@musashishao/agent-kit 1.4.0 → 1.5.0

package/.agent/mcp-gateway/src/index.ts

@@ -3,10 +3,7 @@
  * Agent Kit MCP Gateway Server
  * 
  * Provides AI agents with live access to project context, dependency graphs,
- * and semantic code search with automatic sync capabilities.
- * 
- * Usage:
- *   node dist/index.js [--project-root /path/to/project]
+ * and semantic search with automatic sync capabilities.
  */
 
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
@@ -31,14 +28,11 @@ const timestampChecker = new TimestampChecker(PROJECT_ROOT);
 // ============================================================================
 
 async function autoSyncIfNeeded(): Promise<{ synced: boolean; message: string }> {
-    // Check if data is stale
     const status = timestampChecker.getSyncStatus();
-
     if (!status.needsSync) {
         return { synced: false, message: "Data is fresh" };
     }
 
-    // Try to sync with debouncing
     const result = await debouncer.executeWithDebounce(async () => {
         return await syncer.syncAll();
     });
@@ -57,35 +51,17 @@ async function autoSyncIfNeeded(): Promise<{ synced: boolean; message: string }>
 function readJsonFile(filePath: string): any {
     try {
         const fullPath = path.resolve(PROJECT_ROOT, filePath);
-        if (!fs.existsSync(fullPath)) {
-            return null;
-        }
+        if (!fs.existsSync(fullPath)) return null;
         return JSON.parse(fs.readFileSync(fullPath, "utf-8"));
-    } catch (error) {
-        return null;
-    }
+    } catch { return null; }
 }
 
 function readTextFile(filePath: string): string | null {
     try {
         const fullPath = path.resolve(PROJECT_ROOT, filePath);
-        if (!fs.existsSync(fullPath)) {
-            return null;
-        }
+        if (!fs.existsSync(fullPath)) return null;
         return fs.readFileSync(fullPath, "utf-8");
-    } catch (error) {
-        return null;
-    }
-}
-
-function getFileModTime(filePath: string): number {
-    try {
-        const fullPath = path.resolve(PROJECT_ROOT, filePath);
-        const stats = fs.statSync(fullPath);
-        return stats.mtimeMs;
-    } catch {
-        return 0;
-    }
+    } catch { return null; }
 }
 
 // ============================================================================
@@ -94,68 +70,43 @@ function getFileModTime(filePath: string): number {
 
 const server = new McpServer({
     name: "agent-kit-gateway",
-    version: "1.0.0",
+    version: "1.1.0", // Universal Intelligence Version
 });
 
 // ============================================================================
-// Tool 1: get_project_context
+// Tool: get_project_context
 // ============================================================================
 
 server.tool(
     "get_project_context",
     "Get project overview including tech stack, structure, and conventions from AGENTS.md",
     {
-        section: z
-            .enum(["all", "tech_stack", "structure", "conventions", "commands"])
-            .optional()
-            .default("all")
-            .describe("Specific section to retrieve"),
+        section: z.enum(["all", "tech_stack", "structure", "conventions", "commands"]).optional().default("all")
     },
     async ({ section }) => {
         const agentsMd = readTextFile("AGENTS.md");
-
         if (!agentsMd) {
             return {
-                content: [
-                    {
-                        type: "text",
-                        text: JSON.stringify({
-                            error: "AGENTS.md not found",
-                            suggestion: "Run 'ak init' to generate AI infrastructure",
-                            project_root: PROJECT_ROOT,
-                        }),
-                    },
-                ],
+                content: [{ type: "text" as const, text: "AGENTS.md not found. Run 'ak init' first." }]
             };
         }
 
-        // Parse sections from AGENTS.md
         const sections: Record<string, string> = {};
-        const sectionRegex = /^## (.+)$/gm;
-        let matches;
-        let lastIndex = 0;
-        let lastSection = "";
-
         const lines = agentsMd.split("\n");
         let currentSection = "";
         let currentContent: string[] = [];
 
         for (const line of lines) {
             if (line.startsWith("## ")) {
-                if (currentSection) {
-                    sections[currentSection] = currentContent.join("\n").trim();
-                }
+                if (currentSection) sections[currentSection] = currentContent.join("\n").trim();
                 currentSection = line.replace("## ", "").trim();
                 currentContent = [];
             } else {
                 currentContent.push(line);
             }
         }
-        if (currentSection) {
-            sections[currentSection] = currentContent.join("\n").trim();
-        }
+        if (currentSection) sections[currentSection] = currentContent.join("\n").trim();
 
-        // Map section names to content
         const sectionMap: Record<string, string | undefined> = {
             tech_stack: sections["🛠️ Tech Stack"] || sections["Tech Stack"],
             structure: sections["📁 Directory Map"] || sections["Directory Map"],
@@ -164,427 +115,143 @@ server.tool(
             all: agentsMd,
         };
 
-        const result = sectionMap[section] || agentsMd;
-
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: result,
-                },
-            ],
-        };
+        return { content: [{ type: "text" as const, text: sectionMap[section] || agentsMd }] };
     }
 );
 
 // ============================================================================
