@bundy-lmw/hive-server 1.0.1 → 1.0.3

package/dist/gateway/ws/admin-handler.js

@@ -1,72 +1,77 @@
 /**
- * Admin WebSocket Handler
+ * Admin WebSocket Handler (Router)
  *
- * 处理 /ws/admin 端点的所有管理协议消息：
- * - req/res: 配置管理、服务状态、插件管理、日志查询
- * - event: 日志流推送、状态变更通知
+ * 处理 /ws/admin 端点的管理协议消息。
+ * 业务逻辑委托给 Domain Handler，本类仅负责：
+ * - WS 连接管理
+ * - 消息路由（dispatch）
+ * - HandlerContext 构建
+ * - 生命周期管理
  */
 import { EventEmitter } from 'node:events';
-import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';
-import { resolve, join } from 'node:path';
-import { execSync } from 'node:child_process';
-import { createSuccessResponse, createErrorResponse, createEvent } from './types.js';
-import { HIVE_HOME } from '../../config.js';
-import { LogBuffer } from './log-buffer.js';
+import { createErrorResponse, createEvent } from './types.js';
+import { ConfigStore } from './config-store.js';
+import { ConfigHandler, StatusHandler, PluginHandler, LogHandler, SessionHandler, } from './handlers/index.js';
 // ============================================
