zhamao-framework/src/ZM/script_setup_loader.php

<?php

use Doctrine\Common\Annotations\AnnotationReader;
use ZM\Annotation\Swoole\OnSetup;
use ZM\Annotation\Swoole\SwooleHandler;
use ZM\ConsoleApplication;
use ZM\Exception\InitException;
use ZM\Utils\DataProvider;
use ZM\Utils\ZMUtil;

require_once ((!is_dir(__DIR__ . '/../../vendor')) ? getcwd() : (__DIR__ . "/../..")) . "/vendor/autoload.php";

try {
    try {
        (new ConsoleApplication('zhamao'))->initEnv();
    } catch (InitException $e) {

    }
    $base_path = DataProvider::getSourceRootDir();
    $scan_paths = [];
    $composer = json_decode(file_get_contents($base_path . "/composer.json"), true);
    foreach (($composer["autoload"]["psr-4"] ?? []) as $k => $v) {
        if (is_dir($base_path . "/" . $v) && !in_array($v, $composer["extra"]["exclude_annotate"] ?? [])) {
            $scan_paths[trim($k, "\\")] = $base_path . "/" . $v;
        }
    }
    $all_event_class = [];
    foreach ($scan_paths as $namespace => $autoload_path) {
        $all_event_class = array_merge($all_event_class, ZMUtil::getClassesPsr4($autoload_path, $namespace));
    }

    $reader = new AnnotationReader();
    $event_list = [];
    $setup_list = [];
    foreach ($all_event_class as $v) {
        $reflection_class = new ReflectionClass($v);
        $methods = $reflection_class->getMethods(ReflectionMethod::IS_PUBLIC);
        foreach ($methods as $vs) {
            $method_annotations = $reader->getMethodAnnotations($vs);
            if ($method_annotations != []) {
                $annotation = $method_annotations[0];
                if ($annotation instanceof SwooleHandler) {
                    $event_list[] = [
                        "class" => $v,
                        "method" => $vs->getName(),
                        "event" => $annotation->event
                    ];
                } elseif ($annotation instanceof OnSetup) {
                    $setup_list[] = [
                        "class" => $v,
                        "method" => $vs->getName()
                    ];
                }
            }
        }
    }
    echo json_encode(["setup" => $setup_list, "event" => $event_list]);
} catch (Throwable $e) {
    $stderr = fopen("php://stderr", "w");
    fwrite($stderr, zm_internal_errcode("E00031") . $e->getMessage() . " in " . $e->getFile() . " at line " . $e->getLine() . PHP_EOL);
    fclose($stderr);
    exit(1);
}