ace-tool 0.1.0

package/dist/index.js

@@ -0,0 +1,102 @@
+#!/usr/bin/env node
+/**
+ * Ace Tool - MCP 服务器入口
+ * 代码库索引和语义搜索工具
+ */
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
+import { initConfig } from './config.js';
+import { logger } from './logger.js';
+import { SEARCH_CONTEXT_TOOL } from './prompts/searchContext.js';
+import { searchContextTool } from './tools/searchContext.js';
+/**
+ * 创建 MCP 服务器
+ */
+const server = new Server({
+    name: 'ace-tool',
+    version: '0.1.0',
+}, {
+    capabilities: {
+        tools: {},
+    },
+});
+/**
+ * 列出可用工具
+ */
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+    const tools = [
+        {
+            name: SEARCH_CONTEXT_TOOL.name,
+            description: SEARCH_CONTEXT_TOOL.description,
+            inputSchema: SEARCH_CONTEXT_TOOL.inputSchema,
+        },
+    ];
+    return { tools };
+});
+/**
+ * 处理工具调用
+ */
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    const { name, arguments: args } = request.params;
+    logger.info(`Tool called: ${name}, args: ${JSON.stringify(args)}`);
+    try {
+        if (name === 'search_context') {
+            const result = await searchContextTool(args);
+            logger.info(`Tool result: ${result?.text?.substring(0, 100)}...`);
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: result?.text || 'Error: No result returned',
+                    },
+                ],
+            };
+        }
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Unknown tool: ${name}`,
+                },
+            ],
+        };
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        logger.error(`Tool execution error: ${errorMessage}`);
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Error: ${errorMessage}`,
+                },
+            ],
+        };
+    }
+});
+/**
+ * 主函数
+ */
+async function main() {
+    try {
+        // 初始化配置
+        const config = initConfig();
+        logger.info('Starting ace-tool MCP server...');
+        logger.info(`API: ${config.baseUrl}`);
+        // 启动 MCP 服务器
+        const transport = new StdioServerTransport();
+        await server.connect(transport);
+        logger.info('MCP server connected via stdio');
+    }
+    catch (error) {
+        logger.exception('Server error', error);
+        process.exit(1);
+    }
+}
+// 运行主函数
+main().catch((error) => {
+    console.error('Fatal error:', error);
+    process.exit(1);
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA;;;GAGG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GAEvB,MAAM,oCAAoC,CAAC;AAE5C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAE7D;;GAEG;AACH,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;IACE,IAAI,EAAE,UAAU;IAChB,OAAO,EAAE,OAAO;CACjB,EACD;IACE,YAAY,EAAE;QACZ,KAAK,EAAE,EAAE;KACV;CACF,CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;IAC1D,MAAM,KAAK,GAAW;QACpB;YACE,IAAI,EAAE,mBAAmB,CAAC,IAAI;YAC9B,WAAW,EAAE,mBAAmB,CAAC,WAAW;YAC5C,WAAW,EAAE,mBAAmB,CAAC,WAAW;SAC7C;KACF,CAAC;IAEF,OAAO,EAAE,KAAK,EAAE,CAAC;AACnB,CAAC,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;IAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAEjD,MAAM,CAAC,IAAI,CAAC,gBAAgB,IAAI,WAAW,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEnE,IAAI,CAAC;QACH,IAAI,IAAI,KAAK,gBAAgB,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAsD,CAAC,CAAC;YAC/F,MAAM,CAAC,IAAI,CAAC,gBAAgB,MAAM,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;YAClE,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,MAAM,EAAE,IAAI,IAAI,2BAA2B;qBAClD;iBACF;aACF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iBAAiB,IAAI,EAAE;iBAC9B;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,MAAM,CAAC,KAAK,CAAC,yBAAyB,YAAY,EAAE,CAAC,CAAC;QACtD,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,UAAU,YAAY,EAAE;iBAC/B;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC,CAAC,CAAC;AAEH;;GAEG;AACH,KAAK,UAAU,IAAI;IACjB,IAAI,CAAC;QACH,QAAQ;QACR,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAE5B,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;QAC/C,MAAM,CAAC,IAAI,CAAC,QAAQ,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QAEtC,aAAa;QACb,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEhC,MAAM,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;IAChD,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QACxC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,QAAQ;AACR,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;IACrC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/logger.d.ts

@@ -0,0 +1,34 @@
+/**
+ * 日志模块 - 支持输出到项目 .ace-tool 目录
+ */
+type LogLevel = 'debug' | 'info' | 'warning' | 'error';
+export declare const logger: {
+    /**
+     * 启用日志
+     */
+    enable(): void;
+    /**
+     * 禁用日志
+     */
+    disable(): void;
+    /**
+     * 设置日志级别
+     */
+    setLevel(level: LogLevel): void;
+    /**
+     * 设置日志输出到项目的 .ace-tool 目录
+     * 只有在日志启用时才会创建文件
+     */
+    setProjectPath(projectRoot: string): void;
+    /**
+     * 关闭日志流
+     */
+    close(): void;
+    debug(message: string, ...args: unknown[]): void;
+    info(message: string, ...args: unknown[]): void;
+    warning(message: string, ...args: unknown[]): void;
+    error(message: string, ...args: unknown[]): void;
+    exception(message: string, error: unknown): void;
+};
+export {};
+//# sourceMappingURL=logger.d.ts.map

package/dist/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA;;GAEG;AAKH,KAAK,QAAQ,GAAG,OAAO,GAAG,MAAM,GAAG,SAAS,GAAG,OAAO,CAAC;AAiDvD,eAAO,MAAM,MAAM;IACjB;;OAEG;cACO,IAAI;IAId;;OAEG;eACQ,IAAI;IAIf;;OAEG;oBACa,QAAQ,GAAG,IAAI;IAI/B;;;OAGG;gCACyB,MAAM,GAAG,IAAI;IA4BzC;;OAEG;aACM,IAAI;mBAOE,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;kBAIlC,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;qBAI9B,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;mBAInC,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;uBAI7B,MAAM,SAAS,OAAO,GAAG,IAAI;CAQjD,CAAC"}

package/dist/logger.js

@@ -0,0 +1,116 @@
+/**
+ * 日志模块 - 支持输出到项目 .ace-tool 目录
+ */
+import fs from 'fs';
+import path from 'path';
+const LOG_LEVELS = {
+    debug: 0,
+    info: 1,
+    warning: 2,
+    error: 3
+};
+// 默认日志级别
+let currentLevel = 'info';
+// 日志是否启用（默认关闭）
+let enabled = false;
+// 日志文件路径
+let logFilePath = null;
+// 日志文件流
+let logStream = null;
+function formatTimestamp() {
+    return new Date().toISOString().replace('T', ' ').substring(0, 19);
+}
+function log(level, message, ...args) {
+    // 如果日志未启用，直接返回
+    if (!enabled) {
+        return;
+    }
+    if (LOG_LEVELS[level] < LOG_LEVELS[currentLevel]) {
+        return;
+    }
+    const timestamp = formatTimestamp();
+    const levelStr = level.toUpperCase().padEnd(7);
+    const formattedMessage = args.length > 0
+        ? `${message} ${args.map(a => JSON.stringify(a)).join(' ')}`
+        : message;
+    const logLine = `${timestamp} | ${levelStr} | ${formattedMessage}\n`;
+    // 如果有日志文件，写入文件
+    if (logStream) {
+        logStream.write(logLine);
+    }
+}
+export const logger = {
+    /**
+     * 启用日志
+     */
+    enable() {
+        enabled = true;
+    },
+    /**
+     * 禁用日志
+     */
+    disable() {
+        enabled = false;
+    },
+    /**
+     * 设置日志级别
+     */
+    setLevel(level) {
+        currentLevel = level;
+    },
+    /**
+     * 设置日志输出到项目的 .ace-tool 目录
+     * 只有在日志启用时才会创建文件
+     */
+    setProjectPath(projectRoot) {
+        // 如果日志未启用，不创建文件
+        if (!enabled) {
+            return;
+        }
+        const aceDir = path.join(projectRoot, '.ace-tool');
+        // 确保 .ace-tool 目录存在
+        if (!fs.existsSync(aceDir)) {
+            fs.mkdirSync(aceDir, { recursive: true });
+        }
+        logFilePath = path.join(aceDir, 'ace-tool.log');
+        // 关闭旧的流
+        if (logStream) {
+            logStream.end();
+        }
+        // 创建新的写入流（追加模式）
+        logStream = fs.createWriteStream(logFilePath, { flags: 'a' });
+        // 写入分隔线
+        const separator = `\n${'='.repeat(60)}\n${formatTimestamp()} | Session started\n${'='.repeat(60)}\n`;
+        logStream.write(separator);
+    },
+    /**
+     * 关闭日志流
+     */
+    close() {
+        if (logStream) {
+            logStream.end();
+            logStream = null;
+        }
+    },
+    debug(message, ...args) {
+        log('debug', message, ...args);
+    },
+    info(message, ...args) {
+        log('info', message, ...args);
+    },
+    warning(message, ...args) {
+        log('warning', message, ...args);
+    },
+    error(message, ...args) {
+        log('error', message, ...args);
+    },
+    exception(message, error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        const stack = error instanceof Error ? error.stack : '';
+        log('error', `${message}: ${errorMessage}`);
+        if (stack && logStream) {
+            logStream.write(stack + '\n');
+        }
+    }
+};
+//# sourceMappingURL=logger.js.map

package/dist/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAIxB,MAAM,UAAU,GAA6B;IAC3C,KAAK,EAAE,CAAC;IACR,IAAI,EAAE,CAAC;IACP,OAAO,EAAE,CAAC;IACV,KAAK,EAAE,CAAC;CACT,CAAC;AAEF,SAAS;AACT,IAAI,YAAY,GAAa,MAAM,CAAC;AAEpC,eAAe;AACf,IAAI,OAAO,GAAY,KAAK,CAAC;AAE7B,SAAS;AACT,IAAI,WAAW,GAAkB,IAAI,CAAC;AAEtC,QAAQ;AACR,IAAI,SAAS,GAA0B,IAAI,CAAC;AAE5C,SAAS,eAAe;IACtB,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACrE,CAAC;AAED,SAAS,GAAG,CAAC,KAAe,EAAE,OAAe,EAAE,GAAG,IAAe;IAC/D,eAAe;IACf,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;IACT,CAAC;IAED,IAAI,UAAU,CAAC,KAAK,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QACjD,OAAO;IACT,CAAC;IAED,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;IACpC,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAC/C,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC;QACtC,CAAC,CAAC,GAAG,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;QAC5D,CAAC,CAAC,OAAO,CAAC;IAEZ,MAAM,OAAO,GAAG,GAAG,SAAS,MAAM,QAAQ,MAAM,gBAAgB,IAAI,CAAC;IAErE,eAAe;IACf,IAAI,SAAS,EAAE,CAAC;QACd,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB;;OAEG;IACH,MAAM;QACJ,OAAO,GAAG,IAAI,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,OAAO;QACL,OAAO,GAAG,KAAK,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,KAAe;QACtB,YAAY,GAAG,KAAK,CAAC;IACvB,CAAC;IAED;;;OAGG;IACH,cAAc,CAAC,WAAmB;QAChC,gBAAgB;QAChB,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;QAEnD,oBAAoB;QACpB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,CAAC;QAED,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAEhD,QAAQ;QACR,IAAI,SAAS,EAAE,CAAC;YACd,SAAS,CAAC,GAAG,EAAE,CAAC;QAClB,CAAC;QAED,gBAAgB;QAChB,SAAS,GAAG,EAAE,CAAC,iBAAiB,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAE9D,QAAQ;QACR,MAAM,SAAS,GAAG,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,eAAe,EAAE,uBAAuB,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC;QACrG,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,SAAS,EAAE,CAAC;YACd,SAAS,CAAC,GAAG,EAAE,CAAC;YAChB,SAAS,GAAG,IAAI,CAAC;QACnB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,IAAI,CAAC,OAAe,EAAE,GAAG,IAAe;QACtC,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,CAAC,OAAe,EAAE,GAAG,IAAe;QACzC,GAAG,CAAC,SAAS,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,SAAS,CAAC,OAAe,EAAE,KAAc;QACvC,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,MAAM,KAAK,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACxD,GAAG,CAAC,OAAO,EAAE,GAAG,OAAO,KAAK,YAAY,EAAE,CAAC,CAAC;QAC5C,IAAI,KAAK,IAAI,SAAS,EAAE,CAAC;YACvB,SAAS,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;CACF,CAAC"}

package/dist/prompts/searchContext.d.ts

@@ -0,0 +1,22 @@
+/**
+ * search_context 工具的提示词定义
+ */
+export declare const SEARCH_CONTEXT_TOOL: {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: "object";
+        properties: {
+            project_root_path: {
+                type: string;
+                description: string;
+            };
+            query: {
+                type: string;
+                description: string;
+            };
+        };
+        required: string[];
+    };
+};
+//# sourceMappingURL=searchContext.d.ts.map

package/dist/prompts/searchContext.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"searchContext.d.ts","sourceRoot":"","sources":["../../src/prompts/searchContext.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;kBA8EgB,MAAM,EAAE;;CAEvD,CAAC"}

package/dist/prompts/searchContext.js

@@ -0,0 +1,83 @@
+/**
+ * search_context 工具的提示词定义
+ */
+export const SEARCH_CONTEXT_TOOL = {
+    name: 'search_context',
+    description: `IMPORTANT: This is the primary tool for searching the codebase. Please consider as the FIRST CHOICE for any codebase searches.
+
+This MCP tool is Augment's context engine, the world's best codebase context engine. It:
+1. Takes in a natural language description of the code you are looking for
+2. Uses a proprietary retrieval/embedding model suite that produces the highest-quality recall of relevant code snippets from across the codebase
+3. Maintains a real-time index of the codebase, so the results are always up-to-date and reflects the current state of the codebase
+4. Can retrieve across different programming languages
+5. Only reflects the current state of the codebase on the disk, and has no information on version control or code history
+
+## When to Use
+- When you don't know which files contain the information you need
+- When you want to gather high level information about the task you are trying to accomplish
+- When you want to gather information about the codebase in general
+
+## Good Query Examples
+- "Where is the function that handles user authentication?"
+- "What tests are there for the login functionality?"
+- "How is the database connected to the application?"
+
+## Bad Query Examples (use grep or file view instead)
+- "Find definition of constructor of class Foo" (use grep tool instead)
+- "Find all references to function bar" (use grep tool instead)
+- "Show me how Checkout class is used in services/payment.py" (use file view tool instead)
+- "Show context of the file foo.py" (use file view tool instead)
+
+ALWAYS use this tool when you're unsure of exact file locations. Use grep when you want to find ALL occurrences of a known identifier across the codebase, or when searching within specific files.
+
+## RULES
+
+### Tool Selection for Code Search
+CRITICAL: When searching for code, classes, functions, or understanding the codebase:
+- ALWAYS use this tool as your PRIMARY tool for code search
+- DO NOT use Bash commands (find, grep, ag, rg, etc.) or Grep tool for semantic code understanding
+- This tool uses advanced semantic search and is specifically designed for code understanding
+- Bash/Grep are only appropriate for exact string matching of non-code content (like error messages, config values, or log entries)
+- When in doubt between Bash/Grep and this tool, ALWAYS choose this tool
+
+### Preliminary Tasks and Planning
+Before starting to execute a task, ALWAYS use this tool to make sure you have a clear understanding of the task and the codebase.
+
+### Making Edits
+Before editing a file, ALWAYS first call this tool, asking for highly detailed information about the code you want to edit. Ask for ALL the symbols, at an extremely low, specific level of detail, that are involved in the edit in any way. Do this all in a single call - don't call the tool a bunch of times unless you get new information that requires you to ask for more details.
+
+For example:
+- If you want to call a method in another class, ask for information about the class and the method
+- If the edit involves an instance of a class, ask for information about the class
+- If the edit involves a property of a class, ask for information about the class and the property
+- If several of the above apply, ask for all of them in a single call
+- When in any doubt, include the symbol or object`,
+    inputSchema: {
+        type: 'object',
+        properties: {
+            project_root_path: {
+                type: 'string',
+                description: `Absolute path to the project root directory. Use forward slashes (/) as separators. Example: /Users/username/projects/myproject or C:/Users/username/projects/myproject`
+            },
+            query: {
+                type: 'string',
+                description: `Natural language description of the code you are looking for.
+
+Provide a clear description of the code behavior, workflow, or issue you want to locate. You may also add optional keywords to improve semantic matching.
+
+Recommended format: Natural language description + optional keywords
+
+Examples:
+- "I want to find where the server handles chunk merging in the file upload process. Keywords: upload chunk merge, file service"
+- "Locate where the system refreshes cached data after user permissions are updated. Keywords: permission update, cache refresh"
+- "Find the initialization flow of message queue consumers during startup. Keywords: mq consumer init, subscribe"
+- "Show me how configuration hot-reload is triggered and applied in the code. Keywords: config reload, hot update"
+- "Where is the function that handles user authentication?"
+- "What tests are there for the login functionality?"
+- "How is the database connected to the application?"`
+            }
+        },
+        required: ['project_root_path', 'query']
+    }
+};
+//# sourceMappingURL=searchContext.js.map

package/dist/prompts/searchContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"searchContext.js","sourceRoot":"","sources":["../../src/prompts/searchContext.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAG;IACjC,IAAI,EAAE,gBAAgB;IAEtB,WAAW,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;kDAgDmC;IAEhD,WAAW,EAAE;QACX,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,iBAAiB,EAAE;gBACjB,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yKAAyK;aACvL;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE;;;;;;;;;;;;;sDAaiC;aAC/C;SACF;QACD,QAAQ,EAAE,CAAC,mBAAmB,EAAE,OAAO,CAAa;KACrD;CACF,CAAC"}

package/dist/tools/searchContext.d.ts

@@ -0,0 +1,23 @@
+/**
+ * search_context 工具实现
+ */
+/**
+ * 工具参数接口
+ */
+interface SearchContextArgs {
+    project_root_path?: string;
+    query?: string;
+}
+/**
+ * 工具结果接口
+ */
+interface ToolResult {
+    type: string;
+    text: string;
+}
+/**
+ * search_context 工具实现
+ */
+export declare function searchContextTool(args: SearchContextArgs): Promise<ToolResult>;
+export {};
+//# sourceMappingURL=searchContext.d.ts.map

package/dist/tools/searchContext.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"searchContext.d.ts","sourceRoot":"","sources":["../../src/tools/searchContext.ts"],"names":[],"mappings":"AAAA;;GAEG;AAOH;;GAEG;AACH,UAAU,iBAAiB;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,UAAU,UAAU;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,wBAAsB,iBAAiB,CAAC,IAAI,EAAE,iBAAiB,GAAG,OAAO,CAAC,UAAU,CAAC,CAqDpF"}

