natroc 0.0.1
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/AGENTS.md +494 -0
- package/LICENSE +7 -0
- package/README.md +0 -0
- package/install.ps1 +109 -0
- package/install.sh +132 -0
- package/package.json +77 -0
- package/server/dist/agent/agent-loop.d.ts +71 -0
- package/server/dist/agent/agent-loop.js +171 -0
- package/server/dist/agent/agent-loop.js.map +1 -0
- package/server/dist/agent/home-context.d.ts +29 -0
- package/server/dist/agent/home-context.js +134 -0
- package/server/dist/agent/home-context.js.map +1 -0
- package/server/dist/agent/improvement-engine.d.ts +23 -0
- package/server/dist/agent/improvement-engine.js +107 -0
- package/server/dist/agent/improvement-engine.js.map +1 -0
- package/server/dist/agent/tools/index.d.ts +14 -0
- package/server/dist/agent/tools/index.js +85 -0
- package/server/dist/agent/tools/index.js.map +1 -0
- package/server/dist/agent/tools/list-directory.d.ts +2 -0
- package/server/dist/agent/tools/list-directory.js +27 -0
- package/server/dist/agent/tools/list-directory.js.map +1 -0
- package/server/dist/agent/tools/read-file.d.ts +2 -0
- package/server/dist/agent/tools/read-file.js +30 -0
- package/server/dist/agent/tools/read-file.js.map +1 -0
- package/server/dist/agent/tools/run-command.d.ts +2 -0
- package/server/dist/agent/tools/run-command.js +72 -0
- package/server/dist/agent/tools/run-command.js.map +1 -0
- package/server/dist/agent/tools/system-info.d.ts +2 -0
- package/server/dist/agent/tools/system-info.js +28 -0
- package/server/dist/agent/tools/system-info.js.map +1 -0
- package/server/dist/agent/tools/types.d.ts +18 -0
- package/server/dist/agent/tools/types.js +2 -0
- package/server/dist/agent/tools/types.js.map +1 -0
- package/server/dist/agent/tools/util.d.ts +7 -0
- package/server/dist/agent/tools/util.js +24 -0
- package/server/dist/agent/tools/util.js.map +1 -0
- package/server/dist/agent/tools/write-file.d.ts +2 -0
- package/server/dist/agent/tools/write-file.js +25 -0
- package/server/dist/agent/tools/write-file.js.map +1 -0
- package/server/dist/app.d.ts +8 -0
- package/server/dist/app.js +39 -0
- package/server/dist/app.js.map +1 -0
- package/server/dist/auth/password.d.ts +7 -0
- package/server/dist/auth/password.js +20 -0
- package/server/dist/auth/password.js.map +1 -0
- package/server/dist/channels/channel-runtime.d.ts +32 -0
- package/server/dist/channels/channel-runtime.js +484 -0
- package/server/dist/channels/channel-runtime.js.map +1 -0
- package/server/dist/cli/agent-deliver.d.ts +14 -0
- package/server/dist/cli/agent-deliver.js +183 -0
- package/server/dist/cli/agent-deliver.js.map +1 -0
- package/server/dist/cli/args.d.ts +11 -0
- package/server/dist/cli/args.js +57 -0
- package/server/dist/cli/args.js.map +1 -0
- package/server/dist/cli/cli-token.d.ts +1 -0
- package/server/dist/cli/cli-token.js +22 -0
- package/server/dist/cli/cli-token.js.map +1 -0
- package/server/dist/cli/daemon.d.ts +2 -0
- package/server/dist/cli/daemon.js +252 -0
- package/server/dist/cli/daemon.js.map +1 -0
- package/server/dist/cli.d.ts +2 -0
- package/server/dist/cli.js +416 -0
- package/server/dist/cli.js.map +1 -0
- package/server/dist/config/natroc-home.d.ts +13 -0
- package/server/dist/config/natroc-home.js +730 -0
- package/server/dist/config/natroc-home.js.map +1 -0
- package/server/dist/gateway/agent-service.d.ts +16 -0
- package/server/dist/gateway/agent-service.js +261 -0
- package/server/dist/gateway/agent-service.js.map +1 -0
- package/server/dist/gateway/connection.d.ts +38 -0
- package/server/dist/gateway/connection.js +254 -0
- package/server/dist/gateway/connection.js.map +1 -0
- package/server/dist/gateway/gateway.d.ts +79 -0
- package/server/dist/gateway/gateway.js +150 -0
- package/server/dist/gateway/gateway.js.map +1 -0
- package/server/dist/gateway/index.d.ts +8 -0
- package/server/dist/gateway/index.js +26 -0
- package/server/dist/gateway/index.js.map +1 -0
- package/server/dist/gateway/protocol.d.ts +102 -0
- package/server/dist/gateway/protocol.js +63 -0
- package/server/dist/gateway/protocol.js.map +1 -0
- package/server/dist/gateway/rpc/agent.d.ts +3 -0
- package/server/dist/gateway/rpc/agent.js +174 -0
- package/server/dist/gateway/rpc/agent.js.map +1 -0
- package/server/dist/gateway/rpc/agents.d.ts +2 -0
- package/server/dist/gateway/rpc/agents.js +68 -0
- package/server/dist/gateway/rpc/agents.js.map +1 -0
- package/server/dist/gateway/rpc/auth.d.ts +3 -0
- package/server/dist/gateway/rpc/auth.js +180 -0
- package/server/dist/gateway/rpc/auth.js.map +1 -0
- package/server/dist/gateway/rpc/channels.d.ts +2 -0
- package/server/dist/gateway/rpc/channels.js +230 -0
- package/server/dist/gateway/rpc/channels.js.map +1 -0
- package/server/dist/gateway/rpc/conversations.d.ts +3 -0
- package/server/dist/gateway/rpc/conversations.js +36 -0
- package/server/dist/gateway/rpc/conversations.js.map +1 -0
- package/server/dist/gateway/rpc/projects.d.ts +3 -0
- package/server/dist/gateway/rpc/projects.js +49 -0
- package/server/dist/gateway/rpc/projects.js.map +1 -0
- package/server/dist/gateway/rpc/providers.d.ts +3 -0
- package/server/dist/gateway/rpc/providers.js +106 -0
- package/server/dist/gateway/rpc/providers.js.map +1 -0
- package/server/dist/gateway/rpc/usage.d.ts +2 -0
- package/server/dist/gateway/rpc/usage.js +41 -0
- package/server/dist/gateway/rpc/usage.js.map +1 -0
- package/server/dist/gateway/types.d.ts +43 -0
- package/server/dist/gateway/types.js +20 -0
- package/server/dist/gateway/types.js.map +1 -0
- package/server/dist/gateway/ws-server.d.ts +10 -0
- package/server/dist/gateway/ws-server.js +37 -0
- package/server/dist/gateway/ws-server.js.map +1 -0
- package/server/dist/index.d.ts +1 -0
- package/server/dist/index.js +9 -0
- package/server/dist/index.js.map +1 -0
- package/server/dist/local-runtime.d.ts +9 -0
- package/server/dist/local-runtime.js +16 -0
- package/server/dist/local-runtime.js.map +1 -0
- package/server/dist/providers/configured-adapters.d.ts +9 -0
- package/server/dist/providers/configured-adapters.js +34 -0
- package/server/dist/providers/configured-adapters.js.map +1 -0
- package/server/dist/providers/ollama.d.ts +23 -0
- package/server/dist/providers/ollama.js +164 -0
- package/server/dist/providers/ollama.js.map +1 -0
- package/server/dist/providers/openrouter.d.ts +24 -0
- package/server/dist/providers/openrouter.js +201 -0
- package/server/dist/providers/openrouter.js.map +1 -0
- package/server/dist/providers/thinking.d.ts +18 -0
- package/server/dist/providers/thinking.js +58 -0
- package/server/dist/providers/thinking.js.map +1 -0
- package/server/dist/providers/types.d.ts +55 -0
- package/server/dist/providers/types.js +2 -0
- package/server/dist/providers/types.js.map +1 -0
- package/server/dist/routes/schemas.d.ts +51 -0
- package/server/dist/routes/schemas.js +53 -0
- package/server/dist/routes/schemas.js.map +1 -0
- package/server/dist/runtime.d.ts +47 -0
- package/server/dist/runtime.js +29 -0
- package/server/dist/runtime.js.map +1 -0
- package/server/dist/server.d.ts +11 -0
- package/server/dist/server.js +19 -0
- package/server/dist/server.js.map +1 -0
- package/server/dist/storage/agent-repository.d.ts +59 -0
- package/server/dist/storage/agent-repository.js +192 -0
- package/server/dist/storage/agent-repository.js.map +1 -0
- package/server/dist/storage/auth-repository.d.ts +49 -0
- package/server/dist/storage/auth-repository.js +139 -0
- package/server/dist/storage/auth-repository.js.map +1 -0
- package/server/dist/storage/channel-repository.d.ts +152 -0
- package/server/dist/storage/channel-repository.js +413 -0
- package/server/dist/storage/channel-repository.js.map +1 -0
- package/server/dist/storage/conversation-repository.d.ts +63 -0
- package/server/dist/storage/conversation-repository.js +196 -0
- package/server/dist/storage/conversation-repository.js.map +1 -0
- package/server/dist/storage/database.d.ts +11 -0
- package/server/dist/storage/database.js +360 -0
- package/server/dist/storage/database.js.map +1 -0
- package/server/dist/storage/memory-repository.d.ts +70 -0
- package/server/dist/storage/memory-repository.js +279 -0
- package/server/dist/storage/memory-repository.js.map +1 -0
- package/server/dist/storage/project-repository.d.ts +25 -0
- package/server/dist/storage/project-repository.js +67 -0
- package/server/dist/storage/project-repository.js.map +1 -0
- package/server/dist/storage/provider-repository.d.ts +44 -0
- package/server/dist/storage/provider-repository.js +159 -0
- package/server/dist/storage/provider-repository.js.map +1 -0
- package/server/dist/storage/tool-call-repository.d.ts +35 -0
- package/server/dist/storage/tool-call-repository.js +83 -0
- package/server/dist/storage/tool-call-repository.js.map +1 -0
- package/server/dist/storage/usage-repository.d.ts +76 -0
- package/server/dist/storage/usage-repository.js +249 -0
- package/server/dist/storage/usage-repository.js.map +1 -0
- package/server/dist/storage/vault.d.ts +3 -0
- package/server/dist/storage/vault.js +57 -0
- package/server/dist/storage/vault.js.map +1 -0
- package/ui/README.md +0 -0
- package/ui/dist/assets/geist-cyrillic-ext-wght-normal-DjL33-gN.woff2 +0 -0
- package/ui/dist/assets/geist-cyrillic-wght-normal-BEAKL7Jp.woff2 +0 -0
- package/ui/dist/assets/geist-latin-ext-wght-normal-DC-KSUi6.woff2 +0 -0
- package/ui/dist/assets/geist-latin-wght-normal-BgDaEnEv.woff2 +0 -0
- package/ui/dist/assets/geist-vietnamese-wght-normal-6IgcOCM7.woff2 +0 -0
- package/ui/dist/assets/index-DKaFmZNO.js +114 -0
- package/ui/dist/assets/index-DOfPcjx3.css +2 -0
- package/ui/dist/index.html +14 -0
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
export class ToolCallRepository {
|
|
2
|
+
db;
|
|
3
|
+
constructor(db) {
|
|
4
|
+
this.db = db;
|
|
5
|
+
}
|
|
6
|
+
create(input) {
|
|
7
|
+
const id = input.id ?? crypto.randomUUID();
|
|
8
|
+
const now = new Date().toISOString();
|
|
9
|
+
const argsJson = JSON.stringify(input.args ?? {});
|
|
10
|
+
this.db
|
|
11
|
+
.prepare(`
|
|
12
|
+
INSERT INTO agent_tool_calls (
|
|
13
|
+
id,
|
|
14
|
+
user_id,
|
|
15
|
+
conversation_id,
|
|
16
|
+
tool_name,
|
|
17
|
+
args_json,
|
|
18
|
+
result_output,
|
|
19
|
+
status,
|
|
20
|
+
risk,
|
|
21
|
+
created_at
|
|
22
|
+
)
|
|
23
|
+
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
24
|
+
`)
|
|
25
|
+
.run(id, input.userId ?? null, input.conversationId, input.toolName, argsJson, input.output, input.status, input.risk, now);
|
|
26
|
+
return {
|
|
27
|
+
id,
|
|
28
|
+
userId: input.userId ?? null,
|
|
29
|
+
conversationId: input.conversationId,
|
|
30
|
+
toolName: input.toolName,
|
|
31
|
+
args: input.args ?? {},
|
|
32
|
+
output: input.output,
|
|
33
|
+
status: input.status,
|
|
34
|
+
risk: input.risk,
|
|
35
|
+
createdAt: now,
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
listByConversation(conversationId, options = {}) {
|
|
39
|
+
const rows = options.userId
|
|
40
|
+
? this.db
|
|
41
|
+
.prepare("SELECT * FROM agent_tool_calls WHERE conversation_id = ? AND user_id = ? ORDER BY created_at ASC, rowid ASC")
|
|
42
|
+
.all(conversationId, options.userId)
|
|
43
|
+
: this.db
|
|
44
|
+
.prepare("SELECT * FROM agent_tool_calls WHERE conversation_id = ? ORDER BY created_at ASC, rowid ASC")
|
|
45
|
+
.all(conversationId);
|
|
46
|
+
return rows.map(mapToolCallRow);
|
|
47
|
+
}
|
|
48
|
+
countAll(options = {}) {
|
|
49
|
+
const row = options.userId
|
|
50
|
+
? this.db
|
|
51
|
+
.prepare("SELECT COUNT(*) as count FROM agent_tool_calls WHERE user_id = ?")
|
|
52
|
+
.get(options.userId)
|
|
53
|
+
: this.db
|
|
54
|
+
.prepare("SELECT COUNT(*) as count FROM agent_tool_calls")
|
|
55
|
+
.get();
|
|
56
|
+
return row.count;
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
function mapToolCallRow(row) {
|
|
60
|
+
return {
|
|
61
|
+
id: row.id,
|
|
62
|
+
userId: row.user_id,
|
|
63
|
+
conversationId: row.conversation_id,
|
|
64
|
+
toolName: row.tool_name,
|
|
65
|
+
args: safeParseArgs(row.args_json),
|
|
66
|
+
output: row.result_output,
|
|
67
|
+
status: row.status,
|
|
68
|
+
risk: row.risk,
|
|
69
|
+
createdAt: row.created_at,
|
|
70
|
+
};
|
|
71
|
+
}
|
|
72
|
+
function safeParseArgs(value) {
|
|
73
|
+
try {
|
|
74
|
+
const parsed = JSON.parse(value);
|
|
75
|
+
return parsed && typeof parsed === "object" && !Array.isArray(parsed)
|
|
76
|
+
? parsed
|
|
77
|
+
: {};
|
|
78
|
+
}
|
|
79
|
+
catch {
|
|
80
|
+
return {};
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
//# sourceMappingURL=tool-call-repository.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-call-repository.js","sourceRoot":"","sources":["../../src/storage/tool-call-repository.ts"],"names":[],"mappings":"AAwCA,MAAM,OAAO,kBAAkB;IACA;IAA7B,YAA6B,EAAsB;QAAtB,OAAE,GAAF,EAAE,CAAoB;IAAG,CAAC;IAEvD,MAAM,CAAC,KAA0B;QAC/B,MAAM,EAAE,GAAG,KAAK,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE,CAAA;QAC1C,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAA;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAA;QAEjD,IAAI,CAAC,EAAE;aACJ,OAAO,CACN;;;;;;;;;;;;;OAaD,CACA;aACA,GAAG,CACF,EAAE,EACF,KAAK,CAAC,MAAM,IAAI,IAAI,EACpB,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,QAAQ,EACd,QAAQ,EACR,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,IAAI,EACV,GAAG,CACJ,CAAA;QAEH,OAAO;YACL,EAAE;YACF,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,IAAI;YAC5B,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE;YACtB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,GAAG;SACf,CAAA;IACH,CAAC;IAED,kBAAkB,CAChB,cAAsB,EACtB,UAA+B,EAAE;QAEjC,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM;YACzB,CAAC,CAAE,IAAI,CAAC,EAAE;iBACL,OAAO,CACN,6GAA6G,CAC9G;iBACA,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,MAAM,CAAmB;YAC1D,CAAC,CAAE,IAAI,CAAC,EAAE;iBACL,OAAO,CACN,6FAA6F,CAC9F;iBACA,GAAG,CAAC,cAAc,CAAmB,CAAA;QAE5C,OAAO,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,CAAA;IACjC,CAAC;IAED,QAAQ,CAAC,UAA+B,EAAE;QACxC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM;YACxB,CAAC,CAAE,IAAI,CAAC,EAAE;iBACL,OAAO,CAAC,kEAAkE,CAAC;iBAC3E,GAAG,CAAC,OAAO,CAAC,MAAM,CAAuB;YAC9C,CAAC,CAAE,IAAI,CAAC,EAAE;iBACL,OAAO,CAAC,gDAAgD,CAAC;iBACzD,GAAG,EAAwB,CAAA;QAElC,OAAO,GAAG,CAAC,KAAK,CAAA;IAClB,CAAC;CACF;AAED,SAAS,cAAc,CAAC,GAAgB;IACtC,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,MAAM,EAAE,GAAG,CAAC,OAAO;QACnB,cAAc,EAAE,GAAG,CAAC,eAAe;QACnC,QAAQ,EAAE,GAAG,CAAC,SAAS;QACvB,IAAI,EAAE,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC;QAClC,MAAM,EAAE,GAAG,CAAC,aAAa;QACzB,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,SAAS,EAAE,GAAG,CAAC,UAAU;KAC1B,CAAA;AACH,CAAC;AAED,SAAS,aAAa,CAAC,KAAa;IAClC,IAAI,CAAC;QACH,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QACzC,OAAO,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YACnE,CAAC,CAAE,MAAkC;YACrC,CAAC,CAAC,EAAE,CAAA;IACR,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAA;IACX,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
import type { DatabaseConnection } from "./database.js";
|
|
2
|
+
import type { ProviderName } from "../providers/types.js";
|
|
3
|
+
export type UsageStatus = "actual" | "unknown";
|
|
4
|
+
export type UsageEventStatus = "success" | "error";
|
|
5
|
+
export type UsageEventRecord = {
|
|
6
|
+
id: string;
|
|
7
|
+
userId: string;
|
|
8
|
+
provider: ProviderName;
|
|
9
|
+
model: string;
|
|
10
|
+
agentId: string | null;
|
|
11
|
+
channel: string;
|
|
12
|
+
conversationId: string | null;
|
|
13
|
+
messageId: string | null;
|
|
14
|
+
inputTokens: number | null;
|
|
15
|
+
outputTokens: number | null;
|
|
16
|
+
reasoningTokens: number | null;
|
|
17
|
+
totalTokens: number | null;
|
|
18
|
+
costUsd: number | null;
|
|
19
|
+
durationMs: number | null;
|
|
20
|
+
status: UsageEventStatus;
|
|
21
|
+
usageStatus: UsageStatus;
|
|
22
|
+
rawUsage: Record<string, unknown> | null;
|
|
23
|
+
createdAt: string;
|
|
24
|
+
};
|
|
25
|
+
export type CreateUsageEventInput = Omit<UsageEventRecord, "id" | "createdAt" | "rawUsage"> & {
|
|
26
|
+
rawUsage?: Record<string, unknown> | null;
|
|
27
|
+
};
|
|
28
|
+
export type UsageFilters = {
|
|
29
|
+
userId: string;
|
|
30
|
+
from?: string;
|
|
31
|
+
to?: string;
|
|
32
|
+
provider?: ProviderName;
|
|
33
|
+
model?: string;
|
|
34
|
+
channel?: string;
|
|
35
|
+
agentId?: string;
|
|
36
|
+
};
|
|
37
|
+
export type UsageAggregate = {
|
|
38
|
+
requests: number;
|
|
39
|
+
knownRequests: number;
|
|
40
|
+
unknownRequests: number;
|
|
41
|
+
inputTokens: number;
|
|
42
|
+
outputTokens: number;
|
|
43
|
+
reasoningTokens: number;
|
|
44
|
+
totalTokens: number;
|
|
45
|
+
costUsd: number;
|
|
46
|
+
durationMs: number;
|
|
47
|
+
};
|
|
48
|
+
export type UsageSummary = {
|
|
49
|
+
totals: UsageAggregate;
|
|
50
|
+
byProvider: Array<UsageAggregate & {
|
|
51
|
+
provider: string;
|
|
52
|
+
}>;
|
|
53
|
+
byModel: Array<UsageAggregate & {
|
|
54
|
+
provider: string;
|
|
55
|
+
model: string;
|
|
56
|
+
}>;
|
|
57
|
+
byChannel: Array<UsageAggregate & {
|
|
58
|
+
channel: string;
|
|
59
|
+
}>;
|
|
60
|
+
byAgent: Array<UsageAggregate & {
|
|
61
|
+
agentId: string | null;
|
|
62
|
+
}>;
|
|
63
|
+
};
|
|
64
|
+
export declare class UsageRepository {
|
|
65
|
+
private readonly db;
|
|
66
|
+
constructor(db: DatabaseConnection);
|
|
67
|
+
createEvent(input: CreateUsageEventInput): UsageEventRecord;
|
|
68
|
+
getSummary(filters: UsageFilters): UsageSummary;
|
|
69
|
+
listEvents(filters: UsageFilters & {
|
|
70
|
+
limit?: number;
|
|
71
|
+
offset?: number;
|
|
72
|
+
}): UsageEventRecord[];
|
|
73
|
+
exportCsv(filters: UsageFilters): string;
|
|
74
|
+
private aggregate;
|
|
75
|
+
private groupBy;
|
|
76
|
+
}
|
|
@@ -0,0 +1,249 @@
|
|
|
1
|
+
export class UsageRepository {
|
|
2
|
+
db;
|
|
3
|
+
constructor(db) {
|
|
4
|
+
this.db = db;
|
|
5
|
+
}
|
|
6
|
+
createEvent(input) {
|
|
7
|
+
const id = crypto.randomUUID();
|
|
8
|
+
const now = new Date().toISOString();
|
|
9
|
+
const totalTokens = input.totalTokens ??
|
|
10
|
+
sumNullable(input.inputTokens, input.outputTokens, input.reasoningTokens);
|
|
11
|
+
this.db
|
|
12
|
+
.prepare(`
|
|
13
|
+
INSERT INTO usage_events (
|
|
14
|
+
id,
|
|
15
|
+
user_id,
|
|
16
|
+
provider,
|
|
17
|
+
model,
|
|
18
|
+
agent_id,
|
|
19
|
+
channel,
|
|
20
|
+
conversation_id,
|
|
21
|
+
message_id,
|
|
22
|
+
input_tokens,
|
|
23
|
+
output_tokens,
|
|
24
|
+
reasoning_tokens,
|
|
25
|
+
total_tokens,
|
|
26
|
+
cost_usd,
|
|
27
|
+
duration_ms,
|
|
28
|
+
status,
|
|
29
|
+
usage_status,
|
|
30
|
+
raw_usage_json,
|
|
31
|
+
created_at
|
|
32
|
+
)
|
|
33
|
+
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
34
|
+
`)
|
|
35
|
+
.run(id, input.userId, input.provider, input.model || "unknown", input.agentId, input.channel || "app", input.conversationId, input.messageId, input.inputTokens, input.outputTokens, input.reasoningTokens, totalTokens, input.costUsd, input.durationMs, input.status, input.usageStatus, input.rawUsage ? JSON.stringify(input.rawUsage) : null, now);
|
|
36
|
+
return {
|
|
37
|
+
...input,
|
|
38
|
+
id,
|
|
39
|
+
model: input.model || "unknown",
|
|
40
|
+
channel: input.channel || "app",
|
|
41
|
+
totalTokens,
|
|
42
|
+
rawUsage: input.rawUsage ?? null,
|
|
43
|
+
createdAt: now,
|
|
44
|
+
};
|
|
45
|
+
}
|
|
46
|
+
getSummary(filters) {
|
|
47
|
+
return {
|
|
48
|
+
totals: this.aggregate(filters),
|
|
49
|
+
byProvider: this.groupBy(filters, ["provider"]),
|
|
50
|
+
byModel: this.groupBy(filters, ["provider", "model"]),
|
|
51
|
+
byChannel: this.groupBy(filters, ["channel"]),
|
|
52
|
+
byAgent: this.groupBy(filters, ["agent_id"]),
|
|
53
|
+
};
|
|
54
|
+
}
|
|
55
|
+
listEvents(filters) {
|
|
56
|
+
const query = buildFilterQuery(filters);
|
|
57
|
+
const rows = this.db
|
|
58
|
+
.prepare(`
|
|
59
|
+
SELECT * FROM usage_events
|
|
60
|
+
WHERE ${query.where}
|
|
61
|
+
ORDER BY created_at DESC, rowid DESC
|
|
62
|
+
LIMIT ? OFFSET ?
|
|
63
|
+
`)
|
|
64
|
+
.all(...query.values, Math.max(1, Math.min(filters.limit ?? 100, 500)), Math.max(0, filters.offset ?? 0));
|
|
65
|
+
return rows.map(mapUsageRow);
|
|
66
|
+
}
|
|
67
|
+
exportCsv(filters) {
|
|
68
|
+
const rows = this.listEvents({ ...filters, limit: 500, offset: 0 });
|
|
69
|
+
const header = [
|
|
70
|
+
"createdAt",
|
|
71
|
+
"provider",
|
|
72
|
+
"model",
|
|
73
|
+
"channel",
|
|
74
|
+
"agentId",
|
|
75
|
+
"conversationId",
|
|
76
|
+
"inputTokens",
|
|
77
|
+
"outputTokens",
|
|
78
|
+
"reasoningTokens",
|
|
79
|
+
"totalTokens",
|
|
80
|
+
"costUsd",
|
|
81
|
+
"durationMs",
|
|
82
|
+
"usageStatus",
|
|
83
|
+
"status",
|
|
84
|
+
];
|
|
85
|
+
return [
|
|
86
|
+
header.join(","),
|
|
87
|
+
...rows.map((row) => [
|
|
88
|
+
row.createdAt,
|
|
89
|
+
row.provider,
|
|
90
|
+
row.model,
|
|
91
|
+
row.channel,
|
|
92
|
+
row.agentId ?? "",
|
|
93
|
+
row.conversationId ?? "",
|
|
94
|
+
row.inputTokens ?? "",
|
|
95
|
+
row.outputTokens ?? "",
|
|
96
|
+
row.reasoningTokens ?? "",
|
|
97
|
+
row.totalTokens ?? "",
|
|
98
|
+
row.costUsd ?? "",
|
|
99
|
+
row.durationMs ?? "",
|
|
100
|
+
row.usageStatus,
|
|
101
|
+
row.status,
|
|
102
|
+
]
|
|
103
|
+
.map(csvCell)
|
|
104
|
+
.join(",")),
|
|
105
|
+
].join("\n");
|
|
106
|
+
}
|
|
107
|
+
aggregate(filters) {
|
|
108
|
+
const query = buildFilterQuery(filters);
|
|
109
|
+
const row = this.db
|
|
110
|
+
.prepare(`
|
|
111
|
+
SELECT ${aggregateSelect()}
|
|
112
|
+
FROM usage_events
|
|
113
|
+
WHERE ${query.where}
|
|
114
|
+
`)
|
|
115
|
+
.get(...query.values);
|
|
116
|
+
return mapAggregateRow(row);
|
|
117
|
+
}
|
|
118
|
+
groupBy(filters, columns) {
|
|
119
|
+
const query = buildFilterQuery(filters);
|
|
120
|
+
const selectColumns = columns.join(", ");
|
|
121
|
+
const rows = this.db
|
|
122
|
+
.prepare(`
|
|
123
|
+
SELECT ${selectColumns}, ${aggregateSelect()}
|
|
124
|
+
FROM usage_events
|
|
125
|
+
WHERE ${query.where}
|
|
126
|
+
GROUP BY ${selectColumns}
|
|
127
|
+
ORDER BY total_tokens DESC, requests DESC
|
|
128
|
+
`)
|
|
129
|
+
.all(...query.values);
|
|
130
|
+
return rows.map((row) => ({
|
|
131
|
+
...mapGroupColumns(row),
|
|
132
|
+
...mapAggregateRow(row),
|
|
133
|
+
}));
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
function buildFilterQuery(filters) {
|
|
137
|
+
const conditions = ["user_id = ?"];
|
|
138
|
+
const values = [filters.userId];
|
|
139
|
+
if (filters.from) {
|
|
140
|
+
conditions.push("created_at >= ?");
|
|
141
|
+
values.push(filters.from);
|
|
142
|
+
}
|
|
143
|
+
if (filters.to) {
|
|
144
|
+
conditions.push("created_at <= ?");
|
|
145
|
+
values.push(filters.to);
|
|
146
|
+
}
|
|
147
|
+
if (filters.provider) {
|
|
148
|
+
conditions.push("provider = ?");
|
|
149
|
+
values.push(filters.provider);
|
|
150
|
+
}
|
|
151
|
+
if (filters.model) {
|
|
152
|
+
conditions.push("model = ?");
|
|
153
|
+
values.push(filters.model);
|
|
154
|
+
}
|
|
155
|
+
if (filters.channel) {
|
|
156
|
+
conditions.push("channel = ?");
|
|
157
|
+
values.push(filters.channel);
|
|
158
|
+
}
|
|
159
|
+
if (filters.agentId) {
|
|
160
|
+
conditions.push("agent_id = ?");
|
|
161
|
+
values.push(filters.agentId);
|
|
162
|
+
}
|
|
163
|
+
return { where: conditions.join(" AND "), values };
|
|
164
|
+
}
|
|
165
|
+
function aggregateSelect() {
|
|
166
|
+
return `
|
|
167
|
+
COUNT(*) AS requests,
|
|
168
|
+
SUM(CASE WHEN usage_status = 'actual' THEN 1 ELSE 0 END) AS known_requests,
|
|
169
|
+
SUM(CASE WHEN usage_status = 'unknown' THEN 1 ELSE 0 END) AS unknown_requests,
|
|
170
|
+
SUM(COALESCE(input_tokens, 0)) AS input_tokens,
|
|
171
|
+
SUM(COALESCE(output_tokens, 0)) AS output_tokens,
|
|
172
|
+
SUM(COALESCE(reasoning_tokens, 0)) AS reasoning_tokens,
|
|
173
|
+
SUM(COALESCE(total_tokens, 0)) AS total_tokens,
|
|
174
|
+
SUM(COALESCE(cost_usd, 0)) AS cost_usd,
|
|
175
|
+
SUM(COALESCE(duration_ms, 0)) AS duration_ms
|
|
176
|
+
`;
|
|
177
|
+
}
|
|
178
|
+
function mapAggregateRow(row) {
|
|
179
|
+
return {
|
|
180
|
+
requests: Number(row.requests ?? 0),
|
|
181
|
+
knownRequests: Number(row.known_requests ?? 0),
|
|
182
|
+
unknownRequests: Number(row.unknown_requests ?? 0),
|
|
183
|
+
inputTokens: Number(row.input_tokens ?? 0),
|
|
184
|
+
outputTokens: Number(row.output_tokens ?? 0),
|
|
185
|
+
reasoningTokens: Number(row.reasoning_tokens ?? 0),
|
|
186
|
+
totalTokens: Number(row.total_tokens ?? 0),
|
|
187
|
+
costUsd: Number(row.cost_usd ?? 0),
|
|
188
|
+
durationMs: Number(row.duration_ms ?? 0),
|
|
189
|
+
};
|
|
190
|
+
}
|
|
191
|
+
function mapGroupColumns(row) {
|
|
192
|
+
return {
|
|
193
|
+
provider: typeof row.provider === "string" ? row.provider : undefined,
|
|
194
|
+
model: typeof row.model === "string" ? row.model : undefined,
|
|
195
|
+
channel: typeof row.channel === "string" ? row.channel : undefined,
|
|
196
|
+
agentId: typeof row.agent_id === "string" ? row.agent_id : null,
|
|
197
|
+
};
|
|
198
|
+
}
|
|
199
|
+
function mapUsageRow(row) {
|
|
200
|
+
return {
|
|
201
|
+
id: row.id,
|
|
202
|
+
userId: row.user_id,
|
|
203
|
+
provider: row.provider,
|
|
204
|
+
model: row.model,
|
|
205
|
+
agentId: row.agent_id,
|
|
206
|
+
channel: row.channel,
|
|
207
|
+
conversationId: row.conversation_id,
|
|
208
|
+
messageId: row.message_id,
|
|
209
|
+
inputTokens: row.input_tokens,
|
|
210
|
+
outputTokens: row.output_tokens,
|
|
211
|
+
reasoningTokens: row.reasoning_tokens,
|
|
212
|
+
totalTokens: row.total_tokens,
|
|
213
|
+
costUsd: row.cost_usd,
|
|
214
|
+
durationMs: row.duration_ms,
|
|
215
|
+
status: row.status,
|
|
216
|
+
usageStatus: row.usage_status,
|
|
217
|
+
rawUsage: safeParseRaw(row.raw_usage_json),
|
|
218
|
+
createdAt: row.created_at,
|
|
219
|
+
};
|
|
220
|
+
}
|
|
221
|
+
function safeParseRaw(value) {
|
|
222
|
+
if (!value)
|
|
223
|
+
return null;
|
|
224
|
+
try {
|
|
225
|
+
const parsed = JSON.parse(value);
|
|
226
|
+
return parsed && typeof parsed === "object" && !Array.isArray(parsed)
|
|
227
|
+
? parsed
|
|
228
|
+
: null;
|
|
229
|
+
}
|
|
230
|
+
catch {
|
|
231
|
+
return null;
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
function sumNullable(...values) {
|
|
235
|
+
let total = 0;
|
|
236
|
+
let hasValue = false;
|
|
237
|
+
for (const value of values) {
|
|
238
|
+
if (typeof value !== "number")
|
|
239
|
+
continue;
|
|
240
|
+
total += value;
|
|
241
|
+
hasValue = true;
|
|
242
|
+
}
|
|
243
|
+
return hasValue ? total : null;
|
|
244
|
+
}
|
|
245
|
+
function csvCell(value) {
|
|
246
|
+
const text = String(value);
|
|
247
|
+
return /[",\n\r]/.test(text) ? `"${text.replaceAll('"', '""')}"` : text;
|
|
248
|
+
}
|
|
249
|
+
//# sourceMappingURL=usage-repository.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage-repository.js","sourceRoot":"","sources":["../../src/storage/usage-repository.ts"],"names":[],"mappings":"AAiGA,MAAM,OAAO,eAAe;IACG;IAA7B,YAA6B,EAAsB;QAAtB,OAAE,GAAF,EAAE,CAAoB;IAAG,CAAC;IAEvD,WAAW,CAAC,KAA4B;QACtC,MAAM,EAAE,GAAG,MAAM,CAAC,UAAU,EAAE,CAAA;QAC9B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAA;QACpC,MAAM,WAAW,GACf,KAAK,CAAC,WAAW;YACjB,WAAW,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,eAAe,CAAC,CAAA;QAE3E,IAAI,CAAC,EAAE;aACJ,OAAO,CACN;;;;;;;;;;;;;;;;;;;;;;SAsBC,CACF;aACA,GAAG,CACF,EAAE,EACF,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,QAAQ,EACd,KAAK,CAAC,KAAK,IAAI,SAAS,EACxB,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,OAAO,IAAI,KAAK,EACtB,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,WAAW,EACjB,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,eAAe,EACrB,WAAW,EACX,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,WAAW,EACjB,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,EACtD,GAAG,CACJ,CAAA;QAEH,OAAO;YACL,GAAG,KAAK;YACR,EAAE;YACF,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,SAAS;YAC/B,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,KAAK;YAC/B,WAAW;YACX,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI;YAChC,SAAS,EAAE,GAAG;SACf,CAAA;IACH,CAAC;IAED,UAAU,CAAC,OAAqB;QAC9B,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAC/B,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,CAE7C;YACD,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,UAAU,EAAE,OAAO,CAAC,CAEnD;YACD,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,SAAS,CAAC,CAE3C;YACD,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,CAE1C;SACF,CAAA;IACH,CAAC;IAED,UAAU,CACR,OAA2D;QAE3D,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAA;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACjB,OAAO,CACN;;kBAEU,KAAK,CAAC,KAAK;;;SAGpB,CACF;aACA,GAAG,CACF,GAAG,KAAK,CAAC,MAAM,EACf,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,GAAG,EAAE,GAAG,CAAC,CAAC,EAChD,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC,CACd,CAAA;QAEtB,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAA;IAC9B,CAAC;IAED,SAAS,CAAC,OAAqB;QAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAA;QACnE,MAAM,MAAM,GAAG;YACb,WAAW;YACX,UAAU;YACV,OAAO;YACP,SAAS;YACT,SAAS;YACT,gBAAgB;YAChB,aAAa;YACb,cAAc;YACd,iBAAiB;YACjB,aAAa;YACb,SAAS;YACT,YAAY;YACZ,aAAa;YACb,QAAQ;SACT,CAAA;QAED,OAAO;YACL,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;YAChB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAClB;gBACE,GAAG,CAAC,SAAS;gBACb,GAAG,CAAC,QAAQ;gBACZ,GAAG,CAAC,KAAK;gBACT,GAAG,CAAC,OAAO;gBACX,GAAG,CAAC,OAAO,IAAI,EAAE;gBACjB,GAAG,CAAC,cAAc,IAAI,EAAE;gBACxB,GAAG,CAAC,WAAW,IAAI,EAAE;gBACrB,GAAG,CAAC,YAAY,IAAI,EAAE;gBACtB,GAAG,CAAC,eAAe,IAAI,EAAE;gBACzB,GAAG,CAAC,WAAW,IAAI,EAAE;gBACrB,GAAG,CAAC,OAAO,IAAI,EAAE;gBACjB,GAAG,CAAC,UAAU,IAAI,EAAE;gBACpB,GAAG,CAAC,WAAW;gBACf,GAAG,CAAC,MAAM;aACX;iBACE,GAAG,CAAC,OAAO,CAAC;iBACZ,IAAI,CAAC,GAAG,CAAC,CACb;SACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACd,CAAC;IAEO,SAAS,CAAC,OAAqB;QACrC,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAA;QACvC,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aAChB,OAAO,CACN;mBACW,eAAe,EAAE;;kBAElB,KAAK,CAAC,KAAK;SACpB,CACF;aACA,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAiB,CAAA;QAEvC,OAAO,eAAe,CAAC,GAAG,CAAC,CAAA;IAC7B,CAAC;IAEO,OAAO,CAAC,OAAqB,EAAE,OAAiB;QACtD,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAA;QACvC,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACjB,OAAO,CACN;mBACW,aAAa,KAAK,eAAe,EAAE;;kBAEpC,KAAK,CAAC,KAAK;qBACR,aAAa;;SAEzB,CACF;aACA,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAkD,CAAA;QAExE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,GAAG,eAAe,CAAC,GAAG,CAAC;YACvB,GAAG,eAAe,CAAC,GAAG,CAAC;SACxB,CAAC,CAAC,CAAA;IACL,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,OAAqB;IAC7C,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,CAAA;IAClC,MAAM,MAAM,GAAkB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IAE9C,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QACjB,UAAU,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QAClC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;IAC3B,CAAC;IACD,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;QACf,UAAU,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QAClC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IACzB,CAAC;IACD,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrB,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;QAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;IAC/B,CAAC;IACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAC5B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC5B,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QAC9B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;IAC9B,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;QAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;IAC9B,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAA;AACpD,CAAC;AAED,SAAS,eAAe;IACtB,OAAO;;;;;;;;;;GAUN,CAAA;AACH,CAAC;AAED,SAAS,eAAe,CAAC,GAAiB;IACxC,OAAO;QACL,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,CAAC;QACnC,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,CAAC;QAC9C,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,gBAAgB,IAAI,CAAC,CAAC;QAClD,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC;QAC1C,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,CAAC;QAC5C,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,gBAAgB,IAAI,CAAC,CAAC;QAClD,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC;QAC1C,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,CAAC;QAClC,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC;KACzC,CAAA;AACH,CAAC;AAED,SAAS,eAAe,CAAC,GAA4B;IACnD,OAAO;QACL,QAAQ,EAAE,OAAO,GAAG,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;QACrE,KAAK,EAAE,OAAO,GAAG,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;QAC5D,OAAO,EAAE,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QAClE,OAAO,EAAE,OAAO,GAAG,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI;KAChE,CAAA;AACH,CAAC;AAED,SAAS,WAAW,CAAC,GAAkB;IACrC,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,MAAM,EAAE,GAAG,CAAC,OAAO;QACnB,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,OAAO,EAAE,GAAG,CAAC,QAAQ;QACrB,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,cAAc,EAAE,GAAG,CAAC,eAAe;QACnC,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,YAAY,EAAE,GAAG,CAAC,aAAa;QAC/B,eAAe,EAAE,GAAG,CAAC,gBAAgB;QACrC,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,OAAO,EAAE,GAAG,CAAC,QAAQ;QACrB,UAAU,EAAE,GAAG,CAAC,WAAW;QAC3B,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,QAAQ,EAAE,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC;QAC1C,SAAS,EAAE,GAAG,CAAC,UAAU;KAC1B,CAAA;AACH,CAAC;AAED,SAAS,YAAY,CAAC,KAAoB;IACxC,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAA;IAEvB,IAAI,CAAC;QACH,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QACzC,OAAO,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YACnE,CAAC,CAAE,MAAkC;YACrC,CAAC,CAAC,IAAI,CAAA;IACV,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAA;IACb,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,GAAG,MAA4B;IAClD,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,IAAI,QAAQ,GAAG,KAAK,CAAA;IAEpB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,SAAQ;QACvC,KAAK,IAAI,KAAK,CAAA;QACd,QAAQ,GAAG,IAAI,CAAA;IACjB,CAAC;IAED,OAAO,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAA;AAChC,CAAC;AAED,SAAS,OAAO,CAAC,KAAsB;IACrC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAA;IAC1B,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAA;AACzE,CAAC"}
|
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
import { randomBytes, createCipheriv, createDecipheriv } from "node:crypto";
|
|
2
|
+
import { mkdir, readFile, writeFile, chmod } from "node:fs/promises";
|
|
3
|
+
import { dirname } from "node:path";
|
|
4
|
+
const KEY_LENGTH = 32;
|
|
5
|
+
const IV_LENGTH = 12;
|
|
6
|
+
const FORMAT_VERSION = "v1";
|
|
7
|
+
export async function ensureMasterKey(path) {
|
|
8
|
+
await mkdir(dirname(path), { recursive: true });
|
|
9
|
+
try {
|
|
10
|
+
const existingKey = (await readFile(path, "utf8")).trim();
|
|
11
|
+
return Buffer.from(existingKey, "hex");
|
|
12
|
+
}
|
|
13
|
+
catch (error) {
|
|
14
|
+
if (!(error instanceof Error) || !("code" in error) || error.code !== "ENOENT") {
|
|
15
|
+
throw error;
|
|
16
|
+
}
|
|
17
|
+
}
|
|
18
|
+
const key = randomBytes(KEY_LENGTH);
|
|
19
|
+
await writeFile(path, key.toString("hex"), { mode: 0o600, flag: "wx" });
|
|
20
|
+
await chmod(path, 0o600);
|
|
21
|
+
return key;
|
|
22
|
+
}
|
|
23
|
+
export function encryptSecret(plainText, key) {
|
|
24
|
+
assertValidKey(key);
|
|
25
|
+
const iv = randomBytes(IV_LENGTH);
|
|
26
|
+
const cipher = createCipheriv("aes-256-gcm", key, iv);
|
|
27
|
+
const encrypted = Buffer.concat([
|
|
28
|
+
cipher.update(plainText, "utf8"),
|
|
29
|
+
cipher.final(),
|
|
30
|
+
]);
|
|
31
|
+
const authTag = cipher.getAuthTag();
|
|
32
|
+
return [
|
|
33
|
+
FORMAT_VERSION,
|
|
34
|
+
iv.toString("base64url"),
|
|
35
|
+
authTag.toString("base64url"),
|
|
36
|
+
encrypted.toString("base64url"),
|
|
37
|
+
].join(":");
|
|
38
|
+
}
|
|
39
|
+
export function decryptSecret(encryptedSecret, key) {
|
|
40
|
+
assertValidKey(key);
|
|
41
|
+
const [version, iv, authTag, encrypted] = encryptedSecret.split(":");
|
|
42
|
+
if (version !== FORMAT_VERSION || !iv || !authTag || !encrypted) {
|
|
43
|
+
throw new Error("Invalid encrypted secret format");
|
|
44
|
+
}
|
|
45
|
+
const decipher = createDecipheriv("aes-256-gcm", key, Buffer.from(iv, "base64url"));
|
|
46
|
+
decipher.setAuthTag(Buffer.from(authTag, "base64url"));
|
|
47
|
+
return Buffer.concat([
|
|
48
|
+
decipher.update(Buffer.from(encrypted, "base64url")),
|
|
49
|
+
decipher.final(),
|
|
50
|
+
]).toString("utf8");
|
|
51
|
+
}
|
|
52
|
+
function assertValidKey(key) {
|
|
53
|
+
if (key.length !== KEY_LENGTH) {
|
|
54
|
+
throw new Error("Vault master key must be 32 bytes");
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
//# sourceMappingURL=vault.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vault.js","sourceRoot":"","sources":["../../src/storage/vault.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAA;AAC3E,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAA;AACpE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAA;AAEnC,MAAM,UAAU,GAAG,EAAE,CAAA;AACrB,MAAM,SAAS,GAAG,EAAE,CAAA;AACpB,MAAM,cAAc,GAAG,IAAI,CAAA;AAE3B,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,IAAY;IAChD,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;IAE/C,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;QACzD,OAAO,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,CAAA;IACxC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,CAAC,KAAK,YAAY,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC/E,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;IAED,MAAM,GAAG,GAAG,WAAW,CAAC,UAAU,CAAC,CAAA;IACnC,MAAM,SAAS,CAAC,IAAI,EAAE,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;IACvE,MAAM,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;IAExB,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,SAAiB,EAAE,GAAW;IAC1D,cAAc,CAAC,GAAG,CAAC,CAAA;IAEnB,MAAM,EAAE,GAAG,WAAW,CAAC,SAAS,CAAC,CAAA;IACjC,MAAM,MAAM,GAAG,cAAc,CAAC,aAAa,EAAE,GAAG,EAAE,EAAE,CAAC,CAAA;IACrD,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC;QAChC,MAAM,CAAC,KAAK,EAAE;KACf,CAAC,CAAA;IACF,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,EAAE,CAAA;IAEnC,OAAO;QACL,cAAc;QACd,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC7B,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC;KAChC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,eAAuB,EAAE,GAAW;IAChE,cAAc,CAAC,GAAG,CAAC,CAAA;IAEnB,MAAM,CAAC,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAEpE,IAAI,OAAO,KAAK,cAAc,IAAI,CAAC,EAAE,IAAI,CAAC,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAA;IACpD,CAAC;IAED,MAAM,QAAQ,GAAG,gBAAgB,CAC/B,aAAa,EACb,GAAG,EACH,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,WAAW,CAAC,CAC7B,CAAA;IACD,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,CAAA;IAEtD,OAAO,MAAM,CAAC,MAAM,CAAC;QACnB,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;QACpD,QAAQ,CAAC,KAAK,EAAE;KACjB,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;AACrB,CAAC;AAED,SAAS,cAAc,CAAC,GAAW;IACjC,IAAI,GAAG,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;QAC9B,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAA;IACtD,CAAC;AACH,CAAC"}
|
package/ui/README.md
ADDED
|
File without changes
|
|
Binary file
|
|
Binary file
|
|
Binary file
|
|
Binary file
|
|
Binary file
|