@jefuriiij/synthra 0.1.6 → 0.1.7

package/dist/server/index.js

@@ -1013,7 +1013,7 @@ async function parseRust(f, source) {
 }
 
 // src/scanner/parsers/typescript.ts
-var QUERY12 = `
+var TS_QUERY = `
 (function_declaration name: (identifier) @function.name) @function
 (class_declaration name: (type_identifier) @class.name) @class
 (interface_declaration name: (type_identifier) @interface.name) @interface
@@ -1023,11 +1023,22 @@ var QUERY12 = `
 (lexical_declaration (variable_declarator name: (identifier) @const-fn.name value: [(arrow_function) (function_expression)])) @const-fn
 (import_statement source: (string) @import)
 `;
+var JS_QUERY = `
+(function_declaration name: (identifier) @function.name) @function
+(class_declaration name: (identifier) @class.name) @class
+(method_definition name: (property_identifier) @method.name) @method
+(lexical_declaration (variable_declarator name: (identifier) @const-fn.name value: [(arrow_function) (function_expression)])) @const-fn
+(import_statement source: (string) @import)
+(call_expression function: (identifier) @_require_fn arguments: (arguments . (string) @require_source))
+`;
 function grammarFor(ext) {
   if (ext === ".tsx" || ext === ".jsx") return "tsx";
   if (ext === ".js" || ext === ".cjs" || ext === ".mjs") return "javascript";
   return "typescript";
 }
+function queryFor(grammar) {
+  return grammar === "javascript" ? JS_QUERY : TS_QUERY;
+}
 function unquote(s) {
   return s.replace(/^["'`]|["'`]$/g, "");
 }
@@ -1051,7 +1062,7 @@ async function parseTypeScript(f, source) {
     const { parser, language } = await createParser(grammar);
     const tree = parser.parse(source);
     if (!tree) return { file: f, source, symbols, imports, calls: [] };
-    const query = language.query(QUERY12);
+    const query = language.query(queryFor(grammar));
     const matches = query.matches(tree.rootNode);
     for (const match of matches) {
       const byName = /* @__PURE__ */ new Map();
@@ -1068,7 +1079,15 @@ async function parseTypeScript(f, source) {
         continue;
       }
       const importNode = byName.get("import");
-      if (importNode) imports.push(unquote(importNode.text));
+      if (importNode) {
+        imports.push(unquote(importNode.text));
+        continue;
+      }
+      const requireFn = byName.get("_require_fn");
+      const requireSource = byName.get("require_source");
+      if (requireFn && requireSource && requireFn.text === "require") {
+        imports.push(unquote(requireSource.text));
+      }
     }
     const seen = /* @__PURE__ */ new Set();
     symbols = symbols.filter((s) => {