-// AdminWsHandler
+// AdminWsHandler (Router)
 // ============================================
 export class AdminWsHandler extends EventEmitter {
     clients = new Set();
     logBuffer;
-    configCache = null;
-    configPath;
+    hiveLogger = null;
+    configStore;
     server = null;
-    httpServer = null;
     startTime;
     handlers;
-    constructor() {
+    statusHandler;
+    pluginHandler;
+    constructor(hiveLogger, logBuffer) {
         super();
-        this.logBuffer = new LogBuffer(10_000);
-        this.configPath = join(HIVE_HOME, 'hive.config.json');
+        this.logBuffer = logBuffer;
+        this.hiveLogger = hiveLogger;
+        this.configStore = new ConfigStore();
         this.startTime = Date.now();
-        // 注册方法处理器
-        this.handlers = new Map([
-            ['config.get', this.handleConfigGet.bind(this)],
-            ['config.update', this.handleConfigUpdate.bind(this)],
-            ['config.getProviderPresets', this.handleGetProviderPresets.bind(this)],
-            ['status.get', this.handleStatusGet.bind(this)],
-            ['server.restart', this.handleServerRestart.bind(this)],
-            ['server.getProviders', this.handleGetProviders.bind(this)],
-            ['provider.list', this.handleProviderList.bind(this)],
-            ['provider.getModels', this.handleProviderGetModels.bind(this)],
-            ['plugin.list', this.handlePluginList.bind(this)],
-            ['plugin.install', this.handlePluginInstall.bind(this)],
-            ['plugin.uninstall', this.handlePluginUninstall.bind(this)],
-            ['plugin.updateConfig', this.handlePluginUpdateConfig.bind(this)],
-            ['log.getHistory', this.handleLogGetHistory.bind(this)],
-            ['log.subscribe', this.handleLogSubscribe.bind(this)],
-            ['log.unsubscribe', this.handleLogUnsubscribe.bind(this)],
-            ['session.list', this.handleSessionList.bind(this)],
-            ['session.get', this.handleSessionGet.bind(this)],
-            ['session.delete', this.handleSessionDelete.bind(this)],
-        ]);
-        // 拦截 console 输出
-        this.interceptConsole();
+        const ctx = this.createContext();
+        this.statusHandler = new StatusHandler(ctx, this.startTime);
+        this.pluginHandler = new PluginHandler(ctx);
+        const domains = [
+            new ConfigHandler(ctx),
+            this.statusHandler,
+            this.pluginHandler,
+            new LogHandler(ctx),
+            new SessionHandler(ctx),
+        ];
+        this.handlers = new Map();
+        for (const domain of domains) {
+            for (const [method, handler] of domain.register()) {
+                this.handlers.set(method, handler);
+            }
+        }
     }
     // ============================================
     // 生命周期
     // ============================================
-    /** 注入 Server 实例（在 start 后调用） */
     setServer(server) {
         this.server = server;
     }
-    /** 注入 HttpServer 实例 */
     setHttpServer(httpServer) {
-        this.httpServer = httpServer;
+        this.statusHandler.setHttpServer(httpServer);
+    }
+    setPlugins(plugins) {
+        this.pluginHandler.setPlugins(plugins);
+    }
+    getHiveLogger() {
+        return this.hiveLogger;
     }
-    /** 处理新的 WS 连接 */
+    /** Push a log entry to subscribed admin clients — called by main.ts subscriber */
+    pushLog(entry) {
+        this.broadcastLog(entry);
+    }
+    // ============================================
+    // 连接管理
+    // ============================================
     handleConnection(ws) {
-        const client = { ws, logSubscribed: false };
+        const client = { ws, logSubscribed: false, threadIds: new Set() };
         this.clients.add(client);
         ws.on('message', (raw) => {
             const data = raw.toString();
@@ -74,7 +79,7 @@ export class AdminWsHandler extends EventEmitter {
             if (!msg)
                 return;
             if (msg.type === 'req') {
-                this.handleRequest(msg).then(response => {
+                this.handleRequest(msg, client).then(response => {
                     ws.send(JSON.stringify(response));
                 });
             }
@@ -86,8 +91,7 @@ export class AdminWsHandler extends EventEmitter {
             this.clients.delete(client);
         });
     }
-    /** 关闭所有连接（graceful shutdown 时调用） */
-    closeAll() {
+    async closeAll() {
         this.broadcastEvent('server.shutting_down', { reason: 'shutdown' });
         for (const client of this.clients) {
             client.ws.close();
@@ -95,7 +99,7 @@ export class AdminWsHandler extends EventEmitter {
         this.clients.clear();
     }
     // ============================================
-    // 消息处理
+    // 消息路由
     // ============================================
     parseMessage(data) {
         try {
@@ -110,252 +114,34 @@ export class AdminWsHandler extends EventEmitter {
             return null;
         }
     }
-    async handleRequest(req) {
+    async handleRequest(req, client) {
         const handler = this.handlers.get(req.method);
         if (!handler) {
             return createErrorResponse(req.id, 'NOT_FOUND', `Unknown method: ${req.method}`);
         }
         try {
-            return await handler(req.params, req.id);
+            return await handler(req.params, req.id, client.ws);
         }
         catch (error) {
             return createErrorResponse(req.id, 'INTERNAL', error instanceof Error ? error.message : 'Unknown error');
         }
     }
     // ============================================
-    // Config Handlers
-    // ============================================
-    handleConfigGet(_params, id) {
-        const config = this.loadConfig();
-        return createSuccessResponse(id, this.sensitizeConfig(config));
-    }
-    handleConfigUpdate(params, id) {
-        const updates = params;
-        const config = this.loadConfig();
-        if (updates.server)
-            Object.assign(config.server, updates.server);
-        if (updates.auth)
-            Object.assign(config.auth, updates.auth);
-        if (updates.provider)
-            Object.assign(config.provider, updates.provider);
-        if (updates.heartbeat)
-            Object.assign(config.heartbeat, updates.heartbeat);
-        this.saveConfig(config);
-        this.configCache = null; // 清除缓存
-        const changedKeys = Object.keys(updates);
-        this.broadcastEvent('config.changed', { keys: changedKeys });
-        return createSuccessResponse(id, { success: true });
-    }
-    handleGetProviderPresets(_params, id) {
-        if (!this.server) {
-            return createErrorResponse(id, 'INTERNAL', 'Server not initialized');
-        }
-        try {
-            const presets = this.server.agent.listPresets();
-            return createSuccessResponse(id, presets);
-        }
-        catch {
-            return createSuccessResponse(id, []);
-        }
-    }
-    // ============================================
-    // Status Handlers
+    // HandlerContext
     // ============================================
-    handleStatusGet(_params, id) {
-        const providerReady = this.isProviderReady();
-        const currentProvider = this.server?.agent.currentProvider;
-        const status = {
-            server: {
-                state: 'running',
-                port: this.getPort(),
-                uptime: Math.floor((Date.now() - this.startTime) / 1000),
-                version: this.getVersion(),
-            },
-            agent: {
-                initialized: !!this.server,
-                providerReady,
-                currentProvider: currentProvider?.id ?? null,
-                activePlugins: this.getActivePluginIds(),
-            },
-            system: {
-                memory: {
-                    rss: process.memoryUsage().rss,
-                    heapUsed: process.memoryUsage().heapUsed,
-                    heapTotal: process.memoryUsage().heapTotal,
-                },
-                nodeVersion: process.version,
-                platform: `${process.platform} ${process.arch}`,
-            },
+    createContext() {
+        return {
+            broadcastEvent: (event, data) => this.broadcastEvent(event, data),
+            broadcastLog: (entry) => this.broadcastLog(entry),
+            loadConfig: () => this.configStore.load(),
+            saveConfig: (config) => this.configStore.save(config),
+            sensitizeConfig: (config) => this.configStore.sensitize(config),
+            getServer: () => this.server,
+            getLogBuffer: () => this.logBuffer,
+            getHiveLogger: () => this.hiveLogger,
+            getClients: () => this.clients,
+            findClientByWs: (ws) => this.findClientByWs(ws),
         };
-        return createSuccessResponse(id, status);
-    }
-    handleServerRestart(_params, id) {
-        const response = createSuccessResponse(id, { success: true });
-        // 先广播事件，再退出
-        this.broadcastEvent('server.shutting_down', { reason: 'restart' });
-        setTimeout(() => {
-            process.exit(0);
-        }, 300);
-        return response;
-    }
-    handleGetProviders(_params, id) {
-        if (!this.server) {
-            return createErrorResponse(id, 'INTERNAL', 'Server not initialized');
-        }
-        const providers = this.server.agent.listProviders();
-        return createSuccessResponse(id, providers);
-    }
-    async handleProviderList(_params, id) {
-        if (!this.server) {
-            return createErrorResponse(id, 'INTERNAL', 'Server not initialized');
-        }
-        try {
-            const providers = await this.server.agent.listAllProviders();
-            return createSuccessResponse(id, providers.map(p => ({
-                id: p.id,
-                name: p.name,
-                logo: p.logo,
-                type: p.type,
-                defaultModel: p.models.length > 0 ? p.models[0].id : undefined,
-                modelCount: p.models.length,
-            })));
-        }
-        catch (error) {
-            return createErrorResponse(id, 'INTERNAL', error instanceof Error ? error.message : 'Failed to list providers');
-        }
-    }
-    async handleProviderGetModels(params, id) {
-        const { providerId } = params;
-        if (!providerId) {
-            return createErrorResponse(id, 'VALIDATION', 'providerId is required');
-        }
-        if (!this.server) {
-            return createErrorResponse(id, 'INTERNAL', 'Server not initialized');
-        }
-        try {
-            const models = await this.server.agent.listProviderModels(providerId);
-            return createSuccessResponse(id, models.map(m => ({
-                id: m.id,
-                name: m.name,
-                family: m.family,
-                contextWindow: m.contextWindow,
-                maxOutputTokens: m.maxOutputTokens,
-            })));
-        }
-        catch (error) {
-            return createErrorResponse(id, 'INTERNAL', error instanceof Error ? error.message : 'Failed to get models');
-        }
-    }
-    // ============================================
-    // Plugin Handlers
-    // ============================================
-    handlePluginList(_params, id) {
-        if (!this.server) {
-            return createSuccessResponse(id, []);
-        }
-        // 从已加载的插件获取信息
-        // 注意: 需要从 plugins.ts 的加载结果获取
-        const plugins = [];
-        return createSuccessResponse(id, plugins);
-    }
-    handlePluginInstall(params, id) {
-        const { source } = params;
-        if (!source || typeof source !== 'string') {
-            return createErrorResponse(id, 'VALIDATION', 'source is required');
-        }
-        try {
-            const pluginsDir = join(HIVE_HOME, 'plugins');
-            if (!existsSync(pluginsDir)) {
-                mkdirSync(pluginsDir, { recursive: true });
-            }
-            if (source.startsWith('https://') || source.startsWith('git+')) {
-                // Git install
-                this.installFromGit(source, pluginsDir);
-            }
-            else if (source.startsWith('.') || source.startsWith('/')) {
-                // Local path - copy
-                this.installFromLocal(source, pluginsDir);
-            }
-            else {
-                // npm install
-                this.installFromNpm(source, pluginsDir);
-            }
-            this.broadcastEvent('plugin.installed', { id: source, name: source, version: 'latest' });
-            return createSuccessResponse(id, { id: source, name: source, version: 'latest', enabled: true });
-        }
-        catch (error) {
-            return createErrorResponse(id, 'INTERNAL', error instanceof Error ? error.message : 'Installation failed');
-        }
-    }
-    handlePluginUninstall(params, id) {
-        const { id: pluginId } = params;
-        if (!pluginId) {
-            return createErrorResponse(id, 'VALIDATION', 'id is required');
-        }
-        try {
-            const pluginDir = join(HIVE_HOME, 'plugins', pluginId);
-            if (existsSync(pluginDir)) {
-                execSync(`rm -rf "${pluginDir}"`, { stdio: 'pipe' });
-            }
-            this.broadcastEvent('plugin.uninstalled', { id: pluginId });
-            return createSuccessResponse(id, { success: true });
-        }
-        catch (error) {
-            return createErrorResponse(id, 'INTERNAL', error instanceof Error ? error.message : 'Uninstall failed');
-        }
-    }
-    handlePluginUpdateConfig(params, id) {
-        const { id: pluginId, config } = params;
-        if (!pluginId || !config) {
-            return createErrorResponse(id, 'VALIDATION', 'id and config are required');
-        }
-        try {
-            // TODO: 更新 hive.config.json 中对应插件的配置
-            return createSuccessResponse(id, { success: true });
-        }
-        catch (error) {
-            return createErrorResponse(id, 'INTERNAL', error instanceof Error ? error.message : 'Config update failed');
-        }
-    }
-    // ============================================
-    // Log Handlers
-    // ============================================
-    handleLogGetHistory(params, id) {
-        const options = params;
-        const entries = this.logBuffer.query(options);
-        return createSuccessResponse(id, entries);
-    }
-    handleLogSubscribe(_params, id, _ws) {
-        // 标记当前客户端为 log subscriber
-        // 由于 handler 签名限制，通过 event 方式处理
-        this.emit('log:subscribe');
-        return createSuccessResponse(id, { success: true });
-    }
-    handleLogUnsubscribe(_params, id) {
-        this.emit('log:unsubscribe');
-        return createSuccessResponse(id, { success: true });
-    }
-    // ============================================
-    // Session Handlers
-    // ============================================
-    handleSessionList(_params, id) {
-        // Session 数据由 core 的 SessionCapability 管理
-        // 暂时返回空列表，后续通过 server 实例获取
-        return createSuccessResponse(id, []);
-    }
-    handleSessionGet(params, id) {
-        const { id: sessionId } = params;
-        if (!sessionId) {
-            return createErrorResponse(id, 'VALIDATION', 'id is required');
-        }
-        return createSuccessResponse(id, null);
-    }
-    handleSessionDelete(params, id) {
-        const { id: sessionId } = params;
-        if (!sessionId) {
-            return createErrorResponse(id, 'VALIDATION', 'id is required');
-        }
-        return createSuccessResponse(id, { success: true });
     }
     // ============================================
     // 事件广播
@@ -369,7 +155,6 @@ export class AdminWsHandler extends EventEmitter {
             }
         }
     }
-    /** 推送日志到所有已订阅的客户端 */
     broadcastLog(entry) {
         const msg = createEvent('log', entry);
         const payload = JSON.stringify(msg);
@@ -380,194 +165,20 @@ export class AdminWsHandler extends EventEmitter {
         }
     }
     // ============================================
-    // Console 拦截
-    // ============================================
-    interceptConsole() {
-        const origLog = console.log;
-        const origWarn = console.warn;
-        const origError = console.error;
-        const origDebug = console.debug;
-        const extractSource = (args) => {
-            const first = String(args[0] ?? '');
-            // 匹配 [server], [plugin:feishu], [agent] 等来源标记
-            const match = first.match(/^\[([^\]]+)\]/);
-            return match ? match[1] : 'server';
-        };
-        const formatMessage = (args) => {
-            return args.map(a => typeof a === 'object' ? JSON.stringify(a) : String(a)).join(' ');
-        };
-        console.log = (...args) => {
-            origLog.apply(console, args);
-            const entry = this.logBuffer.add('info', extractSource(args), formatMessage(args));
-            this.broadcastLog(entry);
-        };
-        console.warn = (...args) => {
-            origWarn.apply(console, args);
-            const entry = this.logBuffer.add('warn', extractSource(args), formatMessage(args));
-            this.broadcastLog(entry);
-        };
-        console.error = (...args) => {
-            origError.apply(console, args);
-            const entry = this.logBuffer.add('error', extractSource(args), formatMessage(args));
-            this.broadcastLog(entry);
-        };
-        console.debug = (...args) => {
-            origDebug.apply(console, args);
-            const entry = this.logBuffer.add('debug', extractSource(args), formatMessage(args));
-            this.broadcastLog(entry);
-        };
-    }
-    // ============================================
-    // 配置读写
+    // 辅助
     // ============================================
-    loadConfig() {
-        if (this.configCache)
-            return this.configCache;
-        const defaults = {
-            server: { port: 4450, host: '127.0.0.1', logLevel: 'info' },
-            auth: { enabled: false, apiKey: '' },
-            provider: { id: 'glm', apiKey: '', model: undefined },
-            heartbeat: { enabled: false, intervalMs: 300000 },
-        };
-        if (!existsSync(this.configPath)) {
-            this.configCache = defaults;
-            return defaults;
-        }
-        try {
-            const raw = JSON.parse(readFileSync(this.configPath, 'utf-8'));
-            const config = {
-                server: { ...defaults.server, ...raw.server },
-                auth: { ...defaults.auth, ...raw.auth },
-                provider: { ...defaults.provider, ...raw.provider },
-                heartbeat: { ...defaults.heartbeat, ...raw.heartbeat },
-            };
-            this.configCache = config;
-            return config;
-        }
-        catch {
-            this.configCache = defaults;
-            return defaults;
-        }
-    }
-    saveConfig(config) {
-        const dir = resolve(this.configPath, '..');
-        if (!existsSync(dir)) {
-            mkdirSync(dir, { recursive: true });
-        }
-        writeFileSync(this.configPath, JSON.stringify(config, null, 2), 'utf-8');
-        this.configCache = config;
-    }
-    /** 脱敏配置中的敏感字段 */
-    sensitizeConfig(config) {
-        return {
-            ...config,
-            auth: {
-                ...config.auth,
-                apiKey: this.sensitizeApiKey(config.auth.apiKey),
-            },
-            provider: {
-                ...config.provider,
-                apiKey: this.sensitizeApiKey(config.provider.apiKey),
-            },
-        };
-    }
-    sensitizeApiKey(key) {
-        if (!key || key.length <= 3)
-            return '***';
-        return `***${key.slice(-3)}`;
-    }
-    // ============================================
-    // 辅助方法
-    // ============================================
-    isProviderReady() {
-        const provider = this.server?.agent.currentProvider;
-        return !!provider?.apiKey && provider.apiKey.length > 0;
-    }
-    getPort() {
-        if (this.httpServer) {
-            const addr = this.httpServer.address();
-            if (typeof addr === 'object' && addr)
-                return addr.port;
-        }
-        return 4450;
-    }
-    getVersion() {
-        try {
-            const pkgPath = resolve(HIVE_HOME, 'package.json');
-            const pkg = JSON.parse(readFileSync(pkgPath, 'utf-8'));
-            return pkg.version ?? '0.0.0';
-        }
-        catch {
-            return '0.0.0';
-        }
-    }
-    getActivePluginIds() {
-        // 暂时返回空列表
-        return [];
-    }
-    // ============================================
-    // 插件安装
-    // ============================================
-    installFromNpm(source, pluginsDir) {
-        execSync(`npm install --production --prefix "${pluginsDir}" "${source}"`, {
-            stdio: 'pipe',
-            timeout: 60_000,
-        });
-    }
-    installFromGit(url, pluginsDir) {
-        const tmpDir = resolve(pluginsDir, '.tmp-install');
-        if (existsSync(tmpDir)) {
-            execSync(`rm -rf "${tmpDir}"`, { stdio: 'pipe' });
-        }
-        execSync(`git clone --depth 1 "${url}" "${tmpDir}"`, {
-            stdio: 'pipe',
-            timeout: 60_000,
-        });
-        // 验证 package.json
-        const pkgPath = resolve(tmpDir, 'package.json');
-        if (!existsSync(pkgPath)) {
-            execSync(`rm -rf "${tmpDir}"`, { stdio: 'pipe' });
-            throw new Error('Invalid plugin: no package.json found');
-        }
-        const pkg = JSON.parse(readFileSync(pkgPath, 'utf-8'));
-        if (!pkg.hive?.plugin) {
-            execSync(`rm -rf "${tmpDir}"`, { stdio: 'pipe' });
-            throw new Error('Invalid plugin: missing hive.plugin in package.json');
-        }
-        // 安装依赖
-        if (existsSync(resolve(tmpDir, 'package.json'))) {
-            execSync(`cd "${tmpDir}" && npm install --production`, {
-                stdio: 'pipe',
-                timeout: 60_000,
-            });
-        }
-        // 移动到 plugins 目录
-        const pluginName = pkg.name ?? url.split('/').pop() ?? 'unknown';
-        const targetDir = resolve(pluginsDir, pluginName);
-        execSync(`mv "${tmpDir}" "${targetDir}"`, { stdio: 'pipe' });
-    }
-    installFromLocal(source, pluginsDir) {
-        const resolvedSource = resolve(source);
-        if (!existsSync(resolvedSource)) {
-            throw new Error(`Path not found: ${source}`);
-        }
-        const pkgPath = resolve(resolvedSource, 'package.json');
-        if (!existsSync(pkgPath)) {
-            throw new Error('Invalid plugin: no package.json found');
-        }
-        const pkg = JSON.parse(readFileSync(pkgPath, 'utf-8'));
-        if (!pkg.hive?.plugin) {
-            throw new Error('Invalid plugin: missing hive.plugin in package.json');
+    findClientByWs(ws) {
+        for (const client of this.clients) {
+            if (client.ws === ws)
+                return client;
         }
-        const pluginName = pkg.name ?? 'local-plugin';
-        const targetDir = resolve(pluginsDir, pluginName);
-        execSync(`cp -r "${resolvedSource}" "${targetDir}"`, { stdio: 'pipe' });
+        return undefined;
     }
 }
 // ============================================
 // 工厂函数
 // ============================================
-export function createAdminWsHandler() {
-    return new AdminWsHandler();
+export function createAdminWsHandler(hiveLogger, logBuffer) {
+    return new AdminWsHandler(hiveLogger, logBuffer);
 }
 //# sourceMappingURL=admin-handler.js.map