codemap-ai 3.2.0 → 3.4.0

package/dist/{flow-server-UQEOUP2C.js → flow-server-4XSR5P4N.js}

@@ -29,7 +29,7 @@ function getStorage() {
 var server = new Server(
   {
     name: "codemap-flow",
-    version: "3.2.0"
+    version: "3.4.0"
   },
   {
     capabilities: {
@@ -243,4 +243,4 @@ server.connect(transport).catch((error) => {
   console.error("Failed to start MCP server:", error);
   process.exit(1);
 });
-//# sourceMappingURL=flow-server-UQEOUP2C.js.map
+//# sourceMappingURL=flow-server-4XSR5P4N.js.map

package/dist/{flow-server-UQEOUP2C.js.map → flow-server-4XSR5P4N.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../src/mcp/flow-server.ts"],"sourcesContent":["/**\n * MCP Server for CodeMap Flow Edition\n * Provides 3 focused tools: impact analysis, flow tracing, and caller lookup\n */\n\nimport { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\nimport { StdioServerTransport } from \"@modelcontextprotocol/sdk/server/stdio.js\";\nimport {\n  CallToolRequestSchema,\n  ListToolsRequestSchema,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { FlowStorage } from \"../flow/storage.js\";\nimport { resolve } from \"path\";\nimport { existsSync } from \"fs\";\n\n// Get config from environment\nconst DB_PATH = process.env.CODEMAP_DB_PATH || \".codemap/graph.db\";\nconst PROJECT_ROOT = process.env.CODEMAP_PROJECT_ROOT || process.cwd();\n\nlet storage: FlowStorage | null = null;\n\nfunction getStorage(): FlowStorage {\n  if (!storage) {\n    const dbPath = resolve(PROJECT_ROOT, DB_PATH);\n    if (!existsSync(dbPath)) {\n      throw new Error(`CodeMap database not found at ${dbPath}. Run 'codemap index' first.`);\n    }\n    storage = new FlowStorage(dbPath);\n  }\n  return storage;\n}\n\nconst server = new Server(\n  {\n    name: \"codemap-flow\",\n    version: \"3.2.0\",\n  },\n  {\n    capabilities: {\n      tools: {},\n    },\n  }\n);\n\n// ============ Tools ============\n\nserver.setRequestHandler(ListToolsRequestSchema, async () => {\n  return {\n    tools: [\n      {\n        name: \"codemap_impact\",\n        description: \"Analyze what would break if you change a function/class. Shows all callers and affected files with risk assessment.\",\n        inputSchema: {\n          type: \"object\",\n          properties: {\n            target: {\n              type: \"string\",\n              description: \"Function or class name to analyze\",\n            },\n            depth: {\n              type: \"number\",\n              description: \"How many levels deep to analyze (default: 3)\",\n              default: 3,\n            },\n          },\n          required: [\"target\"],\n        },\n      },\n      {\n        name: \"codemap_trace_flow\",\n        description: \"Trace the execution flow between two functions. Shows the complete call path from source to target.\",\n        inputSchema: {\n          type: \"object\",\n          properties: {\n            from: {\n              type: \"string\",\n              description: \"Starting function name\",\n            },\n            to: {\n              type: \"string\",\n              description: \"Target function name\",\n            },\n          },\n          required: [\"from\", \"to\"],\n        },\n      },\n      {\n        name: \"codemap_callers\",\n        description: \"Find all functions that call a specific function. Shows where and how a function is used.\",\n        inputSchema: {\n          type: \"object\",\n          properties: {\n            function: {\n              type: \"string\",\n              description: \"Function name to find callers for\",\n            },\n          },\n          required: [\"function\"],\n        },\n      },\n    ],\n  };\n});\n\nserver.setRequestHandler(CallToolRequestSchema, async (request) => {\n  const { name, arguments: args } = request.params;\n  const db = getStorage();\n\n  try {\n    switch (name) {\n      case \"codemap_impact\": {\n        const target = args?.target as string;\n        const depth = (args?.depth as number) || 3;\n\n        // Find the target node(s)\n        const nodes = db.searchNodesByName(target);\n\n        if (nodes.length === 0) {\n          return {\n            content: [\n              {\n                type: \"text\",\n                text: `No function or class found with name \"${target}\"`,\n              },\n            ],\n          };\n        }\n\n        const targetNode = nodes[0];\n\n        // Get impacted nodes\n        const impacted = db.getImpactedNodes(targetNode.id, depth);\n\n        // Get unique files\n        const affectedFiles = new Set(impacted.map((n) => n.filePath));\n\n        // Assess risk\n        const directCallers = db.getResolvedCallers(targetNode.id);\n        let risk = \"LOW\";\n        if (directCallers.length > 10) risk = \"HIGH\";\n        else if (directCallers.length > 5) risk = \"MEDIUM\";\n\n        const response = `# Impact Analysis: ${targetNode.name}\n\n**Location:** ${targetNode.filePath}:${targetNode.startLine}\n**Type:** ${targetNode.type}\n**Risk Level:** ${risk}\n\n## Direct Callers (${directCallers.length})\n${directCallers.slice(0, 10).map((c) => `- ${c.name} in ${c.filePath}:${c.startLine}`).join(\"\\n\")}\n${directCallers.length > 10 ? `\\n... and ${directCallers.length - 10} more` : \"\"}\n\n## Total Impact\n- **Affected functions:** ${impacted.length}\n- **Affected files:** ${affectedFiles.size}\n\n## Affected Files\n${[...affectedFiles].slice(0, 20).map((f) => `- ${f}`).join(\"\\n\")}\n${affectedFiles.size > 20 ? `\\n... and ${affectedFiles.size - 20} more` : \"\"}\n\n⚠️ **Recommendation:** ${risk === \"HIGH\" ? \"High-risk change. Review all callers before modifying.\" : risk === \"MEDIUM\" ? \"Moderate risk. Test affected files after changes.\" : \"Low risk. Changes unlikely to cause wide impact.\"}\n`;\n\n        return {\n          content: [{ type: \"text\", text: response }],\n        };\n      }\n\n      case \"codemap_trace_flow\": {\n        const from = args?.from as string;\n        const to = args?.to as string;\n\n        // Find source and target nodes\n        const sourceNodes = db.searchNodesByName(from);\n        const targetNodes = db.searchNodesByName(to);\n\n        if (sourceNodes.length === 0) {\n          return {\n            content: [{ type: \"text\", text: `Source function \"${from}\" not found` }],\n          };\n        }\n\n        if (targetNodes.length === 0) {\n          return {\n            content: [{ type: \"text\", text: `Target function \"${to}\" not found` }],\n          };\n        }\n\n        const sourceNode = sourceNodes[0];\n        const targetNode = targetNodes[0];\n\n        // Trace path\n        const path = db.traceCallPath(sourceNode.id, targetNode.id);\n\n        if (!path) {\n          return {\n            content: [\n              {\n                type: \"text\",\n                text: `No call path found from \"${from}\" to \"${to}\". They may not be connected.`,\n              },\n            ],\n          };\n        }\n\n        const response = `# Execution Flow: ${from} → ${to}\n\n${path.nodes\n  .map(\n    (n, i) =>\n      `${i + 1}. **${n.name}**\\n   File: ${n.file}:${n.line}${i < path.nodes.length - 1 ? \"\\n   ↓ calls\" : \"\"}`\n  )\n  .join(\"\\n\\n\")}\n\n**Total steps:** ${path.nodes.length}\n`;\n\n        return {\n          content: [{ type: \"text\", text: response }],\n        };\n      }\n\n      case \"codemap_callers\": {\n        const functionName = args?.function as string;\n\n        // Find the function\n        const nodes = db.searchNodesByName(functionName);\n\n        if (nodes.length === 0) {\n          return {\n            content: [{ type: \"text\", text: `Function \"${functionName}\" not found` }],\n          };\n        }\n\n        const targetNode = nodes[0];\n\n        // Get all callers\n        const callers = db.getResolvedCallers(targetNode.id);\n\n        if (callers.length === 0) {\n          return {\n            content: [\n              {\n                type: \"text\",\n                text: `No callers found for \"${functionName}\". It may be unused or only called externally.`,\n              },\n            ],\n          };\n        }\n\n        const response = `# Callers of ${functionName}\n\n**Function:** ${targetNode.name} (${targetNode.type})\n**Location:** ${targetNode.filePath}:${targetNode.startLine}\n**Total callers:** ${callers.length}\n\n## Who Calls This Function\n\n${callers\n  .map(\n    (c) => `- **${c.name}** (${c.type})\n  File: ${c.filePath}:${c.startLine}`\n  )\n  .join(\"\\n\\n\")}\n`;\n\n        return {\n          content: [{ type: \"text\", text: response }],\n        };\n      }\n\n      default:\n        return {\n          content: [{ type: \"text\", text: `Unknown tool: ${name}` }],\n          isError: true,\n        };\n    }\n  } catch (error: unknown) {\n    const errorMessage = error instanceof Error ? error.message : String(error);\n    return {\n      content: [{ type: \"text\", text: `Error: ${errorMessage}` }],\n      isError: true,\n    };\n  }\n});\n\n// Start server\nconst transport = new StdioServerTransport();\nserver.connect(transport).catch((error) => {\n  console.error(\"Failed to start MCP server:\", error);\n  process.exit(1);\n});\n"],"mappings":";;;;;;;AAKA,SAAS,cAAc;AACvB,SAAS,4BAA4B;AACrC;AAAA,EACE;AAAA,EACA;AAAA,OACK;AAEP,SAAS,eAAe;AACxB,SAAS,kBAAkB;AAG3B,IAAM,UAAU,QAAQ,IAAI,mBAAmB;AAC/C,IAAM,eAAe,QAAQ,IAAI,wBAAwB,QAAQ,IAAI;AAErE,IAAI,UAA8B;AAElC,SAAS,aAA0B;AACjC,MAAI,CAAC,SAAS;AACZ,UAAM,SAAS,QAAQ,cAAc,OAAO;AAC5C,QAAI,CAAC,WAAW,MAAM,GAAG;AACvB,YAAM,IAAI,MAAM,iCAAiC,MAAM,8BAA8B;AAAA,IACvF;AACA,cAAU,IAAI,YAAY,MAAM;AAAA,EAClC;AACA,SAAO;AACT;AAEA,IAAM,SAAS,IAAI;AAAA,EACjB;AAAA,IACE,MAAM;AAAA,IACN,SAAS;AAAA,EACX;AAAA,EACA;AAAA,IACE,cAAc;AAAA,MACZ,OAAO,CAAC;AAAA,IACV;AAAA,EACF;AACF;AAIA,OAAO,kBAAkB,wBAAwB,YAAY;AAC3D,SAAO;AAAA,IACL,OAAO;AAAA,MACL;AAAA,QACE,MAAM;AAAA,QACN,aAAa;AAAA,QACb,aAAa;AAAA,UACX,MAAM;AAAA,UACN,YAAY;AAAA,YACV,QAAQ;AAAA,cACN,MAAM;AAAA,cACN,aAAa;AAAA,YACf;AAAA,YACA,OAAO;AAAA,cACL,MAAM;AAAA,cACN,aAAa;AAAA,cACb,SAAS;AAAA,YACX;AAAA,UACF;AAAA,UACA,UAAU,CAAC,QAAQ;AAAA,QACrB;AAAA,MACF;AAAA,MACA;AAAA,QACE,MAAM;AAAA,QACN,aAAa;AAAA,QACb,aAAa;AAAA,UACX,MAAM;AAAA,UACN,YAAY;AAAA,YACV,MAAM;AAAA,cACJ,MAAM;AAAA,cACN,aAAa;AAAA,YACf;AAAA,YACA,IAAI;AAAA,cACF,MAAM;AAAA,cACN,aAAa;AAAA,YACf;AAAA,UACF;AAAA,UACA,UAAU,CAAC,QAAQ,IAAI;AAAA,QACzB;AAAA,MACF;AAAA,MACA;AAAA,QACE,MAAM;AAAA,QACN,aAAa;AAAA,QACb,aAAa;AAAA,UACX,MAAM;AAAA,UACN,YAAY;AAAA,YACV,UAAU;AAAA,cACR,MAAM;AAAA,cACN,aAAa;AAAA,YACf;AAAA,UACF;AAAA,UACA,UAAU,CAAC,UAAU;AAAA,QACvB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF,CAAC;AAED,OAAO,kBAAkB,uBAAuB,OAAO,YAAY;AACjE,QAAM,EAAE,MAAM,WAAW,KAAK,IAAI,QAAQ;AAC1C,QAAM,KAAK,WAAW;AAEtB,MAAI;AACF,YAAQ,MAAM;AAAA,MACZ,KAAK,kBAAkB;AACrB,cAAM,SAAS,MAAM;AACrB,cAAM,QAAS,MAAM,SAAoB;AAGzC,cAAM,QAAQ,GAAG,kBAAkB,MAAM;AAEzC,YAAI,MAAM,WAAW,GAAG;AACtB,iBAAO;AAAA,YACL,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,MAAM,yCAAyC,MAAM;AAAA,cACvD;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAEA,cAAM,aAAa,MAAM,CAAC;AAG1B,cAAM,WAAW,GAAG,iBAAiB,WAAW,IAAI,KAAK;AAGzD,cAAM,gBAAgB,IAAI,IAAI,SAAS,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC;AAG7D,cAAM,gBAAgB,GAAG,mBAAmB,WAAW,EAAE;AACzD,YAAI,OAAO;AACX,YAAI,cAAc,SAAS,GAAI,QAAO;AAAA,iBAC7B,cAAc,SAAS,EAAG,QAAO;AAE1C,cAAM,WAAW,sBAAsB,WAAW,IAAI;AAAA;AAAA,gBAE9C,WAAW,QAAQ,IAAI,WAAW,SAAS;AAAA,YAC/C,WAAW,IAAI;AAAA,kBACT,IAAI;AAAA;AAAA,qBAED,cAAc,MAAM;AAAA,EACvC,cAAc,MAAM,GAAG,EAAE,EAAE,IAAI,CAAC,MAAM,KAAK,EAAE,IAAI,OAAO,EAAE,QAAQ,IAAI,EAAE,SAAS,EAAE,EAAE,KAAK,IAAI,CAAC;AAAA,EAC/F,cAAc,SAAS,KAAK;AAAA,UAAa,cAAc,SAAS,EAAE,UAAU,EAAE;AAAA;AAAA;AAAA,4BAGpD,SAAS,MAAM;AAAA,wBACnB,cAAc,IAAI;AAAA;AAAA;AAAA,EAGxC,CAAC,GAAG,aAAa,EAAE,MAAM,GAAG,EAAE,EAAE,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,EAAE,KAAK,IAAI,CAAC;AAAA,EAC/D,cAAc,OAAO,KAAK;AAAA,UAAa,cAAc,OAAO,EAAE,UAAU,EAAE;AAAA;AAAA,mCAEnD,SAAS,SAAS,2DAA2D,SAAS,WAAW,sDAAsD,kDAAkD;AAAA;AAG1N,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,SAAS,CAAC;AAAA,QAC5C;AAAA,MACF;AAAA,MAEA,KAAK,sBAAsB;AACzB,cAAM,OAAO,MAAM;AACnB,cAAM,KAAK,MAAM;AAGjB,cAAM,cAAc,GAAG,kBAAkB,IAAI;AAC7C,cAAM,cAAc,GAAG,kBAAkB,EAAE;AAE3C,YAAI,YAAY,WAAW,GAAG;AAC5B,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,oBAAoB,IAAI,cAAc,CAAC;AAAA,UACzE;AAAA,QACF;AAEA,YAAI,YAAY,WAAW,GAAG;AAC5B,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,oBAAoB,EAAE,cAAc,CAAC;AAAA,UACvE;AAAA,QACF;AAEA,cAAM,aAAa,YAAY,CAAC;AAChC,cAAM,aAAa,YAAY,CAAC;AAGhC,cAAM,OAAO,GAAG,cAAc,WAAW,IAAI,WAAW,EAAE;AAE1D,YAAI,CAAC,MAAM;AACT,iBAAO;AAAA,YACL,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,MAAM,4BAA4B,IAAI,SAAS,EAAE;AAAA,cACnD;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAEA,cAAM,WAAW,qBAAqB,IAAI,WAAM,EAAE;AAAA;AAAA,EAExD,KAAK,MACJ;AAAA,UACC,CAAC,GAAG,MACF,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI;AAAA,WAAgB,EAAE,IAAI,IAAI,EAAE,IAAI,GAAG,IAAI,KAAK,MAAM,SAAS,IAAI,sBAAiB,EAAE;AAAA,QAC3G,EACC,KAAK,MAAM,CAAC;AAAA;AAAA,mBAEI,KAAK,MAAM,MAAM;AAAA;AAG5B,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,SAAS,CAAC;AAAA,QAC5C;AAAA,MACF;AAAA,MAEA,KAAK,mBAAmB;AACtB,cAAM,eAAe,MAAM;AAG3B,cAAM,QAAQ,GAAG,kBAAkB,YAAY;AAE/C,YAAI,MAAM,WAAW,GAAG;AACtB,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,aAAa,YAAY,cAAc,CAAC;AAAA,UAC1E;AAAA,QACF;AAEA,cAAM,aAAa,MAAM,CAAC;AAG1B,cAAM,UAAU,GAAG,mBAAmB,WAAW,EAAE;AAEnD,YAAI,QAAQ,WAAW,GAAG;AACxB,iBAAO;AAAA,YACL,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,MAAM,yBAAyB,YAAY;AAAA,cAC7C;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAEA,cAAM,WAAW,gBAAgB,YAAY;AAAA;AAAA,gBAErC,WAAW,IAAI,KAAK,WAAW,IAAI;AAAA,gBACnC,WAAW,QAAQ,IAAI,WAAW,SAAS;AAAA,qBACtC,QAAQ,MAAM;AAAA;AAAA;AAAA;AAAA,EAIjC,QACC;AAAA,UACC,CAAC,MAAM,OAAO,EAAE,IAAI,OAAO,EAAE,IAAI;AAAA,UAC3B,EAAE,QAAQ,IAAI,EAAE,SAAS;AAAA,QACjC,EACC,KAAK,MAAM,CAAC;AAAA;AAGP,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,SAAS,CAAC;AAAA,QAC5C;AAAA,MACF;AAAA,MAEA;AACE,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,iBAAiB,IAAI,GAAG,CAAC;AAAA,UACzD,SAAS;AAAA,QACX;AAAA,IACJ;AAAA,EACF,SAAS,OAAgB;AACvB,UAAM,eAAe,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAC1E,WAAO;AAAA,MACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,UAAU,YAAY,GAAG,CAAC;AAAA,MAC1D,SAAS;AAAA,IACX;AAAA,EACF;AACF,CAAC;AAGD,IAAM,YAAY,IAAI,qBAAqB;AAC3C,OAAO,QAAQ,SAAS,EAAE,MAAM,CAAC,UAAU;AACzC,UAAQ,MAAM,+BAA+B,KAAK;AAClD,UAAQ,KAAK,CAAC;AAChB,CAAC;","names":[]}
+{"version":3,"sources":["../src/mcp/flow-server.ts"],"sourcesContent":["/**\n * MCP Server for CodeMap Flow Edition\n * Provides 3 focused tools: impact analysis, flow tracing, and caller lookup\n */\n\nimport { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\nimport { StdioServerTransport } from \"@modelcontextprotocol/sdk/server/stdio.js\";\nimport {\n  CallToolRequestSchema,\n  ListToolsRequestSchema,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { FlowStorage } from \"../flow/storage.js\";\nimport { resolve } from \"path\";\nimport { existsSync } from \"fs\";\n\n// Get config from environment\nconst DB_PATH = process.env.CODEMAP_DB_PATH || \".codemap/graph.db\";\nconst PROJECT_ROOT = process.env.CODEMAP_PROJECT_ROOT || process.cwd();\n\nlet storage: FlowStorage | null = null;\n\nfunction getStorage(): FlowStorage {\n  if (!storage) {\n    const dbPath = resolve(PROJECT_ROOT, DB_PATH);\n    if (!existsSync(dbPath)) {\n      throw new Error(`CodeMap database not found at ${dbPath}. Run 'codemap index' first.`);\n    }\n    storage = new FlowStorage(dbPath);\n  }\n  return storage;\n}\n\nconst server = new Server(\n  {\n    name: \"codemap-flow\",\n    version: \"3.4.0\",\n  },\n  {\n    capabilities: {\n      tools: {},\n    },\n  }\n);\n\n// ============ Tools ============\n\nserver.setRequestHandler(ListToolsRequestSchema, async () => {\n  return {\n    tools: [\n      {\n        name: \"codemap_impact\",\n        description: \"Analyze what would break if you change a function/class. Shows all callers and affected files with risk assessment.\",\n        inputSchema: {\n          type: \"object\",\n          properties: {\n            target: {\n              type: \"string\",\n              description: \"Function or class name to analyze\",\n            },\n            depth: {\n              type: \"number\",\n              description: \"How many levels deep to analyze (default: 3)\",\n              default: 3,\n            },\n          },\n          required: [\"target\"],\n        },\n      },\n      {\n        name: \"codemap_trace_flow\",\n        description: \"Trace the execution flow between two functions. Shows the complete call path from source to target.\",\n        inputSchema: {\n          type: \"object\",\n          properties: {\n            from: {\n              type: \"string\",\n              description: \"Starting function name\",\n            },\n            to: {\n              type: \"string\",\n              description: \"Target function name\",\n            },\n          },\n          required: [\"from\", \"to\"],\n        },\n      },\n      {\n        name: \"codemap_callers\",\n        description: \"Find all functions that call a specific function. Shows where and how a function is used.\",\n        inputSchema: {\n          type: \"object\",\n          properties: {\n            function: {\n              type: \"string\",\n              description: \"Function name to find callers for\",\n            },\n          },\n          required: [\"function\"],\n        },\n      },\n    ],\n  };\n});\n\nserver.setRequestHandler(CallToolRequestSchema, async (request) => {\n  const { name, arguments: args } = request.params;\n  const db = getStorage();\n\n  try {\n    switch (name) {\n      case \"codemap_impact\": {\n        const target = args?.target as string;\n        const depth = (args?.depth as number) || 3;\n\n        // Find the target node(s)\n        const nodes = db.searchNodesByName(target);\n\n        if (nodes.length === 0) {\n          return {\n            content: [\n              {\n                type: \"text\",\n                text: `No function or class found with name \"${target}\"`,\n              },\n            ],\n          };\n        }\n\n        const targetNode = nodes[0];\n\n        // Get impacted nodes\n        const impacted = db.getImpactedNodes(targetNode.id, depth);\n\n        // Get unique files\n        const affectedFiles = new Set(impacted.map((n) => n.filePath));\n\n        // Assess risk\n        const directCallers = db.getResolvedCallers(targetNode.id);\n        let risk = \"LOW\";\n        if (directCallers.length > 10) risk = \"HIGH\";\n        else if (directCallers.length > 5) risk = \"MEDIUM\";\n\n        const response = `# Impact Analysis: ${targetNode.name}\n\n**Location:** ${targetNode.filePath}:${targetNode.startLine}\n**Type:** ${targetNode.type}\n**Risk Level:** ${risk}\n\n## Direct Callers (${directCallers.length})\n${directCallers.slice(0, 10).map((c) => `- ${c.name} in ${c.filePath}:${c.startLine}`).join(\"\\n\")}\n${directCallers.length > 10 ? `\\n... and ${directCallers.length - 10} more` : \"\"}\n\n## Total Impact\n- **Affected functions:** ${impacted.length}\n- **Affected files:** ${affectedFiles.size}\n\n## Affected Files\n${[...affectedFiles].slice(0, 20).map((f) => `- ${f}`).join(\"\\n\")}\n${affectedFiles.size > 20 ? `\\n... and ${affectedFiles.size - 20} more` : \"\"}\n\n⚠️ **Recommendation:** ${risk === \"HIGH\" ? \"High-risk change. Review all callers before modifying.\" : risk === \"MEDIUM\" ? \"Moderate risk. Test affected files after changes.\" : \"Low risk. Changes unlikely to cause wide impact.\"}\n`;\n\n        return {\n          content: [{ type: \"text\", text: response }],\n        };\n      }\n\n      case \"codemap_trace_flow\": {\n        const from = args?.from as string;\n        const to = args?.to as string;\n\n        // Find source and target nodes\n        const sourceNodes = db.searchNodesByName(from);\n        const targetNodes = db.searchNodesByName(to);\n\n        if (sourceNodes.length === 0) {\n          return {\n            content: [{ type: \"text\", text: `Source function \"${from}\" not found` }],\n          };\n        }\n\n        if (targetNodes.length === 0) {\n          return {\n            content: [{ type: \"text\", text: `Target function \"${to}\" not found` }],\n          };\n        }\n\n        const sourceNode = sourceNodes[0];\n        const targetNode = targetNodes[0];\n\n        // Trace path\n        const path = db.traceCallPath(sourceNode.id, targetNode.id);\n\n        if (!path) {\n          return {\n            content: [\n              {\n                type: \"text\",\n                text: `No call path found from \"${from}\" to \"${to}\". They may not be connected.`,\n              },\n            ],\n          };\n        }\n\n        const response = `# Execution Flow: ${from} → ${to}\n\n${path.nodes\n  .map(\n    (n, i) =>\n      `${i + 1}. **${n.name}**\\n   File: ${n.file}:${n.line}${i < path.nodes.length - 1 ? \"\\n   ↓ calls\" : \"\"}`\n  )\n  .join(\"\\n\\n\")}\n\n**Total steps:** ${path.nodes.length}\n`;\n\n        return {\n          content: [{ type: \"text\", text: response }],\n        };\n      }\n\n      case \"codemap_callers\": {\n        const functionName = args?.function as string;\n\n        // Find the function\n        const nodes = db.searchNodesByName(functionName);\n\n        if (nodes.length === 0) {\n          return {\n            content: [{ type: \"text\", text: `Function \"${functionName}\" not found` }],\n          };\n        }\n\n        const targetNode = nodes[0];\n\n        // Get all callers\n        const callers = db.getResolvedCallers(targetNode.id);\n\n        if (callers.length === 0) {\n          return {\n            content: [\n              {\n                type: \"text\",\n                text: `No callers found for \"${functionName}\". It may be unused or only called externally.`,\n              },\n            ],\n          };\n        }\n\n        const response = `# Callers of ${functionName}\n\n**Function:** ${targetNode.name} (${targetNode.type})\n**Location:** ${targetNode.filePath}:${targetNode.startLine}\n**Total callers:** ${callers.length}\n\n## Who Calls This Function\n\n${callers\n  .map(\n    (c) => `- **${c.name}** (${c.type})\n  File: ${c.filePath}:${c.startLine}`\n  )\n  .join(\"\\n\\n\")}\n`;\n\n        return {\n          content: [{ type: \"text\", text: response }],\n        };\n      }\n\n      default:\n        return {\n          content: [{ type: \"text\", text: `Unknown tool: ${name}` }],\n          isError: true,\n        };\n    }\n  } catch (error: unknown) {\n    const errorMessage = error instanceof Error ? error.message : String(error);\n    return {\n      content: [{ type: \"text\", text: `Error: ${errorMessage}` }],\n      isError: true,\n    };\n  }\n});\n\n// Start server\nconst transport = new StdioServerTransport();\nserver.connect(transport).catch((error) => {\n  console.error(\"Failed to start MCP server:\", error);\n  process.exit(1);\n});\n"],"mappings":";;;;;;;AAKA,SAAS,cAAc;AACvB,SAAS,4BAA4B;AACrC;AAAA,EACE;AAAA,EACA;AAAA,OACK;AAEP,SAAS,eAAe;AACxB,SAAS,kBAAkB;AAG3B,IAAM,UAAU,QAAQ,IAAI,mBAAmB;AAC/C,IAAM,eAAe,QAAQ,IAAI,wBAAwB,QAAQ,IAAI;AAErE,IAAI,UAA8B;AAElC,SAAS,aAA0B;AACjC,MAAI,CAAC,SAAS;AACZ,UAAM,SAAS,QAAQ,cAAc,OAAO;AAC5C,QAAI,CAAC,WAAW,MAAM,GAAG;AACvB,YAAM,IAAI,MAAM,iCAAiC,MAAM,8BAA8B;AAAA,IACvF;AACA,cAAU,IAAI,YAAY,MAAM;AAAA,EAClC;AACA,SAAO;AACT;AAEA,IAAM,SAAS,IAAI;AAAA,EACjB;AAAA,IACE,MAAM;AAAA,IACN,SAAS;AAAA,EACX;AAAA,EACA;AAAA,IACE,cAAc;AAAA,MACZ,OAAO,CAAC;AAAA,IACV;AAAA,EACF;AACF;AAIA,OAAO,kBAAkB,wBAAwB,YAAY;AAC3D,SAAO;AAAA,IACL,OAAO;AAAA,MACL;AAAA,QACE,MAAM;AAAA,QACN,aAAa;AAAA,QACb,aAAa;AAAA,UACX,MAAM;AAAA,UACN,YAAY;AAAA,YACV,QAAQ;AAAA,cACN,MAAM;AAAA,cACN,aAAa;AAAA,YACf;AAAA,YACA,OAAO;AAAA,cACL,MAAM;AAAA,cACN,aAAa;AAAA,cACb,SAAS;AAAA,YACX;AAAA,UACF;AAAA,UACA,UAAU,CAAC,QAAQ;AAAA,QACrB;AAAA,MACF;AAAA,MACA;AAAA,QACE,MAAM;AAAA,QACN,aAAa;AAAA,QACb,aAAa;AAAA,UACX,MAAM;AAAA,UACN,YAAY;AAAA,YACV,MAAM;AAAA,cACJ,MAAM;AAAA,cACN,aAAa;AAAA,YACf;AAAA,YACA,IAAI;AAAA,cACF,MAAM;AAAA,cACN,aAAa;AAAA,YACf;AAAA,UACF;AAAA,UACA,UAAU,CAAC,QAAQ,IAAI;AAAA,QACzB;AAAA,MACF;AAAA,MACA;AAAA,QACE,MAAM;AAAA,QACN,aAAa;AAAA,QACb,aAAa;AAAA,UACX,MAAM;AAAA,UACN,YAAY;AAAA,YACV,UAAU;AAAA,cACR,MAAM;AAAA,cACN,aAAa;AAAA,YACf;AAAA,UACF;AAAA,UACA,UAAU,CAAC,UAAU;AAAA,QACvB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF,CAAC;AAED,OAAO,kBAAkB,uBAAuB,OAAO,YAAY;AACjE,QAAM,EAAE,MAAM,WAAW,KAAK,IAAI,QAAQ;AAC1C,QAAM,KAAK,WAAW;AAEtB,MAAI;AACF,YAAQ,MAAM;AAAA,MACZ,KAAK,kBAAkB;AACrB,cAAM,SAAS,MAAM;AACrB,cAAM,QAAS,MAAM,SAAoB;AAGzC,cAAM,QAAQ,GAAG,kBAAkB,MAAM;AAEzC,YAAI,MAAM,WAAW,GAAG;AACtB,iBAAO;AAAA,YACL,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,MAAM,yCAAyC,MAAM;AAAA,cACvD;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAEA,cAAM,aAAa,MAAM,CAAC;AAG1B,cAAM,WAAW,GAAG,iBAAiB,WAAW,IAAI,KAAK;AAGzD,cAAM,gBAAgB,IAAI,IAAI,SAAS,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC;AAG7D,cAAM,gBAAgB,GAAG,mBAAmB,WAAW,EAAE;AACzD,YAAI,OAAO;AACX,YAAI,cAAc,SAAS,GAAI,QAAO;AAAA,iBAC7B,cAAc,SAAS,EAAG,QAAO;AAE1C,cAAM,WAAW,sBAAsB,WAAW,IAAI;AAAA;AAAA,gBAE9C,WAAW,QAAQ,IAAI,WAAW,SAAS;AAAA,YAC/C,WAAW,IAAI;AAAA,kBACT,IAAI;AAAA;AAAA,qBAED,cAAc,MAAM;AAAA,EACvC,cAAc,MAAM,GAAG,EAAE,EAAE,IAAI,CAAC,MAAM,KAAK,EAAE,IAAI,OAAO,EAAE,QAAQ,IAAI,EAAE,SAAS,EAAE,EAAE,KAAK,IAAI,CAAC;AAAA,EAC/F,cAAc,SAAS,KAAK;AAAA,UAAa,cAAc,SAAS,EAAE,UAAU,EAAE;AAAA;AAAA;AAAA,4BAGpD,SAAS,MAAM;AAAA,wBACnB,cAAc,IAAI;AAAA;AAAA;AAAA,EAGxC,CAAC,GAAG,aAAa,EAAE,MAAM,GAAG,EAAE,EAAE,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,EAAE,KAAK,IAAI,CAAC;AAAA,EAC/D,cAAc,OAAO,KAAK;AAAA,UAAa,cAAc,OAAO,EAAE,UAAU,EAAE;AAAA;AAAA,mCAEnD,SAAS,SAAS,2DAA2D,SAAS,WAAW,sDAAsD,kDAAkD;AAAA;AAG1N,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,SAAS,CAAC;AAAA,QAC5C;AAAA,MACF;AAAA,MAEA,KAAK,sBAAsB;AACzB,cAAM,OAAO,MAAM;AACnB,cAAM,KAAK,MAAM;AAGjB,cAAM,cAAc,GAAG,kBAAkB,IAAI;AAC7C,cAAM,cAAc,GAAG,kBAAkB,EAAE;AAE3C,YAAI,YAAY,WAAW,GAAG;AAC5B,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,oBAAoB,IAAI,cAAc,CAAC;AAAA,UACzE;AAAA,QACF;AAEA,YAAI,YAAY,WAAW,GAAG;AAC5B,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,oBAAoB,EAAE,cAAc,CAAC;AAAA,UACvE;AAAA,QACF;AAEA,cAAM,aAAa,YAAY,CAAC;AAChC,cAAM,aAAa,YAAY,CAAC;AAGhC,cAAM,OAAO,GAAG,cAAc,WAAW,IAAI,WAAW,EAAE;AAE1D,YAAI,CAAC,MAAM;AACT,iBAAO;AAAA,YACL,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,MAAM,4BAA4B,IAAI,SAAS,EAAE;AAAA,cACnD;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAEA,cAAM,WAAW,qBAAqB,IAAI,WAAM,EAAE;AAAA;AAAA,EAExD,KAAK,MACJ;AAAA,UACC,CAAC,GAAG,MACF,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI;AAAA,WAAgB,EAAE,IAAI,IAAI,EAAE,IAAI,GAAG,IAAI,KAAK,MAAM,SAAS,IAAI,sBAAiB,EAAE;AAAA,QAC3G,EACC,KAAK,MAAM,CAAC;AAAA;AAAA,mBAEI,KAAK,MAAM,MAAM;AAAA;AAG5B,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,SAAS,CAAC;AAAA,QAC5C;AAAA,MACF;AAAA,MAEA,KAAK,mBAAmB;AACtB,cAAM,eAAe,MAAM;AAG3B,cAAM,QAAQ,GAAG,kBAAkB,YAAY;AAE/C,YAAI,MAAM,WAAW,GAAG;AACtB,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,aAAa,YAAY,cAAc,CAAC;AAAA,UAC1E;AAAA,QACF;AAEA,cAAM,aAAa,MAAM,CAAC;AAG1B,cAAM,UAAU,GAAG,mBAAmB,WAAW,EAAE;AAEnD,YAAI,QAAQ,WAAW,GAAG;AACxB,iBAAO;AAAA,YACL,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,MAAM,yBAAyB,YAAY;AAAA,cAC7C;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAEA,cAAM,WAAW,gBAAgB,YAAY;AAAA;AAAA,gBAErC,WAAW,IAAI,KAAK,WAAW,IAAI;AAAA,gBACnC,WAAW,QAAQ,IAAI,WAAW,SAAS;AAAA,qBACtC,QAAQ,MAAM;AAAA;AAAA;AAAA;AAAA,EAIjC,QACC;AAAA,UACC,CAAC,MAAM,OAAO,EAAE,IAAI,OAAO,EAAE,IAAI;AAAA,UAC3B,EAAE,QAAQ,IAAI,EAAE,SAAS;AAAA,QACjC,EACC,KAAK,MAAM,CAAC;AAAA;AAGP,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,SAAS,CAAC;AAAA,QAC5C;AAAA,MACF;AAAA,MAEA;AACE,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,iBAAiB,IAAI,GAAG,CAAC;AAAA,UACzD,SAAS;AAAA,QACX;AAAA,IACJ;AAAA,EACF,SAAS,OAAgB;AACvB,UAAM,eAAe,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAC1E,WAAO;AAAA,MACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,UAAU,YAAY,GAAG,CAAC;AAAA,MAC1D,SAAS;AAAA,IACX;AAAA,EACF;AACF,CAAC;AAGD,IAAM,YAAY,IAAI,qBAAqB;AAC3C,OAAO,QAAQ,SAAS,EAAE,MAAM,CAAC,UAAU;AACzC,UAAQ,MAAM,+BAA+B,KAAK;AAClD,UAAQ,KAAK,CAAC;AAChB,CAAC;","names":[]}

package/dist/index.js

@@ -8,6 +8,392 @@ import { createHash } from "crypto";
 import { glob } from "glob";
 import { resolve, relative, extname } from "path";
 
+// src/analysis/class-hierarchy.ts
+function buildClassHierarchy(nodes) {
+  const classes = /* @__PURE__ */ new Map();
+  const children = /* @__PURE__ */ new Map();
+  const parents = /* @__PURE__ */ new Map();
+  for (const node of nodes) {
+    if (node.type === "class") {
+      const classNode = {
+        id: node.id,
+        name: node.name,
+        file: node.filePath,
+        line: node.startLine,
+        extends: node.extends || [],
+        methods: []
+        // Will populate in next step
+      };
+      classes.set(node.name, classNode);
+    }
+  }
+  for (const node of nodes) {
+    if (node.type === "method" && node.parentClass) {
+      const classNode = classes.get(node.parentClass);
+      if (classNode) {
+        classNode.methods.push(node.name);
+      }
+    }
+  }
+  for (const [className, classNode] of classes) {
+    for (const parentName of classNode.extends) {
+      if (!children.has(parentName)) {
+        children.set(parentName, []);
+      }
+      children.get(parentName).push(className);
+      if (!parents.has(className)) {
+        parents.set(className, []);
+      }
+      parents.get(className).push(parentName);
+    }
+  }
+  return {
+    classes,
+    children,
+    parents
+  };
+}
+function findMethodDefinition(className, methodName, hierarchy) {
+  const visited = /* @__PURE__ */ new Set();
+  const queue = [className];
+  while (queue.length > 0) {
+    const current = queue.shift();
+    if (visited.has(current)) {
+      continue;
+    }
+    visited.add(current);
+    const classNode = hierarchy.classes.get(current);
+    if (classNode) {
+      if (classNode.methods.includes(methodName)) {
+        return classNode;
+      }
+      const parentNames = hierarchy.parents.get(current) || [];
+      for (const parent of parentNames) {
+        queue.push(parent);
+      }
+    }
+  }
+  return null;
+}
+function resolveMethodCall(instanceType, methodName, hierarchy) {
+  return findMethodDefinition(instanceType, methodName, hierarchy);
+}
+
+// src/flow/enhanced-resolver.ts
+var EnhancedResolver = class {
+  constructor(storage) {
+    this.storage = storage;
+  }
+  hierarchy = null;
+  variableTypes = /* @__PURE__ */ new Map();
+  /**
+   * Build class hierarchy from all nodes in storage
+   */
+  buildHierarchy() {
+    const nodes = this.getAllNodes();
+    this.hierarchy = buildClassHierarchy(nodes);
+  }
+  /**
+   * Load variable type information from storage
+   */
+  loadVariableTypes() {
+    const stmt = this.storage.db.prepare(`
+      SELECT * FROM variable_types
+    `);
+    const rows = stmt.all();
+    for (const row of rows) {
+      const scopeId = row.scope_node_id;
+      if (!this.variableTypes.has(scopeId)) {
+        this.variableTypes.set(scopeId, []);
+      }
+      this.variableTypes.get(scopeId).push({
+        name: row.variable_name,
+        type: row.type_name,
+        scope: scopeId,
+        line: row.line
+      });
+    }
+  }
+  /**
+   * Resolve all unresolved edges using type information
+   */
+  resolveAllCalls() {
+    if (!this.hierarchy) {
+      this.buildHierarchy();
+    }
+    this.loadVariableTypes();
+    const resolvedCalls = [];
+    const stmt = this.storage.db.prepare(`
+      SELECT * FROM edges
+      WHERE target_id LIKE 'ref:%'
+      AND type = 'calls'
+    `);
+    const edges = stmt.all();
+    for (const edgeRow of edges) {
+      const edge = {
+        id: edgeRow.id,
+        type: edgeRow.type,
+        sourceId: edgeRow.source_id,
+        targetId: edgeRow.target_id,
+        metadata: edgeRow.metadata ? JSON.parse(edgeRow.metadata) : void 0
+      };
+      const resolved = this.resolveCall(edge);
+      if (resolved.targetNode) {
+        resolvedCalls.push(resolved);
+        this.updateEdgeTarget(edge.id, resolved.targetNode.id);
+      }
+    }
+    return resolvedCalls;
+  }
+  /**
+   * Resolve a single call edge
+   */
+  resolveCall(edge) {
+    const unresolvedName = edge.metadata?.unresolvedName;
+    if (!unresolvedName) {
+      return {
+        originalEdge: edge,
+        targetNode: null,
+        confidence: "LOW",
+        reason: "No unresolved name in metadata"
+      };
+    }
+    const sourceNode = this.storage.getNode(edge.sourceId);
+    if (!sourceNode) {
+      return {
+        originalEdge: edge,
+        targetNode: null,
+        confidence: "LOW",
+        reason: "Source node not found"
+      };
+    }
+    if (!unresolvedName.includes(".")) {
+      return this.resolveSimpleCall(unresolvedName, sourceNode, edge);
+    }
+    return this.resolveMethodCall(unresolvedName, sourceNode, edge);
+  }
+  /**
+   * Resolve simple function call (no dot notation)
+   */
+  resolveSimpleCall(functionName, sourceNode, edge) {
+    const sameFileNodes = this.storage.getNodesByFile(sourceNode.filePath);
+    const localMatch = sameFileNodes.find(
+      (n) => (n.type === "function" || n.type === "method") && n.name === functionName
+    );
+    if (localMatch) {
+      return {
+        originalEdge: edge,
+        targetNode: localMatch,
+        confidence: "HIGH",
+        reason: "Found in same file"
+      };
+    }
+    const globalMatches = this.storage.searchNodesByName(functionName);
+    if (globalMatches.length === 1) {
+      return {
+        originalEdge: edge,
+        targetNode: globalMatches[0],
+        confidence: "MEDIUM",
+        reason: "Single global match"
+      };
+    }
+    if (globalMatches.length > 1) {
+      const sourceDir = sourceNode.filePath.split("/").slice(0, -1).join("/");
+      const sameDirMatch = globalMatches.find((n) => n.filePath.startsWith(sourceDir));
+      if (sameDirMatch) {
+        return {
+          originalEdge: edge,
+          targetNode: sameDirMatch,
+          confidence: "MEDIUM",
+          reason: "Multiple matches, picked same directory"
+        };
+      }
+      return {
+        originalEdge: edge,
+        targetNode: globalMatches[0],
+        confidence: "LOW",
+        reason: `Multiple matches (${globalMatches.length}), picked first`
+      };
+    }
+    return {
+      originalEdge: edge,
+      targetNode: null,
+      confidence: "LOW",
+      reason: "No matches found"
+    };
+  }
+  /**
+   * Resolve method call using type information
+   * Pattern: obj.method() or self.method()
+   */
+  resolveMethodCall(fullName, sourceNode, edge) {
+    const parts = fullName.split(".");
+    const varName = parts[0];
+    const methodName = parts.slice(1).join(".");
+    if (varName === "self" || varName === "this") {
+      return this.resolveSelfMethodCall(methodName, sourceNode, edge);
+    }
+    const scopeTypes = this.variableTypes.get(sourceNode.id) || [];
+    const varType = scopeTypes.find((v) => v.name === varName);
+    if (!varType) {
+      return this.resolveSimpleCall(methodName, sourceNode, edge);
+    }
+    if (!this.hierarchy) {
+      return {
+        originalEdge: edge,
+        targetNode: null,
+        confidence: "LOW",
+        reason: "Class hierarchy not built"
+      };
+    }
+    const classNode = resolveMethodCall(varType.type, methodName, this.hierarchy);
+    if (classNode) {
+      const methodNode = this.findMethodInClass(classNode.name, methodName);
+      if (methodNode) {
+        return {
+          originalEdge: edge,
+          targetNode: methodNode,
+          confidence: "HIGH",
+          reason: `Resolved via type tracking: ${varName}: ${varType.type}`
+        };
+      }
+    }
+    return {
+      originalEdge: edge,
+      targetNode: null,
+      confidence: "LOW",
+      reason: `Type found (${varType.type}) but method not resolved`
+    };
+  }
+  /**
+   * Resolve self.method() or this.method()
+   */
+  resolveSelfMethodCall(methodName, sourceNode, edge) {
+    const parentClass = sourceNode.metadata?.parentClass;
+    if (!parentClass) {
+      return {
+        originalEdge: edge,
+        targetNode: null,
+        confidence: "LOW",
+        reason: "self/this call but no parent class"
+      };
+    }
+    if (!this.hierarchy) {
+      return {
+        originalEdge: edge,
+        targetNode: null,
+        confidence: "LOW",
+        reason: "Class hierarchy not built"
+      };
+    }
+    const classNode = resolveMethodCall(parentClass, methodName, this.hierarchy);
+    if (classNode) {
+      const methodNode2 = this.findMethodInClass(classNode.name, methodName);
+      if (methodNode2) {
+        return {
+          originalEdge: edge,
+          targetNode: methodNode2,
+          confidence: "HIGH",
+          reason: `Resolved self.${methodName} in ${parentClass}`
+        };
+      }
+    }
+    const sameFileNodes = this.storage.getNodesByFile(sourceNode.filePath);
+    const methodNode = sameFileNodes.find(
+      (n) => n.type === "method" && n.name === methodName
+    );
+    if (methodNode) {
+      return {
+        originalEdge: edge,
+        targetNode: methodNode,
+        confidence: "MEDIUM",
+        reason: "Found method in same file"
+      };
+    }
+    return {
+      originalEdge: edge,
+      targetNode: null,
+      confidence: "LOW",
+      reason: `Method ${methodName} not found in ${parentClass}`
+    };
+  }
+  /**
+   * Find method node in a class
+   */
+  findMethodInClass(className, methodName) {
+    const stmt = this.storage.db.prepare(`
+      SELECT * FROM nodes
+      WHERE type = 'method'
+      AND name = ?
+      AND json_extract(metadata, '$.parentClass') = ?
+    `);
+    const row = stmt.get(methodName, className);
+    if (!row) return null;
+    return {
+      id: row.id,
+      type: row.type,
+      name: row.name,
+      filePath: row.file_path,
+      startLine: row.start_line,
+      endLine: row.end_line,
+      language: row.language,
+      metadata: row.metadata ? JSON.parse(row.metadata) : void 0
+    };
+  }
+  /**
+   * Get all nodes from storage
+   */
+  getAllNodes() {
+    const stmt = this.storage.db.prepare("SELECT * FROM nodes");
+    const rows = stmt.all();
+    return rows.map((row) => ({
+      id: row.id,
+      type: row.type,
+      name: row.name,
+      filePath: row.file_path,
+      startLine: row.start_line,
+      endLine: row.end_line,
+      language: row.language,
+      metadata: row.metadata ? JSON.parse(row.metadata) : void 0
+    }));
+  }
+  /**
+   * Update edge target in database
+   */
+  updateEdgeTarget(edgeId, newTargetId) {
+    const stmt = this.storage.db.prepare(`
+      UPDATE edges
+      SET target_id = ?
+      WHERE id = ?
+    `);
+    stmt.run(newTargetId, edgeId);
+  }
+  /**
+   * Get resolution statistics
+   */
+  getStats() {
+    const totalStmt = this.storage.db.prepare(`
+      SELECT COUNT(*) as count FROM edges WHERE type = 'calls'
+    `);
+    const totalRow = totalStmt.get();
+    const totalCalls = totalRow.count;
+    const resolvedStmt = this.storage.db.prepare(`
+      SELECT COUNT(*) as count FROM edges
+      WHERE type = 'calls' AND target_id NOT LIKE 'ref:%'
+    `);
+    const resolvedRow = resolvedStmt.get();
+    const resolvedCalls = resolvedRow.count;
+    const unresolvedCalls = totalCalls - resolvedCalls;
+    const resolutionRate = totalCalls > 0 ? resolvedCalls / totalCalls * 100 : 0;
+    return {
+      totalCalls,
+      resolvedCalls,
+      unresolvedCalls,
+      resolutionRate
+    };
+  }
+};
+
 // src/parsers/base.ts
 var BaseParser = class {
   nodeIdCounter = 0;
@@ -262,31 +648,38 @@ var TypeScriptParser = class extends BaseParser {
   handleClass(node, filePath, analysis, parentId) {
     const nameNode = node.childForFieldName("name");
     if (!nameNode) return;
+    const extends_ = [];
+    for (const child of node.children) {
+      if (child.type === "class_heritage") {
+        const extendsClause = child.children.find((c) => c.type === "extends_clause");
+        if (extendsClause) {
+          const baseClass = extendsClause.children.find((c) => c.type === "identifier");
+          if (baseClass) {
+            extends_.push(baseClass.text);
+          }
+        }
+      }
+    }
     const classNode = this.createNode(
       "class",
       nameNode.text,
       filePath,
       node.startPosition.row + 1,
-      node.endPosition.row + 1
+      node.endPosition.row + 1,
+      {
+        extends: extends_
+      }
     );
     analysis.nodes.push(classNode);
     analysis.edges.push(
       this.createEdge("contains", parentId, classNode.id)
     );
-    for (const child of node.children) {
-      if (child.type === "class_heritage") {
-        const extendsClause = child.children.find((c) => c.type === "extends_clause");
-        if (extendsClause) {
-          const baseClass = extendsClause.children.find((c) => c.type === "identifier");
-          if (baseClass) {
-            analysis.edges.push(
-              this.createEdge("extends", classNode.id, `ref:${baseClass.text}`, {
-                unresolvedName: baseClass.text
-              })
-            );
-          }
-        }
-      }
+    for (const baseClass of extends_) {
+      analysis.edges.push(
+        this.createEdge("extends", classNode.id, `ref:${baseClass}`, {
+          unresolvedName: baseClass
+        })
+      );
     }
     const body = node.childForFieldName("body");
     if (body) {
@@ -300,6 +693,8 @@ var TypeScriptParser = class extends BaseParser {
   handleMethod(node, filePath, analysis, parentId) {
     const nameNode = node.childForFieldName("name");
     if (!nameNode) return;
+    const parentNode = analysis.nodes.find((n) => n.id === parentId);
+    const parentClass = parentNode?.type === "class" ? parentNode.name : void 0;
     const methodNode = this.createNode(
       "method",
       nameNode.text,
@@ -308,7 +703,8 @@ var TypeScriptParser = class extends BaseParser {
       node.endPosition.row + 1,
       {
         static: node.children.some((c) => c.type === "static"),
-        async: node.children.some((c) => c.type === "async")
+        async: node.children.some((c) => c.type === "async"),
+        parentClass
       }
     );
     analysis.nodes.push(methodNode);
@@ -743,6 +1139,8 @@ var PythonParser = class extends BaseParser {
       node.endPosition.row + 1,
       {
         bases,
+        extends: bases,
+        // Add extends for consistency with TypeScript
         isPrivate: name.startsWith("_")
       }
     );
@@ -780,6 +1178,8 @@ var PythonParser = class extends BaseParser {
     const nameNode = node.childForFieldName("name");
     if (!nameNode) return;
     const name = nameNode.text;
+    const parentNode = analysis.nodes.find((n) => n.id === classId);
+    const parentClass = parentNode?.type === "class" ? parentNode.name : void 0;
     const decorators = [];
     if (node.parent?.type === "decorated_definition") {
       for (const sibling of node.parent.children) {
@@ -808,7 +1208,8 @@ var PythonParser = class extends BaseParser {
         classmethod: isClassMethod,
         property: isProperty,
         isPrivate: name.startsWith("_"),
-        isDunder: name.startsWith("__") && name.endsWith("__")
+        isDunder: name.startsWith("__") && name.endsWith("__"),
+        parentClass
       }
     );
     analysis.nodes.push(methodNode);
@@ -1011,6 +1412,12 @@ var FlowBuilder = class {
       currentFile: ""
     });
     this.resolveAllCalls();
+    const enhancedResolver = new EnhancedResolver(this.storage);
+    enhancedResolver.buildHierarchy();
+    const enhancedResolved = enhancedResolver.resolveAllCalls();
+    if (enhancedResolved.length > 0) {
+      console.log(`\u2728 Enhanced resolver: resolved ${enhancedResolved.length} additional calls`);
+    }
     const stats = this.storage.getStats();
     this.emitProgress({
       phase: "complete",