initial 2.0 commit

2026-07-21 15:45:36 +08:00 · 2020-08-31 10:11:06 +08:00
parent b6756179f5
commit beb1f5f063
69 changed files with 1237 additions and 2090 deletions
--- a/src/ZM/Utils/Terminal.php
+++ b/src/ZM/Utils/Terminal.php
@@ -0,0 +1,126 @@
+<?php
+
+
+namespace ZM\Utils;
+
+
+use Exception;
+use ZM\Config\ZMConfig;
+use ZM\ConnectionManager\ConnectionObject;
+use ZM\Console\Console;
+use ZM\Store\ZMBuf;
+use ZM\Annotation\Swoole\SwooleEventAt;
+
+class Terminal
+{
+    /**
+     * @var false|resource
+     */
+    public static $console_proc = null;
+    public static $pipes = [];
+
+    static function listenConsole($terminal_id) {
+        if ($terminal_id === null) {
+            if (ZMBuf::$server->worker_id === 0) Console::info("ConsoleCommand disabled.");
+            return;
+        }
+        global $terminal_id;
+        global $port;
+        $port = ZMConfig::get("global", "port");
+        $vss = new SwooleEventAt();
+        $vss->type = "open";
+        $vss->level = 256;
+        $vss->rule = "connectType:terminal";
+
+        $vss->callback = function (?ConnectionObject $conn) use ($terminal_id) {
+            $req = ctx()->getRequest();
+            if ($conn->getName() != "terminal") return false;
+            Console::debug("Terminal fd: " . $conn->getFd());
+            ZMBuf::set("terminal_fd", $conn->getFd());
+            if (($req->header["x-terminal-id"] ?? "") != $terminal_id) {
+                ZMBuf::$server->close($conn->getFd());
+                return false;
+            }
+            return false;
+        };
+        ZMBuf::$events[SwooleEventAt::class][] = $vss;
+        $vss2 = new SwooleEventAt();
+        $vss2->type = "message";
+        $vss2->rule = "connectType:terminal";
+        $vss2->callback = function (?ConnectionObject $conn) {
+            if ($conn === null) return false;
+            if ($conn->getName() != "terminal") return false;
+            $cmd = ctx()->getFrame()->data;
+            self::executeCommand($cmd);
+            return false;
+        };
+        ZMBuf::$events[SwooleEventAt::class][] = $vss2;
+        if (ZMBuf::$server->worker_id === 0) {
+            go(function () {
+                global $terminal_id, $port;
+                $descriptorspec = array(
+                    0 => STDIN,
+                    1 => STDOUT,
+                    2 => STDERR
+                );
+                self::$console_proc = proc_open('php -r \'$terminal_id = "' . $terminal_id . '";$port = ' . $port . ';require "' . __DIR__ . '/terminal_listener.php";\'', $descriptorspec, self::$pipes);
+            });
+        }
+    }
+
+    /**
+     * @param string $cmd
+     * @return bool
+     */
+    private static function executeCommand(string $cmd) {
+        $it = explodeMsg($cmd);
+        switch ($it[0] ?? '') {
+            case 'logtest':
+                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)");
+                return true;
+            case 'call':
+                $class_name = $it[1];
+                $function_name = $it[2];
+                $class = new $class_name([]);
+                call_user_func_array([$class, $function_name], []);
+                return true;
+            case 'bc':
+                $code = base64_decode($it[1] ?? '', true);
+                try {
+                    eval($code);
+                } catch (Exception $e) {
+                }
+                return true;
+            case 'echo':
+                Console::info($it[1]);
+                return true;
+            case 'color':
+                Console::log($it[2], $it[1]);
+                return true;
+            case 'stop':
+                ZMUtil::stop();
+                return false;
+            case 'reload':
+            case 'r':
+                ZMUtil::reload();
+                return false;
+            case 'save':
+                //$origin = ZMBuf::$atomics["info_level"]->get();
+                //ZMBuf::$atomics["info_level"]->set(3);
+                DataProvider::saveBuffer();
+                //ZMBuf::$atomics["info_level"]->set($origin);
+                return true;
+            case '':
+                return true;
+            default:
+                Console::info("Command not found: " . $cmd);
+                return true;
+        }
+    }
+}