@fairfox/polly 0.72.0 → 0.73.1

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

@@ -1864,14 +1864,37 @@ class HandlerExtractor {
       return null;
     const key = keyArg.getLiteralValue();
     const variableName = this.getVariableNameFromParent(node) || key;
+    const access = this.extractMeshAccess(args[2]);
     return {
       kind,
       key,
       variableName,
       filePath,
-      line: node.getStartLineNumber()
+      line: node.getStartLineNumber(),
+      ...access ? { access } : {}
     };
   }
+  extractMeshAccess(optionsArg) {
+    if (!optionsArg || !Node4.isObjectLiteralExpression(optionsArg))
+      return;
+    const accessProp = optionsArg.getProperty("access");
+    if (!accessProp || !Node4.isPropertyAssignment(accessProp))
+      return;
+    const accessObj = accessProp.getInitializer();
+    if (!accessObj || !Node4.isObjectLiteralExpression(accessObj))
+      return;
+    const predicateText = (name) => {
+      const prop = accessObj.getProperty(name);
+      if (!prop || !Node4.isPropertyAssignment(prop))
+        return;
+      return prop.getInitializer()?.getText().replace(/\s+/g, " ").trim();
+    };
+    const read = predicateText("read");
+    const write = predicateText("write");
+    if (read === undefined && write === undefined)
+      return;
+    return { read: read ?? "unset", write: write ?? "unset" };
+  }
   analyzeFileAndImports(sourceFile, handlers, messageTypes, invalidMessageTypes, stateConstraints, globalStateConstraints, verifiedStates, resources) {
     const filePath = sourceFile.getFilePath();
     if (this.analyzedFiles.has(filePath)) {
@@ -4456,7 +4479,7 @@ class ArchitectureAnalyzer {
       }
     }
     const handlerExtractor = new HandlerExtractor(this.options.tsConfigPath, contextOverrides.size > 0 ? contextOverrides : undefined);
-    const { handlers } = handlerExtractor.extractHandlers();
+    const { handlers, meshOrPeerSignals } = handlerExtractor.extractHandlers();
     const contexts = this.analyzeContexts(entryPoints, handlers);
     const flowAnalyzer = new FlowAnalyzer(this.options.tsConfigPath, handlers, contextOverrides.size > 0 ? contextOverrides : undefined);
     const messageFlows = flowAnalyzer.analyzeFlows();
@@ -4473,6 +4496,7 @@ class ArchitectureAnalyzer {
       contexts,
       messageFlows,
       integrations,
+      ...meshOrPeerSignals.length > 0 ? { meshOrPeerSignals } : {},
       ...adrs.adrs.length > 0 ? { adrs } : {},
       ...repository ? { repository } : {}
     };
@@ -4628,6 +4652,21 @@ var DEFAULT_ELEMENT_STYLES = {
   Container: {
     background: DEFAULT_COLORS.container,
     color: DEFAULT_COLORS.textLight
+  },
+  "Mesh Document": {
+    shape: "Cylinder",
+    background: DEFAULT_COLORS.subscribeHandler,
+    color: DEFAULT_COLORS.textLight
+  },
+  "Peer Document": {
+    shape: "Cylinder",
+    background: DEFAULT_COLORS.queryHandler,
+    color: DEFAULT_COLORS.textDark
+  },
+  "Mesh Transport": {
+    shape: "Pipe",
+    background: DEFAULT_COLORS.service,
+    color: DEFAULT_COLORS.textLight
   }
 };
 var DEFAULT_RELATIONSHIP_STYLES = {
@@ -4654,6 +4693,29 @@ var DEFAULT_RELATIONSHIP_STYLES = {
 var DEFAULT_THEME = "https://static.structurizr.com/themes/default/theme.json";
 
 // tools/visualize/src/codegen/structurizr.ts
+var MESH_TRANSPORT_NODES = [
+  {
+    id: "mesh_net_adapter",
+    name: "MeshNetworkAdapter",
+    description: "Signs and encrypts every mesh operation"
+  },
+  {
+    id: "mesh_webrtc_adapter",
+    name: "MeshWebRTCAdapter",
+    description: "Peer-to-peer WebRTC data channels"
+  },
+  {
+    id: "mesh_signaling_client",
+    name: "MeshSignalingClient",
+    description: "WebRTC offer/answer signalling"
+  },
+  {
+    id: "mesh_signaling_endpoint",
+    name: "Signalling endpoint",
+    description: "The rendezvous server peers exchange WebRTC offers through"
+  }
+];
+
 class StructurizrDSLGenerator {
   analysis;
   options;
@@ -4749,6 +4811,12 @@ class StructurizrDSLGenerator {
     for (const [contextType, contextInfo] of Object.entries(this.analysis.contexts)) {
       parts.push(this.generateContainer(contextType, contextInfo));
     }
+    const meshDocs = this.generateMeshDocuments();
+    if (meshDocs)
+      parts.push(meshDocs);
+    const meshTransport = this.generateMeshTransport();
+    if (meshTransport)
+      parts.push(meshTransport);
     parts.push(this.generateContainerRelationships());
     parts.push("    }");
     return parts.join(`
@@ -4769,6 +4837,49 @@ class StructurizrDSLGenerator {
     return parts.join(`
 `);
   }
+  generateMeshDocuments() {
+    const signals = this.analysis.meshOrPeerSignals ?? [];
+    if (signals.length === 0)
+      return "";
+    const parts = [];
+    const seen = new Set;
+    for (const sig of signals) {
+      if (seen.has(sig.key))
+        continue;
+      seen.add(sig.key);
+      const kindLabel = sig.kind === "mesh" ? "Mesh Document" : "Peer Document";
+      const access = sig.access ? ` · access read=${this.clip(sig.access.read)} write=${this.clip(sig.access.write)}` : "";
+      const description = `${kindLabel} — deriveDocumentId('${sig.key}')${access}`;
+      parts.push(`      ${this.meshDocId(sig.key)} = container "${this.escape(sig.key)}" "${this.escape(description)}" "$${sig.kind}State" {`);
+      parts.push(`        tags "${kindLabel}"`);
+      parts.push("      }");
+    }
+    return parts.join(`
+`);
+  }
+  meshDocId(key) {
+    return `mesh_doc_${this.toId(key)}`;
+  }
+  clip(text, max = 32) {
+    return text.length > max ? `${text.slice(0, max - 1)}…` : text;
+  }
+  contextForFilePath(filePath) {
+    const contextKeys = Object.keys(this.analysis.contexts);
+    if (contextKeys.length === 1)
+      return contextKeys[0];
+    let best;
+    for (const [context, info] of Object.entries(this.analysis.contexts)) {
+      const entry = info.entryPoint;
+      if (!entry)
+        continue;
+      const dir = entry.slice(0, entry.lastIndexOf("/") + 1);
+      if (dir.length > 0 && filePath.includes(dir)) {
+        if (!best || dir.length > best.length)
+          best = { context, length: dir.length };
+      }
+    }
+    return best?.context;
+  }
   generateComponents(contextType, contextInfo) {
     const parts = [];
     const componentDefs = this.buildComponentDefinitionsFromHandlers(contextInfo, contextType);
@@ -5089,9 +5200,74 @@ class StructurizrDSLGenerator {
     parts.push(...this.generateUserRelationships());
     parts.push(...this.generateMessageFlowRelationships());
     parts.push(...this.generateExternalAPIRelationships());
+    parts.push(...this.generateMeshRelationships());
+    parts.push(...this.generateMeshTransportRelationships());
+    return parts.join(`
+`);
+  }
+  generateMeshRelationships() {
+    const signals = this.analysis.meshOrPeerSignals ?? [];
+    const parts = [];
+    const seen = new Set;
+    for (const sig of signals) {
+      const context = this.contextForFilePath(sig.filePath);
+      if (!context)
+        continue;
+      const edgeKey = `${context}->${sig.key}`;
+      if (seen.has(edgeKey))
+        continue;
+      seen.add(edgeKey);
+      parts.push(`      extension.${context} -> extension.${this.meshDocId(sig.key)} "syncs ${this.escape(sig.key)}"`);
+    }
+    return parts;
+  }
+  meshSignalKinds() {
+    const signals = this.analysis.meshOrPeerSignals ?? [];
+    return {
+      mesh: signals.some((s) => s.kind === "mesh"),
+      peer: signals.some((s) => s.kind === "peer")
+    };
+  }
+  generateMeshTransport() {
+    const { mesh, peer } = this.meshSignalKinds();
+    if (!mesh && !peer)
+      return "";
+    const parts = [];
+    if (mesh) {
+      for (const node of MESH_TRANSPORT_NODES) {
+        parts.push(`      ${node.id} = container "${node.name}" "${this.escape(node.description)}" "Mesh Transport" {`);
+        parts.push('        tags "Mesh Transport"');
+        parts.push("      }");
+      }
+    }
+    if (peer) {
+      parts.push(`      peer_relay = container "Peer relay" "${this.escape("The always-on relay $peerState syncs through")}" "Peer Transport" {`);
+      parts.push('        tags "Mesh Transport"');
+      parts.push("      }");
+    }
     return parts.join(`
 `);
   }
+  generateMeshTransportRelationships() {
+    const { mesh, peer } = this.meshSignalKinds();
+    const parts = [];
+    if (mesh) {
+      parts.push('      extension.mesh_net_adapter -> extension.mesh_webrtc_adapter "wraps"');
+      parts.push('      extension.mesh_webrtc_adapter -> extension.mesh_signaling_client "negotiates via"');
+      parts.push('      extension.mesh_signaling_client -> extension.mesh_signaling_endpoint "connects to"');
+    }
+    if (!mesh && !peer)
+      return parts;
+    const seen = new Set;
+    for (const sig of this.analysis.meshOrPeerSignals ?? []) {
+      if (seen.has(sig.key))
+        continue;
+      seen.add(sig.key);
+      const target = sig.kind === "mesh" ? "mesh_net_adapter" : "peer_relay";
+      parts.push(`      extension.${this.meshDocId(sig.key)} -> extension.${target} "syncs through"`);
+    }
+    return parts;
+  }
   generateUserRelationships() {
     const parts = [];
     const uiContexts = ["popup", "options", "devtools"];
@@ -6290,4 +6466,4 @@ main().catch((_error) => {
   process.exit(1);
 });
 
-//# debugId=76D47496DA7C4C5D64756E2164756E21
+//# debugId=2FC255DE0B3BC47C64756E2164756E21