codeprobe 0.1.0
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/LICENSE +21 -0
- package/README.md +406 -0
- package/dist/cli.d.ts +7 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +104 -0
- package/dist/cli.js.map +1 -0
- package/dist/commands/ab.d.ts +7 -0
- package/dist/commands/ab.d.ts.map +1 -0
- package/dist/commands/ab.js +230 -0
- package/dist/commands/ab.js.map +1 -0
- package/dist/commands/agents.d.ts +10 -0
- package/dist/commands/agents.d.ts.map +1 -0
- package/dist/commands/agents.js +326 -0
- package/dist/commands/agents.js.map +1 -0
- package/dist/commands/autotest.d.ts +10 -0
- package/dist/commands/autotest.d.ts.map +1 -0
- package/dist/commands/autotest.js +408 -0
- package/dist/commands/autotest.js.map +1 -0
- package/dist/commands/benchmark.d.ts +6 -0
- package/dist/commands/benchmark.d.ts.map +1 -0
- package/dist/commands/benchmark.js +215 -0
- package/dist/commands/benchmark.js.map +1 -0
- package/dist/commands/check.d.ts +10 -0
- package/dist/commands/check.d.ts.map +1 -0
- package/dist/commands/check.js +333 -0
- package/dist/commands/check.js.map +1 -0
- package/dist/commands/context.d.ts +16 -0
- package/dist/commands/context.d.ts.map +1 -0
- package/dist/commands/context.js +219 -0
- package/dist/commands/context.js.map +1 -0
- package/dist/commands/cost.d.ts +9 -0
- package/dist/commands/cost.d.ts.map +1 -0
- package/dist/commands/cost.js +142 -0
- package/dist/commands/cost.js.map +1 -0
- package/dist/commands/dashboard.d.ts +10 -0
- package/dist/commands/dashboard.d.ts.map +1 -0
- package/dist/commands/dashboard.js +462 -0
- package/dist/commands/dashboard.js.map +1 -0
- package/dist/commands/diff.d.ts +6 -0
- package/dist/commands/diff.d.ts.map +1 -0
- package/dist/commands/diff.js +118 -0
- package/dist/commands/diff.js.map +1 -0
- package/dist/commands/doctor.d.ts +12 -0
- package/dist/commands/doctor.d.ts.map +1 -0
- package/dist/commands/doctor.js +203 -0
- package/dist/commands/doctor.js.map +1 -0
- package/dist/commands/explain.d.ts +7 -0
- package/dist/commands/explain.d.ts.map +1 -0
- package/dist/commands/explain.js +164 -0
- package/dist/commands/explain.js.map +1 -0
- package/dist/commands/flaky.d.ts +10 -0
- package/dist/commands/flaky.d.ts.map +1 -0
- package/dist/commands/flaky.js +141 -0
- package/dist/commands/flaky.js.map +1 -0
- package/dist/commands/generateClaudeMd.d.ts +11 -0
- package/dist/commands/generateClaudeMd.d.ts.map +1 -0
- package/dist/commands/generateClaudeMd.js +278 -0
- package/dist/commands/generateClaudeMd.js.map +1 -0
- package/dist/commands/generateRules.d.ts +11 -0
- package/dist/commands/generateRules.d.ts.map +1 -0
- package/dist/commands/generateRules.js +413 -0
- package/dist/commands/generateRules.js.map +1 -0
- package/dist/commands/heatmap.d.ts +7 -0
- package/dist/commands/heatmap.d.ts.map +1 -0
- package/dist/commands/heatmap.js +117 -0
- package/dist/commands/heatmap.js.map +1 -0
- package/dist/commands/history.d.ts +13 -0
- package/dist/commands/history.d.ts.map +1 -0
- package/dist/commands/history.js +113 -0
- package/dist/commands/history.js.map +1 -0
- package/dist/commands/hooks.d.ts +9 -0
- package/dist/commands/hooks.d.ts.map +1 -0
- package/dist/commands/hooks.js +199 -0
- package/dist/commands/hooks.js.map +1 -0
- package/dist/commands/improve.d.ts +7 -0
- package/dist/commands/improve.d.ts.map +1 -0
- package/dist/commands/improve.js +192 -0
- package/dist/commands/improve.js.map +1 -0
- package/dist/commands/init.d.ts +9 -0
- package/dist/commands/init.d.ts.map +1 -0
- package/dist/commands/init.js +270 -0
- package/dist/commands/init.js.map +1 -0
- package/dist/commands/installHook.d.ts +9 -0
- package/dist/commands/installHook.d.ts.map +1 -0
- package/dist/commands/installHook.js +78 -0
- package/dist/commands/installHook.js.map +1 -0
- package/dist/commands/lint.d.ts +6 -0
- package/dist/commands/lint.d.ts.map +1 -0
- package/dist/commands/lint.js +237 -0
- package/dist/commands/lint.js.map +1 -0
- package/dist/commands/map.d.ts +9 -0
- package/dist/commands/map.d.ts.map +1 -0
- package/dist/commands/map.js +114 -0
- package/dist/commands/map.js.map +1 -0
- package/dist/commands/mcp.d.ts +6 -0
- package/dist/commands/mcp.d.ts.map +1 -0
- package/dist/commands/mcp.js +151 -0
- package/dist/commands/mcp.js.map +1 -0
- package/dist/commands/models.d.ts +9 -0
- package/dist/commands/models.d.ts.map +1 -0
- package/dist/commands/models.js +89 -0
- package/dist/commands/models.js.map +1 -0
- package/dist/commands/pack.d.ts +10 -0
- package/dist/commands/pack.d.ts.map +1 -0
- package/dist/commands/pack.js +248 -0
- package/dist/commands/pack.js.map +1 -0
- package/dist/commands/recommend.d.ts +10 -0
- package/dist/commands/recommend.d.ts.map +1 -0
- package/dist/commands/recommend.js +472 -0
- package/dist/commands/recommend.js.map +1 -0
- package/dist/commands/regression.d.ts +10 -0
- package/dist/commands/regression.d.ts.map +1 -0
- package/dist/commands/regression.js +212 -0
- package/dist/commands/regression.js.map +1 -0
- package/dist/commands/repl.d.ts +9 -0
- package/dist/commands/repl.d.ts.map +1 -0
- package/dist/commands/repl.js +245 -0
- package/dist/commands/repl.js.map +1 -0
- package/dist/commands/scan.d.ts +10 -0
- package/dist/commands/scan.d.ts.map +1 -0
- package/dist/commands/scan.js +352 -0
- package/dist/commands/scan.js.map +1 -0
- package/dist/commands/score.d.ts +10 -0
- package/dist/commands/score.d.ts.map +1 -0
- package/dist/commands/score.js +192 -0
- package/dist/commands/score.js.map +1 -0
- package/dist/commands/security.d.ts +10 -0
- package/dist/commands/security.d.ts.map +1 -0
- package/dist/commands/security.js +211 -0
- package/dist/commands/security.js.map +1 -0
- package/dist/commands/simulate.d.ts +7 -0
- package/dist/commands/simulate.d.ts.map +1 -0
- package/dist/commands/simulate.js +149 -0
- package/dist/commands/simulate.js.map +1 -0
- package/dist/commands/summary.d.ts +9 -0
- package/dist/commands/summary.d.ts.map +1 -0
- package/dist/commands/summary.js +271 -0
- package/dist/commands/summary.js.map +1 -0
- package/dist/commands/test.d.ts +9 -0
- package/dist/commands/test.d.ts.map +1 -0
- package/dist/commands/test.js +219 -0
- package/dist/commands/test.js.map +1 -0
- package/dist/commands/ui.d.ts +8 -0
- package/dist/commands/ui.d.ts.map +1 -0
- package/dist/commands/ui.js +222 -0
- package/dist/commands/ui.js.map +1 -0
- package/dist/commands/validate.d.ts +7 -0
- package/dist/commands/validate.d.ts.map +1 -0
- package/dist/commands/validate.js +254 -0
- package/dist/commands/validate.js.map +1 -0
- package/dist/commands/workflow.d.ts +39 -0
- package/dist/commands/workflow.d.ts.map +1 -0
- package/dist/commands/workflow.js +309 -0
- package/dist/commands/workflow.js.map +1 -0
- package/dist/core/__tests__/contextAnalyzer.test.d.ts +2 -0
- package/dist/core/__tests__/contextAnalyzer.test.d.ts.map +1 -0
- package/dist/core/__tests__/contextAnalyzer.test.js +48 -0
- package/dist/core/__tests__/contextAnalyzer.test.js.map +1 -0
- package/dist/core/__tests__/promptLinter.test.d.ts +2 -0
- package/dist/core/__tests__/promptLinter.test.d.ts.map +1 -0
- package/dist/core/__tests__/promptLinter.test.js +74 -0
- package/dist/core/__tests__/promptLinter.test.js.map +1 -0
- package/dist/core/__tests__/promptRunner.test.d.ts +2 -0
- package/dist/core/__tests__/promptRunner.test.d.ts.map +1 -0
- package/dist/core/__tests__/promptRunner.test.js +84 -0
- package/dist/core/__tests__/promptRunner.test.js.map +1 -0
- package/dist/core/__tests__/securityScanner.test.d.ts +2 -0
- package/dist/core/__tests__/securityScanner.test.d.ts.map +1 -0
- package/dist/core/__tests__/securityScanner.test.js +39 -0
- package/dist/core/__tests__/securityScanner.test.js.map +1 -0
- package/dist/core/agentTracer.d.ts +21 -0
- package/dist/core/agentTracer.d.ts.map +1 -0
- package/dist/core/agentTracer.js +355 -0
- package/dist/core/agentTracer.js.map +1 -0
- package/dist/core/anthropicClient.d.ts +26 -0
- package/dist/core/anthropicClient.d.ts.map +1 -0
- package/dist/core/anthropicClient.js +62 -0
- package/dist/core/anthropicClient.js.map +1 -0
- package/dist/core/benchmarkRunner.d.ts +25 -0
- package/dist/core/benchmarkRunner.d.ts.map +1 -0
- package/dist/core/benchmarkRunner.js +182 -0
- package/dist/core/benchmarkRunner.js.map +1 -0
- package/dist/core/contextAnalyzer.d.ts +19 -0
- package/dist/core/contextAnalyzer.d.ts.map +1 -0
- package/dist/core/contextAnalyzer.js +221 -0
- package/dist/core/contextAnalyzer.js.map +1 -0
- package/dist/core/contextPacker.d.ts +26 -0
- package/dist/core/contextPacker.d.ts.map +1 -0
- package/dist/core/contextPacker.js +358 -0
- package/dist/core/contextPacker.js.map +1 -0
- package/dist/core/datasetRunner.d.ts +10 -0
- package/dist/core/datasetRunner.d.ts.map +1 -0
- package/dist/core/datasetRunner.js +130 -0
- package/dist/core/datasetRunner.js.map +1 -0
- package/dist/core/doctorRunner.d.ts +24 -0
- package/dist/core/doctorRunner.d.ts.map +1 -0
- package/dist/core/doctorRunner.js +278 -0
- package/dist/core/doctorRunner.js.map +1 -0
- package/dist/core/hookScanner.d.ts +24 -0
- package/dist/core/hookScanner.d.ts.map +1 -0
- package/dist/core/hookScanner.js +226 -0
- package/dist/core/hookScanner.js.map +1 -0
- package/dist/core/mcpScanner.d.ts +22 -0
- package/dist/core/mcpScanner.d.ts.map +1 -0
- package/dist/core/mcpScanner.js +290 -0
- package/dist/core/mcpScanner.js.map +1 -0
- package/dist/core/modelRegistry.d.ts +35 -0
- package/dist/core/modelRegistry.d.ts.map +1 -0
- package/dist/core/modelRegistry.js +97 -0
- package/dist/core/modelRegistry.js.map +1 -0
- package/dist/core/promptDiff.d.ts +25 -0
- package/dist/core/promptDiff.d.ts.map +1 -0
- package/dist/core/promptDiff.js +130 -0
- package/dist/core/promptDiff.js.map +1 -0
- package/dist/core/promptExplainer.d.ts +17 -0
- package/dist/core/promptExplainer.d.ts.map +1 -0
- package/dist/core/promptExplainer.js +334 -0
- package/dist/core/promptExplainer.js.map +1 -0
- package/dist/core/promptImprover.d.ts +19 -0
- package/dist/core/promptImprover.d.ts.map +1 -0
- package/dist/core/promptImprover.js +260 -0
- package/dist/core/promptImprover.js.map +1 -0
- package/dist/core/promptLinter.d.ts +24 -0
- package/dist/core/promptLinter.d.ts.map +1 -0
- package/dist/core/promptLinter.js +319 -0
- package/dist/core/promptLinter.js.map +1 -0
- package/dist/core/promptRunner.d.ts +31 -0
- package/dist/core/promptRunner.d.ts.map +1 -0
- package/dist/core/promptRunner.js +427 -0
- package/dist/core/promptRunner.js.map +1 -0
- package/dist/core/providers/anthropic.d.ts +10 -0
- package/dist/core/providers/anthropic.d.ts.map +1 -0
- package/dist/core/providers/anthropic.js +26 -0
- package/dist/core/providers/anthropic.js.map +1 -0
- package/dist/core/providers/base.d.ts +22 -0
- package/dist/core/providers/base.d.ts.map +1 -0
- package/dist/core/providers/base.js +2 -0
- package/dist/core/providers/base.js.map +1 -0
- package/dist/core/providers/factory.d.ts +7 -0
- package/dist/core/providers/factory.d.ts.map +1 -0
- package/dist/core/providers/factory.js +42 -0
- package/dist/core/providers/factory.js.map +1 -0
- package/dist/core/providers/google.d.ts +10 -0
- package/dist/core/providers/google.d.ts.map +1 -0
- package/dist/core/providers/google.js +47 -0
- package/dist/core/providers/google.js.map +1 -0
- package/dist/core/providers/openai.d.ts +19 -0
- package/dist/core/providers/openai.d.ts.map +1 -0
- package/dist/core/providers/openai.js +54 -0
- package/dist/core/providers/openai.js.map +1 -0
- package/dist/core/regressionRunner.d.ts +11 -0
- package/dist/core/regressionRunner.d.ts.map +1 -0
- package/dist/core/regressionRunner.js +116 -0
- package/dist/core/regressionRunner.js.map +1 -0
- package/dist/core/repositorySimulator.d.ts +17 -0
- package/dist/core/repositorySimulator.d.ts.map +1 -0
- package/dist/core/repositorySimulator.js +104 -0
- package/dist/core/repositorySimulator.js.map +1 -0
- package/dist/core/scorer.d.ts +30 -0
- package/dist/core/scorer.d.ts.map +1 -0
- package/dist/core/scorer.js +317 -0
- package/dist/core/scorer.js.map +1 -0
- package/dist/core/securityScanner.d.ts +23 -0
- package/dist/core/securityScanner.d.ts.map +1 -0
- package/dist/core/securityScanner.js +216 -0
- package/dist/core/securityScanner.js.map +1 -0
- package/dist/core/skillValidator.d.ts +41 -0
- package/dist/core/skillValidator.d.ts.map +1 -0
- package/dist/core/skillValidator.js +235 -0
- package/dist/core/skillValidator.js.map +1 -0
- package/dist/core/testHistory.d.ts +44 -0
- package/dist/core/testHistory.d.ts.map +1 -0
- package/dist/core/testHistory.js +91 -0
- package/dist/core/testHistory.js.map +1 -0
- package/dist/tokenizers/claudeTokenizer.d.ts +26 -0
- package/dist/tokenizers/claudeTokenizer.d.ts.map +1 -0
- package/dist/tokenizers/claudeTokenizer.js +83 -0
- package/dist/tokenizers/claudeTokenizer.js.map +1 -0
- package/dist/types/agent.d.ts +26 -0
- package/dist/types/agent.d.ts.map +1 -0
- package/dist/types/agent.js +5 -0
- package/dist/types/agent.js.map +1 -0
- package/dist/types/config.d.ts +30 -0
- package/dist/types/config.d.ts.map +1 -0
- package/dist/types/config.js +5 -0
- package/dist/types/config.js.map +1 -0
- package/dist/types/context.d.ts +77 -0
- package/dist/types/context.d.ts.map +1 -0
- package/dist/types/context.js +5 -0
- package/dist/types/context.js.map +1 -0
- package/dist/types/dataset.d.ts +26 -0
- package/dist/types/dataset.d.ts.map +1 -0
- package/dist/types/dataset.js +5 -0
- package/dist/types/dataset.js.map +1 -0
- package/dist/types/diagnostics.d.ts +31 -0
- package/dist/types/diagnostics.d.ts.map +1 -0
- package/dist/types/diagnostics.js +5 -0
- package/dist/types/diagnostics.js.map +1 -0
- package/dist/types/prompt.d.ts +53 -0
- package/dist/types/prompt.d.ts.map +1 -0
- package/dist/types/prompt.js +5 -0
- package/dist/types/prompt.js.map +1 -0
- package/dist/types/results.d.ts +42 -0
- package/dist/types/results.d.ts.map +1 -0
- package/dist/types/results.js +5 -0
- package/dist/types/results.js.map +1 -0
- package/dist/ui/dashboard.d.ts +57 -0
- package/dist/ui/dashboard.d.ts.map +1 -0
- package/dist/ui/dashboard.js +644 -0
- package/dist/ui/dashboard.js.map +1 -0
- package/dist/utils/__tests__/hashing.test.d.ts +2 -0
- package/dist/utils/__tests__/hashing.test.d.ts.map +1 -0
- package/dist/utils/__tests__/hashing.test.js +28 -0
- package/dist/utils/__tests__/hashing.test.js.map +1 -0
- package/dist/utils/__tests__/output.test.d.ts +2 -0
- package/dist/utils/__tests__/output.test.d.ts.map +1 -0
- package/dist/utils/__tests__/output.test.js +62 -0
- package/dist/utils/__tests__/output.test.js.map +1 -0
- package/dist/utils/cache.d.ts +29 -0
- package/dist/utils/cache.d.ts.map +1 -0
- package/dist/utils/cache.js +87 -0
- package/dist/utils/cache.js.map +1 -0
- package/dist/utils/config.d.ts +15 -0
- package/dist/utils/config.d.ts.map +1 -0
- package/dist/utils/config.js +61 -0
- package/dist/utils/config.js.map +1 -0
- package/dist/utils/errors.d.ts +43 -0
- package/dist/utils/errors.d.ts.map +1 -0
- package/dist/utils/errors.js +83 -0
- package/dist/utils/errors.js.map +1 -0
- package/dist/utils/fs.d.ts +44 -0
- package/dist/utils/fs.d.ts.map +1 -0
- package/dist/utils/fs.js +119 -0
- package/dist/utils/fs.js.map +1 -0
- package/dist/utils/hashing.d.ts +13 -0
- package/dist/utils/hashing.d.ts.map +1 -0
- package/dist/utils/hashing.js +18 -0
- package/dist/utils/hashing.js.map +1 -0
- package/dist/utils/logger.d.ts +32 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +76 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/output.d.ts +34 -0
- package/dist/utils/output.d.ts.map +1 -0
- package/dist/utils/output.js +99 -0
- package/dist/utils/output.js.map +1 -0
- package/dist/utils/paths.d.ts +33 -0
- package/dist/utils/paths.d.ts.map +1 -0
- package/dist/utils/paths.js +51 -0
- package/dist/utils/paths.js.map +1 -0
- package/dist/utils/spinner.d.ts +23 -0
- package/dist/utils/spinner.d.ts.map +1 -0
- package/dist/utils/spinner.js +79 -0
- package/dist/utils/spinner.js.map +1 -0
- package/package.json +60 -0
|
@@ -0,0 +1,211 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* `codeprobe security [path]` — Security checks on prompt specs
|
|
3
|
+
* and configuration files.
|
|
4
|
+
*
|
|
5
|
+
* Detects potential prompt injection patterns, leaked secrets,
|
|
6
|
+
* and unsafe configurations.
|
|
7
|
+
*/
|
|
8
|
+
import { resolve } from 'node:path';
|
|
9
|
+
import { resolvePath } from '../utils/paths.js';
|
|
10
|
+
import { readTextFile, isDirectory, fileExists, getRelativePath as getRelPath } from '../utils/fs.js';
|
|
11
|
+
import { setLogLevel } from '../utils/logger.js';
|
|
12
|
+
/** Security check patterns. */
|
|
13
|
+
const SECURITY_RULES = [
|
|
14
|
+
{
|
|
15
|
+
id: 'injection-ignore',
|
|
16
|
+
pattern: /ignore\s+(previous|above|all)\s+(instructions?|prompts?|rules?)/i,
|
|
17
|
+
severity: 'critical',
|
|
18
|
+
message: 'Potential prompt injection: instruction override pattern detected',
|
|
19
|
+
},
|
|
20
|
+
{
|
|
21
|
+
id: 'injection-pretend',
|
|
22
|
+
pattern: /pretend\s+(you\s+are|to\s+be|that)/i,
|
|
23
|
+
severity: 'high',
|
|
24
|
+
message: 'Potential prompt injection: persona override pattern detected',
|
|
25
|
+
},
|
|
26
|
+
{
|
|
27
|
+
id: 'injection-system-override',
|
|
28
|
+
pattern: /system\s*:\s*you\s+are\s+now/i,
|
|
29
|
+
severity: 'critical',
|
|
30
|
+
message: 'Potential system prompt override detected',
|
|
31
|
+
},
|
|
32
|
+
{
|
|
33
|
+
id: 'leaked-api-key',
|
|
34
|
+
pattern: /(?:api[_-]?key|apikey)\s*[:=]\s*['"][A-Za-z0-9_\-]{20,}/i,
|
|
35
|
+
severity: 'critical',
|
|
36
|
+
message: 'Potential API key detected in prompt spec',
|
|
37
|
+
},
|
|
38
|
+
{
|
|
39
|
+
id: 'leaked-secret',
|
|
40
|
+
pattern: /(?:secret|password|token|credential)\s*[:=]\s*['"][^'"]{8,}/i,
|
|
41
|
+
severity: 'critical',
|
|
42
|
+
message: 'Potential secret or credential detected',
|
|
43
|
+
},
|
|
44
|
+
{
|
|
45
|
+
id: 'leaked-env-var',
|
|
46
|
+
pattern: /(?:sk-|pk_|rk_|ghp_|gho_|github_pat_)[A-Za-z0-9_\-]{20,}/,
|
|
47
|
+
severity: 'critical',
|
|
48
|
+
message: 'Potential service token (Stripe, GitHub, etc.) detected',
|
|
49
|
+
},
|
|
50
|
+
{
|
|
51
|
+
id: 'unsafe-eval',
|
|
52
|
+
pattern: /eval\s*\(|exec\s*\(|Function\s*\(/,
|
|
53
|
+
severity: 'high',
|
|
54
|
+
message: 'Dynamic code execution detected — potential security risk',
|
|
55
|
+
},
|
|
56
|
+
{
|
|
57
|
+
id: 'url-in-prompt',
|
|
58
|
+
pattern: /https?:\/\/[^\s'"]+/,
|
|
59
|
+
severity: 'low',
|
|
60
|
+
message: 'URL found in prompt — verify it is intentional and safe',
|
|
61
|
+
},
|
|
62
|
+
{
|
|
63
|
+
id: 'pii-email',
|
|
64
|
+
pattern: /[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/,
|
|
65
|
+
severity: 'medium',
|
|
66
|
+
message: 'Email address found — may contain PII',
|
|
67
|
+
},
|
|
68
|
+
{
|
|
69
|
+
id: 'excessive-permissions',
|
|
70
|
+
pattern: /\b(sudo|root|admin|superuser)\b.*\b(access|permission|privilege)/i,
|
|
71
|
+
severity: 'medium',
|
|
72
|
+
message: 'Reference to elevated permissions detected',
|
|
73
|
+
},
|
|
74
|
+
{
|
|
75
|
+
id: 'data-exfiltration',
|
|
76
|
+
pattern: /send\s+(to|data|all|everything)\s+(to\s+)?https?:\/\//i,
|
|
77
|
+
severity: 'high',
|
|
78
|
+
message: 'Potential data exfiltration pattern detected',
|
|
79
|
+
},
|
|
80
|
+
];
|
|
81
|
+
/**
|
|
82
|
+
* Scan a file for security issues.
|
|
83
|
+
*/
|
|
84
|
+
function scanFileContent(content, filePath) {
|
|
85
|
+
const findings = [];
|
|
86
|
+
const lines = content.split('\n');
|
|
87
|
+
for (let lineIdx = 0; lineIdx < lines.length; lineIdx++) {
|
|
88
|
+
const line = lines[lineIdx];
|
|
89
|
+
for (const rule of SECURITY_RULES) {
|
|
90
|
+
if (rule.pattern.test(line)) {
|
|
91
|
+
// Extract a snippet (trim to 100 chars)
|
|
92
|
+
const snippet = line.trim().length > 100
|
|
93
|
+
? line.trim().slice(0, 97) + '...'
|
|
94
|
+
: line.trim();
|
|
95
|
+
findings.push({
|
|
96
|
+
file: filePath,
|
|
97
|
+
rule: rule.id,
|
|
98
|
+
severity: rule.severity,
|
|
99
|
+
message: rule.message,
|
|
100
|
+
line: lineIdx + 1,
|
|
101
|
+
snippet,
|
|
102
|
+
});
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
return findings;
|
|
107
|
+
}
|
|
108
|
+
/**
|
|
109
|
+
* Run security scanning on prompt specs.
|
|
110
|
+
*/
|
|
111
|
+
async function securityScanner(targetPath) {
|
|
112
|
+
const { glob } = await import('glob');
|
|
113
|
+
const findings = [];
|
|
114
|
+
let files;
|
|
115
|
+
if (await fileExists(targetPath)) {
|
|
116
|
+
files = [targetPath];
|
|
117
|
+
}
|
|
118
|
+
else if (await isDirectory(targetPath)) {
|
|
119
|
+
files = await glob(resolve(targetPath, '**/*.{yaml,yml,json,md}'), {
|
|
120
|
+
absolute: true,
|
|
121
|
+
ignore: ['**/node_modules/**', '**/.git/**'],
|
|
122
|
+
});
|
|
123
|
+
}
|
|
124
|
+
else {
|
|
125
|
+
throw new Error(`Path not found: ${targetPath}`);
|
|
126
|
+
}
|
|
127
|
+
for (const file of files) {
|
|
128
|
+
const content = await readTextFile(file);
|
|
129
|
+
if (!content)
|
|
130
|
+
continue;
|
|
131
|
+
const relPath = getRelPath(process.cwd(), file);
|
|
132
|
+
const fileFindings = scanFileContent(content, relPath);
|
|
133
|
+
findings.push(...fileFindings);
|
|
134
|
+
}
|
|
135
|
+
// Sort by severity
|
|
136
|
+
const severityOrder = {
|
|
137
|
+
critical: 0,
|
|
138
|
+
high: 1,
|
|
139
|
+
medium: 2,
|
|
140
|
+
low: 3,
|
|
141
|
+
};
|
|
142
|
+
findings.sort((a, b) => severityOrder[a.severity] - severityOrder[b.severity]);
|
|
143
|
+
return findings;
|
|
144
|
+
}
|
|
145
|
+
export function registerSecurityCommand(program) {
|
|
146
|
+
program
|
|
147
|
+
.command('security [path]')
|
|
148
|
+
.description('Security checks — detect injection patterns, leaked secrets, unsafe configs')
|
|
149
|
+
.option('--json', 'Output findings as JSON')
|
|
150
|
+
.option('--output <file>', 'Write JSON results to a file')
|
|
151
|
+
.action(async (pathArg, options) => {
|
|
152
|
+
if (options.json || options.output) {
|
|
153
|
+
setLogLevel('silent');
|
|
154
|
+
}
|
|
155
|
+
const chalk = (await import('chalk')).default;
|
|
156
|
+
const targetPath = resolvePath(pathArg ?? 'prompts');
|
|
157
|
+
const findings = await securityScanner(targetPath);
|
|
158
|
+
if (options.output) {
|
|
159
|
+
const { writeFile } = await import('node:fs/promises');
|
|
160
|
+
await writeFile(resolvePath(options.output), JSON.stringify(findings, null, 2));
|
|
161
|
+
console.log(`Results written to ${options.output}`);
|
|
162
|
+
return;
|
|
163
|
+
}
|
|
164
|
+
if (options.json) {
|
|
165
|
+
console.log(JSON.stringify(findings, null, 2));
|
|
166
|
+
return;
|
|
167
|
+
}
|
|
168
|
+
if (findings.length === 0) {
|
|
169
|
+
console.log(chalk.green('\nNo security issues found.\n'));
|
|
170
|
+
return;
|
|
171
|
+
}
|
|
172
|
+
const severityColor = {
|
|
173
|
+
critical: chalk.bgRed.white,
|
|
174
|
+
high: chalk.red,
|
|
175
|
+
medium: chalk.yellow,
|
|
176
|
+
low: chalk.blue,
|
|
177
|
+
};
|
|
178
|
+
console.log(chalk.bold(`\nSecurity Findings (${findings.length})`));
|
|
179
|
+
console.log('');
|
|
180
|
+
// Group by file
|
|
181
|
+
const grouped = new Map();
|
|
182
|
+
for (const f of findings) {
|
|
183
|
+
const list = grouped.get(f.file) ?? [];
|
|
184
|
+
list.push(f);
|
|
185
|
+
grouped.set(f.file, list);
|
|
186
|
+
}
|
|
187
|
+
for (const [file, fileFindings] of grouped) {
|
|
188
|
+
console.log(chalk.bold(` ${file}`));
|
|
189
|
+
for (const finding of fileFindings) {
|
|
190
|
+
const sevLabel = severityColor[finding.severity](` ${finding.severity.toUpperCase()} `);
|
|
191
|
+
const lineStr = finding.line ? `:${finding.line}` : '';
|
|
192
|
+
console.log(` ${sevLabel} ${finding.rule}${lineStr}`);
|
|
193
|
+
console.log(` ${finding.message}`);
|
|
194
|
+
if (finding.snippet) {
|
|
195
|
+
console.log(chalk.dim(` ${finding.snippet}`));
|
|
196
|
+
}
|
|
197
|
+
}
|
|
198
|
+
console.log('');
|
|
199
|
+
}
|
|
200
|
+
// Summary
|
|
201
|
+
const critCount = findings.filter((f) => f.severity === 'critical').length;
|
|
202
|
+
const highCount = findings.filter((f) => f.severity === 'high').length;
|
|
203
|
+
const medCount = findings.filter((f) => f.severity === 'medium').length;
|
|
204
|
+
const lowCount = findings.filter((f) => f.severity === 'low').length;
|
|
205
|
+
console.log(chalk.dim(` ${critCount} critical, ${highCount} high, ${medCount} medium, ${lowCount} low\n`));
|
|
206
|
+
if (critCount > 0 || highCount > 0) {
|
|
207
|
+
process.exitCode = 1;
|
|
208
|
+
}
|
|
209
|
+
});
|
|
210
|
+
}
|
|
211
|
+
//# sourceMappingURL=security.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"security.js","sourceRoot":"","sources":["../../src/commands/security.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAGH,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,UAAU,EAAE,eAAe,IAAI,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACtG,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAGjD,+BAA+B;AAC/B,MAAM,cAAc,GAKf;IACH;QACE,EAAE,EAAE,kBAAkB;QACtB,OAAO,EAAE,kEAAkE;QAC3E,QAAQ,EAAE,UAAU;QACpB,OAAO,EAAE,mEAAmE;KAC7E;IACD;QACE,EAAE,EAAE,mBAAmB;QACvB,OAAO,EAAE,qCAAqC;QAC9C,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,+DAA+D;KACzE;IACD;QACE,EAAE,EAAE,2BAA2B;QAC/B,OAAO,EAAE,+BAA+B;QACxC,QAAQ,EAAE,UAAU;QACpB,OAAO,EAAE,2CAA2C;KACrD;IACD;QACE,EAAE,EAAE,gBAAgB;QACpB,OAAO,EAAE,0DAA0D;QACnE,QAAQ,EAAE,UAAU;QACpB,OAAO,EAAE,2CAA2C;KACrD;IACD;QACE,EAAE,EAAE,eAAe;QACnB,OAAO,EAAE,8DAA8D;QACvE,QAAQ,EAAE,UAAU;QACpB,OAAO,EAAE,yCAAyC;KACnD;IACD;QACE,EAAE,EAAE,gBAAgB;QACpB,OAAO,EAAE,0DAA0D;QACnE,QAAQ,EAAE,UAAU;QACpB,OAAO,EAAE,yDAAyD;KACnE;IACD;QACE,EAAE,EAAE,aAAa;QACjB,OAAO,EAAE,mCAAmC;QAC5C,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,2DAA2D;KACrE;IACD;QACE,EAAE,EAAE,eAAe;QACnB,OAAO,EAAE,qBAAqB;QAC9B,QAAQ,EAAE,KAAK;QACf,OAAO,EAAE,yDAAyD;KACnE;IACD;QACE,EAAE,EAAE,WAAW;QACf,OAAO,EAAE,gDAAgD;QACzD,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,uCAAuC;KACjD;IACD;QACE,EAAE,EAAE,uBAAuB;QAC3B,OAAO,EAAE,mEAAmE;QAC5E,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,4CAA4C;KACtD;IACD;QACE,EAAE,EAAE,mBAAmB;QACvB,OAAO,EAAE,wDAAwD;QACjE,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,8CAA8C;KACxD;CACF,CAAC;AAEF;;GAEG;AACH,SAAS,eAAe,CACtB,OAAe,EACf,QAAgB;IAEhB,MAAM,QAAQ,GAAsB,EAAE,CAAC;IACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC;QACxD,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAE,CAAC;QAE7B,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5B,wCAAwC;gBACxC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,GAAG;oBACtC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK;oBAClC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBAEhB,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,IAAI,CAAC,EAAE;oBACb,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,IAAI,EAAE,OAAO,GAAG,CAAC;oBACjB,OAAO;iBACR,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,eAAe,CAAC,UAAkB;IAC/C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,CAAC;IACtC,MAAM,QAAQ,GAAsB,EAAE,CAAC;IAEvC,IAAI,KAAe,CAAC;IAEpB,IAAI,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QACjC,KAAK,GAAG,CAAC,UAAU,CAAC,CAAC;IACvB,CAAC;SAAM,IAAI,MAAM,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC;QACzC,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,yBAAyB,CAAC,EAAE;YACjE,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,CAAC,oBAAoB,EAAE,YAAY,CAAC;SAC7C,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,mBAAmB,UAAU,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,OAAO;YAAE,SAAS;QAEvB,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,IAAI,CAAC,CAAC;QAChD,MAAM,YAAY,GAAG,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACvD,QAAQ,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;IACjC,CAAC;IAED,mBAAmB;IACnB,MAAM,aAAa,GAAgD;QACjE,QAAQ,EAAE,CAAC;QACX,IAAI,EAAE,CAAC;QACP,MAAM,EAAE,CAAC;QACT,GAAG,EAAE,CAAC;KACP,CAAC;IACF,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/E,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,OAAgB;IACtD,OAAO;SACJ,OAAO,CAAC,iBAAiB,CAAC;SAC1B,WAAW,CAAC,6EAA6E,CAAC;SAC1F,MAAM,CAAC,QAAQ,EAAE,yBAAyB,CAAC;SAC3C,MAAM,CAAC,iBAAiB,EAAE,8BAA8B,CAAC;SACzD,MAAM,CAAC,KAAK,EACX,OAA2B,EAC3B,OAA4C,EAC5C,EAAE;QACF,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnC,WAAW,CAAC,QAAQ,CAAC,CAAC;QACxB,CAAC;QAED,MAAM,KAAK,GAAG,CAAC,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QAC9C,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,IAAI,SAAS,CAAC,CAAC;QAErD,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,UAAU,CAAC,CAAC;QAEnD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;YACvD,MAAM,SAAS,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAChF,OAAO,CAAC,GAAG,CAAC,sBAAsB,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YACpD,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC/C,OAAO;QACT,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,MAAM,aAAa,GAA+D;YAChF,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,KAAK;YAC3B,IAAI,EAAE,KAAK,CAAC,GAAG;YACf,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,GAAG,EAAE,KAAK,CAAC,IAAI;SAChB,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,wBAAwB,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACpE,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEhB,gBAAgB;QAChB,MAAM,OAAO,GAAG,IAAI,GAAG,EAA6B,CAAC;QACrD,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACvC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACb,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,KAAK,MAAM,CAAC,IAAI,EAAE,YAAY,CAAC,IAAI,OAAO,EAAE,CAAC;YAC3C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;YACrC,KAAK,MAAM,OAAO,IAAI,YAAY,EAAE,CAAC;gBACnC,MAAM,QAAQ,GAAG,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;gBACxF,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvD,OAAO,CAAC,GAAG,CAAC,OAAO,QAAQ,KAAK,OAAO,CAAC,IAAI,GAAG,OAAO,EAAE,CAAC,CAAC;gBAC1D,OAAO,CAAC,GAAG,CAAC,cAAc,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC7C,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBACpB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC1D,CAAC;YACH,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC;QAED,UAAU;QACV,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC,MAAM,CAAC;QAC3E,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,MAAM,CAAC;QACvE,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,MAAM,CAAC;QACxE,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,KAAK,CAAC,CAAC,MAAM,CAAC;QAErE,OAAO,CAAC,GAAG,CACT,KAAK,CAAC,GAAG,CAAC,KAAK,SAAS,cAAc,SAAS,UAAU,QAAQ,YAAY,QAAQ,QAAQ,CAAC,CAC/F,CAAC;QAEF,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YACnC,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"simulate.d.ts","sourceRoot":"","sources":["../../src/commands/simulate.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAyFpC,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAkF9D"}
|
|
@@ -0,0 +1,149 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* `codeprobe simulate [path]` — Simulate context fit against
|
|
3
|
+
* target context windows.
|
|
4
|
+
*/
|
|
5
|
+
import { resolvePath } from '../utils/paths.js';
|
|
6
|
+
import { walkDirectory } from '../utils/fs.js';
|
|
7
|
+
import { estimateTokens } from '../tokenizers/claudeTokenizer.js';
|
|
8
|
+
import { readFile, stat } from 'node:fs/promises';
|
|
9
|
+
import { formatTokens, formatPercentage } from '../utils/output.js';
|
|
10
|
+
import { setLogLevel } from '../utils/logger.js';
|
|
11
|
+
import { getModel } from '../core/modelRegistry.js';
|
|
12
|
+
const DEFAULT_IGNORE_DIRS = new Set([
|
|
13
|
+
'node_modules', '.git', 'dist', 'build', 'coverage',
|
|
14
|
+
'.next', '.nuxt', '__pycache__', '.venv', 'vendor',
|
|
15
|
+
'.cache', '.turbo',
|
|
16
|
+
]);
|
|
17
|
+
const TARGETS = [
|
|
18
|
+
{ size: 200_000, label: '200k' },
|
|
19
|
+
{ size: 1_000_000, label: '1M' },
|
|
20
|
+
];
|
|
21
|
+
/**
|
|
22
|
+
* Run a repository simulation.
|
|
23
|
+
*/
|
|
24
|
+
async function repositorySimulator(rootPath, targetFilter, customTargets) {
|
|
25
|
+
const entries = await walkDirectory(rootPath, { ignoreDirs: DEFAULT_IGNORE_DIRS });
|
|
26
|
+
const fileEntries = entries.filter((e) => e.isFile && e.size < 1_000_000);
|
|
27
|
+
let totalTokens = 0;
|
|
28
|
+
for (const entry of fileEntries) {
|
|
29
|
+
try {
|
|
30
|
+
const content = await readFile(entry.path, 'utf-8');
|
|
31
|
+
totalTokens += estimateTokens(content);
|
|
32
|
+
}
|
|
33
|
+
catch {
|
|
34
|
+
// Skip unreadable files
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
// Use custom targets (from --model) if provided, otherwise default targets
|
|
38
|
+
const baseTargets = customTargets ?? TARGETS;
|
|
39
|
+
const activeTargets = targetFilter
|
|
40
|
+
? baseTargets.filter((t) => t.label.toLowerCase() === targetFilter.toLowerCase())
|
|
41
|
+
: baseTargets;
|
|
42
|
+
if (activeTargets.length === 0) {
|
|
43
|
+
throw new Error(`Unknown target: ${targetFilter}. Use "200k" or "1m".`);
|
|
44
|
+
}
|
|
45
|
+
// Reserve ~20% for system prompt, tool defs, etc.
|
|
46
|
+
const reserveRatio = 0.2;
|
|
47
|
+
const targets = activeTargets.map((t) => {
|
|
48
|
+
const reservedBudget = Math.floor(t.size * reserveRatio);
|
|
49
|
+
const available = t.size - reservedBudget;
|
|
50
|
+
return {
|
|
51
|
+
windowSize: t.size,
|
|
52
|
+
windowLabel: t.label,
|
|
53
|
+
fits: totalTokens <= available,
|
|
54
|
+
utilization: available > 0 ? totalTokens / available : 0,
|
|
55
|
+
headroom: available - totalTokens,
|
|
56
|
+
reservedBudget,
|
|
57
|
+
};
|
|
58
|
+
});
|
|
59
|
+
const recommendations = [];
|
|
60
|
+
if (totalTokens > 200_000) {
|
|
61
|
+
recommendations.push('Consider using `codeprobe pack` to build an optimized context plan.');
|
|
62
|
+
}
|
|
63
|
+
if (totalTokens > 1_000_000) {
|
|
64
|
+
recommendations.push('Repository exceeds 1M tokens. Focus on core files and use summaries for large docs.');
|
|
65
|
+
}
|
|
66
|
+
if (totalTokens < 50_000) {
|
|
67
|
+
recommendations.push('Repository fits comfortably. You can include the full codebase as context.');
|
|
68
|
+
}
|
|
69
|
+
return {
|
|
70
|
+
rootPath,
|
|
71
|
+
totalTokens,
|
|
72
|
+
targets,
|
|
73
|
+
recommendations,
|
|
74
|
+
};
|
|
75
|
+
}
|
|
76
|
+
export function registerSimulateCommand(program) {
|
|
77
|
+
program
|
|
78
|
+
.command('simulate [path]')
|
|
79
|
+
.description('Simulate context fit — estimate whether repo fits in target context windows')
|
|
80
|
+
.option('--json', 'Output simulation as JSON')
|
|
81
|
+
.option('--target <target>', 'Target context window: 200k or 1m')
|
|
82
|
+
.option('--model <model>', 'Simulate against a specific model\'s context window')
|
|
83
|
+
.option('--output <file>', 'Write JSON results to a file')
|
|
84
|
+
.action(async (pathArg, options) => {
|
|
85
|
+
if (options.json || options.output) {
|
|
86
|
+
setLogLevel('silent');
|
|
87
|
+
}
|
|
88
|
+
const chalk = (await import('chalk')).default;
|
|
89
|
+
const targetPath = resolvePath(pathArg ?? '.');
|
|
90
|
+
try {
|
|
91
|
+
await stat(targetPath);
|
|
92
|
+
}
|
|
93
|
+
catch {
|
|
94
|
+
console.error(`Error: path not found: ${targetPath}`);
|
|
95
|
+
process.exitCode = 1;
|
|
96
|
+
return;
|
|
97
|
+
}
|
|
98
|
+
// If --model is provided, use that model's context window as the target
|
|
99
|
+
let customTargets;
|
|
100
|
+
if (options.model) {
|
|
101
|
+
const modelInfo = getModel(options.model);
|
|
102
|
+
if (!modelInfo) {
|
|
103
|
+
console.error(`Error: unknown model "${options.model}". Use a model id from the registry (e.g., gpt-4o, claude-sonnet-4-6, gemini-2.5-pro).`);
|
|
104
|
+
process.exitCode = 1;
|
|
105
|
+
return;
|
|
106
|
+
}
|
|
107
|
+
const windowSize = modelInfo.contextWindow;
|
|
108
|
+
const label = `${modelInfo.name} (${Math.round(windowSize / 1000)}k)`;
|
|
109
|
+
customTargets = [{ size: windowSize, label }];
|
|
110
|
+
}
|
|
111
|
+
const result = await repositorySimulator(targetPath, options.target, customTargets);
|
|
112
|
+
if (options.output) {
|
|
113
|
+
const { writeFile } = await import('node:fs/promises');
|
|
114
|
+
await writeFile(resolvePath(options.output), JSON.stringify(result, null, 2));
|
|
115
|
+
console.log(`Results written to ${options.output}`);
|
|
116
|
+
return;
|
|
117
|
+
}
|
|
118
|
+
if (options.json) {
|
|
119
|
+
console.log(JSON.stringify(result, null, 2));
|
|
120
|
+
return;
|
|
121
|
+
}
|
|
122
|
+
console.log(chalk.bold('\nContext Simulation'));
|
|
123
|
+
console.log(chalk.dim(` Root: ${result.rootPath}`));
|
|
124
|
+
console.log(` Total tokens: ${formatTokens(result.totalTokens)}`);
|
|
125
|
+
console.log('');
|
|
126
|
+
for (const target of result.targets) {
|
|
127
|
+
const fitsIcon = target.fits ? chalk.green('FITS') : chalk.red('DOES NOT FIT');
|
|
128
|
+
console.log(chalk.bold(` ${target.windowLabel} window:`));
|
|
129
|
+
console.log(` Status: ${fitsIcon}`);
|
|
130
|
+
console.log(` Utilization: ${formatPercentage(target.utilization)}`);
|
|
131
|
+
console.log(` Reserved: ${formatTokens(target.reservedBudget)} (system/tools)`);
|
|
132
|
+
if (target.headroom >= 0) {
|
|
133
|
+
console.log(` Headroom: ${chalk.green(formatTokens(target.headroom))}`);
|
|
134
|
+
}
|
|
135
|
+
else {
|
|
136
|
+
console.log(` Over budget: ${chalk.red(formatTokens(Math.abs(target.headroom)))}`);
|
|
137
|
+
}
|
|
138
|
+
console.log('');
|
|
139
|
+
}
|
|
140
|
+
if (result.recommendations.length > 0) {
|
|
141
|
+
console.log(chalk.bold(' Recommendations'));
|
|
142
|
+
for (const rec of result.recommendations) {
|
|
143
|
+
console.log(` * ${rec}`);
|
|
144
|
+
}
|
|
145
|
+
console.log('');
|
|
146
|
+
}
|
|
147
|
+
});
|
|
148
|
+
}
|
|
149
|
+
//# sourceMappingURL=simulate.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"simulate.js","sourceRoot":"","sources":["../../src/commands/simulate.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACpE,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AAGpD,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAAC;IAClC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU;IACnD,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,QAAQ;IAClD,QAAQ,EAAE,QAAQ;CACnB,CAAC,CAAC;AAEH,MAAM,OAAO,GAA2C;IACtD,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE;IAChC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE;CACjC,CAAC;AAEF;;GAEG;AACH,KAAK,UAAU,mBAAmB,CAChC,QAAgB,EAChB,YAAqB,EACrB,aAAsD;IAEtD,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,mBAAmB,EAAE,CAAC,CAAC;IACnF,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,IAAI,GAAG,SAAS,CAAC,CAAC;IAE1E,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACpD,WAAW,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,MAAM,WAAW,GAAG,aAAa,IAAI,OAAO,CAAC;IAE7C,MAAM,aAAa,GAAG,YAAY;QAChC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,YAAY,CAAC,WAAW,EAAE,CAAC;QACjF,CAAC,CAAC,WAAW,CAAC;IAEhB,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,KAAK,CAAC,mBAAmB,YAAY,uBAAuB,CAAC,CAAC;IAC1E,CAAC;IAED,kDAAkD;IAClD,MAAM,YAAY,GAAG,GAAG,CAAC;IAEzB,MAAM,OAAO,GAAuB,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,GAAG,YAAY,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,CAAC,CAAC,IAAI,GAAG,cAAc,CAAC;QAC1C,OAAO;YACL,UAAU,EAAE,CAAC,CAAC,IAAI;YAClB,WAAW,EAAE,CAAC,CAAC,KAAK;YACpB,IAAI,EAAE,WAAW,IAAI,SAAS;YAC9B,WAAW,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;YACxD,QAAQ,EAAE,SAAS,GAAG,WAAW;YACjC,cAAc;SACf,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,eAAe,GAAa,EAAE,CAAC;IACrC,IAAI,WAAW,GAAG,OAAO,EAAE,CAAC;QAC1B,eAAe,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAC;IAC9F,CAAC;IACD,IAAI,WAAW,GAAG,SAAS,EAAE,CAAC;QAC5B,eAAe,CAAC,IAAI,CAAC,qFAAqF,CAAC,CAAC;IAC9G,CAAC;IACD,IAAI,WAAW,GAAG,MAAM,EAAE,CAAC;QACzB,eAAe,CAAC,IAAI,CAAC,4EAA4E,CAAC,CAAC;IACrG,CAAC;IAED,OAAO;QACL,QAAQ;QACR,WAAW;QACX,OAAO;QACP,eAAe;KAChB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,OAAgB;IACtD,OAAO;SACJ,OAAO,CAAC,iBAAiB,CAAC;SAC1B,WAAW,CAAC,6EAA6E,CAAC;SAC1F,MAAM,CAAC,QAAQ,EAAE,2BAA2B,CAAC;SAC7C,MAAM,CAAC,mBAAmB,EAAE,mCAAmC,CAAC;SAChE,MAAM,CAAC,iBAAiB,EAAE,qDAAqD,CAAC;SAChF,MAAM,CAAC,iBAAiB,EAAE,8BAA8B,CAAC;SACzD,MAAM,CAAC,KAAK,EACX,OAA2B,EAC3B,OAA6E,EAC7E,EAAE;QACF,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnC,WAAW,CAAC,QAAQ,CAAC,CAAC;QACxB,CAAC;QAED,MAAM,KAAK,GAAG,CAAC,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QAC9C,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,IAAI,GAAG,CAAC,CAAC;QAE/C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,KAAK,CAAC,0BAA0B,UAAU,EAAE,CAAC,CAAC;YACtD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;YACrB,OAAO;QACT,CAAC;QAED,wEAAwE;QACxE,IAAI,aAAiE,CAAC;QACtE,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC1C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,OAAO,CAAC,KAAK,wFAAwF,CAAC,CAAC;gBAC9I,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;gBACrB,OAAO;YACT,CAAC;YACD,MAAM,UAAU,GAAG,SAAS,CAAC,aAAa,CAAC;YAC3C,MAAM,KAAK,GAAG,GAAG,SAAS,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC;YACtE,aAAa,GAAG,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,UAAU,EAAE,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAEpF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;YACvD,MAAM,SAAS,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9E,OAAO,CAAC,GAAG,CAAC,sBAAsB,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YACpD,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO;QACT,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,CAAC;QAChD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACrD,OAAO,CAAC,GAAG,CAAC,mBAAmB,YAAY,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACnE,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEhB,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC/E,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,WAAW,UAAU,CAAC,CAAC,CAAC;YAC3D,OAAO,CAAC,GAAG,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;YAC7C,OAAO,CAAC,GAAG,CAAC,qBAAqB,gBAAgB,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YACzE,OAAO,CAAC,GAAG,CAAC,qBAAqB,YAAY,CAAC,MAAM,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;YACvF,IAAI,MAAM,CAAC,QAAQ,IAAI,CAAC,EAAE,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC;YACjF,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACzF,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,MAAM,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC;YAC7C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,SAAS,GAAG,EAAE,CAAC,CAAC;YAC9B,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* `codeprobe summary [path]` — Quick, compact project summary.
|
|
3
|
+
*
|
|
4
|
+
* Like `git status` but for AI context. Fast, one-screen output
|
|
5
|
+
* with the most important facts about the project.
|
|
6
|
+
*/
|
|
7
|
+
import { Command } from 'commander';
|
|
8
|
+
export declare function registerSummaryCommand(program: Command): void;
|
|
9
|
+
//# sourceMappingURL=summary.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"summary.d.ts","sourceRoot":"","sources":["../../src/commands/summary.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AA4JpC,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CA4I7D"}
|