@zhin.js/client 1.0.3 → 1.0.4

package/src/websocket/messageHandler.ts

@@ -0,0 +1,166 @@
+/**
+ * WebSocket 消息处理器
+ * 负责处理不同类型的 WebSocket 消息并分发到对应的处理逻辑
+ */
+
+import {
+  store,
+  loadScripts,
+  loadScript,
+  unloadScript,
+  setConnected,
+  updateConfig,
+  updateSchema,
+  updateConfigs,
+  updateSchemas,
+  setError
+} from '../store'
+import type { WebSocketMessage } from './types'
+
+export class MessageHandler {
+  /**
+   * 处理 WebSocket 消息
+   */
+  static handle(message: WebSocketMessage): void {
+    try {
+      switch (message.type) {
+        // 脚本管理相关消息
+        case 'sync':
+          this.handleScriptSync(message)
+          break
+        case 'add':
+          this.handleScriptAdd(message)
+          break
+        case 'delete':
+          this.handleScriptDelete(message)
+          break
+
+        // 配置管理相关消息
+        case 'config:updated':
+          this.handleConfigUpdated(message)
+          break
+        case 'config:batch':
+          this.handleConfigBatch(message)
+          break
+        case 'config:error':
+          this.handleConfigError(message)
+          break
+
+        // Schema 管理相关消息
+        case 'schema:updated':
+          this.handleSchemaUpdated(message)
+          break
+        case 'schema:batch':
+          this.handleSchemaBatch(message)
+          break
+
+        // 系统消息
+        case 'init-data':
+        case 'data-update':
+          this.handleSystemMessage(message)
+          break
+
+        default:
+          console.warn('[WebSocket] Unknown message type:', message.type)
+      }
+    } catch (error) {
+      console.error('[WebSocket] Message handling error:', error)
+    }
+  }
+
+  /**
+   * 处理脚本同步消息
+   */
+  private static handleScriptSync(message: any): void {
+    if (message.data?.key === 'entries') {
+      const entries = Array.isArray(message.data.value) 
+        ? message.data.value 
+        : [message.data.value]
+      
+      store.dispatch({ type: 'script/syncEntries', payload: entries })
+      store.dispatch(loadScripts(entries))
+    }
+  }
+
+  /**
+   * 处理脚本添加消息
+   */
+  private static handleScriptAdd(message: any): void {
+    if (message.data?.key === 'entries') {
+      store.dispatch({ type: 'script/addEntry', payload: message.data.value })
+      store.dispatch(loadScript(message.data.value))
+    }
+  }
+
+  /**
+   * 处理脚本删除消息
+   */
+  private static handleScriptDelete(message: any): void {
+    if (message.data?.key === 'entries') {
+      store.dispatch({ type: 'script/removeEntry', payload: message.data.value })
+      store.dispatch(unloadScript(message.data.value))
+    }
+  }
+
+  /**
+   * 处理配置更新消息
+   */
+  private static handleConfigUpdated(message: any): void {
+    if (message.pluginName && message.data !== undefined) {
+      store.dispatch(updateConfig({
+        pluginName: message.pluginName,
+        config: message.data
+      }))
+    }
+  }
+
+  /**
+   * 处理批量配置消息
+   */
+  private static handleConfigBatch(message: any): void {
+    if (message.data) {
+      store.dispatch(updateConfigs(message.data))
+    }
+  }
+
+  /**
+   * 处理配置错误消息
+   */
+  private static handleConfigError(message: any): void {
+    if (message.pluginName && message.error) {
+      store.dispatch(setError({
+        pluginName: message.pluginName,
+        error: message.error
+      }))
+    }
+  }
+
+  /**
+   * 处理 Schema 更新消息
+   */
+  private static handleSchemaUpdated(message: any): void {
+    if (message.pluginName && message.data !== undefined) {
+      store.dispatch(updateSchema({
+        pluginName: message.pluginName,
+        schema: message.data
+      }))
+    }
+  }
+
+  /**
+   * 处理批量 Schema 消息
+   */
+  private static handleSchemaBatch(message: any): void {
+    if (message.data) {
+      store.dispatch(updateSchemas(message.data))
+    }
+  }
+
+  /**
+   * 处理系统消息
+   */
+  private static handleSystemMessage(message: WebSocketMessage): void {
+    // 系统消息暂时不需要特殊处理
+    // 可以在这里添加系统级别的消息处理逻辑
+  }
+}

package/src/websocket/types.ts

