ts_ast_utils.py

import subprocess
import json

TS_PARSER_JS = r"""
const { Project } = require("ts-morph");
let code = "";
process.stdin.setEncoding("utf8");
process.stdin.on("data", chunk => { code += chunk; });
process.stdin.on("end", () => {
  const project = new Project({ useInMemoryFileSystem: true });
  const sourceFile = project.createSourceFile("temp.ts", code);
  const functions = [];
  sourceFile.getFunctions().forEach(fn => {
    let name = fn.getName() || "<anonymous>";
    let body = fn.getBodyText() || "";
    let calls = [];
    fn.forEachDescendant(n => {
      if(n.getKindName() === "CallExpression") {
         try {
           let callee = n.getExpression().getText();
           calls.push(callee);
         } catch(e) {}
      }
    });
    functions.push({name: name, code: body, calls: Array.from(new Set(calls))});
  });
  console.log(JSON.stringify(functions));
});
"""

def extract_ts_function_info(code):
    try:
        result = subprocess.run([
            "node", "-e", TS_PARSER_JS
        ], input=code, text=True, capture_output=True, check=True)
        return json.loads(result.stdout)
    except (subprocess.CalledProcessError, json.JSONDecodeError) as e:
        print("[DEBUG] TypeScript AST parsing error:", e)
        return []