kiku-agent 1.3.2 → 1.3.4

package/server/web-chat/router.js

@@ -1,157 +1,204 @@
-import { handleStaticRoute } from './routes/staticRoute.js';
-import { handleStatusRoute } from './routes/statusRoute.js';
-import { handleChatRoute } from './routes/chatRoute.js';
-import { handleProviderGetRoute, handleProviderUpdateRoute } from './routes/providerRoute.js';
-import { handleLanDiscoveryRoute, handleLanApplyRoute } from './routes/lanDiscoveryRoute.js';
-import { handleSuggestionsRoute, handleAiSuggestRoute } from './routes/suggestRoute.js';
-import { handleVerifyRoute } from './routes/verifyRoute.js';
-import { handleNewsRoute } from './routes/newsRoute.js';
-import { handleCompareRoute } from './routes/compareRoute.js';
-import { handleQuizRoute } from './routes/quizRoute.js';
-import { handleTimelineRoute } from './routes/timelineRoute.js';
-import { handleBilibiliCoverRoute } from './routes/bilibiliRoute.js';
-import { handleExportRoute } from './routes/exportRoute.js';
-import { handleClarifyRoute } from './routes/clarifyRoute.js';
-import { handleFeedbackRoute, handleFeedbackStatsRoute } from './routes/feedbackRoute.js';
-import { handleClarifyAnalyticsRoute, handleClarifyConfigRoute } from './routes/clarifyConfigRoute.js';
-import { handleSystemHardwareRoute, handleEstimateVramRoute } from './routes/vramRoute.js';
-import { applySecurityHeaders, checkChatRateLimitDistributed, resolveClientIp } from './core/security.js';
-import { sendError } from './core/errorHandler.js';
-import { handleApiRequest } from '../apiRouter.js';
-
-export async function handleWebChatRequest(req, res, options = {}) {
-  const port = Number(options.port || 3002);
-  const urlObj = new URL(req.url, `http://localhost:${port}`);
-  const pathname = urlObj.pathname;
-
-  const originAllowed = applySecurityHeaders(req, res);
-  if (!originAllowed) {
-    return sendError(res, 403, 'Origin is not allowed');
-  }
-
-  if (req.method === 'OPTIONS') {
-    res.writeHead(204);
-    res.end();
-    return;
-  }
-
-  if (pathname.startsWith('/api/')) {
-    const rate = await checkChatRateLimitDistributed(resolveClientIp(req));
-    if (!rate.allowed) {
-      res.setHeader('Retry-After', String(rate.retryAfter || 60));
-      return sendError(res, 429, `Rate limit exceeded. Please retry in ${rate.retryAfter || 60} seconds.`);
-    }
-  }
-
-  try {
-
-    if (handleStaticRoute(req, res, pathname, port)) {
-      return;
-    }
-
-    if (pathname === '/api/status') {
-      return await handleStatusRoute(req, res, urlObj, port);
-    }
-
-    if (pathname === '/api/provider' && req.method === 'GET') {
-      return await handleProviderGetRoute(req, res);
-    }
-
-    if (pathname === '/api/provider' && req.method === 'POST') {
-      return await handleProviderUpdateRoute(req, res);
-    }
-
-    if (pathname === '/api/lan/discovery' && req.method === 'GET') {
-      return await handleLanDiscoveryRoute(req, res);
-    }
-
-    if (pathname === '/api/lan/discovery/apply' && req.method === 'POST') {
-      return await handleLanApplyRoute(req, res);
-    }
-
-    if (pathname === '/api/suggestions') {
-      return await handleSuggestionsRoute(req, res);
-    }
-
-    if (pathname === '/api/ai-suggest' && req.method === 'POST') {
-      return await handleAiSuggestRoute(req, res);
-    }
-
-    if (pathname === '/api/quiz') {
-      return await handleQuizRoute(req, res);
-    }
-
-    if (pathname === '/api/v1/timeline') {
-      return await handleTimelineRoute(req, res, urlObj);
-    }
-
-    if (pathname === '/api/bilibili/cover') {
-      return await handleBilibiliCoverRoute(req, res, urlObj);
-    }
-
-    if (pathname === '/api/chat' && req.method === 'POST') {
-      return await handleChatRoute(req, res, urlObj);
-    }
-
-    if (pathname === '/api/v1/system/hardware' && req.method === 'GET') {
-      return await handleSystemHardwareRoute(req, res);
-    }
-
-    if (pathname === '/api/v1/models/estimate-vram' && (req.method === 'POST' || req.method === 'GET')) {
-      return await handleEstimateVramRoute(req, res, urlObj);
-    }
-
-    if (pathname === '/api/export' && req.method === 'POST') {
-      return await handleExportRoute(req, res);
-    }
-
-    if (pathname === '/api/verify' && req.method === 'POST') {
-      return await handleVerifyRoute(req, res);
-    }
-
-    if (pathname === '/api/news' && req.method === 'POST') {
-      return await handleNewsRoute(req, res);
-    }
-
-    if (pathname === '/api/compare' && req.method === 'POST') {
-      return await handleCompareRoute(req, res);
-    }
-
-    if (pathname === '/api/clarify' && req.method === 'POST') {
-      return await handleClarifyRoute(req, res);
-    }
-
-    if (pathname === '/api/feedback' && req.method === 'POST') {
-      return await handleFeedbackRoute(req, res);
-    }
-
-    if (pathname === '/api/feedback/stats' && req.method === 'GET') {
-      return await handleFeedbackStatsRoute(req, res);
-    }
-
-    if (pathname === '/api/clarify/analytics' && req.method === 'GET') {
-      return await handleClarifyAnalyticsRoute(req, res, options);
-    }
-
-    if (pathname === '/api/clarify/config' && req.method === 'GET') {
-      return await handleClarifyConfigRoute(req, res, options, 'get');
-    }
-
-    if (pathname === '/api/clarify/config' && req.method === 'POST') {
-      return await handleClarifyConfigRoute(req, res, options, 'update');
-    }
-
-    if (pathname === '/api/clarify/config/reset' && req.method === 'POST') {
-      return await handleClarifyConfigRoute(req, res, options, 'reset');
-    }
-
-    if (pathname.startsWith('/api/v1') || pathname === '/api-docs' || pathname === '/api/api-docs' || pathname === '/openapi.json') {
-      return await handleApiRequest(req, res);
-    }
-
-    res.writeHead(404, { 'Content-Type': 'application/json' });
-    res.end(JSON.stringify({ error: 'Not found' }));
-  } catch (err) {
-    sendError(res, 500, err.message);
-  }
-}
+import { handleStaticRoute } from './routes/staticRoute.js';
+import { handleStatusRoute } from './routes/statusRoute.js';
+import { handleChatRoute } from './routes/chatRoute.js';
+import { handleProviderGetRoute, handleProviderUpdateRoute } from './routes/providerRoute.js';
+import { handleLanDiscoveryRoute, handleLanInterfacesRoute, handleLanApplyRoute } from './routes/lanDiscoveryRoute.js';
+import { handleLmStudioLoadRoute, handleLmStudioLoadSettingsRoute } from './routes/lmStudioLoadRoute.js';
+import { handleEngineLoadRoute, handleEngineStatusRoute, handleEngineSettingsRoute, handleEngineStopRoute, handleEngineUnloadRoute } from './routes/engineRoute.js';
+import { handleSuggestionsRoute, handleAiSuggestRoute } from './routes/suggestRoute.js';
+import { handleVerifyRoute } from './routes/verifyRoute.js';
+import { handleNewsRoute } from './routes/newsRoute.js';
+import { handleCompareRoute } from './routes/compareRoute.js';
+import { handleQuizRoute } from './routes/quizRoute.js';
+import { handleTimelineRoute } from './routes/timelineRoute.js';
+import { handleBilibiliCoverRoute } from './routes/bilibiliRoute.js';
+import { handleExportRoute } from './routes/exportRoute.js';
+import { handleClarifyRoute } from './routes/clarifyRoute.js';
+import { handleFeedbackRoute, handleFeedbackStatsRoute } from './routes/feedbackRoute.js';
+import { handleMemoryGetRoute, handleMemoryActionRoute } from './routes/memoryRoute.js';
+import { handleClarifyAnalyticsRoute, handleClarifyConfigRoute } from './routes/clarifyConfigRoute.js';
+import { handleSystemHardwareRoute, handleEstimateVramRoute } from './routes/vramRoute.js';
+import { handleMcpStatusRoute } from './routes/mcpRoute.js';
+import { applySecurityHeaders, checkChatRateLimitDistributed, resolveClientIp } from './core/security.js';
+import { sendError } from './core/errorHandler.js';
+import { handleApiRequest } from '../apiRouter.js';
+
+export async function handleWebChatRequest(req, res, options = {}) {
+  const port = Number(options.port || 3002);
+  const urlObj = new URL(req.url, `http://localhost:${port}`);
+  const pathname = urlObj.pathname;
+
+  const originAllowed = applySecurityHeaders(req, res);
+  if (!originAllowed) {
+    return sendError(res, 403, 'Origin is not allowed');
+  }
+
+  if (req.method === 'OPTIONS') {
+    res.writeHead(204);
+    res.end();
+    return;
+  }
+
+  if (pathname.startsWith('/api/')) {
+    const [bucket, limit] = pathname === '/api/status'
+      ? (urlObj.searchParams.get('light') === '1' ? ['status-light', 120] : ['status-full', 20])
+      : pathname.startsWith('/api/lan/discovery') || pathname === '/api/lan/interfaces' ? ['lan-discovery', 6]
+      : pathname === '/api/provider' ? ['provider', 30]
+      : ['/api/lmstudio/load', '/api/engine/load', '/api/engine/unload', '/api/lmstudio/unload'].includes(pathname) ? ['engine-load', 6]
+      : ['/api/lmstudio/settings', '/api/engine/settings', '/api/engine/status'].includes(pathname) ? ['engine-settings', 30]
+      : ['/api/chat', '/api/ai-suggest'].includes(pathname) ? ['chat', 60] : ['default', 60];
+    const rate = await checkChatRateLimitDistributed(resolveClientIp(req), bucket, limit);
+    if (!rate.allowed) {
+      res.setHeader('Retry-After', String(rate.retryAfter || 60));
+      return sendError(res, 429, `Rate limit exceeded. Please retry in ${rate.retryAfter || 60} seconds.`);
+    }
+  }
+
+  try {
+
+    if (handleStaticRoute(req, res, pathname, port)) {
+      return;
+    }
+
+    if (pathname === '/api/status') {
+      return await handleStatusRoute(req, res, urlObj, port);
+    }
+
+    if (pathname === '/api/provider' && req.method === 'GET') {
+      return await handleProviderGetRoute(req, res);
+    }
+
+    if (pathname === '/api/provider' && req.method === 'POST') {
+      return await handleProviderUpdateRoute(req, res);
+    }
+
+    if (pathname === '/api/lan/discovery' && req.method === 'GET') {
+      return await handleLanDiscoveryRoute(req, res);
+    }
+
+    if (pathname === '/api/lan/interfaces' && req.method === 'GET') {
+      return await handleLanInterfacesRoute(req, res);
+    }
+
+    if (pathname === '/api/lan/discovery/apply' && req.method === 'POST') {
+      return await handleLanApplyRoute(req, res);
+    }
+
+    if ((pathname === '/api/engine/load' || pathname === '/api/lmstudio/load') && req.method === 'POST') {
+      return await handleEngineLoadRoute(req, res);
+    }
+
+    if (pathname === '/api/engine/stop' && req.method === 'POST') {
+      return await handleEngineStopRoute(req, res);
+    }
+
+    if ((pathname === '/api/engine/unload' || pathname === '/api/lmstudio/unload') && req.method === 'POST') {
+      return await handleEngineUnloadRoute(req, res);
+    }
+
+    if ((pathname === '/api/engine/settings' || pathname === '/api/lmstudio/settings') && (req.method === 'GET' || req.method === 'POST')) {
+      return await handleEngineSettingsRoute(req, res);
+    }
+
+    if (pathname === '/api/engine/status' && req.method === 'GET') {
+      return await handleEngineStatusRoute(req, res);
+    }
+
+    if (pathname === '/api/suggestions') {
+      return await handleSuggestionsRoute(req, res);
+    }
+
+    if (pathname === '/api/ai-suggest' && req.method === 'POST') {
+      return await handleAiSuggestRoute(req, res);
+    }
+
+    if (pathname === '/api/quiz') {
+      return await handleQuizRoute(req, res);
+    }
+
+    if (pathname === '/api/v1/timeline') {
+      return await handleTimelineRoute(req, res, urlObj);
+    }
+
+    if (pathname === '/api/bilibili/cover') {
+      return await handleBilibiliCoverRoute(req, res, urlObj);
+    }
+
+    if (pathname === '/api/chat' && req.method === 'POST') {
+      return await handleChatRoute(req, res, urlObj);
+    }
+
+    if (pathname === '/api/v1/system/hardware' && req.method === 'GET') {
+      return await handleSystemHardwareRoute(req, res);
+    }
+
+    if (pathname === '/api/v1/models/estimate-vram' && (req.method === 'POST' || req.method === 'GET')) {
+      return await handleEstimateVramRoute(req, res, urlObj);
+    }
+
+    if (pathname === '/api/export' && req.method === 'POST') {
+      return await handleExportRoute(req, res);
+    }
+
+    if (pathname === '/api/verify' && req.method === 'POST') {
+      return await handleVerifyRoute(req, res);
+    }
+
+    if (pathname === '/api/news' && req.method === 'POST') {
+      return await handleNewsRoute(req, res);
+    }
+
+    if (pathname === '/api/compare' && req.method === 'POST') {
+      return await handleCompareRoute(req, res);
+    }
+
+    if (pathname === '/api/clarify' && req.method === 'POST') {
+      return await handleClarifyRoute(req, res);
+    }
+
+    if (pathname === '/api/feedback' && req.method === 'POST') {
+      return await handleFeedbackRoute(req, res);
+    }
+
+    if (pathname === '/api/memory' && req.method === 'GET') {
+      return await handleMemoryGetRoute(req, res);
+    }
+
+    if (pathname === '/api/memory' && req.method === 'POST') {
+      return await handleMemoryActionRoute(req, res);
+    }
+
+    if (pathname === '/api/feedback/stats' && req.method === 'GET') {
+      return await handleFeedbackStatsRoute(req, res);
+    }
+
+    if (pathname === '/api/clarify/analytics' && req.method === 'GET') {
+      return await handleClarifyAnalyticsRoute(req, res, options);
+    }
+
+    if (pathname === '/api/clarify/config' && req.method === 'GET') {
+      return await handleClarifyConfigRoute(req, res, options, 'get');
+    }
+
+    if (pathname === '/api/clarify/config' && req.method === 'POST') {
+      return await handleClarifyConfigRoute(req, res, options, 'update');
+    }
+
+    if (pathname === '/api/clarify/config/reset' && req.method === 'POST') {
+      return await handleClarifyConfigRoute(req, res, options, 'reset');
+    }
+
+    if (pathname === '/api/mcp/status' || pathname === '/api/v1/mcp/status' || pathname === '/api/mcp/tools') {
+      return await handleMcpStatusRoute(req, res, options);
+    }
+
+    if (pathname === '/mcp' || pathname.startsWith('/mcp/') || pathname.startsWith('/api/v1') || pathname === '/api-docs' || pathname === '/api/api-docs' || pathname === '/openapi.json') {
+      return await handleApiRequest(req, res);
+    }
+
+    res.writeHead(404, { 'Content-Type': 'application/json' });
+    res.end(JSON.stringify({ error: 'Not found' }));
+  } catch (err) {
+    sendError(res, 500, err.message);
+  }
+}