-// Tool 2: analyze_dependencies
+// Tool: get_project_intelligence
 // ============================================================================
 
 server.tool(
-    "analyze_dependencies",
-    "Get dependency graph for a file or module. Shows imports and files that import this file. Auto-syncs if data is stale.",
-    {
-        file_path: z.string().describe("File path to analyze (relative to project root)"),
-        direction: z
-            .enum(["imports", "imported_by", "both"])
-            .optional()
-            .default("both")
-            .describe("Direction of dependencies to show"),
-        depth: z.number().optional().default(2).describe("Depth of traversal"),
-    },
-    async ({ file_path, direction, depth }) => {
-        // Auto-sync if data is stale
-        const syncStatus = await autoSyncIfNeeded();
-
+    "get_project_intelligence",
+    "Get high-level project summary: Code vs Docs distribution, primary types, and impact analysis.",
+    {},
+    async () => {
         const graph = readJsonFile(".agent/graph.json");
-
-        if (!graph) {
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: JSON.stringify({
-                            error: "Dependency graph not found",
-                            suggestion: "Run 'ak sync' to generate dependency graph",
-                        }),
-                    },
-                ],
-            };
-        }
-
-        // Normalize file path
-        const normalizedPath = file_path.replace(/^\.\//, "");
-
-        // Find the node
-        const node = graph.nodes?.find(
-            (n: any) => n.id === normalizedPath || n.path === normalizedPath
-        );
-
-        if (!node) {
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: JSON.stringify({
-                            error: `File not found in graph: ${file_path}`,
-                            available_files: graph.nodes?.slice(0, 10).map((n: any) => n.id),
-                        }),
-                    },
-                ],
-            };
-        }
-
-        // Build result based on direction
-        const result: any = {
-            file: normalizedPath,
-            type: node.type,
+        if (!graph) return { content: [{ type: "text" as const, text: "Project data not generated." }] };
+
+        const stats = graph.metadata?.file_types || {};
+        const total = graph.metadata?.total_files || 1;
+        const docCount = (stats.documentation || 0) + (stats.document || 0) + (stats.readme || 0);
+        const codeCount = (stats.component || 0) + (stats.module || 0) + (stats.utility || 0);
+
+        const intelligence = {
+            project_name: path.basename(PROJECT_ROOT),
+            project_type: docCount > codeCount ? "Documentation-Heavy" : "Code-Centric",
+            distribution: { docs: `${Math.round((docCount / total) * 100)}%`, code: `${Math.round((codeCount / total) * 100)}%` },
+            top_files: graph.nodes?.slice(0, 5).map((n: any) => n.id)
         };
 
-        if (direction === "imports" || direction === "both") {
-            result.imports = node.imports || [];
-        }
-
-        if (direction === "imported_by" || direction === "both") {
-            // Find files that import this file
-            const importedBy = graph.edges
-                ?.filter((e: any) => e.target === normalizedPath)
-                .map((e: any) => e.source) || [];
-            result.imported_by = importedBy;
-        }
-
-        // Calculate impact score
-        result.impact_score = (result.imported_by?.length || 0);
-
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: JSON.stringify(result, null, 2),
-                },
-            ],
-        };
+        return { content: [{ type: "text" as const, text: JSON.stringify(intelligence, null, 2) }] };
     }
 );
 
 // ============================================================================