package/dist/tools/searchContext.js

@@ -0,0 +1,49 @@
+/**
+ * search_context 工具实现
+ */
+import fs from 'fs';
+import { getConfig } from '../config.js';
+import { IndexManager } from '../index/manager.js';
+import { logger } from '../logger.js';
+/**
+ * search_context 工具实现
+ */
+export async function searchContextTool(args) {
+    try {
+        const query = args.query;
+        const projectRootPath = args.project_root_path;
+        if (!query) {
+            return { type: 'text', text: 'Error: query is required' };
+        }
+        if (!projectRootPath) {
+            return { type: 'text', text: 'Error: project_root_path is required' };
+        }
+        // 规范化路径（统一使用正斜杠）
+        const projectRoot = projectRootPath.replace(/\\/g, '/');
+        // 验证路径是否存在
+        if (!fs.existsSync(projectRoot)) {
+            return { type: 'text', text: `Error: Project path does not exist: ${projectRoot}` };
+        }
+        // 验证是否为目录
+        const stats = fs.statSync(projectRoot);
+        if (!stats.isDirectory()) {
+            return { type: 'text', text: `Error: Project path is not a directory: ${projectRoot}` };
+        }
+        const config = getConfig();
+        // 根据配置启用日志
+        if (config.enableLog) {
+            logger.enable();
+            logger.setProjectPath(projectRoot);
+        }
+        logger.info(`Tool invoked: search_context for project ${projectRoot}`);
+        const indexManager = new IndexManager(projectRoot, config.baseUrl, config.token, config.textExtensions, config.batchSize, config.maxLinesPerBlob, config.excludePatterns);
+        const result = await indexManager.searchContext(query);
+        return { type: 'text', text: result };
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        logger.exception('Error in search_context_tool', error);
+        return { type: 'text', text: `Error: ${errorMessage}` };
+    }
+}
+//# sourceMappingURL=searchContext.js.map

