mirror of
https://github.com/danog/MadelineProto.git
synced 2024-11-30 07:18:57 +01:00
107 lines
3.1 KiB
PHP
107 lines
3.1 KiB
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 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
|
|
{
|
|
foreach (self::$callbacks as $callback) {
|
|
$callback();
|
|
}
|
|
self::$callbacks = [];
|
|
|
|
if (\defined('STDIN')) {
|
|
getStdin()->unreference();
|
|
}
|
|
API::finalize();
|
|
MTProto::serializeAll();
|
|
Logger::finalize();
|
|
if (class_exists(Installer::class)) {
|
|
Installer::unlock();
|
|
}
|
|
}
|
|
/**
|
|
* Register shutdown function.
|
|
*/
|
|
public static function init(): void
|
|
{
|
|
if (!self::$registered) {
|
|
register_shutdown_function(static 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;
|
|
}
|
|
}
|