-// Tool 3: search_code_logic
+// Tool: analyze_dependencies
 // ============================================================================
 
 server.tool(
-    "search_code_logic",
-    "Semantic search through code chunks with context. Auto-syncs if data is stale.",
+    "analyze_dependencies",
+    "Analyze imports/references for a file.",
     {
-        query: z.string().describe("Search query describing what you're looking for"),
-        file_filter: z.string().optional().describe("Glob pattern to filter files (e.g., '*.ts')"),
-        top_k: z.number().optional().default(10).describe("Number of results to return"),
+        file_path: z.string(),
+        direction: z.enum(["imports", "imported_by", "both"]).optional().default("both")
     },
-    async ({ query, file_filter, top_k }) => {
-        // Auto-sync if data is stale
-        const syncStatus = await autoSyncIfNeeded();
-
-        const chunks = readJsonFile(".agent/rag/chunks.json");
-
-        if (!chunks || !chunks.chunks) {
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: JSON.stringify({
-                            error: "RAG chunks not found",
-                            suggestion: "Run 'ak sync' to generate code chunks",
-                        }),
-                    },
-                ],
-            };
-        }
-
-        // Simple keyword-based search (in production, use embeddings)
-        const queryWords = query.toLowerCase().split(/\s+/);
+    async ({ file_path, direction }) => {
+        await autoSyncIfNeeded();
+        const graph = readJsonFile(".agent/graph.json");
+        if (!graph) return { content: [{ type: "text" as const, text: "Graph not found." }] };
 
-        const scored = chunks.chunks.map((chunk: any) => {
-            const content = (chunk.content || "").toLowerCase();
-            const metadata = chunk.metadata || {};
+        const normalizedPath = file_path.replace(/^\.\//, "");
+        const node = graph.nodes?.find((n: any) => n.id === normalizedPath);
+        if (!node) return { content: [{ type: "text" as const, text: "File not found in graph." }] };
 
-            // Apply file filter
-            if (file_filter) {
-                const pattern = file_filter.replace("*", ".*");
-                if (!new RegExp(pattern).test(metadata.file_path || "")) {
-                    return { chunk, score: -1 };
-                }
-            }
+        const isDoc = ["documentation", "readme", "document"].includes(node.type);
+        const outLabel = isDoc ? "references" : "imports";
+        const inLabel = isDoc ? "referenced_by" : "imported_by";
 
-            // Score based on keyword matches
-            let score = 0;
-            for (const word of queryWords) {
-                if (content.includes(word)) {
-                    score += 1;
-                }
-                if ((metadata.name || "").toLowerCase().includes(word)) {
-                    score += 2; // Boost for name matches
-                }
-            }
+        const result: any = { file: normalizedPath, type: node.type };
+        if (direction !== "imported_by") result[outLabel] = node.imports || [];
+        if (direction !== "imports") {
+            result[inLabel] = graph.edges?.filter((e: any) => e.target === normalizedPath).map((e: any) => e.source) || [];
+        }
 
-            return { chunk, score };
-        });
-
-        // Sort and take top k
-        const results = scored
-            .filter((s: any) => s.score > 0)
-            .sort((a: any, b: any) => b.score - a.score)
-            .slice(0, top_k)
-            .map((s: any) => ({
-                file: s.chunk.metadata?.file_path,
-                type: s.chunk.metadata?.chunk_type,
-                name: s.chunk.metadata?.name,
-                lines: `${s.chunk.metadata?.start_line}-${s.chunk.metadata?.end_line}`,
-                preview: s.chunk.content?.substring(0, 200) + "...",
-                relevance_score: s.score,
-            }));
-
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: JSON.stringify({
-                        query,
-                        total_results: results.length,
-                        results,
-                    }, null, 2),
-                },
-            ],
-        };
+        return { content: [{ type: "text" as const, text: JSON.stringify(result, null, 2) }] };
     }
 );
 
 // ============================================================================
-// Tool 4: get_impact_zone
+// Tool: search_knowledge (Search Engine)
 // ============================================================================
 