package/dist/tools/searchContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"searchContext.js","sourceRoot":"","sources":["../../src/tools/searchContext.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAkBtC;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,IAAuB;IAC7D,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAE/C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC;QAC5D,CAAC;QAED,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sCAAsC,EAAE,CAAC;QACxE,CAAC;QAED,iBAAiB;QACjB,MAAM,WAAW,GAAG,eAAe,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAExD,WAAW;QACX,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YAChC,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,uCAAuC,WAAW,EAAE,EAAE,CAAC;QACtF,CAAC;QAED,UAAU;QACV,MAAM,KAAK,GAAG,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;YACzB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,2CAA2C,WAAW,EAAE,EAAE,CAAC;QAC1F,CAAC;QAED,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAE3B,WAAW;QACX,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACrB,MAAM,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QACrC,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,4CAA4C,WAAW,EAAE,CAAC,CAAC;QACvE,MAAM,YAAY,GAAG,IAAI,YAAY,CACnC,WAAW,EACX,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,cAAc,EACrB,MAAM,CAAC,SAAS,EAChB,MAAM,CAAC,eAAe,EACtB,MAAM,CAAC,eAAe,CACvB,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAEvD,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IACxC,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,MAAM,CAAC,SAAS,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QACxD,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,YAAY,EAAE,EAAE,CAAC;IAC1D,CAAC;AACH,CAAC"}

