@yu_robotics/remote-cli 1.0.0

package/dist/client/MessageHandler.js

@@ -0,0 +1,496 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MessageHandler = void 0;
+const hooks_1 = require("../hooks");
+const child_process_1 = require("child_process");
+/**
+ * Message Handler
+ * Responsible for handling messages from WebSocket and invoking Claude executor
+ */
+class MessageHandler {
+    wsClient;
+    executor;
+    directoryGuard;
+    isDestroyed = false;
+    isExecuting = false;
+    currentOpenId;
+    notificationAdapter;
+    constructor(wsClient, executor, directoryGuard) {
+        this.wsClient = wsClient;
+        this.executor = executor;
+        this.directoryGuard = directoryGuard;
+        // Initialize Feishu notification adapter
+        this.notificationAdapter = new hooks_1.FeishuNotificationAdapter(wsClient);
+        this.notificationAdapter.register();
+    }
+    /**
+     * Handle message (supports new IncomingMessage format)
+     * @param message Message object
+     */
+    async handleMessage(message) {
+        // Check if already destroyed
+        if (this.isDestroyed) {
+            return;
+        }
+        // Validate message structure
+        if (!message || !this.isValidMessage(message)) {
+            this.sendResponse(message?.messageId || 'unknown', {
+                success: false,
+                error: 'Invalid message format',
+            });
+            return;
+        }
+        // Handle different types of messages
+        switch (message.type) {
+            case 'status':
+                await this.handleStatusQuery(message.messageId);
+                return;
+            case 'command':
+                await this.handleCommandMessage(message);
+                return;
+            case 'heartbeat':
+                // Silently ignore heartbeat responses from server
+                return;
+            case 'binding_confirm':
+                // Silently ignore binding confirmation from server
+                return;
+            default:
+                this.sendResponse(message.messageId, {
+                    success: false,
+                    error: `Unknown message type: ${message.type}`,
+                });
+        }
+    }
+    /**
+     * Handle command message
+     */
+    async handleCommandMessage(message) {
+        const { messageId, content, workingDirectory, openId, isSlashCommand } = message;
+        // Store openId for response routing and notifications
+        this.currentOpenId = openId;
+        this.notificationAdapter.setCurrentOpenId(openId);
+        // Handle /abort command first, even when busy
+        if (content?.trim() === '/abort') {
+            await this.handleAbortCommand(messageId);
+            return;
+        }
+        // Check if executor is waiting for interactive input
+        if ('isWaitingInput' in this.executor && typeof this.executor.isWaitingInput === 'function') {
+            const executor = this.executor;
+            if (executor.isWaitingInput()) {
+                const input = content?.trim();
+                if (input) {
+                    const sent = executor.sendInput(input);
+                    if (sent) {
+                        this.sendResponse(messageId, {
+                            success: true,
+                            output: `✅ Sent: "${input}"`,
+                        });
+                    }
+                    else {
+                        this.sendResponse(messageId, {
+                            success: false,
+                            error: '❌ Failed to send input - executor is no longer waiting',
+                        });
+                    }
+                }
+                else {
+                    this.sendResponse(messageId, {
+                        success: false,
+                        error: '❌ Please provide a non-empty input',
+                    });
+                }
+                return;
+            }
+        }
+        // Check if there is a task currently executing
+        if (this.isExecuting) {
+            this.sendResponse(messageId, {
+                success: false,
+                error: 'Executor is busy, please wait for current task to complete',
+            });
+            return;
+        }
+        // If working directory is provided, validate and set it
+        if (workingDirectory) {
+            // Verify directory is in the whitelist
+            if (!this.directoryGuard.isSafePath(workingDirectory)) {
+                this.sendResponse(messageId, {
+                    success: false,
+                    error: `Directory not in whitelist: ${workingDirectory}\n\nAllowed directories:\n${this.directoryGuard
+                        .getAllowedDirectories()
+                        .map((d) => `• ${d}`)
+                        .join('\n')}`,
+                });
+                return;
+            }
+            // Set working directory
+            this.executor.setWorkingDirectory(workingDirectory);
+        }
+        try {
+            this.isExecuting = true;
+            // Handle built-in commands (except /abort which was handled above)
+            const builtInResult = await this.handleBuiltInCommand(messageId, content);
+            if (builtInResult) {
+                return;
+            }
+            // Check if this is a passthrough slash command from server
+            if (isSlashCommand) {
+                console.log(`[MessageHandler] Executing passthrough slash command: ${content}`);
+                await this.executeSlashCommand(messageId, content);
+                return;
+            }
+            // Expand command shortcuts
+            const expandedContent = this.expandCommandShortcuts(content);
+            // Execute Claude command
+            await this.executeCommand(messageId, expandedContent);
+        }
+        catch (error) {
+            this.sendResponse(messageId, {
+                success: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+            });
+        }
+        finally {
+            this.isExecuting = false;
+        }
+    }
+    /**
+     * Handle abort command
+     * Can be executed even when executor is busy
+     */
+    async handleAbortCommand(messageId) {
+        const wasExecuting = this.isExecuting;
+        const aborted = await this.executor.abort();
+        if (aborted) {
+            this.isExecuting = false;
+            this.sendResponse(messageId, {
+                success: true,
+                output: wasExecuting
+                    ? '✅ Current command has been aborted'
+                    : '⚠️ No command was executing, but executor has been reset',
+            });
+        }
+        else {
+            this.sendResponse(messageId, {
+                success: true,
+                output: 'ℹ️ No command is currently executing',
+            });
+        }
+    }
+    /**
+     * Handle status query
+     */
+    async handleStatusQuery(messageId) {
+        this.wsClient.send({
+            type: 'status',
+            messageId,
+            status: {
+                connected: this.wsClient.isConnected(),
+                allowedDirectories: this.directoryGuard.getAllowedDirectories(),
+                currentWorkingDirectory: this.executor.getCurrentWorkingDirectory(),
+            },
+            timestamp: Date.now(),
+        });
+    }
+    /**
+     * Validate message structure
+     */
+    isValidMessage(message) {
+        if (!message || typeof message !== 'object') {
+            return false;
+        }
+        if (message.type !== 'command') {
+            return true; // Non-command messages don't need further validation
+        }
+        return Boolean(message.messageId && message.content);
+    }
+    /**
+     * Handle built-in commands
+     * @returns Returns true if built-in command was handled, otherwise false
+     */
+    async handleBuiltInCommand(messageId, content) {
+        const trimmed = content.trim();
+        // /status command
+        if (trimmed === '/status') {
+            const cwd = this.executor.getCurrentWorkingDirectory();
+            const allowedDirs = this.directoryGuard.getAllowedDirectories();
+            this.sendResponse(messageId, {
+                success: true,
+                output: `📊 Status:
+- Working Directory: ${cwd}
+- Allowed Directories: ${allowedDirs.join(', ')}
+- Connection: Active`,
+            });
+            return true;
+        }
+        // /help command
+        if (trimmed === '/help') {
+            this.sendResponse(messageId, {
+                success: true,
+                output: `📖 Available commands:
+- /help - Show this help message
+- /status - Show current status
+- /abort - Abort the currently executing command
+- /clear - Clear conversation context
+- /cd <directory> - Change working directory
+- /r or /resume - Resume previous conversation
+- /c or /continue - Continue previous conversation
+- /review - Review changes (supports remote interaction)
+
+💡 Interactive commands like /review will prompt you for input via Feishu when needed.
+
+You can also use natural language commands to control Claude Code CLI.`,
+            });
+            return true;
+        }
+        // /clear command
+        if (trimmed === '/clear') {
+            this.executor.resetContext();
+            this.sendResponse(messageId, {
+                success: true,
+                output: '✅ Conversation context cleared',
+            });
+            return true;
+        }
+        // /cd command
+        if (trimmed.startsWith('/cd')) {
+            const parts = trimmed.split(/\s+/);
+            if (parts.length < 2) {
+                this.sendResponse(messageId, {
+                    success: false,
+                    error: 'Usage: /cd <directory>',
+                });
+                return true;
+            }
+            const targetDir = parts.slice(1).join(' ');
+            try {
+                this.executor.setWorkingDirectory(targetDir);
+                const newCwd = this.executor.getCurrentWorkingDirectory();
+                this.sendResponse(messageId, {
+                    success: true,
+                    output: `✅ Changed working directory to: ${newCwd}`,
+                });
+            }
+            catch (error) {
+                this.sendResponse(messageId, {
+                    success: false,
+                    error: error instanceof Error
+                        ? error.message
+                        : 'Failed to change directory',
+                });
+            }
+            return true;
+        }
+        return false;
+    }
+    /**
+     * Expand command shortcuts
+     */
+    expandCommandShortcuts(content) {
+        const trimmed = content.trim();
+        // Only expand when command is the entire content
+        if (trimmed === '/r' || trimmed === '/resume') {
+            return 'Please resume the previous conversation';
+        }
+        if (trimmed === '/c' || trimmed === '/continue') {
+            return 'Please continue from where we left off';
+        }
+        return content;
+    }
+    /**
+     * Execute passthrough slash command using local Claude CLI
+     * This allows users to use their custom slash commands
+     */
+    async executeSlashCommand(messageId, command) {
+        return new Promise((resolve) => {
+            const chunks = [];
+            const errorChunks = [];
+            console.log(`[MessageHandler] Spawning Claude CLI for command: ${command}`);
+            // Spawn Claude CLI with the slash command
+            // Use --print to get output and exit
+            const child = (0, child_process_1.spawn)('claude', [command, '--print'], {
+                cwd: this.executor.getCurrentWorkingDirectory(),
+                stdio: ['ignore', 'pipe', 'pipe'],
+                env: {
+                    ...process.env,
+                    CLAUDECODE: '', // Prevent nested session error
+                },
+            });
+            // Handle stdout (stream chunks)
+            child.stdout?.on('data', (data) => {
+                const chunk = data.toString();
+                chunks.push(chunk);
+                this.sendStreamChunk(messageId, chunk);
+            });
+            // Handle stderr
+            child.stderr?.on('data', (data) => {
+                const chunk = data.toString();
+                errorChunks.push(chunk);
+                console.error(`[MessageHandler] Claude stderr: ${chunk}`);
+            });
+            // Handle process exit
+            child.on('exit', (code) => {
+                console.log(`[MessageHandler] Claude process exited with code: ${code}`);
+                if (code === 0) {
+                    const output = chunks.join('');
+                    this.sendResponse(messageId, {
+                        success: true,
+                        output: output.trim() || '✅ Command executed successfully',
+                    });
+                }
+                else {
+                    const errorOutput = errorChunks.join('') || chunks.join('');
+                    this.sendResponse(messageId, {
+                        success: false,
+                        error: errorOutput.trim() || `Command failed with exit code ${code}`,
+                    });
+                }
+                resolve();
+            });
+            // Handle process error
+            child.on('error', (error) => {
+                console.error(`[MessageHandler] Failed to spawn Claude:`, error);
+                this.sendResponse(messageId, {
+                    success: false,
+                    error: `Failed to execute command: ${error.message}`,
+                });
+                resolve();
+            });
+        });
+    }
+    /**
+     * Execute Claude command
+     */
+    async executeCommand(messageId, content) {
+        try {
+            const result = await this.executor.execute(content, {
+                onStream: (chunk) => {
+                    this.sendStreamChunk(messageId, chunk);
+                },
+                onToolUse: (toolUse) => {
+                    this.sendToolUse(messageId, toolUse);
+                },
+                onToolResult: (toolResult) => {
+                    this.sendToolResult(messageId, toolResult);
+                },
+            });
+            // Only send success status, not the output
+            // Output has already been streamed via onStream callback
+            this.sendResponse(messageId, {
+                success: result.success,
+                error: result.error,
+            });
+        }
+        catch (error) {
+            this.sendResponse(messageId, {
+                success: false,
+                error: error instanceof Error ? error.message : 'Execution error',
+            });
+        }
+    }
+    /**
+     * Send streaming output chunk
+     */
+    sendStreamChunk(messageId, chunk) {
+        try {
+            this.wsClient.send({
+                type: 'stream',
+                messageId,
+                chunk,
+                streamType: 'text',
+                openId: this.currentOpenId,
+                timestamp: Date.now(),
+            });
+        }
+        catch (error) {
+            // Ignore send errors, don't affect main flow
+            console.error('Failed to send stream chunk:', error);
+        }
+    }
+    /**
+     * Send tool use event
+     */
+    sendToolUse(messageId, toolUse) {
+        try {
+            this.wsClient.send({
+                type: 'stream',
+                messageId,
+                streamType: 'tool_use',
+                toolUse,
+                openId: this.currentOpenId,
+                timestamp: Date.now(),
+            });
+        }
+        catch (error) {
+            console.error('Failed to send tool use:', error);
+        }
+    }
+    /**
+     * Send tool result event
+     */
+    sendToolResult(messageId, toolResult) {
+        try {
+            this.wsClient.send({
+                type: 'stream',
+                messageId,
+                streamType: 'tool_result',
+                toolResult,
+                openId: this.currentOpenId,
+                timestamp: Date.now(),
+            });
+        }
+        catch (error) {
+            console.error('Failed to send tool result:', error);
+        }
+    }
+    /**
+     * Send structured content for rich formatting
+     */
+    sendStructuredContent(messageId, structuredContent) {
+        try {
+            this.wsClient.send({
+                type: 'structured',
+                messageId,
+                structuredContent,
+                openId: this.currentOpenId,
+                timestamp: Date.now(),
+                cwd: this.executor.getCurrentWorkingDirectory(),
+            });
+        }
+        catch (error) {
+            console.error('Failed to send structured content:', error);
+        }
+    }
+    /**
+     * Send response
+     */
+    sendResponse(messageId, result) {
+        try {
+            this.wsClient.send({
+                type: 'response',
+                messageId,
+                success: result.success,
+                output: result.output,
+                error: result.error,
+                sessionAbbr: result.sessionAbbr,
+                openId: this.currentOpenId,
+                timestamp: Date.now(),
+                cwd: this.executor.getCurrentWorkingDirectory(),
+            });
+        }
+        catch (error) {
+            console.error('Failed to send response:', error);
+        }
+    }
+    /**
+     * Destroy handler
+     */
+    destroy() {
+        this.isDestroyed = true;
+        this.isExecuting = false;
+        this.notificationAdapter.unregister();
+    }
+}
+exports.MessageHandler = MessageHandler;
+//# sourceMappingURL=MessageHandler.js.map