+async function handleSearch(args: { query: string, file_filter?: string, top_k: number }) {
+    await autoSyncIfNeeded();
+    const chunks = readJsonFile(".agent/rag/chunks.json");
+    if (!chunks?.chunks) return { content: [{ type: "text" as const, text: "No chunks found." }] };
+
+    const query = args.query.toLowerCase();
+    const results = chunks.chunks
+        .map((chunk: any) => {
+            let score = 0;
+            const content = chunk.content.toLowerCase();
+            if (content.includes(query)) score += 5;
+            if (chunk.metadata.name?.toLowerCase().includes(query)) score += 10;
+            return { chunk, score };
+        })
+        .filter((s: any) => s.score > 0)
+        .sort((a: any, b: any) => b.score - a.score)
+        .slice(0, args.top_k)
+        .map((s: any) => ({
+            file: s.chunk.metadata.file_path,
+            context: s.chunk.metadata.context_path || s.chunk.metadata.name,
+            preview: s.chunk.content.substring(0, 250) + "..."
+        }));
+
+    return { content: [{ type: "text" as const, text: JSON.stringify(results, null, 2) }] };
+}
+
 server.tool(
-    "get_impact_zone",
-    "Get all files that would be affected if a specific file is modified. Auto-syncs if data is stale.",
+    "search_knowledge",
+    "Universal search through project knowledge (Code + Docs).",
     {
-        file_path: z.string().describe("File path to analyze impact for"),
-        depth: z.number().optional().default(3).describe("How many levels of dependencies to traverse"),
+        query: z.string(),
+        file_filter: z.string().optional(),
+        top_k: z.number().optional().default(10)
     },
-    async ({ file_path, depth }) => {
-        // Auto-sync if data is stale
-        const syncStatus = await autoSyncIfNeeded();
-
-        const graph = readJsonFile(".agent/graph.json");
-
-        if (!graph) {
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: JSON.stringify({
-                            error: "Dependency graph not found",
-                            suggestion: "Run 'ak sync' to generate dependency graph",
-                        }),
-                    },
-                ],
-            };
-        }
-
-        const normalizedPath = file_path.replace(/^\.\//, "");
-
-        // Build adjacency list for reverse dependencies (who imports this?)
-        const reverseGraph: Record<string, string[]> = {};
-        for (const edge of graph.edges || []) {
-            if (!reverseGraph[edge.target]) {
-                reverseGraph[edge.target] = [];
-            }
-            reverseGraph[edge.target].push(edge.source);
-        }
-
-        // BFS to find all affected files
-        const visited = new Set<string>();
-        const queue: { file: string; level: number }[] = [{ file: normalizedPath, level: 0 }];
-        const impactByLevel: Record<number, string[]> = {};
-
-        while (queue.length > 0) {
-            const { file, level } = queue.shift()!;
-
-            if (visited.has(file) || level > depth) {
-                continue;
-            }
-            visited.add(file);
-
-            if (level > 0) {
-                if (!impactByLevel[level]) {
-                    impactByLevel[level] = [];
-                }
-                impactByLevel[level].push(file);
-            }
-
-            // Add files that import this file
-            const dependents = reverseGraph[file] || [];
-            for (const dep of dependents) {
-                if (!visited.has(dep)) {
-                    queue.push({ file: dep, level: level + 1 });
-                }
-            }
-        }
-
-        // Flatten results
-        const allAffected = Object.values(impactByLevel).flat();
-
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: JSON.stringify({
-                        source_file: normalizedPath,
-                        total_affected: allAffected.length,
-                        impact_by_level: impactByLevel,
-                        all_affected_files: allAffected,
-                        warning: allAffected.length > 10
-                            ? "High impact change! Review carefully."
-                            : null,
-                    }, null, 2),
-                },
-            ],
-        };
-    }
+    async (args) => await handleSearch(args)
 );
 
