@vibecheckai/cli 3.1.0 → 3.1.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (160) hide show
  1. package/bin/.generated +25 -25
  2. package/bin/dev/run-v2-torture.js +30 -30
  3. package/bin/registry.js +105 -105
  4. package/bin/runners/lib/__tests__/entitlements-v2.test.js +295 -295
  5. package/bin/runners/lib/analysis-core.js +271 -271
  6. package/bin/runners/lib/analyzers.js +579 -579
  7. package/bin/runners/lib/auth-truth.js +193 -193
  8. package/bin/runners/lib/backup.js +62 -62
  9. package/bin/runners/lib/billing.js +107 -107
  10. package/bin/runners/lib/claims.js +118 -118
  11. package/bin/runners/lib/cli-output.js +368 -368
  12. package/bin/runners/lib/cli-ui.js +540 -540
  13. package/bin/runners/lib/contracts/auth-contract.js +202 -202
  14. package/bin/runners/lib/contracts/env-contract.js +181 -181
  15. package/bin/runners/lib/contracts/external-contract.js +206 -206
  16. package/bin/runners/lib/contracts/guard.js +168 -168
  17. package/bin/runners/lib/contracts/index.js +89 -89
  18. package/bin/runners/lib/contracts/plan-validator.js +311 -311
  19. package/bin/runners/lib/contracts/route-contract.js +199 -199
  20. package/bin/runners/lib/contracts.js +804 -804
  21. package/bin/runners/lib/detect.js +89 -89
  22. package/bin/runners/lib/detectors-v2.js +703 -703
  23. package/bin/runners/lib/doctor/autofix.js +254 -254
  24. package/bin/runners/lib/doctor/index.js +37 -37
  25. package/bin/runners/lib/doctor/modules/dependencies.js +325 -325
  26. package/bin/runners/lib/doctor/modules/index.js +46 -46
  27. package/bin/runners/lib/doctor/modules/network.js +250 -250
  28. package/bin/runners/lib/doctor/modules/project.js +312 -312
  29. package/bin/runners/lib/doctor/modules/runtime.js +224 -224
  30. package/bin/runners/lib/doctor/modules/security.js +348 -348
  31. package/bin/runners/lib/doctor/modules/system.js +213 -213
  32. package/bin/runners/lib/doctor/modules/vibecheck.js +394 -394
  33. package/bin/runners/lib/doctor/reporter.js +262 -262
  34. package/bin/runners/lib/doctor/service.js +262 -262
  35. package/bin/runners/lib/doctor/types.js +113 -113
  36. package/bin/runners/lib/doctor/ui.js +263 -263
  37. package/bin/runners/lib/doctor-v2.js +608 -608
  38. package/bin/runners/lib/drift.js +425 -425
  39. package/bin/runners/lib/enforcement.js +72 -72
  40. package/bin/runners/lib/enterprise-detect.js +603 -603
  41. package/bin/runners/lib/enterprise-init.js +942 -942
  42. package/bin/runners/lib/entitlements-v2.js +490 -489
  43. package/bin/runners/lib/entitlements.js +6 -3
  44. package/bin/runners/lib/env-resolver.js +417 -417
  45. package/bin/runners/lib/env-template.js +66 -66
  46. package/bin/runners/lib/env.js +189 -189
  47. package/bin/runners/lib/extractors/client-calls.js +990 -990
  48. package/bin/runners/lib/extractors/fastify-route-dump.js +573 -573
  49. package/bin/runners/lib/extractors/fastify-routes.js +426 -426
  50. package/bin/runners/lib/extractors/index.js +363 -363
  51. package/bin/runners/lib/extractors/next-routes.js +524 -524
  52. package/bin/runners/lib/extractors/proof-graph.js +431 -431
  53. package/bin/runners/lib/extractors/route-matcher.js +451 -451
  54. package/bin/runners/lib/extractors/truthpack-v2.js +377 -377
  55. package/bin/runners/lib/extractors/ui-bindings.js +547 -547
  56. package/bin/runners/lib/findings-schema.js +281 -281
  57. package/bin/runners/lib/firewall-prompt.js +50 -50
  58. package/bin/runners/lib/graph/graph-builder.js +265 -265
  59. package/bin/runners/lib/graph/html-renderer.js +413 -413
  60. package/bin/runners/lib/graph/index.js +32 -32
  61. package/bin/runners/lib/graph/runtime-collector.js +215 -215
  62. package/bin/runners/lib/graph/static-extractor.js +518 -518
  63. package/bin/runners/lib/html-report.js +650 -650
  64. package/bin/runners/lib/init-wizard.js +308 -308
  65. package/bin/runners/lib/llm.js +75 -75
  66. package/bin/runners/lib/meter.js +61 -61
  67. package/bin/runners/lib/missions/evidence.js +126 -126
  68. package/bin/runners/lib/missions/plan.js +69 -69
  69. package/bin/runners/lib/missions/templates.js +192 -192
  70. package/bin/runners/lib/patch.js +40 -40
  71. package/bin/runners/lib/permissions/auth-model.js +213 -213
  72. package/bin/runners/lib/permissions/idor-prover.js +205 -205
  73. package/bin/runners/lib/permissions/index.js +45 -45
  74. package/bin/runners/lib/permissions/matrix-builder.js +198 -198
  75. package/bin/runners/lib/pkgjson.js +28 -28
  76. package/bin/runners/lib/policy.js +295 -295
  77. package/bin/runners/lib/preflight.js +142 -142
  78. package/bin/runners/lib/reality/correlation-detectors.js +359 -359
  79. package/bin/runners/lib/reality/index.js +318 -318
  80. package/bin/runners/lib/reality/request-hashing.js +416 -416
  81. package/bin/runners/lib/reality/request-mapper.js +453 -453
  82. package/bin/runners/lib/reality/safety-rails.js +463 -463
  83. package/bin/runners/lib/reality/semantic-snapshot.js +408 -408
  84. package/bin/runners/lib/reality/toast-detector.js +393 -393
  85. package/bin/runners/lib/reality-findings.js +84 -84
  86. package/bin/runners/lib/receipts.js +179 -179
  87. package/bin/runners/lib/redact.js +29 -29
  88. package/bin/runners/lib/replay/capsule-manager.js +154 -154
  89. package/bin/runners/lib/replay/index.js +263 -263
  90. package/bin/runners/lib/replay/player.js +348 -348
  91. package/bin/runners/lib/replay/recorder.js +331 -331
  92. package/bin/runners/lib/report-engine.js +447 -447
  93. package/bin/runners/lib/report-html.js +1499 -1499
  94. package/bin/runners/lib/report-templates.js +969 -969
  95. package/bin/runners/lib/report.js +135 -135
  96. package/bin/runners/lib/route-detection.js +1140 -1140
  97. package/bin/runners/lib/route-truth.js +477 -477
  98. package/bin/runners/lib/sandbox/index.js +59 -59
  99. package/bin/runners/lib/sandbox/proof-chain.js +399 -399
  100. package/bin/runners/lib/sandbox/sandbox-runner.js +205 -205
  101. package/bin/runners/lib/sandbox/worktree.js +174 -174
  102. package/bin/runners/lib/schema-validator.js +350 -350
  103. package/bin/runners/lib/schemas/contracts.schema.json +160 -160
  104. package/bin/runners/lib/schemas/finding.schema.json +100 -100
  105. package/bin/runners/lib/schemas/mission-pack.schema.json +206 -206
  106. package/bin/runners/lib/schemas/proof-graph.schema.json +176 -176
  107. package/bin/runners/lib/schemas/reality-report.schema.json +162 -162
  108. package/bin/runners/lib/schemas/share-pack.schema.json +180 -180
  109. package/bin/runners/lib/schemas/ship-report.schema.json +117 -117
  110. package/bin/runners/lib/schemas/truthpack-v2.schema.json +303 -303
  111. package/bin/runners/lib/schemas/validator.js +438 -438
  112. package/bin/runners/lib/score-history.js +282 -282
  113. package/bin/runners/lib/server-usage.js +12 -0
  114. package/bin/runners/lib/share-pack.js +239 -239
  115. package/bin/runners/lib/snippets.js +67 -67
  116. package/bin/runners/lib/truth.js +667 -667
  117. package/bin/runners/lib/upsell.js +510 -510
  118. package/bin/runners/lib/usage.js +153 -153
  119. package/bin/runners/lib/validate-patch.js +156 -156
  120. package/bin/runners/lib/verdict-engine.js +628 -628
  121. package/bin/runners/reality/engine.js +917 -917
  122. package/bin/runners/reality/flows.js +122 -122
  123. package/bin/runners/reality/report.js +378 -378
  124. package/bin/runners/reality/session.js +193 -193
  125. package/bin/runners/runAuth.js +51 -0
  126. package/bin/runners/runClaimVerifier.js +483 -483
  127. package/bin/runners/runContext.js +56 -56
  128. package/bin/runners/runContextCompiler.js +385 -385
  129. package/bin/runners/runCtx.js +674 -674
  130. package/bin/runners/runCtxDiff.js +301 -301
  131. package/bin/runners/runCtxGuard.js +176 -176
  132. package/bin/runners/runCtxSync.js +116 -116
  133. package/bin/runners/runGate.js +17 -17
  134. package/bin/runners/runGraph.js +454 -454
  135. package/bin/runners/runGuard.js +168 -168
  136. package/bin/runners/runInitGha.js +164 -164
  137. package/bin/runners/runInstall.js +277 -277
  138. package/bin/runners/runInteractive.js +388 -388
  139. package/bin/runners/runLabs.js +340 -340
  140. package/bin/runners/runMissionGenerator.js +282 -282
  141. package/bin/runners/runPR.js +255 -255
  142. package/bin/runners/runPermissions.js +304 -304
  143. package/bin/runners/runPreflight.js +580 -553
  144. package/bin/runners/runProve.js +1252 -1252
  145. package/bin/runners/runReality.js +1328 -1328
  146. package/bin/runners/runReplay.js +499 -499
  147. package/bin/runners/runReport.js +584 -584
  148. package/bin/runners/runShare.js +212 -212
  149. package/bin/runners/runStatus.js +138 -138
  150. package/bin/runners/runTruthpack.js +636 -636
  151. package/bin/runners/runVerify.js +272 -272
  152. package/bin/runners/runWatch.js +407 -407
  153. package/bin/vibecheck.js +2 -1
  154. package/mcp-server/consolidated-tools.js +804 -804
  155. package/mcp-server/package.json +1 -1
  156. package/mcp-server/tools/index.js +72 -72
  157. package/mcp-server/truth-context.js +581 -581
  158. package/mcp-server/truth-firewall-tools.js +1500 -1500
  159. package/package.json +1 -1
  160. package/bin/runners/runProof.zip +0 -0