package/dist/client/MessageHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MessageHandler.js","sourceRoot":"","sources":["../../src/client/MessageHandler.ts"],"names":[],"mappings":";;;AAIA,oCAAqD;AACrD,iDAAsC;AAYtC;;;GAGG;AACH,MAAa,cAAc;IACjB,QAAQ,CAAkB;IAC1B,QAAQ,CAA4C;IACpD,cAAc,CAAiB;IAC/B,WAAW,GAAG,KAAK,CAAC;IACpB,WAAW,GAAG,KAAK,CAAC;IACpB,aAAa,CAAU;IACvB,mBAAmB,CAA4B;IAEvD,YACE,QAAyB,EACzB,QAAmD,EACnD,cAA8B;QAE9B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QAErC,yCAAyC;QACzC,IAAI,CAAC,mBAAmB,GAAG,IAAI,iCAAyB,CAAC,QAAQ,CAAC,CAAC;QACnE,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,CAAC;IACtC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,aAAa,CAAC,OAAkC;QACpD,6BAA6B;QAC7B,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,6BAA6B;QAC7B,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9C,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,SAAS,IAAI,SAAS,EAAE;gBACjD,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,wBAAwB;aAChC,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,qCAAqC;QACrC,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,QAAQ;gBACX,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,SAAU,CAAC,CAAC;gBACjD,OAAO;YAET,KAAK,SAAS;gBACZ,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAA0B,CAAC,CAAC;gBAC5D,OAAO;YAET,KAAK,WAAW;gBACd,kDAAkD;gBAClD,OAAO;YAET,KAAK,iBAAiB;gBACpB,mDAAmD;gBACnD,OAAO;YAET;gBACE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,SAAU,EAAE;oBACpC,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,yBAAyB,OAAO,CAAC,IAAI,EAAE;iBAC/C,CAAC,CAAC;QACP,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,oBAAoB,CAAC,OAAwB;QACzD,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;QAEjF,sDAAsD;QACtD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;QAC5B,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAElD,8CAA8C;QAC9C,IAAI,OAAO,EAAE,IAAI,EAAE,KAAK,QAAQ,EAAE,CAAC;YACjC,MAAM,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QAED,qDAAqD;QACrD,IAAI,gBAAgB,IAAI,IAAI,CAAC,QAAQ,IAAI,OAAO,IAAI,CAAC,QAAQ,CAAC,cAAc,KAAK,UAAU,EAAE,CAAC;YAC5F,MAAM,QAAQ,GAAG,IAAI,CAAC,QAA4E,CAAC;YACnG,IAAI,QAAQ,CAAC,cAAc,EAAE,EAAE,CAAC;gBAC9B,MAAM,KAAK,GAAG,OAAO,EAAE,IAAI,EAAE,CAAC;gBAC9B,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,IAAI,GAAG,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;oBACvC,IAAI,IAAI,EAAE,CAAC;wBACT,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;4BAC3B,OAAO,EAAE,IAAI;4BACb,MAAM,EAAE,YAAY,KAAK,GAAG;yBAC7B,CAAC,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;4BAC3B,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,wDAAwD;yBAChE,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;wBAC3B,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,oCAAoC;qBAC5C,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO;YACT,CAAC;QACH,CAAC;QAED,+CAA+C;QAC/C,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,4DAA4D;aACpE,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,wDAAwD;QACxD,IAAI,gBAAgB,EAAE,CAAC;YACrB,uCAAuC;YACvC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBACtD,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;oBAC3B,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,+BAA+B,gBAAgB,6BAA6B,IAAI,CAAC,cAAc;yBACnG,qBAAqB,EAAE;yBACvB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;yBACpB,IAAI,CAAC,IAAI,CAAC,EAAE;iBAChB,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,wBAAwB;YACxB,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YAExB,mEAAmE;YACnE,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,OAAQ,CAAC,CAAC;YAC3E,IAAI,aAAa,EAAE,CAAC;gBAClB,OAAO;YACT,CAAC;YAED,2DAA2D;YAC3D,IAAI,cAAc,EAAE,CAAC;gBACnB,OAAO,CAAC,GAAG,CAAC,yDAAyD,OAAO,EAAE,CAAC,CAAC;gBAChF,MAAM,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,OAAQ,CAAC,CAAC;gBACpD,OAAO;YACT,CAAC;YAED,2BAA2B;YAC3B,MAAM,eAAe,GAAG,IAAI,CAAC,sBAAsB,CAAC,OAAQ,CAAC,CAAC;YAE9D,yBAAyB;YACzB,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,kBAAkB,CAAC,SAAiB;QAChD,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC;QACtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAE5C,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;YACzB,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,YAAY;oBAClB,CAAC,CAAC,oCAAoC;oBACtC,CAAC,CAAC,0DAA0D;aAC/D,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,sCAAsC;aAC/C,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QAC/C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,QAAQ;YACd,SAAS;YACT,MAAM,EAAE;gBACN,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE;gBACtC,kBAAkB,EAAE,IAAI,CAAC,cAAc,CAAC,qBAAqB,EAAE;gBAC/D,uBAAuB,EAAE,IAAI,CAAC,QAAQ,CAAC,0BAA0B,EAAE;aACpE;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,OAAgB;QACrC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC5C,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC,CAAC,qDAAqD;QACpE,CAAC;QAED,OAAO,OAAO,CAAC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,oBAAoB,CAChC,SAAiB,EACjB,OAAe;QAEf,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;QAE/B,kBAAkB;QAClB,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,0BAA0B,EAAE,CAAC;YACvD,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,qBAAqB,EAAE,CAAC;YAEhE,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE;uBACO,GAAG;yBACD,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;qBAC1B;aACd,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QACd,CAAC;QAED,gBAAgB;QAChB,IAAI,OAAO,KAAK,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE;;;;;;;;;;;;uEAYuD;aAChE,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QACd,CAAC;QAED,iBAAiB;QACjB,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;YACzB,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC;YAC7B,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,gCAAgC;aACzC,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QACd,CAAC;QAED,cAAc;QACd,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;oBAC3B,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,wBAAwB;iBAChC,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC3C,IAAI,CAAC;gBACH,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;gBAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,0BAA0B,EAAE,CAAC;gBAC1D,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;oBAC3B,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,mCAAmC,MAAM,EAAE;iBACpD,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;oBAC3B,OAAO,EAAE,KAAK;oBACd,KAAK,EACH,KAAK,YAAY,KAAK;wBACpB,CAAC,CAAC,KAAK,CAAC,OAAO;wBACf,CAAC,CAAC,4BAA4B;iBACnC,CAAC,CAAC;YACL,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,OAAe;QAC5C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;QAE/B,iDAAiD;QACjD,IAAI,OAAO,KAAK,IAAI,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YAC9C,OAAO,yCAAyC,CAAC;QACnD,CAAC;QAED,IAAI,OAAO,KAAK,IAAI,IAAI,OAAO,KAAK,WAAW,EAAE,CAAC;YAChD,OAAO,wCAAwC,CAAC;QAClD,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,mBAAmB,CAAC,SAAiB,EAAE,OAAe;QAClE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,MAAM,WAAW,GAAa,EAAE,CAAC;YAEjC,OAAO,CAAC,GAAG,CAAC,qDAAqD,OAAO,EAAE,CAAC,CAAC;YAE5E,0CAA0C;YAC1C,qCAAqC;YACrC,MAAM,KAAK,GAAG,IAAA,qBAAK,EAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,SAAS,CAAC,EAAE;gBAClD,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,0BAA0B,EAAE;gBAC/C,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;gBACjC,GAAG,EAAE;oBACH,GAAG,OAAO,CAAC,GAAG;oBACd,UAAU,EAAE,EAAE,EAAE,+BAA+B;iBAChD;aACF,CAAC,CAAC;YAEH,gCAAgC;YAChC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;gBACxC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC9B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnB,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACzC,CAAC,CAAC,CAAC;YAEH,gBAAgB;YAChB,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;gBACxC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC9B,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACxB,OAAO,CAAC,KAAK,CAAC,mCAAmC,KAAK,EAAE,CAAC,CAAC;YAC5D,CAAC,CAAC,CAAC;YAEH,sBAAsB;YACtB,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;gBACxB,OAAO,CAAC,GAAG,CAAC,qDAAqD,IAAI,EAAE,CAAC,CAAC;gBAEzE,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;oBACf,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAC/B,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;wBAC3B,OAAO,EAAE,IAAI;wBACb,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,iCAAiC;qBAC3D,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAC5D,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;wBAC3B,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,WAAW,CAAC,IAAI,EAAE,IAAI,iCAAiC,IAAI,EAAE;qBACrE,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,uBAAuB;YACvB,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC1B,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;gBACjE,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;oBAC3B,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,8BAA8B,KAAK,CAAC,OAAO,EAAE;iBACrD,CAAC,CAAC;gBACH,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAC1B,SAAiB,EACjB,OAAe;QAEf,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE;gBAClD,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE;oBAC1B,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;gBACzC,CAAC;gBACD,SAAS,EAAE,CAAC,OAAoB,EAAE,EAAE;oBAClC,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;gBACvC,CAAC;gBACD,YAAY,EAAE,CAAC,UAA0B,EAAE,EAAE;oBAC3C,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;gBAC7C,CAAC;aACF,CAAC,CAAC;YAEH,2CAA2C;YAC3C,yDAAyD;YACzD,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,MAAM,CAAC,KAAK;aACpB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;gBAC3B,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB;aAClE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,SAAiB,EAAE,KAAa;QACtD,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,QAAQ;gBACd,SAAS;gBACT,KAAK;gBACL,UAAU,EAAE,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,aAAa;gBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,6CAA6C;YAC7C,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,SAAiB,EAAE,OAAoB;QACzD,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,QAAQ;gBACd,SAAS;gBACT,UAAU,EAAE,UAAU;gBACtB,OAAO;gBACP,MAAM,EAAE,IAAI,CAAC,aAAa;gBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,SAAiB,EAAE,UAA0B;QAClE,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,QAAQ;gBACd,SAAS;gBACT,UAAU,EAAE,aAAa;gBACzB,UAAU;gBACV,MAAM,EAAE,IAAI,CAAC,aAAa;gBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,qBAAqB,CAAC,SAAiB,EAAE,iBAAoC;QACnF,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,YAAY;gBAClB,SAAS;gBACT,iBAAiB;gBACjB,MAAM,EAAE,IAAI,CAAC,aAAa;gBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,0BAA0B,EAAE;aAC7B,CAAC,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED;;OAEG;IACK,YAAY,CAClB,SAAiB,EACjB,MAAmF;QAEnF,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,UAAU;gBAChB,SAAS;gBACT,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,MAAM,EAAE,IAAI,CAAC,aAAa;gBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,0BAA0B,EAAE;aAChD,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,CAAC;IACxC,CAAC;CACF;AA5hBD,wCA4hBC"}