package/dist/utils/projectDetector.d.ts

@@ -0,0 +1,32 @@
+/**
+ * 项目根目录检测工具
+ * 从当前工作目录向上查找 .ace-tool/ 或 .git/ 目录
+ */
+/**
+ * 检测项目根目录
+ * 优先级：.ace-tool/ > .git/
+ *
+ * @returns 项目根目录路径
+ * @throws 如果找不到项目根目录
+ */
+export declare function detectProjectRoot(): string;
+/**
+ * 获取项目的 .ace-tool 目录路径
+ * 如果不存在则创建
+ *
+ * @param projectRoot 项目根目录
+ * @returns .ace-tool 目录路径
+ */
+export declare function getAceDir(projectRoot: string): string;
+/**
+ * 获取索引文件路径
+ *
+ * @param projectRoot 项目根目录
+ * @returns index.json 文件路径
+ */
+export declare function getIndexFilePath(projectRoot: string): string;
+/**
+ * 规范化路径，统一使用正斜杠
+ */
+export declare function normalizePath(filePath: string): string;
+//# sourceMappingURL=projectDetector.d.ts.map

package/dist/utils/projectDetector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"projectDetector.d.ts","sourceRoot":"","sources":["../../src/utils/projectDetector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAMH;;;;;;GAMG;AACH,wBAAgB,iBAAiB,IAAI,MAAM,CAmC1C;AAED;;;;;;GAMG;AACH,wBAAgB,SAAS,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAYrD;AA+BD;;;;;GAKG;AACH,wBAAgB,gBAAgB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAG5D;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAEtD"}

