202 lines
11 KiB
PHP
202 lines
11 KiB
PHP
<?php
|
|
|
|
/**
|
|
* Update loop.
|
|
*
|
|
* 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>
|
|
* @copyright 2016-2020 Daniil Gentili <daniil@daniil.it>
|
|
* @license https://opensource.org/licenses/AGPL-3.0 AGPLv3
|
|
*
|
|
* @link https://docs.madelineproto.xyz MadelineProto documentation
|
|
*/
|
|
|
|
namespace danog\MadelineProto\Loop\Update;
|
|
|
|
use Amp\Loop;
|
|
use danog\MadelineProto\Exception;
|
|
use danog\MadelineProto\Loop\Impl\ResumableSignalLoop;
|
|
use danog\MadelineProto\RPCErrorException;
|
|
|
|
/**
|
|
* Update loop.
|
|
*
|
|
* @author Daniil Gentili <daniil@daniil.it>
|
|
*/
|
|
class UpdateLoop extends ResumableSignalLoop
|
|
{
|
|
private $toPts;
|
|
private $channelId;
|
|
private $feeder;
|
|
public function __construct($API, $channelId)
|
|
{
|
|
$this->API = $API;
|
|
$this->channelId = $channelId;
|
|
}
|
|
public function loop(): \Generator
|
|
{
|
|
$API = $this->API;
|
|
$feeder = $this->feeder = $API->feeders[$this->channelId];
|
|
while (!$API->settings['updates']['handle_updates'] || !$API->hasAllAuth()) {
|
|
if (yield $this->waitSignal($this->pause())) {
|
|
$API->logger->logger("Exiting {$this} due to signal");
|
|
return;
|
|
}
|
|
}
|
|
$this->state = $state = $this->channelId === false ? yield from $API->loadUpdateState() : $API->loadChannelState($this->channelId);
|
|
$timeout = $API->settings['updates']['getdifference_interval'];
|
|
$first = true;
|
|
while (true) {
|
|
while (!$API->settings['updates']['handle_updates'] || !$API->hasAllAuth()) {
|
|
if (yield $this->waitSignal($this->pause())) {
|
|
$API->logger->logger("Exiting {$this} due to signal");
|
|
return;
|
|
}
|
|
}
|
|
$result = [];
|
|
$toPts = $this->toPts;
|
|
$this->toPts = null;
|
|
while (true) {
|
|
if ($this->channelId) {
|
|
$API->logger->logger('Resumed and fetching '.$this->channelId.' difference...', \danog\MadelineProto\Logger::ULTRA_VERBOSE);
|
|
if ($state->pts() <= 1) {
|
|
$limit = 10;
|
|
} elseif ($API->authorization['user']['bot']) {
|
|
$limit = 100000;
|
|
} else {
|
|
$limit = 100;
|
|
}
|
|
$request_pts = $state->pts();
|
|
try {
|
|
$difference = yield from $API->methodCallAsyncRead('updates.getChannelDifference', ['channel' => 'channel#'.$this->channelId, 'filter' => ['_' => 'channelMessagesFilterEmpty'], 'pts' => $request_pts, 'limit' => $limit, 'force' => true], ['datacenter' => $API->datacenter->curdc, 'postpone' => $first]);
|
|
} catch (RPCErrorException $e) {
|
|
if (\in_array($e->rpc, ['CHANNEL_PRIVATE', 'CHAT_FORBIDDEN', 'CHANNEL_INVALID'])) {
|
|
$feeder->signal(true);
|
|
unset($API->updaters[$this->channelId], $API->feeders[$this->channelId]);
|
|
$API->getChannelStates()->remove($this->channelId);
|
|
$API->logger->logger("Channel private, exiting {$this}");
|
|
return true;
|
|
}
|
|
throw $e;
|
|
} catch (Exception $e) {
|
|
if (\in_array($e->getMessage(), ['This peer is not present in the internal peer database'])) {
|
|
$feeder->signal(true);
|
|
$API->getChannelStates()->remove($this->channelId);
|
|
unset($API->updaters[$this->channelId], $API->feeders[$this->channelId]);
|
|
$API->logger->logger("Channel private, exiting {$this}");
|
|
return true;
|
|
}
|
|
throw $e;
|
|
}
|
|
if (isset($difference['timeout'])) {
|
|
$timeout = $difference['timeout'];
|
|
}
|
|
$API->logger->logger('Got '.$difference['_'], \danog\MadelineProto\Logger::VERBOSE);
|
|
switch ($difference['_']) {
|
|
case 'updates.channelDifferenceEmpty':
|
|
$state->update($difference);
|
|
unset($difference);
|
|
break 2;
|
|
case 'updates.channelDifference':
|
|
if ($request_pts >= $difference['pts'] && $request_pts > 1) {
|
|
$API->logger->logger("The PTS ({$difference['pts']}) I got with getDifference is smaller than the PTS I requested ".$state->pts().', using '.($state->pts() + 1), \danog\MadelineProto\Logger::VERBOSE);
|
|
$difference['pts'] = $request_pts + 1;
|
|
}
|
|
$result += (yield from $feeder->feed($difference['other_updates']));
|
|
$state->update($difference);
|
|
$feeder->saveMessages($difference['new_messages']);
|
|
if (!$difference['final']) {
|
|
if ($difference['pts'] >= $toPts) {
|
|
unset($difference);
|
|
break 2;
|
|
}
|
|
unset($difference);
|
|
break;
|
|
}
|
|
unset($difference);
|
|
break 2;
|
|
case 'updates.channelDifferenceTooLong':
|
|
if (isset($difference['dialog']['pts'])) {
|
|
$difference['pts'] = $difference['dialog']['pts'];
|
|
}
|
|
$state->update($difference);
|
|
$feeder->saveMessages($difference['messages']);
|
|
unset($difference);
|
|
break;
|
|
default:
|
|
throw new \danog\MadelineProto\Exception('Unrecognized update difference received: '.\var_export($difference, true));
|
|
}
|
|
} else {
|
|
$API->logger->logger('Resumed and fetching normal difference...', \danog\MadelineProto\Logger::ULTRA_VERBOSE);
|
|
$difference = yield from $API->methodCallAsyncRead('updates.getDifference', ['pts' => $state->pts(), 'date' => $state->date(), 'qts' => $state->qts()], ['datacenter' => $API->settings['connection_settings']['default_dc']]);
|
|
$API->logger->logger('Got '.$difference['_'], \danog\MadelineProto\Logger::ULTRA_VERBOSE);
|
|
switch ($difference['_']) {
|
|
case 'updates.differenceEmpty':
|
|
$state->update($difference);
|
|
unset($difference);
|
|
break 2;
|
|
case 'updates.difference':
|
|
$state->qts($difference['state']['qts']);
|
|
foreach ($difference['new_encrypted_messages'] as &$encrypted) {
|
|
$encrypted = ['_' => 'updateNewEncryptedMessage', 'message' => $encrypted];
|
|
}
|
|
$result += (yield from $feeder->feed($difference['other_updates']));
|
|
$result += (yield from $feeder->feed($difference['new_encrypted_messages']));
|
|
$state->update($difference['state']);
|
|
$feeder->saveMessages($difference['new_messages']);
|
|
unset($difference);
|
|
break 2;
|
|
case 'updates.differenceSlice':
|
|
$state->qts($difference['intermediate_state']['qts']);
|
|
foreach ($difference['new_encrypted_messages'] as &$encrypted) {
|
|
$encrypted = ['_' => 'updateNewEncryptedMessage', 'message' => $encrypted];
|
|
}
|
|
$result += (yield from $feeder->feed($difference['other_updates']));
|
|
$result += (yield from $feeder->feed($difference['new_encrypted_messages']));
|
|
$state->update($difference['intermediate_state']);
|
|
$feeder->saveMessages($difference['new_messages']);
|
|
if ($difference['intermediate_state']['pts'] >= $toPts) {
|
|
unset($difference);
|
|
break 2;
|
|
}
|
|
unset($difference);
|
|
break;
|
|
case 'updates.differenceTooLong':
|
|
$state->update($difference);
|
|
unset($difference);
|
|
break;
|
|
default:
|
|
throw new \danog\MadelineProto\Exception('Unrecognized update difference received: '.\var_export($difference, true));
|
|
}
|
|
}
|
|
}
|
|
$API->logger->logger("Finished parsing updates in {$this}, now resuming feeders");
|
|
foreach ($result as $channelId => $boh) {
|
|
$API->feeders[$channelId]->resumeDefer();
|
|
}
|
|
$API->logger->logger("Finished resuming feeders in {$this}, signaling updates");
|
|
$API->signalUpdate();
|
|
$API->logger->logger("Finished signaling updates in {$this}, pausing");
|
|
$first = false;
|
|
if (yield $this->waitSignal($this->pause($timeout))) {
|
|
$API->logger->logger("Exiting {$this} due to signal");
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
public function setLimit($toPts)
|
|
{
|
|
$this->toPts = $toPts;
|
|
}
|
|
public function __toString(): string
|
|
{
|
|
return !$this->channelId ? 'getUpdate loop generic' : "getUpdate loop channel {$this->channelId}";
|
|
}
|
|
}
|