MadelineProto/src/Shutdown.php

<?php

declare(strict_types=1);

/**
 * Shutdown module.
 *
 * 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-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;

use ReflectionClass;
use Revolt\EventLoop;
use Revolt\EventLoop\Internal\AbstractDriver;

use function Amp\ByteStream\getStdin;

/**
 * Class that controls script shutdown.
 */
final class Shutdown
{
    /**
     * Callbacks to call on shutdown.
     *
     * @var array<callable>
     */
    private static array $callbacks = [];
    /**
     * Whether the main shutdown was registered.
     *
     */
    private static bool $registered = false;
    /**
     * Incremental ID for new callback.
     *
     */
    private static int $id = 0;
    /**
     * Function to be called on shutdown.
     */
    private static function shutdown(): void
    {
        $obj = EventLoop::getSuspension();
        $reflection = new ReflectionClass($obj);
        $reflection->getProperty('pending')->setValue($obj, false);
        $obj = EventLoop::getDriver();
        $reflection = new ReflectionClass(AbstractDriver::class);
        if (!$reflection->getProperty('callbackFiber')->isInitialized($obj)
            || $reflection->getProperty('callbackFiber')->getValue($obj)->isTerminated()
        ) {
            $reflection->getMethod('createCallbackFiber')->invoke($obj);
        }

        foreach (self::$callbacks as $callback) {
            $callback();
        }
        self::$callbacks = [];

        if (\defined('STDIN')) {
            getStdin()->unreference();
        }
        API::finalize();
        MTProto::serializeAll();
        if (\class_exists(Installer::class)) {
            Installer::unlock();
        }
    }
    /**
     * Register shutdown function.
     */
    public static function init(): void
    {
        if (!self::$registered) {
            \register_shutdown_function(fn () => self::shutdown());
            self::$registered = true;
        }
    }
    /**
     * Add a callback for script shutdown.
     *
     * @param callable $callback The callback to set
     * @param null|string $id The optional callback ID
     * @return int|string The callback ID
     */
    public static function addCallback(callable $callback, ?string $id = null): int|string
    {
        if (!$id) {
            $id = self::$id++;
        }
        self::$callbacks[$id] = $callback;
        self::init();
        return $id;
    }
    /**
     * Remove a callback from the script shutdown callable list.
     *
     * @param null|string|int $id The optional callback ID
     * @return bool true if the callback was removed correctly, false otherwise
     */
    public static function removeCallback(string|int|null $id): bool
    {
        if (isset(self::$callbacks[$id])) {
            unset(self::$callbacks[$id]);
            return true;
        }
        return false;
    }
}