package/dist/utils/projectDetector.js

@@ -0,0 +1,102 @@
+/**
+ * 项目根目录检测工具
+ * 从当前工作目录向上查找 .ace-tool/ 或 .git/ 目录
+ */
+import fs from 'fs';
+import path from 'path';
+import { logger } from '../logger.js';
+/**
+ * 检测项目根目录
+ * 优先级：.ace-tool/ > .git/
+ *
+ * @returns 项目根目录路径
+ * @throws 如果找不到项目根目录
+ */
+export function detectProjectRoot() {
+    const cwd = process.cwd();
+    logger.debug(`Detecting project root from: ${cwd}`);
+    let currentDir = cwd;
+    while (true) {
+        // 优先检查 .ace-tool 目录
+        const acePath = path.join(currentDir, '.ace-tool');
+        if (fs.existsSync(acePath) && fs.statSync(acePath).isDirectory()) {
+            logger.info(`Found project root (via .ace-tool): ${currentDir}`);
+            return currentDir;
+        }
+        // 其次检查 .git 目录
+        const gitPath = path.join(currentDir, '.git');
+        if (fs.existsSync(gitPath)) {
+            logger.info(`Found project root (via .git): ${currentDir}`);
+            return currentDir;
+        }
+        // 向上一级目录
+        const parentDir = path.dirname(currentDir);
+        // 已到达根目录
+        if (parentDir === currentDir) {
+            break;
+        }
+        currentDir = parentDir;
+    }
+    // 如果找不到 .ace-tool 或 .git，使用当前工作目录
+    logger.warning(`No .ace-tool or .git found, using current directory: ${cwd}`);
+    return cwd;
+}
+/**
+ * 获取项目的 .ace-tool 目录路径
+ * 如果不存在则创建
+ *
+ * @param projectRoot 项目根目录
+ * @returns .ace-tool 目录路径
+ */
+export function getAceDir(projectRoot) {
+    const aceDir = path.join(projectRoot, '.ace-tool');
+    if (!fs.existsSync(aceDir)) {
+        fs.mkdirSync(aceDir, { recursive: true });
+        logger.info(`Created .ace-tool directory: ${aceDir}`);
+        // 尝试将 .ace-tool 添加到 .gitignore
+        addToGitignore(projectRoot);
+    }
+    return aceDir;
+}
+/**
+ * 将 .ace-tool 添加到 .gitignore
+ */
+function addToGitignore(projectRoot) {
+    const gitignorePath = path.join(projectRoot, '.gitignore');
+    try {
+        let content = '';
+        if (fs.existsSync(gitignorePath)) {
+            content = fs.readFileSync(gitignorePath, 'utf-8');
+            // 检查是否已经包含 .ace-tool
+            if (content.includes('.ace-tool')) {
+                return;
+            }
+        }
+        // 添加 .ace-tool 到 .gitignore
+        const newContent = content.endsWith('\n') || content === ''
+            ? `${content}.ace-tool/\n`
+            : `${content}\n.ace-tool/\n`;
+        fs.writeFileSync(gitignorePath, newContent, 'utf-8');
+        logger.info('Added .ace-tool/ to .gitignore');
+    }
+    catch (error) {
+        logger.warning(`Failed to update .gitignore: ${error}`);
+    }
+}
+/**
+ * 获取索引文件路径
+ *
+ * @param projectRoot 项目根目录
+ * @returns index.json 文件路径
+ */
+export function getIndexFilePath(projectRoot) {
+    const aceDir = getAceDir(projectRoot);
+    return path.join(aceDir, 'index.json');
+}
+/**
+ * 规范化路径，统一使用正斜杠
+ */
+export function normalizePath(filePath) {
+    return path.resolve(filePath).replace(/\\/g, '/');
+}
+//# sourceMappingURL=projectDetector.js.map

