1
0
mirror of https://github.com/danog/MadelineProto.git synced 2024-12-12 15:47:21 +01:00
MadelineProto/src/MTProto/MTProtoOutgoingMessage.php

439 lines
11 KiB
PHP
Raw Normal View History

2022-12-30 21:54:44 +01:00
<?php
declare(strict_types=1);
/**
* Outgoing message.
*
* This file is part of MadelineProto.
* MadelineProto is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
* MadelineProto is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU Affero General Public License for more details.
* You should have received a copy of the GNU General Public License along with MadelineProto.
* If not, see <http://www.gnu.org/licenses/>.
*
* @author Daniil Gentili <daniil@daniil.it>
2023-01-04 12:43:01 +01:00
* @copyright 2016-2023 Daniil Gentili <daniil@daniil.it>
* @license https://opensource.org/licenses/AGPL-3.0 AGPLv3
* @link https://docs.madelineproto.xyz MadelineProto documentation
*/
namespace danog\MadelineProto\MTProto;
2023-06-02 20:49:50 +02:00
use Amp\Cancellation;
use Amp\CancelledException;
2022-12-30 20:24:13 +01:00
use Amp\DeferredFuture;
2022-12-30 21:43:58 +01:00
use Amp\Future;
use danog\MadelineProto\Exception;
2022-12-30 21:43:58 +01:00
use Revolt\EventLoop;
2023-01-08 16:06:50 +01:00
use Throwable;
2022-12-30 19:21:36 +01:00
use function time;
/**
* Outgoing message.
*
* @internal
*/
2023-07-05 21:28:17 +02:00
class MTProtoOutgoingMessage extends MTProtoMessage
{
/**
* The message was created.
*/
const STATE_PENDING = 0;
/**
* The message was sent.
*/
const STATE_SENT = 1;
/**
* The message was acked.
*/
const STATE_ACKED = 2;
/**
* We got a reply to the message.
*/
const STATE_REPLIED = self::STATE_ACKED | 4;
/**
* State of message.
*
2023-01-13 14:36:10 +01:00
* @var self::STATE_*
*/
private int $state = self::STATE_PENDING;
/**
* Send deferred.
2023-01-11 18:47:27 +01:00
*
* @var ?DeferredFuture<null>
*/
2023-01-20 17:24:13 +01:00
private ?DeferredFuture $sendDeferred = null;
/**
* Message body.
*
2023-01-08 19:02:49 +01:00
* @var array|(callable(): array)|null
*/
2023-01-08 19:02:49 +01:00
private $body = null;
/**
* Serialized body.
*/
private ?string $serializedBody = null;
/**
* Whether we should refresh references upon serialization of this message.
*/
private bool $refreshReferences = false;
/**
* When was this message sent.
*/
2023-01-14 19:51:23 +01:00
private ?int $sent = null;
/**
* Number of times this message was sent.
*/
private int $tries = 0;
2023-06-02 20:49:50 +02:00
private ?Cancellation $cancellation = null;
/**
* Whether this message is related to a user, as in getting a successful reply means we have auth.
*/
public readonly bool $userRelated;
/**
2023-09-28 17:28:40 +02:00
* Previous queued message.
*/
2023-09-28 17:28:40 +02:00
private ?self $previousQueuedMessage = null;
/**
* Create outgoing message.
*
2023-09-06 21:12:19 +02:00
* @param array|callable(): array $body Body
2023-01-08 19:02:49 +01:00
* @param string $constructor Constructor name
* @param string $type Constructor type
* @param boolean $isMethod Is this a method?
2023-01-08 19:02:49 +01:00
* @param boolean $unencrypted Is this an unencrypted message?
*/
public function __construct(
array|callable $body,
public readonly string $constructor,
public readonly string $type,
public readonly bool $isMethod,
public readonly bool $unencrypted,
2023-09-27 15:24:59 +02:00
public readonly ?string $subtype = null,
/**
* Whether this message is related to a file upload, as in getting a redirect should redirect to a media server.
*/
public readonly bool $fileRelated = false,
/**
* Queue ID.
*/
public readonly ?string $queueId = null,
/**
* Custom flood wait limit for this message.
*/
public readonly ?int $floodWaitLimit = null,
private ?DeferredFuture $resultDeferred = null,
?Cancellation $cancellation = null
) {
$this->body = $body;
$this->userRelated = $constructor === 'users.getUsers' && $body === ['id' => [['_' => 'inputUserSelf']]] || $constructor === 'auth.exportAuthorization' || $constructor === 'updates.getDifference';
parent::__construct(!isset(MTProtoMessage::NOT_CONTENT_RELATED[$constructor]));
2023-06-02 20:49:50 +02:00
$this->cancellation = $cancellation;
$cancellation?->subscribe(fn (CancelledException $e) => $this->reply(fn () => throw $e));
}
/**
* Whether cancellation is requested.
*/
2023-06-04 15:16:06 +02:00
public function isCancellationRequested(): bool
{
2023-06-02 20:49:50 +02:00
return $this->cancellation?->isRequested() ?? false;
}
/**
* Signal that we're trying to send the message.
*/
public function trySend(): void
{
2023-01-20 17:24:13 +01:00
if (!isset($this->sendDeferred)) {
$this->sendDeferred = new DeferredFuture;
}
$this->tries++;
}
/**
* Signal that the message was sent.
*/
public function sent(): void
{
if ($this->state & self::STATE_REPLIED) {
2021-12-07 17:16:15 +01:00
//throw new Exception("Trying to resend already replied message $this!");
}
$this->state |= self::STATE_SENT;
2023-10-01 20:05:04 +02:00
$this->sent = time();
2023-01-20 17:24:13 +01:00
if (isset($this->sendDeferred)) {
$sendDeferred = $this->sendDeferred;
$this->sendDeferred = null;
$sendDeferred->complete();
}
}
/**
* Set reply to message.
2023-01-20 14:14:29 +01:00
*
* @param mixed|(callable(): Throwable) $result
*/
public function reply($result): void
{
if ($this->state & self::STATE_REPLIED) {
2023-04-28 18:58:46 +02:00
//throw new Exception("Trying to double reply to message $this!");
// It can happen, no big deal
return;
}
$this->serializedBody = null;
$this->body = null;
$this->state |= self::STATE_REPLIED;
2023-01-20 17:24:13 +01:00
if ($this->resultDeferred) { // Sometimes can get an RPC error for constructors
$promise = $this->resultDeferred;
$this->resultDeferred = null;
EventLoop::queue($promise->complete(...), $result);
}
}
/**
* ACK message.
*/
public function ack(): void
{
$this->state |= self::STATE_ACKED;
}
/**
* Get state of message.
*
2023-01-04 16:04:05 +01:00
* @return self::STATE_*
*/
public function getState(): int
{
return $this->state;
}
/**
* Get message body.
*/
public function getBody()
{
2023-01-08 19:27:29 +01:00
return \is_callable($this->body) ? ($this->body)() : $this->body;
}
/**
* Get message body or empty array.
*/
public function getBodyOrEmpty(): array
{
return \is_array($this->body) ? $this->body : [];
}
/**
* Check if we have a body.
*/
public function hasBody(): bool
{
return $this->body !== null;
}
/**
* Get serialized body.
*/
public function getSerializedBody(): ?string
{
return $this->serializedBody;
}
/**
* Check if we have a serialized body.
*/
public function hasSerializedBody(): bool
{
return $this->serializedBody !== null;
}
/**
* Get number of times this message was sent.
*/
public function getTries(): int
{
return $this->tries;
}
/**
* Get constructor name.
*/
public function getConstructor(): string
{
return $this->constructor;
}
/**
* Get whether we should refresh references upon serialization of this message.
*/
public function shouldRefreshReferences(): bool
{
return $this->refreshReferences;
}
/**
* Set serialized body.
*
* @param string $serializedBody Serialized body.
*/
public function setSerializedBody(string $serializedBody): self
{
$this->serializedBody = $serializedBody;
return $this;
}
/**
* Set whether we should refresh references upon serialization of this message.
*
* @param bool $refreshReferences Whether we should refresh references upon serialization of this message.
*/
public function setRefreshReferences(bool $refreshReferences): self
{
$this->refreshReferences = $refreshReferences;
return $this;
}
/**
* Get when was this message sent.
*/
2023-01-14 19:51:23 +01:00
public function getSent(): ?int
{
return $this->sent;
}
/**
* Check if the message was sent.
*/
public function wasSent(): bool
{
return (bool) ($this->state & self::STATE_SENT);
}
2023-09-28 17:28:40 +02:00
/**
* Check if the message has a reply.
*/
public function hasReply(): bool
{
return (bool) ($this->state & self::STATE_REPLIED);
}
/**
* Check if can garbage collect this message.
*/
public function canGarbageCollect(): bool
{
if ($this->state & self::STATE_REPLIED) {
return true;
}
if (!$this->hasPromise()) {
return true;
}
return false;
}
/**
* For logging.
*/
2022-12-30 19:21:36 +01:00
public function __toString(): string
{
2023-01-22 20:03:51 +01:00
if ($this->state & self::STATE_REPLIED) {
$state = 'acked (by reply)';
} elseif ($this->state & self::STATE_ACKED) {
$state = 'acked';
} elseif ($this->state & self::STATE_SENT) {
2023-10-01 20:05:04 +02:00
$state = 'sent '.(time() - $this->sent).' seconds ago';
2023-01-22 20:03:51 +01:00
} else {
$state = 'pending';
}
if ($this->msgId) {
return "{$this->constructor} with message ID {$this->msgId} $state";
}
return "{$this->constructor} $state";
}
/**
* Wait for message to be sent.
2023-01-11 18:47:27 +01:00
*
* @return Future<null>
*/
2022-12-30 21:43:58 +01:00
public function getSendPromise(): Future
{
2023-01-20 17:24:13 +01:00
if (!$this->sendDeferred) {
throw new Exception("Message was already sent, can't get send promise!");
}
2023-01-20 17:24:13 +01:00
return $this->sendDeferred->getFuture();
}
/**
* Check if we have a promise.
*/
public function hasPromise(): bool
{
2023-01-20 17:24:13 +01:00
return $this->resultDeferred !== null;
}
/**
* Get the promise.
*/
public function getResultPromise(): Future
{
\assert($this->resultDeferred !== null);
return $this->resultDeferred->getFuture();
}
/**
* Reset sent time to trigger resending.
*/
public function resetSent(): self
{
$this->sent = 0;
return $this;
}
/**
* Set when was this message sent.
*
* @param int $sent When was this message sent.
*/
public function setSent(int $sent): self
{
$this->sent = $sent;
return $this;
}
/**
2023-09-28 17:28:40 +02:00
* Get previous queued message.
*
2023-09-28 17:28:40 +02:00
* @return ?self
*/
2023-09-28 17:28:40 +02:00
public function getPreviousQueuedMessage(): ?self
{
2023-09-28 17:28:40 +02:00
return $this->previousQueuedMessage;
}
/**
2023-09-28 17:28:40 +02:00
* Set previous queued message.
*
2023-09-28 17:28:40 +02:00
* @param ?self $previousQueuedMessage Previous queued message.
*
*/
2023-09-28 17:28:40 +02:00
public function setPreviousQueuedMessage(?self $previousQueuedMessage): self
{
2023-09-28 17:28:40 +02:00
$this->previousQueuedMessage = $previousQueuedMessage;
return $this;
}
}