@atlisp/mcp 1.3.0 → 1.3.1

package/src/atlisp-mcp.js

@@ -1,630 +0,0 @@
-#!/usr/bin/env node
-import path from 'path';
-import { fileURLToPath } from 'url';
-import { createRequire } from 'module';
-
-const __dirname = path.dirname(fileURLToPath(import.meta.url));
-const require = createRequire(import.meta.url);
-const { version } = require(path.join(__dirname, '..', 'package.json'));
-
-const isMcpScript = process.argv[1]?.includes('atlisp-mcp');
-if (isMcpScript) {
-    const args = process.argv.slice(2);
-    for (let i = 0; i < args.length; i++) {
-        if (args[i] === '--version' || args[i] === '-v') {
-            console.log(`atlisp-mcp v${version}`);
-            process.exit(0);
-        } else if (args[i] === '--help' || args[i] === '-h') {
-            console.log(`
-@lisp MCP Server v${version}
-
-Usage: atlisp-mcp [options]
-
-Options:
-  -v, --version         Show version number
-  --transport <type>   Transport type: http or stdio (default: http)
-  --port <port>        HTTP port (default: 8110)
-  --host <host>        HTTP host (default: 0.0.0.0)
-  --stdio              Shorthand for --transport stdio
-  -h, --help           Show this help message
-
-Examples:
-  atlisp-mcp                       # Start HTTP server on port 8110
-  atlisp-mcp --port 3000           # Start HTTP server on port 3000
-  atlisp-mcp --stdio               # Start in stdio mode for MCP clients
-  atlisp-mcp --transport stdio     # Same as --stdio
-`);
-            process.exit(0);
-        } else if (args[i] === '--transport' && args[i + 1]) {
-            process.env.TRANSPORT = args[i + 1];
-            i++;
-        } else if (args[i] === '--stdio') {
-            process.env.TRANSPORT = 'stdio';
-        } else if (args[i] === '--port' && args[i + 1]) {
-            process.env.PORT = args[i + 1];
-            i++;
-        } else if (args[i] === '--host' && args[i + 1]) {
-            process.env.HOST = args[i + 1];
-            i++;
-        }
-    }
-}
-
-import express from 'express';
-import rawBody from 'raw-body';
-import iconv from 'iconv-lite';
-import rateLimit from 'express-rate-limit';
-import { randomUUID } from 'crypto';
-import { AsyncLocalStorage } from 'async_hooks';
-const { decode: charsetDecode, encodingExists } = iconv;
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { SessionTransport } from './session-transport.js';
-import {
-    ListToolsRequestSchema,
-    CallToolRequestSchema,
-    ListResourcesRequestSchema,
-    ReadResourceRequestSchema,
-    SubscribeRequestSchema,
-    UnsubscribeRequestSchema,
-    ListPromptsRequestSchema,
-    GetPromptRequestSchema,
-} from '@modelcontextprotocol/sdk/types.js';
-import { cad } from './cad.js';
-import { CAD_PLATFORMS, FILE_EXTENSIONS, PROTOCOL_VERSION, SERVER_NAME, SERVER_VERSION } from './constants.js';
-import { log, closeLog } from './logger.js';
-import * as cadHandlers from './handlers/cad-handlers.js';
-import * as packageHandlers from './handlers/package-handlers.js';
-import * as functionHandlers from './handlers/function-handlers.js';
-import { listResources, readResource } from './handlers/resource-handlers.js';
-import { listPrompts, getPrompt } from './handlers/prompt-handlers.js';
-import * as subMgr from './subscription-manager.js';
-import { sessionManager } from './subscription-manager.js';
-import { SseSessionManager } from './sse-session-manager.js';
-import { SseSession } from './sse-session.js';
-import config from './config.js';
-
-const mcpContext = new AsyncLocalStorage();
-const sseSessionManager = new SseSessionManager();
-
-const SERVER_CAPABILITIES = {
-    tools: {},
-    resources: { subscribe: true, listChanged: true },
-    prompts: {}
-};
-
-const SERVER_INFO = {
-    protocolVersion: PROTOCOL_VERSION,
-    capabilities: SERVER_CAPABILITIES,
-    serverInfo: { name: SERVER_NAME, version: SERVER_VERSION }
-};
-
-const { port, host, transport, apiKey, rateLimitWindow, rateLimitMax } = config;
-
-if (apiKey) log('API Key authentication enabled');
-
-const mcpLimiter = rateLimit({
-    windowMs: rateLimitWindow,
-    max: rateLimitMax,
-    standardHeaders: true,
-    legacyHeaders: false,
-    message: { error: '请求过于频繁，请稍后再试' }
-});
-
-export { CAD_PLATFORMS, FILE_EXTENSIONS, MOCK_PACKAGES } from './constants.js';
-
-const FUNCTION_LIB_URL = process.env.FUNCTION_LIB_URL || 'http://s3.atlisp.cn/json/functions.json';
-let cachedFunctionLib = null;
-let cachedAt = null;
-const CACHE_TTL = 5 * 60 * 1000;
-
-export async function loadAtlibFunctionLib() {
-    const now = Date.now();
-    if (cachedFunctionLib && cachedAt && (now - cachedAt) < CACHE_TTL) {
-        return cachedFunctionLib;
-    }
-    try {
-        const response = await fetch(FUNCTION_LIB_URL, { headers: { 'User-Agent': 'atlisp-mcp' } });
-        if (response.ok) {
-            const data = await response.json();
-            cachedFunctionLib = data;
-            cachedAt = now;
-            return data;
-        }
-    } catch (e) {
-        log(`loadAtlibFunctionLib error: ${e.message}`);
-    }
-    return null;
-}
-
-const TOOL_HANDLERS = {
-    connect_cad:           () => cadHandlers.connectCad(),
-    eval_lisp:             (a) => cadHandlers.evalLisp(a.code),
-    eval_lisp_with_result: (a) => cadHandlers.evalLisp(a.code, true, a.encoding),
-    get_cad_info:          () => cadHandlers.getCadInfo(),
-    list_packages:         () => packageHandlers.listPackages(),
-    search_packages:       (a) => packageHandlers.searchPackages(a.query),
-    install_package:       (a) => packageHandlers.installPackage(a.packageName),
-    get_platform_info:     () => ({ content: [{ type: 'text', text: `支持的 CAD 平台:\n${CAD_PLATFORMS.join('\n')}\n当前连接: ${cad.connected ? '已连接' : '未连接'}` }] }),
-    at_command:            (a) => cadHandlers.atCommand(a.command),
-    new_document:          () => cadHandlers.newDocument(),
-    bring_to_front:        () => cadHandlers.bringToFront(),
-    install_atlisp:        () => cadHandlers.installAtlisp(),
-    init_atlisp:          () => cadHandlers.initAtlisp(),
-    get_function_usage:    (a) => functionHandlers.getFunctionUsage(a.name, a.package),
-    list_symbols:          (a) => functionHandlers.listFunctions(a.package),
-    import_funlib: async (a) => {
-        const data = await loadAtlibFunctionLib();
-        if (!data) return { content: [{ type: 'text', text: '加载函数库失败' }], isError: true };
-        if (a.format === 'list') {
-            const items = data.functions?.map(f => `${f.name}: ${f.description || ''}`) || [];
-            return { content: [{ type: 'text', text: items.join('\n') }] };
-        }
-        return { content: [{ type: 'text', text: JSON.stringify(data, null, 2) }] };
-    },
-};
-
-function getOrCreateSessionServer(sessionId) {
-  let entry = sessionServers.get(sessionId);
-  if (!entry) {
-    const server = createMcpServer();
-    const transport = new SessionTransport(sessionId);
-    server.connect(transport);
-    entry = { server, transport };
-    sessionServers.set(sessionId, entry);
-    sessionTranports.set(sessionId, transport);
-  }
-  return entry;
-}
-
-function removeSessionServer(sessionId) {
-  const entry = sessionServers.get(sessionId);
-  if (entry) {
-    entry.transport.close().catch(() => {});
-    entry.server.close().catch(() => {});
-    sessionServers.delete(sessionId);
-    sessionTranports.delete(sessionId);
-  }
-}
-
-function broadcastToAllSessions(uri) {
-  for (const [sid, entry] of sessionServers) {
-    entry.server.sendResourceUpdated({ uri }).catch(err => {
-      log(`Failed to send resource update to session ${sid}: ${err.message}`);
-    });
-  }
-}
-
-export const tools = [
-    {
-	name: 'connect_cad',
-	description: '连接到 CAD (AutoCAD/ZWCAD/GStarCAD/BricsCAD)',
-	inputSchema: { type: 'object', properties: {} }
-    },
-    {
-	name: 'eval_lisp',
-	description: '在 CAD 中执行 AutoLISP 代码（不返回结果）',
-	inputSchema: {
-	    type: 'object',
-	    properties: { code: { type: 'string', description: '要执行的 LISP 代码' } },
-	    required: ['code']
-	}
-    },
-    {
-	name: 'eval_lisp_with_result',
-	description: '在 CAD 中执行 AutoLISP 代码并返回结果',
-	inputSchema: {
-	    type: 'object',
-	    properties: { 
-	        code: { type: 'string', description: '要执行的 LISP 代码' },
-	        encoding: { type: 'string', description: '结果编码，如 utf-8, gbk, gb2312, gb18030（可选，默认自动检测）' }
-	    },
-	    required: ['code']
-	}
-    },
-    {
-	name: 'get_cad_info',
-	description: '获取当前 CAD 信息',
-	inputSchema: { type: 'object', properties: {} }
-    },
-    {
-	name: 'list_packages',
-	description: '列出已安装的 @lisp 包',
-	inputSchema: { type: 'object', properties: {} }
-    },
-    {
-	name: 'search_packages',
-	description: '搜索 @lisp 包',
-	inputSchema: {
-	    type: 'object',
-	    properties: { query: { type: 'string', description: '搜索关键词' } },
-	    required: ['query']
-	}
-    },
-    {
-	name: 'install_package',
-	description: '安装 @lisp 包到 CAD',
-	inputSchema: {
-	    type: 'object',
-	    properties: { packageName: { type: 'string', description: '包名称' } },
-	    required: ['packageName']
-	}
-    },
-    {
-	name: 'get_platform_info',
-	description: '获取 CAD 平台信息',
-	inputSchema: { type: 'object', properties: {} }
-    },
-    {
-	name: 'at_command',
-	description: '执行 @lisp 命令',
-	inputSchema: {
-	    type: 'object',
-	    properties: { command: { type: 'string', description: '@lisp 命令' } },
-	    required: ['command']
-	}
-    },
-    {
-	name: 'new_document',
-	description: '在 CAD 中新建空白文档',
-	inputSchema: { type: 'object', properties: {} }
-    },
-    {
-	name: 'bring_to_front',
-	description: '将 CAD 窗口切换到前台（从后台运行状态唤醒）',
-	inputSchema: { type: 'object', properties: {} }
-    },
-    {
-	name: 'install_atlisp',
-	description: '在 CAD 中安装 @lisp',
-	inputSchema: { type: 'object', properties: {} }
-    },
-    {
-	name: 'init_atlisp',
-	description: '初始化 CAD 中的 @lisp 环境，加载函数库和初始变量',
-	inputSchema: { type: 'object', properties: {} }
-    },
-    {
-	name: 'get_function_usage',
-	description: '获取 @lisp 函数用法',
-	inputSchema: {
-	    type: 'object',
-	    properties: {
-		name: { type: 'string', description: '函数名，如 string:length' },
-		package: { type: 'string', description: '包名，如 string（可选，不传则自动搜索）' }
-	    },
-	    required: ['name']
-	}
-    },
-    {
-	name: 'list_symbols',
-	description: '列出 @lisp 所有符号，用于检查函数和变量',
-	inputSchema: {
-	    type: 'object',
-	    properties: {
-		package: { type: 'string', description: '包名，如 string（可选）' }
-	    }
-	}
-    },
-    {
-	name: 'import_funlib',
-	description: '从网络导入 @lisp 函数库数据到 AI Agent',
-	inputSchema: {
-	    type: 'object',
-	    properties: {
-		format: { type: 'string', enum: ['json', 'list'], description: '返回格式：json（完整JSON）或 list（函数名:描述 列表），默认 json' }
-	    }
-	}
-    }
-];
-
-export async function handleToolCall(name, args) {
-    const handler = TOOL_HANDLERS[name];
-    if (!handler) return { content: [{ type: 'text', text: `未知工具: ${name}` }], isError: true };
-    try {
-        const result = await handler(args || {});
-        notifyResourceChanges(name);
-        return result;
-    } catch (e) {
-        log(`Tool call error [${name}]: ${e.message}`);
-        return { content: [{ type: 'text', text: `工具执行错误: ${e.message}` }], isError: true };
-    }
-}
-
-function notifyResourceChanges(toolName) {
-    const resourceMap = {
-        connect_cad:    ['atlisp://cad/info'],
-        new_document:   ['atlisp://dwg/name', 'atlisp://dwg/path', 'atlisp://cad/layers', 'atlisp://cad/entities'],
-        install_package: ['atlisp://packages'],
-        install_atlisp: ['atlisp://packages'],
-    };
-    const uris = resourceMap[toolName];
-    if (uris) {
-        for (const uri of uris) {
-            if (subMgr.isSubscribed(uri)) {
-                subMgr.notify(uri, null);
-            }
-        }
-    }
-}
-
-async function initCadConnection() {
-    log('Attempting to connect to CAD on startup...');
-    try {
-	const connected = await cad.connect();
-	if (connected) {
-	    log('Auto-connected to CAD: ' + cad.getPlatform() + ' ' + cad.getVersion());
-	    console.error('已自动连接到 CAD: ' + cad.getPlatform() + ' ' + cad.getVersion());
-	} else {
-	    log('No CAD found on startup');
-	}
-    } catch (e) {
-	log('Auto-connect failed: ' + e.message);
-    }
-}
-
-const sessionServers = new Map();
-const sessionTranports = new Map();
-
-export function createMcpServer() {
-    const server = new Server(
-	{ name: SERVER_NAME, version: SERVER_VERSION },
-	{ capabilities: SERVER_CAPABILITIES }
-    );
-
-    server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools }));
-
-    server.setRequestHandler(CallToolRequestSchema, async (request) => {
-	const { name, arguments: args } = request.params;
-	return await handleToolCall(name, args || {});
-    });
-
-    server.setRequestHandler(ListResourcesRequestSchema, async (request) => {
-        const ctx = mcpContext.getStore();
-        const sessionId = ctx?.sessionId;
-        const subscribedUris = sessionId ? subMgr.list(sessionId) : subMgr.list();
-        return {
-            resources: await listResources(subscribedUris)
-        };
-    });
-
-    server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
-        const { uri } = request.params;
-        try {
-            const data = await readResource(uri);
-            return {
-                contents: [{
-                    uri,
-                    mimeType: 'application/json',
-                    text: JSON.stringify(data)
-                }]
-            };
-        } catch (e) {
-            return {
-                contents: [],
-                isError: true,
-                error: { code: -32603, message: e.message }
-            };
-        }
-    });
-
-    server.setRequestHandler(SubscribeRequestSchema, async (request) => {
-        const { uri } = request.params;
-        const ctx = mcpContext.getStore();
-        subMgr.subscribe(uri, ctx?.sessionId);
-        return { success: true };
-    });
-
-    server.setRequestHandler(UnsubscribeRequestSchema, async (request) => {
-        const { uri } = request.params;
-        const ctx = mcpContext.getStore();
-        subMgr.unsubscribe(uri, ctx?.sessionId);
-        return { success: true };
-    });
-
-    server.setRequestHandler(ListPromptsRequestSchema, async () => ({
-        prompts: await listPrompts()
-    }));
-
-    server.setRequestHandler(GetPromptRequestSchema, async (request) => {
-        const { name, arguments: args } = request.params;
-        return await getPrompt(name, args || {});
-    });
-
-    return server;
-}
-
-
-
-export async function startServer() {
-    if (config.transport === 'stdio') {
-        await initCadConnection();
-        const mcpServer = createMcpServer();
-        const transport = new StdioServerTransport();
-        const sessionId = 'stdio-session';
-        sessionManager.createSession(sessionId);
-        await mcpContext.run({ sessionId }, async () => {
-            await mcpServer.connect(transport);
-        });
-        console.error(`${SERVER_NAME} 运行在 stdio 模式`);
-
-        subMgr.setNotify((uri, data) => {
-            mcpServer.sendResourceUpdated({ uri }).catch(err => {
-                log(`Failed to send resource update: ${err.message}`);
-            });
-        });
-    } else {
-        const app = express();
-
-        const JSON_CONTENT_TYPES = ['application/json', 'application/json-rpc', 'application/vnd.api+json'];
-        app.use(async (req, res, next) => {
-            const ct = (req.headers['content-type'] || '').toLowerCase();
-            if (!JSON_CONTENT_TYPES.some(t => ct.startsWith(t))) return next();
-            try {
-                const buf = await rawBody(req, { limit: '10mb' });
-                const m = ct.match(/charset\s*=\s*([^\s;]+)/i);
-                let charset = m ? m[1].toLowerCase() : 'utf-8';
-                const charsetMap = { 'gb2312': 'gbk', 'gb18030': 'gbk', 'gbk': 'gbk', '936': 'gbk' };
-                charset = charsetMap[charset] || charset;
-                const str = encodingExists(charset) ? charsetDecode(buf, charset) : buf.toString('utf-8');
-                req.body = JSON.parse(str);
-                next();
-            } catch (e) {
-                res.status(400).json({ error: 'Invalid JSON body', message: e.message });
-            }
-        });
-
-        app.use((req, res, next) => {
-            log(`${req.method} ${req.path}`);
-            next();
-        });
-
-        const PUBLIC_PATHS = ['/health'];
-        if (apiKey) {
-            app.use((req, res, next) => {
-                if (PUBLIC_PATHS.includes(req.path)) return next();
-                const auth = req.get('Authorization');
-                if (auth === `Bearer ${apiKey}`) return next();
-                res.status(401).json({ error: 'Unauthorized' });
-            });
-        }
-
-        app.get('/health', (req, res) => {
-            res.json({ status: 'ok', cad: cad.connected ? 'connected' : 'disconnected' });
-        });
-
-        if (config.enableCors) {
-            app.use((req, res, next) => {
-                res.setHeader('Access-Control-Allow-Origin', config.corsOrigin);
-                res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS, CONNECT');
-                res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization, Accept, mcp-session-id, last-event-id');
-                res.setHeader('Access-Control-Expose-Headers', 'mcp-session-id, content-type, x-accel-buffering');
-                res.setHeader('Access-Control-Max-Age', '86400');
-                res.setHeader('Vary', 'Accept');
-                if (req.method === 'OPTIONS') return res.status(204).end();
-                next();
-            });
-        }
-
-        subMgr.setNotify((uri, data) => {
-            broadcastToAllSessions(uri);
-            for (const session of sseSessionManager.listActive()) {
-                sseSessionManager.sendToClient(session.clientId, 'resource_updated', { uri });
-            }
-        });
-
-        async function handleMcpPost(req, res) {
-            const sessionId = req.headers['mcp-session-id'] || req.query.sessionId || randomUUID();
-            sessionManager.createSession(sessionId);
-            const body = req.body;
-            if (!body || !body.method) {
-                return res.status(400).json({ error: 'Invalid request' });
-            }
-
-            const { server, transport } = getOrCreateSessionServer(sessionId);
-            transport.resetCleanupTimer();
-
-            await mcpContext.run({ sessionId }, async () => {
-                try {
-                    const response = await transport.handleRequest(body);
-                    if (response) {
-                        const headers = { 'Content-Type': 'application/json' };
-                        if (transport.sessionId) {
-                            headers['mcp-session-id'] = transport.sessionId;
-                        }
-                        res.writeHead(200, headers);
-                        res.end(JSON.stringify(response));
-                    } else {
-                        if (!res.headersSent) {
-                            res.status(202).end();
-                        }
-                    }
-                } catch (error) {
-                    log(`Transport error [${sessionId}]: ${error.message}`);
-                    if (!res.headersSent) {
-                        res.status(500).json({ error: 'Internal server error', message: error.message });
-                    }
-                }
-            });
-        }
-
-        app.all('/mcp', mcpLimiter, handleMcpPost);
-        app.all('/mcp/:path', mcpLimiter, handleMcpPost);
-        app.get('/mcp/stream', mcpLimiter, handleMcpPost);
-
-        const sseEndpoint = '/sse';
-        app.get(sseEndpoint, async (req, res) => {
-            const sessionId = randomUUID();
-            sessionManager.createSession(sessionId);
-            getOrCreateSessionServer(sessionId);
-
-            const sseSession = new SseSession(res, { clientId: sessionId });
-            sseSessionManager.add(sessionId, sseSession);
-
-            sseSession.sendEndpoint(`/message?sessionId=${sessionId}`);
-
-            req.on('close', () => {
-                sseSessionManager.remove(sessionId);
-                removeSessionServer(sessionId);
-            });
-        });
-
-        app.post('/message', mcpLimiter, async (req, res) => {
-            const sessionId = req.query.sessionId;
-            if (!sessionId) {
-                return res.status(400).json({ error: 'Missing sessionId' });
-            }
-            const sseSession = sseSessionManager.get(sessionId);
-            if (!sseSession) {
-                return res.status(404).json({ error: 'Session not found' });
-            }
-            sessionManager.createSession(sessionId);
-            const { server, transport } = getOrCreateSessionServer(sessionId);
-            transport.resetCleanupTimer();
-
-            const body = req.body;
-            if (body && body.method) {
-                try {
-                    const response = await transport.handleRequest(body);
-                    if (response) {
-                        sseSession.sendResponse(response);
-                    }
-                    if (!res.headersSent) {
-                        res.status(202).end();
-                    }
-                } catch (error) {
-                    log(`SSE message error [${sessionId}]: ${error.message}`);
-                    if (!res.headersSent) {
-                        res.status(500).json({ error: 'Internal server error' });
-                    }
-                }
-            } else {
-                res.status(400).json({ error: 'Invalid request' });
-            }
-        });
-
-        const httpServer = app.listen(config.port, config.host, async () => {
-            await initCadConnection();
-            console.error(`@lisp MCP Server 启动 - http://${config.host}:${config.port} (多会话模式)`);
-        });
-
-        async function shutdown(signal) {
-            console.error(`\n收到 ${signal}，正在优雅关闭...`);
-            sseSessionManager.closeAll();
-            sessionManager.clear();
-            for (const [sid] of sessionServers) {
-                removeSessionServer(sid);
-            }
-            await cad.disconnect();
-            closeLog();
-            httpServer.close(() => process.exit(0));
-            setTimeout(() => process.exit(1), 5000);
-        }
-        process.on('SIGINT', () => shutdown('SIGINT'));
-        process.on('SIGTERM', () => shutdown('SIGTERM'));
-    }
-}
-
-if (!process.env.VITEST) {
-    await startServer();
-}