@@ -0,0 +1,208 @@
+/**
+ * WebSocket 相关类型定义
+ */
+
+// ============================================================================
+// 基础消息类型
+// ============================================================================
+
+export interface BaseMessage {
+  type: string
+  timestamp?: number
+  requestId?: number
+  error?: string
+}
+
+// ============================================================================
+// 脚本管理消息类型
+// ============================================================================
+
+export interface ScriptSyncMessage extends BaseMessage {
+  type: 'sync'
+  data: {
+    key: 'entries'
+    value: string[]
+  }
+}
+
+export interface ScriptAddMessage extends BaseMessage {
+  type: 'add'
+  data: {
+    key: 'entries'
+    value: string
+  }
+}
+
+export interface ScriptDeleteMessage extends BaseMessage {
+  type: 'delete'
+  data: {
+    key: 'entries'
+    value: string
+  }
+}
+
+// ============================================================================
+// 配置管理消息类型
+// ============================================================================
+
+export interface ConfigGetMessage extends BaseMessage {
+  type: 'config:get'
+  pluginName: string
+}
+
+export interface ConfigSetMessage extends BaseMessage {
+  type: 'config:set'
+  pluginName: string
+  data: any
+}
+
+export interface ConfigUpdatedMessage extends BaseMessage {
+  type: 'config:updated'
+  pluginName: string
+  data: any
+}
+
+export interface ConfigBatchMessage extends BaseMessage {
+  type: 'config:batch'
+  data: Record<string, any>
+}
+
+export interface ConfigErrorMessage extends BaseMessage {
+  type: 'config:error'
+  pluginName: string
+  error: string
+}
+
+export interface SchemaGetMessage extends BaseMessage {
+  type: 'schema:get'
+  pluginName: string
+}
+
+export interface SchemaUpdatedMessage extends BaseMessage {
+  type: 'schema:updated'
+  pluginName: string
+  data: any
+}
+
+export interface SchemaBatchMessage extends BaseMessage {
+  type: 'schema:batch'
+  data: Record<string, any>
+}
+
+// ============================================================================
+// 系统消息类型
+// ============================================================================
+
+export interface InitDataMessage extends BaseMessage {
+  type: 'init-data'
+}
+
+export interface DataUpdateMessage extends BaseMessage {
+  type: 'data-update'
+}
+
+// ============================================================================
+// 联合类型
+// ============================================================================
+
+export type WebSocketMessage = 
+  | ScriptSyncMessage
+  | ScriptAddMessage
+  | ScriptDeleteMessage
+  | ConfigGetMessage
+  | ConfigSetMessage
+  | ConfigUpdatedMessage
+  | ConfigBatchMessage
+  | ConfigErrorMessage
+  | SchemaGetMessage
+  | SchemaUpdatedMessage
+  | SchemaBatchMessage
+  | InitDataMessage
+  | DataUpdateMessage
+
+// ============================================================================
+// 配置选项类型
+// ============================================================================
+
+export interface WebSocketConfig {
+  /** WebSocket 服务器 URL */
+  url?: string
+  /** 重连间隔时间（毫秒） */
+  reconnectInterval?: number
+  /** 最大重连尝试次数 */
+  maxReconnectAttempts?: number
+  /** 请求超时时间（毫秒） */
+  requestTimeout?: number
+}
+
+export interface WebSocketCallbacks {
+  /** 连接建立回调 */
+  onConnect?: () => void
+  /** 连接断开回调 */
+  onDisconnect?: () => void
+  /** 连接错误回调 */
+  onError?: (error: Event) => void
+  /** 消息接收回调 */
+  onMessage?: (message: WebSocketMessage) => void
+}
+
+// ============================================================================
+// Hook 配置类型
+// ============================================================================
+
+export interface UseConfigOptions {
+  /** 是否自动加载配置 */
+  autoLoad?: boolean
+  /** 是否自动加载 Schema */
+  autoLoadSchema?: boolean
+}
+
+export interface UseWebSocketOptions {
+  /** 是否自动连接 */
+  autoConnect?: boolean
+}
+
+// ============================================================================
+// 连接状态枚举
+// ============================================================================
+
+export enum ConnectionState {
+  DISCONNECTED = 'disconnected',
+  CONNECTING = 'connecting',
+  CONNECTED = 'connected',
+  RECONNECTING = 'reconnecting',
+  ERROR = 'error'
+}
+
+// ============================================================================
+// 错误类型
+// ============================================================================
+
+export class WebSocketError extends Error {
+  constructor(
+    message: string,
+    public code: string,
+    public originalError?: Error
+  ) {
+    super(message)
+    this.name = 'WebSocketError'
+  }
+}
+
+export class RequestTimeoutError extends WebSocketError {
+  constructor(requestId: number) {
+    super(`Request ${requestId} timed out`, 'REQUEST_TIMEOUT')
+  }
+}
+
+export class ConnectionError extends WebSocketError {
+  constructor(message: string, originalError?: Error) {
+    super(message, 'CONNECTION_ERROR', originalError)
+  }
+}
+
+export class MessageError extends WebSocketError {
+  constructor(message: string, originalError?: Error) {
+    super(message, 'MESSAGE_ERROR', originalError)
+  }
+}

package/src/websocket/useWebSocket.ts

@@ -1,42 +0,0 @@
-import { useCallback } from 'react'
-import { useSelector } from '../store'
-import { getWebSocketManager } from './index'
-
-/**
- * WebSocket Hook
- * 用于在 React 组件中访问 WebSocket 状态和方法
- * WebSocket 在模块加载时已自动连接，无需手动初始化
- * 状态由 Redux store 管理
- */
-export function useWebSocket() {
-    // 从 Redux 读取状态
-    const entries = useSelector((state: any) => state.script.entries || [])
-    const loadedScripts = useSelector((state: any) => state.script.loadedScripts || [])
-
-    const connect = useCallback(() => {
-        getWebSocketManager().connect()
-    }, [])
-
-    const disconnect = useCallback(() => {
-        getWebSocketManager().disconnect()
-    }, [])
-
-    const send = useCallback((message: any) => {
-        getWebSocketManager().send(message)
-    }, [])
-
-    const isConnected = useCallback(() => {
-        return getWebSocketManager().isConnected()
-    }, [])
-
-    return {
-        entries,
-        loadedScripts,
-        connect,
-        disconnect,
-        send,
-        isConnected
-    }
-}
-
-export default useWebSocket