@fairfox/polly 0.6.0 → 0.6.1

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

@@ -1671,6 +1671,7 @@ class HandlerExtractor {
   extractHandlers() {
     const handlers = [];
     const messageTypes = new Set;
+    const invalidMessageTypes = new Set;
     const sourceFiles = this.project.getSourceFiles();
     if (process.env["POLLY_DEBUG"]) {
       console.log(`[DEBUG] Loaded ${sourceFiles.length} source files`);
@@ -1684,17 +1685,30 @@ class HandlerExtractor {
       const fileHandlers = this.extractFromFile(sourceFile);
       handlers.push(...fileHandlers);
       for (const handler of fileHandlers) {
-        messageTypes.add(handler.messageType);
+        if (this.isValidTLAIdentifier(handler.messageType)) {
+          messageTypes.add(handler.messageType);
+        } else {
+          invalidMessageTypes.add(handler.messageType);
+        }
       }
     }
     if (process.env["POLLY_DEBUG"]) {
       console.log(`[DEBUG] Total handlers extracted: ${handlers.length}`);
+      if (invalidMessageTypes.size > 0) {
+        console.log(`[DEBUG] Filtered ${invalidMessageTypes.size} invalid message type(s) from handlers`);
+      }
     }
     return {
       handlers,
       messageTypes
     };
   }
+  isValidTLAIdentifier(s) {
+    if (!s || s.length === 0) {
+      return false;
+    }
+    return /^[a-zA-Z][a-zA-Z0-9_]*$/.test(s);
+  }
   extractFromFile(sourceFile) {
     const handlers = [];
     const filePath = sourceFile.getFilePath();
@@ -4003,4 +4017,4 @@ Stack trace:`, COLORS.gray));
   process.exit(1);
 });
 
-//# debugId=3ECE56D6DA69F32764756E2164756E21
+//# debugId=698279082E1D6F3764756E2164756E21