@zhin.js/client 1.0.17 → 1.1.0

package/dist/websocket/manager.js

@@ -1,411 +1,403 @@
-/**
- * WebSocket 核心管理器
- * 负责 WebSocket 连接管理、消息发送、重连逻辑等核心功能
- */
-import { store, setConnected, updateConfigs, updateSchemas } from '../store';
-import { MessageHandler } from './messageHandler';
-import { ConnectionState, WebSocketError, ConnectionError, MessageError, RequestTimeoutError } from './types';
+import { ConnectionState, WebSocketError, ConnectionError, MessageError, RequestTimeoutError, } from "./types";
+import { getStoredToken, resolveApiUrl } from "./remote-settings.js";
+import { applyConsoleEvent } from "../persistence/idb-store.js";
 export class WebSocketManager {
-    // ============================================================================
-    // 私有属性
-    // ============================================================================
     ws = null;
     config;
     callbacks;
     state = ConnectionState.DISCONNECTED;
-    // 重连相关
     reconnectAttempts = 0;
     reconnectTimer = null;
-    // 请求管理
     requestId = 0;
     pendingRequests = new Map();
-    // ============================================================================
-    // 构造函数
-    // ============================================================================
+    connectedListeners = new Set();
+    sseAbort = null;
+    useRestTransport = true;
     constructor(config = {}, callbacks = {}) {
         this.config = {
             url: this.buildWebSocketUrl(config.url),
             reconnectInterval: config.reconnectInterval ?? 3000,
             maxReconnectAttempts: config.maxReconnectAttempts ?? 10,
-            requestTimeout: config.requestTimeout ?? 10000
+            requestTimeout: config.requestTimeout ?? 10000,
         };
         this.callbacks = callbacks;
     }
-    // ============================================================================
-    // 公共 API
-    // ============================================================================
-    /**
-     * 建立 WebSocket 连接
-     */
+    onConnectionChange(listener) {
+        this.connectedListeners.add(listener);
+        return () => this.connectedListeners.delete(listener);
+    }
+    notifyConnection(connected) {
+        for (const l of this.connectedListeners)
+            l(connected);
+    }
     connect() {
-        if (this.state === ConnectionState.CONNECTED || this.state === ConnectionState.CONNECTING) {
+        if (this.useRestTransport) {
+            void this.connectRestSse();
             return;
         }
+        if (this.state === ConnectionState.CONNECTED || this.state === ConnectionState.CONNECTING)
+            return;
         this.setState(ConnectionState.CONNECTING);
         try {
             this.ws = new WebSocket(this.config.url);
             this.attachEventHandlers();
         }
         catch (error) {
-            this.handleConnectionError(new ConnectionError('Failed to create WebSocket', error));
+            this.handleConnectionError(new ConnectionError("Failed to create WebSocket", error));
+        }
+    }
+    async connectRestSse() {
+        const token = getStoredToken();
+        if (!token) {
+            this.setState(ConnectionState.DISCONNECTED);
+            this.notifyConnection(false);
+            return;
+        }
+        this.setState(ConnectionState.CONNECTING);
+        this.sseAbort?.abort();
+        this.sseAbort = new AbortController();
+        try {
+            const res = await fetch(resolveApiUrl("/api/events"), {
+                headers: {
+                    Authorization: `Bearer ${token}`,
+                    Accept: "text/event-stream",
+                },
+                signal: this.sseAbort.signal,
+            });
+            if (!res.ok || !res.body) {
+                throw new ConnectionError(`SSE failed: HTTP ${res.status}`);
+            }
+            this.setState(ConnectionState.CONNECTED);
+            this.reconnectAttempts = 0;
+            this.notifyConnection(true);
+            this.callbacks.onConnect?.();
+            void this.pumpSse(res.body);
+        }
+        catch (error) {
+            if (error.name === "AbortError")
+                return;
+            this.handleConnectionError(error instanceof ConnectionError ? error : new ConnectionError("SSE connect failed", error));
+            this.scheduleReconnect();
+        }
+    }
+    async pumpSse(body) {
+        const reader = body.getReader();
+        const decoder = new TextDecoder();
+        let buffer = "";
+        try {
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                buffer += decoder.decode(value, { stream: true });
+                const parts = buffer.split("\n\n");
+                buffer = parts.pop() ?? "";
+                for (const part of parts) {
+                    const line = part.split("\n").find((l) => l.startsWith("data: "));
+                    if (!line)
+                        continue;
+                    const json = line.slice(6);
+                    try {
+                        const message = JSON.parse(json);
+                        void applyConsoleEvent(message);
+                        this.handleBroadcast(message);
+                        this.callbacks.onMessage?.(message);
+                    }
+                    catch {
+                        /* skip */
+                    }
+                }
+            }
+        }
+        catch {
+            /* closed */
+        }
+        finally {
+            if (this.state === ConnectionState.CONNECTED) {
+                this.notifyConnection(false);
+                this.setState(ConnectionState.RECONNECTING);
+                this.scheduleReconnect();
+            }
         }
     }
-    /**
-     * 断开 WebSocket 连接
-     */
     disconnect() {
         this.clearReconnectTimer();
         this.clearPendingRequests();
+        this.sseAbort?.abort();
+        this.sseAbort = null;
         if (this.ws) {
             this.ws.close();
             this.ws = null;
         }
         this.setState(ConnectionState.DISCONNECTED);
-        store.dispatch(setConnected(false));
+        this.notifyConnection(false);
     }
-    /**
-     * 发送普通消息
-     */
     send(message) {
-        if (!this.isConnected()) {
-            throw new WebSocketError('WebSocket is not connected', 'NOT_CONNECTED');
+        if (this.useRestTransport) {
+            void this.sendRequest(message);
+            return;
         }
+        if (!this.isConnected())
+            throw new WebSocketError("WebSocket is not connected", "NOT_CONNECTED");
         try {
             this.ws.send(JSON.stringify(message));
         }
         catch (error) {
-            throw new MessageError('Failed to send message', error);
+            throw new MessageError("Failed to send message", error);
         }
     }
-    /**
-     * 发送请求并等待响应
-     */
     async sendRequest(message) {
-        if (!this.isConnected()) {
-            throw new WebSocketError('WebSocket is not connected', 'NOT_CONNECTED');
+        if (this.useRestTransport) {
+            return this.sendRestRequest(message);
         }
+        if (!this.isConnected())
+            throw new WebSocketError("WebSocket is not connected", "NOT_CONNECTED");
         return new Promise((resolve, reject) => {
             const requestId = ++this.requestId;
             const messageWithId = { ...message, requestId };
-            // 设置超时计时器
             const timer = setTimeout(() => {
                 this.pendingRequests.delete(requestId);
                 reject(new RequestTimeoutError(requestId));
             }, this.config.requestTimeout);
-            // 存储请求信息
-            this.pendingRequests.set(requestId, { resolve, reject, timer });
+            this.pendingRequests.set(requestId, { resolve: resolve, reject, timer });
             try {
                 this.ws.send(JSON.stringify(messageWithId));
             }
             catch (error) {
                 this.pendingRequests.delete(requestId);
                 clearTimeout(timer);
-                reject(new MessageError('Failed to send request', error));
+                reject(new MessageError("Failed to send request", error));
             }
         });
     }
-    /**
-     * 检查连接状态
-     */
+    async sendRestRequest(message) {
+        const token = getStoredToken();
+        if (!token)
+            throw new WebSocketError("Not authenticated", "NOT_CONNECTED");
+        const requestId = ++this.requestId;
+        const body = { ...message, requestId };
+        const timer = setTimeout(() => {
+            /* handled below */
+        }, this.config.requestTimeout);
+        try {
+            const res = await fetch(resolveApiUrl("/api/console/request"), {
+                method: "POST",
+                headers: {
+                    Authorization: `Bearer ${token}`,
+                    "Content-Type": "application/json",
+                },
+                body: JSON.stringify(body),
+                signal: AbortSignal.timeout(this.config.requestTimeout),
+            });
+            clearTimeout(timer);
+            const json = (await res.json());
+            if (!res.ok || json.success === false) {
+                throw new WebSocketError(json.error ?? `HTTP ${res.status}`, "SERVER_ERROR");
+            }
+            return json.data;
+        }
+        catch (error) {
+            clearTimeout(timer);
+            if (error instanceof WebSocketError)
+                throw error;
+            throw new MessageError("REST request failed", error);
+        }
+    }
     isConnected() {
+        if (this.useRestTransport) {
+            return this.state === ConnectionState.CONNECTED;
+        }
         return this.ws?.readyState === WebSocket.OPEN;
     }
-    /**
-     * 获取当前状态
-     */
     getState() {
         return this.state;
     }
-    // ============================================================================
-    // 配置管理 API
-    // ============================================================================
-    /**
-     * 获取插件配置
-     */
     async getConfig(pluginName) {
-        return this.sendRequest({
-            type: 'config:get',
-            pluginName
-        });
+        return this.sendRequest({ type: "config:get", pluginName });
     }
-    /**
-     * 设置插件配置
-     */
     async setConfig(pluginName, config) {
         return this.sendRequest({
-            type: 'config:set',
+            type: "config:set",
             pluginName,
-            data: config
+            data: config,
         });
     }
-    /**
-     * 获取插件 Schema
-     */
     async getSchema(pluginName) {
-        return this.sendRequest({
-            type: 'schema:get',
-            pluginName
-        });
+        return this.sendRequest({ type: "schema:get", pluginName });
     }
-    /**
-     * 获取所有配置
-     */
     async getAllConfigs() {
-        return this.sendRequest({
-            type: 'config:get-all'
-        });
+        return this.sendRequest({ type: "config:get-all" });
     }
-    /**
-     * 获取所有 Schema
-     */
     async getAllSchemas() {
-        return this.sendRequest({
-            type: 'schema:get-all'
-        });
+        return this.sendRequest({ type: "schema:get-all" });
     }
-    // ============================================================================
-    // 配置文件原始 YAML 读写
-    // ============================================================================
     async getConfigYaml() {
-        return this.sendRequest({ type: 'config:get-yaml' });
+        return this.sendRequest({ type: "config:get-yaml" });
     }
     async saveConfigYaml(yaml) {
-        return this.sendRequest({ type: 'config:save-yaml', yaml });
+        return this.sendRequest({ type: "config:save-yaml", yaml });
     }
-    // ============================================================================
-    // 环境变量文件管理
-    // ============================================================================
     async getEnvList() {
-        return this.sendRequest({ type: 'env:list' });
+        return this.sendRequest({ type: "env:list" });
     }
     async getEnvFile(filename) {
-        return this.sendRequest({ type: 'env:get', filename });
+        return this.sendRequest({ type: "env:get", filename });
     }
     async saveEnvFile(filename, content) {
-        return this.sendRequest({ type: 'env:save', filename, content });
+        return this.sendRequest({ type: "env:save", filename, content });
     }
-    // ============================================================================
-    // 文件管理
-    // ============================================================================
     async getFileTree() {
-        return this.sendRequest({ type: 'files:tree' });
+        return this.sendRequest({ type: "files:tree" });
     }
     async readFile(filePath) {
-        return this.sendRequest({ type: 'files:read', filePath });
+        return this.sendRequest({ type: "files:read", filePath });
     }
     async saveFile(filePath, content) {
-        return this.sendRequest({ type: 'files:save', filePath, content });
+        return this.sendRequest({ type: "files:save", filePath, content });
     }
-    // ============================================================================
-    // 数据库管理
-    // ============================================================================
     async getDbInfo() {
-        return this.sendRequest({ type: 'db:info' });
+        return this.sendRequest({ type: "db:info" });
     }
     async getDbTables() {
-        return this.sendRequest({ type: 'db:tables' });
+        return this.sendRequest({ type: "db:tables" });
     }
     async dbSelect(table, page, pageSize, where) {
-        return this.sendRequest({ type: 'db:select', table, page, pageSize, where });
+        return this.sendRequest({ type: "db:select", table, page, pageSize, where });
     }
     async dbInsert(table, row) {
-        return this.sendRequest({ type: 'db:insert', table, row });
+        return this.sendRequest({ type: "db:insert", table, row });
     }
     async dbUpdate(table, row, where) {
-        return this.sendRequest({ type: 'db:update', table, row, where });
+        return this.sendRequest({ type: "db:update", table, row, where });
     }
     async dbDelete(table, where) {
-        return this.sendRequest({ type: 'db:delete', table, where });
+        return this.sendRequest({ type: "db:delete", table, where });
     }
     async dbDropTable(table) {
-        return this.sendRequest({ type: 'db:drop-table', table });
+        return this.sendRequest({ type: "db:drop-table", table });
     }
     async kvGet(table, key) {
-        return this.sendRequest({ type: 'db:kv:get', table, key });
+        return this.sendRequest({ type: "db:kv:get", table, key });
     }
     async kvSet(table, key, value, ttl) {
-        return this.sendRequest({ type: 'db:kv:set', table, key, value, ttl });
+        return this.sendRequest({ type: "db:kv:set", table, key, value, ttl });
     }
     async kvDelete(table, key) {
-        return this.sendRequest({ type: 'db:kv:delete', table, key });
+        return this.sendRequest({ type: "db:kv:delete", table, key });
     }
     async kvGetEntries(table) {
-        return this.sendRequest({ type: 'db:kv:entries', table });
-    }
-    // ============================================================================
-    // 私有方法
-    // ============================================================================
-    /**
-     * 构建 WebSocket URL
-     */
+        return this.sendRequest({ type: "db:kv:entries", table });
+    }
     buildWebSocketUrl(customUrl) {
-        if (customUrl) {
+        if (customUrl)
             return customUrl;
-        }
-        const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
-        const host = window.location.host;
-        return `${protocol}//${host}/server`;
+        const protocol = window.location.protocol === "https:" ? "wss:" : "ws:";
+        return `${protocol}//${window.location.host}/server`;
     }
-    /**
-     * 设置连接状态
-     */
     setState(newState) {
-        const oldState = this.state;
         this.state = newState;
-        console.debug(`[WebSocket] State changed: ${oldState} -> ${newState}`);
     }
-    /**
-     * 绑定事件处理器
-     */
     attachEventHandlers() {
         if (!this.ws)
             return;
-        this.ws.onopen = () => {
-            this.handleConnectionOpen();
-        };
-        this.ws.onmessage = (event) => {
-            this.handleMessage(event);
-        };
-        this.ws.onclose = (event) => {
-            this.handleConnectionClose(event);
-        };
-        this.ws.onerror = (event) => {
-            this.handleConnectionError(new ConnectionError('WebSocket error', event));
-        };
+        this.ws.onopen = () => this.handleConnectionOpen();
+        this.ws.onmessage = (event) => this.handleMessage(event);
+        this.ws.onclose = () => this.handleConnectionClose();
+        this.ws.onerror = (event) => this.handleConnectionError(new ConnectionError("WebSocket error", event));
     }
-    /**
-     * 处理连接打开
-     */
     handleConnectionOpen() {
         this.setState(ConnectionState.CONNECTED);
         this.reconnectAttempts = 0;
-        // 更新 Redux 状态
-        store.dispatch(setConnected(true));
-        // 初始化数据
-        this.initializeData();
-        // 触发回调
+        this.notifyConnection(true);
         this.callbacks.onConnect?.();
     }
-    /**
-     * 处理消息接收
-     */
     handleMessage(event) {
         try {
             const message = JSON.parse(event.data);
-            // 处理请求响应
             if (message.requestId && this.pendingRequests.has(message.requestId)) {
                 this.handleRequestResponse(message);
                 return;
             }
-            // 处理广播消息
-            MessageHandler.handle(message);
+            void applyConsoleEvent(message);
+            this.handleBroadcast(message);
             this.callbacks.onMessage?.(message);
         }
         catch (error) {
-            console.error('[WebSocket] Message parsing error:', error);
+            console.error("[WebSocket] Message parsing error:", error);
+        }
+    }
+    handleBroadcast(message) {
+        const t = message.type;
+        if (t === "bot:request" || t === "bot:notice" || t === "bot:message") {
+            if (typeof window !== "undefined") {
+                window.dispatchEvent(new CustomEvent("zhin-console-bot-push", { detail: message }));
+            }
+            return;
+        }
+        if (t === "hmr:reload") {
+            console.info(`[HMR] File changed, reloading...`);
+            window.location.reload();
+            return;
+        }
+        if (t === "system:restarting") {
+            console.info("[System] Server restarting, will reload shortly...");
+            setTimeout(() => window.location.reload(), 3000);
+            return;
         }
     }
-    /**
-     * 处理请求响应
-     */
     handleRequestResponse(message) {
         const { requestId } = message;
-        const pendingRequest = this.pendingRequests.get(requestId);
-        if (!pendingRequest) {
+        const pending = this.pendingRequests.get(requestId);
+        if (!pending)
             return;
-        }
         this.pendingRequests.delete(requestId);
-        clearTimeout(pendingRequest.timer);
+        clearTimeout(pending.timer);
         if (message.error) {
-            pendingRequest.reject(new WebSocketError(message.error, 'SERVER_ERROR'));
+            pending.reject(new WebSocketError(message.error, "SERVER_ERROR"));
         }
         else {
-            pendingRequest.resolve(message.data);
+            pending.resolve(message.data);
         }
     }
-    /**
-     * 处理连接关闭
-     */
-    handleConnectionClose(event) {
+    handleConnectionClose() {
         this.ws = null;
-        store.dispatch(setConnected(false));
-        if (this.state === ConnectionState.DISCONNECTED) {
-            // 主动断开，不重连
+        this.notifyConnection(false);
+        if (this.state === ConnectionState.DISCONNECTED)
             return;
-        }
         this.setState(ConnectionState.RECONNECTING);
         this.callbacks.onDisconnect?.();
         this.scheduleReconnect();
     }
-    /**
-     * 处理连接错误
-     */
     handleConnectionError(error) {
         this.clearReconnectTimer();
         this.setState(ConnectionState.ERROR);
-        console.error('[WebSocket] Connection error:', error);
+        console.error("[Console transport] Connection error:", error);
         this.callbacks.onError?.(error);
     }
-    /**
-     * 安排重连
-     */
     scheduleReconnect() {
         if (this.reconnectAttempts >= this.config.maxReconnectAttempts) {
-            console.error('[WebSocket] Max reconnect attempts reached');
             this.setState(ConnectionState.ERROR);
             return;
         }
         this.reconnectAttempts++;
         const delay = this.config.reconnectInterval * this.reconnectAttempts;
-        console.debug(`[WebSocket] Reconnecting in ${delay}ms (attempt ${this.reconnectAttempts}/${this.config.maxReconnectAttempts})`);
         this.reconnectTimer = setTimeout(() => {
-            if (this.state === ConnectionState.RECONNECTING) {
+            if (this.state === ConnectionState.RECONNECTING)
                 this.connect();
-            }
         }, delay);
     }
-    /**
-     * 清除重连计时器
-     */
     clearReconnectTimer() {
         if (this.reconnectTimer) {
             clearTimeout(this.reconnectTimer);
             this.reconnectTimer = null;
         }
     }
-    /**
-     * 清除所有待处理请求
-     */
     clearPendingRequests() {
-        for (const [requestId, { reject, timer }] of this.pendingRequests) {
+        for (const [, { reject, timer }] of this.pendingRequests) {
             clearTimeout(timer);
-            reject(new WebSocketError('Connection closed', 'CONNECTION_CLOSED'));
+            reject(new WebSocketError("Connection closed", "CONNECTION_CLOSED"));
         }
         this.pendingRequests.clear();
     }
-    /**
-     * 初始化数据
-     */
-    async initializeData() {
-        try {
-            // 并行获取所有配置和 Schema
-            const [configs, schemas] = await Promise.all([
-                this.getAllConfigs().catch(error => {
-                    console.warn('[WebSocket] Failed to load configs:', error);
-                    return {};
-                }),
-                this.getAllSchemas().catch(error => {
-                    console.warn('[WebSocket] Failed to load schemas:', error);
-                    return {};
-                })
-            ]);
-            // 更新 Redux 状态
-            store.dispatch(updateConfigs(configs));
-            store.dispatch(updateSchemas(schemas));
-        }
-        catch (error) {
-            console.error('[WebSocket] Data initialization failed:', error);
-        }
-    }
 }
 //# sourceMappingURL=manager.js.map