@fairfox/polly 0.3.4 → 0.3.8

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

@@ -1321,11 +1321,7 @@ class HandlerExtractor {
   typeGuardCache;
   constructor(tsConfigPath) {
     this.project = new Project4({
-      tsConfigFilePath: tsConfigPath,
-      compilerOptions: {
-        allowImportingTsExtensions: true,
-        moduleResolution: 99
-      }
+      tsConfigFilePath: tsConfigPath
     });
     this.typeGuardCache = new WeakMap;
   }
@@ -1382,8 +1378,12 @@ class HandlerExtractor {
         handlers.push(...mapHandlers);
       }
       if (Node4.isIfStatement(node)) {
-        const typeGuardHandlers = this.extractTypeGuardHandlers(node, context, filePath);
-        handlers.push(...typeGuardHandlers);
+        const parent = node.getParent();
+        const isElseIf = parent && Node4.isIfStatement(parent);
+        if (!isElseIf) {
+          const typeGuardHandlers = this.extractTypeGuardHandlers(node, context, filePath);
+          handlers.push(...typeGuardHandlers);
+        }
       }
     });
     return handlers;
@@ -1685,9 +1685,8 @@ class HandlerExtractor {
     const typeGuards = new Map;
     sourceFile.forEachDescendant((node) => {
       if (Node4.isFunctionDeclaration(node) || Node4.isFunctionExpression(node) || Node4.isArrowFunction(node)) {
-        const returnType = node.getReturnType();
-        const returnTypeText = returnType.getText();
-        if (/is\s+\w+/.test(returnTypeText)) {
+        const returnTypeNode = node.getReturnTypeNode();
+        if (returnTypeNode && Node4.isTypePredicate(returnTypeNode)) {
           let functionName;
           if (Node4.isFunctionDeclaration(node)) {
             functionName = node.getName();
@@ -1703,10 +1702,10 @@ class HandlerExtractor {
             }
           }
           if (functionName) {
+            const typeNode = returnTypeNode.getTypeNode();
             let messageType = null;
-            const typeMatch = returnTypeText.match(/is\s+(\w+)/);
-            if (typeMatch) {
-              const typeName = typeMatch[1];
+            if (typeNode) {
+              const typeName = typeNode.getText();
               messageType = this.extractMessageTypeFromTypeName(typeName);
             }
             if (!messageType) {
@@ -1740,34 +1739,38 @@ class HandlerExtractor {
       }
       for (const def of definitions) {
         if (Node4.isFunctionDeclaration(def) || Node4.isFunctionExpression(def) || Node4.isArrowFunction(def)) {
-          const returnType = def.getReturnType();
-          const returnTypeText = returnType.getText();
+          const returnTypeNode = def.getReturnTypeNode();
           if (process.env.POLLY_DEBUG) {
-            console.log(`[DEBUG] Function ${funcName} return type: ${returnTypeText}`);
+            const returnType = def.getReturnType().getText();
+            console.log(`[DEBUG] Function ${funcName} return type (resolved): ${returnType}`);
+            console.log(`[DEBUG] Has return type node: ${!!returnTypeNode}`);
+            console.log(`[DEBUG] Is type predicate node: ${returnTypeNode && Node4.isTypePredicate(returnTypeNode)}`);
           }
-          if (/is\s+\w+/.test(returnTypeText)) {
-            const typeMatch = returnTypeText.match(/is\s+(\w+)/);
-            if (typeMatch) {
-              const typeName = typeMatch[1];
+          if (returnTypeNode && Node4.isTypePredicate(returnTypeNode)) {
+            const typeNode = returnTypeNode.getTypeNode();
+            if (typeNode) {
+              const typeName = typeNode.getText();
               const messageType = this.extractMessageTypeFromTypeName(typeName);
-              if (process.env.POLLY_DEBUG) {
-                console.log(`[DEBUG] Resolved ${funcName} → ${messageType}`);
-              }
-              return messageType;
-            }
-            const body = def.getBody();
-            if (body) {
-              const bodyText = body.getText();
-              const typeValueMatch = bodyText.match(/\.type\s*===?\s*['"](\w+)['"]/);
-              if (typeValueMatch) {
-                const messageType = typeValueMatch[1];
+              if (messageType) {
                 if (process.env.POLLY_DEBUG) {
-                  console.log(`[DEBUG] Resolved ${funcName} → ${messageType} (from body)`);
+                  console.log(`[DEBUG] Resolved ${funcName} → ${messageType} (from AST type predicate)`);
                 }
                 return messageType;
               }
             }
           }
+          const body = def.getBody();
+          if (body) {
+            const bodyText = body.getText();
+            const typeValueMatch = bodyText.match(/\.type\s*===?\s*['"](\w+)['"]/);
+            if (typeValueMatch) {
+              const messageType = typeValueMatch[1];
+              if (process.env.POLLY_DEBUG) {
+                console.log(`[DEBUG] Resolved ${funcName} → ${messageType} (from body)`);
+              }
+              return messageType;
+            }
+          }
         }
       }
     } catch (error) {
@@ -2768,10 +2771,11 @@ async function generateCommand() {
     console.log(color(`
 \uD83D\uDCDD Generating Structurizr DSL...
 `, COLORS.blue));
+    const contextTypes = Object.keys(analysis.contexts);
     const dsl = generateStructurizrDSL(analysis, {
       includeDynamicDiagrams: true,
       includeComponentDiagrams: true,
-      componentDiagramContexts: ["background"]
+      componentDiagramContexts: contextTypes.length > 0 ? contextTypes : ["background"]
     });
     const outputDir = path6.join(process.cwd(), "docs");
     if (!fs6.existsSync(outputDir)) {
@@ -3003,4 +3007,4 @@ Stack trace:`, COLORS.gray));
   process.exit(1);
 });
 
-//# debugId=FFB153526798575464756E2164756E21
+//# debugId=78733C0D94700D7D64756E2164756E21