@vibecheckai/cli 3.2.2 → 3.2.4
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.
- package/bin/.generated +25 -25
- package/bin/dev/run-v2-torture.js +30 -30
- package/bin/runners/ENHANCEMENT_GUIDE.md +121 -121
- package/bin/runners/lib/__tests__/entitlements-v2.test.js +295 -295
- package/bin/runners/lib/agent-firewall/ai/false-positive-analyzer.js +474 -0
- package/bin/runners/lib/agent-firewall/claims/extractor.js +117 -28
- package/bin/runners/lib/agent-firewall/evidence/env-evidence.js +23 -14
- package/bin/runners/lib/agent-firewall/evidence/route-evidence.js +72 -1
- package/bin/runners/lib/agent-firewall/interceptor/base.js +2 -2
- package/bin/runners/lib/agent-firewall/policy/default-policy.json +6 -0
- package/bin/runners/lib/agent-firewall/policy/engine.js +34 -3
- package/bin/runners/lib/agent-firewall/policy/rules/fake-success.js +29 -4
- package/bin/runners/lib/agent-firewall/policy/rules/ghost-route.js +12 -0
- package/bin/runners/lib/agent-firewall/truthpack/loader.js +21 -0
- package/bin/runners/lib/agent-firewall/utils/ignore-checker.js +118 -0
- package/bin/runners/lib/analyzers.js +606 -325
- package/bin/runners/lib/auth-truth.js +193 -193
- package/bin/runners/lib/backup.js +62 -62
- package/bin/runners/lib/billing.js +107 -107
- package/bin/runners/lib/claims.js +118 -118
- package/bin/runners/lib/cli-ui.js +540 -540
- package/bin/runners/lib/contracts/auth-contract.js +202 -202
- package/bin/runners/lib/contracts/env-contract.js +181 -181
- package/bin/runners/lib/contracts/external-contract.js +206 -206
- package/bin/runners/lib/contracts/guard.js +168 -168
- package/bin/runners/lib/contracts/index.js +89 -89
- package/bin/runners/lib/contracts/plan-validator.js +311 -311
- package/bin/runners/lib/contracts/route-contract.js +199 -199
- package/bin/runners/lib/contracts.js +804 -804
- package/bin/runners/lib/detect.js +89 -89
- package/bin/runners/lib/doctor/autofix.js +254 -254
- package/bin/runners/lib/doctor/index.js +37 -37
- package/bin/runners/lib/doctor/modules/dependencies.js +325 -325
- package/bin/runners/lib/doctor/modules/index.js +46 -46
- package/bin/runners/lib/doctor/modules/network.js +250 -250
- package/bin/runners/lib/doctor/modules/project.js +312 -312
- package/bin/runners/lib/doctor/modules/runtime.js +224 -224
- package/bin/runners/lib/doctor/modules/security.js +348 -348
- package/bin/runners/lib/doctor/modules/system.js +213 -213
- package/bin/runners/lib/doctor/modules/vibecheck.js +394 -394
- package/bin/runners/lib/doctor/reporter.js +262 -262
- package/bin/runners/lib/doctor/service.js +262 -262
- package/bin/runners/lib/doctor/types.js +113 -113
- package/bin/runners/lib/doctor/ui.js +263 -263
- package/bin/runners/lib/doctor-v2.js +608 -608
- package/bin/runners/lib/drift.js +425 -425
- package/bin/runners/lib/enforcement.js +72 -72
- package/bin/runners/lib/engines/accessibility-engine.js +190 -0
- package/bin/runners/lib/engines/api-consistency-engine.js +162 -0
- package/bin/runners/lib/engines/ast-cache.js +99 -0
- package/bin/runners/lib/engines/code-quality-engine.js +255 -0
- package/bin/runners/lib/engines/console-logs-engine.js +115 -0
- package/bin/runners/lib/engines/cross-file-analysis-engine.js +268 -0
- package/bin/runners/lib/engines/dead-code-engine.js +198 -0
- package/bin/runners/lib/engines/deprecated-api-engine.js +226 -0
- package/bin/runners/lib/engines/empty-catch-engine.js +150 -0
- package/bin/runners/lib/engines/file-filter.js +131 -0
- package/bin/runners/lib/engines/hardcoded-secrets-engine.js +251 -0
- package/bin/runners/lib/engines/mock-data-engine.js +272 -0
- package/bin/runners/lib/engines/parallel-processor.js +71 -0
- package/bin/runners/lib/engines/performance-issues-engine.js +265 -0
- package/bin/runners/lib/engines/security-vulnerabilities-engine.js +243 -0
- package/bin/runners/lib/engines/todo-fixme-engine.js +115 -0
- package/bin/runners/lib/engines/type-aware-engine.js +152 -0
- package/bin/runners/lib/engines/unsafe-regex-engine.js +225 -0
- package/bin/runners/lib/engines/vibecheck-engines/README.md +53 -0
- package/bin/runners/lib/engines/vibecheck-engines/index.js +15 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/ast-cache.js +164 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/code-quality-engine.js +291 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/console-logs-engine.js +83 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/dead-code-engine.js +198 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/deprecated-api-engine.js +275 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/empty-catch-engine.js +167 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/file-filter.js +217 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/hardcoded-secrets-engine.js +139 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/mock-data-engine.js +140 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/parallel-processor.js +164 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/performance-issues-engine.js +234 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/type-aware-engine.js +217 -0
- package/bin/runners/lib/engines/vibecheck-engines/lib/unsafe-regex-engine.js +78 -0
- package/bin/runners/lib/engines/vibecheck-engines/package.json +13 -0
- package/bin/runners/lib/enterprise-detect.js +603 -603
- package/bin/runners/lib/enterprise-init.js +942 -942
- package/bin/runners/lib/env-resolver.js +417 -417
- package/bin/runners/lib/env-template.js +66 -66
- package/bin/runners/lib/env.js +189 -189
- package/bin/runners/lib/extractors/client-calls.js +990 -990
- package/bin/runners/lib/extractors/fastify-route-dump.js +573 -573
- package/bin/runners/lib/extractors/fastify-routes.js +426 -426
- package/bin/runners/lib/extractors/index.js +363 -363
- package/bin/runners/lib/extractors/next-routes.js +524 -524
- package/bin/runners/lib/extractors/proof-graph.js +431 -431
- package/bin/runners/lib/extractors/route-matcher.js +451 -451
- package/bin/runners/lib/extractors/truthpack-v2.js +377 -377
- package/bin/runners/lib/extractors/ui-bindings.js +547 -547
- package/bin/runners/lib/findings-schema.js +281 -281
- package/bin/runners/lib/firewall-prompt.js +50 -50
- package/bin/runners/lib/global-flags.js +213 -213
- package/bin/runners/lib/graph/graph-builder.js +265 -265
- package/bin/runners/lib/graph/html-renderer.js +413 -413
- package/bin/runners/lib/graph/index.js +32 -32
- package/bin/runners/lib/graph/runtime-collector.js +215 -215
- package/bin/runners/lib/graph/static-extractor.js +518 -518
- package/bin/runners/lib/html-report.js +650 -650
- package/bin/runners/lib/interactive-menu.js +1496 -1496
- package/bin/runners/lib/llm.js +75 -75
- package/bin/runners/lib/meter.js +61 -61
- package/bin/runners/lib/missions/evidence.js +126 -126
- package/bin/runners/lib/patch.js +40 -40
- package/bin/runners/lib/permissions/auth-model.js +213 -213
- package/bin/runners/lib/permissions/idor-prover.js +205 -205
- package/bin/runners/lib/permissions/index.js +45 -45
- package/bin/runners/lib/permissions/matrix-builder.js +198 -198
- package/bin/runners/lib/pkgjson.js +28 -28
- package/bin/runners/lib/policy.js +295 -295
- package/bin/runners/lib/preflight.js +142 -142
- package/bin/runners/lib/reality/correlation-detectors.js +359 -359
- package/bin/runners/lib/reality/index.js +318 -318
- package/bin/runners/lib/reality/request-hashing.js +416 -416
- package/bin/runners/lib/reality/request-mapper.js +453 -453
- package/bin/runners/lib/reality/safety-rails.js +463 -463
- package/bin/runners/lib/reality/semantic-snapshot.js +408 -408
- package/bin/runners/lib/reality/toast-detector.js +393 -393
- package/bin/runners/lib/reality-findings.js +84 -84
- package/bin/runners/lib/receipts.js +179 -179
- package/bin/runners/lib/redact.js +29 -29
- package/bin/runners/lib/replay/capsule-manager.js +154 -154
- package/bin/runners/lib/replay/index.js +263 -263
- package/bin/runners/lib/replay/player.js +348 -348
- package/bin/runners/lib/replay/recorder.js +331 -331
- package/bin/runners/lib/report-output.js +187 -187
- package/bin/runners/lib/report.js +135 -135
- package/bin/runners/lib/route-detection.js +1140 -1140
- package/bin/runners/lib/sandbox/index.js +59 -59
- package/bin/runners/lib/sandbox/proof-chain.js +399 -399
- package/bin/runners/lib/sandbox/sandbox-runner.js +205 -205
- package/bin/runners/lib/sandbox/worktree.js +174 -174
- package/bin/runners/lib/scan-output.js +525 -190
- package/bin/runners/lib/schema-validator.js +350 -350
- package/bin/runners/lib/schemas/contracts.schema.json +160 -160
- package/bin/runners/lib/schemas/finding.schema.json +100 -100
- package/bin/runners/lib/schemas/mission-pack.schema.json +206 -206
- package/bin/runners/lib/schemas/proof-graph.schema.json +176 -176
- package/bin/runners/lib/schemas/reality-report.schema.json +162 -162
- package/bin/runners/lib/schemas/share-pack.schema.json +180 -180
- package/bin/runners/lib/schemas/ship-report.schema.json +117 -117
- package/bin/runners/lib/schemas/truthpack-v2.schema.json +303 -303
- package/bin/runners/lib/schemas/validator.js +438 -438
- package/bin/runners/lib/score-history.js +282 -282
- package/bin/runners/lib/share-pack.js +239 -239
- package/bin/runners/lib/snippets.js +67 -67
- package/bin/runners/lib/status-output.js +253 -253
- package/bin/runners/lib/terminal-ui.js +351 -271
- package/bin/runners/lib/upsell.js +510 -510
- package/bin/runners/lib/usage.js +153 -153
- package/bin/runners/lib/validate-patch.js +156 -156
- package/bin/runners/lib/verdict-engine.js +628 -628
- package/bin/runners/reality/engine.js +917 -917
- package/bin/runners/reality/flows.js +122 -122
- package/bin/runners/reality/report.js +378 -378
- package/bin/runners/reality/session.js +193 -193
- package/bin/runners/runGuard.js +168 -168
- package/bin/runners/runProof.zip +0 -0
- package/bin/runners/runProve.js +8 -0
- package/bin/runners/runReality.js +14 -0
- package/bin/runners/runScan.js +17 -1
- package/bin/runners/runTruth.js +15 -3
- package/mcp-server/tier-auth.js +4 -4
- package/mcp-server/tools/index.js +72 -72
- package/package.json +1 -1
|
@@ -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
|
+
};
|