package/dist/client/WebSocketClient.d.ts

@@ -0,0 +1,109 @@
+/**
+ * WebSocket client configuration
+ */
+export interface WebSocketClientOptions {
+    /** Reconnect interval (milliseconds), default 5000 */
+    reconnectInterval?: number;
+    /** Heartbeat interval (milliseconds), default 15000 */
+    heartbeatInterval?: number;
+}
+/**
+ * Connection status
+ */
+export interface ConnectionStatus {
+    connected: boolean;
+    serverUrl: string;
+    deviceId: string;
+    lastHeartbeat?: number;
+}
+/**
+ * WebSocket Client
+ * Responsible for establishing and maintaining WebSocket connection with router server
+ */
+export declare class WebSocketClient {
+    private serverUrl;
+    private deviceId;
+    private ws;
+    private reconnectInterval;
+    private heartbeatInterval;
+    private reconnectTimer;
+    private heartbeatTimer;
+    private manualDisconnect;
+    private connected;
+    private messageHandlers;
+    private errorHandlers;
+    private closeHandlers;
+    private connectHandlers;
+    constructor(serverUrl: string, deviceId: string, options?: WebSocketClientOptions);
+    /**
+     * Connect to server
+     */
+    connect(): Promise<void>;
+    /**
+     * Disconnect
+     */
+    disconnect(): void;
+    /**
+     * Send message
+     * @param message Message object
+     */
+    send(message: any): void;
+    /**
+     * Check if connected
+     */
+    isConnected(): boolean;
+    /**
+     * Get connection status
+     */
+    getStatus(): ConnectionStatus;
+    /**
+     * Register message handler
+     * @param handler Message handler function
+     */
+    onMessage(handler: (message: any) => void): void;
+    /**
+     * Register error handler
+     * @param handler Error handler function
+     */
+    onError(handler: (error: Error) => void): void;
+    /**
+     * Register close handler
+     * @param handler Close handler function
+     */
+    onClose(handler: (code: number, reason: string) => void): void;
+    /**
+     * Register connect handler
+     * @param handler Connect handler function
+     */
+    onConnect(handler: () => void): void;
+    /**
+     * Event emitter style: on method
+     * @param event Event name
+     * @param handler Event handler
+     */
+    on(event: 'connected', handler: () => void): void;
+    on(event: 'disconnected', handler: () => void): void;
+    on(event: 'error', handler: (error: Error) => void): void;
+    on(event: 'message', handler: (message: any) => void): void;
+    /**
+     * Send device registration message
+     */
+    private sendRegistration;
+    /**
+     * Start heartbeat
+     */
+    private startHeartbeat;
+    /**
+     * Stop heartbeat
+     */
+    private stopHeartbeat;
+    /**
+     * Schedule reconnection
+     */
+    private scheduleReconnect;
+    /**
+     * Clear reconnect timer
+     */
+    private clearReconnectTimer;
+}
+//# sourceMappingURL=WebSocketClient.d.ts.map

