@ai-setting/roy-agent-core 1.3.10 → 1.3.14
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/config/index.js +1647 -0
- package/dist/index.js +12579 -89691
- package/package.json +19 -56
- package/src/config/config-component.test.ts +0 -627
- package/src/config/config-component.ts +0 -906
- package/src/config/config-parser.test.ts +0 -319
- package/src/config/config-parser.ts +0 -203
- package/src/config/decentralized-config.test.ts +0 -740
- package/src/config/env-key.ts +0 -210
- package/src/config/env-source.test.ts +0 -252
- package/src/config/env-source.ts +0 -301
- package/src/config/file-source.test.ts +0 -357
- package/src/config/file-source.ts +0 -421
- package/src/config/index.ts +0 -24
- package/src/config/protocol-resolver.test.ts +0 -217
- package/src/config/protocol-resolver.ts +0 -228
- package/src/env/agent/agent-component.abort.test.ts +0 -511
- package/src/env/agent/agent-component.record-session.test.ts +0 -349
- package/src/env/agent/agent-component.test.ts +0 -1389
- package/src/env/agent/agent-component.tool-error.test.ts +0 -327
- package/src/env/agent/agent-component.ts +0 -1711
- package/src/env/agent/agent-config-registration.test.ts +0 -226
- package/src/env/agent/agent-config-registration.ts +0 -46
- package/src/env/agent/agent-reminder-plugin.integration.test.ts +0 -243
- package/src/env/agent/index.ts +0 -10
- package/src/env/agent/summary-agent.parse-hint.test.ts +0 -360
- package/src/env/agent/summary-agent.ts +0 -508
- package/src/env/agent/types.ts +0 -536
- package/src/env/commands/commands-component.test.ts +0 -364
- package/src/env/commands/commands-component.ts +0 -604
- package/src/env/commands/commands-config-registration.test.ts +0 -198
- package/src/env/commands/commands-config-registration.ts +0 -38
- package/src/env/commands/index.ts +0 -21
- package/src/env/commands/parser.test.ts +0 -203
- package/src/env/commands/parser.ts +0 -115
- package/src/env/commands/types.ts +0 -184
- package/src/env/commands-prompt-integration.test.ts +0 -243
- package/src/env/component-env.test.ts +0 -119
- package/src/env/component.ts +0 -335
- package/src/env/constants.test.ts +0 -72
- package/src/env/constants.ts +0 -123
- package/src/env/debug/debug-component.test.ts +0 -114
- package/src/env/debug/debug-component.ts +0 -547
- package/src/env/debug/formatters/index.ts +0 -9
- package/src/env/debug/formatters/repl-formatter.test.ts +0 -139
- package/src/env/debug/formatters/repl-formatter.ts +0 -358
- package/src/env/debug/formatters/trace-formatter.test.ts +0 -119
- package/src/env/debug/formatters/trace-formatter.ts +0 -191
- package/src/env/debug/formatters/tree-formatter.test.ts +0 -107
- package/src/env/debug/formatters/tree-formatter.ts +0 -325
- package/src/env/debug/index.ts +0 -38
- package/src/env/debug/parser/regex-parser.test.ts +0 -201
- package/src/env/debug/parser/regex-parser.ts +0 -196
- package/src/env/debug/parser/span-builder.test.ts +0 -241
- package/src/env/debug/parser/span-builder.ts +0 -386
- package/src/env/debug/reader/log-reader.test.ts +0 -170
- package/src/env/debug/reader/log-reader.ts +0 -186
- package/src/env/debug/reader/span-db-reader.test.ts +0 -118
- package/src/env/debug/reader/span-db-reader.ts +0 -201
- package/src/env/debug/types.test.ts +0 -187
- package/src/env/debug/types.ts +0 -171
- package/src/env/environment-init.test.ts +0 -183
- package/src/env/environment-lifecycle.test.ts +0 -516
- package/src/env/environment-service.test.ts +0 -332
- package/src/env/environment.handle-query.test.ts +0 -96
- package/src/env/environment.test.ts +0 -232
- package/src/env/environment.ts +0 -708
- package/src/env/errors.test.ts +0 -165
- package/src/env/errors.ts +0 -157
- package/src/env/event-source/event-source-agent-handler.test.ts +0 -193
- package/src/env/event-source/event-source-agent-handler.ts +0 -111
- package/src/env/event-source/event-source-component.process-cleanup.test.ts +0 -236
- package/src/env/event-source/event-source-component.stop.test.ts +0 -346
- package/src/env/event-source/event-source-component.test.ts +0 -1207
- package/src/env/event-source/event-source-component.ts +0 -1379
- package/src/env/event-source/event-source-config-registration.test.ts +0 -242
- package/src/env/event-source/event-source-config-registration.ts +0 -37
- package/src/env/event-source/event-source-integration.test.ts +0 -320
- package/src/env/event-source/event-source-platform.test.ts +0 -630
- package/src/env/event-source/types.ts +0 -298
- package/src/env/hook/global-hook-manager.ts +0 -162
- package/src/env/hook/hook-manager.test.ts +0 -374
- package/src/env/hook/hook-manager.ts +0 -309
- package/src/env/hook/index.ts +0 -38
- package/src/env/hook/types.ts +0 -138
- package/src/env/index.ts +0 -144
- package/src/env/interface.ts +0 -203
- package/src/env/llm/hooks.test.ts +0 -293
- package/src/env/llm/hooks.ts +0 -316
- package/src/env/llm/index.ts +0 -61
- package/src/env/llm/invoke-threshold-check.test.ts +0 -88
- package/src/env/llm/invoke-timeout.test.ts +0 -54
- package/src/env/llm/invoke.test.ts +0 -71
- package/src/env/llm/invoke.ts +0 -1039
- package/src/env/llm/llm-config.test.ts +0 -523
- package/src/env/llm/llm.test.ts +0 -233
- package/src/env/llm/llm.ts +0 -568
- package/src/env/llm/provider.test.ts +0 -182
- package/src/env/llm/provider.ts +0 -108
- package/src/env/llm/transform.test.ts +0 -251
- package/src/env/llm/transform.ts +0 -286
- package/src/env/llm/types.test.ts +0 -580
- package/src/env/llm/types.ts +0 -424
- package/src/env/log-trace/decorator-otel.test.ts +0 -182
- package/src/env/log-trace/decorator.ts +0 -230
- package/src/env/log-trace/index.ts +0 -79
- package/src/env/log-trace/log-trace-component.test.ts +0 -242
- package/src/env/log-trace/log-trace-component.ts +0 -497
- package/src/env/log-trace/log-trace-config-registration.test.ts +0 -348
- package/src/env/log-trace/log-trace-config-registration.ts +0 -45
- package/src/env/log-trace/logger.test.ts +0 -149
- package/src/env/log-trace/logger.ts +0 -522
- package/src/env/log-trace/opentelemetry/cli-propagation.test.ts +0 -147
- package/src/env/log-trace/opentelemetry/cli-propagation.ts +0 -194
- package/src/env/log-trace/opentelemetry/integration.test.ts +0 -668
- package/src/env/log-trace/opentelemetry/mod.ts +0 -25
- package/src/env/log-trace/opentelemetry/propagation-env.test.ts +0 -181
- package/src/env/log-trace/opentelemetry/propagation-env.ts +0 -136
- package/src/env/log-trace/opentelemetry/propagation.test.ts +0 -259
- package/src/env/log-trace/opentelemetry/propagation.ts +0 -215
- package/src/env/log-trace/opentelemetry/tracer-provider-context.test.ts +0 -166
- package/src/env/log-trace/opentelemetry/tracer-provider.test.ts +0 -379
- package/src/env/log-trace/opentelemetry/tracer-provider.ts +0 -612
- package/src/env/log-trace/span-storage.test.ts +0 -145
- package/src/env/log-trace/span-storage.ts +0 -230
- package/src/env/log-trace/trace-context.test.ts +0 -187
- package/src/env/log-trace/trace-context.ts +0 -162
- package/src/env/log-trace/types.test.ts +0 -63
- package/src/env/log-trace/types.ts +0 -172
- package/src/env/mcp/README.md +0 -244
- package/src/env/mcp/__integration__/mcp-component.integration.test.ts +0 -373
- package/src/env/mcp/config.test.ts +0 -74
- package/src/env/mcp/config.ts +0 -116
- package/src/env/mcp/index.ts +0 -41
- package/src/env/mcp/loader.test.ts +0 -161
- package/src/env/mcp/loader.ts +0 -209
- package/src/env/mcp/mcp-component.test.ts +0 -111
- package/src/env/mcp/mcp-component.ts +0 -358
- package/src/env/mcp/mcp-config-registration.test.ts +0 -304
- package/src/env/mcp/mcp-config-registration.ts +0 -50
- package/src/env/mcp/scanner.test.ts +0 -170
- package/src/env/mcp/scanner.ts +0 -246
- package/src/env/mcp/tool/adapter.test.ts +0 -520
- package/src/env/mcp/tool/adapter.ts +0 -521
- package/src/env/mcp/tool/index.ts +0 -5
- package/src/env/mcp/types.test.ts +0 -171
- package/src/env/mcp/types.ts +0 -79
- package/src/env/memory/README.md +0 -177
- package/src/env/memory/built-in/index.ts +0 -59
- package/src/env/memory/built-in/recall-memory.ts +0 -103
- package/src/env/memory/built-in/record-memory.ts +0 -148
- package/src/env/memory/index.ts +0 -20
- package/src/env/memory/memory-component.test.ts +0 -239
- package/src/env/memory/memory-component.ts +0 -503
- package/src/env/memory/memory-config-registration.test.ts +0 -67
- package/src/env/memory/memory-config-registration.ts +0 -48
- package/src/env/memory/memory-config.ts +0 -45
- package/src/env/memory/memory-file.test.ts +0 -268
- package/src/env/memory/plugin/index.ts +0 -48
- package/src/env/memory/plugin/memory-agent.test.ts +0 -249
- package/src/env/memory/plugin/memory-agent.ts +0 -365
- package/src/env/memory/plugin/memory-manager.ts +0 -198
- package/src/env/memory/plugin/memory-plugin-agent.test.ts +0 -145
- package/src/env/memory/plugin/memory-plugin.ts +0 -210
- package/src/env/memory/plugin/plugin-simplified.test.ts +0 -51
- package/src/env/memory/plugin/recall-memory.test.ts +0 -106
- package/src/env/memory/plugin/recall-memory.ts +0 -53
- package/src/env/memory/plugin/types.ts +0 -101
- package/src/env/memory/tools/memory-agent-tools.ts +0 -228
- package/src/env/memory/types.ts +0 -85
- package/src/env/paths.ts +0 -118
- package/src/env/prompt/index.ts +0 -18
- package/src/env/prompt/memory-prompts.test.ts +0 -91
- package/src/env/prompt/prompt-component.test.ts +0 -491
- package/src/env/prompt/prompt-component.ts +0 -619
- package/src/env/prompt/prompt-config-registration.test.ts +0 -213
- package/src/env/prompt/prompt-config-registration.ts +0 -39
- package/src/env/prompt/prompts-index.ts +0 -504
- package/src/env/prompt/renderer.ts +0 -67
- package/src/env/prompt/types.ts +0 -136
- package/src/env/session/hooks.ts +0 -18
- package/src/env/session/index.ts +0 -37
- package/src/env/session/search-query-parser.test.ts +0 -425
- package/src/env/session/search-query-parser.ts +0 -171
- package/src/env/session/session-checkpoint.test.ts +0 -523
- package/src/env/session/session-component.extract-recent-messages.test.ts +0 -209
- package/src/env/session/session-component.test.ts +0 -132
- package/src/env/session/session-component.ts +0 -1249
- package/src/env/session/session-config-registration.test.ts +0 -138
- package/src/env/session/session-config-registration.ts +0 -52
- package/src/env/session/session-message-converter.test.ts +0 -763
- package/src/env/session/session-message-converter.ts +0 -415
- package/src/env/session/session-message-e2e.test.ts +0 -448
- package/src/env/session/session-search.test.ts +0 -391
- package/src/env/session/session-store.test.ts +0 -362
- package/src/env/session/session-store.ts +0 -141
- package/src/env/session/storage/index.ts +0 -6
- package/src/env/session/storage/memory.ts +0 -502
- package/src/env/session/storage/sqlite.ts +0 -794
- package/src/env/session/types.ts +0 -742
- package/src/env/skill/config.ts +0 -39
- package/src/env/skill/index.ts +0 -6
- package/src/env/skill/parser.test.ts +0 -116
- package/src/env/skill/parser.ts +0 -77
- package/src/env/skill/scanner.test.ts +0 -211
- package/src/env/skill/scanner.ts +0 -119
- package/src/env/skill/skill-component.test.ts +0 -234
- package/src/env/skill/skill-component.ts +0 -352
- package/src/env/skill/skill-config-registration.test.ts +0 -60
- package/src/env/skill/skill-config-registration.ts +0 -43
- package/src/env/skill/tool/index.ts +0 -1
- package/src/env/skill/tool/skill-tool.test.ts +0 -100
- package/src/env/skill/tool/skill-tool.ts +0 -72
- package/src/env/skill/types.ts +0 -64
- package/src/env/task/delegate/delegate-tool.test.ts +0 -498
- package/src/env/task/delegate/delegate-tool.ts +0 -1014
- package/src/env/task/delegate/index.ts +0 -18
- package/src/env/task/delegate/stop-tool.test.ts +0 -140
- package/src/env/task/delegate/stop-tool.ts +0 -119
- package/src/env/task/delegate/task-events.test.ts +0 -178
- package/src/env/task/delegate/task-events.ts +0 -143
- package/src/env/task/hooks/contexts.test.ts +0 -92
- package/src/env/task/hooks/contexts.ts +0 -192
- package/src/env/task/hooks/index.ts +0 -23
- package/src/env/task/hooks/task-hook-points.test.ts +0 -32
- package/src/env/task/hooks/task-hook-points.ts +0 -54
- package/src/env/task/index.ts +0 -7
- package/src/env/task/plugins/index.ts +0 -13
- package/src/env/task/plugins/task-plugin.test.ts +0 -74
- package/src/env/task/plugins/task-plugin.ts +0 -89
- package/src/env/task/plugins/task-tag-plugin.test.ts +0 -377
- package/src/env/task/plugins/task-tag-plugin.ts +0 -319
- package/src/env/task/plugins/task-workflow-extractor.integration.test.ts +0 -226
- package/src/env/task/plugins/workflow-extractor-agent.test.ts +0 -107
- package/src/env/task/plugins/workflow-extractor-agent.ts +0 -225
- package/src/env/task/storage/index.ts +0 -6
- package/src/env/task/storage/sqlite-task-store.test.ts +0 -283
- package/src/env/task/storage/sqlite-task-store.ts +0 -903
- package/src/env/task/storage/task-search.test.ts +0 -291
- package/src/env/task/tag-service.test.ts +0 -198
- package/src/env/task/tag-service.ts +0 -264
- package/src/env/task/task-component.test.ts +0 -193
- package/src/env/task/task-component.ts +0 -658
- package/src/env/task/task-config-registration.test.ts +0 -57
- package/src/env/task/task-config-registration.ts +0 -37
- package/src/env/task/task-types.test.ts +0 -137
- package/src/env/task/tools/complete-tool.ts +0 -44
- package/src/env/task/tools/create-tool.ts +0 -49
- package/src/env/task/tools/delete-tool.ts +0 -43
- package/src/env/task/tools/get-tool.ts +0 -59
- package/src/env/task/tools/index.ts +0 -10
- package/src/env/task/tools/list-tool.ts +0 -40
- package/src/env/task/tools/operation/create-tool.ts +0 -48
- package/src/env/task/tools/operation/delete-tool.ts +0 -43
- package/src/env/task/tools/operation/get-tool.ts +0 -43
- package/src/env/task/tools/operation/index.ts +0 -9
- package/src/env/task/tools/operation/list-tool.ts +0 -40
- package/src/env/task/tools/operation/operation-tools.test.ts +0 -274
- package/src/env/task/tools/operation/operation-types.ts +0 -75
- package/src/env/task/tools/operation/update-tool.ts +0 -47
- package/src/env/task/tools/task-tools.test.ts +0 -203
- package/src/env/task/tools/task-types.test.ts +0 -75
- package/src/env/task/tools/task-types.ts +0 -68
- package/src/env/task/tools/update-tool.ts +0 -70
- package/src/env/task/types.ts +0 -160
- package/src/env/tool/built-in/bash.ts +0 -201
- package/src/env/tool/built-in/echo.ts +0 -29
- package/src/env/tool/built-in/edit-file.test.ts +0 -136
- package/src/env/tool/built-in/edit-file.ts +0 -92
- package/src/env/tool/built-in/glob.test.ts +0 -94
- package/src/env/tool/built-in/glob.ts +0 -65
- package/src/env/tool/built-in/grep.test.ts +0 -122
- package/src/env/tool/built-in/grep.ts +0 -108
- package/src/env/tool/built-in/index.ts +0 -44
- package/src/env/tool/built-in/read-file.test.ts +0 -84
- package/src/env/tool/built-in/read-file.ts +0 -75
- package/src/env/tool/built-in/write-file.test.ts +0 -119
- package/src/env/tool/built-in/write-file.ts +0 -68
- package/src/env/tool/index.ts +0 -24
- package/src/env/tool/registry.test.ts +0 -257
- package/src/env/tool/registry.ts +0 -167
- package/src/env/tool/tool-component.test.ts +0 -559
- package/src/env/tool/tool-component.ts +0 -563
- package/src/env/tool/tool-config-registration.test.ts +0 -249
- package/src/env/tool/tool-config-registration.ts +0 -46
- package/src/env/tool/types.ts +0 -267
- package/src/env/tool/validator.test.ts +0 -143
- package/src/env/tool/validator.ts +0 -44
- package/src/env/types.ts +0 -180
- package/src/env/workflow/ask-user-tool-registration.test.ts +0 -216
- package/src/env/workflow/complex-workflow.integration.test.ts +0 -1900
- package/src/env/workflow/decorators/decorator-node.ts +0 -229
- package/src/env/workflow/decorators/decorator.test.ts +0 -196
- package/src/env/workflow/decorators/edge.ts +0 -82
- package/src/env/workflow/decorators/index.ts +0 -31
- package/src/env/workflow/decorators/node-as.ts +0 -98
- package/src/env/workflow/decorators/workflow.ts +0 -54
- package/src/env/workflow/engine/dag-manager.test.ts +0 -570
- package/src/env/workflow/engine/dag-manager.ts +0 -594
- package/src/env/workflow/engine/engine.ts +0 -1422
- package/src/env/workflow/engine/event-bus.test.ts +0 -359
- package/src/env/workflow/engine/event-bus.ts +0 -156
- package/src/env/workflow/engine/executor-agent-session.test.ts +0 -84
- package/src/env/workflow/engine/executor.test.ts +0 -619
- package/src/env/workflow/engine/executor.ts +0 -593
- package/src/env/workflow/engine/index.ts +0 -24
- package/src/env/workflow/engine/node-registry.test.ts +0 -560
- package/src/env/workflow/engine/node-registry.ts +0 -289
- package/src/env/workflow/engine/resume-removed.test.ts +0 -22
- package/src/env/workflow/engine/scheduler.test.ts +0 -715
- package/src/env/workflow/engine/scheduler.ts +0 -318
- package/src/env/workflow/engine/workflow-engine.test.ts +0 -815
- package/src/env/workflow/extractor/workflow-converter.ts +0 -306
- package/src/env/workflow/fixtures.ts +0 -380
- package/src/env/workflow/index.ts +0 -38
- package/src/env/workflow/integration/run-resume-unified.test.ts +0 -186
- package/src/env/workflow/integration/service-integration.test.ts +0 -267
- package/src/env/workflow/metadata/keys.ts +0 -12
- package/src/env/workflow/nodes/agent-component-adapter.test.ts +0 -318
- package/src/env/workflow/nodes/agent-component-adapter.ts +0 -448
- package/src/env/workflow/nodes/agent-node.test.ts +0 -371
- package/src/env/workflow/nodes/agent-node.ts +0 -598
- package/src/env/workflow/nodes/ask-user-node.ts +0 -113
- package/src/env/workflow/nodes/condition-node.ts +0 -200
- package/src/env/workflow/nodes/index.ts +0 -9
- package/src/env/workflow/nodes/merge-node.ts +0 -141
- package/src/env/workflow/nodes/skill-node.test.ts +0 -253
- package/src/env/workflow/nodes/skill-node.ts +0 -393
- package/src/env/workflow/nodes/tool-node.test.ts +0 -251
- package/src/env/workflow/nodes/tool-node.ts +0 -493
- package/src/env/workflow/nodes/workflow-llm-history.test.ts +0 -455
- package/src/env/workflow/nodes/workflow-node.test.ts +0 -315
- package/src/env/workflow/nodes/workflow-node.ts +0 -311
- package/src/env/workflow/service/index.ts +0 -27
- package/src/env/workflow/service/registry.test.ts +0 -133
- package/src/env/workflow/service/registry.ts +0 -71
- package/src/env/workflow/service/workflow-service.test.ts +0 -310
- package/src/env/workflow/service/workflow-service.ts +0 -393
- package/src/env/workflow/storage/index.ts +0 -28
- package/src/env/workflow/storage/mock-repositories.ts +0 -385
- package/src/env/workflow/storage/sqlite.test.ts +0 -179
- package/src/env/workflow/storage/sqlite.ts +0 -163
- package/src/env/workflow/storage/workflow-repo.test.ts +0 -780
- package/src/env/workflow/storage/workflow-repo.ts +0 -342
- package/src/env/workflow/tools/ask-user-tool.ts +0 -82
- package/src/env/workflow/tools/index.ts +0 -26
- package/src/env/workflow/tools/run-workflow.test.ts +0 -352
- package/src/env/workflow/tools/run-workflow.ts +0 -214
- package/src/env/workflow/types/context.ts +0 -18
- package/src/env/workflow/types/decorators-types.ts +0 -198
- package/src/env/workflow/types/event.test.ts +0 -515
- package/src/env/workflow/types/event.ts +0 -193
- package/src/env/workflow/types/index.ts +0 -49
- package/src/env/workflow/types/run.test.ts +0 -437
- package/src/env/workflow/types/run.ts +0 -173
- package/src/env/workflow/types/workflow-hil.ts +0 -114
- package/src/env/workflow/types/workflow-message.test.ts +0 -138
- package/src/env/workflow/types/workflow-message.ts +0 -196
- package/src/env/workflow/types/workflow-session.test.ts +0 -95
- package/src/env/workflow/types/workflow-session.ts +0 -59
- package/src/env/workflow/types/workflow.test.ts +0 -495
- package/src/env/workflow/types/workflow.ts +0 -195
- package/src/env/workflow/types_compat.ts +0 -51
- package/src/env/workflow/utils/create-workflow.ts +0 -47
- package/src/env/workflow/utils/execution-state.ts +0 -245
- package/src/env/workflow/utils/index.ts +0 -18
- package/src/env/workflow/utils/node-registry-helper.ts +0 -58
- package/src/env/workflow/utils/recovery-validator.test.ts +0 -460
- package/src/env/workflow/utils/recovery-validator.ts +0 -377
- package/src/env/workflow/utils/session-parser.test.ts +0 -111
- package/src/env/workflow/utils/session-parser.ts +0 -94
- package/src/env/workflow/utils/session-recovery.test.ts +0 -334
- package/src/env/workflow/utils/session-recovery.ts +0 -188
- package/src/env/workflow/utils/template-resolver.test.ts +0 -258
- package/src/env/workflow/utils/template-resolver.ts +0 -436
- package/src/env/workflow/utils/validation-rules.ts +0 -149
- package/src/env/workflow/workflow-component.ts +0 -544
- package/src/index.ts +0 -422
- package/src/utils/id.ts +0 -21
|
@@ -1,794 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @fileoverview SQLite Session Store
|
|
3
|
-
*
|
|
4
|
-
* SQLite-based implementation of SessionStore with checkpoint support.
|
|
5
|
-
* Uses Bun's built-in bun:sqlite module for persistence.
|
|
6
|
-
*/
|
|
7
|
-
|
|
8
|
-
import { join } from "path";
|
|
9
|
-
import { randomUUID } from "crypto";
|
|
10
|
-
import type {
|
|
11
|
-
Session,
|
|
12
|
-
CreateSessionOptions,
|
|
13
|
-
UpdateSessionOptions,
|
|
14
|
-
ListSessionsOptions,
|
|
15
|
-
SessionMessage,
|
|
16
|
-
SessionCheckpoint,
|
|
17
|
-
GetMessagesOptions,
|
|
18
|
-
SessionCheckpointsMeta,
|
|
19
|
-
SessionMetadata,
|
|
20
|
-
SearchMessagesOptions,
|
|
21
|
-
MessageMatch,
|
|
22
|
-
} from "../types";
|
|
23
|
-
import type { SessionStore } from "../session-store";
|
|
24
|
-
import { SessionMessageConverter } from "../session-message-converter";
|
|
25
|
-
|
|
26
|
-
// ============================================================================
|
|
27
|
-
// Helpers
|
|
28
|
-
// ============================================================================
|
|
29
|
-
|
|
30
|
-
/**
|
|
31
|
-
* Generate a unique session ID using crypto UUID
|
|
32
|
-
*/
|
|
33
|
-
function generateSessionId(): string {
|
|
34
|
-
return `session_${randomUUID()}`;
|
|
35
|
-
}
|
|
36
|
-
|
|
37
|
-
/**
|
|
38
|
-
* Generate a unique message ID using crypto UUID
|
|
39
|
-
*/
|
|
40
|
-
function generateMessageId(): string {
|
|
41
|
-
return `msg_${randomUUID()}`;
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
/**
|
|
45
|
-
* Get default data directory for session storage
|
|
46
|
-
*/
|
|
47
|
-
function getDefaultDataDir(): string {
|
|
48
|
-
const home = process.env.HOME || process.env.USERPROFILE || "/tmp";
|
|
49
|
-
return join(home, ".local", "share");
|
|
50
|
-
}
|
|
51
|
-
|
|
52
|
-
/**
|
|
53
|
-
* Get default database path for session storage
|
|
54
|
-
*/
|
|
55
|
-
function getDefaultSessionDbPath(): string {
|
|
56
|
-
return join(getDefaultDataDir(), "roy-agent", "sessions.db");
|
|
57
|
-
}
|
|
58
|
-
|
|
59
|
-
// ============================================================================
|
|
60
|
-
// SQLiteSessionStore
|
|
61
|
-
// ============================================================================
|
|
62
|
-
|
|
63
|
-
/**
|
|
64
|
-
* SQLite-based Session Store with checkpoint support
|
|
65
|
-
*
|
|
66
|
-
* Provides persistent storage for sessions and messages using SQLite.
|
|
67
|
-
* Follows the same interface as MemorySessionStore but persists to disk.
|
|
68
|
-
*/
|
|
69
|
-
export class SQLiteSessionStore implements SessionStore {
|
|
70
|
-
private db: any = null;
|
|
71
|
-
private dbPath: string;
|
|
72
|
-
private initialized = false;
|
|
73
|
-
|
|
74
|
-
// In-memory cache for performance
|
|
75
|
-
private sessionsCache = new Map<string, Session>();
|
|
76
|
-
private messagesCache = new Map<string, SessionMessage[]>();
|
|
77
|
-
private checkpointsCache = new Map<string, SessionCheckpoint[]>();
|
|
78
|
-
private messageConverter = new SessionMessageConverter();
|
|
79
|
-
|
|
80
|
-
constructor(dbPath?: string) {
|
|
81
|
-
this.dbPath = dbPath || getDefaultSessionDbPath();
|
|
82
|
-
}
|
|
83
|
-
|
|
84
|
-
// ========================================================================
|
|
85
|
-
// Initialization
|
|
86
|
-
// ========================================================================
|
|
87
|
-
|
|
88
|
-
private async initialize(): Promise<void> {
|
|
89
|
-
if (this.initialized) return;
|
|
90
|
-
|
|
91
|
-
// 使用 Bun 内置的 bun:sqlite
|
|
92
|
-
const { Database } = require("bun:sqlite");
|
|
93
|
-
|
|
94
|
-
// 确保目录存在
|
|
95
|
-
if (this.dbPath !== ":memory:") {
|
|
96
|
-
const fs = require("fs");
|
|
97
|
-
const path = require("path");
|
|
98
|
-
const dir = path.dirname(this.dbPath);
|
|
99
|
-
if (!fs.existsSync(dir)) {
|
|
100
|
-
fs.mkdirSync(dir, { recursive: true });
|
|
101
|
-
}
|
|
102
|
-
}
|
|
103
|
-
|
|
104
|
-
this.db = new Database(this.dbPath);
|
|
105
|
-
|
|
106
|
-
// 创建表
|
|
107
|
-
this.db.run(`
|
|
108
|
-
CREATE TABLE IF NOT EXISTS session (
|
|
109
|
-
id TEXT PRIMARY KEY,
|
|
110
|
-
title TEXT NOT NULL,
|
|
111
|
-
directory TEXT NOT NULL,
|
|
112
|
-
parent_id TEXT,
|
|
113
|
-
created_at INTEGER NOT NULL,
|
|
114
|
-
updated_at INTEGER NOT NULL,
|
|
115
|
-
message_count INTEGER DEFAULT 0,
|
|
116
|
-
metadata TEXT
|
|
117
|
-
)
|
|
118
|
-
`);
|
|
119
|
-
|
|
120
|
-
this.db.run(`
|
|
121
|
-
CREATE TABLE IF NOT EXISTS message (
|
|
122
|
-
id TEXT PRIMARY KEY,
|
|
123
|
-
session_id TEXT NOT NULL,
|
|
124
|
-
role TEXT NOT NULL,
|
|
125
|
-
content TEXT NOT NULL,
|
|
126
|
-
timestamp INTEGER NOT NULL,
|
|
127
|
-
parts TEXT,
|
|
128
|
-
is_archived INTEGER DEFAULT 0,
|
|
129
|
-
archived_at INTEGER,
|
|
130
|
-
checkpoint_id TEXT,
|
|
131
|
-
metadata TEXT,
|
|
132
|
-
FOREIGN KEY (session_id) REFERENCES session(id) ON DELETE CASCADE
|
|
133
|
-
)
|
|
134
|
-
`);
|
|
135
|
-
|
|
136
|
-
this.db.run("CREATE INDEX IF NOT EXISTS idx_message_session ON message(session_id)");
|
|
137
|
-
this.db.run("CREATE INDEX IF NOT EXISTS idx_message_checkpoint ON message(checkpoint_id)");
|
|
138
|
-
|
|
139
|
-
this.initialized = true;
|
|
140
|
-
}
|
|
141
|
-
|
|
142
|
-
// ========================================================================
|
|
143
|
-
// Session CRUD
|
|
144
|
-
// ========================================================================
|
|
145
|
-
|
|
146
|
-
async create(options: CreateSessionOptions): Promise<Session> {
|
|
147
|
-
await this.initialize();
|
|
148
|
-
|
|
149
|
-
const now = Date.now();
|
|
150
|
-
const session: Session = {
|
|
151
|
-
id: options.id ?? generateSessionId(), // Use custom ID if provided
|
|
152
|
-
title: options.title ?? "Session",
|
|
153
|
-
directory: options.directory ?? "/tmp",
|
|
154
|
-
parentID: options.parentID,
|
|
155
|
-
createdAt: now,
|
|
156
|
-
updatedAt: now,
|
|
157
|
-
messageCount: 0,
|
|
158
|
-
metadata: {
|
|
159
|
-
...options.metadata,
|
|
160
|
-
messageCount: 0,
|
|
161
|
-
checkpoints: { checkpoints: [] },
|
|
162
|
-
} as SessionMetadata,
|
|
163
|
-
};
|
|
164
|
-
|
|
165
|
-
// Check for duplicate ID (in case)
|
|
166
|
-
const existing = this.db.prepare("SELECT id FROM session WHERE id = ?").get(session.id);
|
|
167
|
-
if (existing) {
|
|
168
|
-
throw new Error(`Session with ID '${session.id}' already exists`);
|
|
169
|
-
}
|
|
170
|
-
|
|
171
|
-
// Insert into database
|
|
172
|
-
const stmt = this.db.prepare(`
|
|
173
|
-
INSERT INTO session (id, title, directory, parent_id, created_at, updated_at, message_count, metadata)
|
|
174
|
-
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
|
175
|
-
`);
|
|
176
|
-
stmt.run(
|
|
177
|
-
session.id,
|
|
178
|
-
session.title,
|
|
179
|
-
session.directory,
|
|
180
|
-
session.parentID || null,
|
|
181
|
-
session.createdAt,
|
|
182
|
-
session.updatedAt,
|
|
183
|
-
session.messageCount,
|
|
184
|
-
JSON.stringify(session.metadata)
|
|
185
|
-
);
|
|
186
|
-
|
|
187
|
-
// Update cache
|
|
188
|
-
this.sessionsCache.set(session.id, session);
|
|
189
|
-
this.messagesCache.set(session.id, []);
|
|
190
|
-
this.checkpointsCache.set(session.id, []);
|
|
191
|
-
|
|
192
|
-
return session;
|
|
193
|
-
}
|
|
194
|
-
|
|
195
|
-
async get(id: string): Promise<Session | undefined> {
|
|
196
|
-
await this.initialize();
|
|
197
|
-
|
|
198
|
-
// Check cache first
|
|
199
|
-
const cached = this.sessionsCache.get(id);
|
|
200
|
-
if (cached) return cached;
|
|
201
|
-
|
|
202
|
-
// Query from database
|
|
203
|
-
const row = this.db.prepare("SELECT * FROM session WHERE id = ?").get(id);
|
|
204
|
-
if (!row) return undefined;
|
|
205
|
-
|
|
206
|
-
const session = this.rowToSession(row);
|
|
207
|
-
this.sessionsCache.set(session.id, session);
|
|
208
|
-
return session;
|
|
209
|
-
}
|
|
210
|
-
|
|
211
|
-
async list(options?: ListSessionsOptions): Promise<Session[]> {
|
|
212
|
-
await this.initialize();
|
|
213
|
-
|
|
214
|
-
let sql = "SELECT * FROM session";
|
|
215
|
-
const params: any[] = [];
|
|
216
|
-
const conditions: string[] = [];
|
|
217
|
-
|
|
218
|
-
// Filter by metadata
|
|
219
|
-
if (options?.filter?.metadata) {
|
|
220
|
-
for (const [key, value] of Object.entries(options.filter.metadata)) {
|
|
221
|
-
conditions.push(`json_extract(metadata, '$.${key}') = ?`);
|
|
222
|
-
params.push(JSON.stringify(value));
|
|
223
|
-
}
|
|
224
|
-
}
|
|
225
|
-
|
|
226
|
-
// Filter by time range
|
|
227
|
-
if (options?.filter?.startTime) {
|
|
228
|
-
conditions.push("created_at >= ?");
|
|
229
|
-
params.push(options.filter.startTime);
|
|
230
|
-
}
|
|
231
|
-
if (options?.filter?.endTime) {
|
|
232
|
-
conditions.push("created_at <= ?");
|
|
233
|
-
params.push(options.filter.endTime);
|
|
234
|
-
}
|
|
235
|
-
|
|
236
|
-
if (conditions.length > 0) {
|
|
237
|
-
sql += " WHERE " + conditions.join(" AND ");
|
|
238
|
-
}
|
|
239
|
-
|
|
240
|
-
// Sort
|
|
241
|
-
const sortField = options?.sort?.field ?? "updatedAt";
|
|
242
|
-
const sortOrder = options?.sort?.order ?? "desc";
|
|
243
|
-
const orderColumn = sortField === "title" ? "title"
|
|
244
|
-
: sortField === "createdAt" ? "created_at"
|
|
245
|
-
: "updated_at";
|
|
246
|
-
sql += ` ORDER BY ${orderColumn} ${sortOrder.toUpperCase()}`;
|
|
247
|
-
|
|
248
|
-
// Paginate
|
|
249
|
-
const offset = options?.offset ?? 0;
|
|
250
|
-
const limit = options?.limit ?? 100;
|
|
251
|
-
sql += " LIMIT ? OFFSET ?";
|
|
252
|
-
params.push(limit, offset);
|
|
253
|
-
|
|
254
|
-
const rows = this.db.prepare(sql).all(...params);
|
|
255
|
-
return rows.map((row: any) => this.rowToSession(row));
|
|
256
|
-
}
|
|
257
|
-
|
|
258
|
-
async getCount(): Promise<number> {
|
|
259
|
-
await this.initialize();
|
|
260
|
-
const result = this.db.prepare("SELECT COUNT(*) as count FROM session").get() as { count: number };
|
|
261
|
-
return result.count;
|
|
262
|
-
}
|
|
263
|
-
|
|
264
|
-
async update(id: string, updates: UpdateSessionOptions): Promise<boolean> {
|
|
265
|
-
await this.initialize();
|
|
266
|
-
|
|
267
|
-
const session = await this.get(id);
|
|
268
|
-
if (!session) return false;
|
|
269
|
-
|
|
270
|
-
// Update fields
|
|
271
|
-
if (updates.title !== undefined) {
|
|
272
|
-
session.title = updates.title;
|
|
273
|
-
}
|
|
274
|
-
if (updates.metadata !== undefined) {
|
|
275
|
-
session.metadata = { ...session.metadata, ...updates.metadata } as SessionMetadata;
|
|
276
|
-
}
|
|
277
|
-
session.updatedAt = Date.now();
|
|
278
|
-
|
|
279
|
-
// Update database
|
|
280
|
-
const stmt = this.db.prepare(`
|
|
281
|
-
UPDATE session
|
|
282
|
-
SET title = ?, metadata = ?, updated_at = ?
|
|
283
|
-
WHERE id = ?
|
|
284
|
-
`);
|
|
285
|
-
stmt.run(session.title, JSON.stringify(session.metadata), session.updatedAt, id);
|
|
286
|
-
|
|
287
|
-
// Update cache
|
|
288
|
-
this.sessionsCache.set(id, session);
|
|
289
|
-
return true;
|
|
290
|
-
}
|
|
291
|
-
|
|
292
|
-
async delete(id: string): Promise<boolean> {
|
|
293
|
-
await this.initialize();
|
|
294
|
-
|
|
295
|
-
const session = await this.get(id);
|
|
296
|
-
if (!session) return false;
|
|
297
|
-
|
|
298
|
-
// Delete from database (messages will cascade)
|
|
299
|
-
this.db.prepare("DELETE FROM session WHERE id = ?").run(id);
|
|
300
|
-
|
|
301
|
-
// Clean up cache
|
|
302
|
-
this.sessionsCache.delete(id);
|
|
303
|
-
this.messagesCache.delete(id);
|
|
304
|
-
this.checkpointsCache.delete(id);
|
|
305
|
-
|
|
306
|
-
return true;
|
|
307
|
-
}
|
|
308
|
-
|
|
309
|
-
// ========================================================================
|
|
310
|
-
// Messages
|
|
311
|
-
// ========================================================================
|
|
312
|
-
|
|
313
|
-
async addMessage(
|
|
314
|
-
sessionId: string,
|
|
315
|
-
message: { role: string; content: string; parts?: unknown[]; metadata?: Record<string, unknown> }
|
|
316
|
-
): Promise<string> {
|
|
317
|
-
await this.initialize();
|
|
318
|
-
|
|
319
|
-
const session = await this.get(sessionId);
|
|
320
|
-
if (!session) {
|
|
321
|
-
throw new Error(`Session not found: ${sessionId}`);
|
|
322
|
-
}
|
|
323
|
-
|
|
324
|
-
const now = Date.now();
|
|
325
|
-
const msg: SessionMessage = {
|
|
326
|
-
id: generateMessageId(),
|
|
327
|
-
sessionID: sessionId,
|
|
328
|
-
role: message.role as SessionMessage["role"],
|
|
329
|
-
content: message.content,
|
|
330
|
-
timestamp: now,
|
|
331
|
-
parts: message.parts as SessionMessage["parts"],
|
|
332
|
-
metadata: message.metadata as SessionMessage["metadata"],
|
|
333
|
-
};
|
|
334
|
-
|
|
335
|
-
// Insert into database
|
|
336
|
-
const stmt = this.db.prepare(`
|
|
337
|
-
INSERT INTO message (id, session_id, role, content, timestamp, parts, metadata)
|
|
338
|
-
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
339
|
-
`);
|
|
340
|
-
stmt.run(
|
|
341
|
-
msg.id,
|
|
342
|
-
msg.sessionID,
|
|
343
|
-
msg.role,
|
|
344
|
-
msg.content,
|
|
345
|
-
msg.timestamp,
|
|
346
|
-
msg.parts ? JSON.stringify(msg.parts) : null,
|
|
347
|
-
msg.metadata ? JSON.stringify(msg.metadata) : null
|
|
348
|
-
);
|
|
349
|
-
|
|
350
|
-
// Update message count
|
|
351
|
-
const countResult = this.db.prepare(
|
|
352
|
-
"SELECT COUNT(*) as count FROM message WHERE session_id = ? AND is_archived = 0"
|
|
353
|
-
).get(sessionId);
|
|
354
|
-
session.messageCount = countResult?.count || 0;
|
|
355
|
-
session.updatedAt = now;
|
|
356
|
-
|
|
357
|
-
// Update metadata with messageCount
|
|
358
|
-
session.metadata = {
|
|
359
|
-
...session.metadata,
|
|
360
|
-
messageCount: session.messageCount
|
|
361
|
-
} as SessionMetadata;
|
|
362
|
-
|
|
363
|
-
this.db.prepare(`
|
|
364
|
-
UPDATE session SET message_count = ?, metadata = ?, updated_at = ? WHERE id = ?
|
|
365
|
-
`).run(session.messageCount, JSON.stringify(session.metadata), session.updatedAt, sessionId);
|
|
366
|
-
|
|
367
|
-
// Update cache
|
|
368
|
-
const messages = this.messagesCache.get(sessionId) ?? [];
|
|
369
|
-
messages.push(msg);
|
|
370
|
-
this.messagesCache.set(sessionId, messages);
|
|
371
|
-
this.sessionsCache.set(sessionId, session);
|
|
372
|
-
|
|
373
|
-
return msg.id;
|
|
374
|
-
}
|
|
375
|
-
|
|
376
|
-
async getMessages(
|
|
377
|
-
sessionId: string,
|
|
378
|
-
offset = 0,
|
|
379
|
-
limit = 50,
|
|
380
|
-
options?: GetMessagesOptions
|
|
381
|
-
): Promise<SessionMessage[]> {
|
|
382
|
-
await this.initialize();
|
|
383
|
-
|
|
384
|
-
let sql = "SELECT * FROM message WHERE session_id = ?";
|
|
385
|
-
let countSql = "SELECT COUNT(*) as count FROM message WHERE session_id = ?";
|
|
386
|
-
const params: any[] = [sessionId];
|
|
387
|
-
const countParams: any[] = [sessionId];
|
|
388
|
-
|
|
389
|
-
// Filter by archived status
|
|
390
|
-
if (!options?.includeArchived) {
|
|
391
|
-
sql += " AND is_archived = 0";
|
|
392
|
-
countSql += " AND is_archived = 0";
|
|
393
|
-
}
|
|
394
|
-
|
|
395
|
-
// Handle fromCheckpointId
|
|
396
|
-
if (options?.fromCheckpointId) {
|
|
397
|
-
sql += " AND timestamp >= (SELECT MIN(timestamp) FROM message WHERE session_id = ? AND json_extract(metadata, '$.checkpointId') = ?)";
|
|
398
|
-
params.push(sessionId, options.fromCheckpointId);
|
|
399
|
-
}
|
|
400
|
-
|
|
401
|
-
// Handle negative offset: count total messages first, then calculate actual offset
|
|
402
|
-
// offset = -1, limit = 50: latest 50 messages → offset = total - limit
|
|
403
|
-
// offset = -2, limit = 50: 2nd latest 50 messages → offset = total - 2*limit
|
|
404
|
-
let actualOffset = offset;
|
|
405
|
-
if (offset < 0) {
|
|
406
|
-
const totalResult = this.db.prepare(countSql).get(...countParams);
|
|
407
|
-
const total = totalResult?.count || 0;
|
|
408
|
-
actualOffset = Math.max(0, total + offset * limit);
|
|
409
|
-
}
|
|
410
|
-
|
|
411
|
-
// Sort
|
|
412
|
-
if (options?.reverse) {
|
|
413
|
-
sql += " ORDER BY timestamp DESC";
|
|
414
|
-
} else {
|
|
415
|
-
sql += " ORDER BY timestamp ASC";
|
|
416
|
-
}
|
|
417
|
-
|
|
418
|
-
// Paginate
|
|
419
|
-
sql += " LIMIT ? OFFSET ?";
|
|
420
|
-
params.push(limit, actualOffset);
|
|
421
|
-
|
|
422
|
-
const rows = this.db.prepare(sql).all(...params);
|
|
423
|
-
return rows.map((row: any) => this.rowToMessage(row));
|
|
424
|
-
}
|
|
425
|
-
|
|
426
|
-
async getMessageCount(sessionId: string, includeArchived = false): Promise<number> {
|
|
427
|
-
await this.initialize();
|
|
428
|
-
|
|
429
|
-
let sql = "SELECT COUNT(*) as count FROM message WHERE session_id = ?";
|
|
430
|
-
if (!includeArchived) {
|
|
431
|
-
sql += " AND is_archived = 0";
|
|
432
|
-
}
|
|
433
|
-
|
|
434
|
-
const result = this.db.prepare(sql).get(sessionId);
|
|
435
|
-
return result?.count || 0;
|
|
436
|
-
}
|
|
437
|
-
|
|
438
|
-
async getMessageIndexes(sessionId: string, role: string): Promise<number[]> {
|
|
439
|
-
await this.initialize();
|
|
440
|
-
|
|
441
|
-
// Use ROW_NUMBER() to get the offset index based on timestamp ASC order
|
|
442
|
-
// This ensures the index matches what getMessages returns with default ordering
|
|
443
|
-
const sql = `
|
|
444
|
-
SELECT idx FROM (
|
|
445
|
-
SELECT
|
|
446
|
-
ROW_NUMBER() OVER (ORDER BY timestamp ASC) - 1 as idx,
|
|
447
|
-
role
|
|
448
|
-
FROM message
|
|
449
|
-
WHERE session_id = ? AND is_archived = 0
|
|
450
|
-
) ranked
|
|
451
|
-
WHERE role = ?
|
|
452
|
-
`;
|
|
453
|
-
|
|
454
|
-
const rows = this.db.prepare(sql).all(sessionId, role) as { idx: number }[];
|
|
455
|
-
return rows.map(row => row.idx);
|
|
456
|
-
}
|
|
457
|
-
|
|
458
|
-
// ========================================================================
|
|
459
|
-
// Checkpoint Management
|
|
460
|
-
// ========================================================================
|
|
461
|
-
|
|
462
|
-
async saveCheckpoint(sessionId: string, checkpoint: SessionCheckpoint): Promise<void> {
|
|
463
|
-
await this.initialize();
|
|
464
|
-
|
|
465
|
-
const session = await this.get(sessionId);
|
|
466
|
-
if (!session) {
|
|
467
|
-
throw new Error(`Session not found: ${sessionId}`);
|
|
468
|
-
}
|
|
469
|
-
|
|
470
|
-
// Save to checkpoints cache
|
|
471
|
-
const cps = this.checkpointsCache.get(sessionId) ?? [];
|
|
472
|
-
cps.push(checkpoint);
|
|
473
|
-
this.checkpointsCache.set(sessionId, cps);
|
|
474
|
-
|
|
475
|
-
// Update session metadata
|
|
476
|
-
const meta = session.metadata as Record<string, unknown> || {};
|
|
477
|
-
const checkpointsMeta: SessionCheckpointsMeta = (meta.checkpoints as SessionCheckpointsMeta) || { checkpoints: [] };
|
|
478
|
-
checkpointsMeta.checkpoints.push({
|
|
479
|
-
id: checkpoint.id,
|
|
480
|
-
messageIndex: checkpoint.messageIndex,
|
|
481
|
-
title: checkpoint.title,
|
|
482
|
-
createdAt: checkpoint.createdAt,
|
|
483
|
-
type: checkpoint.type,
|
|
484
|
-
});
|
|
485
|
-
checkpointsMeta.latestCheckpointId = checkpoint.id;
|
|
486
|
-
|
|
487
|
-
meta.checkpoints = checkpointsMeta;
|
|
488
|
-
|
|
489
|
-
// Store checkpoint details separately
|
|
490
|
-
if (!meta.checkpointDetails) {
|
|
491
|
-
meta.checkpointDetails = {};
|
|
492
|
-
}
|
|
493
|
-
(meta.checkpointDetails as Record<string, SessionCheckpoint>)[checkpoint.id] = checkpoint;
|
|
494
|
-
|
|
495
|
-
session.metadata = meta as SessionMetadata;
|
|
496
|
-
session.updatedAt = Date.now();
|
|
497
|
-
|
|
498
|
-
// Update database
|
|
499
|
-
const stmt = this.db.prepare(`
|
|
500
|
-
UPDATE session SET metadata = ?, updated_at = ? WHERE id = ?
|
|
501
|
-
`);
|
|
502
|
-
stmt.run(JSON.stringify(session.metadata), session.updatedAt, sessionId);
|
|
503
|
-
|
|
504
|
-
// Update cache
|
|
505
|
-
this.sessionsCache.set(sessionId, session);
|
|
506
|
-
}
|
|
507
|
-
|
|
508
|
-
async getCheckpoints(sessionId: string): Promise<SessionCheckpoint[]> {
|
|
509
|
-
await this.initialize();
|
|
510
|
-
|
|
511
|
-
const session = await this.get(sessionId);
|
|
512
|
-
if (!session) return [];
|
|
513
|
-
|
|
514
|
-
const meta = session.metadata as Record<string, unknown>;
|
|
515
|
-
const checkpointDetails = meta?.checkpointDetails as Record<string, SessionCheckpoint>;
|
|
516
|
-
if (!checkpointDetails) return [];
|
|
517
|
-
|
|
518
|
-
return Object.values(checkpointDetails);
|
|
519
|
-
}
|
|
520
|
-
|
|
521
|
-
async getCheckpoint(sessionId: string, checkpointId: string): Promise<SessionCheckpoint | undefined> {
|
|
522
|
-
await this.initialize();
|
|
523
|
-
|
|
524
|
-
const checkpoints = await this.getCheckpoints(sessionId);
|
|
525
|
-
return checkpoints.find((cp) => cp.id === checkpointId);
|
|
526
|
-
}
|
|
527
|
-
|
|
528
|
-
async deleteCheckpoint(sessionId: string, checkpointId: string): Promise<boolean> {
|
|
529
|
-
await this.initialize();
|
|
530
|
-
|
|
531
|
-
const session = await this.get(sessionId);
|
|
532
|
-
if (!session) return false;
|
|
533
|
-
|
|
534
|
-
const checkpoints = this.checkpointsCache.get(sessionId) ?? [];
|
|
535
|
-
const index = checkpoints.findIndex((cp) => cp.id === checkpointId);
|
|
536
|
-
if (index === -1) return false;
|
|
537
|
-
|
|
538
|
-
// Remove checkpoint
|
|
539
|
-
checkpoints.splice(index, 1);
|
|
540
|
-
this.checkpointsCache.set(sessionId, checkpoints);
|
|
541
|
-
|
|
542
|
-
// Restore archived messages
|
|
543
|
-
this.db.prepare(`
|
|
544
|
-
UPDATE message SET is_archived = 0, archived_at = NULL, checkpoint_id = NULL
|
|
545
|
-
WHERE session_id = ? AND checkpoint_id = ?
|
|
546
|
-
`).run(sessionId, checkpointId);
|
|
547
|
-
|
|
548
|
-
// Update session metadata
|
|
549
|
-
const meta = session.metadata as Record<string, unknown> || {};
|
|
550
|
-
const checkpointsMeta: SessionCheckpointsMeta = (meta.checkpoints as SessionCheckpointsMeta) || { checkpoints: [] };
|
|
551
|
-
checkpointsMeta.checkpoints = checkpointsMeta.checkpoints.filter((cp) => cp.id !== checkpointId);
|
|
552
|
-
checkpointsMeta.latestCheckpointId = checkpoints[checkpoints.length - 1]?.id;
|
|
553
|
-
|
|
554
|
-
if (meta.checkpointDetails) {
|
|
555
|
-
delete (meta.checkpointDetails as Record<string, SessionCheckpoint>)[checkpointId];
|
|
556
|
-
}
|
|
557
|
-
|
|
558
|
-
session.metadata = meta as SessionMetadata;
|
|
559
|
-
|
|
560
|
-
// Update message count
|
|
561
|
-
const countResult = this.db.prepare(
|
|
562
|
-
"SELECT COUNT(*) as count FROM message WHERE session_id = ? AND is_archived = 0"
|
|
563
|
-
).get(sessionId);
|
|
564
|
-
session.messageCount = countResult?.count || 0;
|
|
565
|
-
session.updatedAt = Date.now();
|
|
566
|
-
|
|
567
|
-
// Update database
|
|
568
|
-
this.db.prepare(`
|
|
569
|
-
UPDATE session SET metadata = ?, message_count = ?, updated_at = ? WHERE id = ?
|
|
570
|
-
`).run(JSON.stringify(session.metadata), session.messageCount, session.updatedAt, sessionId);
|
|
571
|
-
|
|
572
|
-
// Update cache
|
|
573
|
-
this.sessionsCache.set(sessionId, session);
|
|
574
|
-
|
|
575
|
-
return true;
|
|
576
|
-
}
|
|
577
|
-
|
|
578
|
-
async archiveMessages(sessionId: string, checkpointId: string, beforeIndex: number): Promise<number> {
|
|
579
|
-
await this.initialize();
|
|
580
|
-
|
|
581
|
-
const session = await this.get(sessionId);
|
|
582
|
-
if (!session) return 0;
|
|
583
|
-
|
|
584
|
-
const now = Date.now();
|
|
585
|
-
|
|
586
|
-
// Archive messages before the index
|
|
587
|
-
const result = this.db.prepare(`
|
|
588
|
-
UPDATE message
|
|
589
|
-
SET is_archived = 1, archived_at = ?, checkpoint_id = ?
|
|
590
|
-
WHERE session_id = ? AND id IN (
|
|
591
|
-
SELECT id FROM message WHERE session_id = ? AND is_archived = 0
|
|
592
|
-
ORDER BY timestamp ASC LIMIT ?
|
|
593
|
-
)
|
|
594
|
-
`).run(now, checkpointId, sessionId, sessionId, beforeIndex);
|
|
595
|
-
|
|
596
|
-
// Update message count
|
|
597
|
-
const countResult = this.db.prepare(
|
|
598
|
-
"SELECT COUNT(*) as count FROM message WHERE session_id = ? AND is_archived = 0"
|
|
599
|
-
).get(sessionId);
|
|
600
|
-
session.messageCount = countResult?.count || 0;
|
|
601
|
-
session.updatedAt = now;
|
|
602
|
-
|
|
603
|
-
// Update database
|
|
604
|
-
this.db.prepare(`
|
|
605
|
-
UPDATE session SET message_count = ?, updated_at = ? WHERE id = ?
|
|
606
|
-
`).run(session.messageCount, session.updatedAt, sessionId);
|
|
607
|
-
|
|
608
|
-
// Update cache
|
|
609
|
-
this.sessionsCache.set(sessionId, session);
|
|
610
|
-
this.messagesCache.delete(sessionId); // Clear message cache
|
|
611
|
-
|
|
612
|
-
return result.changes;
|
|
613
|
-
}
|
|
614
|
-
|
|
615
|
-
async getActiveMessages(sessionId: string, offset = 0, limit = 50): Promise<SessionMessage[]> {
|
|
616
|
-
return this.getMessages(sessionId, offset, limit, { includeArchived: false });
|
|
617
|
-
}
|
|
618
|
-
|
|
619
|
-
// ========================================================================
|
|
620
|
-
// Search [new]
|
|
621
|
-
// ========================================================================
|
|
622
|
-
|
|
623
|
-
async searchMessages(options: SearchMessagesOptions): Promise<MessageMatch[]> {
|
|
624
|
-
await this.initialize();
|
|
625
|
-
|
|
626
|
-
const matches: MessageMatch[] = [];
|
|
627
|
-
const {
|
|
628
|
-
query,
|
|
629
|
-
sessionId,
|
|
630
|
-
limit = 10,
|
|
631
|
-
beforeTime,
|
|
632
|
-
afterTime,
|
|
633
|
-
includeArchived = false,
|
|
634
|
-
} = options;
|
|
635
|
-
|
|
636
|
-
// Build SQL query
|
|
637
|
-
let sql = "SELECT * FROM message WHERE 1=1";
|
|
638
|
-
const params: any[] = [];
|
|
639
|
-
|
|
640
|
-
// Filter by session
|
|
641
|
-
if (sessionId) {
|
|
642
|
-
sql += " AND session_id = ?";
|
|
643
|
-
params.push(sessionId);
|
|
644
|
-
}
|
|
645
|
-
|
|
646
|
-
// Filter by archived status
|
|
647
|
-
if (!includeArchived) {
|
|
648
|
-
sql += " AND is_archived = 0";
|
|
649
|
-
}
|
|
650
|
-
|
|
651
|
-
// Filter by time
|
|
652
|
-
if (beforeTime) {
|
|
653
|
-
sql += " AND timestamp <= ?";
|
|
654
|
-
params.push(beforeTime);
|
|
655
|
-
}
|
|
656
|
-
if (afterTime) {
|
|
657
|
-
sql += " AND timestamp >= ?";
|
|
658
|
-
params.push(afterTime);
|
|
659
|
-
}
|
|
660
|
-
|
|
661
|
-
// Order by timestamp
|
|
662
|
-
sql += " ORDER BY timestamp DESC";
|
|
663
|
-
|
|
664
|
-
// Limit
|
|
665
|
-
sql += " LIMIT ?";
|
|
666
|
-
params.push(limit * 10); // Get more than limit to filter by query
|
|
667
|
-
|
|
668
|
-
const rows = this.db.prepare(sql).all(...params);
|
|
669
|
-
|
|
670
|
-
// Parse query for simple text matching
|
|
671
|
-
const queryTerms = this.parseQuery(query);
|
|
672
|
-
const lowerQuery = query.toLowerCase();
|
|
673
|
-
|
|
674
|
-
// Filter and match messages
|
|
675
|
-
for (const row of rows) {
|
|
676
|
-
const content = (row.content as string).toLowerCase();
|
|
677
|
-
|
|
678
|
-
// Check if content matches query
|
|
679
|
-
let matchesQuery = true;
|
|
680
|
-
|
|
681
|
-
// Check excluded terms
|
|
682
|
-
for (const term of queryTerms.excluded) {
|
|
683
|
-
if (content.includes(term.toLowerCase())) {
|
|
684
|
-
matchesQuery = false;
|
|
685
|
-
break;
|
|
686
|
-
}
|
|
687
|
-
}
|
|
688
|
-
|
|
689
|
-
// Check required terms
|
|
690
|
-
if (matchesQuery && queryTerms.required.length > 0) {
|
|
691
|
-
for (const term of queryTerms.required) {
|
|
692
|
-
if (!content.includes(term.toLowerCase())) {
|
|
693
|
-
matchesQuery = false;
|
|
694
|
-
break;
|
|
695
|
-
}
|
|
696
|
-
}
|
|
697
|
-
}
|
|
698
|
-
|
|
699
|
-
// Simple fallback: check if query string appears in content
|
|
700
|
-
if (!matchesQuery && lowerQuery && content.includes(lowerQuery)) {
|
|
701
|
-
matchesQuery = true;
|
|
702
|
-
}
|
|
703
|
-
|
|
704
|
-
if (matchesQuery) {
|
|
705
|
-
matches.push({
|
|
706
|
-
messageId: row.id,
|
|
707
|
-
sessionId: row.session_id,
|
|
708
|
-
role: row.role as MessageMatch["role"],
|
|
709
|
-
content: row.content,
|
|
710
|
-
timestamp: row.timestamp,
|
|
711
|
-
});
|
|
712
|
-
|
|
713
|
-
if (matches.length >= limit) break;
|
|
714
|
-
}
|
|
715
|
-
}
|
|
716
|
-
|
|
717
|
-
return matches;
|
|
718
|
-
}
|
|
719
|
-
|
|
720
|
-
/**
|
|
721
|
-
* Parse simple query into terms
|
|
722
|
-
*/
|
|
723
|
-
private parseQuery(query: string): { required: string[]; excluded: string[] } {
|
|
724
|
-
// Remove quoted phrases (simplified)
|
|
725
|
-
const terms = query.split(/\s+/).filter((t) => !t.startsWith('"') && !t.endsWith('"'));
|
|
726
|
-
|
|
727
|
-
const required: string[] = [];
|
|
728
|
-
const excluded: string[] = [];
|
|
729
|
-
|
|
730
|
-
for (const term of terms) {
|
|
731
|
-
if (term.startsWith("-") || term.toUpperCase().startsWith("NOT ")) {
|
|
732
|
-
excluded.push(term.replace(/^-|^NOT /i, ""));
|
|
733
|
-
} else {
|
|
734
|
-
required.push(term);
|
|
735
|
-
}
|
|
736
|
-
}
|
|
737
|
-
|
|
738
|
-
return { required, excluded };
|
|
739
|
-
}
|
|
740
|
-
|
|
741
|
-
// ========================================================================
|
|
742
|
-
// Lifecycle
|
|
743
|
-
// ========================================================================
|
|
744
|
-
|
|
745
|
-
async close(): Promise<void> {
|
|
746
|
-
if (this.db) {
|
|
747
|
-
this.db.close();
|
|
748
|
-
this.db = null;
|
|
749
|
-
}
|
|
750
|
-
this.sessionsCache.clear();
|
|
751
|
-
this.messagesCache.clear();
|
|
752
|
-
this.checkpointsCache.clear();
|
|
753
|
-
this.initialized = false;
|
|
754
|
-
}
|
|
755
|
-
|
|
756
|
-
// ========================================================================
|
|
757
|
-
// Row Conversions
|
|
758
|
-
// ========================================================================
|
|
759
|
-
|
|
760
|
-
private rowToSession(row: any): Session {
|
|
761
|
-
return {
|
|
762
|
-
id: row.id,
|
|
763
|
-
title: row.title,
|
|
764
|
-
directory: row.directory,
|
|
765
|
-
parentID: row.parent_id,
|
|
766
|
-
createdAt: row.created_at,
|
|
767
|
-
updatedAt: row.updated_at,
|
|
768
|
-
messageCount: row.message_count,
|
|
769
|
-
metadata: row.metadata ? JSON.parse(row.metadata) : undefined,
|
|
770
|
-
};
|
|
771
|
-
}
|
|
772
|
-
|
|
773
|
-
private rowToMessage(row: any): SessionMessage {
|
|
774
|
-
// Parse the raw message from database
|
|
775
|
-
const rawMessage: Partial<SessionMessage> = {
|
|
776
|
-
id: row.id,
|
|
777
|
-
sessionID: row.session_id,
|
|
778
|
-
role: row.role as SessionMessage["role"],
|
|
779
|
-
content: row.content,
|
|
780
|
-
timestamp: row.timestamp,
|
|
781
|
-
parts: row.parts ? JSON.parse(row.parts) : undefined,
|
|
782
|
-
isArchived: row.is_archived === 1,
|
|
783
|
-
archivedAt: row.archived_at || undefined,
|
|
784
|
-
checkpointId: row.checkpoint_id || undefined,
|
|
785
|
-
metadata: row.metadata ? JSON.parse(row.metadata) : undefined,
|
|
786
|
-
};
|
|
787
|
-
|
|
788
|
-
// Apply migration for backward compatibility with old format
|
|
789
|
-
return this.messageConverter.migrateMessage(rawMessage);
|
|
790
|
-
}
|
|
791
|
-
}
|
|
792
|
-
|
|
793
|
-
// Export helper functions
|
|
794
|
-
export { getDefaultSessionDbPath, getDefaultDataDir };
|