@@ -0,0 +1,360 @@
+<?php
+
+namespace app\common\utils;
+
+use app\common\model\UserRoles;
+use GatewayWorker\Lib\Gateway;
+use app\common\utils\WsResponse;
+use app\common\service\MessageNotificationsService;
+use support\Redis;
+
+class WsEvents
+{
+    
+    private $registerAddress = '127.0.0.1:1236';
+    protected $clientId;
+    protected $eventData;
+
+    public function __construct()
+    {
+        Gateway::$registerAddress = $this->registerAddress;
+    }
+
+    /**
+     * 监听消息接收
+     */
+    public function message($client_id, $message)
+    {
+        $this->clientId = $client_id;
+
+        $payload = json_decode($message, true);
+        $event = $payload['event'];
+        $this->eventData = $payload['data'] ?? [];
+        $traceId = $payload['trace_id'] ?? null;
+
+        $method = $this->eventToMethod($event);
+
+        if (method_exists($this, $method)) {
+            try {
+                $this->$method($traceId);
+            } catch (\Throwable $th) {
+                Gateway::sendToClient($client_id, WsResponse::error(
+                    $event,
+                    500,
+                    $th->getMessage(),
+                    $traceId
+                ));
+            }
+        } else {
+            Gateway::sendToClient($client_id, WsResponse::error(
+                $event,
+                404,
+                "Event handler not found: $method",
+                $traceId
+            ));
+        }
+    }
+
+    protected function eventToMethod(string $event): string
+    {
+        $parts = explode('.', $event);
+        $method = array_shift($parts);
+        foreach ($parts as $part) {
+            $method .= ucfirst($part);
+        }
+        return $method;
+    }
+
+
+    // ================= 系统事件 =================
+    /**
+     * 认证
+     */
+    public function auth($client_id, $payload)
+    {
+        try {
+            // 检查来源
+            if (!isset($payload['server']['HTTP_ORIGIN'])) {
+                flg('[x] [TOKEN认证] 未定义HTTP_ORIGIN', false);
+                Gateway::sendToClient($client_id, WsResponse::error(
+                    'system.auth',
+                    1001,
+                    '[x] [TOKEN认证] 未定义HTTP_ORIGIN'
+                ));
+                Gateway::closeClient($client_id);
+                return;
+            }
+
+            // 连接是否满足条件
+            if (getenv('ENVIRONMENT') === 'master' && $payload['server']['HTTP_ORIGIN'] === getenv('MAIN_DOMAIN')) {
+                flg('[x] [TOKEN认证] HTTP_ORIGIN不满足条件', false);
+                Gateway::sendToClient($client_id, WsResponse::error(
+                    'system.auth',
+                    1001,
+                    '[x] [TOKEN认证] HTTP_ORIGIN不满足条件'
+                ));
+                Gateway::closeClient($client_id);
+                return;
+            }
+
+            // 签名检查
+            if (!isset($payload['get']['token'])) {
+                flg('[x] [TOKEN认证] 未携带TOKEN参数', false);
+                Gateway::sendToClient($client_id, WsResponse::error(
+                    'system.auth',
+                    1001,
+                    '[x] [TOKEN认证] 未携带TOKEN参数'
+                ));
+                Gateway::closeClient($client_id);
+                return;
+            }
+
+            // 解析TOKEN
+            $user = (new Jwt())->verifyToken($payload['get']['token']);
+
+            // 绑定用户ID
+            Gateway::bindUid($client_id, $user->uid);
+
+            // 绑定SESSION
+            $_SESSION['uid'] = $user->uid;
+            // $_SESSION['name'] = $user->uname;
+
+            // 响应
+            Gateway::sendToClient($client_id, WsResponse::success(
+                'system.auth',
+                ['uid' => $user->uid]
+            ));
+        } catch (\Throwable $th) {
+            flg('[x] [TOKEN认证] TOKEN认证失败：'.$th->getMessage(), false);
+            Gateway::sendToClient($client_id, WsResponse::error(
+                'system.auth',
+                1001,
+                '[x] [TOKEN认证] TOKEN认证失败：'.$th->getMessage()
+            ));
+            Gateway::closeClient($client_id);
+            return;
+        }
+    }
+
+    public function logout(string $client_id)
+    {
+        // var_dump('====== ====== ====== start ====== ====== ======');
+        // var_dump('关闭客户端ID：'.$client_id);
+        // if ($_SESSION !== null) {
+        //     var_dump($_SESSION);
+        // }
+        // var_dump('====== ====== ====== end ====== ====== ======');
+    }
+
+    /**
+     * 系统心跳
+     *
+     * @param string|null $traceId
+     */
+    public function systemPing(?string $traceId)
+    {
+        Gateway::sendToClient($this->clientId, WsResponse::success(
+            'system.pong',
+            ['time' => time()],
+            $traceId
+        ));
+    }
+
+
+    public function systemConnect(?string $traceId)
+    {
+        // 响应
+        Gateway::sendToClient($this->clientId, WsResponse::success(
+            'system.connect.receive',
+            ['time' => time()],
+            $traceId
+        ));
+
+        // 获取审批待办数量
+        $num = (new \app\common\utils\approval\service\ApprovalFlowService())->getInitTodoCount($_SESSION['uid']);
+        $this->todoCounts($_SESSION['uid'], $num);
+
+        // 绑定检查
+        $initData = $this->initCheck($_SESSION['uid']);
+        Gateway::sendToClient($this->clientId, WsResponse::success(
+            'system.init.check',
+            $initData,
+            $traceId
+        ));
+    }
+
+    // ================= 消息通知事件 =================
+    /**
+     * 管理端主动推送通知
+     */
+    public function notifyPush(?string $traceId)
+    {
+        $title = $this->eventData['title'] ?? null;
+        $content = $this->eventData['content'] ?? null;
+        $toUids = $this->eventData['to_uids'] ?? [];
+
+        if (!$title || !$content) {
+            Gateway::sendToClient($this->clientId, WsResponse::error(
+                'notify.push.result',
+                400,
+                '标题和内容不能为空',
+                $traceId
+            ));
+            return;
+        }
+
+        // 写消息
+        $msgService = new MessageNotificationsService();
+        $msgId = $msgService->createMessage([
+            'sender_id'   => $this->eventData['sender_id'] ?? 0,
+            'sender_type' => $this->eventData['sender_type'] ?? 'system',
+            'title'       => $title,
+            'content'     => $content,
+            'scene'       => $this->eventData['scene'] ?? 0,
+            'recipient'   => $this->eventData['to_uids'] ? implode(',', $this->eventData['to_uids']) : '',
+            'push_method' => isset($this->eventData['push_method']) ? $this->eventData['push_method'] : 0
+        ]);
+
+        // 写用户状态
+        $uids = !empty($toUids) ? $toUids : $msgService->getAllUserIds();
+        $msgService->addUserStatus($msgId, $uids);
+
+        $payload = WsResponse::success('notify.receive', [
+            'msg_id' => $msgId,
+            'title' => $title,
+            'content' => $content,
+            'created_at' => time()
+        ], $traceId);
+
+        if (!empty($toUids)) {
+            // 推送给指定用户
+            foreach ($toUids as $uid) {
+                Gateway::sendToUid($uid, $payload);
+            }
+        } else {
+            // 推送给全体在线用户
+            Gateway::sendToAll($payload);
+        }
+
+        // 回复发送者确认
+        Gateway::sendToClient($this->clientId, WsResponse::success(
+            'notify.push.result',
+            ['msg_id' => $msgId],
+            $traceId,
+            '消息发送成功'
+        ));
+    }
+
+    /**
+     * 客户端标记通知已读
+     */
+    public function notifyRead(?string $traceId)
+    {
+        $msgIds = $this->eventData['msg_ids'];
+        if (!is_array($msgIds) && $msgIds !== 'all') {
+            Gateway::sendToClient($this->clientId, WsResponse::error(
+                'notify.read.result',
+                400,
+                '消息ID不能为空',
+                $traceId
+            ));
+            return;
+        }
+
+        // 更新消息为已读
+        $msgService = new MessageNotificationsService();
+        $msgService->markAsRead($msgIds, Gateway::getUidByClientId($this->clientId));
+
+        Gateway::sendToClient($this->clientId, WsResponse::success(
+            'notify.read.result',
+            ['msg_ids' => $msgIds],
+            $traceId
+        ));
+    }
+
+
+    /*
+        approval.todo.counts 审批待办数量
+
+     */
+
+    // ================= 主动推送 =================
+
+    public function pushWebsocket(string $event, int|string $uid, array $data)
+    {
+        return match (gettype($uid)) {
+            'integer' => Gateway::sendToUid($uid, WsResponse::success($event, $data)),
+            'string' => Gateway::sendToClient($uid, WsResponse::success($event, $data)),
+            default => throw new \InvalidArgumentException('UID must be integer or string'),
+        };
+    }
+
+
+
+
+
+
+
+
+
+
+
+    // ================= 业务处理 =================
+
+    /**
+     * 初始化检查
+     */
+    public function initCheck(int $userId)
+    {
+        $data = [
+            // 'is_report_to'      => false,
+            // 'is_office_clerk'   => false,
+            'is_sign'           => false
+        ];
+
+        $user = \app\common\model\User::find($userId);
+        // if ($user->report_to > 0) { 
+        //     $data['is_report_to'] = true;
+        // }
+
+        if ($user->sign_img !== null && $user->sign_img !== '') { 
+            $data['is_sign'] = true;
+        }
+
+        // $ug = \app\common\model\UserGroupMembers::with(['group' => function($query) {
+        //     $query->where('scenario_code', 'stock_orders');
+        // }])->where('user_id', $userId)->first();
+        // if ($ug !== null) { 
+        //     $data['is_office_clerk'] = true;
+        // }
+
+        if ($data['is_sign']) {
+            Redis::hSet('user:check:sign'.$userId, 'status', 1);
+        }
+
+        return $data;
+    }
+
+    // 审批待办数量
+    public function todoCounts(int $userId, int $num = 0)
+    {
+        $ly = [
+            1 => 186,
+            3 => 224,
+        ];
+
+        if ($num === 0) {
+            $num = (new \app\common\utils\approval\service\ApprovalFlowService())->getInitTodoCount($userId);
+        }
+        
+        $systemIds = \app\common\model\UserSystems::where('user_id', $userId)->pluck('system_id')->toArray();
+        if (count($systemIds) > 0) {
+            foreach ($systemIds as $systemId) {
+                if (isset($ly[$systemId])) {
+                    $chain = \app\common\service\MenuBadgeService::buildBadgeChain(menuId: $ly[$systemId], leafCount: $num);
+                    $this->pushWebsocket('approval.todo.counts', $userId, $chain);
+                }
+            }
+        } 
+    }
+}