gsd-pi 2.80.0-dev.c5f2443b3 → 2.80.0-dev.e146beb20
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/dist/resources/.managed-resources-content-hash +1 -1
- package/dist/resources/GSD-WORKFLOW.md +2 -2
- package/dist/resources/extensions/github-sync/templates.js +39 -8
- package/dist/resources/extensions/gsd/auto/loop.js +48 -10
- package/dist/resources/extensions/gsd/auto/phases.js +37 -30
- package/dist/resources/extensions/gsd/auto/run-unit.js +19 -15
- package/dist/resources/extensions/gsd/auto-dashboard.js +51 -15
- package/dist/resources/extensions/gsd/auto-dispatch.js +10 -0
- package/dist/resources/extensions/gsd/auto-post-unit.js +10 -10
- package/dist/resources/extensions/gsd/auto-prompts.js +111 -1
- package/dist/resources/extensions/gsd/auto-recovery.js +154 -8
- package/dist/resources/extensions/gsd/auto-start.js +2 -3
- package/dist/resources/extensions/gsd/auto.js +9 -1
- package/dist/resources/extensions/gsd/bootstrap/agent-end-recovery.js +15 -1
- package/dist/resources/extensions/gsd/bootstrap/register-hooks.js +11 -1
- package/dist/resources/extensions/gsd/bootstrap/write-gate.js +129 -1
- package/dist/resources/extensions/gsd/clean-root-preflight.js +42 -4
- package/dist/resources/extensions/gsd/commands/dispatcher.js +5 -0
- package/dist/resources/extensions/gsd/commands-extract-learnings.js +17 -12
- package/dist/resources/extensions/gsd/crash-recovery.js +56 -10
- package/dist/resources/extensions/gsd/custom-workflow-engine.js +22 -2
- package/dist/resources/extensions/gsd/db-base-schema.js +14 -0
- package/dist/resources/extensions/gsd/db-migration-steps.js +16 -0
- package/dist/resources/extensions/gsd/detection.js +106 -0
- package/dist/resources/extensions/gsd/graph.js +9 -3
- package/dist/resources/extensions/gsd/gsd-db.js +102 -2
- package/dist/resources/extensions/gsd/guided-flow.js +49 -12
- package/dist/resources/extensions/gsd/planning-path-scope.js +26 -0
- package/dist/resources/extensions/gsd/pr-evidence.js +57 -16
- package/dist/resources/extensions/gsd/prompts/complete-milestone.md +7 -8
- package/dist/resources/extensions/gsd/prompts/plan-milestone.md +3 -1
- package/dist/resources/extensions/gsd/prompts/plan-slice.md +1 -1
- package/dist/resources/extensions/gsd/safety/evidence-collector.js +10 -2
- package/dist/resources/extensions/gsd/tools/plan-slice.js +9 -0
- package/dist/resources/extensions/gsd/tools/plan-task.js +9 -0
- package/dist/resources/extensions/gsd/unit-runtime.js +11 -0
- package/dist/resources/extensions/gsd/working-output-messages.js +64 -0
- package/dist/resources/extensions/gsd/worktree-manager.js +16 -14
- package/dist/tsconfig.extensions.tsbuildinfo +1 -1
- package/dist/web/standalone/.next/BUILD_ID +1 -1
- package/dist/web/standalone/.next/app-path-routes-manifest.json +16 -16
- package/dist/web/standalone/.next/build-manifest.json +3 -3
- package/dist/web/standalone/.next/prerender-manifest.json +3 -3
- package/dist/web/standalone/.next/react-loadable-manifest.json +1 -1
- package/dist/web/standalone/.next/server/app/_global-error.html +1 -1
- package/dist/web/standalone/.next/server/app/_global-error.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_global-error.segments/_full.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_global-error.segments/_global-error/__PAGE__.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_global-error.segments/_global-error.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_global-error.segments/_head.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_global-error.segments/_index.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_global-error.segments/_tree.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_not-found.html +1 -1
- package/dist/web/standalone/.next/server/app/_not-found.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_not-found.segments/_full.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_not-found.segments/_head.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_not-found.segments/_index.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_not-found.segments/_not-found/__PAGE__.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_not-found.segments/_not-found.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/_not-found.segments/_tree.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/api/onboarding/route.js +1 -1
- package/dist/web/standalone/.next/server/app/index.html +1 -1
- package/dist/web/standalone/.next/server/app/index.rsc +1 -1
- package/dist/web/standalone/.next/server/app/index.segments/__PAGE__.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/index.segments/_full.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/index.segments/_head.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/index.segments/_index.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app/index.segments/_tree.segment.rsc +1 -1
- package/dist/web/standalone/.next/server/app-paths-manifest.json +16 -16
- package/dist/web/standalone/.next/server/chunks/6897.js +3 -3
- package/dist/web/standalone/.next/server/middleware-build-manifest.js +1 -1
- package/dist/web/standalone/.next/server/middleware-react-loadable-manifest.js +1 -1
- package/dist/web/standalone/.next/server/pages/404.html +1 -1
- package/dist/web/standalone/.next/server/pages/500.html +1 -1
- package/dist/web/standalone/.next/server/server-reference-manifest.json +1 -1
- package/dist/web/standalone/.next/static/chunks/{8336.6f6f30e410419aff.js → 8336.631939fb583761fa.js} +1 -1
- package/dist/web/standalone/.next/static/chunks/{webpack-d82dbee6356c1733.js → webpack-0481f1221120a7c6.js} +1 -1
- package/package.json +10 -6
- package/packages/contracts/package.json +1 -1
- package/packages/pi-ai/dist/models/fake-model.d.ts +12 -0
- package/packages/pi-ai/dist/models/fake-model.d.ts.map +1 -0
- package/packages/pi-ai/dist/models/fake-model.js +27 -0
- package/packages/pi-ai/dist/models/fake-model.js.map +1 -0
- package/packages/pi-ai/dist/models/index.d.ts.map +1 -1
- package/packages/pi-ai/dist/models/index.js +8 -0
- package/packages/pi-ai/dist/models/index.js.map +1 -1
- package/packages/pi-ai/dist/providers/fake.d.ts +42 -0
- package/packages/pi-ai/dist/providers/fake.d.ts.map +1 -0
- package/packages/pi-ai/dist/providers/fake.js +319 -0
- package/packages/pi-ai/dist/providers/fake.js.map +1 -0
- package/packages/pi-ai/dist/providers/register-builtins.d.ts.map +1 -1
- package/packages/pi-ai/dist/providers/register-builtins.js +24 -0
- package/packages/pi-ai/dist/providers/register-builtins.js.map +1 -1
- package/packages/pi-ai/src/models/fake-model.ts +30 -0
- package/packages/pi-ai/src/models/index.ts +9 -0
- package/packages/pi-ai/src/providers/fake.ts +376 -0
- package/packages/pi-ai/src/providers/register-builtins.ts +23 -0
- package/packages/pi-ai/tsconfig.tsbuildinfo +1 -1
- package/packages/pi-coding-agent/dist/core/chat-controller-ordering.test.js +74 -0
- package/packages/pi-coding-agent/dist/core/chat-controller-ordering.test.js.map +1 -1
- package/packages/pi-coding-agent/dist/core/db-snapshot.d.ts +15 -0
- package/packages/pi-coding-agent/dist/core/db-snapshot.d.ts.map +1 -0
- package/packages/pi-coding-agent/dist/core/db-snapshot.js +66 -0
- package/packages/pi-coding-agent/dist/core/db-snapshot.js.map +1 -0
- package/packages/pi-coding-agent/dist/core/db-snapshot.test.d.ts +2 -0
- package/packages/pi-coding-agent/dist/core/db-snapshot.test.d.ts.map +1 -0
- package/packages/pi-coding-agent/dist/core/db-snapshot.test.js +24 -0
- package/packages/pi-coding-agent/dist/core/db-snapshot.test.js.map +1 -0
- package/packages/pi-coding-agent/dist/core/extensions/runner.d.ts +2 -0
- package/packages/pi-coding-agent/dist/core/extensions/runner.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/core/extensions/runner.js +14 -1
- package/packages/pi-coding-agent/dist/core/extensions/runner.js.map +1 -1
- package/packages/pi-coding-agent/dist/core/extensions/runner.test.js +97 -0
- package/packages/pi-coding-agent/dist/core/extensions/runner.test.js.map +1 -1
- package/packages/pi-coding-agent/dist/core/model-registry.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/core/model-registry.js +5 -0
- package/packages/pi-coding-agent/dist/core/model-registry.js.map +1 -1
- package/packages/pi-coding-agent/dist/core/settings-manager.d.ts +4 -0
- package/packages/pi-coding-agent/dist/core/settings-manager.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/core/settings-manager.js +8 -0
- package/packages/pi-coding-agent/dist/core/settings-manager.js.map +1 -1
- package/packages/pi-coding-agent/dist/core/slash-commands.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/core/slash-commands.js +1 -0
- package/packages/pi-coding-agent/dist/core/slash-commands.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/__tests__/chat-frame-compaction-tone.test.js +6 -4
- package/packages/pi-coding-agent/dist/modes/interactive/components/__tests__/chat-frame-compaction-tone.test.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/__tests__/tool-execution.test.js +54 -15
- package/packages/pi-coding-agent/dist/modes/interactive/components/__tests__/tool-execution.test.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/adaptive-layout.d.ts +26 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/adaptive-layout.d.ts.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/adaptive-layout.js +112 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/adaptive-layout.js.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/adaptive-layout.test.d.ts +2 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/adaptive-layout.test.d.ts.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/adaptive-layout.test.js +51 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/adaptive-layout.test.js.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/assistant-message.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/assistant-message.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/chat-frame.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/chat-frame.js +10 -9
- package/packages/pi-coding-agent/dist/modes/interactive/components/chat-frame.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/settings-selector.d.ts +3 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/settings-selector.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/settings-selector.js +11 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/settings-selector.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/tool-card-cleanup-and-success-runtime.test.js +7 -6
- package/packages/pi-coding-agent/dist/modes/interactive/components/tool-card-cleanup-and-success-runtime.test.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/tool-execution.d.ts +16 -0
- package/packages/pi-coding-agent/dist/modes/interactive/components/tool-execution.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/components/tool-execution.js +106 -17
- package/packages/pi-coding-agent/dist/modes/interactive/components/tool-execution.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/controllers/chat-controller.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/controllers/chat-controller.js +60 -1
- package/packages/pi-coding-agent/dist/modes/interactive/controllers/chat-controller.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/controllers/chat-controller.test.js +40 -1
- package/packages/pi-coding-agent/dist/modes/interactive/controllers/chat-controller.test.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/controllers/input-controller.test.js +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/controllers/input-controller.test.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/interactive-mode-state.d.ts +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/interactive-mode-state.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/interactive-mode-state.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/interactive-mode.d.ts +3 -0
- package/packages/pi-coding-agent/dist/modes/interactive/interactive-mode.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/interactive-mode.js +23 -0
- package/packages/pi-coding-agent/dist/modes/interactive/interactive-mode.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/slash-command-handlers.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/slash-command-handlers.js +20 -0
- package/packages/pi-coding-agent/dist/modes/interactive/slash-command-handlers.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/slash-command-handlers.test.d.ts +2 -0
- package/packages/pi-coding-agent/dist/modes/interactive/slash-command-handlers.test.d.ts.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/slash-command-handlers.test.js +79 -0
- package/packages/pi-coding-agent/dist/modes/interactive/slash-command-handlers.test.js.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/theme/theme-schema.d.ts +12 -0
- package/packages/pi-coding-agent/dist/modes/interactive/theme/theme-schema.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/theme/theme-schema.js +13 -0
- package/packages/pi-coding-agent/dist/modes/interactive/theme/theme-schema.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/theme/theme.d.ts +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/theme/theme.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/theme/theme.js +18 -1
- package/packages/pi-coding-agent/dist/modes/interactive/theme/theme.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/theme/themes.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/theme/themes.js +36 -27
- package/packages/pi-coding-agent/dist/modes/interactive/theme/themes.js.map +1 -1
- package/packages/pi-coding-agent/dist/modes/interactive/tui-mode.d.ts +11 -0
- package/packages/pi-coding-agent/dist/modes/interactive/tui-mode.d.ts.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/tui-mode.js +18 -0
- package/packages/pi-coding-agent/dist/modes/interactive/tui-mode.js.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/tui-mode.test.d.ts +2 -0
- package/packages/pi-coding-agent/dist/modes/interactive/tui-mode.test.d.ts.map +1 -0
- package/packages/pi-coding-agent/dist/modes/interactive/tui-mode.test.js +48 -0
- package/packages/pi-coding-agent/dist/modes/interactive/tui-mode.test.js.map +1 -0
- package/packages/pi-coding-agent/dist/resources/extensions/memory/storage-safety-guard.test.d.ts +2 -0
- package/packages/pi-coding-agent/dist/resources/extensions/memory/storage-safety-guard.test.d.ts.map +1 -0
- package/packages/pi-coding-agent/dist/resources/extensions/memory/storage-safety-guard.test.js +10 -0
- package/packages/pi-coding-agent/dist/resources/extensions/memory/storage-safety-guard.test.js.map +1 -0
- package/packages/pi-coding-agent/dist/resources/extensions/memory/storage.d.ts.map +1 -1
- package/packages/pi-coding-agent/dist/resources/extensions/memory/storage.js +3 -2
- package/packages/pi-coding-agent/dist/resources/extensions/memory/storage.js.map +1 -1
- package/packages/pi-coding-agent/src/core/chat-controller-ordering.test.ts +87 -0
- package/packages/pi-coding-agent/src/core/db-snapshot.test.ts +32 -0
- package/packages/pi-coding-agent/src/core/db-snapshot.ts +66 -0
- package/packages/pi-coding-agent/src/core/extensions/runner.test.ts +108 -0
- package/packages/pi-coding-agent/src/core/extensions/runner.ts +16 -1
- package/packages/pi-coding-agent/src/core/model-registry.ts +4 -0
- package/packages/pi-coding-agent/src/core/settings-manager.ts +12 -0
- package/packages/pi-coding-agent/src/core/slash-commands.ts +1 -0
- package/packages/pi-coding-agent/src/modes/interactive/components/__tests__/chat-frame-compaction-tone.test.ts +7 -5
- package/packages/pi-coding-agent/src/modes/interactive/components/__tests__/tool-execution.test.ts +78 -15
- package/packages/pi-coding-agent/src/modes/interactive/components/adaptive-layout.test.ts +59 -0
- package/packages/pi-coding-agent/src/modes/interactive/components/adaptive-layout.ts +160 -0
- package/packages/pi-coding-agent/src/modes/interactive/components/assistant-message.ts +1 -0
- package/packages/pi-coding-agent/src/modes/interactive/components/chat-frame.ts +10 -9
- package/packages/pi-coding-agent/src/modes/interactive/components/settings-selector.ts +15 -0
- package/packages/pi-coding-agent/src/modes/interactive/components/tool-card-cleanup-and-success-runtime.test.ts +10 -9
- package/packages/pi-coding-agent/src/modes/interactive/components/tool-execution.ts +118 -17
- package/packages/pi-coding-agent/src/modes/interactive/controllers/chat-controller.test.ts +43 -1
- package/packages/pi-coding-agent/src/modes/interactive/controllers/chat-controller.ts +75 -1
- package/packages/pi-coding-agent/src/modes/interactive/controllers/input-controller.test.ts +1 -0
- package/packages/pi-coding-agent/src/modes/interactive/interactive-mode-state.ts +1 -1
- package/packages/pi-coding-agent/src/modes/interactive/interactive-mode.ts +25 -0
- package/packages/pi-coding-agent/src/modes/interactive/slash-command-handlers.test.ts +95 -0
- package/packages/pi-coding-agent/src/modes/interactive/slash-command-handlers.ts +24 -1
- package/packages/pi-coding-agent/src/modes/interactive/theme/theme-schema.ts +13 -0
- package/packages/pi-coding-agent/src/modes/interactive/theme/theme.ts +32 -2
- package/packages/pi-coding-agent/src/modes/interactive/theme/themes.ts +36 -27
- package/packages/pi-coding-agent/src/modes/interactive/tui-mode.test.ts +65 -0
- package/packages/pi-coding-agent/src/modes/interactive/tui-mode.ts +29 -0
- package/packages/pi-coding-agent/src/resources/extensions/memory/storage-safety-guard.test.ts +14 -0
- package/packages/pi-coding-agent/src/resources/extensions/memory/storage.ts +3 -2
- package/packages/pi-coding-agent/tsconfig.tsbuildinfo +1 -1
- package/packages/pi-tui/dist/__tests__/style.test.d.ts +2 -0
- package/packages/pi-tui/dist/__tests__/style.test.d.ts.map +1 -0
- package/packages/pi-tui/dist/__tests__/style.test.js +63 -0
- package/packages/pi-tui/dist/__tests__/style.test.js.map +1 -0
- package/packages/pi-tui/dist/__tests__/tui.test.js +24 -3
- package/packages/pi-tui/dist/__tests__/tui.test.js.map +1 -1
- package/packages/pi-tui/dist/index.d.ts +1 -0
- package/packages/pi-tui/dist/index.d.ts.map +1 -1
- package/packages/pi-tui/dist/index.js +2 -0
- package/packages/pi-tui/dist/index.js.map +1 -1
- package/packages/pi-tui/dist/style.d.ts +41 -0
- package/packages/pi-tui/dist/style.d.ts.map +1 -0
- package/packages/pi-tui/dist/style.js +158 -0
- package/packages/pi-tui/dist/style.js.map +1 -0
- package/packages/pi-tui/dist/tui.d.ts +0 -1
- package/packages/pi-tui/dist/tui.d.ts.map +1 -1
- package/packages/pi-tui/dist/tui.js +3 -8
- package/packages/pi-tui/dist/tui.js.map +1 -1
- package/packages/pi-tui/src/__tests__/style.test.ts +76 -0
- package/packages/pi-tui/src/__tests__/tui.test.ts +29 -3
- package/packages/pi-tui/src/index.ts +9 -0
- package/packages/pi-tui/src/style.ts +225 -0
- package/packages/pi-tui/src/tui.ts +3 -8
- package/packages/pi-tui/tsconfig.tsbuildinfo +1 -1
- package/pkg/dist/modes/interactive/theme/theme-schema.d.ts +12 -0
- package/pkg/dist/modes/interactive/theme/theme-schema.d.ts.map +1 -1
- package/pkg/dist/modes/interactive/theme/theme-schema.js +13 -0
- package/pkg/dist/modes/interactive/theme/theme-schema.js.map +1 -1
- package/pkg/dist/modes/interactive/theme/theme.d.ts +1 -1
- package/pkg/dist/modes/interactive/theme/theme.d.ts.map +1 -1
- package/pkg/dist/modes/interactive/theme/theme.js +18 -1
- package/pkg/dist/modes/interactive/theme/theme.js.map +1 -1
- package/pkg/dist/modes/interactive/theme/themes.d.ts.map +1 -1
- package/pkg/dist/modes/interactive/theme/themes.js +36 -27
- package/pkg/dist/modes/interactive/theme/themes.js.map +1 -1
- package/src/resources/GSD-WORKFLOW.md +2 -2
- package/src/resources/extensions/github-sync/templates.ts +38 -8
- package/src/resources/extensions/github-sync/tests/inline-code.test.ts +66 -0
- package/src/resources/extensions/gsd/auto/loop-deps.ts +1 -0
- package/src/resources/extensions/gsd/auto/loop.ts +67 -18
- package/src/resources/extensions/gsd/auto/phases.ts +42 -28
- package/src/resources/extensions/gsd/auto/run-unit.ts +24 -14
- package/src/resources/extensions/gsd/auto-dashboard.ts +57 -8
- package/src/resources/extensions/gsd/auto-dispatch.ts +17 -0
- package/src/resources/extensions/gsd/auto-post-unit.ts +10 -10
- package/src/resources/extensions/gsd/auto-prompts.ts +116 -1
- package/src/resources/extensions/gsd/auto-recovery.ts +153 -7
- package/src/resources/extensions/gsd/auto-start.ts +7 -6
- package/src/resources/extensions/gsd/auto.ts +12 -1
- package/src/resources/extensions/gsd/bootstrap/agent-end-recovery.ts +16 -1
- package/src/resources/extensions/gsd/bootstrap/register-hooks.ts +17 -1
- package/src/resources/extensions/gsd/bootstrap/write-gate.ts +135 -1
- package/src/resources/extensions/gsd/clean-root-preflight.ts +41 -3
- package/src/resources/extensions/gsd/commands/dispatcher.ts +6 -0
- package/src/resources/extensions/gsd/commands-extract-learnings.ts +17 -12
- package/src/resources/extensions/gsd/crash-recovery.ts +67 -10
- package/src/resources/extensions/gsd/custom-workflow-engine.ts +24 -1
- package/src/resources/extensions/gsd/db-base-schema.ts +15 -0
- package/src/resources/extensions/gsd/db-migration-steps.ts +17 -0
- package/src/resources/extensions/gsd/detection.ts +128 -0
- package/src/resources/extensions/gsd/graph.ts +12 -5
- package/src/resources/extensions/gsd/gsd-db.ts +119 -1
- package/src/resources/extensions/gsd/guided-flow.ts +49 -12
- package/src/resources/extensions/gsd/planning-path-scope.ts +35 -0
- package/src/resources/extensions/gsd/pr-evidence.ts +63 -5
- package/src/resources/extensions/gsd/prompts/complete-milestone.md +7 -8
- package/src/resources/extensions/gsd/prompts/plan-milestone.md +3 -1
- package/src/resources/extensions/gsd/prompts/plan-slice.md +1 -1
- package/src/resources/extensions/gsd/safety/evidence-collector.ts +11 -2
- package/src/resources/extensions/gsd/tests/auto-abort-pause-regression.test.ts +7 -1
- package/src/resources/extensions/gsd/tests/auto-dashboard.test.ts +33 -0
- package/src/resources/extensions/gsd/tests/auto-loop.test.ts +84 -5
- package/src/resources/extensions/gsd/tests/auto-recovery.test.ts +170 -1
- package/src/resources/extensions/gsd/tests/clean-root-preflight.test.ts +88 -2
- package/src/resources/extensions/gsd/tests/commands-extract-learnings.test.ts +9 -0
- package/src/resources/extensions/gsd/tests/crash-handler-secondary.test.ts +55 -0
- package/src/resources/extensions/gsd/tests/crash-recovery-via-db.test.ts +22 -0
- package/src/resources/extensions/gsd/tests/custom-engine-loop-integration.test.ts +112 -6
- package/src/resources/extensions/gsd/tests/custom-workflow-engine.test.ts +40 -2
- package/src/resources/extensions/gsd/tests/db-migration-steps.integration.test.ts +428 -0
- package/src/resources/extensions/gsd/tests/db-schema-metadata.test.ts +2 -2
- package/src/resources/extensions/gsd/tests/detection.test.ts +140 -0
- package/src/resources/extensions/gsd/tests/fixtures/pr-body/commands-ship-basic.md +52 -0
- package/src/resources/extensions/gsd/tests/fixtures/pr-body/commands-ship-empty-optionals.md +42 -0
- package/src/resources/extensions/gsd/tests/fixtures/pr-body/swarm-lane-no-blockers.md +55 -0
- package/src/resources/extensions/gsd/tests/fixtures/pr-body/swarm-lane-with-blockers.md +60 -0
- package/src/resources/extensions/gsd/tests/graph-operations.test.ts +10 -0
- package/src/resources/extensions/gsd/tests/gsd-db.test.ts +44 -0
- package/src/resources/extensions/gsd/tests/has-pending-deep-stage.test.ts +33 -1
- package/src/resources/extensions/gsd/tests/plan-slice.test.ts +50 -0
- package/src/resources/extensions/gsd/tests/plan-task.test.ts +21 -0
- package/src/resources/extensions/gsd/tests/pr-evidence-equivalence.test.ts +102 -0
- package/src/resources/extensions/gsd/tests/pr-evidence-hardening.test.ts +165 -0
- package/src/resources/extensions/gsd/tests/right-sized-workflow-prompts.test.ts +192 -0
- package/src/resources/extensions/gsd/tests/safety-harness-false-positives.test.ts +29 -0
- package/src/resources/extensions/gsd/tests/smart-entry-complete.test.ts +38 -0
- package/src/resources/extensions/gsd/tests/start-auto-detached.test.ts +46 -2
- package/src/resources/extensions/gsd/tests/unit-runtime.test.ts +7 -0
- package/src/resources/extensions/gsd/tests/uok-plan-v2-wiring.test.ts +1 -1
- package/src/resources/extensions/gsd/tests/working-output-messages.test.ts +93 -0
- package/src/resources/extensions/gsd/tests/worktree-health-dispatch.test.ts +37 -6
- package/src/resources/extensions/gsd/tests/worktree-manager.test.ts +7 -0
- package/src/resources/extensions/gsd/tests/worktree-nested-git-safety.test.ts +9 -2
- package/src/resources/extensions/gsd/tests/worktree-write-gate.test.ts +179 -0
- package/src/resources/extensions/gsd/tools/plan-slice.ts +13 -0
- package/src/resources/extensions/gsd/tools/plan-task.ts +10 -0
- package/src/resources/extensions/gsd/unit-runtime.ts +14 -0
- package/src/resources/extensions/gsd/working-output-messages.ts +120 -0
- package/src/resources/extensions/gsd/worktree-manager.ts +15 -4
- package/packages/contracts/tsconfig.tsbuildinfo +0 -1
- /package/dist/web/standalone/.next/static/{bQDK5_LtkGVS64AirQgQG → y73quA-XdLo9n41nxphjW}/_buildManifest.js +0 -0
- /package/dist/web/standalone/.next/static/{bQDK5_LtkGVS64AirQgQG → y73quA-XdLo9n41nxphjW}/_ssgManifest.js +0 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tui-mode.test.js","sourceRoot":"","sources":["../../../src/modes/interactive/tui-mode.test.ts"],"names":[],"mappings":"AAAA,+CAA+C;AAE/C,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,SAAS,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC1E,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;AAEzB,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC/B,IAAI,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,KAAK,CACX,cAAc,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,sBAAsB,EAAE,CAAC,EAC1F,OAAO,CACP,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,iDAAiD,EAAE,GAAG,EAAE;QAC5D,MAAM,CAAC,KAAK,CACX,cAAc,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,QAAQ,EAAE,sBAAsB,EAAE,CAAC,EACjH,SAAS,CACT,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wDAAwD,EAAE,GAAG,EAAE;QACnE,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,aAAa,EAAE,GAAG,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACtE,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,aAAa,EAAE,GAAG,EAAE,QAAQ,EAAE,sBAAsB,EAAE,CAAC,EAAE,YAAY,CAAC,CAAC;QACrG,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,aAAa,EAAE,GAAG,EAAE,QAAQ,EAAE,oBAAoB,EAAE,CAAC,EAAE,YAAY,CAAC,CAAC;IACpG,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC9E,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,aAAa,EAAE,GAAG,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;QACrF,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,aAAa,EAAE,GAAG,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,gDAAgD,EAAE,GAAG,EAAE;QAC3D,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACxC,IAAI,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC/D,MAAM,MAAM,GAAG,IAAI,uBAAuB,CAAC,GAAG,EAAE,CAAC,CAAC;YACjD,QAAQ,EAAE,UAAU;YACpB,eAAe,EAAE,CAAC;YAClB,QAAQ,EAAE,cAAc;YACxB,WAAW,EAAE,MAAM;YACnB,GAAG,EAAE,wBAAwB;SAC7B,CAAC,CAAC,CAAC;QAEJ,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEhD,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,gDAAgD,CAAC,CAAC;QAChF,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,EAAE,8BAA8B,CAAC,CAAC;QACrG,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,+BAA+B,CAAC,CAAC;QAC3F,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,EAAE,wCAAwC,CAAC,CAAC;QACrG,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,oDAAoD,CAAC,CAAC;IAC7G,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["// GSD2 - Tests for adaptive TUI mode selection\n\nimport assert from \"node:assert/strict\";\nimport { describe, test } from \"node:test\";\nimport stripAnsi from \"strip-ansi\";\n\nimport { AdaptiveLayoutComponent } from \"./components/adaptive-layout.js\";\nimport { initTheme } from \"./theme/theme.js\";\nimport { resolveTuiMode } from \"./tui-mode.js\";\n\ninitTheme(\"dark\", false);\n\ndescribe(\"resolveTuiMode\", () => {\n\ttest(\"explicit overrides beat auto selection\", () => {\n\t\tassert.equal(\n\t\t\tresolveTuiMode({ terminalWidth: 60, override: \"debug\", gsdPhase: \"validating-milestone\" }),\n\t\t\t\"debug\",\n\t\t);\n\t});\n\n\ttest(\"prioritizes compact layouts on narrow terminals\", () => {\n\t\tassert.equal(\n\t\t\tresolveTuiMode({ terminalWidth: 60, override: \"auto\", hasBlockingError: true, gsdPhase: \"validating-milestone\" }),\n\t\t\t\"compact\",\n\t\t);\n\t});\n\n\ttest(\"uses debug mode for blocking errors on roomy terminals\", () => {\n\t\tassert.equal(resolveTuiMode({ terminalWidth: 100, hasBlockingError: true }), \"debug\");\n\t});\n\n\ttest(\"uses validation mode for validation and completion phases\", () => {\n\t\tassert.equal(resolveTuiMode({ terminalWidth: 100, gsdPhase: \"validating-milestone\" }), \"validation\");\n\t\tassert.equal(resolveTuiMode({ terminalWidth: 100, gsdPhase: \"complete-milestone\" }), \"validation\");\n\t});\n\n\ttest(\"uses workflow mode when tools or non-validation phases are active\", () => {\n\t\tassert.equal(resolveTuiMode({ terminalWidth: 100, activeToolCount: 1 }), \"workflow\");\n\t\tassert.equal(resolveTuiMode({ terminalWidth: 100, gsdPhase: \"execute-phase\" }), \"workflow\");\n\t});\n\n\ttest(\"falls back to chat mode for plain conversation\", () => {\n\t\tassert.equal(resolveTuiMode({ terminalWidth: 100 }), \"chat\");\n\t});\n});\n\ndescribe(\"AdaptiveLayoutComponent\", () => {\n\ttest(\"renders workflow layout with prototype rule frames\", () => {\n\t\tconst layout = new AdaptiveLayoutComponent(() => ({\n\t\t\toverride: \"workflow\",\n\t\t\tactiveToolCount: 2,\n\t\t\tgsdPhase: \"execute-task\",\n\t\t\tsessionName: \"main\",\n\t\t\tcwd: \"/Users/example/project\",\n\t\t}));\n\n\t\tconst plain = layout.render(120).map(stripAnsi);\n\n\t\tassert.match(plain[0], /^─+/, \"workflow layout should start with a rule frame\");\n\t\tassert.ok(plain.some((line) => line.includes(\"GSD Command Center\")), \"workflow title should render\");\n\t\tassert.ok(plain.some((line) => line.includes(\"signals\")), \"inspector title should render\");\n\t\tassert.ok(plain.some((line) => line.includes(\"│ Active\")), \"body rows should keep prototype gutter\");\n\t\tassert.ok(!plain.some((line) => /[╭╮╰╯]/.test(line)), \"workflow layout should not use rounded box corners\");\n\t});\n});\n"]}
|
package/packages/pi-coding-agent/dist/resources/extensions/memory/storage-safety-guard.test.d.ts.map
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage-safety-guard.test.d.ts","sourceRoot":"","sources":["../../../../src/resources/extensions/memory/storage-safety-guard.test.ts"],"names":[],"mappings":""}
|
package/packages/pi-coding-agent/dist/resources/extensions/memory/storage-safety-guard.test.js
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import assert from "node:assert/strict";
|
|
2
|
+
import { readFileSync } from "node:fs";
|
|
3
|
+
import { join } from "node:path";
|
|
4
|
+
import test from "node:test";
|
|
5
|
+
test("MemoryStorage persists sql.js snapshots through the atomic DB snapshot writer", () => {
|
|
6
|
+
const src = readFileSync(join(process.cwd(), "packages", "pi-coding-agent", "src", "resources", "extensions", "memory", "storage.ts"), "utf-8");
|
|
7
|
+
assert.match(src, /atomicWriteDbSnapshotSync\(/, "storage must use atomic DB snapshot writes");
|
|
8
|
+
assert.doesNotMatch(src, /writeFileSync\(this\.dbPath/, "direct live DB overwrite is forbidden");
|
|
9
|
+
});
|
|
10
|
+
//# sourceMappingURL=storage-safety-guard.test.js.map
|
package/packages/pi-coding-agent/dist/resources/extensions/memory/storage-safety-guard.test.js.map
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage-safety-guard.test.js","sourceRoot":"","sources":["../../../../src/resources/extensions/memory/storage-safety-guard.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,IAAI,CAAC,+EAA+E,EAAE,GAAG,EAAE;IAC1F,MAAM,GAAG,GAAG,YAAY,CACvB,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,UAAU,EAAE,iBAAiB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,EAAE,YAAY,CAAC,EAC5G,OAAO,CACP,CAAC;IAEF,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,6BAA6B,EAAE,4CAA4C,CAAC,CAAC;IAC/F,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE,6BAA6B,EAAE,uCAAuC,CAAC,CAAC;AAClG,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport { readFileSync } from \"node:fs\";\nimport { join } from \"node:path\";\nimport test from \"node:test\";\n\ntest(\"MemoryStorage persists sql.js snapshots through the atomic DB snapshot writer\", () => {\n\tconst src = readFileSync(\n\t\tjoin(process.cwd(), \"packages\", \"pi-coding-agent\", \"src\", \"resources\", \"extensions\", \"memory\", \"storage.ts\"),\n\t\t\"utf-8\",\n\t);\n\n\tassert.match(src, /atomicWriteDbSnapshotSync\\(/, \"storage must use atomic DB snapshot writes\");\n\tassert.doesNotMatch(src, /writeFileSync\\(this\\.dbPath/, \"direct live DB overwrite is forbidden\");\n});\n"]}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../../../src/resources/extensions/memory/storage.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;
|
|
1
|
+
{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../../../src/resources/extensions/memory/storage.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAQH,MAAM,WAAW,SAAS;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,SAAS,GAAG,YAAY,GAAG,MAAM,GAAG,OAAO,CAAC;IACpD,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,eAAe;IAC/B,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,MAAM;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,QAAQ,GAAG,QAAQ,CAAC;IAC3B,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,MAAM,EAAE,SAAS,GAAG,SAAS,GAAG,MAAM,GAAG,OAAO,CAAC;IACjD,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,aAAa;IACzB,OAAO,CAAC,EAAE,CAAgB;IAC1B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,YAAY,CAA8C;IAElE,OAAO;WAKM,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAmB3D,OAAO,CAAC,OAAO;IAKf,OAAO,CAAC,eAAe;IAUvB,OAAO,CAAC,UAAU;IA0ClB,OAAO,CAAC,QAAQ;IAWhB,OAAO,CAAC,QAAQ;IAKhB;;;OAGG;IACH,aAAa,CACZ,OAAO,EAAE,KAAK,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,GAAG,EAAE,MAAM,CAAC;KACZ,CAAC,GACA;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE;IA0CzD;;;OAGG;IACH,eAAe,CACd,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,YAAY,EAAE,MAAM,GAClB,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE,CAAC;IAkCrE;;OAEG;IACH,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAgBzD;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,IAAI;IAY3D;;;OAGG;IACH,uBAAuB,CACtB,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,MAAM,GAClB;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;IAsCnD;;OAEG;IACH,iBAAiB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAQtC;;OAEG;IACH,gBAAgB,IAAI,KAAK,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE,CAAC;IAWvE;;OAEG;IACH,sBAAsB,CAAC,GAAG,EAAE,MAAM,GAAG,KAAK,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE,CAAC;IAcxF;;OAEG;IACH,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAOlD;;OAEG;IACH,QAAQ,IAAI;QACX,YAAY,EAAE,MAAM,CAAC;QACrB,cAAc,EAAE,MAAM,CAAC;QACvB,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,kBAAkB,EAAE,MAAM,CAAC;QAC3B,iBAAiB,EAAE,MAAM,CAAC;KAC1B;IA4BD;;OAEG;IACH,QAAQ,IAAI,IAAI;IAOhB;;OAEG;IACH,WAAW,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAa9B;;OAEG;IACH,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IA4BjC,KAAK,IAAI,IAAI;CAQb"}
|
|
@@ -8,8 +8,9 @@
|
|
|
8
8
|
*/
|
|
9
9
|
import initSqlJs from "sql.js";
|
|
10
10
|
import { randomUUID } from "crypto";
|
|
11
|
-
import { existsSync, mkdirSync, readFileSync
|
|
11
|
+
import { existsSync, mkdirSync, readFileSync } from "fs";
|
|
12
12
|
import { dirname } from "path";
|
|
13
|
+
import { atomicWriteDbSnapshotSync } from "../../../core/db-snapshot.js";
|
|
13
14
|
export class MemoryStorage {
|
|
14
15
|
constructor(db, dbPath) {
|
|
15
16
|
this.persistTimer = null;
|
|
@@ -33,7 +34,7 @@ export class MemoryStorage {
|
|
|
33
34
|
}
|
|
34
35
|
persist() {
|
|
35
36
|
const data = this.db.export();
|
|
36
|
-
|
|
37
|
+
atomicWriteDbSnapshotSync(this.dbPath, data);
|
|
37
38
|
}
|
|
38
39
|
schedulePersist() {
|
|
39
40
|
if (this.persistTimer) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"storage.js","sourceRoot":"","sources":["../../../../src/resources/extensions/memory/storage.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,SAA6C,MAAM,QAAQ,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxE,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAiC/B,MAAM,OAAO,aAAa;IAKzB,YAAoB,EAAiB,EAAE,MAAc;QAF7C,iBAAY,GAAyC,IAAI,CAAC;QAGjE,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAc;QACjC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC5B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACtB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,SAAS,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrE,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;QAElE,EAAE,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QACpC,EAAE,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;QACtC,EAAE,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAErC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC9C,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IAChB,CAAC;IAEO,OAAO;QACd,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC;QAC9B,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/C,CAAC;IAEO,eAAe;QACtB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,CAAC,YAAY,GAAG,UAAU,CAAC,GAAG,EAAE;YACnC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YACzB,IAAI,CAAC,OAAO,EAAE,CAAC;QAChB,CAAC,EAAE,GAAG,CAAC,CAAC;IACT,CAAC;IAEO,UAAU;QACjB,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC;;;;;;;;;;;;GAYX,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC;;;;;;;GAOX,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC;;;;;;;;;;;;;GAaX,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,yEAAyE,CAAC,CAAC;QACvF,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,kEAAkE,CAAC,CAAC;QAChF,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,4DAA4D,CAAC,CAAC;QAC1E,IAAI,CAAC,OAAO,EAAE,CAAC;IAChB,CAAC;IAEO,QAAQ,CAAI,GAAW,EAAE,SAAoB,EAAE;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,MAAiD,CAAC,CAAC;QAC7D,MAAM,IAAI,GAAQ,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YACpB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAO,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,OAAO,IAAI,CAAC;IACb,CAAC;IAEO,QAAQ,CAAI,GAAW,EAAE,SAAoB,EAAE;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAI,GAAG,EAAE,MAAM,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,aAAa,CACZ,OAME;QAEF,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAC7B,uEAAuE,EACvE,CAAC,CAAC,CAAC,QAAQ,CAAC,CACZ,CAAC;YAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACf,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,kHAAkH,EAClH,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CACxD,CAAC;gBACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,8FAA8F,EAC9F,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAC1B,CAAC;gBACF,QAAQ,EAAE,CAAC;YACZ,CAAC;iBAAM,IAAI,QAAQ,CAAC,SAAS,KAAK,CAAC,CAAC,QAAQ,IAAI,QAAQ,CAAC,UAAU,KAAK,CAAC,CAAC,SAAS,EAAE,CAAC;gBACrF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,gJAAgJ,EAChJ,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,CAAC,CACxD,CAAC;gBACF,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAC/D,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,8FAA8F,EAC9F,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAC1B,CAAC;gBACH,CAAC;gBACD,OAAO,EAAE,CAAC;YACX,CAAC;iBAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACX,CAAC;QACF,CAAC;QAED,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;IACvC,CAAC;IAED;;;OAGG;IACH,eAAe,CACd,QAAgB,EAChB,KAAa,EACb,YAAoB;QAEpB,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAE3E,IAAI,CAAC,EAAE,CAAC,GAAG,CACV;;;;;;;;;;;KAWE,EACF,CAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,CACnC,CAAC;QAEF,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CACzB,iFAAiF,EACjF,CAAC,KAAK,CAAC,CACP,CAAC;QAEF,IAAI,CAAC,eAAe,EAAE,CAAC;QAEvB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,KAAK,EAAE,CAAC,CAAC,EAAE;YACX,QAAQ,EAAE,CAAC,CAAC,SAAS;YACrB,cAAc,EAAE,KAAK;SACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,QAAgB,EAAE,MAAc;QACjD,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,+HAA+H,EAC/H,CAAC,QAAQ,CAAC,CACV,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,+GAA+G,EAC/G,CAAC,QAAQ,EAAE,MAAM,CAAC,CAClB,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,sFAAsF,EACtF,CAAC,QAAQ,CAAC,CACV,CAAC;QACF,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAgB,EAAE,YAAoB;QACnD,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,mJAAmJ,EACnJ,CAAC,YAAY,EAAE,QAAQ,CAAC,CACxB,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,0GAA0G,EAC1G,CAAC,YAAY,EAAE,QAAQ,CAAC,CACxB,CAAC;QACF,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;;OAGG;IACH,uBAAuB,CACtB,QAAgB,EAChB,YAAoB;QAEpB,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAE3E,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAClC,8FAA8F,CAC9F,CAAC;QAEF,IAAI,aAAa,IAAI,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC;QACb,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CACnC,0GAA0G,CAC1G,CAAC;QAEF,IAAI,cAAc,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC;QACb,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAChC,4CAA4C,CAC5C,CAAC;QAEF,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,GAAG,KAAK,CAAC,EAAE,CAAC;YAC3C,OAAO,IAAI,CAAC;QACb,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,6HAA6H,EAC7H,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CACnC,CAAC;QAEF,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,KAAa;QAC9B,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,iGAAiG,EACjG,CAAC,KAAK,CAAC,CACP,CAAC;QACF,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACf,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CACzB,uDAAuD,CACvD,CAAC;QAEF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,QAAQ,EAAE,CAAC,CAAC,SAAS;YACrB,cAAc,EAAE,CAAC,CAAC,eAAe;SACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,sBAAsB,CAAC,GAAW;QACjC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CACzB;;mBAEgB,EAChB,CAAC,GAAG,CAAC,CACL,CAAC;QAEF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,QAAQ,EAAE,CAAC,CAAC,SAAS;YACrB,cAAc,EAAE,CAAC,CAAC,eAAe;SACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,QAAgB;QACzB,OAAO,IAAI,CAAC,QAAQ,CACnB,2CAA2C,EAC3C,CAAC,QAAQ,CAAC,CACV,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ;QAQP,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAmE;;;;;;;GAO/F,CAAE,CAAC;QAEJ,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAC5B,4CAA4C,CAC3C,CAAC;QAEH,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAChC,8FAA8F,CAC7F,CAAC;QAEH,OAAO;YACN,YAAY,EAAE,OAAO,CAAC,KAAK;YAC3B,cAAc,EAAE,OAAO,CAAC,OAAO;YAC/B,WAAW,EAAE,OAAO,CAAC,IAAI;YACzB,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,kBAAkB,EAAE,OAAO,CAAC,GAAG;YAC/B,iBAAiB,EAAE,WAAW,CAAC,GAAG;SAClC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ;QACP,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC1C,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAChC,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QACnC,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,GAAW;QACtB,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,6FAA6F,EAC7F,CAAC,GAAG,CAAC,CACL,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,mFAAmF,EACnF,CAAC,GAAG,CAAC,CACL,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,mCAAmC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QACxD,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,GAAW;QACzB,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,6FAA6F,EAC7F,CAAC,GAAG,CAAC,CACL,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,mFAAmF,EACnF,CAAC,GAAG,CAAC,CACL,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,mFAAmF,EACnF,CAAC,GAAG,CAAC,CACL,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAC5B,6CAA6C,EAC7C,CAAC,GAAG,CAAC,CACL,CAAC;QAEF,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACzB,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,oFAAoF,EACpF,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAC3B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED,KAAK;QACJ,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAChC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC1B,CAAC;QACD,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC;CACD","sourcesContent":["/**\n * SQLite storage for the memory extraction pipeline.\n *\n * Tables:\n * - threads: tracks session files and their processing state\n * - stage1_outputs: stores per-thread extraction results\n * - jobs: lease-based job queue for pipeline phases\n */\n\nimport initSqlJs, { type Database as SqlJsDatabase } from \"sql.js\";\nimport { randomUUID } from \"crypto\";\nimport { existsSync, mkdirSync, readFileSync, writeFileSync } from \"fs\";\nimport { dirname } from \"path\";\n\nexport interface ThreadRow {\n\tthread_id: string;\n\tfile_path: string;\n\tfile_size: number;\n\tfile_mtime: number;\n\tcwd: string;\n\tstatus: \"pending\" | \"processing\" | \"done\" | \"error\";\n\terror_message: string | null;\n\tcreated_at: string;\n\tupdated_at: string;\n}\n\nexport interface Stage1OutputRow {\n\tthread_id: string;\n\textraction_json: string;\n\tcreated_at: string;\n}\n\nexport interface JobRow {\n\tid: string;\n\tphase: \"stage1\" | \"stage2\";\n\tthread_id: string | null;\n\tstatus: \"pending\" | \"claimed\" | \"done\" | \"error\";\n\tworker_id: string | null;\n\townership_token: string | null;\n\tlease_expires_at: string | null;\n\terror_message: string | null;\n\tcreated_at: string;\n\tupdated_at: string;\n}\n\nexport class MemoryStorage {\n\tprivate db: SqlJsDatabase;\n\tprivate dbPath: string;\n\tprivate persistTimer: ReturnType<typeof setTimeout> | null = null;\n\n\tprivate constructor(db: SqlJsDatabase, dbPath: string) {\n\t\tthis.db = db;\n\t\tthis.dbPath = dbPath;\n\t}\n\n\tstatic async create(dbPath: string): Promise<MemoryStorage> {\n\t\tconst dir = dirname(dbPath);\n\t\tif (!existsSync(dir)) {\n\t\t\tmkdirSync(dir, { recursive: true });\n\t\t}\n\n\t\tconst SQL = await initSqlJs();\n\t\tconst buffer = existsSync(dbPath) ? readFileSync(dbPath) : undefined;\n\t\tconst db = buffer ? new SQL.Database(buffer) : new SQL.Database();\n\n\t\tdb.run(\"PRAGMA journal_mode = WAL\");\n\t\tdb.run(\"PRAGMA synchronous = NORMAL\");\n\t\tdb.run(\"PRAGMA busy_timeout = 5000\");\n\n\t\tconst storage = new MemoryStorage(db, dbPath);\n\t\tstorage.initSchema();\n\t\treturn storage;\n\t}\n\n\tprivate persist(): void {\n\t\tconst data = this.db.export();\n\t\twriteFileSync(this.dbPath, Buffer.from(data));\n\t}\n\n\tprivate schedulePersist(): void {\n\t\tif (this.persistTimer) {\n\t\t\tclearTimeout(this.persistTimer);\n\t\t}\n\t\tthis.persistTimer = setTimeout(() => {\n\t\t\tthis.persistTimer = null;\n\t\t\tthis.persist();\n\t\t}, 500);\n\t}\n\n\tprivate initSchema(): void {\n\t\tthis.db.run(`\n\t\t\tCREATE TABLE IF NOT EXISTS threads (\n\t\t\t\tthread_id TEXT PRIMARY KEY,\n\t\t\t\tfile_path TEXT NOT NULL,\n\t\t\t\tfile_size INTEGER NOT NULL DEFAULT 0,\n\t\t\t\tfile_mtime INTEGER NOT NULL DEFAULT 0,\n\t\t\t\tcwd TEXT NOT NULL DEFAULT '',\n\t\t\t\tstatus TEXT NOT NULL DEFAULT 'pending',\n\t\t\t\terror_message TEXT,\n\t\t\t\tcreated_at TEXT NOT NULL DEFAULT (datetime('now')),\n\t\t\t\tupdated_at TEXT NOT NULL DEFAULT (datetime('now'))\n\t\t\t)\n\t\t`);\n\t\tthis.db.run(`\n\t\t\tCREATE TABLE IF NOT EXISTS stage1_outputs (\n\t\t\t\tthread_id TEXT PRIMARY KEY,\n\t\t\t\textraction_json TEXT NOT NULL,\n\t\t\t\tcreated_at TEXT NOT NULL DEFAULT (datetime('now')),\n\t\t\t\tFOREIGN KEY (thread_id) REFERENCES threads(thread_id) ON DELETE CASCADE\n\t\t\t)\n\t\t`);\n\t\tthis.db.run(`\n\t\t\tCREATE TABLE IF NOT EXISTS jobs (\n\t\t\t\tid TEXT PRIMARY KEY,\n\t\t\t\tphase TEXT NOT NULL,\n\t\t\t\tthread_id TEXT,\n\t\t\t\tstatus TEXT NOT NULL DEFAULT 'pending',\n\t\t\t\tworker_id TEXT,\n\t\t\t\townership_token TEXT,\n\t\t\t\tlease_expires_at TEXT,\n\t\t\t\terror_message TEXT,\n\t\t\t\tcreated_at TEXT NOT NULL DEFAULT (datetime('now')),\n\t\t\t\tupdated_at TEXT NOT NULL DEFAULT (datetime('now'))\n\t\t\t)\n\t\t`);\n\t\tthis.db.run(\"CREATE INDEX IF NOT EXISTS idx_jobs_phase_status ON jobs(phase, status)\");\n\t\tthis.db.run(\"CREATE INDEX IF NOT EXISTS idx_threads_status ON threads(status)\");\n\t\tthis.db.run(\"CREATE INDEX IF NOT EXISTS idx_threads_cwd ON threads(cwd)\");\n\t\tthis.persist();\n\t}\n\n\tprivate queryAll<T>(sql: string, params: unknown[] = []): T[] {\n\t\tconst stmt = this.db.prepare(sql);\n\t\tstmt.bind(params as (string | number | null | Uint8Array)[]);\n\t\tconst rows: T[] = [];\n\t\twhile (stmt.step()) {\n\t\t\trows.push(stmt.getAsObject() as T);\n\t\t}\n\t\tstmt.free();\n\t\treturn rows;\n\t}\n\n\tprivate queryOne<T>(sql: string, params: unknown[] = []): T | undefined {\n\t\tconst rows = this.queryAll<T>(sql, params);\n\t\treturn rows[0];\n\t}\n\n\t/**\n\t * Insert or update thread records. Skips threads whose file hasn't changed\n\t * (same size + mtime = watermark match).\n\t */\n\tupsertThreads(\n\t\tthreads: Array<{\n\t\t\tthreadId: string;\n\t\t\tfilePath: string;\n\t\t\tfileSize: number;\n\t\t\tfileMtime: number;\n\t\t\tcwd: string;\n\t\t}>,\n\t): { inserted: number; updated: number; skipped: number } {\n\t\tlet inserted = 0;\n\t\tlet updated = 0;\n\t\tlet skipped = 0;\n\n\t\tfor (const t of threads) {\n\t\t\tconst existing = this.queryOne<{ file_size: number; file_mtime: number; status: string }>(\n\t\t\t\t\"SELECT file_size, file_mtime, status FROM threads WHERE thread_id = ?\",\n\t\t\t\t[t.threadId],\n\t\t\t);\n\n\t\t\tif (!existing) {\n\t\t\t\tthis.db.run(\n\t\t\t\t\t\"INSERT INTO threads (thread_id, file_path, file_size, file_mtime, cwd, status) VALUES (?, ?, ?, ?, ?, 'pending')\",\n\t\t\t\t\t[t.threadId, t.filePath, t.fileSize, t.fileMtime, t.cwd],\n\t\t\t\t);\n\t\t\t\tthis.db.run(\n\t\t\t\t\t\"INSERT OR IGNORE INTO jobs (id, phase, thread_id, status) VALUES (?, 'stage1', ?, 'pending')\",\n\t\t\t\t\t[randomUUID(), t.threadId],\n\t\t\t\t);\n\t\t\t\tinserted++;\n\t\t\t} else if (existing.file_size !== t.fileSize || existing.file_mtime !== t.fileMtime) {\n\t\t\t\tthis.db.run(\n\t\t\t\t\t\"UPDATE threads SET file_path = ?, file_size = ?, file_mtime = ?, cwd = ?, status = 'pending', updated_at = datetime('now') WHERE thread_id = ?\",\n\t\t\t\t\t[t.filePath, t.fileSize, t.fileMtime, t.cwd, t.threadId],\n\t\t\t\t);\n\t\t\t\tif (existing.status === \"done\" || existing.status === \"error\") {\n\t\t\t\t\tthis.db.run(\n\t\t\t\t\t\t\"INSERT OR IGNORE INTO jobs (id, phase, thread_id, status) VALUES (?, 'stage1', ?, 'pending')\",\n\t\t\t\t\t\t[randomUUID(), t.threadId],\n\t\t\t\t\t);\n\t\t\t\t}\n\t\t\t\tupdated++;\n\t\t\t} else {\n\t\t\t\tskipped++;\n\t\t\t}\n\t\t}\n\n\t\tthis.schedulePersist();\n\t\treturn { inserted, updated, skipped };\n\t}\n\n\t/**\n\t * Claim up to `limit` stage1 jobs for the given worker.\n\t * Uses lease-based ownership with an ownership_token UUID.\n\t */\n\tclaimStage1Jobs(\n\t\tworkerId: string,\n\t\tlimit: number,\n\t\tleaseSeconds: number,\n\t): Array<{ jobId: string; threadId: string; ownershipToken: string }> {\n\t\tconst token = randomUUID();\n\t\tconst expiresAt = new Date(Date.now() + leaseSeconds * 1000).toISOString();\n\n\t\tthis.db.run(\n\t\t\t`UPDATE jobs SET\n\t\t\t\tstatus = 'claimed',\n\t\t\t\tworker_id = ?,\n\t\t\t\townership_token = ?,\n\t\t\t\tlease_expires_at = ?,\n\t\t\t\tupdated_at = datetime('now')\n\t\t\tWHERE id IN (\n\t\t\t\tSELECT id FROM jobs\n\t\t\t\tWHERE phase = 'stage1'\n\t\t\t\t\tAND (status = 'pending' OR (status = 'claimed' AND lease_expires_at < datetime('now')))\n\t\t\t\tLIMIT ?\n\t\t\t)`,\n\t\t\t[workerId, token, expiresAt, limit],\n\t\t);\n\n\t\tconst rows = this.queryAll<{ id: string; thread_id: string }>(\n\t\t\t\"SELECT id, thread_id FROM jobs WHERE ownership_token = ? AND status = 'claimed'\",\n\t\t\t[token],\n\t\t);\n\n\t\tthis.schedulePersist();\n\n\t\treturn rows.map((r) => ({\n\t\t\tjobId: r.id,\n\t\t\tthreadId: r.thread_id,\n\t\t\townershipToken: token,\n\t\t}));\n\t}\n\n\t/**\n\t * Mark a stage1 job as complete and store the extraction output.\n\t */\n\tcompleteStage1Job(threadId: string, output: string): void {\n\t\tthis.db.run(\n\t\t\t\"UPDATE jobs SET status = 'done', updated_at = datetime('now') WHERE thread_id = ? AND phase = 'stage1' AND status = 'claimed'\",\n\t\t\t[threadId],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"INSERT OR REPLACE INTO stage1_outputs (thread_id, extraction_json, created_at) VALUES (?, ?, datetime('now'))\",\n\t\t\t[threadId, output],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"UPDATE threads SET status = 'done', updated_at = datetime('now') WHERE thread_id = ?\",\n\t\t\t[threadId],\n\t\t);\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Mark a stage1 job as errored.\n\t */\n\tfailStage1Job(threadId: string, errorMessage: string): void {\n\t\tthis.db.run(\n\t\t\t\"UPDATE jobs SET status = 'error', error_message = ?, updated_at = datetime('now') WHERE thread_id = ? AND phase = 'stage1' AND status = 'claimed'\",\n\t\t\t[errorMessage, threadId],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"UPDATE threads SET status = 'error', error_message = ?, updated_at = datetime('now') WHERE thread_id = ?\",\n\t\t\t[errorMessage, threadId],\n\t\t);\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Try to claim the global phase 2 consolidation job.\n\t * Only one worker can hold this at a time.\n\t */\n\ttryClaimGlobalPhase2Job(\n\t\tworkerId: string,\n\t\tleaseSeconds: number,\n\t): { jobId: string; ownershipToken: string } | null {\n\t\tconst token = randomUUID();\n\t\tconst expiresAt = new Date(Date.now() + leaseSeconds * 1000).toISOString();\n\n\t\tconst pendingStage1 = this.queryOne<{ cnt: number }>(\n\t\t\t\"SELECT COUNT(*) as cnt FROM jobs WHERE phase = 'stage1' AND status IN ('pending', 'claimed')\",\n\t\t);\n\n\t\tif (pendingStage1 && pendingStage1.cnt > 0) {\n\t\t\treturn null;\n\t\t}\n\n\t\tconst existingPhase2 = this.queryOne<{ id: string }>(\n\t\t\t\"SELECT id FROM jobs WHERE phase = 'stage2' AND status = 'claimed' AND lease_expires_at > datetime('now')\",\n\t\t);\n\n\t\tif (existingPhase2) {\n\t\t\treturn null;\n\t\t}\n\n\t\tconst outputCount = this.queryOne<{ cnt: number }>(\n\t\t\t\"SELECT COUNT(*) as cnt FROM stage1_outputs\",\n\t\t);\n\n\t\tif (!outputCount || outputCount.cnt === 0) {\n\t\t\treturn null;\n\t\t}\n\n\t\tconst jobId = randomUUID();\n\t\tthis.db.run(\n\t\t\t\"INSERT INTO jobs (id, phase, status, worker_id, ownership_token, lease_expires_at) VALUES (?, 'stage2', 'claimed', ?, ?, ?)\",\n\t\t\t[jobId, workerId, token, expiresAt],\n\t\t);\n\n\t\tthis.schedulePersist();\n\t\treturn { jobId, ownershipToken: token };\n\t}\n\n\t/**\n\t * Complete the phase 2 consolidation job.\n\t */\n\tcompletePhase2Job(jobId: string): void {\n\t\tthis.db.run(\n\t\t\t\"UPDATE jobs SET status = 'done', updated_at = datetime('now') WHERE id = ? AND phase = 'stage2'\",\n\t\t\t[jobId],\n\t\t);\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Get all stage1 extraction outputs.\n\t */\n\tgetStage1Outputs(): Array<{ threadId: string; extractionJson: string }> {\n\t\tconst rows = this.queryAll<{ thread_id: string; extraction_json: string }>(\n\t\t\t\"SELECT thread_id, extraction_json FROM stage1_outputs\",\n\t\t);\n\n\t\treturn rows.map((r) => ({\n\t\t\tthreadId: r.thread_id,\n\t\t\textractionJson: r.extraction_json,\n\t\t}));\n\t}\n\n\t/**\n\t * Get all stage1 outputs for a specific cwd.\n\t */\n\tgetStage1OutputsForCwd(cwd: string): Array<{ threadId: string; extractionJson: string }> {\n\t\tconst rows = this.queryAll<{ thread_id: string; extraction_json: string }>(\n\t\t\t`SELECT s.thread_id, s.extraction_json FROM stage1_outputs s\n\t\t\tINNER JOIN threads t ON t.thread_id = s.thread_id\n\t\t\tWHERE t.cwd = ?`,\n\t\t\t[cwd],\n\t\t);\n\n\t\treturn rows.map((r) => ({\n\t\t\tthreadId: r.thread_id,\n\t\t\textractionJson: r.extraction_json,\n\t\t}));\n\t}\n\n\t/**\n\t * Get thread info by ID.\n\t */\n\tgetThread(threadId: string): ThreadRow | undefined {\n\t\treturn this.queryOne<ThreadRow>(\n\t\t\t\"SELECT * FROM threads WHERE thread_id = ?\",\n\t\t\t[threadId],\n\t\t);\n\t}\n\n\t/**\n\t * Get pipeline statistics.\n\t */\n\tgetStats(): {\n\t\ttotalThreads: number;\n\t\tpendingThreads: number;\n\t\tdoneThreads: number;\n\t\terrorThreads: number;\n\t\ttotalStage1Outputs: number;\n\t\tpendingStage1Jobs: number;\n\t} {\n\t\tconst threads = this.queryOne<{ total: number; pending: number; done: number; errors: number }>(`\n\t\t\tSELECT\n\t\t\t\tCOUNT(*) as total,\n\t\t\t\tSUM(CASE WHEN status = 'pending' THEN 1 ELSE 0 END) as pending,\n\t\t\t\tSUM(CASE WHEN status = 'done' THEN 1 ELSE 0 END) as done,\n\t\t\t\tSUM(CASE WHEN status = 'error' THEN 1 ELSE 0 END) as errors\n\t\t\tFROM threads\n\t\t`)!;\n\n\t\tconst outputs = this.queryOne<{ cnt: number }>(\n\t\t\t\"SELECT COUNT(*) as cnt FROM stage1_outputs\",\n\t\t)!;\n\n\t\tconst pendingJobs = this.queryOne<{ cnt: number }>(\n\t\t\t\"SELECT COUNT(*) as cnt FROM jobs WHERE phase = 'stage1' AND status IN ('pending', 'claimed')\",\n\t\t)!;\n\n\t\treturn {\n\t\t\ttotalThreads: threads.total,\n\t\t\tpendingThreads: threads.pending,\n\t\t\tdoneThreads: threads.done,\n\t\t\terrorThreads: threads.errors,\n\t\t\ttotalStage1Outputs: outputs.cnt,\n\t\t\tpendingStage1Jobs: pendingJobs.cnt,\n\t\t};\n\t}\n\n\t/**\n\t * Clear all data (for /memory clear).\n\t */\n\tclearAll(): void {\n\t\tthis.db.run(\"DELETE FROM stage1_outputs\");\n\t\tthis.db.run(\"DELETE FROM jobs\");\n\t\tthis.db.run(\"DELETE FROM threads\");\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Clear data for a specific cwd (for /memory clear in project scope).\n\t */\n\tclearForCwd(cwd: string): void {\n\t\tthis.db.run(\n\t\t\t\"DELETE FROM stage1_outputs WHERE thread_id IN (SELECT thread_id FROM threads WHERE cwd = ?)\",\n\t\t\t[cwd],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"DELETE FROM jobs WHERE thread_id IN (SELECT thread_id FROM threads WHERE cwd = ?)\",\n\t\t\t[cwd],\n\t\t);\n\t\tthis.db.run(\"DELETE FROM threads WHERE cwd = ?\", [cwd]);\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Reset all threads to pending (for /memory rebuild).\n\t */\n\tresetAllForCwd(cwd: string): void {\n\t\tthis.db.run(\n\t\t\t\"DELETE FROM stage1_outputs WHERE thread_id IN (SELECT thread_id FROM threads WHERE cwd = ?)\",\n\t\t\t[cwd],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"DELETE FROM jobs WHERE thread_id IN (SELECT thread_id FROM threads WHERE cwd = ?)\",\n\t\t\t[cwd],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"UPDATE threads SET status = 'pending', updated_at = datetime('now') WHERE cwd = ?\",\n\t\t\t[cwd],\n\t\t);\n\n\t\tconst threads = this.queryAll<{ thread_id: string }>(\n\t\t\t\"SELECT thread_id FROM threads WHERE cwd = ?\",\n\t\t\t[cwd],\n\t\t);\n\n\t\tfor (const t of threads) {\n\t\t\tthis.db.run(\n\t\t\t\t\"INSERT INTO jobs (id, phase, thread_id, status) VALUES (?, 'stage1', ?, 'pending')\",\n\t\t\t\t[randomUUID(), t.thread_id],\n\t\t\t);\n\t\t}\n\t\tthis.schedulePersist();\n\t}\n\n\tclose(): void {\n\t\tif (this.persistTimer) {\n\t\t\tclearTimeout(this.persistTimer);\n\t\t\tthis.persistTimer = null;\n\t\t}\n\t\tthis.persist();\n\t\tthis.db.close();\n\t}\n}\n"]}
|
|
1
|
+
{"version":3,"file":"storage.js","sourceRoot":"","sources":["../../../../src/resources/extensions/memory/storage.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,SAA6C,MAAM,QAAQ,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AACzD,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC/B,OAAO,EAAE,yBAAyB,EAAE,MAAM,8BAA8B,CAAC;AAiCzE,MAAM,OAAO,aAAa;IAKzB,YAAoB,EAAiB,EAAE,MAAc;QAF7C,iBAAY,GAAyC,IAAI,CAAC;QAGjE,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAc;QACjC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC5B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACtB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,SAAS,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrE,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;QAElE,EAAE,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QACpC,EAAE,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;QACtC,EAAE,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAErC,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC9C,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IAChB,CAAC;IAEO,OAAO;QACd,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC;QAC9B,yBAAyB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC9C,CAAC;IAEO,eAAe;QACtB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,CAAC,YAAY,GAAG,UAAU,CAAC,GAAG,EAAE;YACnC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YACzB,IAAI,CAAC,OAAO,EAAE,CAAC;QAChB,CAAC,EAAE,GAAG,CAAC,CAAC;IACT,CAAC;IAEO,UAAU;QACjB,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC;;;;;;;;;;;;GAYX,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC;;;;;;;GAOX,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC;;;;;;;;;;;;;GAaX,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,yEAAyE,CAAC,CAAC;QACvF,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,kEAAkE,CAAC,CAAC;QAChF,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,4DAA4D,CAAC,CAAC;QAC1E,IAAI,CAAC,OAAO,EAAE,CAAC;IAChB,CAAC;IAEO,QAAQ,CAAI,GAAW,EAAE,SAAoB,EAAE;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,MAAiD,CAAC,CAAC;QAC7D,MAAM,IAAI,GAAQ,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YACpB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAO,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,OAAO,IAAI,CAAC;IACb,CAAC;IAEO,QAAQ,CAAI,GAAW,EAAE,SAAoB,EAAE;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAI,GAAG,EAAE,MAAM,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,aAAa,CACZ,OAME;QAEF,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAC7B,uEAAuE,EACvE,CAAC,CAAC,CAAC,QAAQ,CAAC,CACZ,CAAC;YAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACf,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,kHAAkH,EAClH,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CACxD,CAAC;gBACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,8FAA8F,EAC9F,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAC1B,CAAC;gBACF,QAAQ,EAAE,CAAC;YACZ,CAAC;iBAAM,IAAI,QAAQ,CAAC,SAAS,KAAK,CAAC,CAAC,QAAQ,IAAI,QAAQ,CAAC,UAAU,KAAK,CAAC,CAAC,SAAS,EAAE,CAAC;gBACrF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,gJAAgJ,EAChJ,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,CAAC,CACxD,CAAC;gBACF,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAC/D,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,8FAA8F,EAC9F,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAC1B,CAAC;gBACH,CAAC;gBACD,OAAO,EAAE,CAAC;YACX,CAAC;iBAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACX,CAAC;QACF,CAAC;QAED,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;IACvC,CAAC;IAED;;;OAGG;IACH,eAAe,CACd,QAAgB,EAChB,KAAa,EACb,YAAoB;QAEpB,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAE3E,IAAI,CAAC,EAAE,CAAC,GAAG,CACV;;;;;;;;;;;KAWE,EACF,CAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,CACnC,CAAC;QAEF,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CACzB,iFAAiF,EACjF,CAAC,KAAK,CAAC,CACP,CAAC;QAEF,IAAI,CAAC,eAAe,EAAE,CAAC;QAEvB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,KAAK,EAAE,CAAC,CAAC,EAAE;YACX,QAAQ,EAAE,CAAC,CAAC,SAAS;YACrB,cAAc,EAAE,KAAK;SACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,QAAgB,EAAE,MAAc;QACjD,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,+HAA+H,EAC/H,CAAC,QAAQ,CAAC,CACV,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,+GAA+G,EAC/G,CAAC,QAAQ,EAAE,MAAM,CAAC,CAClB,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,sFAAsF,EACtF,CAAC,QAAQ,CAAC,CACV,CAAC;QACF,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAgB,EAAE,YAAoB;QACnD,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,mJAAmJ,EACnJ,CAAC,YAAY,EAAE,QAAQ,CAAC,CACxB,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,0GAA0G,EAC1G,CAAC,YAAY,EAAE,QAAQ,CAAC,CACxB,CAAC;QACF,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;;OAGG;IACH,uBAAuB,CACtB,QAAgB,EAChB,YAAoB;QAEpB,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAE3E,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAClC,8FAA8F,CAC9F,CAAC;QAEF,IAAI,aAAa,IAAI,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC;QACb,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CACnC,0GAA0G,CAC1G,CAAC;QAEF,IAAI,cAAc,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC;QACb,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAChC,4CAA4C,CAC5C,CAAC;QAEF,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,GAAG,KAAK,CAAC,EAAE,CAAC;YAC3C,OAAO,IAAI,CAAC;QACb,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,6HAA6H,EAC7H,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CACnC,CAAC;QAEF,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,KAAa;QAC9B,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,iGAAiG,EACjG,CAAC,KAAK,CAAC,CACP,CAAC;QACF,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACf,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CACzB,uDAAuD,CACvD,CAAC;QAEF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,QAAQ,EAAE,CAAC,CAAC,SAAS;YACrB,cAAc,EAAE,CAAC,CAAC,eAAe;SACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,sBAAsB,CAAC,GAAW;QACjC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CACzB;;mBAEgB,EAChB,CAAC,GAAG,CAAC,CACL,CAAC;QAEF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,QAAQ,EAAE,CAAC,CAAC,SAAS;YACrB,cAAc,EAAE,CAAC,CAAC,eAAe;SACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,QAAgB;QACzB,OAAO,IAAI,CAAC,QAAQ,CACnB,2CAA2C,EAC3C,CAAC,QAAQ,CAAC,CACV,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ;QAQP,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAmE;;;;;;;GAO/F,CAAE,CAAC;QAEJ,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAC5B,4CAA4C,CAC3C,CAAC;QAEH,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAChC,8FAA8F,CAC7F,CAAC;QAEH,OAAO;YACN,YAAY,EAAE,OAAO,CAAC,KAAK;YAC3B,cAAc,EAAE,OAAO,CAAC,OAAO;YAC/B,WAAW,EAAE,OAAO,CAAC,IAAI;YACzB,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,kBAAkB,EAAE,OAAO,CAAC,GAAG;YAC/B,iBAAiB,EAAE,WAAW,CAAC,GAAG;SAClC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ;QACP,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC1C,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAChC,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QACnC,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,GAAW;QACtB,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,6FAA6F,EAC7F,CAAC,GAAG,CAAC,CACL,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,mFAAmF,EACnF,CAAC,GAAG,CAAC,CACL,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,mCAAmC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QACxD,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,GAAW;QACzB,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,6FAA6F,EAC7F,CAAC,GAAG,CAAC,CACL,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,mFAAmF,EACnF,CAAC,GAAG,CAAC,CACL,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,mFAAmF,EACnF,CAAC,GAAG,CAAC,CACL,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAC5B,6CAA6C,EAC7C,CAAC,GAAG,CAAC,CACL,CAAC;QAEF,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACzB,IAAI,CAAC,EAAE,CAAC,GAAG,CACV,oFAAoF,EACpF,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAC3B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAED,KAAK;QACJ,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAChC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC1B,CAAC;QACD,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC;CACD","sourcesContent":["/**\n * SQLite storage for the memory extraction pipeline.\n *\n * Tables:\n * - threads: tracks session files and their processing state\n * - stage1_outputs: stores per-thread extraction results\n * - jobs: lease-based job queue for pipeline phases\n */\n\nimport initSqlJs, { type Database as SqlJsDatabase } from \"sql.js\";\nimport { randomUUID } from \"crypto\";\nimport { existsSync, mkdirSync, readFileSync } from \"fs\";\nimport { dirname } from \"path\";\nimport { atomicWriteDbSnapshotSync } from \"../../../core/db-snapshot.js\";\n\nexport interface ThreadRow {\n\tthread_id: string;\n\tfile_path: string;\n\tfile_size: number;\n\tfile_mtime: number;\n\tcwd: string;\n\tstatus: \"pending\" | \"processing\" | \"done\" | \"error\";\n\terror_message: string | null;\n\tcreated_at: string;\n\tupdated_at: string;\n}\n\nexport interface Stage1OutputRow {\n\tthread_id: string;\n\textraction_json: string;\n\tcreated_at: string;\n}\n\nexport interface JobRow {\n\tid: string;\n\tphase: \"stage1\" | \"stage2\";\n\tthread_id: string | null;\n\tstatus: \"pending\" | \"claimed\" | \"done\" | \"error\";\n\tworker_id: string | null;\n\townership_token: string | null;\n\tlease_expires_at: string | null;\n\terror_message: string | null;\n\tcreated_at: string;\n\tupdated_at: string;\n}\n\nexport class MemoryStorage {\n\tprivate db: SqlJsDatabase;\n\tprivate dbPath: string;\n\tprivate persistTimer: ReturnType<typeof setTimeout> | null = null;\n\n\tprivate constructor(db: SqlJsDatabase, dbPath: string) {\n\t\tthis.db = db;\n\t\tthis.dbPath = dbPath;\n\t}\n\n\tstatic async create(dbPath: string): Promise<MemoryStorage> {\n\t\tconst dir = dirname(dbPath);\n\t\tif (!existsSync(dir)) {\n\t\t\tmkdirSync(dir, { recursive: true });\n\t\t}\n\n\t\tconst SQL = await initSqlJs();\n\t\tconst buffer = existsSync(dbPath) ? readFileSync(dbPath) : undefined;\n\t\tconst db = buffer ? new SQL.Database(buffer) : new SQL.Database();\n\n\t\tdb.run(\"PRAGMA journal_mode = WAL\");\n\t\tdb.run(\"PRAGMA synchronous = NORMAL\");\n\t\tdb.run(\"PRAGMA busy_timeout = 5000\");\n\n\t\tconst storage = new MemoryStorage(db, dbPath);\n\t\tstorage.initSchema();\n\t\treturn storage;\n\t}\n\n\tprivate persist(): void {\n\t\tconst data = this.db.export();\n\t\tatomicWriteDbSnapshotSync(this.dbPath, data);\n\t}\n\n\tprivate schedulePersist(): void {\n\t\tif (this.persistTimer) {\n\t\t\tclearTimeout(this.persistTimer);\n\t\t}\n\t\tthis.persistTimer = setTimeout(() => {\n\t\t\tthis.persistTimer = null;\n\t\t\tthis.persist();\n\t\t}, 500);\n\t}\n\n\tprivate initSchema(): void {\n\t\tthis.db.run(`\n\t\t\tCREATE TABLE IF NOT EXISTS threads (\n\t\t\t\tthread_id TEXT PRIMARY KEY,\n\t\t\t\tfile_path TEXT NOT NULL,\n\t\t\t\tfile_size INTEGER NOT NULL DEFAULT 0,\n\t\t\t\tfile_mtime INTEGER NOT NULL DEFAULT 0,\n\t\t\t\tcwd TEXT NOT NULL DEFAULT '',\n\t\t\t\tstatus TEXT NOT NULL DEFAULT 'pending',\n\t\t\t\terror_message TEXT,\n\t\t\t\tcreated_at TEXT NOT NULL DEFAULT (datetime('now')),\n\t\t\t\tupdated_at TEXT NOT NULL DEFAULT (datetime('now'))\n\t\t\t)\n\t\t`);\n\t\tthis.db.run(`\n\t\t\tCREATE TABLE IF NOT EXISTS stage1_outputs (\n\t\t\t\tthread_id TEXT PRIMARY KEY,\n\t\t\t\textraction_json TEXT NOT NULL,\n\t\t\t\tcreated_at TEXT NOT NULL DEFAULT (datetime('now')),\n\t\t\t\tFOREIGN KEY (thread_id) REFERENCES threads(thread_id) ON DELETE CASCADE\n\t\t\t)\n\t\t`);\n\t\tthis.db.run(`\n\t\t\tCREATE TABLE IF NOT EXISTS jobs (\n\t\t\t\tid TEXT PRIMARY KEY,\n\t\t\t\tphase TEXT NOT NULL,\n\t\t\t\tthread_id TEXT,\n\t\t\t\tstatus TEXT NOT NULL DEFAULT 'pending',\n\t\t\t\tworker_id TEXT,\n\t\t\t\townership_token TEXT,\n\t\t\t\tlease_expires_at TEXT,\n\t\t\t\terror_message TEXT,\n\t\t\t\tcreated_at TEXT NOT NULL DEFAULT (datetime('now')),\n\t\t\t\tupdated_at TEXT NOT NULL DEFAULT (datetime('now'))\n\t\t\t)\n\t\t`);\n\t\tthis.db.run(\"CREATE INDEX IF NOT EXISTS idx_jobs_phase_status ON jobs(phase, status)\");\n\t\tthis.db.run(\"CREATE INDEX IF NOT EXISTS idx_threads_status ON threads(status)\");\n\t\tthis.db.run(\"CREATE INDEX IF NOT EXISTS idx_threads_cwd ON threads(cwd)\");\n\t\tthis.persist();\n\t}\n\n\tprivate queryAll<T>(sql: string, params: unknown[] = []): T[] {\n\t\tconst stmt = this.db.prepare(sql);\n\t\tstmt.bind(params as (string | number | null | Uint8Array)[]);\n\t\tconst rows: T[] = [];\n\t\twhile (stmt.step()) {\n\t\t\trows.push(stmt.getAsObject() as T);\n\t\t}\n\t\tstmt.free();\n\t\treturn rows;\n\t}\n\n\tprivate queryOne<T>(sql: string, params: unknown[] = []): T | undefined {\n\t\tconst rows = this.queryAll<T>(sql, params);\n\t\treturn rows[0];\n\t}\n\n\t/**\n\t * Insert or update thread records. Skips threads whose file hasn't changed\n\t * (same size + mtime = watermark match).\n\t */\n\tupsertThreads(\n\t\tthreads: Array<{\n\t\t\tthreadId: string;\n\t\t\tfilePath: string;\n\t\t\tfileSize: number;\n\t\t\tfileMtime: number;\n\t\t\tcwd: string;\n\t\t}>,\n\t): { inserted: number; updated: number; skipped: number } {\n\t\tlet inserted = 0;\n\t\tlet updated = 0;\n\t\tlet skipped = 0;\n\n\t\tfor (const t of threads) {\n\t\t\tconst existing = this.queryOne<{ file_size: number; file_mtime: number; status: string }>(\n\t\t\t\t\"SELECT file_size, file_mtime, status FROM threads WHERE thread_id = ?\",\n\t\t\t\t[t.threadId],\n\t\t\t);\n\n\t\t\tif (!existing) {\n\t\t\t\tthis.db.run(\n\t\t\t\t\t\"INSERT INTO threads (thread_id, file_path, file_size, file_mtime, cwd, status) VALUES (?, ?, ?, ?, ?, 'pending')\",\n\t\t\t\t\t[t.threadId, t.filePath, t.fileSize, t.fileMtime, t.cwd],\n\t\t\t\t);\n\t\t\t\tthis.db.run(\n\t\t\t\t\t\"INSERT OR IGNORE INTO jobs (id, phase, thread_id, status) VALUES (?, 'stage1', ?, 'pending')\",\n\t\t\t\t\t[randomUUID(), t.threadId],\n\t\t\t\t);\n\t\t\t\tinserted++;\n\t\t\t} else if (existing.file_size !== t.fileSize || existing.file_mtime !== t.fileMtime) {\n\t\t\t\tthis.db.run(\n\t\t\t\t\t\"UPDATE threads SET file_path = ?, file_size = ?, file_mtime = ?, cwd = ?, status = 'pending', updated_at = datetime('now') WHERE thread_id = ?\",\n\t\t\t\t\t[t.filePath, t.fileSize, t.fileMtime, t.cwd, t.threadId],\n\t\t\t\t);\n\t\t\t\tif (existing.status === \"done\" || existing.status === \"error\") {\n\t\t\t\t\tthis.db.run(\n\t\t\t\t\t\t\"INSERT OR IGNORE INTO jobs (id, phase, thread_id, status) VALUES (?, 'stage1', ?, 'pending')\",\n\t\t\t\t\t\t[randomUUID(), t.threadId],\n\t\t\t\t\t);\n\t\t\t\t}\n\t\t\t\tupdated++;\n\t\t\t} else {\n\t\t\t\tskipped++;\n\t\t\t}\n\t\t}\n\n\t\tthis.schedulePersist();\n\t\treturn { inserted, updated, skipped };\n\t}\n\n\t/**\n\t * Claim up to `limit` stage1 jobs for the given worker.\n\t * Uses lease-based ownership with an ownership_token UUID.\n\t */\n\tclaimStage1Jobs(\n\t\tworkerId: string,\n\t\tlimit: number,\n\t\tleaseSeconds: number,\n\t): Array<{ jobId: string; threadId: string; ownershipToken: string }> {\n\t\tconst token = randomUUID();\n\t\tconst expiresAt = new Date(Date.now() + leaseSeconds * 1000).toISOString();\n\n\t\tthis.db.run(\n\t\t\t`UPDATE jobs SET\n\t\t\t\tstatus = 'claimed',\n\t\t\t\tworker_id = ?,\n\t\t\t\townership_token = ?,\n\t\t\t\tlease_expires_at = ?,\n\t\t\t\tupdated_at = datetime('now')\n\t\t\tWHERE id IN (\n\t\t\t\tSELECT id FROM jobs\n\t\t\t\tWHERE phase = 'stage1'\n\t\t\t\t\tAND (status = 'pending' OR (status = 'claimed' AND lease_expires_at < datetime('now')))\n\t\t\t\tLIMIT ?\n\t\t\t)`,\n\t\t\t[workerId, token, expiresAt, limit],\n\t\t);\n\n\t\tconst rows = this.queryAll<{ id: string; thread_id: string }>(\n\t\t\t\"SELECT id, thread_id FROM jobs WHERE ownership_token = ? AND status = 'claimed'\",\n\t\t\t[token],\n\t\t);\n\n\t\tthis.schedulePersist();\n\n\t\treturn rows.map((r) => ({\n\t\t\tjobId: r.id,\n\t\t\tthreadId: r.thread_id,\n\t\t\townershipToken: token,\n\t\t}));\n\t}\n\n\t/**\n\t * Mark a stage1 job as complete and store the extraction output.\n\t */\n\tcompleteStage1Job(threadId: string, output: string): void {\n\t\tthis.db.run(\n\t\t\t\"UPDATE jobs SET status = 'done', updated_at = datetime('now') WHERE thread_id = ? AND phase = 'stage1' AND status = 'claimed'\",\n\t\t\t[threadId],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"INSERT OR REPLACE INTO stage1_outputs (thread_id, extraction_json, created_at) VALUES (?, ?, datetime('now'))\",\n\t\t\t[threadId, output],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"UPDATE threads SET status = 'done', updated_at = datetime('now') WHERE thread_id = ?\",\n\t\t\t[threadId],\n\t\t);\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Mark a stage1 job as errored.\n\t */\n\tfailStage1Job(threadId: string, errorMessage: string): void {\n\t\tthis.db.run(\n\t\t\t\"UPDATE jobs SET status = 'error', error_message = ?, updated_at = datetime('now') WHERE thread_id = ? AND phase = 'stage1' AND status = 'claimed'\",\n\t\t\t[errorMessage, threadId],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"UPDATE threads SET status = 'error', error_message = ?, updated_at = datetime('now') WHERE thread_id = ?\",\n\t\t\t[errorMessage, threadId],\n\t\t);\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Try to claim the global phase 2 consolidation job.\n\t * Only one worker can hold this at a time.\n\t */\n\ttryClaimGlobalPhase2Job(\n\t\tworkerId: string,\n\t\tleaseSeconds: number,\n\t): { jobId: string; ownershipToken: string } | null {\n\t\tconst token = randomUUID();\n\t\tconst expiresAt = new Date(Date.now() + leaseSeconds * 1000).toISOString();\n\n\t\tconst pendingStage1 = this.queryOne<{ cnt: number }>(\n\t\t\t\"SELECT COUNT(*) as cnt FROM jobs WHERE phase = 'stage1' AND status IN ('pending', 'claimed')\",\n\t\t);\n\n\t\tif (pendingStage1 && pendingStage1.cnt > 0) {\n\t\t\treturn null;\n\t\t}\n\n\t\tconst existingPhase2 = this.queryOne<{ id: string }>(\n\t\t\t\"SELECT id FROM jobs WHERE phase = 'stage2' AND status = 'claimed' AND lease_expires_at > datetime('now')\",\n\t\t);\n\n\t\tif (existingPhase2) {\n\t\t\treturn null;\n\t\t}\n\n\t\tconst outputCount = this.queryOne<{ cnt: number }>(\n\t\t\t\"SELECT COUNT(*) as cnt FROM stage1_outputs\",\n\t\t);\n\n\t\tif (!outputCount || outputCount.cnt === 0) {\n\t\t\treturn null;\n\t\t}\n\n\t\tconst jobId = randomUUID();\n\t\tthis.db.run(\n\t\t\t\"INSERT INTO jobs (id, phase, status, worker_id, ownership_token, lease_expires_at) VALUES (?, 'stage2', 'claimed', ?, ?, ?)\",\n\t\t\t[jobId, workerId, token, expiresAt],\n\t\t);\n\n\t\tthis.schedulePersist();\n\t\treturn { jobId, ownershipToken: token };\n\t}\n\n\t/**\n\t * Complete the phase 2 consolidation job.\n\t */\n\tcompletePhase2Job(jobId: string): void {\n\t\tthis.db.run(\n\t\t\t\"UPDATE jobs SET status = 'done', updated_at = datetime('now') WHERE id = ? AND phase = 'stage2'\",\n\t\t\t[jobId],\n\t\t);\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Get all stage1 extraction outputs.\n\t */\n\tgetStage1Outputs(): Array<{ threadId: string; extractionJson: string }> {\n\t\tconst rows = this.queryAll<{ thread_id: string; extraction_json: string }>(\n\t\t\t\"SELECT thread_id, extraction_json FROM stage1_outputs\",\n\t\t);\n\n\t\treturn rows.map((r) => ({\n\t\t\tthreadId: r.thread_id,\n\t\t\textractionJson: r.extraction_json,\n\t\t}));\n\t}\n\n\t/**\n\t * Get all stage1 outputs for a specific cwd.\n\t */\n\tgetStage1OutputsForCwd(cwd: string): Array<{ threadId: string; extractionJson: string }> {\n\t\tconst rows = this.queryAll<{ thread_id: string; extraction_json: string }>(\n\t\t\t`SELECT s.thread_id, s.extraction_json FROM stage1_outputs s\n\t\t\tINNER JOIN threads t ON t.thread_id = s.thread_id\n\t\t\tWHERE t.cwd = ?`,\n\t\t\t[cwd],\n\t\t);\n\n\t\treturn rows.map((r) => ({\n\t\t\tthreadId: r.thread_id,\n\t\t\textractionJson: r.extraction_json,\n\t\t}));\n\t}\n\n\t/**\n\t * Get thread info by ID.\n\t */\n\tgetThread(threadId: string): ThreadRow | undefined {\n\t\treturn this.queryOne<ThreadRow>(\n\t\t\t\"SELECT * FROM threads WHERE thread_id = ?\",\n\t\t\t[threadId],\n\t\t);\n\t}\n\n\t/**\n\t * Get pipeline statistics.\n\t */\n\tgetStats(): {\n\t\ttotalThreads: number;\n\t\tpendingThreads: number;\n\t\tdoneThreads: number;\n\t\terrorThreads: number;\n\t\ttotalStage1Outputs: number;\n\t\tpendingStage1Jobs: number;\n\t} {\n\t\tconst threads = this.queryOne<{ total: number; pending: number; done: number; errors: number }>(`\n\t\t\tSELECT\n\t\t\t\tCOUNT(*) as total,\n\t\t\t\tSUM(CASE WHEN status = 'pending' THEN 1 ELSE 0 END) as pending,\n\t\t\t\tSUM(CASE WHEN status = 'done' THEN 1 ELSE 0 END) as done,\n\t\t\t\tSUM(CASE WHEN status = 'error' THEN 1 ELSE 0 END) as errors\n\t\t\tFROM threads\n\t\t`)!;\n\n\t\tconst outputs = this.queryOne<{ cnt: number }>(\n\t\t\t\"SELECT COUNT(*) as cnt FROM stage1_outputs\",\n\t\t)!;\n\n\t\tconst pendingJobs = this.queryOne<{ cnt: number }>(\n\t\t\t\"SELECT COUNT(*) as cnt FROM jobs WHERE phase = 'stage1' AND status IN ('pending', 'claimed')\",\n\t\t)!;\n\n\t\treturn {\n\t\t\ttotalThreads: threads.total,\n\t\t\tpendingThreads: threads.pending,\n\t\t\tdoneThreads: threads.done,\n\t\t\terrorThreads: threads.errors,\n\t\t\ttotalStage1Outputs: outputs.cnt,\n\t\t\tpendingStage1Jobs: pendingJobs.cnt,\n\t\t};\n\t}\n\n\t/**\n\t * Clear all data (for /memory clear).\n\t */\n\tclearAll(): void {\n\t\tthis.db.run(\"DELETE FROM stage1_outputs\");\n\t\tthis.db.run(\"DELETE FROM jobs\");\n\t\tthis.db.run(\"DELETE FROM threads\");\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Clear data for a specific cwd (for /memory clear in project scope).\n\t */\n\tclearForCwd(cwd: string): void {\n\t\tthis.db.run(\n\t\t\t\"DELETE FROM stage1_outputs WHERE thread_id IN (SELECT thread_id FROM threads WHERE cwd = ?)\",\n\t\t\t[cwd],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"DELETE FROM jobs WHERE thread_id IN (SELECT thread_id FROM threads WHERE cwd = ?)\",\n\t\t\t[cwd],\n\t\t);\n\t\tthis.db.run(\"DELETE FROM threads WHERE cwd = ?\", [cwd]);\n\t\tthis.schedulePersist();\n\t}\n\n\t/**\n\t * Reset all threads to pending (for /memory rebuild).\n\t */\n\tresetAllForCwd(cwd: string): void {\n\t\tthis.db.run(\n\t\t\t\"DELETE FROM stage1_outputs WHERE thread_id IN (SELECT thread_id FROM threads WHERE cwd = ?)\",\n\t\t\t[cwd],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"DELETE FROM jobs WHERE thread_id IN (SELECT thread_id FROM threads WHERE cwd = ?)\",\n\t\t\t[cwd],\n\t\t);\n\t\tthis.db.run(\n\t\t\t\"UPDATE threads SET status = 'pending', updated_at = datetime('now') WHERE cwd = ?\",\n\t\t\t[cwd],\n\t\t);\n\n\t\tconst threads = this.queryAll<{ thread_id: string }>(\n\t\t\t\"SELECT thread_id FROM threads WHERE cwd = ?\",\n\t\t\t[cwd],\n\t\t);\n\n\t\tfor (const t of threads) {\n\t\t\tthis.db.run(\n\t\t\t\t\"INSERT INTO jobs (id, phase, thread_id, status) VALUES (?, 'stage1', ?, 'pending')\",\n\t\t\t\t[randomUUID(), t.thread_id],\n\t\t\t);\n\t\t}\n\t\tthis.schedulePersist();\n\t}\n\n\tclose(): void {\n\t\tif (this.persistTimer) {\n\t\t\tclearTimeout(this.persistTimer);\n\t\t\tthis.persistTimer = null;\n\t\t}\n\t\tthis.persist();\n\t\tthis.db.close();\n\t}\n}\n"]}
|
|
@@ -1002,6 +1002,93 @@ test("chat-controller does not pin when there are no tool calls", async () => {
|
|
|
1002
1002
|
assert.equal(host.pinnedMessageContainer.children.length, 0, "pinned zone should stay empty without tool calls");
|
|
1003
1003
|
});
|
|
1004
1004
|
|
|
1005
|
+
test("chat-controller rolls up only contiguous low-signal tool runs on message_end", async () => {
|
|
1006
|
+
(globalThis as any)[Symbol.for("@gsd/pi-coding-agent:theme")] = {
|
|
1007
|
+
fg: (_key: string, text: string) => text,
|
|
1008
|
+
bg: (_key: string, text: string) => text,
|
|
1009
|
+
bold: (text: string) => text,
|
|
1010
|
+
italic: (text: string) => text,
|
|
1011
|
+
truncate: (text: string) => text,
|
|
1012
|
+
};
|
|
1013
|
+
|
|
1014
|
+
const host = createHost();
|
|
1015
|
+
host.getMarkdownThemeWithSettings = () => ({});
|
|
1016
|
+
|
|
1017
|
+
const t1 = { type: "toolCall", id: "t1", name: "bash", arguments: { command: "true" } };
|
|
1018
|
+
const t2 = { type: "toolCall", id: "t2", name: "bash", arguments: { command: "true" } };
|
|
1019
|
+
const text = { type: "text", text: "middle output" };
|
|
1020
|
+
const t3 = { type: "toolCall", id: "t3", name: "read", arguments: { path: "/tmp/a" } };
|
|
1021
|
+
const t4 = { type: "toolCall", id: "t4", name: "read", arguments: { path: "/tmp/b" } };
|
|
1022
|
+
const content = [t1, t2, text, t3, t4];
|
|
1023
|
+
|
|
1024
|
+
await handleAgentEvent(host, { type: "message_start", message: makeAssistant([]) } as any);
|
|
1025
|
+
await handleAgentEvent(host, {
|
|
1026
|
+
type: "message_update",
|
|
1027
|
+
message: makeAssistant(content),
|
|
1028
|
+
assistantMessageEvent: {
|
|
1029
|
+
type: "text_delta",
|
|
1030
|
+
contentIndex: 2,
|
|
1031
|
+
delta: text.text,
|
|
1032
|
+
partial: makeAssistant(content),
|
|
1033
|
+
},
|
|
1034
|
+
} as any);
|
|
1035
|
+
|
|
1036
|
+
for (const tool of [t1, t2, t3, t4]) {
|
|
1037
|
+
await handleAgentEvent(host, {
|
|
1038
|
+
type: "tool_execution_end",
|
|
1039
|
+
toolCallId: tool.id,
|
|
1040
|
+
isError: false,
|
|
1041
|
+
result: { content: [], details: {} },
|
|
1042
|
+
} as any);
|
|
1043
|
+
}
|
|
1044
|
+
|
|
1045
|
+
await handleAgentEvent(host, { type: "message_end", message: makeAssistant(content) } as any);
|
|
1046
|
+
|
|
1047
|
+
assert.equal(host.chatContainer.children.length, 3, "two separated tool runs should become two summaries around text");
|
|
1048
|
+
assert.equal(host.chatContainer.children[0]?.constructor?.name, "ToolPhaseSummaryComponent");
|
|
1049
|
+
assert.equal(host.chatContainer.children[1]?.constructor?.name, "AssistantMessageComponent");
|
|
1050
|
+
assert.equal(host.chatContainer.children[2]?.constructor?.name, "ToolPhaseSummaryComponent");
|
|
1051
|
+
assert.match(host.chatContainer.children[0].render(120).join("\n"), /Setup \/ shell 2 actions/);
|
|
1052
|
+
assert.match(host.chatContainer.children[2].render(120).join("\n"), /Context reads 2 actions/);
|
|
1053
|
+
assert.equal(host.chatContainer._prevRender, null, "summary reposition must invalidate the chat container render cache");
|
|
1054
|
+
});
|
|
1055
|
+
|
|
1056
|
+
test("chat-controller rolls up low-signal direct tool execution events on agent_end", async () => {
|
|
1057
|
+
(globalThis as any)[Symbol.for("@gsd/pi-coding-agent:theme")] = {
|
|
1058
|
+
fg: (_key: string, text: string) => text,
|
|
1059
|
+
bg: (_key: string, text: string) => text,
|
|
1060
|
+
bold: (text: string) => text,
|
|
1061
|
+
italic: (text: string) => text,
|
|
1062
|
+
truncate: (text: string) => text,
|
|
1063
|
+
};
|
|
1064
|
+
|
|
1065
|
+
const host = createHost();
|
|
1066
|
+
host.getMarkdownThemeWithSettings = () => ({});
|
|
1067
|
+
|
|
1068
|
+
for (const toolCallId of ["bash-1", "bash-2", "bash-3"]) {
|
|
1069
|
+
await handleAgentEvent(host, {
|
|
1070
|
+
type: "tool_execution_start",
|
|
1071
|
+
toolCallId,
|
|
1072
|
+
toolName: "bash",
|
|
1073
|
+
args: { command: "true" },
|
|
1074
|
+
} as any);
|
|
1075
|
+
await handleAgentEvent(host, {
|
|
1076
|
+
type: "tool_execution_end",
|
|
1077
|
+
toolCallId,
|
|
1078
|
+
isError: false,
|
|
1079
|
+
result: { content: [], details: {} },
|
|
1080
|
+
} as any);
|
|
1081
|
+
}
|
|
1082
|
+
|
|
1083
|
+
assert.equal(host.chatContainer.children.length, 3, "direct tool events render as individual rows while running");
|
|
1084
|
+
|
|
1085
|
+
await handleAgentEvent(host, { type: "agent_end" } as any);
|
|
1086
|
+
|
|
1087
|
+
assert.equal(host.chatContainer.children.length, 1, "direct low-signal tool rows should roll up on agent_end");
|
|
1088
|
+
assert.equal(host.chatContainer.children[0]?.constructor?.name, "ToolPhaseSummaryComponent");
|
|
1089
|
+
assert.match(host.chatContainer.children[0].render(120).join("\n"), /Setup \/ shell 3 actions/);
|
|
1090
|
+
});
|
|
1091
|
+
|
|
1005
1092
|
// Regression test for issue #4144: interleaved text/tool content must render in content[] index order.
|
|
1006
1093
|
// Stream: [text "A", toolCall T1, text "B", toolCall T2, text "C"]
|
|
1007
1094
|
// Expected chatContainer order: textRun(A), toolExec(T1), textRun(B), toolExec(T2), textRun(C)
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
import assert from "node:assert/strict";
|
|
2
|
+
import { describe, it, afterEach } from "node:test";
|
|
3
|
+
import { existsSync, mkdtempSync, readFileSync, readdirSync, rmSync } from "node:fs";
|
|
4
|
+
import { join } from "node:path";
|
|
5
|
+
import { tmpdir } from "node:os";
|
|
6
|
+
|
|
7
|
+
import { atomicWriteDbSnapshotSync } from "./db-snapshot.js";
|
|
8
|
+
|
|
9
|
+
describe("atomicWriteDbSnapshotSync", () => {
|
|
10
|
+
let dir: string;
|
|
11
|
+
|
|
12
|
+
afterEach(() => {
|
|
13
|
+
if (dir) {
|
|
14
|
+
rmSync(dir, { recursive: true, force: true });
|
|
15
|
+
}
|
|
16
|
+
});
|
|
17
|
+
|
|
18
|
+
it("writes the full snapshot and leaves no temp file after success", () => {
|
|
19
|
+
dir = mkdtempSync(join(tmpdir(), "gsd-db-snapshot-test-"));
|
|
20
|
+
const dbPath = join(dir, "agent.db");
|
|
21
|
+
const snapshot = new Uint8Array([0x53, 0x51, 0x4c, 0x69, 0x74, 0x65]);
|
|
22
|
+
|
|
23
|
+
atomicWriteDbSnapshotSync(dbPath, snapshot);
|
|
24
|
+
|
|
25
|
+
assert.deepEqual(readFileSync(dbPath), Buffer.from(snapshot));
|
|
26
|
+
assert.equal(existsSync(`${dbPath}.tmp`), false);
|
|
27
|
+
assert.deepEqual(
|
|
28
|
+
readdirSync(dir).filter((entry) => entry.includes("agent.db") && entry.includes(".tmp")),
|
|
29
|
+
[],
|
|
30
|
+
);
|
|
31
|
+
});
|
|
32
|
+
});
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { randomUUID } from "node:crypto";
|
|
2
|
+
import { closeSync, fsyncSync, openSync, renameSync, unlinkSync, writeFileSync } from "node:fs";
|
|
3
|
+
import { basename, dirname, join } from "node:path";
|
|
4
|
+
|
|
5
|
+
export type DbSnapshot = Buffer | Uint8Array;
|
|
6
|
+
|
|
7
|
+
function closeBestEffort(fd: number | null): void {
|
|
8
|
+
if (fd === null) return;
|
|
9
|
+
try {
|
|
10
|
+
closeSync(fd);
|
|
11
|
+
} catch {
|
|
12
|
+
// Preserve the original write/rename failure when cleaning up.
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
function fsyncDirectoryBestEffort(dirPath: string): void {
|
|
17
|
+
let fd: number | null = null;
|
|
18
|
+
try {
|
|
19
|
+
fd = openSync(dirPath, "r");
|
|
20
|
+
fsyncSync(fd);
|
|
21
|
+
} catch {
|
|
22
|
+
// Directory fsync is unsupported on some platforms/filesystems.
|
|
23
|
+
} finally {
|
|
24
|
+
closeBestEffort(fd);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
/**
|
|
29
|
+
* Persist a sql.js database export without exposing callers to a torn live file.
|
|
30
|
+
*
|
|
31
|
+
* The snapshot is written to a unique temp file in the same directory, flushed,
|
|
32
|
+
* then renamed over the target. A hard kill during the temp write leaves the
|
|
33
|
+
* previous target intact; after rename, readers see the complete new snapshot.
|
|
34
|
+
*
|
|
35
|
+
* The rename replaces the target inode: existing file mode/ownership is not
|
|
36
|
+
* preserved, and a symlink at dbPath is replaced rather than written through.
|
|
37
|
+
* Agent DB snapshots are owned runtime files, so this trade-off favors a
|
|
38
|
+
* private 0600 replacement over retaining prior target metadata.
|
|
39
|
+
*/
|
|
40
|
+
export function atomicWriteDbSnapshotSync(dbPath: string, snapshot: DbSnapshot): void {
|
|
41
|
+
const dirPath = dirname(dbPath);
|
|
42
|
+
const tmpPath = join(dirPath, `.${basename(dbPath)}.${process.pid}.${Date.now()}.${randomUUID()}.tmp`);
|
|
43
|
+
let fd: number | null = null;
|
|
44
|
+
let renamed = false;
|
|
45
|
+
|
|
46
|
+
try {
|
|
47
|
+
fd = openSync(tmpPath, "wx", 0o600);
|
|
48
|
+
writeFileSync(fd, Buffer.from(snapshot));
|
|
49
|
+
fsyncSync(fd);
|
|
50
|
+
closeSync(fd);
|
|
51
|
+
fd = null;
|
|
52
|
+
|
|
53
|
+
renameSync(tmpPath, dbPath);
|
|
54
|
+
renamed = true;
|
|
55
|
+
fsyncDirectoryBestEffort(dirPath);
|
|
56
|
+
} finally {
|
|
57
|
+
closeBestEffort(fd);
|
|
58
|
+
if (!renamed) {
|
|
59
|
+
try {
|
|
60
|
+
unlinkSync(tmpPath);
|
|
61
|
+
} catch {
|
|
62
|
+
// The temp file may not have been created yet.
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
}
|
|
@@ -102,6 +102,60 @@ describe("ExtensionRunner.emitToolCall", () => {
|
|
|
102
102
|
assert.equal(errors[0].event, "tool_call");
|
|
103
103
|
assert.equal(errors[0].extensionPath, "/test/throwing-ext");
|
|
104
104
|
});
|
|
105
|
+
|
|
106
|
+
it("preserves shutdown in tool_call handler context", async (t) => {
|
|
107
|
+
const dir = mkdtempSync(join(tmpdir(), "runner-test-"));
|
|
108
|
+
t.after(() => {
|
|
109
|
+
rmSync(dir, { recursive: true, force: true });
|
|
110
|
+
});
|
|
111
|
+
|
|
112
|
+
const sessionManager = SessionManager.create(dir, dir);
|
|
113
|
+
const authStorage = AuthStorage.create();
|
|
114
|
+
const modelRegistry = new ModelRegistry(authStorage, join(dir, "models.json"));
|
|
115
|
+
const runtime = makeMinimalRuntime();
|
|
116
|
+
let shutdownCount = 0;
|
|
117
|
+
const handlers = new Map();
|
|
118
|
+
handlers.set("tool_call", [
|
|
119
|
+
async (_event: unknown, ctx: { shutdown: () => void }) => {
|
|
120
|
+
ctx.shutdown();
|
|
121
|
+
},
|
|
122
|
+
]);
|
|
123
|
+
const extension = {
|
|
124
|
+
path: "/test/shutdown-on-tool-call",
|
|
125
|
+
handlers,
|
|
126
|
+
commands: new Map(),
|
|
127
|
+
shortcuts: new Map(),
|
|
128
|
+
tools: new Map(),
|
|
129
|
+
flags: new Map(),
|
|
130
|
+
diagnostics: [],
|
|
131
|
+
} as unknown as Extension;
|
|
132
|
+
const runner = new ExtensionRunner([extension], runtime, dir, sessionManager, modelRegistry);
|
|
133
|
+
runner.bindCore({} as any, {
|
|
134
|
+
getModel: () => undefined,
|
|
135
|
+
isIdle: () => true,
|
|
136
|
+
abort: () => {},
|
|
137
|
+
hasPendingMessages: () => false,
|
|
138
|
+
shutdown: () => {
|
|
139
|
+
shutdownCount += 1;
|
|
140
|
+
},
|
|
141
|
+
getContextUsage: () => undefined,
|
|
142
|
+
compact: () => {},
|
|
143
|
+
getSystemPrompt: () => "",
|
|
144
|
+
});
|
|
145
|
+
|
|
146
|
+
const errors: any[] = [];
|
|
147
|
+
runner.onError((err) => errors.push(err));
|
|
148
|
+
|
|
149
|
+
await runner.emitToolCall({
|
|
150
|
+
type: "tool_call",
|
|
151
|
+
toolCallId: "test-123",
|
|
152
|
+
toolName: "test_tool",
|
|
153
|
+
input: {},
|
|
154
|
+
} as ToolCallEvent);
|
|
155
|
+
|
|
156
|
+
assert.equal(shutdownCount, 1);
|
|
157
|
+
assert.equal(errors.length, 0);
|
|
158
|
+
});
|
|
105
159
|
});
|
|
106
160
|
|
|
107
161
|
describe("ExtensionRunner.createContext", () => {
|
|
@@ -127,6 +181,60 @@ describe("ExtensionRunner.createContext", () => {
|
|
|
127
181
|
assert.equal(runner.createContext().cwd, realProjectDir);
|
|
128
182
|
assert.equal(runner.createCommandContext().cwd, realProjectDir);
|
|
129
183
|
});
|
|
184
|
+
|
|
185
|
+
it("does not let lifecycle event handlers close the TUI", async (t) => {
|
|
186
|
+
const dir = mkdtempSync(join(tmpdir(), "runner-test-"));
|
|
187
|
+
t.after(() => {
|
|
188
|
+
rmSync(dir, { recursive: true, force: true });
|
|
189
|
+
});
|
|
190
|
+
|
|
191
|
+
const sessionManager = SessionManager.create(dir, dir);
|
|
192
|
+
const authStorage = AuthStorage.create();
|
|
193
|
+
const modelRegistry = new ModelRegistry(authStorage, join(dir, "models.json"));
|
|
194
|
+
const runtime = makeMinimalRuntime();
|
|
195
|
+
let shutdownCount = 0;
|
|
196
|
+
const handlers = new Map();
|
|
197
|
+
handlers.set("agent_end", [
|
|
198
|
+
async (_event: unknown, ctx: { shutdown: () => void }) => {
|
|
199
|
+
ctx.shutdown();
|
|
200
|
+
},
|
|
201
|
+
]);
|
|
202
|
+
const extension = {
|
|
203
|
+
path: "/test/shutdown-on-agent-end",
|
|
204
|
+
handlers,
|
|
205
|
+
commands: new Map(),
|
|
206
|
+
shortcuts: new Map(),
|
|
207
|
+
tools: new Map(),
|
|
208
|
+
flags: new Map(),
|
|
209
|
+
diagnostics: [],
|
|
210
|
+
} as unknown as Extension;
|
|
211
|
+
const runner = new ExtensionRunner([extension], runtime, dir, sessionManager, modelRegistry);
|
|
212
|
+
runner.bindCore({} as any, {
|
|
213
|
+
getModel: () => undefined,
|
|
214
|
+
isIdle: () => true,
|
|
215
|
+
abort: () => {},
|
|
216
|
+
hasPendingMessages: () => false,
|
|
217
|
+
shutdown: () => {
|
|
218
|
+
shutdownCount += 1;
|
|
219
|
+
},
|
|
220
|
+
getContextUsage: () => undefined,
|
|
221
|
+
compact: () => {},
|
|
222
|
+
getSystemPrompt: () => "",
|
|
223
|
+
});
|
|
224
|
+
|
|
225
|
+
const errors: any[] = [];
|
|
226
|
+
runner.onError((err) => errors.push(err));
|
|
227
|
+
|
|
228
|
+
await runner.emit({ type: "agent_end", messages: [] } as any);
|
|
229
|
+
|
|
230
|
+
assert.equal(shutdownCount, 0);
|
|
231
|
+
assert.equal(errors.length, 1);
|
|
232
|
+
assert.equal(errors[0].event, "agent_end");
|
|
233
|
+
assert.match(errors[0].error, /cannot request TUI shutdown/);
|
|
234
|
+
|
|
235
|
+
runner.createCommandContext().shutdown();
|
|
236
|
+
assert.equal(shutdownCount, 1);
|
|
237
|
+
});
|
|
130
238
|
});
|
|
131
239
|
|
|
132
240
|
describe("ExtensionRunner protected commands", () => {
|
|
@@ -717,6 +717,19 @@ export class ExtensionRunner {
|
|
|
717
717
|
};
|
|
718
718
|
}
|
|
719
719
|
|
|
720
|
+
private createEventContext(eventType: string): ExtensionContext {
|
|
721
|
+
return {
|
|
722
|
+
...this.createContext(),
|
|
723
|
+
shutdown: () => {
|
|
724
|
+
throw new Error(`Extension event '${eventType}' cannot request TUI shutdown`);
|
|
725
|
+
},
|
|
726
|
+
};
|
|
727
|
+
}
|
|
728
|
+
|
|
729
|
+
private isShutdownGuardedEvent(eventType: string): boolean {
|
|
730
|
+
return eventType === "agent_end" || eventType === "stop" || eventType === "session_end";
|
|
731
|
+
}
|
|
732
|
+
|
|
720
733
|
createCommandContext(): ExtensionCommandContext {
|
|
721
734
|
return {
|
|
722
735
|
...this.createContext(),
|
|
@@ -757,7 +770,9 @@ export class ExtensionRunner {
|
|
|
757
770
|
getEvent: () => unknown,
|
|
758
771
|
processResult: (handlerResult: unknown, extensionPath: string) => { done: boolean },
|
|
759
772
|
): Promise<void> {
|
|
760
|
-
const ctx = this.
|
|
773
|
+
const ctx = this.isShutdownGuardedEvent(eventType)
|
|
774
|
+
? this.createEventContext(eventType)
|
|
775
|
+
: this.createContext();
|
|
761
776
|
|
|
762
777
|
for (const ext of this.extensions) {
|
|
763
778
|
const handlers = ext.handlers.get(eventType);
|
|
@@ -578,6 +578,10 @@ export class ModelRegistry {
|
|
|
578
578
|
* Defaults to "apiKey" for built-ins and providers without explicit mode.
|
|
579
579
|
*/
|
|
580
580
|
getProviderAuthMode(provider: string): ProviderAuthMode {
|
|
581
|
+
// E2E-test-only: the fake provider is keyless. Sentinel is project-
|
|
582
|
+
// internal ("gsd-fake") so it cannot collide with a real provider.
|
|
583
|
+
// See packages/pi-ai/src/providers/fake.ts.
|
|
584
|
+
if (provider === "gsd-fake") return "none";
|
|
581
585
|
const config = this.registeredProviders.get(provider);
|
|
582
586
|
if (!config) return "apiKey";
|
|
583
587
|
if (config.authMode) return config.authMode;
|
|
@@ -32,6 +32,7 @@ export interface RetrySettings {
|
|
|
32
32
|
export interface TerminalSettings {
|
|
33
33
|
showImages?: boolean; // default: true (only relevant if terminal supports images)
|
|
34
34
|
clearOnShrink?: boolean; // default: false (clear empty rows when content shrinks)
|
|
35
|
+
adaptiveMode?: AdaptiveTuiMode; // default: "auto"
|
|
35
36
|
}
|
|
36
37
|
|
|
37
38
|
export interface ImageSettings {
|
|
@@ -149,6 +150,7 @@ export interface HooksSettings {
|
|
|
149
150
|
}
|
|
150
151
|
|
|
151
152
|
export type TransportSetting = Transport;
|
|
153
|
+
export type AdaptiveTuiMode = "auto" | "chat" | "workflow" | "validation" | "debug" | "compact";
|
|
152
154
|
|
|
153
155
|
/**
|
|
154
156
|
* Package source for npm/git packages.
|
|
@@ -978,6 +980,16 @@ export class SettingsManager {
|
|
|
978
980
|
this.setNestedGlobalSetting("terminal", "clearOnShrink", enabled);
|
|
979
981
|
}
|
|
980
982
|
|
|
983
|
+
getAdaptiveMode(): AdaptiveTuiMode {
|
|
984
|
+
const mode = this.settings.terminal?.adaptiveMode;
|
|
985
|
+
const valid: AdaptiveTuiMode[] = ["auto", "chat", "workflow", "validation", "debug", "compact"];
|
|
986
|
+
return mode && valid.includes(mode) ? mode : "auto";
|
|
987
|
+
}
|
|
988
|
+
|
|
989
|
+
setAdaptiveMode(mode: AdaptiveTuiMode): void {
|
|
990
|
+
this.setNestedGlobalSetting("terminal", "adaptiveMode", mode);
|
|
991
|
+
}
|
|
992
|
+
|
|
981
993
|
getImageAutoResize(): boolean {
|
|
982
994
|
return this.settings.images?.autoResize ?? true;
|
|
983
995
|
}
|
|
@@ -38,5 +38,6 @@ export const BUILTIN_SLASH_COMMANDS: ReadonlyArray<BuiltinSlashCommand> = [
|
|
|
38
38
|
{ name: "thinking", description: "Set thinking level (off/minimal/low/medium/high/xhigh)" },
|
|
39
39
|
{ name: "edit-mode", description: "Toggle edit mode (standard/hashline)" },
|
|
40
40
|
{ name: "terminal", description: "Run a shell command directly (e.g. /terminal ping -c3 1.1.1.1)" },
|
|
41
|
+
{ name: "tui", description: "Configure TUI layout mode (e.g. /tui mode workflow)" },
|
|
41
42
|
{ name: "quit", description: "Quit pi" },
|
|
42
43
|
];
|
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
// GSD2 TUI Tests - Chat frame card visual contract coverage.
|
|
1
2
|
import { test, describe } from "node:test";
|
|
2
3
|
import assert from "node:assert/strict";
|
|
3
4
|
import stripAnsi from "strip-ansi";
|
|
@@ -8,12 +9,12 @@ initTheme("dark", false);
|
|
|
8
9
|
|
|
9
10
|
// Regression tests for the "compaction" tone added to renderChatFrame.
|
|
10
11
|
// The compaction notice shares the same visual frame as user / assistant
|
|
11
|
-
// messages (top rule,
|
|
12
|
+
// messages (top rule, label header, `│ ` body prefix) but uses the
|
|
12
13
|
// purple `customMessageLabel` color key so it is visually distinct from
|
|
13
14
|
// conversation turns.
|
|
14
15
|
|
|
15
16
|
describe("renderChatFrame — compaction tone", () => {
|
|
16
|
-
test("produces a top rule,
|
|
17
|
+
test("produces a top rule, compaction header row, and a │ body margin", () => {
|
|
17
18
|
const lines = renderChatFrame(
|
|
18
19
|
["Compacted from 1,224,262 tokens (ctrl+o to expand)"],
|
|
19
20
|
60,
|
|
@@ -33,11 +34,12 @@ describe("renderChatFrame — compaction tone", () => {
|
|
|
33
34
|
// Top rule is a solid horizontal bar
|
|
34
35
|
assert.match(plain[0], /^─+$/, "first line should be the solid top rule");
|
|
35
36
|
|
|
36
|
-
// Header row contains
|
|
37
|
+
// Header row contains `compaction`
|
|
37
38
|
assert.ok(
|
|
38
|
-
plain[1].includes("
|
|
39
|
-
`expected header to contain "
|
|
39
|
+
plain[1].includes("compaction"),
|
|
40
|
+
`expected header to contain "compaction", got ${JSON.stringify(plain[1])}`,
|
|
40
41
|
);
|
|
42
|
+
assert.ok(!plain[1].includes("•"), `header should not render a bullet prefix, got ${JSON.stringify(plain[1])}`);
|
|
41
43
|
|
|
42
44
|
// Body line(s) start with `│ `
|
|
43
45
|
assert.ok(
|