@aion0/bastion 0.1.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +183 -0
- package/README.zh.md +468 -0
- package/config/default.yaml +73 -0
- package/dist/cli/commands/config.d.ts +3 -0
- package/dist/cli/commands/config.d.ts.map +1 -0
- package/dist/cli/commands/config.js +31 -0
- package/dist/cli/commands/config.js.map +1 -0
- package/dist/cli/commands/env.d.ts +3 -0
- package/dist/cli/commands/env.d.ts.map +1 -0
- package/dist/cli/commands/env.js +83 -0
- package/dist/cli/commands/env.js.map +1 -0
- package/dist/cli/commands/health.d.ts +3 -0
- package/dist/cli/commands/health.d.ts.map +1 -0
- package/dist/cli/commands/health.js +45 -0
- package/dist/cli/commands/health.js.map +1 -0
- package/dist/cli/commands/openclaw.d.ts +3 -0
- package/dist/cli/commands/openclaw.d.ts.map +1 -0
- package/dist/cli/commands/openclaw.js +1062 -0
- package/dist/cli/commands/openclaw.js.map +1 -0
- package/dist/cli/commands/proxy.d.ts +8 -0
- package/dist/cli/commands/proxy.d.ts.map +1 -0
- package/dist/cli/commands/proxy.js +433 -0
- package/dist/cli/commands/proxy.js.map +1 -0
- package/dist/cli/commands/start.d.ts +3 -0
- package/dist/cli/commands/start.d.ts.map +1 -0
- package/dist/cli/commands/start.js +62 -0
- package/dist/cli/commands/start.js.map +1 -0
- package/dist/cli/commands/stats.d.ts +3 -0
- package/dist/cli/commands/stats.d.ts.map +1 -0
- package/dist/cli/commands/stats.js +32 -0
- package/dist/cli/commands/stats.js.map +1 -0
- package/dist/cli/commands/stop.d.ts +3 -0
- package/dist/cli/commands/stop.d.ts.map +1 -0
- package/dist/cli/commands/stop.js +28 -0
- package/dist/cli/commands/stop.js.map +1 -0
- package/dist/cli/commands/token.d.ts +3 -0
- package/dist/cli/commands/token.d.ts.map +1 -0
- package/dist/cli/commands/token.js +32 -0
- package/dist/cli/commands/token.js.map +1 -0
- package/dist/cli/commands/trust-ca.d.ts +3 -0
- package/dist/cli/commands/trust-ca.d.ts.map +1 -0
- package/dist/cli/commands/trust-ca.js +44 -0
- package/dist/cli/commands/trust-ca.js.map +1 -0
- package/dist/cli/commands/wrap.d.ts +3 -0
- package/dist/cli/commands/wrap.d.ts.map +1 -0
- package/dist/cli/commands/wrap.js +70 -0
- package/dist/cli/commands/wrap.js.map +1 -0
- package/dist/cli/daemon.d.ts +11 -0
- package/dist/cli/daemon.d.ts.map +1 -0
- package/dist/cli/daemon.js +82 -0
- package/dist/cli/daemon.js.map +1 -0
- package/dist/cli/index.d.ts +3 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +35 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/config/index.d.ts +3 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +60 -0
- package/dist/config/index.js.map +1 -0
- package/dist/config/manager.d.ts +12 -0
- package/dist/config/manager.d.ts.map +1 -0
- package/dist/config/manager.js +73 -0
- package/dist/config/manager.js.map +1 -0
- package/dist/config/paths.d.ts +10 -0
- package/dist/config/paths.d.ts.map +1 -0
- package/dist/config/paths.js +16 -0
- package/dist/config/paths.js.map +1 -0
- package/dist/config/schema.d.ts +85 -0
- package/dist/config/schema.d.ts.map +1 -0
- package/dist/config/schema.js +3 -0
- package/dist/config/schema.js.map +1 -0
- package/dist/dashboard/api-routes.d.ts +6 -0
- package/dist/dashboard/api-routes.d.ts.map +1 -0
- package/dist/dashboard/api-routes.js +671 -0
- package/dist/dashboard/api-routes.js.map +1 -0
- package/dist/dashboard/api.d.ts +4 -0
- package/dist/dashboard/api.d.ts.map +1 -0
- package/dist/dashboard/api.js +25 -0
- package/dist/dashboard/api.js.map +1 -0
- package/dist/dashboard/page.d.ts +3 -0
- package/dist/dashboard/page.d.ts.map +1 -0
- package/dist/dashboard/page.js +1622 -0
- package/dist/dashboard/page.js.map +1 -0
- package/dist/dlp/actions.d.ts +13 -0
- package/dist/dlp/actions.d.ts.map +1 -0
- package/dist/dlp/actions.js +3 -0
- package/dist/dlp/actions.js.map +1 -0
- package/dist/dlp/ai-validator.d.ts +28 -0
- package/dist/dlp/ai-validator.d.ts.map +1 -0
- package/dist/dlp/ai-validator.js +214 -0
- package/dist/dlp/ai-validator.js.map +1 -0
- package/dist/dlp/engine.d.ts +34 -0
- package/dist/dlp/engine.d.ts.map +1 -0
- package/dist/dlp/engine.js +342 -0
- package/dist/dlp/engine.js.map +1 -0
- package/dist/dlp/entropy.d.ts +22 -0
- package/dist/dlp/entropy.d.ts.map +1 -0
- package/dist/dlp/entropy.js +43 -0
- package/dist/dlp/entropy.js.map +1 -0
- package/dist/dlp/message-cache.d.ts +45 -0
- package/dist/dlp/message-cache.d.ts.map +1 -0
- package/dist/dlp/message-cache.js +251 -0
- package/dist/dlp/message-cache.js.map +1 -0
- package/dist/dlp/patterns/context-aware.d.ts +4 -0
- package/dist/dlp/patterns/context-aware.d.ts.map +1 -0
- package/dist/dlp/patterns/context-aware.js +45 -0
- package/dist/dlp/patterns/context-aware.js.map +1 -0
- package/dist/dlp/patterns/high-confidence.d.ts +4 -0
- package/dist/dlp/patterns/high-confidence.d.ts.map +1 -0
- package/dist/dlp/patterns/high-confidence.js +140 -0
- package/dist/dlp/patterns/high-confidence.js.map +1 -0
- package/dist/dlp/patterns/prompt-injection.d.ts +4 -0
- package/dist/dlp/patterns/prompt-injection.d.ts.map +1 -0
- package/dist/dlp/patterns/prompt-injection.js +244 -0
- package/dist/dlp/patterns/prompt-injection.js.map +1 -0
- package/dist/dlp/patterns/validated.d.ts +4 -0
- package/dist/dlp/patterns/validated.d.ts.map +1 -0
- package/dist/dlp/patterns/validated.js +21 -0
- package/dist/dlp/patterns/validated.js.map +1 -0
- package/dist/dlp/remote-sync.d.ts +47 -0
- package/dist/dlp/remote-sync.d.ts.map +1 -0
- package/dist/dlp/remote-sync.js +252 -0
- package/dist/dlp/remote-sync.js.map +1 -0
- package/dist/dlp/semantics.d.ts +27 -0
- package/dist/dlp/semantics.d.ts.map +1 -0
- package/dist/dlp/semantics.js +93 -0
- package/dist/dlp/semantics.js.map +1 -0
- package/dist/dlp/structure.d.ts +25 -0
- package/dist/dlp/structure.d.ts.map +1 -0
- package/dist/dlp/structure.js +86 -0
- package/dist/dlp/structure.js.map +1 -0
- package/dist/dlp/validators.d.ts +6 -0
- package/dist/dlp/validators.d.ts.map +1 -0
- package/dist/dlp/validators.js +46 -0
- package/dist/dlp/validators.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +200 -0
- package/dist/index.js.map +1 -0
- package/dist/license/verify.d.ts +18 -0
- package/dist/license/verify.d.ts.map +1 -0
- package/dist/license/verify.js +71 -0
- package/dist/license/verify.js.map +1 -0
- package/dist/metrics/collector.d.ts +11 -0
- package/dist/metrics/collector.d.ts.map +1 -0
- package/dist/metrics/collector.js +17 -0
- package/dist/metrics/collector.js.map +1 -0
- package/dist/metrics/dashboard.d.ts +6 -0
- package/dist/metrics/dashboard.d.ts.map +1 -0
- package/dist/metrics/dashboard.js +66 -0
- package/dist/metrics/dashboard.js.map +1 -0
- package/dist/metrics/pricing.d.ts +10 -0
- package/dist/metrics/pricing.d.ts.map +1 -0
- package/dist/metrics/pricing.js +62 -0
- package/dist/metrics/pricing.js.map +1 -0
- package/dist/optimizer/cache.d.ts +14 -0
- package/dist/optimizer/cache.d.ts.map +1 -0
- package/dist/optimizer/cache.js +58 -0
- package/dist/optimizer/cache.js.map +1 -0
- package/dist/optimizer/estimator.d.ts +6 -0
- package/dist/optimizer/estimator.d.ts.map +1 -0
- package/dist/optimizer/estimator.js +12 -0
- package/dist/optimizer/estimator.js.map +1 -0
- package/dist/optimizer/reorder.d.ts +9 -0
- package/dist/optimizer/reorder.d.ts.map +1 -0
- package/dist/optimizer/reorder.js +27 -0
- package/dist/optimizer/reorder.js.map +1 -0
- package/dist/optimizer/trimmer.d.ts +9 -0
- package/dist/optimizer/trimmer.d.ts.map +1 -0
- package/dist/optimizer/trimmer.js +47 -0
- package/dist/optimizer/trimmer.js.map +1 -0
- package/dist/plugin-api/index.d.ts +3 -0
- package/dist/plugin-api/index.d.ts.map +1 -0
- package/dist/plugin-api/index.js +6 -0
- package/dist/plugin-api/index.js.map +1 -0
- package/dist/plugin-api/types.d.ts +77 -0
- package/dist/plugin-api/types.d.ts.map +1 -0
- package/dist/plugin-api/types.js +6 -0
- package/dist/plugin-api/types.js.map +1 -0
- package/dist/plugins/adapter.d.ts +12 -0
- package/dist/plugins/adapter.d.ts.map +1 -0
- package/dist/plugins/adapter.js +116 -0
- package/dist/plugins/adapter.js.map +1 -0
- package/dist/plugins/builtin/audit-logger.d.ts +9 -0
- package/dist/plugins/builtin/audit-logger.d.ts.map +1 -0
- package/dist/plugins/builtin/audit-logger.js +53 -0
- package/dist/plugins/builtin/audit-logger.js.map +1 -0
- package/dist/plugins/builtin/dlp-scanner.d.ts +19 -0
- package/dist/plugins/builtin/dlp-scanner.d.ts.map +1 -0
- package/dist/plugins/builtin/dlp-scanner.js +284 -0
- package/dist/plugins/builtin/dlp-scanner.js.map +1 -0
- package/dist/plugins/builtin/metrics-collector.d.ts +4 -0
- package/dist/plugins/builtin/metrics-collector.d.ts.map +1 -0
- package/dist/plugins/builtin/metrics-collector.js +111 -0
- package/dist/plugins/builtin/metrics-collector.js.map +1 -0
- package/dist/plugins/builtin/token-optimizer.d.ts +10 -0
- package/dist/plugins/builtin/token-optimizer.d.ts.map +1 -0
- package/dist/plugins/builtin/token-optimizer.js +120 -0
- package/dist/plugins/builtin/token-optimizer.js.map +1 -0
- package/dist/plugins/builtin/tool-guard.d.ts +20 -0
- package/dist/plugins/builtin/tool-guard.d.ts.map +1 -0
- package/dist/plugins/builtin/tool-guard.js +259 -0
- package/dist/plugins/builtin/tool-guard.js.map +1 -0
- package/dist/plugins/context.d.ts +8 -0
- package/dist/plugins/context.d.ts.map +1 -0
- package/dist/plugins/context.js +33 -0
- package/dist/plugins/context.js.map +1 -0
- package/dist/plugins/event-bus.d.ts +9 -0
- package/dist/plugins/event-bus.d.ts.map +1 -0
- package/dist/plugins/event-bus.js +25 -0
- package/dist/plugins/event-bus.js.map +1 -0
- package/dist/plugins/index.d.ts +18 -0
- package/dist/plugins/index.d.ts.map +1 -0
- package/dist/plugins/index.js +148 -0
- package/dist/plugins/index.js.map +1 -0
- package/dist/plugins/loader.d.ts +14 -0
- package/dist/plugins/loader.d.ts.map +1 -0
- package/dist/plugins/loader.js +98 -0
- package/dist/plugins/loader.js.map +1 -0
- package/dist/plugins/types.d.ts +91 -0
- package/dist/plugins/types.d.ts.map +1 -0
- package/dist/plugins/types.js +3 -0
- package/dist/plugins/types.js.map +1 -0
- package/dist/proxy/certs.d.ts +10 -0
- package/dist/proxy/certs.d.ts.map +1 -0
- package/dist/proxy/certs.js +110 -0
- package/dist/proxy/certs.js.map +1 -0
- package/dist/proxy/connect.d.ts +11 -0
- package/dist/proxy/connect.d.ts.map +1 -0
- package/dist/proxy/connect.js +298 -0
- package/dist/proxy/connect.js.map +1 -0
- package/dist/proxy/forwarder.d.ts +14 -0
- package/dist/proxy/forwarder.d.ts.map +1 -0
- package/dist/proxy/forwarder.js +342 -0
- package/dist/proxy/forwarder.js.map +1 -0
- package/dist/proxy/passthrough.d.ts +4 -0
- package/dist/proxy/passthrough.d.ts.map +1 -0
- package/dist/proxy/passthrough.js +68 -0
- package/dist/proxy/passthrough.js.map +1 -0
- package/dist/proxy/providers/anthropic.d.ts +4 -0
- package/dist/proxy/providers/anthropic.d.ts.map +1 -0
- package/dist/proxy/providers/anthropic.js +46 -0
- package/dist/proxy/providers/anthropic.js.map +1 -0
- package/dist/proxy/providers/classify.d.ts +14 -0
- package/dist/proxy/providers/classify.d.ts.map +1 -0
- package/dist/proxy/providers/classify.js +37 -0
- package/dist/proxy/providers/classify.js.map +1 -0
- package/dist/proxy/providers/claude-web.d.ts +8 -0
- package/dist/proxy/providers/claude-web.d.ts.map +1 -0
- package/dist/proxy/providers/claude-web.js +50 -0
- package/dist/proxy/providers/claude-web.js.map +1 -0
- package/dist/proxy/providers/gemini.d.ts +4 -0
- package/dist/proxy/providers/gemini.d.ts.map +1 -0
- package/dist/proxy/providers/gemini.js +38 -0
- package/dist/proxy/providers/gemini.js.map +1 -0
- package/dist/proxy/providers/index.d.ts +27 -0
- package/dist/proxy/providers/index.d.ts.map +1 -0
- package/dist/proxy/providers/index.js +32 -0
- package/dist/proxy/providers/index.js.map +1 -0
- package/dist/proxy/providers/messaging.d.ts +2 -0
- package/dist/proxy/providers/messaging.d.ts.map +1 -0
- package/dist/proxy/providers/messaging.js +53 -0
- package/dist/proxy/providers/messaging.js.map +1 -0
- package/dist/proxy/providers/openai.d.ts +4 -0
- package/dist/proxy/providers/openai.d.ts.map +1 -0
- package/dist/proxy/providers/openai.js +38 -0
- package/dist/proxy/providers/openai.js.map +1 -0
- package/dist/proxy/providers/telegram.d.ts +8 -0
- package/dist/proxy/providers/telegram.d.ts.map +1 -0
- package/dist/proxy/providers/telegram.js +35 -0
- package/dist/proxy/providers/telegram.js.map +1 -0
- package/dist/proxy/router.d.ts +12 -0
- package/dist/proxy/router.d.ts.map +1 -0
- package/dist/proxy/router.js +26 -0
- package/dist/proxy/router.js.map +1 -0
- package/dist/proxy/safety.d.ts +13 -0
- package/dist/proxy/safety.d.ts.map +1 -0
- package/dist/proxy/safety.js +58 -0
- package/dist/proxy/safety.js.map +1 -0
- package/dist/proxy/server.d.ts +8 -0
- package/dist/proxy/server.d.ts.map +1 -0
- package/dist/proxy/server.js +126 -0
- package/dist/proxy/server.js.map +1 -0
- package/dist/proxy/streaming.d.ts +21 -0
- package/dist/proxy/streaming.d.ts.map +1 -0
- package/dist/proxy/streaming.js +70 -0
- package/dist/proxy/streaming.js.map +1 -0
- package/dist/storage/database.d.ts +6 -0
- package/dist/storage/database.d.ts.map +1 -0
- package/dist/storage/database.js +44 -0
- package/dist/storage/database.js.map +1 -0
- package/dist/storage/encryption.d.ts +11 -0
- package/dist/storage/encryption.d.ts.map +1 -0
- package/dist/storage/encryption.js +47 -0
- package/dist/storage/encryption.js.map +1 -0
- package/dist/storage/migrations.d.ts +3 -0
- package/dist/storage/migrations.d.ts.map +1 -0
- package/dist/storage/migrations.js +265 -0
- package/dist/storage/migrations.js.map +1 -0
- package/dist/storage/repositories/audit-log.d.ts +115 -0
- package/dist/storage/repositories/audit-log.d.ts.map +1 -0
- package/dist/storage/repositories/audit-log.js +586 -0
- package/dist/storage/repositories/audit-log.js.map +1 -0
- package/dist/storage/repositories/cache.d.ts +26 -0
- package/dist/storage/repositories/cache.d.ts.map +1 -0
- package/dist/storage/repositories/cache.js +44 -0
- package/dist/storage/repositories/cache.js.map +1 -0
- package/dist/storage/repositories/dlp-config-history.d.ts +17 -0
- package/dist/storage/repositories/dlp-config-history.d.ts.map +1 -0
- package/dist/storage/repositories/dlp-config-history.js +30 -0
- package/dist/storage/repositories/dlp-config-history.js.map +1 -0
- package/dist/storage/repositories/dlp-events.d.ts +35 -0
- package/dist/storage/repositories/dlp-events.d.ts.map +1 -0
- package/dist/storage/repositories/dlp-events.js +57 -0
- package/dist/storage/repositories/dlp-events.js.map +1 -0
- package/dist/storage/repositories/dlp-patterns.d.ts +70 -0
- package/dist/storage/repositories/dlp-patterns.d.ts.map +1 -0
- package/dist/storage/repositories/dlp-patterns.js +187 -0
- package/dist/storage/repositories/dlp-patterns.js.map +1 -0
- package/dist/storage/repositories/optimizer-events.d.ts +28 -0
- package/dist/storage/repositories/optimizer-events.d.ts.map +1 -0
- package/dist/storage/repositories/optimizer-events.js +49 -0
- package/dist/storage/repositories/optimizer-events.js.map +1 -0
- package/dist/storage/repositories/plugin-events.d.ts +34 -0
- package/dist/storage/repositories/plugin-events.d.ts.map +1 -0
- package/dist/storage/repositories/plugin-events.js +64 -0
- package/dist/storage/repositories/plugin-events.js.map +1 -0
- package/dist/storage/repositories/requests.d.ts +68 -0
- package/dist/storage/repositories/requests.d.ts.map +1 -0
- package/dist/storage/repositories/requests.js +113 -0
- package/dist/storage/repositories/requests.js.map +1 -0
- package/dist/storage/repositories/sessions.d.ts +23 -0
- package/dist/storage/repositories/sessions.d.ts.map +1 -0
- package/dist/storage/repositories/sessions.js +42 -0
- package/dist/storage/repositories/sessions.js.map +1 -0
- package/dist/storage/repositories/tool-calls.d.ts +49 -0
- package/dist/storage/repositories/tool-calls.d.ts.map +1 -0
- package/dist/storage/repositories/tool-calls.js +61 -0
- package/dist/storage/repositories/tool-calls.js.map +1 -0
- package/dist/storage/repositories/tool-guard-rules.d.ts +50 -0
- package/dist/storage/repositories/tool-guard-rules.d.ts.map +1 -0
- package/dist/storage/repositories/tool-guard-rules.js +120 -0
- package/dist/storage/repositories/tool-guard-rules.js.map +1 -0
- package/dist/tool-guard/alert.d.ts +30 -0
- package/dist/tool-guard/alert.d.ts.map +1 -0
- package/dist/tool-guard/alert.js +113 -0
- package/dist/tool-guard/alert.js.map +1 -0
- package/dist/tool-guard/extractor.d.ts +10 -0
- package/dist/tool-guard/extractor.d.ts.map +1 -0
- package/dist/tool-guard/extractor.js +309 -0
- package/dist/tool-guard/extractor.js.map +1 -0
- package/dist/tool-guard/rules.d.ts +18 -0
- package/dist/tool-guard/rules.d.ts.map +1 -0
- package/dist/tool-guard/rules.js +255 -0
- package/dist/tool-guard/rules.js.map +1 -0
- package/dist/tool-guard/streaming-guard.d.ts +57 -0
- package/dist/tool-guard/streaming-guard.d.ts.map +1 -0
- package/dist/tool-guard/streaming-guard.js +389 -0
- package/dist/tool-guard/streaming-guard.js.map +1 -0
- package/dist/utils/hash.d.ts +2 -0
- package/dist/utils/hash.d.ts.map +1 -0
- package/dist/utils/hash.js +8 -0
- package/dist/utils/hash.js.map +1 -0
- package/dist/utils/logger.d.ts +11 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +54 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/timeout.d.ts +5 -0
- package/dist/utils/timeout.d.ts.map +1 -0
- package/dist/utils/timeout.js +26 -0
- package/dist/utils/timeout.js.map +1 -0
- package/dist/version.d.ts +5 -0
- package/dist/version.d.ts.map +1 -0
- package/dist/version.js +23 -0
- package/dist/version.js.map +1 -0
- package/package.json +67 -0
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.PluginEventsRepository = void 0;
|
|
7
|
+
const node_crypto_1 = __importDefault(require("node:crypto"));
|
|
8
|
+
class PluginEventsRepository {
|
|
9
|
+
db;
|
|
10
|
+
insertStmt;
|
|
11
|
+
constructor(db) {
|
|
12
|
+
this.db = db;
|
|
13
|
+
this.insertStmt = db.prepare(`
|
|
14
|
+
INSERT INTO plugin_events (id, plugin_name, request_id, type, severity, rule, detail, matched_text)
|
|
15
|
+
VALUES (@id, @plugin_name, @request_id, @type, @severity, @rule, @detail, @matched_text)
|
|
16
|
+
`);
|
|
17
|
+
}
|
|
18
|
+
insert(record) {
|
|
19
|
+
this.insertStmt.run(record);
|
|
20
|
+
}
|
|
21
|
+
insertEvent(pluginName, requestId, event) {
|
|
22
|
+
this.insert({
|
|
23
|
+
id: node_crypto_1.default.randomUUID(),
|
|
24
|
+
plugin_name: pluginName,
|
|
25
|
+
request_id: requestId,
|
|
26
|
+
type: event.type,
|
|
27
|
+
severity: event.severity,
|
|
28
|
+
rule: event.rule,
|
|
29
|
+
detail: event.detail,
|
|
30
|
+
matched_text: event.matchedText ?? null,
|
|
31
|
+
});
|
|
32
|
+
}
|
|
33
|
+
getRecent(limit = 20, sinceHours) {
|
|
34
|
+
if (sinceHours) {
|
|
35
|
+
return this.db.prepare(`
|
|
36
|
+
SELECT * FROM plugin_events
|
|
37
|
+
WHERE created_at > datetime('now', '-' || ? || ' hours')
|
|
38
|
+
ORDER BY created_at DESC LIMIT ?
|
|
39
|
+
`).all(sinceHours, limit);
|
|
40
|
+
}
|
|
41
|
+
return this.db.prepare('SELECT * FROM plugin_events ORDER BY created_at DESC LIMIT ?').all(limit);
|
|
42
|
+
}
|
|
43
|
+
getByPlugin(pluginName, limit = 20) {
|
|
44
|
+
return this.db.prepare('SELECT * FROM plugin_events WHERE plugin_name = ? ORDER BY created_at DESC LIMIT ?').all(pluginName, limit);
|
|
45
|
+
}
|
|
46
|
+
purgeOlderThan(hours) {
|
|
47
|
+
const result = this.db.prepare(`DELETE FROM plugin_events WHERE created_at < datetime('now', '-' || ? || ' hours')`).run(hours);
|
|
48
|
+
return result.changes;
|
|
49
|
+
}
|
|
50
|
+
getStats() {
|
|
51
|
+
const total = this.db.prepare('SELECT COUNT(*) as count FROM plugin_events').get();
|
|
52
|
+
const pluginRows = this.db.prepare('SELECT plugin_name, COUNT(*) as count FROM plugin_events GROUP BY plugin_name').all();
|
|
53
|
+
const typeRows = this.db.prepare('SELECT type, COUNT(*) as count FROM plugin_events GROUP BY type').all();
|
|
54
|
+
const by_plugin = {};
|
|
55
|
+
for (const row of pluginRows)
|
|
56
|
+
by_plugin[row.plugin_name] = row.count;
|
|
57
|
+
const by_type = {};
|
|
58
|
+
for (const row of typeRows)
|
|
59
|
+
by_type[row.type] = row.count;
|
|
60
|
+
return { total_events: total.count, by_plugin, by_type };
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
exports.PluginEventsRepository = PluginEventsRepository;
|
|
64
|
+
//# sourceMappingURL=plugin-events.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"plugin-events.js","sourceRoot":"","sources":["../../../src/storage/repositories/plugin-events.ts"],"names":[],"mappings":";;;;;;AAAA,8DAAiC;AAejC,MAAa,sBAAsB;IACzB,EAAE,CAAoB;IACtB,UAAU,CAAqB;IAEvC,YAAY,EAAqB;QAC/B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;;KAG5B,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,MAA6C;QAClD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,WAAW,CAAC,UAAkB,EAAE,SAAwB,EAAE,KAA6F;QACrJ,IAAI,CAAC,MAAM,CAAC;YACV,EAAE,EAAE,qBAAM,CAAC,UAAU,EAAE;YACvB,WAAW,EAAE,UAAU;YACvB,UAAU,EAAE,SAAS;YACrB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,YAAY,EAAE,KAAK,CAAC,WAAW,IAAI,IAAI;SACxC,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,QAAgB,EAAE,EAAE,UAAmB;QAC/C,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;OAItB,CAAC,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAwB,CAAC;QACnD,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CACpB,8DAA8D,CAC/D,CAAC,GAAG,CAAC,KAAK,CAAwB,CAAC;IACtC,CAAC;IAED,WAAW,CAAC,UAAkB,EAAE,QAAgB,EAAE;QAChD,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CACpB,oFAAoF,CACrF,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAwB,CAAC;IAClD,CAAC;IAED,cAAc,CAAC,KAAa;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC5B,oFAAoF,CACrF,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACb,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;IAED,QAAQ;QACN,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC,GAAG,EAAuB,CAAC;QACxG,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAChC,+EAA+E,CAChF,CAAC,GAAG,EAA8C,CAAC;QACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC9B,iEAAiE,CAClE,CAAC,GAAG,EAAuC,CAAC;QAE7C,MAAM,SAAS,GAA2B,EAAE,CAAC;QAC7C,KAAK,MAAM,GAAG,IAAI,UAAU;YAAE,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC;QAErE,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,KAAK,MAAM,GAAG,IAAI,QAAQ;YAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC;QAE1D,OAAO,EAAE,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;IAC3D,CAAC;CACF;AAxED,wDAwEC"}
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
import type Database from 'better-sqlite3';
|
|
2
|
+
export interface RequestRecord {
|
|
3
|
+
id: string;
|
|
4
|
+
provider: string;
|
|
5
|
+
model: string;
|
|
6
|
+
method: string;
|
|
7
|
+
path: string;
|
|
8
|
+
status_code: number | null;
|
|
9
|
+
input_tokens: number;
|
|
10
|
+
output_tokens: number;
|
|
11
|
+
cache_creation_tokens: number;
|
|
12
|
+
cache_read_tokens: number;
|
|
13
|
+
cost_usd: number;
|
|
14
|
+
latency_ms: number;
|
|
15
|
+
cached: number;
|
|
16
|
+
dlp_action: string | null;
|
|
17
|
+
dlp_findings: number;
|
|
18
|
+
session_id: string | null;
|
|
19
|
+
api_key_hash: string | null;
|
|
20
|
+
created_at: string;
|
|
21
|
+
}
|
|
22
|
+
export interface RequestStats {
|
|
23
|
+
total_requests: number;
|
|
24
|
+
total_cost_usd: number;
|
|
25
|
+
total_input_tokens: number;
|
|
26
|
+
total_output_tokens: number;
|
|
27
|
+
cache_hits: number;
|
|
28
|
+
avg_latency_ms: number;
|
|
29
|
+
by_provider: Record<string, {
|
|
30
|
+
requests: number;
|
|
31
|
+
cost_usd: number;
|
|
32
|
+
}>;
|
|
33
|
+
by_model: Record<string, {
|
|
34
|
+
requests: number;
|
|
35
|
+
cost_usd: number;
|
|
36
|
+
}>;
|
|
37
|
+
}
|
|
38
|
+
export interface StatsFilter {
|
|
39
|
+
sinceHours?: number;
|
|
40
|
+
sessionId?: string;
|
|
41
|
+
apiKeyHash?: string;
|
|
42
|
+
}
|
|
43
|
+
export interface SessionInfo {
|
|
44
|
+
session_id: string;
|
|
45
|
+
request_count: number;
|
|
46
|
+
total_cost_usd: number;
|
|
47
|
+
first_seen: string;
|
|
48
|
+
last_seen: string;
|
|
49
|
+
label: string | null;
|
|
50
|
+
source: string | null;
|
|
51
|
+
project_path: string | null;
|
|
52
|
+
}
|
|
53
|
+
export declare class RequestsRepository {
|
|
54
|
+
private db;
|
|
55
|
+
private insertStmt;
|
|
56
|
+
constructor(db: Database.Database);
|
|
57
|
+
insert(record: Omit<RequestRecord, 'created_at'>): void;
|
|
58
|
+
getStats(filter?: StatsFilter): RequestStats;
|
|
59
|
+
getRecent(limit?: number, sinceHours?: number): RequestRecord[];
|
|
60
|
+
purgeOlderThan(hours: number): number;
|
|
61
|
+
getSessions(): SessionInfo[];
|
|
62
|
+
getApiKeys(): {
|
|
63
|
+
api_key_hash: string;
|
|
64
|
+
request_count: number;
|
|
65
|
+
total_cost_usd: number;
|
|
66
|
+
}[];
|
|
67
|
+
}
|
|
68
|
+
//# sourceMappingURL=requests.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"requests.d.ts","sourceRoot":"","sources":["../../../src/storage/repositories/requests.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAE3C,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,qBAAqB,EAAE,MAAM,CAAC;IAC9B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,YAAY;IAC3B,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;IACvB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACpE,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAClE;AAED,MAAM,WAAW,WAAW;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,WAAW;IAC1B,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B;AAED,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,EAAE,CAAoB;IAC9B,OAAO,CAAC,UAAU,CAAqB;gBAE3B,EAAE,EAAE,QAAQ,CAAC,QAAQ;IAYjC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,aAAa,EAAE,YAAY,CAAC,GAAG,IAAI;IAIvD,QAAQ,CAAC,MAAM,CAAC,EAAE,WAAW,GAAG,YAAY;IAgE5C,SAAS,CAAC,KAAK,GAAE,MAAW,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,aAAa,EAAE;IAWnE,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;IAOrC,WAAW,IAAI,WAAW,EAAE;IAiB5B,UAAU,IAAI;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE,EAAE;CAYxF"}
|
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.RequestsRepository = void 0;
|
|
4
|
+
class RequestsRepository {
|
|
5
|
+
db;
|
|
6
|
+
insertStmt;
|
|
7
|
+
constructor(db) {
|
|
8
|
+
this.db = db;
|
|
9
|
+
this.insertStmt = db.prepare(`
|
|
10
|
+
INSERT INTO requests (id, provider, model, method, path, status_code,
|
|
11
|
+
input_tokens, output_tokens, cache_creation_tokens, cache_read_tokens,
|
|
12
|
+
cost_usd, latency_ms, cached, dlp_action, dlp_findings, session_id, api_key_hash)
|
|
13
|
+
VALUES (@id, @provider, @model, @method, @path, @status_code,
|
|
14
|
+
@input_tokens, @output_tokens, @cache_creation_tokens, @cache_read_tokens,
|
|
15
|
+
@cost_usd, @latency_ms, @cached, @dlp_action, @dlp_findings, @session_id, @api_key_hash)
|
|
16
|
+
`);
|
|
17
|
+
}
|
|
18
|
+
insert(record) {
|
|
19
|
+
this.insertStmt.run(record);
|
|
20
|
+
}
|
|
21
|
+
getStats(filter) {
|
|
22
|
+
const conditions = [];
|
|
23
|
+
const params = [];
|
|
24
|
+
if (filter?.sinceHours) {
|
|
25
|
+
conditions.push(`created_at >= datetime('now', '-' || ? || ' hours')`);
|
|
26
|
+
params.push(filter.sinceHours);
|
|
27
|
+
}
|
|
28
|
+
if (filter?.sessionId) {
|
|
29
|
+
conditions.push('session_id = ?');
|
|
30
|
+
params.push(filter.sessionId);
|
|
31
|
+
}
|
|
32
|
+
if (filter?.apiKeyHash) {
|
|
33
|
+
conditions.push('api_key_hash = ?');
|
|
34
|
+
params.push(filter.apiKeyHash);
|
|
35
|
+
}
|
|
36
|
+
const whereClause = conditions.length > 0
|
|
37
|
+
? 'WHERE ' + conditions.join(' AND ')
|
|
38
|
+
: '';
|
|
39
|
+
const totals = this.db.prepare(`
|
|
40
|
+
SELECT
|
|
41
|
+
COUNT(*) as total_requests,
|
|
42
|
+
COALESCE(SUM(cost_usd), 0) as total_cost_usd,
|
|
43
|
+
COALESCE(SUM(input_tokens), 0) as total_input_tokens,
|
|
44
|
+
COALESCE(SUM(output_tokens), 0) as total_output_tokens,
|
|
45
|
+
COALESCE(SUM(cached), 0) as cache_hits,
|
|
46
|
+
COALESCE(AVG(latency_ms), 0) as avg_latency_ms
|
|
47
|
+
FROM requests ${whereClause}
|
|
48
|
+
`).get(...params);
|
|
49
|
+
const providerRows = this.db.prepare(`
|
|
50
|
+
SELECT provider, COUNT(*) as requests, COALESCE(SUM(cost_usd), 0) as cost_usd
|
|
51
|
+
FROM requests ${whereClause}
|
|
52
|
+
GROUP BY provider
|
|
53
|
+
`).all(...params);
|
|
54
|
+
const modelRows = this.db.prepare(`
|
|
55
|
+
SELECT model, COUNT(*) as requests, COALESCE(SUM(cost_usd), 0) as cost_usd
|
|
56
|
+
FROM requests ${whereClause}
|
|
57
|
+
GROUP BY model
|
|
58
|
+
`).all(...params);
|
|
59
|
+
const by_provider = {};
|
|
60
|
+
for (const row of providerRows) {
|
|
61
|
+
by_provider[row.provider] = { requests: row.requests, cost_usd: row.cost_usd };
|
|
62
|
+
}
|
|
63
|
+
const by_model = {};
|
|
64
|
+
for (const row of modelRows) {
|
|
65
|
+
by_model[row.model] = { requests: row.requests, cost_usd: row.cost_usd };
|
|
66
|
+
}
|
|
67
|
+
return { ...totals, by_provider, by_model };
|
|
68
|
+
}
|
|
69
|
+
getRecent(limit = 10, sinceHours) {
|
|
70
|
+
if (sinceHours) {
|
|
71
|
+
return this.db.prepare(`
|
|
72
|
+
SELECT * FROM requests WHERE created_at > datetime('now', '-' || ? || ' hours') ORDER BY created_at DESC LIMIT ?
|
|
73
|
+
`).all(sinceHours, limit);
|
|
74
|
+
}
|
|
75
|
+
return this.db.prepare(`
|
|
76
|
+
SELECT * FROM requests ORDER BY created_at DESC LIMIT ?
|
|
77
|
+
`).all(limit);
|
|
78
|
+
}
|
|
79
|
+
purgeOlderThan(hours) {
|
|
80
|
+
const result = this.db.prepare(`DELETE FROM requests WHERE created_at < datetime('now', '-' || ? || ' hours')`).run(hours);
|
|
81
|
+
return result.changes;
|
|
82
|
+
}
|
|
83
|
+
getSessions() {
|
|
84
|
+
return this.db.prepare(`
|
|
85
|
+
SELECT
|
|
86
|
+
r.session_id,
|
|
87
|
+
COUNT(*) as request_count,
|
|
88
|
+
COALESCE(SUM(r.cost_usd), 0) as total_cost_usd,
|
|
89
|
+
MIN(r.created_at) as first_seen,
|
|
90
|
+
MAX(r.created_at) as last_seen,
|
|
91
|
+
s.label, s.source, s.project_path
|
|
92
|
+
FROM requests r
|
|
93
|
+
LEFT JOIN sessions s ON s.id = r.session_id
|
|
94
|
+
WHERE r.session_id IS NOT NULL
|
|
95
|
+
GROUP BY r.session_id
|
|
96
|
+
ORDER BY last_seen DESC
|
|
97
|
+
`).all();
|
|
98
|
+
}
|
|
99
|
+
getApiKeys() {
|
|
100
|
+
return this.db.prepare(`
|
|
101
|
+
SELECT
|
|
102
|
+
api_key_hash,
|
|
103
|
+
COUNT(*) as request_count,
|
|
104
|
+
COALESCE(SUM(cost_usd), 0) as total_cost_usd
|
|
105
|
+
FROM requests
|
|
106
|
+
WHERE api_key_hash IS NOT NULL
|
|
107
|
+
GROUP BY api_key_hash
|
|
108
|
+
ORDER BY request_count DESC
|
|
109
|
+
`).all();
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
exports.RequestsRepository = RequestsRepository;
|
|
113
|
+
//# sourceMappingURL=requests.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"requests.js","sourceRoot":"","sources":["../../../src/storage/repositories/requests.ts"],"names":[],"mappings":";;;AAmDA,MAAa,kBAAkB;IACrB,EAAE,CAAoB;IACtB,UAAU,CAAqB;IAEvC,YAAY,EAAqB;QAC/B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;;;;;;KAO5B,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,MAAyC;QAC9C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,QAAQ,CAAC,MAAoB;QAC3B,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAc,EAAE,CAAC;QAE7B,IAAI,MAAM,EAAE,UAAU,EAAE,CAAC;YACvB,UAAU,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAC;YACvE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,MAAM,EAAE,SAAS,EAAE,CAAC;YACtB,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAClC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,MAAM,EAAE,UAAU,EAAE,CAAC;YACvB,UAAU,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC;QAED,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC;YACvC,CAAC,CAAC,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC;YACrC,CAAC,CAAC,EAAE,CAAC;QAEP,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;;;sBAQb,WAAW;KAC5B,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAOf,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;sBAEnB,WAAW;;KAE5B,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAA+D,CAAC;QAEhF,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;sBAEhB,WAAW;;KAE5B,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAA4D,CAAC;QAE7E,MAAM,WAAW,GAA2D,EAAE,CAAC;QAC/E,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;QACjF,CAAC;QAED,MAAM,QAAQ,GAA2D,EAAE,CAAC;QAC5E,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;YAC5B,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC3E,CAAC;QAED,OAAO,EAAE,GAAG,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC;IAC9C,CAAC;IAED,SAAS,CAAC,QAAgB,EAAE,EAAE,UAAmB;QAC/C,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;OAEtB,CAAC,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAoB,CAAC;QAC/C,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;KAEtB,CAAC,CAAC,GAAG,CAAC,KAAK,CAAoB,CAAC;IACnC,CAAC;IAED,cAAc,CAAC,KAAa;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC5B,+EAA+E,CAChF,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACb,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;;;;;;;;KAatB,CAAC,CAAC,GAAG,EAAmB,CAAC;IAC5B,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;;;;KAStB,CAAC,CAAC,GAAG,EAA+E,CAAC;IACxF,CAAC;CACF;AAnID,gDAmIC"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import type Database from 'better-sqlite3';
|
|
2
|
+
export interface SessionRecord {
|
|
3
|
+
id: string;
|
|
4
|
+
label: string | null;
|
|
5
|
+
source: string;
|
|
6
|
+
project_path: string | null;
|
|
7
|
+
created_at: string;
|
|
8
|
+
last_seen_at: string;
|
|
9
|
+
}
|
|
10
|
+
export declare class SessionsRepository {
|
|
11
|
+
private db;
|
|
12
|
+
constructor(db: Database.Database);
|
|
13
|
+
upsert(id: string, info: {
|
|
14
|
+
label?: string;
|
|
15
|
+
source?: string;
|
|
16
|
+
projectPath?: string;
|
|
17
|
+
}): void;
|
|
18
|
+
get(id: string): SessionRecord | undefined;
|
|
19
|
+
touch(id: string): void;
|
|
20
|
+
getAll(limit?: number): SessionRecord[];
|
|
21
|
+
purgeOlderThan(hours: number): number;
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=sessions.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sessions.d.ts","sourceRoot":"","sources":["../../../src/storage/repositories/sessions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAE3C,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,EAAE,CAAoB;gBAElB,EAAE,EAAE,QAAQ,CAAC,QAAQ;IAIjC,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;IAkBzF,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS;IAI1C,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAIvB,MAAM,CAAC,KAAK,GAAE,MAAW,GAAG,aAAa,EAAE;IAI3C,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;CAMtC"}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.SessionsRepository = void 0;
|
|
4
|
+
class SessionsRepository {
|
|
5
|
+
db;
|
|
6
|
+
constructor(db) {
|
|
7
|
+
this.db = db;
|
|
8
|
+
}
|
|
9
|
+
upsert(id, info) {
|
|
10
|
+
const existing = this.get(id);
|
|
11
|
+
if (existing) {
|
|
12
|
+
// Update: only overwrite label/project_path if provided and currently null
|
|
13
|
+
const label = existing.label ?? info.label ?? null;
|
|
14
|
+
const projectPath = existing.project_path ?? info.projectPath ?? null;
|
|
15
|
+
this.db.prepare(`
|
|
16
|
+
UPDATE sessions SET label = ?, project_path = ?, last_seen_at = datetime('now')
|
|
17
|
+
WHERE id = ?
|
|
18
|
+
`).run(label, projectPath, id);
|
|
19
|
+
}
|
|
20
|
+
else {
|
|
21
|
+
this.db.prepare(`
|
|
22
|
+
INSERT INTO sessions (id, label, source, project_path)
|
|
23
|
+
VALUES (?, ?, ?, ?)
|
|
24
|
+
`).run(id, info.label ?? null, info.source ?? 'auto', info.projectPath ?? null);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
get(id) {
|
|
28
|
+
return this.db.prepare('SELECT * FROM sessions WHERE id = ?').get(id);
|
|
29
|
+
}
|
|
30
|
+
touch(id) {
|
|
31
|
+
this.db.prepare("UPDATE sessions SET last_seen_at = datetime('now') WHERE id = ?").run(id);
|
|
32
|
+
}
|
|
33
|
+
getAll(limit = 50) {
|
|
34
|
+
return this.db.prepare('SELECT * FROM sessions ORDER BY last_seen_at DESC LIMIT ?').all(limit);
|
|
35
|
+
}
|
|
36
|
+
purgeOlderThan(hours) {
|
|
37
|
+
const result = this.db.prepare(`DELETE FROM sessions WHERE last_seen_at < datetime('now', '-' || ? || ' hours')`).run(hours);
|
|
38
|
+
return result.changes;
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
exports.SessionsRepository = SessionsRepository;
|
|
42
|
+
//# sourceMappingURL=sessions.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sessions.js","sourceRoot":"","sources":["../../../src/storage/repositories/sessions.ts"],"names":[],"mappings":";;;AAWA,MAAa,kBAAkB;IACrB,EAAE,CAAoB;IAE9B,YAAY,EAAqB;QAC/B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,CAAC;IAED,MAAM,CAAC,EAAU,EAAE,IAA+D;QAChF,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC9B,IAAI,QAAQ,EAAE,CAAC;YACb,2EAA2E;YAC3E,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC;YACnD,MAAM,WAAW,GAAG,QAAQ,CAAC,YAAY,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC;YACtE,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;OAGf,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;OAGf,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,MAAM,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC;QAClF,CAAC;IACH,CAAC;IAED,GAAG,CAAC,EAAU;QACZ,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC,GAAG,CAAC,EAAE,CAA8B,CAAC;IACrG,CAAC;IAED,KAAK,CAAC,EAAU;QACd,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,iEAAiE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC7F,CAAC;IAED,MAAM,CAAC,QAAgB,EAAE;QACvB,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,2DAA2D,CAAC,CAAC,GAAG,CAAC,KAAK,CAAoB,CAAC;IACpH,CAAC;IAED,cAAc,CAAC,KAAa;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC5B,iFAAiF,CAClF,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACb,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;CACF;AA3CD,gDA2CC"}
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import type Database from 'better-sqlite3';
|
|
2
|
+
export interface ToolCallRecord {
|
|
3
|
+
id: string;
|
|
4
|
+
request_id: string;
|
|
5
|
+
tool_name: string;
|
|
6
|
+
tool_input: string | null;
|
|
7
|
+
rule_id: string | null;
|
|
8
|
+
rule_name: string | null;
|
|
9
|
+
severity: string | null;
|
|
10
|
+
category: string | null;
|
|
11
|
+
action: string | null;
|
|
12
|
+
provider: string | null;
|
|
13
|
+
session_id: string | null;
|
|
14
|
+
created_at: string;
|
|
15
|
+
}
|
|
16
|
+
export interface ToolCallStats {
|
|
17
|
+
total: number;
|
|
18
|
+
flagged: number;
|
|
19
|
+
bySeverity: Record<string, number>;
|
|
20
|
+
byCategory: Record<string, number>;
|
|
21
|
+
topToolNames: Array<{
|
|
22
|
+
tool_name: string;
|
|
23
|
+
count: number;
|
|
24
|
+
}>;
|
|
25
|
+
}
|
|
26
|
+
export declare class ToolCallsRepository {
|
|
27
|
+
private db;
|
|
28
|
+
private insertStmt;
|
|
29
|
+
constructor(db: Database.Database);
|
|
30
|
+
insert(record: {
|
|
31
|
+
id: string;
|
|
32
|
+
request_id: string;
|
|
33
|
+
tool_name: string;
|
|
34
|
+
tool_input: string;
|
|
35
|
+
rule_id: string | null;
|
|
36
|
+
rule_name: string | null;
|
|
37
|
+
severity: string | null;
|
|
38
|
+
category: string | null;
|
|
39
|
+
action: string | null;
|
|
40
|
+
provider: string;
|
|
41
|
+
session_id: string | null;
|
|
42
|
+
}): void;
|
|
43
|
+
getByRequestId(requestId: string): ToolCallRecord[];
|
|
44
|
+
getRecent(limit?: number, sinceHours?: number): ToolCallRecord[];
|
|
45
|
+
getBySession(sessionId: string): ToolCallRecord[];
|
|
46
|
+
getStats(): ToolCallStats;
|
|
47
|
+
purgeOlderThan(hours: number): number;
|
|
48
|
+
}
|
|
49
|
+
//# sourceMappingURL=tool-calls.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-calls.d.ts","sourceRoot":"","sources":["../../../src/storage/repositories/tool-calls.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAE3C,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnC,YAAY,EAAE,KAAK,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAC3D;AAID,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,EAAE,CAAoB;IAC9B,OAAO,CAAC,UAAU,CAAqB;gBAE3B,EAAE,EAAE,QAAQ,CAAC,QAAQ;IAQjC,MAAM,CAAC,MAAM,EAAE;QACb,EAAE,EAAE,MAAM,CAAC;QACX,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;QACzB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QACxB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;QACxB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;QACtB,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;KAC3B,GAAG,IAAI;IASR,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,cAAc,EAAE;IAMnD,SAAS,CAAC,KAAK,GAAE,MAAW,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,cAAc,EAAE;IAWpE,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,cAAc,EAAE;IAMjD,QAAQ,IAAI,aAAa;IAuBzB,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;CAMtC"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ToolCallsRepository = void 0;
|
|
4
|
+
const MAX_INPUT_BYTES = 2048;
|
|
5
|
+
class ToolCallsRepository {
|
|
6
|
+
db;
|
|
7
|
+
insertStmt;
|
|
8
|
+
constructor(db) {
|
|
9
|
+
this.db = db;
|
|
10
|
+
this.insertStmt = db.prepare(`
|
|
11
|
+
INSERT INTO tool_calls (id, request_id, tool_name, tool_input, rule_id, rule_name, severity, category, action, provider, session_id)
|
|
12
|
+
VALUES (@id, @request_id, @tool_name, @tool_input, @rule_id, @rule_name, @severity, @category, @action, @provider, @session_id)
|
|
13
|
+
`);
|
|
14
|
+
}
|
|
15
|
+
insert(record) {
|
|
16
|
+
this.insertStmt.run({
|
|
17
|
+
...record,
|
|
18
|
+
tool_input: record.tool_input.length > MAX_INPUT_BYTES
|
|
19
|
+
? record.tool_input.slice(0, MAX_INPUT_BYTES)
|
|
20
|
+
: record.tool_input,
|
|
21
|
+
});
|
|
22
|
+
}
|
|
23
|
+
getByRequestId(requestId) {
|
|
24
|
+
return this.db.prepare('SELECT * FROM tool_calls WHERE request_id = ? ORDER BY created_at ASC').all(requestId);
|
|
25
|
+
}
|
|
26
|
+
getRecent(limit = 50, sinceHours) {
|
|
27
|
+
if (sinceHours) {
|
|
28
|
+
return this.db.prepare(`
|
|
29
|
+
SELECT * FROM tool_calls WHERE created_at > datetime('now', '-' || ? || ' hours') ORDER BY created_at DESC LIMIT ?
|
|
30
|
+
`).all(sinceHours, limit);
|
|
31
|
+
}
|
|
32
|
+
return this.db.prepare(`
|
|
33
|
+
SELECT * FROM tool_calls ORDER BY created_at DESC LIMIT ?
|
|
34
|
+
`).all(limit);
|
|
35
|
+
}
|
|
36
|
+
getBySession(sessionId) {
|
|
37
|
+
return this.db.prepare(`
|
|
38
|
+
SELECT * FROM tool_calls WHERE session_id = ? ORDER BY created_at DESC
|
|
39
|
+
`).all(sessionId);
|
|
40
|
+
}
|
|
41
|
+
getStats() {
|
|
42
|
+
const total = this.db.prepare('SELECT COUNT(*) as c FROM tool_calls').get().c;
|
|
43
|
+
const flagged = this.db.prepare('SELECT COUNT(*) as c FROM tool_calls WHERE severity IS NOT NULL').get().c;
|
|
44
|
+
const bySeverityRows = this.db.prepare('SELECT severity, COUNT(*) as c FROM tool_calls WHERE severity IS NOT NULL GROUP BY severity').all();
|
|
45
|
+
const bySeverity = {};
|
|
46
|
+
for (const row of bySeverityRows)
|
|
47
|
+
bySeverity[row.severity] = row.c;
|
|
48
|
+
const byCategoryRows = this.db.prepare('SELECT category, COUNT(*) as c FROM tool_calls WHERE category IS NOT NULL GROUP BY category ORDER BY c DESC').all();
|
|
49
|
+
const byCategory = {};
|
|
50
|
+
for (const row of byCategoryRows)
|
|
51
|
+
byCategory[row.category] = row.c;
|
|
52
|
+
const topToolNames = this.db.prepare('SELECT tool_name, COUNT(*) as count FROM tool_calls GROUP BY tool_name ORDER BY count DESC LIMIT 10').all();
|
|
53
|
+
return { total, flagged, bySeverity, byCategory, topToolNames };
|
|
54
|
+
}
|
|
55
|
+
purgeOlderThan(hours) {
|
|
56
|
+
const result = this.db.prepare(`DELETE FROM tool_calls WHERE created_at < datetime('now', '-' || ? || ' hours')`).run(hours);
|
|
57
|
+
return result.changes;
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
exports.ToolCallsRepository = ToolCallsRepository;
|
|
61
|
+
//# sourceMappingURL=tool-calls.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-calls.js","sourceRoot":"","sources":["../../../src/storage/repositories/tool-calls.ts"],"names":[],"mappings":";;;AAyBA,MAAM,eAAe,GAAG,IAAI,CAAC;AAE7B,MAAa,mBAAmB;IACtB,EAAE,CAAoB;IACtB,UAAU,CAAqB;IAEvC,YAAY,EAAqB;QAC/B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;;KAG5B,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,MAYN;QACC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YAClB,GAAG,MAAM;YACT,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,eAAe;gBACpD,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC;gBAC7C,CAAC,CAAC,MAAM,CAAC,UAAU;SACtB,CAAC,CAAC;IACL,CAAC;IAED,cAAc,CAAC,SAAiB;QAC9B,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CACpB,uEAAuE,CACxE,CAAC,GAAG,CAAC,SAAS,CAAqB,CAAC;IACvC,CAAC;IAED,SAAS,CAAC,QAAgB,EAAE,EAAE,UAAmB;QAC/C,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;OAEtB,CAAC,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAqB,CAAC;QAChD,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;KAEtB,CAAC,CAAC,GAAG,CAAC,KAAK,CAAqB,CAAC;IACpC,CAAC;IAED,YAAY,CAAC,SAAiB;QAC5B,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;KAEtB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAqB,CAAC;IACxC,CAAC;IAED,QAAQ;QACN,MAAM,KAAK,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC,GAAG,EAAoB,CAAC,CAAC,CAAC;QACjG,MAAM,OAAO,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,iEAAiE,CAAC,CAAC,GAAG,EAAoB,CAAC,CAAC,CAAC;QAE9H,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CACpC,6FAA6F,CAC9F,CAAC,GAAG,EAA4C,CAAC;QAClD,MAAM,UAAU,GAA2B,EAAE,CAAC;QAC9C,KAAK,MAAM,GAAG,IAAI,cAAc;YAAE,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;QAEnE,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CACpC,6GAA6G,CAC9G,CAAC,GAAG,EAA4C,CAAC;QAClD,MAAM,UAAU,GAA2B,EAAE,CAAC;QAC9C,KAAK,MAAM,GAAG,IAAI,cAAc;YAAE,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;QAEnE,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAClC,qGAAqG,CACtG,CAAC,GAAG,EAAiD,CAAC;QAEvD,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC;IAClE,CAAC;IAED,cAAc,CAAC,KAAa;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC5B,iFAAiF,CAClF,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACb,OAAO,MAAM,CAAC,OAAO,CAAC;IACxB,CAAC;CACF;AArFD,kDAqFC"}
|
|
@@ -0,0 +1,50 @@
|
|
|
1
|
+
import type Database from 'better-sqlite3';
|
|
2
|
+
import type { ToolGuardRule } from '../../tool-guard/rules.js';
|
|
3
|
+
export interface ToolGuardRuleRecord {
|
|
4
|
+
id: string;
|
|
5
|
+
name: string;
|
|
6
|
+
description: string | null;
|
|
7
|
+
severity: string;
|
|
8
|
+
category: string;
|
|
9
|
+
tool_name_pattern: string | null;
|
|
10
|
+
tool_name_flags: string | null;
|
|
11
|
+
input_pattern: string;
|
|
12
|
+
input_flags: string;
|
|
13
|
+
enabled: number;
|
|
14
|
+
is_builtin: number;
|
|
15
|
+
created_at: string;
|
|
16
|
+
}
|
|
17
|
+
export declare class ToolGuardRulesRepository {
|
|
18
|
+
private db;
|
|
19
|
+
constructor(db: Database.Database);
|
|
20
|
+
/**
|
|
21
|
+
* Seed built-in rules.
|
|
22
|
+
*
|
|
23
|
+
* On first run: inserts all rules with enabled=1.
|
|
24
|
+
* On subsequent runs: updates rule definitions (patterns, severity, category)
|
|
25
|
+
* while preserving the user's enabled/disabled toggle state.
|
|
26
|
+
*/
|
|
27
|
+
seedBuiltins(rules: ToolGuardRule[]): void;
|
|
28
|
+
/** Get all enabled rules, converted to ToolGuardRule objects */
|
|
29
|
+
getEnabled(): ToolGuardRule[];
|
|
30
|
+
/** Get all rules for UI listing */
|
|
31
|
+
getAll(): ToolGuardRuleRecord[];
|
|
32
|
+
/** Toggle enabled flag */
|
|
33
|
+
toggle(id: string, enabled: boolean): void;
|
|
34
|
+
/** Insert or update a custom rule */
|
|
35
|
+
upsert(record: {
|
|
36
|
+
id: string;
|
|
37
|
+
name: string;
|
|
38
|
+
description?: string | null;
|
|
39
|
+
severity?: string;
|
|
40
|
+
category?: string;
|
|
41
|
+
tool_name_pattern?: string | null;
|
|
42
|
+
tool_name_flags?: string | null;
|
|
43
|
+
input_pattern: string;
|
|
44
|
+
input_flags?: string;
|
|
45
|
+
enabled?: boolean;
|
|
46
|
+
}): void;
|
|
47
|
+
/** Delete a custom rule. Rejects if is_builtin=1. Returns true if deleted. */
|
|
48
|
+
remove(id: string): boolean;
|
|
49
|
+
}
|
|
50
|
+
//# sourceMappingURL=tool-guard-rules.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-guard-rules.d.ts","sourceRoot":"","sources":["../../../src/storage/repositories/tool-guard-rules.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAC3C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE/D,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,wBAAwB;IACnC,OAAO,CAAC,EAAE,CAAoB;gBAElB,EAAE,EAAE,QAAQ,CAAC,QAAQ;IAIjC;;;;;;OAMG;IACH,YAAY,CAAC,KAAK,EAAE,aAAa,EAAE,GAAG,IAAI;IAqC1C,gEAAgE;IAChE,UAAU,IAAI,aAAa,EAAE;IAO7B,mCAAmC;IACnC,MAAM,IAAI,mBAAmB,EAAE;IAM/B,0BAA0B;IAC1B,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAM1C,qCAAqC;IACrC,MAAM,CAAC,MAAM,EAAE;QACb,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC5B,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAClC,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAChC,aAAa,EAAE,MAAM,CAAC;QACtB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,OAAO,CAAC,EAAE,OAAO,CAAC;KACnB,GAAG,IAAI;IA8BR,8EAA8E;IAC9E,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;CAa5B"}
|
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ToolGuardRulesRepository = void 0;
|
|
4
|
+
class ToolGuardRulesRepository {
|
|
5
|
+
db;
|
|
6
|
+
constructor(db) {
|
|
7
|
+
this.db = db;
|
|
8
|
+
}
|
|
9
|
+
/**
|
|
10
|
+
* Seed built-in rules.
|
|
11
|
+
*
|
|
12
|
+
* On first run: inserts all rules with enabled=1.
|
|
13
|
+
* On subsequent runs: updates rule definitions (patterns, severity, category)
|
|
14
|
+
* while preserving the user's enabled/disabled toggle state.
|
|
15
|
+
*/
|
|
16
|
+
seedBuiltins(rules) {
|
|
17
|
+
const stmt = this.db.prepare(`
|
|
18
|
+
INSERT INTO tool_guard_rules
|
|
19
|
+
(id, name, description, severity, category, tool_name_pattern, tool_name_flags, input_pattern, input_flags, enabled, is_builtin)
|
|
20
|
+
VALUES
|
|
21
|
+
(@id, @name, @description, @severity, @category, @tool_name_pattern, @tool_name_flags, @input_pattern, @input_flags, 1, 1)
|
|
22
|
+
ON CONFLICT(id) DO UPDATE SET
|
|
23
|
+
name = excluded.name,
|
|
24
|
+
description = excluded.description,
|
|
25
|
+
severity = excluded.severity,
|
|
26
|
+
category = excluded.category,
|
|
27
|
+
tool_name_pattern = excluded.tool_name_pattern,
|
|
28
|
+
tool_name_flags = excluded.tool_name_flags,
|
|
29
|
+
input_pattern = excluded.input_pattern,
|
|
30
|
+
input_flags = excluded.input_flags,
|
|
31
|
+
is_builtin = 1
|
|
32
|
+
-- enabled is intentionally omitted to preserve user toggle state
|
|
33
|
+
`);
|
|
34
|
+
const seed = this.db.transaction(() => {
|
|
35
|
+
for (const r of rules) {
|
|
36
|
+
stmt.run({
|
|
37
|
+
id: r.id,
|
|
38
|
+
name: r.name,
|
|
39
|
+
description: r.description ?? null,
|
|
40
|
+
severity: r.severity,
|
|
41
|
+
category: r.category,
|
|
42
|
+
tool_name_pattern: r.match.toolName?.source ?? null,
|
|
43
|
+
tool_name_flags: r.match.toolName?.flags ?? null,
|
|
44
|
+
input_pattern: r.match.inputPattern?.source ?? '',
|
|
45
|
+
input_flags: r.match.inputPattern?.flags ?? 'i',
|
|
46
|
+
});
|
|
47
|
+
}
|
|
48
|
+
});
|
|
49
|
+
seed();
|
|
50
|
+
}
|
|
51
|
+
/** Get all enabled rules, converted to ToolGuardRule objects */
|
|
52
|
+
getEnabled() {
|
|
53
|
+
const rows = this.db.prepare('SELECT * FROM tool_guard_rules WHERE enabled = 1').all();
|
|
54
|
+
return rows.map(rowToRule);
|
|
55
|
+
}
|
|
56
|
+
/** Get all rules for UI listing */
|
|
57
|
+
getAll() {
|
|
58
|
+
return this.db.prepare('SELECT * FROM tool_guard_rules ORDER BY is_builtin DESC, category, name').all();
|
|
59
|
+
}
|
|
60
|
+
/** Toggle enabled flag */
|
|
61
|
+
toggle(id, enabled) {
|
|
62
|
+
this.db.prepare('UPDATE tool_guard_rules SET enabled = ? WHERE id = ?').run(enabled ? 1 : 0, id);
|
|
63
|
+
}
|
|
64
|
+
/** Insert or update a custom rule */
|
|
65
|
+
upsert(record) {
|
|
66
|
+
this.db.prepare(`
|
|
67
|
+
INSERT INTO tool_guard_rules
|
|
68
|
+
(id, name, description, severity, category, tool_name_pattern, tool_name_flags, input_pattern, input_flags, enabled, is_builtin)
|
|
69
|
+
VALUES
|
|
70
|
+
(@id, @name, @description, @severity, @category, @tool_name_pattern, @tool_name_flags, @input_pattern, @input_flags, @enabled, 0)
|
|
71
|
+
ON CONFLICT(id) DO UPDATE SET
|
|
72
|
+
name = @name,
|
|
73
|
+
description = @description,
|
|
74
|
+
severity = @severity,
|
|
75
|
+
category = @category,
|
|
76
|
+
tool_name_pattern = @tool_name_pattern,
|
|
77
|
+
tool_name_flags = @tool_name_flags,
|
|
78
|
+
input_pattern = @input_pattern,
|
|
79
|
+
input_flags = @input_flags,
|
|
80
|
+
enabled = @enabled
|
|
81
|
+
`).run({
|
|
82
|
+
id: record.id,
|
|
83
|
+
name: record.name,
|
|
84
|
+
description: record.description ?? null,
|
|
85
|
+
severity: record.severity ?? 'medium',
|
|
86
|
+
category: record.category ?? 'custom',
|
|
87
|
+
tool_name_pattern: record.tool_name_pattern ?? null,
|
|
88
|
+
tool_name_flags: record.tool_name_flags ?? null,
|
|
89
|
+
input_pattern: record.input_pattern,
|
|
90
|
+
input_flags: record.input_flags ?? 'i',
|
|
91
|
+
enabled: record.enabled === false ? 0 : 1,
|
|
92
|
+
});
|
|
93
|
+
}
|
|
94
|
+
/** Delete a custom rule. Rejects if is_builtin=1. Returns true if deleted. */
|
|
95
|
+
remove(id) {
|
|
96
|
+
const row = this.db.prepare('SELECT is_builtin FROM tool_guard_rules WHERE id = ?').get(id);
|
|
97
|
+
if (!row)
|
|
98
|
+
return false;
|
|
99
|
+
if (row.is_builtin === 1) {
|
|
100
|
+
throw new Error('Cannot delete built-in rule');
|
|
101
|
+
}
|
|
102
|
+
this.db.prepare('DELETE FROM tool_guard_rules WHERE id = ?').run(id);
|
|
103
|
+
return true;
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
exports.ToolGuardRulesRepository = ToolGuardRulesRepository;
|
|
107
|
+
function rowToRule(row) {
|
|
108
|
+
return {
|
|
109
|
+
id: row.id,
|
|
110
|
+
name: row.name,
|
|
111
|
+
description: row.description ?? '',
|
|
112
|
+
severity: row.severity,
|
|
113
|
+
category: row.category,
|
|
114
|
+
match: {
|
|
115
|
+
toolName: row.tool_name_pattern ? new RegExp(row.tool_name_pattern, row.tool_name_flags ?? '') : undefined,
|
|
116
|
+
inputPattern: row.input_pattern ? new RegExp(row.input_pattern, row.input_flags ?? 'i') : undefined,
|
|
117
|
+
},
|
|
118
|
+
};
|
|
119
|
+
}
|
|
120
|
+
//# sourceMappingURL=tool-guard-rules.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-guard-rules.js","sourceRoot":"","sources":["../../../src/storage/repositories/tool-guard-rules.ts"],"names":[],"mappings":";;;AAkBA,MAAa,wBAAwB;IAC3B,EAAE,CAAoB;IAE9B,YAAY,EAAqB;QAC/B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,CAAC;IAED;;;;;;OAMG;IACH,YAAY,CAAC,KAAsB;QACjC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;;;;;;;;;;;KAgB5B,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YACpC,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;gBACtB,IAAI,CAAC,GAAG,CAAC;oBACP,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,WAAW,EAAE,CAAC,CAAC,WAAW,IAAI,IAAI;oBAClC,QAAQ,EAAE,CAAC,CAAC,QAAQ;oBACpB,QAAQ,EAAE,CAAC,CAAC,QAAQ;oBACpB,iBAAiB,EAAE,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,IAAI,IAAI;oBACnD,eAAe,EAAE,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,IAAI,IAAI;oBAChD,aAAa,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,IAAI,EAAE;oBACjD,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,IAAI,GAAG;iBAChD,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,EAAE,CAAC;IACT,CAAC;IAED,gEAAgE;IAChE,UAAU;QACR,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,kDAAkD,CACnD,CAAC,GAAG,EAA2B,CAAC;QACjC,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,mCAAmC;IACnC,MAAM;QACJ,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CACpB,yEAAyE,CAC1E,CAAC,GAAG,EAA2B,CAAC;IACnC,CAAC;IAED,0BAA0B;IAC1B,MAAM,CAAC,EAAU,EAAE,OAAgB;QACjC,IAAI,CAAC,EAAE,CAAC,OAAO,CACb,sDAAsD,CACvD,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED,qCAAqC;IACrC,MAAM,CAAC,MAWN;QACC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;;;;;;;;;;KAef,CAAC,CAAC,GAAG,CAAC;YACL,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,IAAI;YACvC,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,QAAQ;YACrC,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,QAAQ;YACrC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,IAAI;YACnD,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,IAAI;YAC/C,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,GAAG;YACtC,OAAO,EAAE,MAAM,CAAC,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;SAC1C,CAAC,CAAC;IACL,CAAC;IAED,8EAA8E;IAC9E,MAAM,CAAC,EAAU;QACf,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CACzB,sDAAsD,CACvD,CAAC,GAAG,CAAC,EAAE,CAAuC,CAAC;QAEhD,IAAI,CAAC,GAAG;YAAE,OAAO,KAAK,CAAC;QACvB,IAAI,GAAG,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,2CAA2C,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrE,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAjID,4DAiIC;AAED,SAAS,SAAS,CAAC,GAAwB;IACzC,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE;QAClC,QAAQ,EAAE,GAAG,CAAC,QAAqC;QACnD,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,KAAK,EAAE;YACL,QAAQ,EAAE,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,iBAAiB,EAAE,GAAG,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;YAC1G,YAAY,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,aAAa,EAAE,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS;SACpG;KACF,CAAC;AACJ,CAAC"}
|