freeform-modeling-mcp 1.0.29 → 1.0.30

package/.cursor/mcp.json

@@ -0,0 +1,26 @@
+{
+    "mcpServers": {
+        "freeform-modeling-mcp": {
+            "command": "npx",
+            "args": [
+                "tsx",
+                "src/cli.ts",
+                "start",
+                "--stdio"
+            ],
+            "cwd": "D:\\wain\\mcp-server"
+        },
+        "aholo-sdk-mcp": {
+            "type": "stdio",
+            "command": "npx",
+            "args": [
+                "-y",
+                "@manycore/aholo-sdk-mcp"
+            ],
+            "env": {
+                "AHOLO_API_KEY": "urSuPSeAcEnNEEn",
+                "AHOLO_REGION": "cn"
+            }
+        }
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "freeform-modeling-mcp",
-  "version": "1.0.29",
+  "version": "1.0.30",
   "description": "KooMaster MCP Server with WebSocket Bridge",
   "main": "index.cjs",
   "type": "module",

package/src/commands/start.ts

@@ -18,7 +18,7 @@ export async function startMCPServer(options: { dev?: boolean, stdio?: boolean }
         const wsBridge = new WebSocketBridge();
         await wsBridge.initialize();
         const wsPort = wsBridge.getPort();
-        const wsUrl = `ws://localhost:${wsPort}`;
+        const wsUrl = `ws://127.0.0.1:${wsPort}`;
 
         await llmClient.connect(wsUrl);
 

package/src/lib/wsbridge/bridge.ts

@@ -6,6 +6,25 @@ import * as http from 'http';
 import { BridgeLockManager, getActiveBridges } from './lock';
 import logger from '../logger';
 
+function isAllowedOrigin(origin: string): boolean {
+    try {
+        const { hostname } = new URL(origin);
+        return hostname === 'localhost' || hostname === '127.0.0.1'
+            || hostname === 'kujiale.com' || hostname.endsWith('.kujiale.com')
+            || hostname === 'coohom.com' || hostname.endsWith('.coohom.com');
+    } catch {
+        return false;
+    }
+}
+
+function setCors(req: http.IncomingMessage, res: http.ServerResponse): void {
+    const origin = req.headers.origin;
+    if (origin && isAllowedOrigin(origin)) {
+        res.setHeader('Access-Control-Allow-Origin', origin);
+        res.setHeader('Vary', 'Origin');
+    }
+}
+
 // 客户端类型
 export enum ClientType {
     MCP = 'mcp',
@@ -101,14 +120,21 @@ export class WebSocketBridge extends EventEmitter {
             // 使用锁管理器获取可用端口
             this.port = await this.lockManager.initialize();
 
-            // 创建 HTTP 服务器
+            // 创建 HTTP 服务器（与 WebSocket 共用端口；/bridge-status 供浏览器发现）
             this.httpServer = http.createServer((req, res) => {
-                // 处理HTTP请求
+                if (req.method === 'OPTIONS') {
+                    setCors(req, res);
+                    res.writeHead(204);
+                    res.end();
+                    return;
+                }
                 if (req.url === '/bridge-status') {
+                    setCors(req, res);
                     res.writeHead(200, { 'Content-Type': 'application/json' });
                     res.end(JSON.stringify({
                         service: 'WebSocketBridge',
                         version: '1.0.0',
+                        port: this.port,
                         status: this._status,
                         clients: {
                             total: this.clients.size,
@@ -122,8 +148,10 @@ export class WebSocketBridge extends EventEmitter {
                 }
             });
 
-            // 创建 WebSocket 服务器
-            this.server = new WebSocketServer({ server: this.httpServer });
+            this.server = new WebSocketServer({
+                server: this.httpServer,
+                verifyClient: ({ origin }) => !origin || isAllowedOrigin(origin),
+            });
 
             // 设置事件处理程序
             this.setupEventHandlers();
@@ -184,8 +212,7 @@ export class WebSocketBridge extends EventEmitter {
                     }
                 });
 
-                // 启动 HTTP 服务器
-                this.httpServer!.listen(this.port);
+                this.httpServer!.listen(this.port, '127.0.0.1');
             } catch (error) {
                 logger.error('Failed to start WebSocketBridge:', error);
                 reject(error);

package/src/lib/wsbridge/lock.ts

@@ -75,7 +75,7 @@ function isProcessRunning(pid: number): boolean {
 // 检查特定端口是否有WebSocketBridge服务响应
 async function checkBridgeAtPort(port: number): Promise<boolean> {
     return new Promise((resolve) => {
-        const req = http.get(`http://localhost:${port}/bridge-status`, (res) => {
+        const req = http.get(`http://127.0.0.1:${port}/bridge-status`, (res) => {
             // 收到响应，检查是否是我们的服务
             let data = '';
             res.on('data', (chunk) => {
@@ -186,7 +186,7 @@ export class BridgeLockManager extends EventEmitter {
         if (!portAvailable) {
             logger.info(`Port ${this.port} is not available, searching for another port...`);
             try {
-                this.port = await findAvailablePort(this.port + 1);
+                this.port = await findAvailablePort(this.port + 1, 10);
                 logger.info(`Found available port: ${this.port}`);
             } catch (e) {
                 logger.error(`Cannot find available port:`, e.message);