package/dist/utils/projectDetector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"projectDetector.js","sourceRoot":"","sources":["../../src/utils/projectDetector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC;;;;;;GAMG;AACH,MAAM,UAAU,iBAAiB;IAC/B,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC1B,MAAM,CAAC,KAAK,CAAC,gCAAgC,GAAG,EAAE,CAAC,CAAC;IAEpD,IAAI,UAAU,GAAG,GAAG,CAAC;IAErB,OAAO,IAAI,EAAE,CAAC;QACZ,oBAAoB;QACpB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACnD,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC;YACjE,MAAM,CAAC,IAAI,CAAC,uCAAuC,UAAU,EAAE,CAAC,CAAC;YACjE,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,eAAe;QACf,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC9C,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,CAAC,kCAAkC,UAAU,EAAE,CAAC,CAAC;YAC5D,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,SAAS;QACT,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAE3C,SAAS;QACT,IAAI,SAAS,KAAK,UAAU,EAAE,CAAC;YAC7B,MAAM;QACR,CAAC;QAED,UAAU,GAAG,SAAS,CAAC;IACzB,CAAC;IAED,kCAAkC;IAClC,MAAM,CAAC,OAAO,CAAC,wDAAwD,GAAG,EAAE,CAAC,CAAC;IAC9E,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,SAAS,CAAC,WAAmB;IAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;IAEnD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,IAAI,CAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QAEtD,+BAA+B;QAC/B,cAAc,CAAC,WAAW,CAAC,CAAC;IAC9B,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,WAAmB;IACzC,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;IAE3D,IAAI,CAAC;QACH,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;YACjC,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;YAElD,qBAAqB;YACrB,IAAI,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gBAClC,OAAO;YACT,CAAC;QACH,CAAC;QAED,4BAA4B;QAC5B,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,OAAO,KAAK,EAAE;YACzD,CAAC,CAAC,GAAG,OAAO,cAAc;YAC1B,CAAC,CAAC,GAAG,OAAO,gBAAgB,CAAC;QAE/B,EAAE,CAAC,aAAa,CAAC,aAAa,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;IAChD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,OAAO,CAAC,gCAAgC,KAAK,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAAC,WAAmB;IAClD,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,CAAC;IACtC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;AACzC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,QAAgB;IAC5C,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;AACpD,CAAC"}

package/package.json

@@ -0,0 +1,48 @@
+{
+  "name": "ace-tool",
+  "version": "0.1.0",
+  "description": "MCP server for codebase indexing and semantic search",
+  "type": "module",
+  "main": "dist/index.js",
+  "bin": {
+    "ace-tool": "dist/index.js"
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "start": "node dist/index.js",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "mcp",
+    "codebase",
+    "indexing",
+    "semantic-search",
+    "ai",
+    "augment",
+    "context-engine",
+    "claude"
+  ],
+  "author": "xie",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/xie/ace-tool"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "axios": "^1.6.0",
+    "ignore": "^5.3.0",
+    "iconv-lite": "^0.6.3"
+  },
+  "devDependencies": {
+    "@types/node": "^20.10.0",
+    "typescript": "^5.3.0"
+  }
+}