@marktoflow/core 2.0.0-alpha.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/bundle.d.ts +43 -0
- package/dist/bundle.d.ts.map +1 -0
- package/dist/bundle.js +202 -0
- package/dist/bundle.js.map +1 -0
- package/dist/config.d.ts +33 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +27 -0
- package/dist/config.js.map +1 -0
- package/dist/costs.d.ts +182 -0
- package/dist/costs.d.ts.map +1 -0
- package/dist/costs.js +464 -0
- package/dist/costs.js.map +1 -0
- package/dist/credentials.d.ts +162 -0
- package/dist/credentials.d.ts.map +1 -0
- package/dist/credentials.js +646 -0
- package/dist/credentials.js.map +1 -0
- package/dist/engine.d.ts +137 -0
- package/dist/engine.d.ts.map +1 -0
- package/dist/engine.js +514 -0
- package/dist/engine.js.map +1 -0
- package/dist/env.d.ts +59 -0
- package/dist/env.d.ts.map +1 -0
- package/dist/env.js +256 -0
- package/dist/env.js.map +1 -0
- package/dist/failover.d.ts +43 -0
- package/dist/failover.d.ts.map +1 -0
- package/dist/failover.js +53 -0
- package/dist/failover.js.map +1 -0
- package/dist/filewatcher.d.ts +32 -0
- package/dist/filewatcher.d.ts.map +1 -0
- package/dist/filewatcher.js +92 -0
- package/dist/filewatcher.js.map +1 -0
- package/dist/index.d.ts +36 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +69 -0
- package/dist/index.js.map +1 -0
- package/dist/logging.d.ts +62 -0
- package/dist/logging.d.ts.map +1 -0
- package/dist/logging.js +211 -0
- package/dist/logging.js.map +1 -0
- package/dist/mcp-loader.d.ts +29 -0
- package/dist/mcp-loader.d.ts.map +1 -0
- package/dist/mcp-loader.js +60 -0
- package/dist/mcp-loader.js.map +1 -0
- package/dist/metrics.d.ts +19 -0
- package/dist/metrics.d.ts.map +1 -0
- package/dist/metrics.js +65 -0
- package/dist/metrics.js.map +1 -0
- package/dist/models.d.ts +419 -0
- package/dist/models.d.ts.map +1 -0
- package/dist/models.js +111 -0
- package/dist/models.js.map +1 -0
- package/dist/parser.d.ts +40 -0
- package/dist/parser.d.ts.map +1 -0
- package/dist/parser.js +287 -0
- package/dist/parser.js.map +1 -0
- package/dist/plugins.d.ts +105 -0
- package/dist/plugins.d.ts.map +1 -0
- package/dist/plugins.js +182 -0
- package/dist/plugins.js.map +1 -0
- package/dist/queue.d.ts +114 -0
- package/dist/queue.d.ts.map +1 -0
- package/dist/queue.js +385 -0
- package/dist/queue.js.map +1 -0
- package/dist/rollback.d.ts +117 -0
- package/dist/rollback.d.ts.map +1 -0
- package/dist/rollback.js +374 -0
- package/dist/rollback.js.map +1 -0
- package/dist/routing.d.ts +144 -0
- package/dist/routing.d.ts.map +1 -0
- package/dist/routing.js +457 -0
- package/dist/routing.js.map +1 -0
- package/dist/scheduler.d.ts +91 -0
- package/dist/scheduler.d.ts.map +1 -0
- package/dist/scheduler.js +259 -0
- package/dist/scheduler.js.map +1 -0
- package/dist/script-tool.d.ts +22 -0
- package/dist/script-tool.d.ts.map +1 -0
- package/dist/script-tool.js +90 -0
- package/dist/script-tool.js.map +1 -0
- package/dist/sdk-registry.d.ts +81 -0
- package/dist/sdk-registry.d.ts.map +1 -0
- package/dist/sdk-registry.js +264 -0
- package/dist/sdk-registry.js.map +1 -0
- package/dist/security.d.ts +155 -0
- package/dist/security.d.ts.map +1 -0
- package/dist/security.js +362 -0
- package/dist/security.js.map +1 -0
- package/dist/state.d.ts +67 -0
- package/dist/state.d.ts.map +1 -0
- package/dist/state.js +276 -0
- package/dist/state.js.map +1 -0
- package/dist/templates.d.ts +70 -0
- package/dist/templates.d.ts.map +1 -0
- package/dist/templates.js +244 -0
- package/dist/templates.js.map +1 -0
- package/dist/tool-base.d.ts +54 -0
- package/dist/tool-base.d.ts.map +1 -0
- package/dist/tool-base.js +43 -0
- package/dist/tool-base.js.map +1 -0
- package/dist/tool-registry.d.ts +24 -0
- package/dist/tool-registry.d.ts.map +1 -0
- package/dist/tool-registry.js +164 -0
- package/dist/tool-registry.js.map +1 -0
- package/dist/tools/custom-tool.d.ts +16 -0
- package/dist/tools/custom-tool.d.ts.map +1 -0
- package/dist/tools/custom-tool.js +85 -0
- package/dist/tools/custom-tool.js.map +1 -0
- package/dist/tools/mcp-tool.d.ts +16 -0
- package/dist/tools/mcp-tool.d.ts.map +1 -0
- package/dist/tools/mcp-tool.js +98 -0
- package/dist/tools/mcp-tool.js.map +1 -0
- package/dist/tools/openapi-tool.d.ts +17 -0
- package/dist/tools/openapi-tool.d.ts.map +1 -0
- package/dist/tools/openapi-tool.js +165 -0
- package/dist/tools/openapi-tool.js.map +1 -0
- package/dist/trigger-manager.d.ts +26 -0
- package/dist/trigger-manager.d.ts.map +1 -0
- package/dist/trigger-manager.js +107 -0
- package/dist/trigger-manager.js.map +1 -0
- package/dist/webhook.d.ts +95 -0
- package/dist/webhook.d.ts.map +1 -0
- package/dist/webhook.js +261 -0
- package/dist/webhook.js.map +1 -0
- package/package.json +60 -0
package/dist/costs.js
ADDED
|
@@ -0,0 +1,464 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Cost tracking module for marktoflow.
|
|
3
|
+
*
|
|
4
|
+
* Provides token usage monitoring, API cost estimation, and cost limits.
|
|
5
|
+
*/
|
|
6
|
+
import Database from 'better-sqlite3';
|
|
7
|
+
import { existsSync, mkdirSync } from 'node:fs';
|
|
8
|
+
import { randomUUID } from 'node:crypto';
|
|
9
|
+
export var CostUnit;
|
|
10
|
+
(function (CostUnit) {
|
|
11
|
+
CostUnit["TOKENS"] = "tokens";
|
|
12
|
+
CostUnit["REQUESTS"] = "requests";
|
|
13
|
+
CostUnit["MINUTES"] = "minutes";
|
|
14
|
+
CostUnit["CREDITS"] = "credits";
|
|
15
|
+
})(CostUnit || (CostUnit = {}));
|
|
16
|
+
export var CostAlertLevel;
|
|
17
|
+
(function (CostAlertLevel) {
|
|
18
|
+
CostAlertLevel["INFO"] = "info";
|
|
19
|
+
CostAlertLevel["WARNING"] = "warning";
|
|
20
|
+
CostAlertLevel["CRITICAL"] = "critical";
|
|
21
|
+
})(CostAlertLevel || (CostAlertLevel = {}));
|
|
22
|
+
export const DEFAULT_PRICING = {
|
|
23
|
+
'gpt-4o': { modelName: 'gpt-4o', provider: 'openai', inputPricePerMillion: 2.5, outputPricePerMillion: 10.0, currency: 'USD' },
|
|
24
|
+
'gpt-4o-mini': { modelName: 'gpt-4o-mini', provider: 'openai', inputPricePerMillion: 0.15, outputPricePerMillion: 0.6, currency: 'USD' },
|
|
25
|
+
'gpt-4-turbo': { modelName: 'gpt-4-turbo', provider: 'openai', inputPricePerMillion: 10.0, outputPricePerMillion: 30.0, currency: 'USD' },
|
|
26
|
+
'gpt-3.5-turbo': { modelName: 'gpt-3.5-turbo', provider: 'openai', inputPricePerMillion: 0.5, outputPricePerMillion: 1.5, currency: 'USD' },
|
|
27
|
+
'claude-3-5-sonnet': { modelName: 'claude-3-5-sonnet', provider: 'anthropic', inputPricePerMillion: 3.0, outputPricePerMillion: 15.0, currency: 'USD' },
|
|
28
|
+
'claude-3-opus': { modelName: 'claude-3-opus', provider: 'anthropic', inputPricePerMillion: 15.0, outputPricePerMillion: 75.0, currency: 'USD' },
|
|
29
|
+
'claude-3-sonnet': { modelName: 'claude-3-sonnet', provider: 'anthropic', inputPricePerMillion: 3.0, outputPricePerMillion: 15.0, currency: 'USD' },
|
|
30
|
+
'claude-3-haiku': { modelName: 'claude-3-haiku', provider: 'anthropic', inputPricePerMillion: 0.25, outputPricePerMillion: 1.25, currency: 'USD' },
|
|
31
|
+
'gemini-1.5-pro': { modelName: 'gemini-1.5-pro', provider: 'google', inputPricePerMillion: 3.5, outputPricePerMillion: 10.5, currency: 'USD' },
|
|
32
|
+
'gemini-1.5-flash': { modelName: 'gemini-1.5-flash', provider: 'google', inputPricePerMillion: 0.075, outputPricePerMillion: 0.3, currency: 'USD' },
|
|
33
|
+
};
|
|
34
|
+
export class LoggingAlertHandler {
|
|
35
|
+
logFunc;
|
|
36
|
+
constructor(logFunc = console.log) {
|
|
37
|
+
this.logFunc = logFunc;
|
|
38
|
+
}
|
|
39
|
+
handleAlert(alert) {
|
|
40
|
+
this.logFunc(`[${alert.level.toUpperCase()}] ${alert.message}`);
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
export class CallbackAlertHandler {
|
|
44
|
+
callback;
|
|
45
|
+
constructor(callback) {
|
|
46
|
+
this.callback = callback;
|
|
47
|
+
}
|
|
48
|
+
handleAlert(alert) {
|
|
49
|
+
this.callback(alert);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
export class PricingRegistry {
|
|
53
|
+
pricing = new Map();
|
|
54
|
+
constructor() {
|
|
55
|
+
for (const p of Object.values(DEFAULT_PRICING)) {
|
|
56
|
+
this.pricing.set(p.modelName, p);
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
register(pricing) {
|
|
60
|
+
this.pricing.set(pricing.modelName, pricing);
|
|
61
|
+
}
|
|
62
|
+
get(modelName) {
|
|
63
|
+
if (this.pricing.has(modelName))
|
|
64
|
+
return this.pricing.get(modelName);
|
|
65
|
+
for (const [key, value] of this.pricing.entries()) {
|
|
66
|
+
if (modelName.startsWith(key))
|
|
67
|
+
return value;
|
|
68
|
+
}
|
|
69
|
+
return undefined;
|
|
70
|
+
}
|
|
71
|
+
listModels() {
|
|
72
|
+
return Array.from(this.pricing.keys());
|
|
73
|
+
}
|
|
74
|
+
calculateCost(modelName, inputTokens, outputTokens) {
|
|
75
|
+
const pricing = this.get(modelName);
|
|
76
|
+
if (!pricing)
|
|
77
|
+
return null;
|
|
78
|
+
const inputCost = (inputTokens / 1_000_000) * pricing.inputPricePerMillion;
|
|
79
|
+
const outputCost = (outputTokens / 1_000_000) * pricing.outputPricePerMillion;
|
|
80
|
+
return inputCost + outputCost;
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
export class CostTracker {
|
|
84
|
+
pricingRegistry;
|
|
85
|
+
alertHandlers;
|
|
86
|
+
records = [];
|
|
87
|
+
limits = [];
|
|
88
|
+
constructor(pricingRegistry = new PricingRegistry(), alertHandlers = []) {
|
|
89
|
+
this.pricingRegistry = pricingRegistry;
|
|
90
|
+
this.alertHandlers = alertHandlers;
|
|
91
|
+
}
|
|
92
|
+
addAlertHandler(handler) {
|
|
93
|
+
this.alertHandlers.push(handler);
|
|
94
|
+
}
|
|
95
|
+
addLimit(limit) {
|
|
96
|
+
this.limits.push(limit);
|
|
97
|
+
}
|
|
98
|
+
removeLimit(name) {
|
|
99
|
+
const index = this.limits.findIndex((limit) => limit.name === name);
|
|
100
|
+
if (index === -1)
|
|
101
|
+
return false;
|
|
102
|
+
this.limits.splice(index, 1);
|
|
103
|
+
return true;
|
|
104
|
+
}
|
|
105
|
+
recordUsage(params) {
|
|
106
|
+
const cost = this.pricingRegistry.calculateCost(params.modelName, params.tokenUsage.inputTokens, params.tokenUsage.outputTokens) ?? 0;
|
|
107
|
+
const record = {
|
|
108
|
+
id: params.recordId ?? randomUUID(),
|
|
109
|
+
timestamp: new Date(),
|
|
110
|
+
workflowId: params.workflowId,
|
|
111
|
+
runId: params.runId,
|
|
112
|
+
stepName: params.stepName,
|
|
113
|
+
agentName: params.agentName,
|
|
114
|
+
modelName: params.modelName,
|
|
115
|
+
tokenUsage: {
|
|
116
|
+
inputTokens: params.tokenUsage.inputTokens,
|
|
117
|
+
outputTokens: params.tokenUsage.outputTokens,
|
|
118
|
+
cachedTokens: params.tokenUsage.cachedTokens ?? 0,
|
|
119
|
+
reasoningTokens: params.tokenUsage.reasoningTokens ?? 0,
|
|
120
|
+
},
|
|
121
|
+
estimatedCost: cost,
|
|
122
|
+
currency: 'USD',
|
|
123
|
+
metadata: params.metadata ?? {},
|
|
124
|
+
};
|
|
125
|
+
this.records.push(record);
|
|
126
|
+
this.checkLimits(record);
|
|
127
|
+
return record;
|
|
128
|
+
}
|
|
129
|
+
checkLimits(_record) {
|
|
130
|
+
for (const limit of this.limits) {
|
|
131
|
+
const currentCost = this.calculateLimitUsage(limit);
|
|
132
|
+
const percentage = limit.maxCost > 0 ? (currentCost / limit.maxCost) * 100 : 0;
|
|
133
|
+
if (limit.alertThreshold !== undefined && percentage >= limit.alertThreshold) {
|
|
134
|
+
const level = percentage >= 100 ? CostAlertLevel.CRITICAL : CostAlertLevel.WARNING;
|
|
135
|
+
const alert = {
|
|
136
|
+
timestamp: new Date(),
|
|
137
|
+
level,
|
|
138
|
+
limitName: limit.name,
|
|
139
|
+
currentCost,
|
|
140
|
+
limitCost: limit.maxCost,
|
|
141
|
+
percentage,
|
|
142
|
+
message: `Cost limit '${limit.name}' at ${percentage.toFixed(1)}%: $${currentCost.toFixed(4)} / $${limit.maxCost.toFixed(4)}`,
|
|
143
|
+
};
|
|
144
|
+
for (const handler of this.alertHandlers) {
|
|
145
|
+
try {
|
|
146
|
+
handler.handleAlert(alert);
|
|
147
|
+
}
|
|
148
|
+
catch {
|
|
149
|
+
// ignore handler errors
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
calculateLimitUsage(limit) {
|
|
156
|
+
const now = Date.now();
|
|
157
|
+
const startTime = limit.periodMs ? now - limit.periodMs : 0;
|
|
158
|
+
let total = 0;
|
|
159
|
+
for (const record of this.records) {
|
|
160
|
+
if (record.timestamp.getTime() < startTime)
|
|
161
|
+
continue;
|
|
162
|
+
if (limit.scope === 'workflow' && limit.scopeId && record.workflowId !== limit.scopeId)
|
|
163
|
+
continue;
|
|
164
|
+
if (limit.scope === 'agent' && limit.scopeId && record.agentName !== limit.scopeId)
|
|
165
|
+
continue;
|
|
166
|
+
if (limit.scope === 'model' && limit.scopeId && record.modelName !== limit.scopeId)
|
|
167
|
+
continue;
|
|
168
|
+
total += record.estimatedCost;
|
|
169
|
+
}
|
|
170
|
+
return total;
|
|
171
|
+
}
|
|
172
|
+
getRecords(filters = {}) {
|
|
173
|
+
return this.records.filter((record) => {
|
|
174
|
+
if (filters.workflowId && record.workflowId !== filters.workflowId)
|
|
175
|
+
return false;
|
|
176
|
+
if (filters.runId && record.runId !== filters.runId)
|
|
177
|
+
return false;
|
|
178
|
+
if (filters.agentName && record.agentName !== filters.agentName)
|
|
179
|
+
return false;
|
|
180
|
+
if (filters.startTime && record.timestamp < filters.startTime)
|
|
181
|
+
return false;
|
|
182
|
+
if (filters.endTime && record.timestamp > filters.endTime)
|
|
183
|
+
return false;
|
|
184
|
+
return true;
|
|
185
|
+
});
|
|
186
|
+
}
|
|
187
|
+
getSummary(startTime, endTime) {
|
|
188
|
+
const start = startTime ?? new Date(0);
|
|
189
|
+
const end = endTime ?? new Date();
|
|
190
|
+
const records = this.getRecords({ startTime: start, endTime: end });
|
|
191
|
+
let totalCost = 0;
|
|
192
|
+
let totalInput = 0;
|
|
193
|
+
let totalOutput = 0;
|
|
194
|
+
const byWorkflow = {};
|
|
195
|
+
const byAgent = {};
|
|
196
|
+
const byModel = {};
|
|
197
|
+
for (const record of records) {
|
|
198
|
+
totalCost += record.estimatedCost;
|
|
199
|
+
totalInput += record.tokenUsage.inputTokens;
|
|
200
|
+
totalOutput += record.tokenUsage.outputTokens;
|
|
201
|
+
byWorkflow[record.workflowId] = (byWorkflow[record.workflowId] ?? 0) + record.estimatedCost;
|
|
202
|
+
byAgent[record.agentName] = (byAgent[record.agentName] ?? 0) + record.estimatedCost;
|
|
203
|
+
byModel[record.modelName] = (byModel[record.modelName] ?? 0) + record.estimatedCost;
|
|
204
|
+
}
|
|
205
|
+
return {
|
|
206
|
+
startTime: start,
|
|
207
|
+
endTime: end,
|
|
208
|
+
totalCost,
|
|
209
|
+
totalInputTokens: totalInput,
|
|
210
|
+
totalOutputTokens: totalOutput,
|
|
211
|
+
totalRequests: records.length,
|
|
212
|
+
byWorkflow,
|
|
213
|
+
byAgent,
|
|
214
|
+
byModel,
|
|
215
|
+
currency: 'USD',
|
|
216
|
+
};
|
|
217
|
+
}
|
|
218
|
+
getWorkflowCost(workflowId, runId) {
|
|
219
|
+
const records = this.getRecords({ workflowId, runId });
|
|
220
|
+
return records.reduce((sum, record) => sum + record.estimatedCost, 0);
|
|
221
|
+
}
|
|
222
|
+
clear() {
|
|
223
|
+
this.records = [];
|
|
224
|
+
}
|
|
225
|
+
}
|
|
226
|
+
export class CostStore {
|
|
227
|
+
db;
|
|
228
|
+
pricing;
|
|
229
|
+
constructor(dbPath = '.marktoflow/state/costs.db') {
|
|
230
|
+
const dir = dbPath.substring(0, dbPath.lastIndexOf('/'));
|
|
231
|
+
if (dir && !existsSync(dir)) {
|
|
232
|
+
mkdirSync(dir, { recursive: true });
|
|
233
|
+
}
|
|
234
|
+
this.db = new Database(dbPath);
|
|
235
|
+
this.init();
|
|
236
|
+
this.pricing = new PricingRegistry();
|
|
237
|
+
}
|
|
238
|
+
getPricingRegistry() {
|
|
239
|
+
return this.pricing;
|
|
240
|
+
}
|
|
241
|
+
calculateCost(modelName, inputTokens, outputTokens) {
|
|
242
|
+
return this.pricing.calculateCost(modelName, inputTokens, outputTokens) ?? 0;
|
|
243
|
+
}
|
|
244
|
+
record(params) {
|
|
245
|
+
const cost = this.calculateCost(params.modelName, params.tokenUsage.inputTokens, params.tokenUsage.outputTokens);
|
|
246
|
+
const record = {
|
|
247
|
+
id: randomUUID(),
|
|
248
|
+
timestamp: new Date(),
|
|
249
|
+
workflowId: params.workflowId,
|
|
250
|
+
runId: params.runId,
|
|
251
|
+
stepName: params.stepName,
|
|
252
|
+
agentName: params.agentName,
|
|
253
|
+
modelName: params.modelName,
|
|
254
|
+
tokenUsage: {
|
|
255
|
+
inputTokens: params.tokenUsage.inputTokens,
|
|
256
|
+
outputTokens: params.tokenUsage.outputTokens,
|
|
257
|
+
cachedTokens: params.tokenUsage.cachedTokens ?? 0,
|
|
258
|
+
reasoningTokens: params.tokenUsage.reasoningTokens ?? 0,
|
|
259
|
+
},
|
|
260
|
+
estimatedCost: cost,
|
|
261
|
+
currency: 'USD',
|
|
262
|
+
metadata: params.metadata ?? {},
|
|
263
|
+
};
|
|
264
|
+
this.save(record);
|
|
265
|
+
return record;
|
|
266
|
+
}
|
|
267
|
+
init() {
|
|
268
|
+
this.db.exec(`
|
|
269
|
+
CREATE TABLE IF NOT EXISTS cost_records (
|
|
270
|
+
id TEXT PRIMARY KEY,
|
|
271
|
+
timestamp TEXT NOT NULL,
|
|
272
|
+
workflow_id TEXT NOT NULL,
|
|
273
|
+
run_id TEXT NOT NULL,
|
|
274
|
+
step_name TEXT,
|
|
275
|
+
agent_name TEXT NOT NULL,
|
|
276
|
+
model_name TEXT NOT NULL,
|
|
277
|
+
input_tokens INTEGER NOT NULL,
|
|
278
|
+
output_tokens INTEGER NOT NULL,
|
|
279
|
+
cached_tokens INTEGER DEFAULT 0,
|
|
280
|
+
reasoning_tokens INTEGER DEFAULT 0,
|
|
281
|
+
estimated_cost REAL NOT NULL,
|
|
282
|
+
currency TEXT DEFAULT 'USD',
|
|
283
|
+
metadata TEXT
|
|
284
|
+
);
|
|
285
|
+
CREATE INDEX IF NOT EXISTS idx_cost_workflow_id ON cost_records(workflow_id);
|
|
286
|
+
CREATE INDEX IF NOT EXISTS idx_cost_timestamp ON cost_records(timestamp);
|
|
287
|
+
CREATE INDEX IF NOT EXISTS idx_cost_agent ON cost_records(agent_name);
|
|
288
|
+
`);
|
|
289
|
+
}
|
|
290
|
+
save(record) {
|
|
291
|
+
this.db.prepare(`
|
|
292
|
+
INSERT OR REPLACE INTO cost_records
|
|
293
|
+
(id, timestamp, workflow_id, run_id, step_name, agent_name, model_name,
|
|
294
|
+
input_tokens, output_tokens, cached_tokens, reasoning_tokens,
|
|
295
|
+
estimated_cost, currency, metadata)
|
|
296
|
+
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
297
|
+
`).run(record.id, record.timestamp.toISOString(), record.workflowId, record.runId, record.stepName ?? null, record.agentName, record.modelName, record.tokenUsage.inputTokens, record.tokenUsage.outputTokens, record.tokenUsage.cachedTokens ?? 0, record.tokenUsage.reasoningTokens ?? 0, record.estimatedCost, record.currency, record.metadata ? JSON.stringify(record.metadata) : null);
|
|
298
|
+
}
|
|
299
|
+
get(recordId) {
|
|
300
|
+
const row = this.db.prepare('SELECT * FROM cost_records WHERE id = ?').get(recordId);
|
|
301
|
+
return row ? this.rowToRecord(row) : null;
|
|
302
|
+
}
|
|
303
|
+
query(filters = {}) {
|
|
304
|
+
const conditions = [];
|
|
305
|
+
const params = [];
|
|
306
|
+
if (filters.workflowId) {
|
|
307
|
+
conditions.push('workflow_id = ?');
|
|
308
|
+
params.push(filters.workflowId);
|
|
309
|
+
}
|
|
310
|
+
if (filters.runId) {
|
|
311
|
+
conditions.push('run_id = ?');
|
|
312
|
+
params.push(filters.runId);
|
|
313
|
+
}
|
|
314
|
+
if (filters.agentName) {
|
|
315
|
+
conditions.push('agent_name = ?');
|
|
316
|
+
params.push(filters.agentName);
|
|
317
|
+
}
|
|
318
|
+
if (filters.modelName) {
|
|
319
|
+
conditions.push('model_name = ?');
|
|
320
|
+
params.push(filters.modelName);
|
|
321
|
+
}
|
|
322
|
+
if (filters.startTime) {
|
|
323
|
+
conditions.push('timestamp >= ?');
|
|
324
|
+
params.push(filters.startTime.toISOString());
|
|
325
|
+
}
|
|
326
|
+
if (filters.endTime) {
|
|
327
|
+
conditions.push('timestamp <= ?');
|
|
328
|
+
params.push(filters.endTime.toISOString());
|
|
329
|
+
}
|
|
330
|
+
let sql = 'SELECT * FROM cost_records';
|
|
331
|
+
if (conditions.length > 0) {
|
|
332
|
+
sql += ' WHERE ' + conditions.join(' AND ');
|
|
333
|
+
}
|
|
334
|
+
sql += ' ORDER BY timestamp DESC';
|
|
335
|
+
if (filters.limit) {
|
|
336
|
+
sql += ` LIMIT ${filters.limit}`;
|
|
337
|
+
}
|
|
338
|
+
const rows = this.db.prepare(sql).all(...params);
|
|
339
|
+
return rows.map((row) => this.rowToRecord(row));
|
|
340
|
+
}
|
|
341
|
+
getSummary(startTime, endTime) {
|
|
342
|
+
const conditions = [];
|
|
343
|
+
const params = [];
|
|
344
|
+
if (startTime) {
|
|
345
|
+
conditions.push('timestamp >= ?');
|
|
346
|
+
params.push(startTime.toISOString());
|
|
347
|
+
}
|
|
348
|
+
if (endTime) {
|
|
349
|
+
conditions.push('timestamp <= ?');
|
|
350
|
+
params.push(endTime.toISOString());
|
|
351
|
+
}
|
|
352
|
+
const whereClause = conditions.length ? ` WHERE ${conditions.join(' AND ')}` : '';
|
|
353
|
+
const totals = this.db.prepare(`
|
|
354
|
+
SELECT
|
|
355
|
+
COALESCE(SUM(estimated_cost), 0) as total_cost,
|
|
356
|
+
COALESCE(SUM(input_tokens), 0) as total_input,
|
|
357
|
+
COALESCE(SUM(output_tokens), 0) as total_output,
|
|
358
|
+
COUNT(*) as total_requests,
|
|
359
|
+
MIN(timestamp) as min_time,
|
|
360
|
+
MAX(timestamp) as max_time
|
|
361
|
+
FROM cost_records
|
|
362
|
+
${whereClause}
|
|
363
|
+
`).get(...params);
|
|
364
|
+
const byWorkflowRows = this.db.prepare(`
|
|
365
|
+
SELECT workflow_id, SUM(estimated_cost) as cost
|
|
366
|
+
FROM cost_records
|
|
367
|
+
${whereClause}
|
|
368
|
+
GROUP BY workflow_id
|
|
369
|
+
`).all(...params);
|
|
370
|
+
const byAgentRows = this.db.prepare(`
|
|
371
|
+
SELECT agent_name, SUM(estimated_cost) as cost
|
|
372
|
+
FROM cost_records
|
|
373
|
+
${whereClause}
|
|
374
|
+
GROUP BY agent_name
|
|
375
|
+
`).all(...params);
|
|
376
|
+
const byModelRows = this.db.prepare(`
|
|
377
|
+
SELECT model_name, SUM(estimated_cost) as cost
|
|
378
|
+
FROM cost_records
|
|
379
|
+
${whereClause}
|
|
380
|
+
GROUP BY model_name
|
|
381
|
+
`).all(...params);
|
|
382
|
+
const byWorkflow = {};
|
|
383
|
+
const byAgent = {};
|
|
384
|
+
const byModel = {};
|
|
385
|
+
for (const row of byWorkflowRows)
|
|
386
|
+
byWorkflow[row.workflow_id] = row.cost;
|
|
387
|
+
for (const row of byAgentRows)
|
|
388
|
+
byAgent[row.agent_name] = row.cost;
|
|
389
|
+
for (const row of byModelRows)
|
|
390
|
+
byModel[row.model_name] = row.cost;
|
|
391
|
+
return {
|
|
392
|
+
startTime: totals.min_time ? new Date(totals.min_time) : (startTime ?? new Date(0)),
|
|
393
|
+
endTime: totals.max_time ? new Date(totals.max_time) : (endTime ?? new Date()),
|
|
394
|
+
totalCost: totals.total_cost,
|
|
395
|
+
totalInputTokens: totals.total_input,
|
|
396
|
+
totalOutputTokens: totals.total_output,
|
|
397
|
+
totalRequests: totals.total_requests,
|
|
398
|
+
byWorkflow,
|
|
399
|
+
byAgent,
|
|
400
|
+
byModel,
|
|
401
|
+
currency: 'USD',
|
|
402
|
+
};
|
|
403
|
+
}
|
|
404
|
+
deleteBefore(before) {
|
|
405
|
+
const res = this.db.prepare('DELETE FROM cost_records WHERE timestamp < ?').run(before.toISOString());
|
|
406
|
+
return res.changes ?? 0;
|
|
407
|
+
}
|
|
408
|
+
rowToRecord(row) {
|
|
409
|
+
return {
|
|
410
|
+
id: row.id,
|
|
411
|
+
timestamp: new Date(row.timestamp),
|
|
412
|
+
workflowId: row.workflow_id,
|
|
413
|
+
runId: row.run_id,
|
|
414
|
+
stepName: row.step_name ?? undefined,
|
|
415
|
+
agentName: row.agent_name,
|
|
416
|
+
modelName: row.model_name,
|
|
417
|
+
tokenUsage: {
|
|
418
|
+
inputTokens: row.input_tokens,
|
|
419
|
+
outputTokens: row.output_tokens,
|
|
420
|
+
cachedTokens: row.cached_tokens ?? 0,
|
|
421
|
+
reasoningTokens: row.reasoning_tokens ?? 0,
|
|
422
|
+
},
|
|
423
|
+
estimatedCost: row.estimated_cost,
|
|
424
|
+
currency: row.currency ?? 'USD',
|
|
425
|
+
metadata: row.metadata ? JSON.parse(row.metadata) : {},
|
|
426
|
+
};
|
|
427
|
+
}
|
|
428
|
+
}
|
|
429
|
+
export class PersistentCostTracker extends CostTracker {
|
|
430
|
+
store;
|
|
431
|
+
constructor(store, pricingRegistry, alertHandlers) {
|
|
432
|
+
super(pricingRegistry ?? new PricingRegistry(), alertHandlers ?? []);
|
|
433
|
+
this.store = store;
|
|
434
|
+
}
|
|
435
|
+
recordUsage(params) {
|
|
436
|
+
const record = super.recordUsage(params);
|
|
437
|
+
this.store.save(record);
|
|
438
|
+
return record;
|
|
439
|
+
}
|
|
440
|
+
getSummary(startTime, endTime) {
|
|
441
|
+
return this.store.getSummary(startTime, endTime);
|
|
442
|
+
}
|
|
443
|
+
}
|
|
444
|
+
export class WorkflowCostEstimator {
|
|
445
|
+
pricingRegistry;
|
|
446
|
+
constructor(pricingRegistry = new PricingRegistry()) {
|
|
447
|
+
this.pricingRegistry = pricingRegistry;
|
|
448
|
+
}
|
|
449
|
+
estimateStepCost(modelName, inputTokens, outputTokens) {
|
|
450
|
+
return this.pricingRegistry.calculateCost(modelName, inputTokens, outputTokens);
|
|
451
|
+
}
|
|
452
|
+
estimateWorkflowCost(modelName, stepCount, avgInputTokensPerStep = 1000, avgOutputTokensPerStep = 500) {
|
|
453
|
+
const stepCost = this.estimateStepCost(modelName, avgInputTokensPerStep, avgOutputTokensPerStep);
|
|
454
|
+
return stepCost !== null ? stepCost * stepCount : null;
|
|
455
|
+
}
|
|
456
|
+
compareModels(models, inputTokens, outputTokens) {
|
|
457
|
+
const result = {};
|
|
458
|
+
for (const model of models) {
|
|
459
|
+
result[model] = this.pricingRegistry.calculateCost(model, inputTokens, outputTokens);
|
|
460
|
+
}
|
|
461
|
+
return result;
|
|
462
|
+
}
|
|
463
|
+
}
|
|
464
|
+
//# sourceMappingURL=costs.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"costs.js","sourceRoot":"","sources":["../src/costs.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAChD,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,CAAN,IAAY,QAKX;AALD,WAAY,QAAQ;IAClB,6BAAiB,CAAA;IACjB,iCAAqB,CAAA;IACrB,+BAAmB,CAAA;IACnB,+BAAmB,CAAA;AACrB,CAAC,EALW,QAAQ,KAAR,QAAQ,QAKnB;AAED,MAAM,CAAN,IAAY,cAIX;AAJD,WAAY,cAAc;IACxB,+BAAa,CAAA;IACb,qCAAmB,CAAA;IACnB,uCAAqB,CAAA;AACvB,CAAC,EAJW,cAAc,KAAd,cAAc,QAIzB;AAYD,MAAM,CAAC,MAAM,eAAe,GAAiC;IAC3D,QAAQ,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IAC9H,aAAa,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,QAAQ,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE,qBAAqB,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE;IACxI,aAAa,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,QAAQ,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE,qBAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IACzI,eAAe,EAAE,EAAE,SAAS,EAAE,eAAe,EAAE,QAAQ,EAAE,QAAQ,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE;IAC3I,mBAAmB,EAAE,EAAE,SAAS,EAAE,mBAAmB,EAAE,QAAQ,EAAE,WAAW,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IACvJ,eAAe,EAAE,EAAE,SAAS,EAAE,eAAe,EAAE,QAAQ,EAAE,WAAW,EAAE,oBAAoB,EAAE,IAAI,EAAE,qBAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IAChJ,iBAAiB,EAAE,EAAE,SAAS,EAAE,iBAAiB,EAAE,QAAQ,EAAE,WAAW,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IACnJ,gBAAgB,EAAE,EAAE,SAAS,EAAE,gBAAgB,EAAE,QAAQ,EAAE,WAAW,EAAE,oBAAoB,EAAE,IAAI,EAAE,qBAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IAClJ,gBAAgB,EAAE,EAAE,SAAS,EAAE,gBAAgB,EAAE,QAAQ,EAAE,QAAQ,EAAE,oBAAoB,EAAE,GAAG,EAAE,qBAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IAC9I,kBAAkB,EAAE,EAAE,SAAS,EAAE,kBAAkB,EAAE,QAAQ,EAAE,QAAQ,EAAE,oBAAoB,EAAE,KAAK,EAAE,qBAAqB,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE;CACpJ,CAAC;AA4DF,MAAM,OAAO,mBAAmB;IACV;IAApB,YAAoB,UAAqC,OAAO,CAAC,GAAG;QAAhD,YAAO,GAAP,OAAO,CAAyC;IAAG,CAAC;IAExE,WAAW,CAAC,KAAgB;QAC1B,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IAClE,CAAC;CACF;AAED,MAAM,OAAO,oBAAoB;IACX;IAApB,YAAoB,QAAoC;QAApC,aAAQ,GAAR,QAAQ,CAA4B;IAAG,CAAC;IAE5D,WAAW,CAAC,KAAgB;QAC1B,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,eAAe;IAClB,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;IAElD;QACE,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE,CAAC;YAC/C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,OAAqB;QAC5B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED,GAAG,CAAC,SAAiB;QACnB,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YAClD,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,OAAO,KAAK,CAAC;QAC9C,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,UAAU;QACR,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACzC,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,WAAmB,EAAE,YAAoB;QACxE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpC,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAC1B,MAAM,SAAS,GAAG,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,oBAAoB,CAAC;QAC3E,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,qBAAqB,CAAC;QAC9E,OAAO,SAAS,GAAG,UAAU,CAAC;IAChC,CAAC;CACF;AAED,MAAM,OAAO,WAAW;IAKb;IACA;IALD,OAAO,GAAiB,EAAE,CAAC;IAC3B,MAAM,GAAgB,EAAE,CAAC;IAEjC,YACS,kBAAmC,IAAI,eAAe,EAAE,EACxD,gBAAoC,EAAE;QADtC,oBAAe,GAAf,eAAe,CAAyC;QACxD,kBAAa,GAAb,aAAa,CAAyB;IAC5C,CAAC;IAEJ,eAAe,CAAC,OAAyB;QACvC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CAAC,KAAgB;QACvB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,WAAW,CAAC,IAAY;QACtB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QACpE,IAAI,KAAK,KAAK,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,WAAW,CAAC,MASX;QACC,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,CAC7C,MAAM,CAAC,SAAS,EAChB,MAAM,CAAC,UAAU,CAAC,WAAW,EAC7B,MAAM,CAAC,UAAU,CAAC,YAAY,CAC/B,IAAI,CAAC,CAAC;QAEP,MAAM,MAAM,GAAe;YACzB,EAAE,EAAE,MAAM,CAAC,QAAQ,IAAI,UAAU,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,UAAU,EAAE;gBACV,WAAW,EAAE,MAAM,CAAC,UAAU,CAAC,WAAW;gBAC1C,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY;gBAC5C,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC;gBACjD,eAAe,EAAE,MAAM,CAAC,UAAU,CAAC,eAAe,IAAI,CAAC;aACxD;YACD,aAAa,EAAE,IAAI;YACnB,QAAQ,EAAE,KAAK;YACf,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE;SAChC,CAAC;QAEF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACzB,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,WAAW,CAAC,OAAmB;QACrC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,MAAM,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YACpD,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAE/E,IAAI,KAAK,CAAC,cAAc,KAAK,SAAS,IAAI,UAAU,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;gBAC7E,MAAM,KAAK,GAAG,UAAU,IAAI,GAAG,CAAC,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC;gBACnF,MAAM,KAAK,GAAc;oBACvB,SAAS,EAAE,IAAI,IAAI,EAAE;oBACrB,KAAK;oBACL,SAAS,EAAE,KAAK,CAAC,IAAI;oBACrB,WAAW;oBACX,SAAS,EAAE,KAAK,CAAC,OAAO;oBACxB,UAAU;oBACV,OAAO,EAAE,eAAe,KAAK,CAAC,IAAI,QAAQ,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;iBAC9H,CAAC;gBACF,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACzC,IAAI,CAAC;wBACH,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;oBAC7B,CAAC;oBAAC,MAAM,CAAC;wBACP,wBAAwB;oBAC1B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAEO,mBAAmB,CAAC,KAAgB;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5D,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,SAAS;gBAAE,SAAS;YACrD,IAAI,KAAK,CAAC,KAAK,KAAK,UAAU,IAAI,KAAK,CAAC,OAAO,IAAI,MAAM,CAAC,UAAU,KAAK,KAAK,CAAC,OAAO;gBAAE,SAAS;YACjG,IAAI,KAAK,CAAC,KAAK,KAAK,OAAO,IAAI,KAAK,CAAC,OAAO,IAAI,MAAM,CAAC,SAAS,KAAK,KAAK,CAAC,OAAO;gBAAE,SAAS;YAC7F,IAAI,KAAK,CAAC,KAAK,KAAK,OAAO,IAAI,KAAK,CAAC,OAAO,IAAI,MAAM,CAAC,SAAS,KAAK,KAAK,CAAC,OAAO;gBAAE,SAAS;YAC7F,KAAK,IAAI,MAAM,CAAC,aAAa,CAAC;QAChC,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,UAAU,CAAC,UAMP,EAAE;QACJ,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;YACpC,IAAI,OAAO,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,KAAK,OAAO,CAAC,UAAU;gBAAE,OAAO,KAAK,CAAC;YACjF,IAAI,OAAO,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK;gBAAE,OAAO,KAAK,CAAC;YAClE,IAAI,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,CAAC,SAAS;gBAAE,OAAO,KAAK,CAAC;YAC9E,IAAI,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS;gBAAE,OAAO,KAAK,CAAC;YAC5E,IAAI,OAAO,CAAC,OAAO,IAAI,MAAM,CAAC,SAAS,GAAG,OAAO,CAAC,OAAO;gBAAE,OAAO,KAAK,CAAC;YACxE,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,SAAgB,EAAE,OAAc;QACzC,MAAM,KAAK,GAAG,SAAS,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,GAAG,GAAG,OAAO,IAAI,IAAI,IAAI,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAEpE,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,MAAM,UAAU,GAA2B,EAAE,CAAC;QAC9C,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,SAAS,IAAI,MAAM,CAAC,aAAa,CAAC;YAClC,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC;YAC5C,WAAW,IAAI,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC;YAC9C,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC;YAC5F,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC;YACpF,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC;QACtF,CAAC;QAED,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,GAAG;YACZ,SAAS;YACT,gBAAgB,EAAE,UAAU;YAC5B,iBAAiB,EAAE,WAAW;YAC9B,aAAa,EAAE,OAAO,CAAC,MAAM;YAC7B,UAAU;YACV,OAAO;YACP,OAAO;YACP,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,UAAkB,EAAE,KAAc;QAChD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QACvD,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACxE,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;CACF;AAED,MAAM,OAAO,SAAS;IACZ,EAAE,CAAoB;IACtB,OAAO,CAAkB;IAEjC,YAAY,SAAiB,4BAA4B;QACvD,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACzD,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC5B,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;IACvC,CAAC;IAED,kBAAkB;QAChB,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,WAAmB,EAAE,YAAoB;QACxE,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,SAAS,EAAE,WAAW,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;IAC/E,CAAC;IAED,MAAM,CAAC,MAQN;QACC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QACjH,MAAM,MAAM,GAAe;YACzB,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,UAAU,EAAE;gBACV,WAAW,EAAE,MAAM,CAAC,UAAU,CAAC,WAAW;gBAC1C,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY;gBAC5C,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC;gBACjD,eAAe,EAAE,MAAM,CAAC,UAAU,CAAC,eAAe,IAAI,CAAC;aACxD;YACD,aAAa,EAAE,IAAI;YACnB,QAAQ,EAAE,KAAK;YACf,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE;SAChC,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClB,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,IAAI;QACV,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;;;;KAoBZ,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC,MAAkB;QACrB,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;KAMf,CAAC,CAAC,GAAG,CACJ,MAAM,CAAC,EAAE,EACT,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE,EAC9B,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,QAAQ,IAAI,IAAI,EACvB,MAAM,CAAC,SAAS,EAChB,MAAM,CAAC,SAAS,EAChB,MAAM,CAAC,UAAU,CAAC,WAAW,EAC7B,MAAM,CAAC,UAAU,CAAC,YAAY,EAC9B,MAAM,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,EACnC,MAAM,CAAC,UAAU,CAAC,eAAe,IAAI,CAAC,EACtC,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,QAAQ,EACf,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CACzD,CAAC;IACJ,CAAC;IAED,GAAG,CAAC,QAAgB;QAClB,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAQ,CAAC;QAC5F,OAAO,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,UAQF,EAAE;QACJ,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAU,EAAE,CAAC;QAEzB,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAAC,CAAC;QAChG,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAAC,CAAC;QACjF,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAAC,CAAC;QAC7F,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAAC,CAAC;QAC7F,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC;QAAC,CAAC;QAC3G,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;QAAC,CAAC;QAEvG,IAAI,GAAG,GAAG,4BAA4B,CAAC;QACvC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,GAAG,IAAI,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;QACD,GAAG,IAAI,0BAA0B,CAAC;QAClC,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,GAAG,IAAI,UAAU,OAAO,CAAC,KAAK,EAAE,CAAC;QACnC,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;QAC1D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;IAClD,CAAC;IAED,UAAU,CAAC,SAAgB,EAAE,OAAc;QACzC,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAU,EAAE,CAAC;QACzB,IAAI,SAAS,EAAE,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC;QAAC,CAAC;QAC3F,IAAI,OAAO,EAAE,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;QAAC,CAAC;QACvF,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAElF,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;;;;QAS3B,WAAW;KACd,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAQ,CAAC;QAEzB,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;QAGnC,WAAW;;KAEd,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;QAC3B,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;QAGhC,WAAW;;KAEd,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;QAC3B,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;QAGhC,WAAW;;KAEd,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;QAE3B,MAAM,UAAU,GAA2B,EAAE,CAAC;QAC9C,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,KAAK,MAAM,GAAG,IAAI,cAAc;YAAE,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC;QACzE,KAAK,MAAM,GAAG,IAAI,WAAW;YAAE,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC;QAClE,KAAK,MAAM,GAAG,IAAI,WAAW;YAAE,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC;QAElE,OAAO;YACL,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC;YACnF,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,IAAI,IAAI,EAAE,CAAC;YAC9E,SAAS,EAAE,MAAM,CAAC,UAAU;YAC5B,gBAAgB,EAAE,MAAM,CAAC,WAAW;YACpC,iBAAiB,EAAE,MAAM,CAAC,YAAY;YACtC,aAAa,EAAE,MAAM,CAAC,cAAc;YACpC,UAAU;YACV,OAAO;YACP,OAAO;YACP,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,MAAY;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,8CAA8C,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;QACtG,OAAO,GAAG,CAAC,OAAO,IAAI,CAAC,CAAC;IAC1B,CAAC;IAEO,WAAW,CAAC,GAAQ;QAC1B,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC;YAClC,UAAU,EAAE,GAAG,CAAC,WAAW;YAC3B,KAAK,EAAE,GAAG,CAAC,MAAM;YACjB,QAAQ,EAAE,GAAG,CAAC,SAAS,IAAI,SAAS;YACpC,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,UAAU,EAAE;gBACV,WAAW,EAAE,GAAG,CAAC,YAAY;gBAC7B,YAAY,EAAE,GAAG,CAAC,aAAa;gBAC/B,YAAY,EAAE,GAAG,CAAC,aAAa,IAAI,CAAC;gBACpC,eAAe,EAAE,GAAG,CAAC,gBAAgB,IAAI,CAAC;aAC3C;YACD,aAAa,EAAE,GAAG,CAAC,cAAc;YACjC,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,KAAK;YAC/B,QAAQ,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE;SACvD,CAAC;IACJ,CAAC;CACF;AAED,MAAM,OAAO,qBAAsB,SAAQ,WAAW;IACjC;IAAnB,YAAmB,KAAgB,EAAE,eAAiC,EAAE,aAAkC;QACxG,KAAK,CAAC,eAAe,IAAI,IAAI,eAAe,EAAE,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC;QADpD,UAAK,GAAL,KAAK,CAAW;IAEnC,CAAC;IAED,WAAW,CAAC,MASX;QACC,MAAM,MAAM,GAAG,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU,CAAC,SAAgB,EAAE,OAAc;QACzC,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACnD,CAAC;CACF;AAED,MAAM,OAAO,qBAAqB;IACb;IAAnB,YAAmB,kBAAmC,IAAI,eAAe,EAAE;QAAxD,oBAAe,GAAf,eAAe,CAAyC;IAAG,CAAC;IAE/E,gBAAgB,CAAC,SAAiB,EAAE,WAAmB,EAAE,YAAoB;QAC3E,OAAO,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,SAAS,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;IAClF,CAAC;IAED,oBAAoB,CAClB,SAAiB,EACjB,SAAiB,EACjB,wBAAgC,IAAI,EACpC,yBAAiC,GAAG;QAEpC,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,qBAAqB,EAAE,sBAAsB,CAAC,CAAC;QACjG,OAAO,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;IACzD,CAAC;IAED,aAAa,CAAC,MAAgB,EAAE,WAAmB,EAAE,YAAoB;QACvE,MAAM,MAAM,GAAkC,EAAE,CAAC;QACjD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACvF,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}
|
|
@@ -0,0 +1,162 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Credential encryption and management.
|
|
3
|
+
*
|
|
4
|
+
* Provides secure credential storage with encryption support for:
|
|
5
|
+
* - Age encryption (external binary)
|
|
6
|
+
* - GPG encryption (external binary)
|
|
7
|
+
* - Fernet encryption (built-in via Node crypto)
|
|
8
|
+
*/
|
|
9
|
+
export declare enum EncryptionBackend {
|
|
10
|
+
AGE = "age",
|
|
11
|
+
GPG = "gpg",
|
|
12
|
+
FERNET = "fernet"
|
|
13
|
+
}
|
|
14
|
+
export declare enum CredentialType {
|
|
15
|
+
API_KEY = "api_key",
|
|
16
|
+
TOKEN = "token",
|
|
17
|
+
PASSWORD = "password",
|
|
18
|
+
CERTIFICATE = "certificate",
|
|
19
|
+
SSH_KEY = "ssh_key",
|
|
20
|
+
SECRET = "secret",
|
|
21
|
+
OAUTH_TOKEN = "oauth_token",
|
|
22
|
+
CUSTOM = "custom"
|
|
23
|
+
}
|
|
24
|
+
export interface Credential {
|
|
25
|
+
name: string;
|
|
26
|
+
credentialType: CredentialType;
|
|
27
|
+
value: string;
|
|
28
|
+
description?: string | undefined;
|
|
29
|
+
metadata: Record<string, unknown>;
|
|
30
|
+
createdAt: Date;
|
|
31
|
+
updatedAt: Date;
|
|
32
|
+
expiresAt?: Date | undefined;
|
|
33
|
+
tags: string[];
|
|
34
|
+
}
|
|
35
|
+
export declare class EncryptionError extends Error {
|
|
36
|
+
}
|
|
37
|
+
export declare class KeyNotFoundError extends Error {
|
|
38
|
+
}
|
|
39
|
+
export declare class CredentialNotFoundError extends Error {
|
|
40
|
+
}
|
|
41
|
+
export interface Encryptor {
|
|
42
|
+
encrypt(plaintext: string): string;
|
|
43
|
+
decrypt(ciphertext: string): string;
|
|
44
|
+
isAvailable(): boolean;
|
|
45
|
+
generateKey(): string;
|
|
46
|
+
}
|
|
47
|
+
/**
|
|
48
|
+
* Fernet-compatible encryption using Node's built-in crypto.
|
|
49
|
+
*
|
|
50
|
+
* Fernet format:
|
|
51
|
+
* - Version (1 byte): 0x80
|
|
52
|
+
* - Timestamp (8 bytes): big-endian seconds since epoch
|
|
53
|
+
* - IV (16 bytes): random initialization vector
|
|
54
|
+
* - Ciphertext: AES-128-CBC encrypted, PKCS7 padded
|
|
55
|
+
* - HMAC (32 bytes): SHA256 HMAC of version || timestamp || iv || ciphertext
|
|
56
|
+
*
|
|
57
|
+
* Key format: URL-safe base64 encoded 32 bytes (16 bytes signing key + 16 bytes encryption key)
|
|
58
|
+
*/
|
|
59
|
+
export declare class FernetEncryptor implements Encryptor {
|
|
60
|
+
private signingKey;
|
|
61
|
+
private encryptionKey;
|
|
62
|
+
constructor(key?: string, keyFile?: string);
|
|
63
|
+
private decodeKey;
|
|
64
|
+
setKey(key: string): void;
|
|
65
|
+
encrypt(plaintext: string): string;
|
|
66
|
+
decrypt(ciphertext: string): string;
|
|
67
|
+
isAvailable(): boolean;
|
|
68
|
+
generateKey(): string;
|
|
69
|
+
}
|
|
70
|
+
export declare class AgeEncryptor implements Encryptor {
|
|
71
|
+
private recipient?;
|
|
72
|
+
private identityFile?;
|
|
73
|
+
private passphrase?;
|
|
74
|
+
constructor(recipient?: string | undefined, identityFile?: string | undefined, passphrase?: string | undefined);
|
|
75
|
+
private findBinary;
|
|
76
|
+
encrypt(plaintext: string): string;
|
|
77
|
+
decrypt(ciphertext: string): string;
|
|
78
|
+
isAvailable(): boolean;
|
|
79
|
+
generateKey(): string;
|
|
80
|
+
static extractPublicKey(identityContent: string): string;
|
|
81
|
+
}
|
|
82
|
+
export declare class GPGEncryptor implements Encryptor {
|
|
83
|
+
private recipient?;
|
|
84
|
+
private passphrase?;
|
|
85
|
+
private symmetric;
|
|
86
|
+
constructor(recipient?: string | undefined, _keyId?: string, passphrase?: string | undefined, symmetric?: boolean);
|
|
87
|
+
private findBinary;
|
|
88
|
+
encrypt(plaintext: string): string;
|
|
89
|
+
decrypt(ciphertext: string): string;
|
|
90
|
+
isAvailable(): boolean;
|
|
91
|
+
generateKey(): string;
|
|
92
|
+
}
|
|
93
|
+
export interface CredentialStore {
|
|
94
|
+
save(credential: Credential): void;
|
|
95
|
+
get(name: string): Credential | null;
|
|
96
|
+
delete(name: string): boolean;
|
|
97
|
+
list(tag?: string): Credential[];
|
|
98
|
+
exists(name: string): boolean;
|
|
99
|
+
}
|
|
100
|
+
export declare class InMemoryCredentialStore implements CredentialStore {
|
|
101
|
+
private credentials;
|
|
102
|
+
save(credential: Credential): void;
|
|
103
|
+
get(name: string): Credential | null;
|
|
104
|
+
delete(name: string): boolean;
|
|
105
|
+
list(tag?: string): Credential[];
|
|
106
|
+
exists(name: string): boolean;
|
|
107
|
+
}
|
|
108
|
+
export declare class SQLiteCredentialStore implements CredentialStore {
|
|
109
|
+
private db;
|
|
110
|
+
constructor(dbPath: string);
|
|
111
|
+
private init;
|
|
112
|
+
save(credential: Credential): void;
|
|
113
|
+
get(name: string): Credential | null;
|
|
114
|
+
delete(name: string): boolean;
|
|
115
|
+
list(tag?: string): Credential[];
|
|
116
|
+
exists(name: string): boolean;
|
|
117
|
+
}
|
|
118
|
+
export declare class CredentialManager {
|
|
119
|
+
store: CredentialStore;
|
|
120
|
+
encryptor: Encryptor;
|
|
121
|
+
constructor(store: CredentialStore, encryptor: Encryptor);
|
|
122
|
+
set(params: {
|
|
123
|
+
name: string;
|
|
124
|
+
value: string;
|
|
125
|
+
credentialType?: CredentialType;
|
|
126
|
+
description?: string;
|
|
127
|
+
metadata?: Record<string, unknown>;
|
|
128
|
+
expiresAt?: Date;
|
|
129
|
+
tags?: string[];
|
|
130
|
+
}): Credential;
|
|
131
|
+
get(name: string, decrypt?: boolean): string;
|
|
132
|
+
getCredential(name: string): Credential | null;
|
|
133
|
+
delete(name: string): boolean;
|
|
134
|
+
exists(name: string): boolean;
|
|
135
|
+
list(tag?: string, includeExpired?: boolean): Credential[];
|
|
136
|
+
rotate(name: string, newValue: string): Credential;
|
|
137
|
+
export(name: string): Record<string, unknown>;
|
|
138
|
+
importCredential(data: Record<string, unknown>): Credential;
|
|
139
|
+
}
|
|
140
|
+
export declare class KeyManager {
|
|
141
|
+
private keyDir;
|
|
142
|
+
constructor(keyDir: string);
|
|
143
|
+
private keyFile;
|
|
144
|
+
generateFernetKey(name?: string): string;
|
|
145
|
+
generateAgeIdentity(name?: string): {
|
|
146
|
+
identity: string;
|
|
147
|
+
publicKey: string;
|
|
148
|
+
};
|
|
149
|
+
getFernetKey(name?: string): string | null;
|
|
150
|
+
getAgeIdentityFile(name?: string): string | null;
|
|
151
|
+
getAgePublicKey(name?: string): string | null;
|
|
152
|
+
listKeys(): Array<Record<string, unknown>>;
|
|
153
|
+
deleteKey(name: string): boolean;
|
|
154
|
+
}
|
|
155
|
+
export declare function createCredentialManager(params: {
|
|
156
|
+
stateDir: string;
|
|
157
|
+
backend?: EncryptionBackend;
|
|
158
|
+
keyName?: string;
|
|
159
|
+
passphrase?: string;
|
|
160
|
+
}): CredentialManager;
|
|
161
|
+
export declare function getAvailableBackends(): EncryptionBackend[];
|
|
162
|
+
//# sourceMappingURL=credentials.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"credentials.d.ts","sourceRoot":"","sources":["../src/credentials.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAQH,oBAAY,iBAAiB;IAC3B,GAAG,QAAQ;IACX,GAAG,QAAQ;IACX,MAAM,WAAW;CAClB;AAED,oBAAY,cAAc;IACxB,OAAO,YAAY;IACnB,KAAK,UAAU;IACf,QAAQ,aAAa;IACrB,WAAW,gBAAgB;IAC3B,OAAO,YAAY;IACnB,MAAM,WAAW;IACjB,WAAW,gBAAgB;IAC3B,MAAM,WAAW;CAClB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,cAAc,EAAE,cAAc,CAAC;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,CAAC,EAAE,IAAI,GAAG,SAAS,CAAC;IAC7B,IAAI,EAAE,MAAM,EAAE,CAAC;CAChB;AAED,qBAAa,eAAgB,SAAQ,KAAK;CAAG;AAC7C,qBAAa,gBAAiB,SAAQ,KAAK;CAAG;AAC9C,qBAAa,uBAAwB,SAAQ,KAAK;CAAG;AAErD,MAAM,WAAW,SAAS;IACxB,OAAO,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC;IACnC,OAAO,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAAC;IACpC,WAAW,IAAI,OAAO,CAAC;IACvB,WAAW,IAAI,MAAM,CAAC;CACvB;AAED;;;;;;;;;;;GAWG;AACH,qBAAa,eAAgB,YAAW,SAAS;IAC/C,OAAO,CAAC,UAAU,CAAuB;IACzC,OAAO,CAAC,aAAa,CAAuB;gBAEhC,GAAG,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM;IAQ1C,OAAO,CAAC,SAAS;IAejB,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAMzB,OAAO,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAkClC,OAAO,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM;IAsDnC,WAAW,IAAI,OAAO;IAItB,WAAW,IAAI,MAAM;CAKtB;AAED,qBAAa,YAAa,YAAW,SAAS;IAE1C,OAAO,CAAC,SAAS,CAAC;IAClB,OAAO,CAAC,YAAY,CAAC;IACrB,OAAO,CAAC,UAAU,CAAC;gBAFX,SAAS,CAAC,EAAE,MAAM,YAAA,EAClB,YAAY,CAAC,EAAE,MAAM,YAAA,EACrB,UAAU,CAAC,EAAE,MAAM,YAAA;IAG7B,OAAO,CAAC,UAAU;IASlB,OAAO,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAqBlC,OAAO,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM;IAiBnC,WAAW,IAAI,OAAO;IAStB,WAAW,IAAI,MAAM;IASrB,MAAM,CAAC,gBAAgB,CAAC,eAAe,EAAE,MAAM,GAAG,MAAM;CAYzD;AAED,qBAAa,YAAa,YAAW,SAAS;IAE1C,OAAO,CAAC,SAAS,CAAC;IAGlB,OAAO,CAAC,UAAU,CAAC;IACnB,OAAO,CAAC,SAAS;gBAJT,SAAS,CAAC,EAAE,MAAM,YAAA,EAE1B,MAAM,CAAC,EAAE,MAAM,EACP,UAAU,CAAC,EAAE,MAAM,YAAA,EACnB,SAAS,GAAE,OAAe;IAGpC,OAAO,CAAC,UAAU;IAWlB,OAAO,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAsBlC,OAAO,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM;IAcnC,WAAW,IAAI,OAAO;IAStB,WAAW,IAAI,MAAM;CAWtB;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI,CAAC;IACnC,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI,CAAC;IACrC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC;IAC9B,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,GAAG,UAAU,EAAE,CAAC;IACjC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC;CAC/B;AAED,qBAAa,uBAAwB,YAAW,eAAe;IAC7D,OAAO,CAAC,WAAW,CAAiC;IAEpD,IAAI,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI;IAKlC,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI;IAIpC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI7B,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,GAAG,UAAU,EAAE;IAMhC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;CAG9B;AAED,qBAAa,qBAAsB,YAAW,eAAe;IAC3D,OAAO,CAAC,EAAE,CAAoB;gBAElB,MAAM,EAAE,MAAM;IAS1B,OAAO,CAAC,IAAI;IAiBZ,IAAI,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI;IAoBlC,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI;IA+BpC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAK7B,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,GAAG,UAAU,EAAE;IA2BhC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;CAI9B;AAED,qBAAa,iBAAiB;IAEnB,KAAK,EAAE,eAAe;IACtB,SAAS,EAAE,SAAS;gBADpB,KAAK,EAAE,eAAe,EACtB,SAAS,EAAE,SAAS;IAG7B,GAAG,CAAC,MAAM,EAAE;QACV,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACnC,SAAS,CAAC,EAAE,IAAI,CAAC;QACjB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;KACjB,GAAG,UAAU;IAkBd,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,GAAE,OAAc,GAAG,MAAM;IAWlD,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI;IAI9C,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI7B,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI7B,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,cAAc,GAAE,OAAe,GAAG,UAAU,EAAE;IAMjE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,UAAU;IAgBlD,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAkB7C,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,UAAU;CAgB5D;AAED,qBAAa,UAAU;IACT,OAAO,CAAC,MAAM;gBAAN,MAAM,EAAE,MAAM;IAMlC,OAAO,CAAC,OAAO;IAIf,iBAAiB,CAAC,IAAI,GAAE,MAAkB,GAAG,MAAM;IAQnD,mBAAmB,CAAC,IAAI,GAAE,MAAkB,GAAG;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE;IAWtF,YAAY,CAAC,IAAI,GAAE,MAAkB,GAAG,MAAM,GAAG,IAAI;IAKrD,kBAAkB,CAAC,IAAI,GAAE,MAAkB,GAAG,MAAM,GAAG,IAAI;IAK3D,eAAe,CAAC,IAAI,GAAE,MAAkB,GAAG,MAAM,GAAG,IAAI;IAKxD,QAAQ,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAwB1C,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;CAmBjC;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE;IAC9C,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,iBAAiB,CAAC;IAC5B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,iBAAiB,CA4CpB;AAED,wBAAgB,oBAAoB,IAAI,iBAAiB,EAAE,CAW1D"}
|