zhamao-framework/src/ZM/Utils/Terminal.php

<?php /** @noinspection PhpUnused */


namespace ZM\Utils;


use Doctrine\Common\Annotations\AnnotationReader;
use Exception;
use ReflectionClass;
use Swoole\Process;
use ZM\Annotation\Command\TerminalCommand;
use ZM\ConnectionManager\ManagerGM;
use ZM\Console\Console;
use ZM\Event\EventDispatcher;
use ZM\Event\EventManager;

class Terminal
{
    public static $default_commands = false;

    /**
     * @param string $cmd
     * @return bool
     * @noinspection PhpMissingReturnTypeInspection
     * @noinspection PhpUnused
     * @throws Exception
     */
    public static function executeCommand(string $cmd) {
        if (self::$default_commands === false) {
            self::init();
        }
        $it = explodeMsg($cmd);
        $dispatcher = new EventDispatcher(TerminalCommand::class);
        $dispatcher->setRuleFunction(function ($v) use ($it) {
            /** @var TerminalCommand $v */
            return !empty($it) && ($v->command == $it[0] || $v->alias == $it[0]);
        });
        $dispatcher->setReturnFunction(function () {
            EventDispatcher::interrupt('none');
        });
        $dispatcher->dispatchEvents($it);
        if ($dispatcher->store !== 'none' && $cmd !== "") {
            Console::info("Command not found: " . $cmd);
            return true;
        }
        return false;
    }

    public static function log($type, $log_msg) {
        ob_start();
        if (!in_array($type, ["log", "info", "debug", "success", "warning", "error", "verbose"])) {
            ob_get_clean();
            return;
        }
        Console::$type($log_msg);
        $r = ob_get_clean();
        $all = ManagerGM::getAllByName("terminal");
        foreach ($all as $v) {
            server()->send($v->getFd(), "\r" . $r);
            server()->send($v->getFd(), ">>> ");
        }
    }

    public static function init() {
        Console::debug("Initializing Terminal...");
        foreach ((EventManager::$events[TerminalCommand::class] ?? []) as $v) {
            if ($v->command == "help") {
                self::$default_commands = true;
                break;
            }
        }
        $class = new Terminal();
        $reader = new AnnotationReader();
        $reflection = new ReflectionClass($class);
        foreach ($reflection->getMethods() as $v) {
            $r = $reader->getMethodAnnotation($v, TerminalCommand::class);
            if ($r !== null) {
                Console::debug("adding command " . $r->command);
                $r->class = Terminal::class;
                $r->method = $v->getName();
                EventManager::addEvent(TerminalCommand::class, $r);
            }
        }
        self::$default_commands = true;
    }

    /**
     * @TerminalCommand(command="help",alias="h",description="显示帮助菜单")
     */
    public function help() {
        $help = [];
        foreach ((EventManager::$events[TerminalCommand::class] ?? []) as $v) {
            /** @var TerminalCommand $v */
            $cmd = $v->command . ($v->alias !== "" ? (" | " . $v->alias) : "");
            $painted = Console::setColor($v->command, "green") . ($v->alias !== "" ? (" | " . Console::setColor($v->alias, "green")) : "");
            $help[] = $painted . ":" . str_pad("", 16 - strlen($cmd) - 1) . ($v->description === "" ? "<无描述>" : $v->description);
        }
        echo implode("\n", $help) . PHP_EOL;
    }

    /**
     * @TerminalCommand(command="status",description="显示Swoole Server运行状态（需要安装league/climate组件）")
     * @noinspection PhpFullyQualifiedNameUsageInspection
     */
    public function status() {
        if (!class_exists("\League\CLImate\CLImate")) {
            Console::warning("你还没有安装 league/climate 组件，无法使用此功能！");
            return;
        }
        $climate = new \League\CLImate\CLImate;
        $climate->output->addDefault('buffer');

        $objs = server()->stats();
        $climate->columns($objs);
        $obj = $climate->output->get('buffer')->get();
        $climate->output->get("buffer")->clean();
        echo $obj;
    }

    /**
     * @TerminalCommand(command="logtest",description="测试log的显示等级")
     */
    public function testlog() {
        Console::log(date("[H:i:s]") . " [L] This is normal msg. (0)");
        Console::error("This is error msg. (0)");
        Console::warning("This is warning msg. (1)");
        Console::info("This is info msg. (2)");
        Console::success("This is success msg. (2)");
        Console::verbose("This is verbose msg. (3)");
        Console::debug("This is debug msg. (4)");
    }

    /**
     * @TerminalCommand(command="call",description="用于执行不需要参数的动态函数，比如 `call \Module\Example\Hello hitokoto`")
     * @param $it
     */
    public function call($it) {
        $class_name = $it[1];
        $function_name = $it[2];
        $class = new $class_name([]);
        $r = $class->$function_name();
        if (is_string($r)) Console::success($r);
    }

    /**
     * @TerminalCommand(command="level",description="设置log等级，例如 `level 0|1|2|3|4`")
     * @param $it
     */
    public function level($it) {
        $level = intval(is_numeric($it[1] ?? 99) ? ($it[1] ?? 99) : 99);
        if ($level > 4 || $level < 0) Console::warning("Usage: 'level 0|1|2|3|4'");
        else Console::setLevel($level) || Console::success("Success!!");
    }

    /**
     * @TerminalCommand(command="bc",description="eval执行代码，但输入必须是将代码base64之后的，如 `bc em1faW5mbygn5L2g5aW9Jyk7`")
     * @param $it
     */
    public function bc($it) {
        $code = base64_decode($it[1] ?? '', true);
        try {
            eval($code);
        } catch (Exception $e) {
        }
    }

    /**
     * @TerminalCommand(command="echo",description="输出内容，用法：`echo hello`")
     * @param $it
     */
    public function echoI($it) {
        Console::info($it[1]);
    }

    /**
     * @TerminalCommand(command="stop",description="停止框架")
     */
    public function stop() {
        posix_kill(server()->master_pid, SIGTERM);
    }

    /**
     * @TerminalCommand(command="reload",alias="r",description="重启框架（重载用户代码）")
     */
    public function reload() {
        Process::kill(server()->master_pid, SIGUSR1);
    }
}