@fairfox/polly 0.2.0 → 0.3.0

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

@@ -757,7 +757,7 @@ class HandlerExtractor {
         const expression = node.getExpression();
         if (Node.isPropertyAccessExpression(expression)) {
           const methodName = expression.getName();
-          if (methodName === "on") {
+          if (methodName === "on" || methodName === "addEventListener") {
             const handler = this.extractHandler(node, context, filePath);
             if (handler) {
               handlers.push(handler);
@@ -765,6 +765,14 @@ class HandlerExtractor {
           }
         }
       }
+      if (Node.isSwitchStatement(node)) {
+        const switchHandlers = this.extractSwitchCaseHandlers(node, context, filePath);
+        handlers.push(...switchHandlers);
+      }
+      if (Node.isVariableDeclaration(node)) {
+        const mapHandlers = this.extractHandlerMapPattern(node, context, filePath);
+        handlers.push(...mapHandlers);
+      }
     });
     return handlers;
   }
@@ -906,6 +914,75 @@ class HandlerExtractor {
     }
     return;
   }
+  extractSwitchCaseHandlers(switchNode, context, filePath) {
+    const handlers = [];
+    try {
+      const expression = switchNode.getExpression();
+      const expressionText = expression.getText();
+      if (!/\.(type|kind|event|action)/.test(expressionText)) {
+        return handlers;
+      }
+      const caseClauses = switchNode.getClauses();
+      for (const clause of caseClauses) {
+        if (Node.isCaseClause(clause)) {
+          const caseExpr = clause.getExpression();
+          let messageType = null;
+          if (Node.isStringLiteral(caseExpr)) {
+            messageType = caseExpr.getLiteralValue();
+          }
+          if (messageType) {
+            const line = clause.getStartLineNumber();
+            handlers.push({
+              messageType,
+              node: context,
+              assignments: [],
+              preconditions: [],
+              postconditions: [],
+              location: { file: filePath, line }
+            });
+          }
+        }
+      }
+    } catch (error) {}
+    return handlers;
+  }
+  extractHandlerMapPattern(varDecl, context, filePath) {
+    const handlers = [];
+    try {
+      const initializer = varDecl.getInitializer();
+      if (!initializer || !Node.isObjectLiteralExpression(initializer)) {
+        return handlers;
+      }
+      const varName = varDecl.getName().toLowerCase();
+      if (!/(handler|listener|callback|event)s?/.test(varName)) {
+        return handlers;
+      }
+      const properties = initializer.getProperties();
+      for (const prop of properties) {
+        if (Node.isPropertyAssignment(prop)) {
+          const nameNode = prop.getNameNode();
+          let messageType = null;
+          if (Node.isStringLiteral(nameNode)) {
+            messageType = nameNode.getLiteralValue();
+          } else if (Node.isIdentifier(nameNode)) {
+            messageType = nameNode.getText();
+          }
+          if (messageType) {
+            const line = prop.getStartLineNumber();
+            handlers.push({
+              messageType,
+              node: context,
+              assignments: [],
+              preconditions: [],
+              postconditions: [],
+              location: { file: filePath, line }
+            });
+          }
+        }
+      }
+    } catch (error) {}
+    return handlers;
+  }
   inferContext(filePath) {
     const path = filePath.toLowerCase();
     if (path.includes("/background/") || path.includes("\\background\\")) {
@@ -926,6 +1003,15 @@ class HandlerExtractor {
     if (path.includes("/offscreen/") || path.includes("\\offscreen\\")) {
       return "offscreen";
     }
+    if (path.includes("/server/") || path.includes("\\server\\") || path.includes("/server.")) {
+      return "server";
+    }
+    if (path.includes("/client/") || path.includes("\\client\\") || path.includes("/client.")) {
+      return "client";
+    }
+    if (path.includes("/worker/") || path.includes("\\worker\\") || path.includes("service-worker")) {
+      return "worker";
+    }
     return "unknown";
   }
 }
@@ -1172,7 +1258,6 @@ async function analyzeCodebase(options) {
   const extractor = new TypeExtractor(options.tsConfigPath);
   return extractor.analyzeCodebase(options.stateFilePath);
 }
-
 // vendor/verify/src/codegen/config.ts
 class ConfigGenerator {
   lines = [];
@@ -1721,6 +1806,7 @@ function validateConfig(configPath) {
 }
 
 // vendor/verify/src/cli.ts
+var __dirname = "/Users/AJT/projects/polly/packages/polly/vendor/verify/src";
 var COLORS = {
   reset: "\x1B[0m",
   red: "\x1B[31m",
@@ -1935,12 +2021,30 @@ async function runFullVerification(configPath) {
   const cfgPath = path3.join(specDir, "UserApp.cfg");
   fs3.writeFileSync(specPath, spec);
   fs3.writeFileSync(cfgPath, cfg);
-  const baseSpecPath = path3.join(process.cwd(), "specs", "tla", "MessageRouter.tla");
-  if (fs3.existsSync(baseSpecPath)) {
+  const possiblePaths = [
+    path3.join(process.cwd(), "specs", "tla", "MessageRouter.tla"),
+    path3.join(__dirname, "..", "specs", "tla", "MessageRouter.tla"),
+    path3.join(__dirname, "..", "..", "specs", "tla", "MessageRouter.tla"),
+    path3.join(process.cwd(), "external", "polly", "packages", "verify", "specs", "tla", "MessageRouter.tla"),
+    path3.join(process.cwd(), "node_modules", "@fairfox", "polly-verify", "specs", "tla", "MessageRouter.tla")
+  ];
+  let baseSpecPath = null;
+  for (const candidatePath of possiblePaths) {
+    if (fs3.existsSync(candidatePath)) {
+      baseSpecPath = candidatePath;
+      break;
+    }
+  }
+  if (baseSpecPath) {
     const destSpecPath = path3.join(specDir, "MessageRouter.tla");
     fs3.copyFileSync(baseSpecPath, destSpecPath);
+    console.log(color("✓ Copied MessageRouter.tla", COLORS.green));
   } else {
     console.log(color("⚠️  Warning: MessageRouter.tla not found, verification may fail", COLORS.yellow));
+    console.log(color(`   Searched in:`, COLORS.gray));
+    for (const searchPath of possiblePaths) {
+      console.log(color(`   - ${searchPath}`, COLORS.gray));
+    }
   }
   console.log(color("✓ Specification generated", COLORS.green));
   console.log(color(`   ${specPath}`, COLORS.gray));
@@ -2071,4 +2175,4 @@ Stack trace:`, COLORS.gray));
   process.exit(1);
 });
 
-//# debugId=1B244FC1308B4F2264756E2164756E21
+//# debugId=8C53367A419D15AA64756E2164756E21