@fairfox/polly 0.5.0 → 0.5.1

package/dist/vendor/visualize/src/cli.js

@@ -1332,15 +1332,37 @@ class RelationshipExtractor {
         const expr = descendant.getExpression();
         if (Node4.isIdentifier(expr)) {
           const functionName = expr.getText();
-          const functionDecl = sourceFile.getFunction(functionName);
+          let functionDecl = sourceFile.getFunction(functionName);
+          let targetSourceFile = sourceFile;
+          if (!functionDecl) {
+            const resolved = this.resolveImportedFunction(functionName, sourceFile);
+            if (resolved) {
+              functionDecl = resolved.functionDecl;
+              targetSourceFile = resolved.sourceFile;
+            }
+          }
           if (functionDecl && !visited.has(functionName)) {
             visited.add(functionName);
             const body = functionDecl.getBody();
             if (body) {
-              this.extractFromNode(body, sourceFile, handlerName, relationships, visited);
+              this.extractFromNode(body, targetSourceFile, handlerName, relationships, visited);
             }
             return;
           }
+          if (!functionDecl) {
+            const componentFromName = this.inferComponentFromFunctionName(functionName);
+            if (componentFromName) {
+              relationships.push({
+                from: this.toComponentId(handlerName),
+                to: componentFromName,
+                description: `Calls ${functionName}()`,
+                technology: "Function Call",
+                confidence: "medium",
+                evidence: [`Function call: ${functionName}`]
+              });
+              return;
+            }
+          }
         }
         if (Node4.isPropertyAccessExpression(expr)) {
           const rel2 = this.extractFromPropertyAccess(expr, handlerName);
@@ -1420,9 +1442,12 @@ class RelationshipExtractor {
       return null;
     }
     const fullChain = propAccess.getText();
-    const objectExpr = propAccess.getExpression();
-    const objectName = objectExpr.getText();
     const methodName = propAccess.getName();
+    let rootObject = propAccess.getExpression();
+    while (Node4.isPropertyAccessExpression(rootObject)) {
+      rootObject = rootObject.getExpression();
+    }
+    const objectName = rootObject.getText();
     const targetComponent = this.inferComponentFromCall(objectName, methodName);
     if (!targetComponent) {
       return null;
@@ -1485,6 +1510,7 @@ class RelationshipExtractor {
       database: "database",
       repos: "repositories",
       repository: "repositories",
+      repositories: "repositories",
       cache: "cache",
       storage: "storage",
       ai: "ai_service",
@@ -1498,6 +1524,49 @@ class RelationshipExtractor {
     const normalized = objectName.toLowerCase();
     return mappings[normalized] || null;
   }
+  inferComponentFromFunctionName(functionName) {
+    const normalized = functionName.toLowerCase();
+    if (normalized.startsWith("get") || normalized.startsWith("create")) {
+      const suffix = functionName.substring(normalized.startsWith("get") ? 3 : 6);
+      const suffixLower = suffix.toLowerCase();
+      if (suffixLower.includes("database") || suffixLower === "db" || suffixLower.includes("dbconnection") || suffixLower.includes("connection")) {
+        return "db_client";
+      }
+      if (suffixLower.includes("repositories") || suffixLower.includes("repos") || suffixLower.includes("repository")) {
+        return "repositories";
+      }
+      if (suffixLower.includes("service")) {
+        const serviceMatch = suffix.match(/^(.+?)Service/i);
+        if (serviceMatch) {
+          return this.toComponentId(`${serviceMatch[1]}_service`);
+        }
+        return "service";
+      }
+      if (suffixLower.includes("cache")) {
+        return "cache";
+      }
+      if (suffixLower.includes("storage")) {
+        return "storage";
+      }
+      if (suffixLower.includes("ai") || suffixLower.includes("llm")) {
+        return "ai_service";
+      }
+      if (suffixLower.includes("logger")) {
+        return "logger";
+      }
+    }
+    if (normalized.startsWith("init") || normalized.startsWith("setup")) {
+      const suffix = functionName.substring(normalized.startsWith("init") ? 4 : 5);
+      const suffixLower = suffix.toLowerCase();
+      if (suffixLower.includes("database") || suffixLower === "db") {
+        return "db_client";
+      }
+      if (suffixLower.includes("cache")) {
+        return "cache";
+      }
+    }
+    return null;
+  }
   resolveComponentFromImport(functionName, sourceFile) {
     for (const importDecl of sourceFile.getImportDeclarations()) {
       const namedImports = importDecl.getNamedImports();
@@ -1521,6 +1590,40 @@ class RelationshipExtractor {
     }
     return null;
   }
+  resolveImportedFunction(functionName, sourceFile) {
+    try {
+      for (const importDecl of sourceFile.getImportDeclarations()) {
+        const namedImports = importDecl.getNamedImports();
+        for (const namedImport of namedImports) {
+          if (namedImport.getName() === functionName) {
+            const moduleSpecifier = importDecl.getModuleSpecifierSourceFile();
+            if (!moduleSpecifier)
+              continue;
+            const functionDecl = moduleSpecifier.getFunction(functionName);
+            if (functionDecl) {
+              return {
+                functionDecl,
+                sourceFile: moduleSpecifier
+              };
+            }
+            const variableDecl = moduleSpecifier.getVariableDeclaration(functionName);
+            if (variableDecl) {
+              const initializer = variableDecl.getInitializer();
+              if (initializer && (Node4.isArrowFunction(initializer) || Node4.isFunctionExpression(initializer))) {
+                return {
+                  functionDecl: initializer,
+                  sourceFile: moduleSpecifier
+                };
+              }
+            }
+          }
+        }
+      }
+    } catch (error) {
+      return null;
+    }
+    return null;
+  }
   inferDatabaseOperation(callExpr) {
     if (callExpr.includes("query") || callExpr.includes("select")) {
       return "Reads from database";
@@ -3900,4 +4003,4 @@ Stack trace:`, COLORS.gray));
   process.exit(1);
 });
 
-//# debugId=59C09397E2AB6BFF64756E2164756E21
+//# debugId=5BFF854178F00CEB64756E2164756E21