@pugi/cli 0.1.0-beta.5 → 0.1.0-beta.50
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/THIRD_PARTY_NOTICES.md +40 -0
- package/assets/pugi-mascot.ansi +15 -25
- package/assets/pugi-prozr2-mascot.ansi +9 -0
- package/bin/run.js +33 -1
- package/dist/commands/jobs-watch.js +201 -0
- package/dist/commands/jobs.js +15 -0
- package/dist/commands/smoke.js +133 -0
- package/dist/core/agent-progress/cleanup.js +134 -0
- package/dist/core/agent-progress/schema.js +144 -0
- package/dist/core/agent-progress/writer.js +101 -0
- package/dist/core/artifact-chain/dispatcher.js +148 -0
- package/dist/core/artifact-chain/exporter.js +164 -0
- package/dist/core/artifact-chain/state.js +243 -0
- package/dist/core/artifact-chain/steps.js +169 -0
- package/dist/core/auth/ensure-authenticated.js +129 -0
- package/dist/core/auth/env-provider.js +238 -0
- package/dist/core/auto-update/channels.js +122 -0
- package/dist/core/auto-update/checker.js +241 -0
- package/dist/core/auto-update/state.js +235 -0
- package/dist/core/bare-mode/index.js +107 -0
- package/dist/core/bash-classifier.js +400 -4
- package/dist/core/checkpoint/resumer.js +149 -0
- package/dist/core/checkpoint/rewinder.js +291 -0
- package/dist/core/codegraph/decision-store.js +248 -0
- package/dist/core/codegraph/detect-repo.js +459 -0
- package/dist/core/codegraph/install.js +134 -0
- package/dist/core/codegraph/offer-hook.js +220 -0
- package/dist/core/compact/auto-trigger.js +96 -0
- package/dist/core/compact/buffer-rewriter.js +115 -0
- package/dist/core/compact/summarizer.js +208 -0
- package/dist/core/compact/token-counter.js +108 -0
- package/dist/core/consensus/diff-capture.js +112 -3
- package/dist/core/context/index.js +7 -0
- package/dist/core/context/markdown-traverse.js +255 -0
- package/dist/core/cost/rate-card.js +129 -0
- package/dist/core/cost/tracker.js +221 -0
- package/dist/core/denial-tracking/index.js +8 -0
- package/dist/core/denial-tracking/state.js +264 -0
- package/dist/core/diagnostics/probe-runner.js +93 -0
- package/dist/core/diagnostics/probes/api.js +46 -0
- package/dist/core/diagnostics/probes/auth.js +86 -0
- package/dist/core/diagnostics/probes/bare-mode.js +42 -0
- package/dist/core/diagnostics/probes/cli-version.js +127 -0
- package/dist/core/diagnostics/probes/config.js +72 -0
- package/dist/core/diagnostics/probes/denial-tracking.js +57 -0
- package/dist/core/diagnostics/probes/disk.js +81 -0
- package/dist/core/diagnostics/probes/git.js +65 -0
- package/dist/core/diagnostics/probes/hooks.js +118 -0
- package/dist/core/diagnostics/probes/mcp.js +75 -0
- package/dist/core/diagnostics/probes/node.js +59 -0
- package/dist/core/diagnostics/probes/pnpm.js +36 -0
- package/dist/core/diagnostics/probes/pugi-md.js +89 -0
- package/dist/core/diagnostics/probes/sandbox.js +40 -0
- package/dist/core/diagnostics/probes/session.js +74 -0
- package/dist/core/diagnostics/probes/status-snapshot.js +488 -0
- package/dist/core/diagnostics/probes/workspace.js +63 -0
- package/dist/core/diagnostics/types.js +70 -0
- package/dist/core/dispatch/cache-cleanup.js +197 -0
- package/dist/core/dispatch/cache-handoff.js +295 -0
- package/dist/core/edits/dispatch.js +218 -2
- package/dist/core/edits/journal.js +199 -0
- package/dist/core/edits/layer-d-ast.js +557 -14
- package/dist/core/edits/verify-hook.js +273 -0
- package/dist/core/edits/worktree.js +322 -0
- package/dist/core/engine/anvil-client.js +115 -5
- package/dist/core/engine/budgets.js +98 -0
- package/dist/core/engine/context-prefix.js +155 -0
- package/dist/core/engine/intent.js +260 -0
- package/dist/core/engine/native-pugi.js +860 -211
- package/dist/core/engine/prompts.js +88 -2
- package/dist/core/engine/strip-internal-fields.js +124 -0
- package/dist/core/engine/tool-bridge.js +1045 -36
- package/dist/core/feedback/queue.js +177 -0
- package/dist/core/feedback/submitter.js +145 -0
- package/dist/core/file-cache.js +113 -1
- package/dist/core/hooks/events.js +44 -0
- package/dist/core/hooks/index.js +15 -0
- package/dist/core/hooks/registry.js +213 -0
- package/dist/core/hooks/runner.js +236 -0
- package/dist/core/hooks/v2/event-emitter.js +115 -0
- package/dist/core/hooks/v2/executor.js +282 -0
- package/dist/core/hooks/v2/index.js +25 -0
- package/dist/core/hooks/v2/lifecycle.js +104 -0
- package/dist/core/hooks/v2/loader.js +216 -0
- package/dist/core/hooks/v2/matcher.js +125 -0
- package/dist/core/hooks/v2/trust.js +143 -0
- package/dist/core/hooks/v2/types.js +86 -0
- package/dist/core/lsp/cache.js +105 -0
- package/dist/core/lsp/client.js +776 -0
- package/dist/core/lsp/language-detect.js +66 -0
- package/dist/core/lsp/post-edit-diagnostics.js +171 -0
- package/dist/core/mcp/client.js +75 -6
- package/dist/core/mcp/http-server.js +553 -0
- package/dist/core/mcp/orchestrator-tools.js +662 -0
- package/dist/core/mcp/permission.js +190 -0
- package/dist/core/mcp/registry.js +24 -2
- package/dist/core/mcp/server-tools.js +219 -0
- package/dist/core/mcp/server.js +397 -0
- package/dist/core/memory/dual-write.js +416 -0
- package/dist/core/memory/phase1-kinds.js +20 -0
- package/dist/core/memory-sync/queue.js +158 -0
- package/dist/core/onboarding/ensure-initialized.js +133 -0
- package/dist/core/onboarding/marker.js +111 -0
- package/dist/core/onboarding/telemetry-state.js +108 -0
- package/dist/core/output-style/presets.js +176 -0
- package/dist/core/output-style/state.js +185 -0
- package/dist/core/path-security.js +284 -2
- package/dist/core/permissions/auto-classifier.js +124 -0
- package/dist/core/permissions/circuit-breaker.js +83 -0
- package/dist/core/permissions/gate.js +278 -0
- package/dist/core/permissions/index.js +20 -0
- package/dist/core/permissions/mode.js +174 -0
- package/dist/core/permissions/state.js +241 -0
- package/dist/core/permissions/tool-class.js +93 -0
- package/dist/core/prd-check/parser.js +215 -0
- package/dist/core/prd-check/reporter.js +127 -0
- package/dist/core/prd-check/session-review.js +557 -0
- package/dist/core/prd-check/verifiers.js +223 -0
- package/dist/core/pugi-md/context-injector.js +76 -0
- package/dist/core/pugi-md/walk-up.js +207 -0
- package/dist/core/release-notes/parser.js +241 -0
- package/dist/core/release-notes/state.js +116 -0
- package/dist/core/repl/history.js +11 -1
- package/dist/core/repl/model-pricing.js +135 -0
- package/dist/core/repl/session.js +1897 -37
- package/dist/core/repl/slash-commands.js +430 -15
- package/dist/core/repl/store/session-store.js +31 -2
- package/dist/core/repl/workspace-context.js +22 -0
- package/dist/core/repo-map/build.js +125 -0
- package/dist/core/repo-map/cache.js +185 -0
- package/dist/core/repo-map/extractor.js +254 -0
- package/dist/core/repo-map/formatter.js +145 -0
- package/dist/core/repo-map/scanner.js +211 -0
- package/dist/core/retry-budget/budget.js +284 -0
- package/dist/core/retry-budget/index.js +5 -0
- package/dist/core/session.js +92 -0
- package/dist/core/settings.js +80 -0
- package/dist/core/share/formatter.js +271 -0
- package/dist/core/share/redactor.js +221 -0
- package/dist/core/share/uploader.js +267 -0
- package/dist/core/skills/defaults.js +457 -0
- package/dist/core/smoke/headless-driver.js +174 -0
- package/dist/core/smoke/orchestrator.js +194 -0
- package/dist/core/smoke/runner.js +238 -0
- package/dist/core/smoke/scenario-parser.js +316 -0
- package/dist/core/subagents/dispatcher-real.js +600 -0
- package/dist/core/subagents/dispatcher.js +113 -24
- package/dist/core/subagents/index.js +18 -5
- package/dist/core/subagents/isolation-matrix.js +213 -0
- package/dist/core/subagents/spawn.js +19 -4
- package/dist/core/telemetry/emitter.js +229 -0
- package/dist/core/telemetry/queue.js +251 -0
- package/dist/core/theme/context.js +91 -0
- package/dist/core/theme/presets.js +228 -0
- package/dist/core/theme/state.js +181 -0
- package/dist/core/todos/invariant.js +10 -0
- package/dist/core/todos/state.js +177 -0
- package/dist/core/transport/version-interceptor.js +166 -0
- package/dist/core/vim/keymap.js +288 -0
- package/dist/core/vim/state.js +92 -0
- package/dist/core/worktree-manager/cleanup.js +123 -0
- package/dist/core/worktree-manager/manager.js +303 -0
- package/dist/index.js +28 -0
- package/dist/runtime/bootstrap.js +190 -0
- package/dist/runtime/cli.js +3241 -343
- package/dist/runtime/commands/cancel.js +231 -0
- package/dist/runtime/commands/chain.js +489 -0
- package/dist/runtime/commands/codegraph-status.js +227 -0
- package/dist/runtime/commands/compact.js +297 -0
- package/dist/runtime/commands/cost.js +199 -0
- package/dist/runtime/commands/delegate.js +242 -11
- package/dist/runtime/commands/dispatch.js +126 -0
- package/dist/runtime/commands/doctor.js +412 -0
- package/dist/runtime/commands/feedback.js +184 -0
- package/dist/runtime/commands/hooks.js +184 -0
- package/dist/runtime/commands/lsp.js +368 -0
- package/dist/runtime/commands/mcp.js +879 -0
- package/dist/runtime/commands/memory.js +508 -0
- package/dist/runtime/commands/model.js +237 -0
- package/dist/runtime/commands/onboarding.js +275 -0
- package/dist/runtime/commands/patch.js +128 -0
- package/dist/runtime/commands/permissions.js +112 -0
- package/dist/runtime/commands/plan.js +143 -0
- package/dist/runtime/commands/prd-check.js +285 -0
- package/dist/runtime/commands/redo-blob-store.js +92 -0
- package/dist/runtime/commands/redo.js +361 -0
- package/dist/runtime/commands/release-notes.js +229 -0
- package/dist/runtime/commands/repo-map.js +95 -0
- package/dist/runtime/commands/report.js +299 -0
- package/dist/runtime/commands/resume.js +118 -0
- package/dist/runtime/commands/review-consensus.js +17 -2
- package/dist/runtime/commands/rewind.js +333 -0
- package/dist/runtime/commands/sessions.js +163 -0
- package/dist/runtime/commands/share.js +316 -0
- package/dist/runtime/commands/status.js +186 -0
- package/dist/runtime/commands/stickers.js +82 -0
- package/dist/runtime/commands/style.js +194 -0
- package/dist/runtime/commands/theme.js +196 -0
- package/dist/runtime/commands/undo.js +32 -0
- package/dist/runtime/commands/update.js +289 -0
- package/dist/runtime/commands/vim.js +140 -0
- package/dist/runtime/commands/worktree.js +177 -0
- package/dist/runtime/commands/worktrees.js +155 -0
- package/dist/runtime/headless-repl.js +195 -0
- package/dist/runtime/headless.js +543 -0
- package/dist/runtime/load-hooks-or-exit.js +71 -0
- package/dist/runtime/plan-decompose.js +531 -0
- package/dist/runtime/version.js +65 -0
- package/dist/tools/agent-tool.js +229 -0
- package/dist/tools/apply-patch.js +556 -0
- package/dist/tools/ask-user-question.js +213 -0
- package/dist/tools/ask-user.js +115 -0
- package/dist/tools/bash.js +203 -4
- package/dist/tools/file-tools.js +85 -14
- package/dist/tools/lsp-tools.js +189 -0
- package/dist/tools/mcp-tool.js +260 -0
- package/dist/tools/multi-edit.js +361 -0
- package/dist/tools/powershell.js +268 -0
- package/dist/tools/registry.js +51 -0
- package/dist/tools/skill-tool.js +96 -0
- package/dist/tools/tasks.js +208 -0
- package/dist/tools/todo-write.js +184 -0
- package/dist/tools/web-fetch.js +147 -2
- package/dist/tools/web-search.js +458 -0
- package/dist/tui/agent-progress-card.js +111 -0
- package/dist/tui/agent-tree.js +10 -0
- package/dist/tui/ask-modal.js +2 -2
- package/dist/tui/ask-user-question-prompt.js +192 -0
- package/dist/tui/compact-banner.js +81 -0
- package/dist/tui/conversation-pane.js +82 -8
- package/dist/tui/cost-table.js +111 -0
- package/dist/tui/doctor-table.js +46 -0
- package/dist/tui/feedback-prompt.js +156 -0
- package/dist/tui/input-box.js +218 -3
- package/dist/tui/markdown-render.js +4 -4
- package/dist/tui/onboarding-wizard.js +240 -0
- package/dist/tui/permissions-picker.js +86 -0
- package/dist/tui/render.js +35 -0
- package/dist/tui/repl-render.js +313 -35
- package/dist/tui/repl-splash-art.js +1 -1
- package/dist/tui/repl-splash-mascot.js +32 -8
- package/dist/tui/repl-splash.js +2 -2
- package/dist/tui/repl.js +85 -5
- package/dist/tui/splash.js +1 -1
- package/dist/tui/status-bar.js +94 -16
- package/dist/tui/status-table.js +7 -0
- package/dist/tui/stickers-art.js +136 -0
- package/dist/tui/style-table.js +28 -0
- package/dist/tui/theme-table.js +29 -0
- package/dist/tui/thinking-spinner.js +123 -0
- package/dist/tui/tool-stream-pane.js +52 -3
- package/dist/tui/update-banner.js +27 -2
- package/dist/tui/vim-input.js +267 -0
- package/dist/tui/welcome-banner.js +107 -0
- package/dist/tui/welcome-data.js +293 -0
- package/docs/examples/codegraph.mcp.json +10 -0
- package/package.json +12 -6
- package/test/scenarios/codegen-create-file.scenario.txt +13 -0
- package/test/scenarios/compact-force.scenario.txt +11 -0
- package/test/scenarios/identity.scenario.txt +11 -0
- package/test/scenarios/persona-handoff.scenario.txt +11 -0
- package/test/scenarios/walkback.scenario.txt +12 -0
- package/dist/core/engine/compaction-hook.js +0 -154
|
@@ -0,0 +1,221 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Persisted per-session cost tracker — L19 sprint (2026-05-27).
|
|
3
|
+
*
|
|
4
|
+
* Mission: every Anvil-mediated LLM call goes through `recordCall`, which
|
|
5
|
+
* aggregates per-model token + USD totals and atomically persists them
|
|
6
|
+
* to `.pugi/cost.json` so the operator can read `/cost` across REPL
|
|
7
|
+
* restarts and reconcile a 14-min session that crossed a process boundary.
|
|
8
|
+
*
|
|
9
|
+
* Why a fresh module instead of bolting onto `core/repl/session.ts`?
|
|
10
|
+
*
|
|
11
|
+
* - `session.ts` accumulates in-memory state for the live TUI status
|
|
12
|
+
* row, which is by-design ephemeral and cleared on REPL boot. The
|
|
13
|
+
* operator's "what did I spend across the project?" question needs
|
|
14
|
+
* a durable surface that survives a process restart.
|
|
15
|
+
* - L19 also has to read `--all-sessions` (last 30 days). The natural
|
|
16
|
+
* store for that is a per-workspace history of session aggregates,
|
|
17
|
+
* which is easy with the JSON file pattern below and would be
|
|
18
|
+
* awkward stitched into the REPL reducer.
|
|
19
|
+
*
|
|
20
|
+
* On-disk shape (single JSON file, atomic tmp+rename writes):
|
|
21
|
+
*
|
|
22
|
+
* {
|
|
23
|
+
* "version": 1,
|
|
24
|
+
* "current": { sessionId, startedAt, models: { <slug>: ModelEntry } },
|
|
25
|
+
* "history": [
|
|
26
|
+
* { sessionId, startedAt, endedAt, models: { ... } }
|
|
27
|
+
* ]
|
|
28
|
+
* }
|
|
29
|
+
*
|
|
30
|
+
* History rotation: when `recordCall` is invoked with a sessionId
|
|
31
|
+
* different from `current.sessionId`, the existing `current` row is
|
|
32
|
+
* stamped with `endedAt = now()` and pushed onto `history`, then a new
|
|
33
|
+
* `current` row is initialised. History is capped at 90 entries (the L19
|
|
34
|
+
* `--all-sessions` window is 30 days; 90 gives a generous buffer for
|
|
35
|
+
* operators on >1 session/day cadence without unbounded growth).
|
|
36
|
+
*
|
|
37
|
+
* The tracker is workspace-scoped — every workspace has its own
|
|
38
|
+
* `.pugi/cost.json`. This matches the existing `.pugi/events.jsonl` /
|
|
39
|
+
* `.pugi/index.json` pattern and means a multi-repo operator's costs
|
|
40
|
+
* are billed against the repo they were incurred in.
|
|
41
|
+
*/
|
|
42
|
+
import { existsSync, mkdirSync, readFileSync, renameSync, writeFileSync, unlinkSync } from 'node:fs';
|
|
43
|
+
import { dirname, resolve } from 'node:path';
|
|
44
|
+
import { estimateUsd } from './rate-card.js';
|
|
45
|
+
/** On-disk schema version. Bump if the file shape changes. */
|
|
46
|
+
export const COST_FILE_SCHEMA_VERSION = 1;
|
|
47
|
+
/** Maximum number of historical sessions persisted in `.pugi/cost.json`. */
|
|
48
|
+
export const COST_HISTORY_CAP = 90;
|
|
49
|
+
export function createCostTracker(opts) {
|
|
50
|
+
const filePath = resolve(opts.workspaceRoot, '.pugi/cost.json');
|
|
51
|
+
const now = opts.now ?? Date.now;
|
|
52
|
+
let state = readOrInit(filePath);
|
|
53
|
+
function ensureCurrent(sessionId) {
|
|
54
|
+
if (state.current && state.current.sessionId === sessionId) {
|
|
55
|
+
return state.current;
|
|
56
|
+
}
|
|
57
|
+
// Session rotation: stamp the previous current with endedAt and push
|
|
58
|
+
// onto history. Idempotent — calling rotate twice with the same
|
|
59
|
+
// session id is a no-op.
|
|
60
|
+
if (state.current) {
|
|
61
|
+
const ended = {
|
|
62
|
+
...state.current,
|
|
63
|
+
endedAt: new Date(now()).toISOString(),
|
|
64
|
+
};
|
|
65
|
+
state.history = [ended, ...state.history].slice(0, COST_HISTORY_CAP);
|
|
66
|
+
}
|
|
67
|
+
state.current = {
|
|
68
|
+
sessionId,
|
|
69
|
+
startedAt: new Date(now()).toISOString(),
|
|
70
|
+
models: {},
|
|
71
|
+
};
|
|
72
|
+
return state.current;
|
|
73
|
+
}
|
|
74
|
+
function persist() {
|
|
75
|
+
try {
|
|
76
|
+
mkdirSync(dirname(filePath), { recursive: true });
|
|
77
|
+
}
|
|
78
|
+
catch {
|
|
79
|
+
// best-effort directory create; the write below surfaces the real
|
|
80
|
+
// error if the parent is genuinely unwritable
|
|
81
|
+
}
|
|
82
|
+
const tmp = `${filePath}.tmp`;
|
|
83
|
+
writeFileSync(tmp, JSON.stringify(state, null, 2), 'utf8');
|
|
84
|
+
renameSync(tmp, filePath);
|
|
85
|
+
}
|
|
86
|
+
return {
|
|
87
|
+
recordCall(input) {
|
|
88
|
+
const sessionId = opts.sessionIdProvider();
|
|
89
|
+
if (!sessionId)
|
|
90
|
+
return;
|
|
91
|
+
const current = ensureCurrent(sessionId);
|
|
92
|
+
const slug = typeof input.model === 'string' && input.model.length > 0 ? input.model : 'unknown';
|
|
93
|
+
const safeIn = Number.isFinite(input.inputTokens) && input.inputTokens > 0 ? input.inputTokens : 0;
|
|
94
|
+
const safeOut = Number.isFinite(input.outputTokens) && input.outputTokens > 0 ? input.outputTokens : 0;
|
|
95
|
+
const existing = current.models[slug] ?? { input: 0, output: 0, callCount: 0 };
|
|
96
|
+
current.models[slug] = {
|
|
97
|
+
input: existing.input + safeIn,
|
|
98
|
+
output: existing.output + safeOut,
|
|
99
|
+
callCount: existing.callCount + 1,
|
|
100
|
+
};
|
|
101
|
+
persist();
|
|
102
|
+
},
|
|
103
|
+
current() {
|
|
104
|
+
return state.current;
|
|
105
|
+
},
|
|
106
|
+
history() {
|
|
107
|
+
return state.history;
|
|
108
|
+
},
|
|
109
|
+
aggregateWithin(withinDays) {
|
|
110
|
+
const cutoffMs = now() - withinDays * 24 * 60 * 60 * 1000;
|
|
111
|
+
const aggregate = {
|
|
112
|
+
sessionId: 'aggregate',
|
|
113
|
+
startedAt: new Date(cutoffMs).toISOString(),
|
|
114
|
+
endedAt: new Date(now()).toISOString(),
|
|
115
|
+
models: {},
|
|
116
|
+
};
|
|
117
|
+
const rows = [];
|
|
118
|
+
if (state.current)
|
|
119
|
+
rows.push(state.current);
|
|
120
|
+
for (const row of state.history) {
|
|
121
|
+
const stamp = Date.parse(row.startedAt);
|
|
122
|
+
if (Number.isFinite(stamp) && stamp >= cutoffMs)
|
|
123
|
+
rows.push(row);
|
|
124
|
+
}
|
|
125
|
+
for (const row of rows) {
|
|
126
|
+
for (const [slug, entry] of Object.entries(row.models)) {
|
|
127
|
+
const existing = aggregate.models[slug] ?? { input: 0, output: 0, callCount: 0 };
|
|
128
|
+
aggregate.models[slug] = {
|
|
129
|
+
input: existing.input + entry.input,
|
|
130
|
+
output: existing.output + entry.output,
|
|
131
|
+
callCount: existing.callCount + entry.callCount,
|
|
132
|
+
};
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
return aggregate;
|
|
136
|
+
},
|
|
137
|
+
resetCurrent() {
|
|
138
|
+
const wiped = state.current;
|
|
139
|
+
state.current = null;
|
|
140
|
+
persist();
|
|
141
|
+
return wiped;
|
|
142
|
+
},
|
|
143
|
+
flush() {
|
|
144
|
+
persist();
|
|
145
|
+
},
|
|
146
|
+
};
|
|
147
|
+
}
|
|
148
|
+
/**
|
|
149
|
+
* Compute the per-session USD total from a `SessionAggregate`. Pure —
|
|
150
|
+
* uses the rate card to bind a price to every model entry. Open-weight
|
|
151
|
+
* models contribute $0 (their entries always have $0/$0 rate).
|
|
152
|
+
*/
|
|
153
|
+
export function totalUsd(aggregate) {
|
|
154
|
+
let total = 0;
|
|
155
|
+
for (const [slug, entry] of Object.entries(aggregate.models)) {
|
|
156
|
+
total += estimateUsd(slug, entry.input, entry.output);
|
|
157
|
+
}
|
|
158
|
+
return total;
|
|
159
|
+
}
|
|
160
|
+
/**
|
|
161
|
+
* Compute total input + output token sums across all models in an
|
|
162
|
+
* aggregate. Used by the CLI table footer.
|
|
163
|
+
*/
|
|
164
|
+
export function totalTokens(aggregate) {
|
|
165
|
+
let input = 0;
|
|
166
|
+
let output = 0;
|
|
167
|
+
for (const entry of Object.values(aggregate.models)) {
|
|
168
|
+
input += entry.input;
|
|
169
|
+
output += entry.output;
|
|
170
|
+
}
|
|
171
|
+
return { input, output };
|
|
172
|
+
}
|
|
173
|
+
/**
|
|
174
|
+
* Read the persisted file (or initialise an empty one). Tolerates a
|
|
175
|
+
* corrupted file by returning a fresh empty state — losing one
|
|
176
|
+
* session's history is preferable to throwing from the boot path of
|
|
177
|
+
* every `pugi cost` invocation.
|
|
178
|
+
*/
|
|
179
|
+
function readOrInit(filePath) {
|
|
180
|
+
if (!existsSync(filePath)) {
|
|
181
|
+
return { version: COST_FILE_SCHEMA_VERSION, current: null, history: [] };
|
|
182
|
+
}
|
|
183
|
+
try {
|
|
184
|
+
const raw = readFileSync(filePath, 'utf8');
|
|
185
|
+
const parsed = JSON.parse(raw);
|
|
186
|
+
if (!parsed || typeof parsed !== 'object' || Array.isArray(parsed)) {
|
|
187
|
+
return { version: COST_FILE_SCHEMA_VERSION, current: null, history: [] };
|
|
188
|
+
}
|
|
189
|
+
const obj = parsed;
|
|
190
|
+
return {
|
|
191
|
+
version: typeof obj.version === 'number' ? obj.version : COST_FILE_SCHEMA_VERSION,
|
|
192
|
+
current: isAggregate(obj.current) ? obj.current : null,
|
|
193
|
+
history: Array.isArray(obj.history) ? obj.history.filter(isAggregate) : [],
|
|
194
|
+
};
|
|
195
|
+
}
|
|
196
|
+
catch {
|
|
197
|
+
return { version: COST_FILE_SCHEMA_VERSION, current: null, history: [] };
|
|
198
|
+
}
|
|
199
|
+
}
|
|
200
|
+
function isAggregate(v) {
|
|
201
|
+
if (!v || typeof v !== 'object' || Array.isArray(v))
|
|
202
|
+
return false;
|
|
203
|
+
const obj = v;
|
|
204
|
+
if (typeof obj.sessionId !== 'string' || typeof obj.startedAt !== 'string')
|
|
205
|
+
return false;
|
|
206
|
+
if (!obj.models || typeof obj.models !== 'object')
|
|
207
|
+
return false;
|
|
208
|
+
return true;
|
|
209
|
+
}
|
|
210
|
+
/**
|
|
211
|
+
* Test helper — wipe the `.pugi/cost.json` file. Not exported through the
|
|
212
|
+
* public CostTracker surface because production code must never call
|
|
213
|
+
* this; an operator-facing reset goes through `resetCurrent()` which
|
|
214
|
+
* preserves history.
|
|
215
|
+
*/
|
|
216
|
+
export function _danger_wipeCostFile_forTests(workspaceRoot) {
|
|
217
|
+
const filePath = resolve(workspaceRoot, '.pugi/cost.json');
|
|
218
|
+
if (existsSync(filePath))
|
|
219
|
+
unlinkSync(filePath);
|
|
220
|
+
}
|
|
221
|
+
//# sourceMappingURL=tracker.js.map
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Public re-exports for the denial-tracking surface. Engine adapter
|
|
3
|
+
* code imports from here so we can move internals around (e.g. split
|
|
4
|
+
* the diagnostics probe into a sibling file) without churning every
|
|
5
|
+
* call site.
|
|
6
|
+
*/
|
|
7
|
+
export { DenialTrackingState, buildDenialContext, canonicalArgHash, DENIAL_TRACKING_MAX_ENTRIES, DENIAL_REMINDER_THRESHOLD, DENIAL_ARGS_SUMMARY_BYTES, } from './state.js';
|
|
8
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1,264 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* α7 L11 (2026-05-27) — DenialTrackingState surface (Claude Code parity).
|
|
3
|
+
*
|
|
4
|
+
* Per the openclaude / anarchic-CC leak research (`gap-analysis-3-repos`
|
|
5
|
+
* §5.2): Claude Code's `QueryEngine.ts` maintains a per-session
|
|
6
|
+
* `DenialTrackingState` that records every tool-dispatch denial.
|
|
7
|
+
* Subsequent turns receive a compact reminder so the model does not
|
|
8
|
+
* loop on the same refused operation, the operator can audit pressure
|
|
9
|
+
* points via `/permissions denials` / `/doctor`, and hooks can react
|
|
10
|
+
* to recurring patterns.
|
|
11
|
+
*
|
|
12
|
+
* Pugi pre-L11 surfaced denials only as a one-line sentinel inside the
|
|
13
|
+
* tool error (`HOOK_BLOCKED:`, `PLAN_MODE_REFUSED:`, `STALE_READ:`).
|
|
14
|
+
* The model saw each refusal in isolation; no aggregate, no count, no
|
|
15
|
+
* "do not retry" context across turns. This module closes that gap.
|
|
16
|
+
*
|
|
17
|
+
* Design contract:
|
|
18
|
+
*
|
|
19
|
+
* - Per-session in-memory state. The engine adapter creates one
|
|
20
|
+
* instance at session open and threads it through `buildExecutor`
|
|
21
|
+
* + `runEngineLoop`. No disk persistence — denial history is a
|
|
22
|
+
* turn-scoped signal, not a forensic log (the audit ledger at
|
|
23
|
+
* `.pugi/events.jsonl` already carries every refused dispatch).
|
|
24
|
+
*
|
|
25
|
+
* - Pure: no I/O, no logging. Inputs are passed by caller, output
|
|
26
|
+
* is in-memory only. The diagnostics probe + system-reminder
|
|
27
|
+
* splice are pure functions over the snapshot.
|
|
28
|
+
*
|
|
29
|
+
* - Canonical argHash: sha256 of `canonicalize(args)`. Same call
|
|
30
|
+
* to same tool with same args = same key. Reordering JSON object
|
|
31
|
+
* keys does NOT change the hash (we sort keys recursively before
|
|
32
|
+
* stringification). This makes "repeated denial" detection
|
|
33
|
+
* deterministic across model retries.
|
|
34
|
+
*
|
|
35
|
+
* - Bounded: the map caps at 256 entries per session. Once full,
|
|
36
|
+
* `recordDenial` evicts the oldest entry (lowest `lastDeniedAt`).
|
|
37
|
+
* A session with > 256 distinct denials is already pathological;
|
|
38
|
+
* the cap is defensive insurance against pathological loops
|
|
39
|
+
* blowing memory.
|
|
40
|
+
*
|
|
41
|
+
* - Reminder threshold: the reminder splice only injects when at
|
|
42
|
+
* least one record has `count >= 2`. One-off denials are normal
|
|
43
|
+
* (the operator is shaping the session); repeated denials = the
|
|
44
|
+
* model is failing to learn from the refusal and we should
|
|
45
|
+
* reinforce it.
|
|
46
|
+
*/
|
|
47
|
+
import { createHash } from 'node:crypto';
|
|
48
|
+
/** Bound on the per-session map. See module header. */
|
|
49
|
+
export const DENIAL_TRACKING_MAX_ENTRIES = 256;
|
|
50
|
+
/** Threshold above which `buildDenialContext` injects a system reminder. */
|
|
51
|
+
export const DENIAL_REMINDER_THRESHOLD = 2;
|
|
52
|
+
/** Per-denial-args summary cap surfaced to the model. */
|
|
53
|
+
export const DENIAL_ARGS_SUMMARY_BYTES = 240;
|
|
54
|
+
/**
|
|
55
|
+
* Per-session denial tracker. Instances are mutable but threaded
|
|
56
|
+
* through `buildExecutor` so every dispatcher in the loop sees the
|
|
57
|
+
* same view.
|
|
58
|
+
*/
|
|
59
|
+
export class DenialTrackingState {
|
|
60
|
+
/** Insertion-ordered map. Eviction walks oldest `lastDeniedAt`. */
|
|
61
|
+
records = new Map();
|
|
62
|
+
/**
|
|
63
|
+
* Wall clock. Defaults to `Date.now`. Injected so tests can drive
|
|
64
|
+
* deterministic timestamps.
|
|
65
|
+
*/
|
|
66
|
+
clock;
|
|
67
|
+
constructor(options = {}) {
|
|
68
|
+
this.clock = options.clock ?? Date.now;
|
|
69
|
+
}
|
|
70
|
+
/**
|
|
71
|
+
* Record a denial. Increments the count when the (tool, args) pair
|
|
72
|
+
* has been denied before; otherwise inserts a new record. The reason
|
|
73
|
+
* is always overwritten with the latest value so the system reminder
|
|
74
|
+
* surfaces the most-recent context.
|
|
75
|
+
*/
|
|
76
|
+
recordDenial(toolName, args, reason) {
|
|
77
|
+
const argHash = canonicalArgHash(args);
|
|
78
|
+
const key = `${toolName}:${argHash}`;
|
|
79
|
+
const now = new Date(this.clock());
|
|
80
|
+
const existing = this.records.get(key);
|
|
81
|
+
if (existing) {
|
|
82
|
+
existing.count += 1;
|
|
83
|
+
existing.lastDeniedAt = now;
|
|
84
|
+
existing.lastReason = truncateReason(reason);
|
|
85
|
+
// Re-insert so the iteration order tracks lastDeniedAt (LRU-ish).
|
|
86
|
+
this.records.delete(key);
|
|
87
|
+
this.records.set(key, existing);
|
|
88
|
+
return existing;
|
|
89
|
+
}
|
|
90
|
+
if (this.records.size >= DENIAL_TRACKING_MAX_ENTRIES) {
|
|
91
|
+
// Evict the oldest entry (insertion order = LRU after our delete
|
|
92
|
+
// + set pattern above). Map iteration order is insertion order
|
|
93
|
+
// in JS; the first key is the LRU one.
|
|
94
|
+
const oldestKey = this.records.keys().next().value;
|
|
95
|
+
if (oldestKey !== undefined) {
|
|
96
|
+
this.records.delete(oldestKey);
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
const record = {
|
|
100
|
+
key,
|
|
101
|
+
toolName,
|
|
102
|
+
argsSummary: summariseArgs(args),
|
|
103
|
+
firstDeniedAt: now,
|
|
104
|
+
lastDeniedAt: now,
|
|
105
|
+
count: 1,
|
|
106
|
+
lastReason: truncateReason(reason),
|
|
107
|
+
};
|
|
108
|
+
this.records.set(key, record);
|
|
109
|
+
return record;
|
|
110
|
+
}
|
|
111
|
+
/**
|
|
112
|
+
* Snapshot of every recorded denial. Returned in insertion order
|
|
113
|
+
* (LRU-ish: most-recently-touched at the tail). The caller MUST
|
|
114
|
+
* treat the array as read-only — mutating it does not affect state.
|
|
115
|
+
*/
|
|
116
|
+
getDenialHistory() {
|
|
117
|
+
return Array.from(this.records.values()).map((r) => ({ ...r }));
|
|
118
|
+
}
|
|
119
|
+
/** Lookup the record for a specific (tool, args) pair, or undefined. */
|
|
120
|
+
getPatternFor(toolName, args) {
|
|
121
|
+
const argHash = canonicalArgHash(args);
|
|
122
|
+
const key = `${toolName}:${argHash}`;
|
|
123
|
+
const record = this.records.get(key);
|
|
124
|
+
return record ? { ...record } : undefined;
|
|
125
|
+
}
|
|
126
|
+
/**
|
|
127
|
+
* Aggregate counts — handy for `/doctor`, `/permissions denials`,
|
|
128
|
+
* and the system-reminder threshold check.
|
|
129
|
+
*/
|
|
130
|
+
summary() {
|
|
131
|
+
let totalDenials = 0;
|
|
132
|
+
let repeatedPatterns = 0;
|
|
133
|
+
for (const record of this.records.values()) {
|
|
134
|
+
totalDenials += record.count;
|
|
135
|
+
if (record.count >= DENIAL_REMINDER_THRESHOLD)
|
|
136
|
+
repeatedPatterns += 1;
|
|
137
|
+
}
|
|
138
|
+
return {
|
|
139
|
+
totalDenials,
|
|
140
|
+
uniquePatterns: this.records.size,
|
|
141
|
+
repeatedPatterns,
|
|
142
|
+
};
|
|
143
|
+
}
|
|
144
|
+
/** Drop every recorded denial. Called on session end. */
|
|
145
|
+
clear() {
|
|
146
|
+
this.records.clear();
|
|
147
|
+
}
|
|
148
|
+
/** Number of distinct (tool, args) pairs currently tracked. */
|
|
149
|
+
size() {
|
|
150
|
+
return this.records.size;
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
/**
|
|
154
|
+
* Build the operator-facing system-reminder block. Returns an empty
|
|
155
|
+
* string when no record meets the threshold — the caller short-circuits
|
|
156
|
+
* the splice in that case.
|
|
157
|
+
*
|
|
158
|
+
* Rendered shape:
|
|
159
|
+
*
|
|
160
|
+
* <denial-context>
|
|
161
|
+
* - Tool `bash` denied 3 times in this session.
|
|
162
|
+
* Last reason: HOOK_BLOCKED: PreToolUse refused (exit=1).
|
|
163
|
+
* Args: {"command":"rm -rf node_modules"}
|
|
164
|
+
* Do not retry without operator intervention.
|
|
165
|
+
* - Tool `edit` denied 2 times in this session.
|
|
166
|
+
* Last reason: STALE_READ: file changed since last read.
|
|
167
|
+
* Args: {"path":"src/index.ts","oldString":"..."}
|
|
168
|
+
* Do not retry without operator intervention.
|
|
169
|
+
* </denial-context>
|
|
170
|
+
*
|
|
171
|
+
* Determinism: same snapshot always produces the same block. We sort
|
|
172
|
+
* entries by `count` descending then `lastDeniedAt` descending so the
|
|
173
|
+
* most-pressing patterns surface first.
|
|
174
|
+
*/
|
|
175
|
+
export function buildDenialContext(state) {
|
|
176
|
+
const records = state.getDenialHistory()
|
|
177
|
+
.filter((r) => r.count >= DENIAL_REMINDER_THRESHOLD)
|
|
178
|
+
.sort((a, b) => {
|
|
179
|
+
if (b.count !== a.count)
|
|
180
|
+
return b.count - a.count;
|
|
181
|
+
return b.lastDeniedAt.getTime() - a.lastDeniedAt.getTime();
|
|
182
|
+
});
|
|
183
|
+
if (records.length === 0)
|
|
184
|
+
return '';
|
|
185
|
+
const lines = ['<denial-context>'];
|
|
186
|
+
for (const record of records) {
|
|
187
|
+
lines.push(` - Tool \`${record.toolName}\` denied ${record.count} times in this session.`);
|
|
188
|
+
lines.push(` Last reason: ${record.lastReason}`);
|
|
189
|
+
lines.push(` Args: ${record.argsSummary}`);
|
|
190
|
+
lines.push(' Do not retry without operator intervention.');
|
|
191
|
+
}
|
|
192
|
+
lines.push('</denial-context>');
|
|
193
|
+
return lines.join('\n');
|
|
194
|
+
}
|
|
195
|
+
/**
|
|
196
|
+
* Canonical sha256 of the args. Object keys are sorted recursively
|
|
197
|
+
* before stringification so `{a:1,b:2}` and `{b:2,a:1}` hash to the
|
|
198
|
+
* same value. Undefined values + functions are stripped (JSON.stringify
|
|
199
|
+
* already does this); circular refs throw — the caller is expected to
|
|
200
|
+
* pass plain JSON objects (every Pugi tool argument set is a JSON
|
|
201
|
+
* object by contract).
|
|
202
|
+
*
|
|
203
|
+
* Returns the full 64-char hex digest. The denial key prepends the
|
|
204
|
+
* tool name (`<tool>:<hash>`) so a hash collision across tools cannot
|
|
205
|
+
* confuse the matcher.
|
|
206
|
+
*/
|
|
207
|
+
export function canonicalArgHash(args) {
|
|
208
|
+
const canonical = canonicalize(args);
|
|
209
|
+
const hash = createHash('sha256');
|
|
210
|
+
hash.update(canonical);
|
|
211
|
+
return hash.digest('hex');
|
|
212
|
+
}
|
|
213
|
+
function canonicalize(value) {
|
|
214
|
+
if (value === null)
|
|
215
|
+
return 'null';
|
|
216
|
+
if (value === undefined)
|
|
217
|
+
return 'null';
|
|
218
|
+
const t = typeof value;
|
|
219
|
+
if (t === 'string')
|
|
220
|
+
return JSON.stringify(value);
|
|
221
|
+
if (t === 'number' || t === 'boolean')
|
|
222
|
+
return JSON.stringify(value);
|
|
223
|
+
if (t === 'bigint')
|
|
224
|
+
return JSON.stringify(value.toString());
|
|
225
|
+
if (Array.isArray(value)) {
|
|
226
|
+
return `[${value.map((entry) => canonicalize(entry)).join(',')}]`;
|
|
227
|
+
}
|
|
228
|
+
if (t === 'object') {
|
|
229
|
+
const obj = value;
|
|
230
|
+
const keys = Object.keys(obj).sort();
|
|
231
|
+
const parts = keys.map((k) => `${JSON.stringify(k)}:${canonicalize(obj[k])}`);
|
|
232
|
+
return `{${parts.join(',')}}`;
|
|
233
|
+
}
|
|
234
|
+
// Functions / symbols — neither should reach here for tool args.
|
|
235
|
+
// Stringify defensively so the hash is stable.
|
|
236
|
+
return JSON.stringify(String(value));
|
|
237
|
+
}
|
|
238
|
+
/**
|
|
239
|
+
* Short, operator-readable preview of the args object. Cap so a
|
|
240
|
+
* 32 KB write payload does not blow up the reminder block. Falls back
|
|
241
|
+
* к `[non-JSON args]` when stringification fails (e.g. a circular ref
|
|
242
|
+
* snuck through — unlikely for tool args by contract).
|
|
243
|
+
*/
|
|
244
|
+
function summariseArgs(args) {
|
|
245
|
+
let stringified;
|
|
246
|
+
try {
|
|
247
|
+
stringified = JSON.stringify(args);
|
|
248
|
+
}
|
|
249
|
+
catch {
|
|
250
|
+
return '[non-JSON args]';
|
|
251
|
+
}
|
|
252
|
+
if (stringified === undefined)
|
|
253
|
+
return '{}';
|
|
254
|
+
if (stringified.length <= DENIAL_ARGS_SUMMARY_BYTES)
|
|
255
|
+
return stringified;
|
|
256
|
+
return `${stringified.slice(0, DENIAL_ARGS_SUMMARY_BYTES)}…`;
|
|
257
|
+
}
|
|
258
|
+
/** Cap reason at a sane length so a hook script's stdout cannot flood the reminder. */
|
|
259
|
+
function truncateReason(reason, cap = 240) {
|
|
260
|
+
if (reason.length <= cap)
|
|
261
|
+
return reason;
|
|
262
|
+
return `${reason.slice(0, cap)}…`;
|
|
263
|
+
}
|
|
264
|
+
//# sourceMappingURL=state.js.map
|
|
@@ -0,0 +1,93 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Probe runner — orchestrates a set of diagnostic probes in parallel
|
|
3
|
+
* with per-probe fail-isolation + a global wall-clock budget.
|
|
4
|
+
*
|
|
5
|
+
* Design contract:
|
|
6
|
+
*
|
|
7
|
+
* - Probes are independent: one probe's throw or timeout NEVER stops
|
|
8
|
+
* the others. The runner wraps each call in a try/catch + a
|
|
9
|
+
* `Promise.race` against a timeout sentinel.
|
|
10
|
+
*
|
|
11
|
+
* - Order preservation: the returned `probes[]` array preserves the
|
|
12
|
+
* input order so the table renderer always lists rows in the same
|
|
13
|
+
* sequence (operators rely on muscle memory: "auth is the first
|
|
14
|
+
* row, api is the second").
|
|
15
|
+
*
|
|
16
|
+
* - No I/O ownership: the runner owns ZERO file or network calls.
|
|
17
|
+
* Every external dependency is injected via the probe function
|
|
18
|
+
* itself. This keeps the test surface minimal and verifies the
|
|
19
|
+
* fail-isolation contract without spinning real subprocesses.
|
|
20
|
+
*
|
|
21
|
+
* - Crashes attributed to the probe: a probe that throws maps to a
|
|
22
|
+
* synthetic `error` ProbeResult with the probe name + the error
|
|
23
|
+
* message. A probe that exceeds the timeout becomes a synthetic
|
|
24
|
+
* `warn` (timing out is asymmetric — the host is reachable but
|
|
25
|
+
* slow).
|
|
26
|
+
*
|
|
27
|
+
* - The aggregate `DoctorReport` shape comes straight from
|
|
28
|
+
* `types.ts` so the doctor command + the Ink table renderer both
|
|
29
|
+
* consume the same struct without any glue layer.
|
|
30
|
+
*/
|
|
31
|
+
import { computeOverall, countProbes, } from './types.js';
|
|
32
|
+
/**
|
|
33
|
+
* Run a set of probes in parallel and produce a structured report.
|
|
34
|
+
* Never throws — every failure mode maps to a structured ProbeResult.
|
|
35
|
+
*/
|
|
36
|
+
export async function runProbes(probes, options = {}) {
|
|
37
|
+
const now = options.now ?? Date.now;
|
|
38
|
+
const defaultTimeoutMs = options.defaultTimeoutMs ?? 5_000;
|
|
39
|
+
const startedAt = now();
|
|
40
|
+
// Each probe is wrapped in a fail-isolation envelope.
|
|
41
|
+
const results = await Promise.all(probes.map(async (entry) => runOne(entry, defaultTimeoutMs)));
|
|
42
|
+
const overall = computeOverall(results);
|
|
43
|
+
const counts = countProbes(results);
|
|
44
|
+
const durationMs = now() - startedAt;
|
|
45
|
+
return {
|
|
46
|
+
probes: results,
|
|
47
|
+
overall,
|
|
48
|
+
counts,
|
|
49
|
+
durationMs,
|
|
50
|
+
};
|
|
51
|
+
}
|
|
52
|
+
async function runOne(entry, defaultTimeoutMs) {
|
|
53
|
+
const budget = entry.timeoutMs ?? defaultTimeoutMs;
|
|
54
|
+
try {
|
|
55
|
+
return await raceWithTimeout(entry, budget);
|
|
56
|
+
}
|
|
57
|
+
catch (error) {
|
|
58
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
59
|
+
if (error instanceof ProbeTimeoutError) {
|
|
60
|
+
return {
|
|
61
|
+
name: entry.name,
|
|
62
|
+
status: 'warn',
|
|
63
|
+
detail: `Probe timed out after ${budget}ms`,
|
|
64
|
+
remediation: 'Re-run later; the host or registry may be slow',
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
return {
|
|
68
|
+
name: entry.name,
|
|
69
|
+
status: 'error',
|
|
70
|
+
detail: `Probe crashed: ${message}`,
|
|
71
|
+
};
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
class ProbeTimeoutError extends Error {
|
|
75
|
+
constructor(name, budgetMs) {
|
|
76
|
+
super(`Probe ${name} exceeded ${budgetMs}ms`);
|
|
77
|
+
this.name = 'ProbeTimeoutError';
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
async function raceWithTimeout(entry, budgetMs) {
|
|
81
|
+
let timer;
|
|
82
|
+
const timeoutPromise = new Promise((_, reject) => {
|
|
83
|
+
timer = setTimeout(() => reject(new ProbeTimeoutError(entry.name, budgetMs)), budgetMs);
|
|
84
|
+
});
|
|
85
|
+
try {
|
|
86
|
+
return await Promise.race([entry.run(), timeoutPromise]);
|
|
87
|
+
}
|
|
88
|
+
finally {
|
|
89
|
+
if (timer !== undefined)
|
|
90
|
+
clearTimeout(timer);
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
//# sourceMappingURL=probe-runner.js.map
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* API probe — verifies `api.pugi.io` (or the active apiUrl) is
|
|
3
|
+
* reachable WITHOUT requiring a valid auth token. The auth probe
|
|
4
|
+
* covers the credentialed path; this probe answers the orthogonal
|
|
5
|
+
* "is the network broken" question so the operator can disambiguate
|
|
6
|
+
* "I'm offline" from "my token is bad" without reading two probe
|
|
7
|
+
* details together.
|
|
8
|
+
*
|
|
9
|
+
* Success criteria: any HTTP response (including 401 unauthenticated)
|
|
10
|
+
* proves the host is reachable. A thrown fetch (DNS / TCP / TLS
|
|
11
|
+
* failure) is the only true failure mode.
|
|
12
|
+
*/
|
|
13
|
+
export async function probeApi(ctx, deps) {
|
|
14
|
+
const apiUrl = deps.resolveApiUrl(ctx.env);
|
|
15
|
+
const startedAt = deps.now();
|
|
16
|
+
try {
|
|
17
|
+
const response = await deps.fetchImpl(`${stripTrailingSlash(apiUrl)}/api/pugi/health`, {
|
|
18
|
+
method: 'GET',
|
|
19
|
+
});
|
|
20
|
+
const latencyMs = deps.now() - startedAt;
|
|
21
|
+
// Any HTTP response confirms the host is reachable. The auth probe
|
|
22
|
+
// is responsible for verdicting the 401 / 200 split — here we just
|
|
23
|
+
// confirm we can talk to the server.
|
|
24
|
+
return {
|
|
25
|
+
name: 'API',
|
|
26
|
+
status: 'ok',
|
|
27
|
+
detail: `${apiUrl} reachable (${response.status})`,
|
|
28
|
+
latencyMs,
|
|
29
|
+
};
|
|
30
|
+
}
|
|
31
|
+
catch (error) {
|
|
32
|
+
const latencyMs = deps.now() - startedAt;
|
|
33
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
34
|
+
return {
|
|
35
|
+
name: 'API',
|
|
36
|
+
status: 'error',
|
|
37
|
+
detail: `Cannot reach ${apiUrl}`,
|
|
38
|
+
latencyMs,
|
|
39
|
+
remediation: `Check network or override with PUGI_API_URL: ${message}`,
|
|
40
|
+
};
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
function stripTrailingSlash(url) {
|
|
44
|
+
return url.endsWith('/') ? url.slice(0, -1) : url;
|
|
45
|
+
}
|
|
46
|
+
//# sourceMappingURL=api.js.map
|