package/dist/client/WebSocketClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebSocketClient.d.ts","sourceRoot":"","sources":["../../src/client/WebSocketClient.ts"],"names":[],"mappings":"AAGA;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,sDAAsD;IACtD,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,uDAAuD;IACvD,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;;GAGG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,iBAAiB,CAAS;IAClC,OAAO,CAAC,iBAAiB,CAAS;IAClC,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,eAAe,CAAqC;IAC5D,OAAO,CAAC,aAAa,CAAqC;IAC1D,OAAO,CAAC,aAAa,CAAqD;IAC1E,OAAO,CAAC,eAAe,CAAyB;gBAEpC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,GAAE,sBAA2B;IAOrF;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAoD9B;;OAEG;IACH,UAAU,IAAI,IAAI;IAkBlB;;;OAGG;IACH,IAAI,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI;IAQxB;;OAEG;IACH,WAAW,IAAI,OAAO;IAItB;;OAEG;IACH,SAAS,IAAI,gBAAgB;IAQ7B;;;OAGG;IACH,SAAS,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,GAAG,KAAK,IAAI,GAAG,IAAI;IAIhD;;;OAGG;IACH,OAAO,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,GAAG,IAAI;IAI9C;;;OAGG;IACH,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAI9D;;;OAGG;IACH,SAAS,CAAC,OAAO,EAAE,MAAM,IAAI,GAAG,IAAI;IAIpC;;;;OAIG;IACH,EAAE,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,IAAI,GAAG,IAAI;IACjD,EAAE,CAAC,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,IAAI,GAAG,IAAI;IACpD,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,GAAG,IAAI;IACzD,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,GAAG,KAAK,IAAI,GAAG,IAAI;IAoB3D;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAaxB;;OAEG;IACH,OAAO,CAAC,cAAc;IAYtB;;OAEG;IACH,OAAO,CAAC,aAAa;IAOrB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAYzB;;OAEG;IACH,OAAO,CAAC,mBAAmB;CAM5B"}