@@ -1,518 +1,518 @@
1
- /**
2
- * Static Edge Extractor
3
- * Extracts causal edges from AST analysis:
4
- * - UI actions → client functions
5
- * - Client functions → network calls
6
- * - Network calls → server routes
7
- * - Server routes → handlers
8
- * - Handlers → DB/external calls
9
- */
10
-
11
- "use strict";
12
-
13
- const fg = require("fast-glob");
14
- const fs = require("fs");
15
- const path = require("path");
16
- const crypto = require("crypto");
17
- const parser = require("@babel/parser");
18
- const traverse = require("@babel/traverse").default;
19
- const t = require("@babel/types");
20
-
21
- function sha256(text) {
22
- return crypto.createHash("sha256").update(text).digest("hex").slice(0, 16);
23
- }
24
-
25
- function nodeId(type, file, line) {
26
- return `${type}_${sha256(file + ":" + line)}`;
27
- }
28
-
29
- function parseFile(code) {
30
- return parser.parse(code, {
31
- sourceType: "unambiguous",
32
- plugins: ["typescript", "jsx", "decorators-legacy"]
33
- });
34
- }
35
-
36
- function safeRead(fileAbs) {
37
- return fs.readFileSync(fileAbs, "utf8");
38
- }
39
-
40
- function getSnippet(code, loc) {
41
- if (!loc) return "";
42
- const lines = code.split(/\r?\n/);
43
- const start = Math.max(0, (loc.start?.line || 1) - 1);
44
- const end = Math.min(lines.length, (loc.end?.line || start + 1));
45
- return lines.slice(start, end).join("\n").slice(0, 200);
46
- }
47
-
48
- /**
49
- * Extract UI action nodes (onClick, onSubmit, etc.)
50
- */
51
- async function extractUIActions(repoRoot) {
52
- const files = await fg(["**/*.{tsx,jsx}"], {
53
- cwd: repoRoot,
54
- absolute: true,
55
- ignore: ["**/node_modules/**", "**/.next/**", "**/dist/**", "**/build/**"]
56
- });
57
-
58
- const nodes = [];
59
- const edges = [];
60
-
61
- for (const fileAbs of files) {
62
- const fileRel = path.relative(repoRoot, fileAbs).replace(/\\/g, "/");
63
- const code = safeRead(fileAbs);
64
-
65
- let ast;
66
- try { ast = parseFile(code); } catch { continue; }
67
-
68
- traverse(ast, {
69
- JSXAttribute(p) {
70
- const name = p.node.name?.name;
71
- if (!name || !["onClick", "onSubmit", "onPress", "onChange"].includes(name)) return;
72
-
73
- const line = p.node.loc?.start?.line || 0;
74
- const snippet = getSnippet(code, p.node.loc);
75
- const id = nodeId("ui_action", fileRel, line);
76
-
77
- nodes.push({
78
- id,
79
- type: "ui_action",
80
- file: fileRel,
81
- line,
82
- snippet,
83
- snippetHash: sha256(snippet),
84
- actionType: name
85
- });
86
-
87
- // Try to find what function this calls
88
- const value = p.node.value;
89
- if (t.isJSXExpressionContainer(value)) {
90
- const expr = value.expression;
91
-
92
- // Direct function reference: onClick={handleClick}
93
- if (t.isIdentifier(expr)) {
94
- edges.push({
95
- id: `edge_${sha256(id + "_" + expr.name)}`,
96
- from: id,
97
- toRef: expr.name,
98
- type: "calls",
99
- confidence: "high",
100
- file: fileRel
101
- });
102
- }
103
-
104
- // Arrow function with call: onClick={() => handleClick()}
105
- if (t.isArrowFunctionExpression(expr) || t.isFunctionExpression(expr)) {
106
- traverse(expr.body, {
107
- CallExpression(cp) {
108
- if (t.isIdentifier(cp.node.callee)) {
109
- edges.push({
110
- id: `edge_${sha256(id + "_" + cp.node.callee.name)}`,
111
- from: id,
112
- toRef: cp.node.callee.name,
113
- type: "calls",
114
- confidence: "med",
115
- file: fileRel
116
- });
117
- }
118
- }
119
- }, p.scope, p);
120
- }
121
- }
122
- }
123
- });
124
- }
125
-
126
- return { nodes, edges };
127
- }
128
-
129
- /**
130
- * Extract client function nodes that make network calls
131
- */
132
- async function extractClientFunctions(repoRoot) {
133
- const files = await fg(["**/*.{ts,tsx,js,jsx}"], {
134
- cwd: repoRoot,
135
- absolute: true,
136
- ignore: ["**/node_modules/**", "**/.next/**", "**/dist/**", "**/build/**", "**/server/**", "**/api/**"]
137
- });
138
-
139
- const nodes = [];
140
- const edges = [];
141
-
142
- for (const fileAbs of files) {
143
- const fileRel = path.relative(repoRoot, fileAbs).replace(/\\/g, "/");
144
- const code = safeRead(fileAbs);
145
-
146
- let ast;
147
- try { ast = parseFile(code); } catch { continue; }
148
-
149
- // Track function declarations that contain fetch/axios
150
- const funcMap = new Map();
151
-
152
- traverse(ast, {
153
- "FunctionDeclaration|ArrowFunctionExpression|FunctionExpression"(p) {
154
- const funcName = p.node.id?.name ||
155
- (t.isVariableDeclarator(p.parent) && t.isIdentifier(p.parent.id) ? p.parent.id.name : null);
156
-
157
- if (!funcName) return;
158
-
159
- const line = p.node.loc?.start?.line || 0;
160
- const snippet = getSnippet(code, p.node.loc);
161
- const id = nodeId("client_function", fileRel, line);
162
-
163
- funcMap.set(funcName, { id, line, snippet });
164
- }
165
- });
166
-
167
- traverse(ast, {
168
- CallExpression(p) {
169
- const callee = p.node.callee;
170
- let fetchUrl = null;
171
- let method = "*";
172
- let callType = null;
173
-
174
- // fetch("/api/x")
175
- if (t.isIdentifier(callee) && callee.name === "fetch") {
176
- const arg0 = p.node.arguments[0];
177
- if (t.isStringLiteral(arg0)) {
178
- fetchUrl = arg0.value;
179
- callType = "fetch";
180
- }
181
- // Check method in options
182
- const arg1 = p.node.arguments[1];
183
- if (t.isObjectExpression(arg1)) {
184
- for (const prop of arg1.properties) {
185
- if (t.isObjectProperty(prop) &&
186
- ((t.isIdentifier(prop.key) && prop.key.name === "method") ||
187
- (t.isStringLiteral(prop.key) && prop.key.value === "method")) &&
188
- t.isStringLiteral(prop.value)) {
189
- method = prop.value.value.toUpperCase();
190
- }
191
- }
192
- }
193
- }
194
-
195
- // axios.get("/api/x")
196
- if (t.isMemberExpression(callee) &&
197
- t.isIdentifier(callee.object) && callee.object.name === "axios" &&
198
- t.isIdentifier(callee.property)) {
199
- const verb = callee.property.name;
200
- if (["get", "post", "put", "patch", "delete"].includes(verb)) {
201
- const arg0 = p.node.arguments[0];
202
- if (t.isStringLiteral(arg0)) {
203
- fetchUrl = arg0.value;
204
- method = verb.toUpperCase();
205
- callType = "axios";
206
- }
207
- }
208
- }
209
-
210
- if (fetchUrl && fetchUrl.startsWith("/")) {
211
- const line = p.node.loc?.start?.line || 0;
212
- const snippet = getSnippet(code, p.node.loc);
213
- const networkId = nodeId("network_call", fileRel, line);
214
-
215
- nodes.push({
216
- id: networkId,
217
- type: "network_call",
218
- file: fileRel,
219
- line,
220
- snippet,
221
- snippetHash: sha256(snippet),
222
- url: fetchUrl,
223
- method,
224
- callType
225
- });
226
-
227
- // Find enclosing function
228
- let funcScope = p.scope;
229
- while (funcScope) {
230
- const funcNode = funcScope.block;
231
- if (t.isFunction(funcNode)) {
232
- const funcName = funcNode.id?.name ||
233
- (t.isVariableDeclarator(funcScope.parentBlock) ? funcScope.parentBlock.id?.name : null);
234
-
235
- if (funcName && funcMap.has(funcName)) {
236
- const funcData = funcMap.get(funcName);
237
-
238
- // Add function node if not already added
239
- if (!nodes.find(n => n.id === funcData.id)) {
240
- nodes.push({
241
- id: funcData.id,
242
- type: "client_function",
243
- file: fileRel,
244
- line: funcData.line,
245
- snippet: funcData.snippet,
246
- snippetHash: sha256(funcData.snippet),
247
- name: funcName
248
- });
249
- }
250
-
251
- edges.push({
252
- id: `edge_${sha256(funcData.id + "_" + networkId)}`,
253
- from: funcData.id,
254
- to: networkId,
255
- type: "fetches",
256
- confidence: "high"
257
- });
258
- }
259
- break;
260
- }
261
- funcScope = funcScope.parent;
262
- }
263
-
264
- // Create edge to server route (to be resolved later)
265
- edges.push({
266
- id: `edge_${sha256(networkId + "_route_" + fetchUrl)}`,
267
- from: networkId,
268
- toRoute: { method, path: fetchUrl },
269
- type: "calls_route",
270
- confidence: "high"
271
- });
272
- }
273
- }
274
- });
275
- }
276
-
277
- return { nodes, edges };
278
- }
279
-
280
- /**
281
- * Extract server route nodes from truthpack
282
- */
283
- function extractServerRoutes(truthpack) {
284
- const nodes = [];
285
- const serverRoutes = truthpack?.routes?.server || [];
286
-
287
- for (const route of serverRoutes) {
288
- const id = nodeId("server_route", route.handler || "unknown", route.path.length);
289
-
290
- nodes.push({
291
- id,
292
- type: "server_route",
293
- file: route.handler || "unknown",
294
- line: 0,
295
- snippet: `${route.method} ${route.path}`,
296
- snippetHash: sha256(`${route.method} ${route.path}`),
297
- method: route.method,
298
- path: route.path,
299
- confidence: route.confidence
300
- });
301
- }
302
-
303
- return { nodes, edges: [] };
304
- }
305
-
306
- /**
307
- * Extract handler → DB/external call edges
308
- */
309
- async function extractHandlerCalls(repoRoot) {
310
- const files = await fg(["**/api/**/*.{ts,js}", "**/routes/**/*.{ts,js}", "**/server/**/*.{ts,js}"], {
311
- cwd: repoRoot,
312
- absolute: true,
313
- ignore: ["**/node_modules/**", "**/.next/**", "**/dist/**", "**/build/**"]
314
- });
315
-
316
- const nodes = [];
317
- const edges = [];
318
-
319
- for (const fileAbs of files) {
320
- const fileRel = path.relative(repoRoot, fileAbs).replace(/\\/g, "/");
321
- const code = safeRead(fileAbs);
322
-
323
- let ast;
324
- try { ast = parseFile(code); } catch { continue; }
325
-
326
- traverse(ast, {
327
- CallExpression(p) {
328
- const callee = p.node.callee;
329
- let callType = null;
330
- let target = null;
331
-
332
- // Prisma: prisma.user.findMany(), prisma.$transaction()
333
- if (t.isMemberExpression(callee)) {
334
- const obj = callee.object;
335
- if (t.isIdentifier(obj) && obj.name === "prisma") {
336
- callType = "db_call";
337
- target = "prisma";
338
- } else if (t.isMemberExpression(obj) &&
339
- t.isIdentifier(obj.object) && obj.object.name === "prisma") {
340
- callType = "db_call";
341
- target = `prisma.${obj.property?.name || "unknown"}`;
342
- }
343
- }
344
-
345
- // External: stripe.customers.create()
346
- if (t.isMemberExpression(callee)) {
347
- const rootObj = getRootObject(callee);
348
- if (rootObj && ["stripe", "github", "sendgrid", "twilio", "aws"].includes(rootObj.toLowerCase())) {
349
- callType = "external_call";
350
- target = rootObj;
351
- }
352
- }
353
-
354
- if (callType) {
355
- const line = p.node.loc?.start?.line || 0;
356
- const snippet = getSnippet(code, p.node.loc);
357
- const id = nodeId(callType, fileRel, line);
358
-
359
- nodes.push({
360
- id,
361
- type: callType,
362
- file: fileRel,
363
- line,
364
- snippet,
365
- snippetHash: sha256(snippet),
366
- target
367
- });
368
- }
369
- }
370
- });
371
- }
372
-
373
- return { nodes, edges };
374
- }
375
-
376
- function getRootObject(node) {
377
- if (t.isIdentifier(node)) return node.name;
378
- if (t.isMemberExpression(node)) return getRootObject(node.object);
379
- return null;
380
- }
381
-
382
- /**
383
- * Build complete static graph
384
- */
385
- async function extractStaticGraph(repoRoot, truthpack) {
386
- const uiActions = await extractUIActions(repoRoot);
387
- const clientFuncs = await extractClientFunctions(repoRoot);
388
- const serverRoutes = extractServerRoutes(truthpack);
389
- const handlerCalls = await extractHandlerCalls(repoRoot);
390
-
391
- const allNodes = [
392
- ...uiActions.nodes,
393
- ...clientFuncs.nodes,
394
- ...serverRoutes.nodes,
395
- ...handlerCalls.nodes
396
- ];
397
-
398
- const allEdges = [
399
- ...uiActions.edges,
400
- ...clientFuncs.edges,
401
- ...serverRoutes.edges,
402
- ...handlerCalls.edges
403
- ];
404
-
405
- // Resolve function reference edges
406
- const resolvedEdges = resolveEdges(allNodes, allEdges, truthpack);
407
-
408
- return {
409
- nodes: dedupeNodes(allNodes),
410
- edges: resolvedEdges
411
- };
412
- }
413
-
414
- function dedupeNodes(nodes) {
415
- const seen = new Map();
416
- for (const n of nodes) {
417
- if (!seen.has(n.id)) seen.set(n.id, n);
418
- }
419
- return Array.from(seen.values());
420
- }
421
-
422
- function resolveEdges(nodes, edges, truthpack) {
423
- const resolved = [];
424
- const nodeById = new Map(nodes.map(n => [n.id, n]));
425
- const funcByName = new Map();
426
- const routeNodes = nodes.filter(n => n.type === "server_route");
427
-
428
- // Build function name → node map
429
- for (const n of nodes) {
430
- if (n.type === "client_function" && n.name) {
431
- funcByName.set(n.name, n);
432
- }
433
- }
434
-
435
- for (const edge of edges) {
436
- // Resolve toRef (function name) → actual node
437
- if (edge.toRef) {
438
- const target = funcByName.get(edge.toRef);
439
- if (target) {
440
- resolved.push({
441
- ...edge,
442
- to: target.id,
443
- toRef: undefined
444
- });
445
- } else {
446
- // Unresolved function call - might be external
447
- resolved.push({
448
- ...edge,
449
- to: `unresolved_${edge.toRef}`,
450
- confidence: "low"
451
- });
452
- }
453
- continue;
454
- }
455
-
456
- // Resolve toRoute → server route node
457
- if (edge.toRoute) {
458
- const route = findMatchingRoute(routeNodes, edge.toRoute.method, edge.toRoute.path);
459
- if (route) {
460
- resolved.push({
461
- ...edge,
462
- to: route.id,
463
- toRoute: undefined
464
- });
465
- } else {
466
- // No matching route - this is a broken edge
467
- resolved.push({
468
- ...edge,
469
- to: `missing_route_${edge.toRoute.method}_${edge.toRoute.path}`,
470
- toRoute: edge.toRoute,
471
- broken: true,
472
- brokenReason: `Route ${edge.toRoute.method} ${edge.toRoute.path} not found on server`
473
- });
474
- }
475
- continue;
476
- }
477
-
478
- resolved.push(edge);
479
- }
480
-
481
- return resolved;
482
- }
483
-
484
- function findMatchingRoute(routeNodes, method, path) {
485
- // Normalize path
486
- const normPath = path.replace(/\/$/, "") || "/";
487
-
488
- for (const r of routeNodes) {
489
- if (r.method === "*" || r.method === method || method === "*") {
490
- if (r.path === normPath) return r;
491
- // Check parameterized match
492
- if (matchesParameterized(r.path, normPath)) return r;
493
- }
494
- }
495
- return null;
496
- }
497
-
498
- function matchesParameterized(pattern, actual) {
499
- const patternParts = pattern.split("/").filter(Boolean);
500
- const actualParts = actual.split("/").filter(Boolean);
501
-
502
- if (patternParts.length !== actualParts.length) return false;
503
-
504
- for (let i = 0; i < patternParts.length; i++) {
505
- const p = patternParts[i];
506
- if (p.startsWith(":") || p.startsWith("*")) continue;
507
- if (p !== actualParts[i]) return false;
508
- }
509
- return true;
510
- }
511
-
512
- module.exports = {
513
- extractStaticGraph,
514
- extractUIActions,
515
- extractClientFunctions,
516
- extractServerRoutes,
517
- extractHandlerCalls
518
- };
1
+ /**
2
+ * Static Edge Extractor
3
+ * Extracts causal edges from AST analysis:
4
+ * - UI actions → client functions
5
+ * - Client functions → network calls
6
+ * - Network calls → server routes
7
+ * - Server routes → handlers
8
+ * - Handlers → DB/external calls
9
+ */
10
+
11
+ "use strict";
12
+
13
+ const fg = require("fast-glob");
14
+ const fs = require("fs");
15
+ const path = require("path");
16
+ const crypto = require("crypto");
17
+ const parser = require("@babel/parser");
18
+ const traverse = require("@babel/traverse").default;
19
+ const t = require("@babel/types");
20
+
21
+ function sha256(text) {
22
+ return crypto.createHash("sha256").update(text).digest("hex").slice(0, 16);
23
+ }
24
+
25
+ function nodeId(type, file, line) {
26
+ return `${type}_${sha256(file + ":" + line)}`;
27
+ }
28
+
29
+ function parseFile(code) {
30
+ return parser.parse(code, {
31
+ sourceType: "unambiguous",
32
+ plugins: ["typescript", "jsx", "decorators-legacy"]
33
+ });
34
+ }
35
+
36
+ function safeRead(fileAbs) {
37
+ return fs.readFileSync(fileAbs, "utf8");
38
+ }
39
+
40
+ function getSnippet(code, loc) {
41
+ if (!loc) return "";
42
+ const lines = code.split(/\r?\n/);
43
+ const start = Math.max(0, (loc.start?.line || 1) - 1);
44
+ const end = Math.min(lines.length, (loc.end?.line || start + 1));
45
+ return lines.slice(start, end).join("\n").slice(0, 200);
46
+ }
47
+
48
+ /**
49
+ * Extract UI action nodes (onClick, onSubmit, etc.)
50
+ */
51
+ async function extractUIActions(repoRoot) {
52
+ const files = await fg(["**/*.{tsx,jsx}"], {
53
+ cwd: repoRoot,
54
+ absolute: true,
55
+ ignore: ["**/node_modules/**", "**/.next/**", "**/dist/**", "**/build/**"]
56
+ });
57
+
58
+ const nodes = [];
59
+ const edges = [];
60
+
61
+ for (const fileAbs of files) {
62
+ const fileRel = path.relative(repoRoot, fileAbs).replace(/\\/g, "/");
63
+ const code = safeRead(fileAbs);
64
+
65
+ let ast;
66
+ try { ast = parseFile(code); } catch { continue; }
67
+
68
+ traverse(ast, {
69
+ JSXAttribute(p) {
70
+ const name = p.node.name?.name;
71
+ if (!name || !["onClick", "onSubmit", "onPress", "onChange"].includes(name)) return;
72
+
73
+ const line = p.node.loc?.start?.line || 0;
74
+ const snippet = getSnippet(code, p.node.loc);
75
+ const id = nodeId("ui_action", fileRel, line);
76
+
77
+ nodes.push({
78
+ id,
79
+ type: "ui_action",
80
+ file: fileRel,
81
+ line,
82
+ snippet,
83
+ snippetHash: sha256(snippet),
84
+ actionType: name
85
+ });
86
+
87
+ // Try to find what function this calls
88
+ const value = p.node.value;
89
+ if (t.isJSXExpressionContainer(value)) {
90
+ const expr = value.expression;
91
+
92
+ // Direct function reference: onClick={handleClick}
93
+ if (t.isIdentifier(expr)) {
94
+ edges.push({
95
+ id: `edge_${sha256(id + "_" + expr.name)}`,
96
+ from: id,
97
+ toRef: expr.name,
98
+ type: "calls",
99
+ confidence: "high",
100
+ file: fileRel
101
+ });
102
+ }
103
+
104
+ // Arrow function with call: onClick={() => handleClick()}
105
+ if (t.isArrowFunctionExpression(expr) || t.isFunctionExpression(expr)) {
106
+ traverse(expr.body, {
107
+ CallExpression(cp) {
108
+ if (t.isIdentifier(cp.node.callee)) {
109
+ edges.push({
110
+ id: `edge_${sha256(id + "_" + cp.node.callee.name)}`,
111
+ from: id,
112
+ toRef: cp.node.callee.name,
113
+ type: "calls",
114
+ confidence: "med",
115
+ file: fileRel
116
+ });
117
+ }
118
+ }
119
+ }, p.scope, p);
120
+ }
121
+ }
122
+ }
123
+ });
124
+ }
125
+
126
+ return { nodes, edges };
127
+ }
128
+
129
+ /**
130
+ * Extract client function nodes that make network calls
131
+ */
132
+ async function extractClientFunctions(repoRoot) {
133
+ const files = await fg(["**/*.{ts,tsx,js,jsx}"], {
134
+ cwd: repoRoot,
135
+ absolute: true,
136
+ ignore: ["**/node_modules/**", "**/.next/**", "**/dist/**", "**/build/**", "**/server/**", "**/api/**"]
137
+ });
138
+
139
+ const nodes = [];
140
+ const edges = [];
141
+
142
+ for (const fileAbs of files) {
143
+ const fileRel = path.relative(repoRoot, fileAbs).replace(/\\/g, "/");
144
+ const code = safeRead(fileAbs);
145
+
146
+ let ast;
147
+ try { ast = parseFile(code); } catch { continue; }
148
+
149
+ // Track function declarations that contain fetch/axios
150
+ const funcMap = new Map();
151
+
152
+ traverse(ast, {
153
+ "FunctionDeclaration|ArrowFunctionExpression|FunctionExpression"(p) {
154
+ const funcName = p.node.id?.name ||
155
+ (t.isVariableDeclarator(p.parent) && t.isIdentifier(p.parent.id) ? p.parent.id.name : null);
156
+
157
+ if (!funcName) return;
158
+
159
+ const line = p.node.loc?.start?.line || 0;
160
+ const snippet = getSnippet(code, p.node.loc);
161
+ const id = nodeId("client_function", fileRel, line);
162
+
163
+ funcMap.set(funcName, { id, line, snippet });
164
+ }
165
+ });
166
+
167
+ traverse(ast, {
168
+ CallExpression(p) {
169
+ const callee = p.node.callee;
170
+ let fetchUrl = null;
171
+ let method = "*";
172
+ let callType = null;
173
+
174
+ // fetch("/api/x")
175
+ if (t.isIdentifier(callee) && callee.name === "fetch") {
176
+ const arg0 = p.node.arguments[0];
177
+ if (t.isStringLiteral(arg0)) {
178
+ fetchUrl = arg0.value;
179
+ callType = "fetch";
180
+ }
181
+ // Check method in options
182
+ const arg1 = p.node.arguments[1];
183
+ if (t.isObjectExpression(arg1)) {
184
+ for (const prop of arg1.properties) {
185
+ if (t.isObjectProperty(prop) &&
186
+ ((t.isIdentifier(prop.key) && prop.key.name === "method") ||
187
+ (t.isStringLiteral(prop.key) && prop.key.value === "method")) &&
188
+ t.isStringLiteral(prop.value)) {
189
+ method = prop.value.value.toUpperCase();
190
+ }
191
+ }
192
+ }
193
+ }
194
+
195
+ // axios.get("/api/x")
196
+ if (t.isMemberExpression(callee) &&
197
+ t.isIdentifier(callee.object) && callee.object.name === "axios" &&
198
+ t.isIdentifier(callee.property)) {
199
+ const verb = callee.property.name;
200
+ if (["get", "post", "put", "patch", "delete"].includes(verb)) {
201
+ const arg0 = p.node.arguments[0];
202
+ if (t.isStringLiteral(arg0)) {
203
+ fetchUrl = arg0.value;
204
+ method = verb.toUpperCase();
205
+ callType = "axios";
206
+ }
207
+ }
208
+ }
209
+
210
+ if (fetchUrl && fetchUrl.startsWith("/")) {
211
+ const line = p.node.loc?.start?.line || 0;
212
+ const snippet = getSnippet(code, p.node.loc);
213
+ const networkId = nodeId("network_call", fileRel, line);
214
+
215
+ nodes.push({
216
+ id: networkId,
217
+ type: "network_call",
218
+ file: fileRel,
219
+ line,
220
+ snippet,
221
+ snippetHash: sha256(snippet),
222
+ url: fetchUrl,
223
+ method,
224
+ callType
225
+ });
226
+
227
+ // Find enclosing function
228
+ let funcScope = p.scope;
229
+ while (funcScope) {
230
+ const funcNode = funcScope.block;
231
+ if (t.isFunction(funcNode)) {
232
+ const funcName = funcNode.id?.name ||
233
+ (t.isVariableDeclarator(funcScope.parentBlock) ? funcScope.parentBlock.id?.name : null);
234
+
235
+ if (funcName && funcMap.has(funcName)) {
236
+ const funcData = funcMap.get(funcName);
237
+
238
+ // Add function node if not already added
239
+ if (!nodes.find(n => n.id === funcData.id)) {
240
+ nodes.push({
241
+ id: funcData.id,
242
+ type: "client_function",
243
+ file: fileRel,
244
+ line: funcData.line,
245
+ snippet: funcData.snippet,
246
+ snippetHash: sha256(funcData.snippet),
247
+ name: funcName
248
+ });
249
+ }
250
+
251
+ edges.push({
252
+ id: `edge_${sha256(funcData.id + "_" + networkId)}`,
253
+ from: funcData.id,
254
+ to: networkId,
255
+ type: "fetches",
256
+ confidence: "high"
257
+ });
258
+ }
259
+ break;
260
+ }
261
+ funcScope = funcScope.parent;
262
+ }
263
+
264
+ // Create edge to server route (to be resolved later)
265
+ edges.push({
266
+ id: `edge_${sha256(networkId + "_route_" + fetchUrl)}`,
267
+ from: networkId,
268
+ toRoute: { method, path: fetchUrl },
269
+ type: "calls_route",
270
+ confidence: "high"
271
+ });
272
+ }
273
+ }
274
+ });
275
+ }
276
+
277
+ return { nodes, edges };
278
+ }
279
+
280
+ /**
281
+ * Extract server route nodes from truthpack
282
+ */
283
+ function extractServerRoutes(truthpack) {
284
+ const nodes = [];
285
+ const serverRoutes = truthpack?.routes?.server || [];
286
+
287
+ for (const route of serverRoutes) {
288
+ const id = nodeId("server_route", route.handler || "unknown", route.path.length);
289
+
290
+ nodes.push({
291
+ id,
292
+ type: "server_route",
293
+ file: route.handler || "unknown",
294
+ line: 0,
295
+ snippet: `${route.method} ${route.path}`,
296
+ snippetHash: sha256(`${route.method} ${route.path}`),
297
+ method: route.method,
298
+ path: route.path,
299
+ confidence: route.confidence
300
+ });
301
+ }
302
+
303
+ return { nodes, edges: [] };
304
+ }
305
+
306
+ /**
307
+ * Extract handler → DB/external call edges
308
+ */
309
+ async function extractHandlerCalls(repoRoot) {
310
+ const files = await fg(["**/api/**/*.{ts,js}", "**/routes/**/*.{ts,js}", "**/server/**/*.{ts,js}"], {
311
+ cwd: repoRoot,
312
+ absolute: true,
313
+ ignore: ["**/node_modules/**", "**/.next/**", "**/dist/**", "**/build/**"]
314
+ });
315
+
316
+ const nodes = [];
317
+ const edges = [];
318
+
319
+ for (const fileAbs of files) {
320
+ const fileRel = path.relative(repoRoot, fileAbs).replace(/\\/g, "/");
321
+ const code = safeRead(fileAbs);
322
+
323
+ let ast;
324
+ try { ast = parseFile(code); } catch { continue; }
325
+
326
+ traverse(ast, {
327
+ CallExpression(p) {
328
+ const callee = p.node.callee;
329
+ let callType = null;
330
+ let target = null;
331
+
332
+ // Prisma: prisma.user.findMany(), prisma.$transaction()
333
+ if (t.isMemberExpression(callee)) {
334
+ const obj = callee.object;
335
+ if (t.isIdentifier(obj) && obj.name === "prisma") {
336
+ callType = "db_call";
337
+ target = "prisma";
338
+ } else if (t.isMemberExpression(obj) &&
339
+ t.isIdentifier(obj.object) && obj.object.name === "prisma") {
340
+ callType = "db_call";
341
+ target = `prisma.${obj.property?.name || "unknown"}`;
342
+ }
343
+ }
344
+
345
+ // External: stripe.customers.create()
346
+ if (t.isMemberExpression(callee)) {
347
+ const rootObj = getRootObject(callee);
348
+ if (rootObj && ["stripe", "github", "sendgrid", "twilio", "aws"].includes(rootObj.toLowerCase())) {
349
+ callType = "external_call";
350
+ target = rootObj;
351
+ }
352
+ }
353
+
354
+ if (callType) {
355
+ const line = p.node.loc?.start?.line || 0;
356
+ const snippet = getSnippet(code, p.node.loc);
357
+ const id = nodeId(callType, fileRel, line);
358
+
359
+ nodes.push({
360
+ id,
361
+ type: callType,
362
+ file: fileRel,
363
+ line,
364
+ snippet,
365
+ snippetHash: sha256(snippet),
366
+ target
367
+ });
368
+ }
369
+ }
370
+ });
371
+ }
372
+
373
+ return { nodes, edges };
374
+ }
375
+
376
+ function getRootObject(node) {
377
+ if (t.isIdentifier(node)) return node.name;
378
+ if (t.isMemberExpression(node)) return getRootObject(node.object);
379
+ return null;
380
+ }
381
+
382
+ /**
383
+ * Build complete static graph
384
+ */
385
+ async function extractStaticGraph(repoRoot, truthpack) {
386
+ const uiActions = await extractUIActions(repoRoot);
387
+ const clientFuncs = await extractClientFunctions(repoRoot);
388
+ const serverRoutes = extractServerRoutes(truthpack);
389
+ const handlerCalls = await extractHandlerCalls(repoRoot);
390
+
391
+ const allNodes = [
392
+ ...uiActions.nodes,
393
+ ...clientFuncs.nodes,
394
+ ...serverRoutes.nodes,
395
+ ...handlerCalls.nodes
396
+ ];
397
+
398
+ const allEdges = [
399
+ ...uiActions.edges,
400
+ ...clientFuncs.edges,
401
+ ...serverRoutes.edges,
402
+ ...handlerCalls.edges
403
+ ];
404
+
405
+ // Resolve function reference edges
406
+ const resolvedEdges = resolveEdges(allNodes, allEdges, truthpack);
407
+
408
+ return {
409
+ nodes: dedupeNodes(allNodes),
410
+ edges: resolvedEdges
411
+ };
412
+ }
413
+
414
+ function dedupeNodes(nodes) {
415
+ const seen = new Map();
416
+ for (const n of nodes) {
417
+ if (!seen.has(n.id)) seen.set(n.id, n);
418
+ }
419
+ return Array.from(seen.values());
420
+ }
421
+
422
+ function resolveEdges(nodes, edges, truthpack) {
423
+ const resolved = [];
424
+ const nodeById = new Map(nodes.map(n => [n.id, n]));
425
+ const funcByName = new Map();
426
+ const routeNodes = nodes.filter(n => n.type === "server_route");
427
+
428
+ // Build function name → node map
429
+ for (const n of nodes) {
430
+ if (n.type === "client_function" && n.name) {
431
+ funcByName.set(n.name, n);
432
+ }
433
+ }
434
+
435
+ for (const edge of edges) {
436
+ // Resolve toRef (function name) → actual node
437
+ if (edge.toRef) {
438
+ const target = funcByName.get(edge.toRef);
439
+ if (target) {
440
+ resolved.push({
441
+ ...edge,
442
+ to: target.id,
443
+ toRef: undefined
444
+ });
445
+ } else {
446
+ // Unresolved function call - might be external
447
+ resolved.push({
448
+ ...edge,
449
+ to: `unresolved_${edge.toRef}`,
450
+ confidence: "low"
451
+ });
452
+ }
453
+ continue;
454
+ }
455
+
456
+ // Resolve toRoute → server route node
457
+ if (edge.toRoute) {
458
+ const route = findMatchingRoute(routeNodes, edge.toRoute.method, edge.toRoute.path);
459
+ if (route) {
460
+ resolved.push({
461
+ ...edge,
462
+ to: route.id,
463
+ toRoute: undefined
464
+ });
465
+ } else {
466
+ // No matching route - this is a broken edge
467
+ resolved.push({
468
+ ...edge,
469
+ to: `missing_route_${edge.toRoute.method}_${edge.toRoute.path}`,
470
+ toRoute: edge.toRoute,
471
+ broken: true,
472
+ brokenReason: `Route ${edge.toRoute.method} ${edge.toRoute.path} not found on server`
473
+ });
474
+ }
475
+ continue;
476
+ }
477
+
478
+ resolved.push(edge);
479
+ }
480
+
481
+ return resolved;
482
+ }
483
+
484
+ function findMatchingRoute(routeNodes, method, path) {
485
+ // Normalize path
486
+ const normPath = path.replace(/\/$/, "") || "/";
487
+
488
+ for (const r of routeNodes) {
489
+ if (r.method === "*" || r.method === method || method === "*") {
490
+ if (r.path === normPath) return r;
491
+ // Check parameterized match
492
+ if (matchesParameterized(r.path, normPath)) return r;
493
+ }
494
+ }
495
+ return null;
496
+ }
497
+
498
+ function matchesParameterized(pattern, actual) {
499
+ const patternParts = pattern.split("/").filter(Boolean);
500
+ const actualParts = actual.split("/").filter(Boolean);
501
+
502
+ if (patternParts.length !== actualParts.length) return false;
503
+
504
+ for (let i = 0; i < patternParts.length; i++) {
505
+ const p = patternParts[i];
506
+ if (p.startsWith(":") || p.startsWith("*")) continue;
507
+ if (p !== actualParts[i]) return false;
508
+ }
509
+ return true;
510
+ }
511
+
512
+ module.exports = {
513
+ extractStaticGraph,
514
+ extractUIActions,
515
+ extractClientFunctions,
516
+ extractServerRoutes,
517
+ extractHandlerCalls
518
+ };