-// ============================================================================
-// Tool 5: force_sync
-// ============================================================================
-
 server.tool(
-    "force_sync",
-    "Force refresh all AI infrastructure data (AGENTS.md, graph, RAG chunks)",
+    "search_code_logic",
+    "Alias for search_knowledge.",
     {
-        target: z
-            .enum(["all", "graph", "rag", "agents_md"])
-            .optional()
-            .default("all")
-            .describe("What to sync"),
+        query: z.string(),
+        file_filter: z.string().optional(),
+        top_k: z.number().optional().default(10)
     },
-    async ({ target }) => {
-        const { spawn } = await import("child_process");
-        const results: Record<string, string> = {};
-
-        const runScript = (scriptPath: string, args: string[]): Promise<string> => {
-            return new Promise((resolve, reject) => {
-                const proc = spawn("python", [scriptPath, ...args], {
-                    cwd: PROJECT_ROOT,
-                });
-
-                let output = "";
-                let error = "";
-
-                proc.stdout.on("data", (data) => {
-                    output += data.toString();
-                });
-
-                proc.stderr.on("data", (data) => {
-                    error += data.toString();
-                });
-
-                proc.on("close", (code) => {
-                    if (code === 0) {
-                        resolve(output);
-                    } else {
-                        resolve(`Error: ${error || output}`);
-                    }
-                });
-
-                proc.on("error", (err) => {
-                    resolve(`Failed to run: ${err.message}`);
-                });
-            });
-        };
-
-        // Determine script paths (relative to kit installation)
-        const kitPath = process.env.AGENT_KIT_PATH || path.join(__dirname, "..", "..");
-
-        if (target === "all" || target === "graph") {
-            const graphScript = path.join(kitPath, "skills/graph-mapper/scripts/generate_graph.py");
-            if (fs.existsSync(graphScript)) {
-                results.graph = await runScript(graphScript, [
-                    "--src", path.join(PROJECT_ROOT, "src"),
-                    "--output", path.join(PROJECT_ROOT, ".agent/graph.json"),
-                    "--format", "both"
-                ]);
-            } else {
-                results.graph = "Graph script not found";
-            }
-        }
-
-        if (target === "all" || target === "rag") {
-            const ragScript = path.join(kitPath, "skills/rag-engineering/scripts/chunk_code.py");
-            if (fs.existsSync(ragScript)) {
-                results.rag = await runScript(ragScript, [
-                    "--src", path.join(PROJECT_ROOT, "src"),
-                    "--output", path.join(PROJECT_ROOT, ".agent/rag/chunks.json")
-                ]);
-            } else {
-                results.rag = "RAG script not found";
-            }
-        }
-
-        if (target === "all" || target === "agents_md") {
-            const infraScript = path.join(kitPath, "skills/app-builder/scripts/generate_ai_infra.py");
-            if (fs.existsSync(infraScript)) {
-                results.agents_md = await runScript(infraScript, [
-                    "--project-root", PROJECT_ROOT
-                ]);
-            } else {
-                results.agents_md = "AI infra script not found";
-            }
-        }
-
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: JSON.stringify({
-                        status: "Sync completed",
-                        target,
-                        results,
-                        timestamp: new Date().toISOString(),
-                    }, null, 2),
-                },
-            ],
-        };
-    }
+    async (args) => await handleSearch(args)
 );
 
 // ============================================================================
-// Resource: Project Status
+// Tool: get_impact_zone
 // ============================================================================
 
-server.resource(
-    "project-status",
-    "status://project",
-    async (uri) => {
-        const agentsMd = fs.existsSync(path.join(PROJECT_ROOT, "AGENTS.md"));
-        const graphJson = fs.existsSync(path.join(PROJECT_ROOT, ".agent/graph.json"));
-        const ragChunks = fs.existsSync(path.join(PROJECT_ROOT, ".agent/rag/chunks.json"));
-
-        const status = {
-            project_root: PROJECT_ROOT,
-            infrastructure: {
-                agents_md: agentsMd ? "✅ Present" : "❌ Missing",
-                graph: graphJson ? "✅ Present" : "❌ Missing",
-                rag: ragChunks ? "✅ Present" : "❌ Missing",
-            },
-            last_checked: new Date().toISOString(),
-        };
+server.tool(
+    "get_impact_zone",
+    "Find files affected by changing a file.",
+    { file_path: z.string() },
+    async ({ file_path }) => {
+        await autoSyncIfNeeded();
+        const graph = readJsonFile(".agent/graph.json");
+        if (!graph) return { content: [{ type: "text" as const, text: "Graph not found." }] };
 
-        return {
-            contents: [
-                {
-                    uri: uri.href,
-                    mimeType: "application/json",
-                    text: JSON.stringify(status, null, 2),
-                },
-            ],
-        };
+        const normalizedPath = file_path.replace(/^\.\//, "");
+        const affected = graph.edges
+            ?.filter((e: any) => e.target === normalizedPath)
+            .map((e: any) => e.source) || [];
+
+        return { content: [{ type: "text" as const, text: JSON.stringify({ file: normalizedPath, affected }, null, 2) }] };
     }
 );
 
@@ -593,16 +260,11 @@ server.resource(
 // ============================================================================
 
 async function main() {
-    console.error(`Agent Kit MCP Gateway starting...`);
-    console.error(`Project root: ${PROJECT_ROOT}`);
-
     const transport = new StdioServerTransport();
     await server.connect(transport);
-
-    console.error("MCP Gateway ready and listening on stdio");
 }
 
-main().catch((error) => {
-    console.error("Fatal error:", error);
+main().catch(err => {
+    console.error(err);
     process.exit(1);
 });