@neat.is/core 0.6.0 → 0.6.1

package/dist/cli.cjs

@@ -5363,6 +5363,7 @@ init_cjs_shims();
 var import_node_path23 = __toESM(require("path"), 1);
 var import_tree_sitter2 = __toESM(require("tree-sitter"), 1);
 var import_tree_sitter_javascript2 = __toESM(require("tree-sitter-javascript"), 1);
+var import_tree_sitter_python2 = __toESM(require("tree-sitter-python"), 1);
 var import_types11 = require("@neat.is/types");
 var PARSE_CHUNK2 = 16384;
 function parseSource2(parser, source) {
@@ -5375,6 +5376,11 @@ function makeJsParser2() {
   p.setLanguage(import_tree_sitter_javascript2.default);
   return p;
 }
+function makePyParser2() {
+  const p = new import_tree_sitter2.default();
+  p.setLanguage(import_tree_sitter_python2.default);
+  return p;
+}
 var ROUTER_METHODS = /* @__PURE__ */ new Set([
   "get",
   "post",
@@ -5387,6 +5393,7 @@ var ROUTER_METHODS = /* @__PURE__ */ new Set([
 ]);
 var NEXT_APP_METHODS = /* @__PURE__ */ new Set(["GET", "POST", "PUT", "PATCH", "DELETE", "HEAD", "OPTIONS"]);
 var JS_ROUTE_EXTENSIONS = /* @__PURE__ */ new Set([".js", ".jsx", ".mjs", ".cjs", ".ts", ".tsx"]);
+var FASTAPI_METHODS = /* @__PURE__ */ new Set(["get", "post", "put", "patch", "delete", "options", "head", "trace"]);
 function canonicalizeTemplate(raw) {
   let p = raw.split("?")[0].split("#")[0];
   if (!p.startsWith("/")) p = "/" + p;
@@ -5614,8 +5621,102 @@ function nextRoutesFromFile(source, relFile, parser) {
   }
   return [];
 }
+function pyStaticStringText(node) {
+  if (node.type !== "string") return null;
+  for (let i = 0; i < node.namedChildCount; i++) {
+    const child = node.namedChild(i);
+    if (child?.type === "interpolation") return null;
+    if (child?.type === "string_content") return child.text;
+  }
+  return "";
+}
+function keywordArrayStrings(argsNode, key) {
+  for (let i = 0; i < argsNode.namedChildCount; i++) {
+    const arg = argsNode.namedChild(i);
+    if (arg?.type !== "keyword_argument") continue;
+    if (arg.childForFieldName("name")?.text !== key) continue;
+    const val = arg.childForFieldName("value");
+    if (!val || val.type !== "list") return [];
+    const out = [];
+    for (let j = 0; j < val.namedChildCount; j++) {
+      const el = val.namedChild(j);
+      if (el?.type === "string") {
+        const s = pyStaticStringText(el);
+        if (s) out.push(s);
+      }
+    }
+    return out;
+  }
+  return [];
+}
+function collectApiRouterPrefixes(root) {
+  const prefixes = /* @__PURE__ */ new Map();
+  walk(root, (node) => {
+    if (node.type !== "assignment") return;
+    const right = node.childForFieldName("right");
+    if (!right || right.type !== "call") return;
+    const fn = right.childForFieldName("function");
+    if (!fn) return;
+    const ctor = fn.type === "attribute" ? fn.childForFieldName("attribute")?.text : fn.text;
+    if (ctor !== "APIRouter") return;
+    const left = node.childForFieldName("left");
+    if (!left || left.type !== "identifier") return;
+    const args = right.childForFieldName("arguments");
+    if (!args) return;
+    for (let i = 0; i < args.namedChildCount; i++) {
+      const arg = args.namedChild(i);
+      if (arg?.type !== "keyword_argument") continue;
+      if (arg.childForFieldName("name")?.text !== "prefix") continue;
+      const val = arg.childForFieldName("value");
+      const p = val ? pyStaticStringText(val) : null;
+      if (p !== null) prefixes.set(left.text, p);
+    }
+  });
+  return prefixes;
+}
+function fastapiRoutesFromSource(source, parser) {
+  const tree = parseSource2(parser, source);
+  const prefixes = collectApiRouterPrefixes(tree.rootNode);
+  const out = [];
+  walk(tree.rootNode, (node) => {
+    if (node.type !== "decorator") return;
+    const call = node.namedChild(0);
+    if (!call || call.type !== "call") return;
+    const fn = call.childForFieldName("function");
+    if (!fn || fn.type !== "attribute") return;
+    const method = fn.childForFieldName("attribute")?.text?.toLowerCase();
+    if (!method) return;
+    const isVerb = FASTAPI_METHODS.has(method);
+    if (!isVerb && method !== "api_route") return;
+    const args = call.childForFieldName("arguments");
+    const first = args?.namedChild(0);
+    if (!first || first.type !== "string") return;
+    const rawPath = pyStaticStringText(first);
+    if (rawPath === null || !rawPath.startsWith("/")) return;
+    const obj = fn.childForFieldName("object")?.text;
+    const prefix = obj ? prefixes.get(obj) ?? "" : "";
+    const pathTemplate = canonicalizeTemplate(prefix + rawPath);
+    const line = node.startPosition.row + 1;
+    if (isVerb) {
+      out.push({ method: method.toUpperCase(), pathTemplate, line, framework: "fastapi" });
+      return;
+    }
+    const methods = keywordArrayStrings(args, "methods");
+    const list = methods.length > 0 ? methods : ["ALL"];
+    for (const m of list) {
+      out.push({
+        method: m === "ALL" ? "ALL" : m.toUpperCase(),
+        pathTemplate,
+        line,
+        framework: "fastapi"
+      });
+    }
+  });
+  return out;
+}
 async function addRoutes(graph, services) {
   const jsParser = makeJsParser2();
+  const pyParser = makePyParser2();
   let nodesAdded = 0;
   let edgesAdded = 0;
   for (const service of services) {
@@ -5627,15 +5728,20 @@ async function addRoutes(graph, services) {
     const hasFastify = deps["fastify"] !== void 0;
     const hasHono = deps["hono"] !== void 0;
     const hasNext = deps["next"] !== void 0;
-    if (!hasExpress && !hasFastify && !hasHono && !hasNext) continue;
+    const hasFastapi = deps["fastapi"] !== void 0;
+    if (!hasExpress && !hasFastify && !hasHono && !hasNext && !hasFastapi) continue;
     const files = await loadSourceFiles(service.dir);
     for (const file of files) {
       if (isTestPath(file.path)) continue;
-      if (!JS_ROUTE_EXTENSIONS.has(import_node_path23.default.extname(file.path))) continue;
+      const ext = import_node_path23.default.extname(file.path);
+      const isPy = ext === ".py";
+      if (!JS_ROUTE_EXTENSIONS.has(ext) && !isPy) continue;
       const relFile = toPosix2(import_node_path23.default.relative(service.dir, file.path));
       let routes;
       try {
-        if (hasNext && (isNextAppRouteFile(relFile) || isNextPagesApiFile(relFile))) {
+        if (isPy) {
+          routes = hasFastapi ? fastapiRoutesFromSource(file.content, pyParser) : [];
+        } else if (hasNext && (isNextAppRouteFile(relFile) || isNextPagesApiFile(relFile))) {
           routes = nextRoutesFromFile(file.content, relFile, jsParser);
         } else if (hasExpress || hasFastify || hasHono) {
           routes = serverRoutesFromSource(file.content, jsParser, hasExpress, hasFastify, hasHono);
@@ -5819,7 +5925,7 @@ init_cjs_shims();
 var import_node_path25 = __toESM(require("path"), 1);
 var import_tree_sitter3 = __toESM(require("tree-sitter"), 1);
 var import_tree_sitter_javascript3 = __toESM(require("tree-sitter-javascript"), 1);
-var import_tree_sitter_python2 = __toESM(require("tree-sitter-python"), 1);
+var import_tree_sitter_python3 = __toESM(require("tree-sitter-python"), 1);
 var import_types13 = require("@neat.is/types");
 var STRING_LITERAL_NODE_TYPES = /* @__PURE__ */ new Set(["string_fragment", "string_content"]);
 var JSX_EXTERNAL_LINK_TAGS = /* @__PURE__ */ new Set(["a", "Link", "NavLink", "ExternalLink", "Anchor"]);
@@ -5874,14 +5980,14 @@ function makeJsParser3() {
   p.setLanguage(import_tree_sitter_javascript3.default);
   return p;
 }
-function makePyParser2() {
+function makePyParser3() {
   const p = new import_tree_sitter3.default();
-  p.setLanguage(import_tree_sitter_python2.default);
+  p.setLanguage(import_tree_sitter_python3.default);
   return p;
 }
 async function addHttpCallEdges(graph, services) {
   const jsParser = makeJsParser3();
-  const pyParser = makePyParser2();
+  const pyParser = makePyParser3();
   const { knownHosts, hostToNodeId } = buildServiceHostIndex(services);
   let nodesAdded = 0;
   let edgesAdded = 0;