tokenlens-cli 1.0.0

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.
@@ -0,0 +1,243 @@
1
+ import { createHash } from "crypto";
2
+ import { createReadStream, existsSync, readdirSync, statSync } from "fs";
3
+ import { basename, extname, join, resolve } from "path";
4
+ import { createInterface } from "readline";
5
+ import { defaultImportPaths } from "../config.js";
6
+ function listDataFiles(root) {
7
+ if (!existsSync(root))
8
+ return [];
9
+ const stat = statSync(root);
10
+ if (stat.isFile())
11
+ return [root];
12
+ const files = [];
13
+ for (const entry of readdirSync(root, { withFileTypes: true })) {
14
+ const path = join(root, entry.name);
15
+ if (entry.isDirectory())
16
+ files.push(...listDataFiles(path));
17
+ else if ([".jsonl", ".json", ".ndjson"].includes(extname(entry.name).toLowerCase()))
18
+ files.push(path);
19
+ }
20
+ return files;
21
+ }
22
+ function textContent(value) {
23
+ if (typeof value === "string")
24
+ return value;
25
+ if (Array.isArray(value)) {
26
+ return value
27
+ .map((item) => typeof item === "string" ? item : item?.text ?? item?.content ?? "")
28
+ .filter(Boolean)
29
+ .join(" ");
30
+ }
31
+ return "";
32
+ }
33
+ function usageFrom(record) {
34
+ const usage = record.usage ?? record.message?.usage ?? record.response?.usage ?? record.usageMetadata;
35
+ if (usage) {
36
+ const cached = usage.cached_input_tokens ??
37
+ usage.cache_read_input_tokens ??
38
+ usage.prompt_tokens_details?.cached_tokens ??
39
+ 0;
40
+ const cacheCreation = usage.cache_creation_input_tokens ?? 0;
41
+ const prompt = usage.prompt_tokens ??
42
+ usage.input_tokens ??
43
+ usage.promptTokenCount ??
44
+ 0;
45
+ const completion = usage.completion_tokens ??
46
+ usage.output_tokens ??
47
+ usage.candidatesTokenCount ??
48
+ 0;
49
+ const reasoning = usage.reasoning_output_tokens ??
50
+ usage.completion_tokens_details?.reasoning_tokens ??
51
+ 0;
52
+ return {
53
+ prompt: prompt + cached + cacheCreation,
54
+ completion,
55
+ cached,
56
+ reasoning,
57
+ };
58
+ }
59
+ if (record.prompt_eval_count !== undefined || record.eval_count !== undefined) {
60
+ return {
61
+ prompt: record.prompt_eval_count ?? 0,
62
+ completion: record.eval_count ?? 0,
63
+ cached: 0,
64
+ reasoning: 0,
65
+ };
66
+ }
67
+ return null;
68
+ }
69
+ function externalId(source, path, line) {
70
+ return createHash("sha1").update(`${source}:${path}:${line}`).digest("hex");
71
+ }
72
+ function normalizePrompt(value) {
73
+ return value.toLowerCase().replace(/\s+/g, " ").trim();
74
+ }
75
+ function promptHash(value) {
76
+ const normalized = normalizePrompt(value);
77
+ if (!normalized)
78
+ return null;
79
+ return createHash("sha1").update(normalized).digest("hex");
80
+ }
81
+ function promptFamilyHash(value) {
82
+ const normalized = normalizePrompt(value);
83
+ if (!normalized)
84
+ return null;
85
+ const family = normalized.split(" ").slice(0, 40).join(" ");
86
+ if (!family)
87
+ return null;
88
+ return createHash("sha1").update(family).digest("hex");
89
+ }
90
+ async function importFile(store, path, sourceHint) {
91
+ const normalized = path.toLowerCase();
92
+ const source = sourceHint ??
93
+ (normalized.includes(`${join(".codex", "sessions").toLowerCase()}`) ? "codex" :
94
+ normalized.includes(`${join(".claude", "projects").toLowerCase()}`) ? "claude" :
95
+ "generic");
96
+ const sessionId = basename(path, extname(path)).replace(/^rollout-[^-]+-[^-]+-/, "");
97
+ const reader = createInterface({ input: createReadStream(path, { encoding: "utf8" }), crlfDelay: Infinity });
98
+ let imported = 0;
99
+ let skipped = 0;
100
+ let lineNumber = 0;
101
+ let model = source === "codex" ? "codex" : "unknown";
102
+ let promptText = "";
103
+ for await (const line of reader) {
104
+ lineNumber++;
105
+ if (!line.trim())
106
+ continue;
107
+ let record;
108
+ try {
109
+ record = JSON.parse(line);
110
+ }
111
+ catch {
112
+ skipped++;
113
+ continue;
114
+ }
115
+ if (source === "codex") {
116
+ if (record.type === "turn_context" && record.payload?.model)
117
+ model = record.payload.model;
118
+ if (record.type === "response_item" && record.payload?.role === "user") {
119
+ promptText = textContent(record.payload.content);
120
+ }
121
+ const last = record.payload?.type === "token_count"
122
+ ? record.payload?.info?.last_token_usage
123
+ : null;
124
+ if (!last)
125
+ continue;
126
+ const currentPrompt = promptText.slice(0, 500);
127
+ const result = store.insertCall({
128
+ toolName: "Codex",
129
+ source,
130
+ provider: "OpenAI",
131
+ model,
132
+ endpoint: "local-log",
133
+ promptTokens: last.input_tokens ?? 0,
134
+ cachedTokens: last.cached_input_tokens ?? 0,
135
+ completionTokens: last.output_tokens ?? 0,
136
+ reasoningTokens: last.reasoning_output_tokens ?? 0,
137
+ promptPreview: currentPrompt,
138
+ completionPreview: "",
139
+ sessionId,
140
+ externalId: externalId(source, path, lineNumber),
141
+ promptHash: promptHash(promptText),
142
+ promptFamilyHash: promptFamilyHash(promptText),
143
+ createdAt: record.timestamp,
144
+ });
145
+ if (result > 0)
146
+ imported++;
147
+ else
148
+ skipped++;
149
+ continue;
150
+ }
151
+ if (source === "claude") {
152
+ if (record.type === "user")
153
+ promptText = textContent(record.message?.content);
154
+ if (!record.message?.usage)
155
+ continue;
156
+ const usage = usageFrom(record);
157
+ if (!usage || usage.prompt + usage.completion === 0)
158
+ continue;
159
+ const currentPrompt = promptText.slice(0, 500);
160
+ const result = store.insertCall({
161
+ toolName: "Claude Code",
162
+ source,
163
+ provider: "Anthropic",
164
+ model: record.message?.model ?? "claude",
165
+ endpoint: "local-log",
166
+ promptTokens: usage.prompt,
167
+ cachedTokens: usage.cached,
168
+ completionTokens: usage.completion,
169
+ reasoningTokens: usage.reasoning,
170
+ promptPreview: currentPrompt,
171
+ completionPreview: textContent(record.message?.content).slice(0, 500),
172
+ sessionId: record.sessionId ?? sessionId,
173
+ externalId: externalId(source, path, lineNumber),
174
+ promptHash: promptHash(promptText),
175
+ promptFamilyHash: promptFamilyHash(promptText),
176
+ createdAt: record.timestamp,
177
+ });
178
+ if (result > 0)
179
+ imported++;
180
+ else
181
+ skipped++;
182
+ continue;
183
+ }
184
+ const usage = usageFrom(record);
185
+ if (!usage || usage.prompt + usage.completion === 0)
186
+ continue;
187
+ const genericPrompt = textContent(record.prompt ?? record.messages ?? record.request?.messages);
188
+ const result = store.insertCall({
189
+ toolName: record.tool ?? record.provider ?? sourceHint ?? "Imported",
190
+ source,
191
+ provider: record.provider ?? record.vendor ?? sourceHint ?? "Imported",
192
+ model: record.model ?? record.response?.model ?? "unknown",
193
+ endpoint: record.endpoint ?? record.provider ?? "import",
194
+ promptTokens: usage.prompt,
195
+ cachedTokens: usage.cached,
196
+ completionTokens: usage.completion,
197
+ reasoningTokens: usage.reasoning,
198
+ promptPreview: genericPrompt.slice(0, 500),
199
+ completionPreview: textContent(record.response ?? record.message?.content).slice(0, 500),
200
+ sessionId: record.session_id ?? record.sessionId ?? null,
201
+ externalId: externalId(source, path, lineNumber),
202
+ promptHash: promptHash(genericPrompt),
203
+ promptFamilyHash: promptFamilyHash(genericPrompt),
204
+ createdAt: record.created_at ?? record.timestamp,
205
+ });
206
+ if (result > 0)
207
+ imported++;
208
+ else
209
+ skipped++;
210
+ }
211
+ return { imported, skipped, source };
212
+ }
213
+ export async function importLogs(store, paths = [], force = false, configuredDefaults = defaultImportPaths()) {
214
+ const defaults = configuredDefaults.length ? configuredDefaults : defaultImportPaths();
215
+ const roots = paths.length ? paths.map((path) => resolve(path)) : defaults;
216
+ const files = roots.flatMap(listDataFiles);
217
+ const result = {
218
+ filesFound: files.length,
219
+ filesRead: 0,
220
+ callsImported: 0,
221
+ callsSkipped: 0,
222
+ errors: [],
223
+ sources: {},
224
+ };
225
+ for (const path of files) {
226
+ try {
227
+ const stat = statSync(path);
228
+ if (!force && store.isFileImported(path, stat.size, Math.round(stat.mtimeMs)))
229
+ continue;
230
+ const fileResult = await importFile(store, path);
231
+ result.filesRead++;
232
+ result.callsImported += fileResult.imported;
233
+ result.callsSkipped += fileResult.skipped;
234
+ result.sources[fileResult.source] = (result.sources[fileResult.source] ?? 0) + fileResult.imported;
235
+ store.markFileImported(path, stat.size, Math.round(stat.mtimeMs));
236
+ }
237
+ catch (error) {
238
+ result.errors.push(`${path}: ${error instanceof Error ? error.message : String(error)}`);
239
+ }
240
+ }
241
+ return result;
242
+ }
243
+ //# sourceMappingURL=logs.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"logs.js","sourceRoot":"","sources":["../../src/importers/logs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,gBAAgB,EAAE,UAAU,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AACzE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3C,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAclD,SAAS,aAAa,CAAC,IAAY;IACjC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IACjC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC5B,IAAI,IAAI,CAAC,MAAM,EAAE;QAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAC/D,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,KAAK,CAAC,WAAW,EAAE;YAAE,KAAK,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;aACvD,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxG,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK;aACT,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,IAAI,IAAI,EAAE,OAAO,IAAI,EAAE,CAAC;aAClF,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,SAAS,CAAC,MAAkB;IAMnC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,KAAK,IAAI,MAAM,CAAC,QAAQ,EAAE,KAAK,IAAI,MAAM,CAAC,aAAa,CAAC;IACtG,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,MAAM,GACV,KAAK,CAAC,mBAAmB;YACzB,KAAK,CAAC,uBAAuB;YAC7B,KAAK,CAAC,qBAAqB,EAAE,aAAa;YAC1C,CAAC,CAAC;QACJ,MAAM,aAAa,GAAG,KAAK,CAAC,2BAA2B,IAAI,CAAC,CAAC;QAC7D,MAAM,MAAM,GACV,KAAK,CAAC,aAAa;YACnB,KAAK,CAAC,YAAY;YAClB,KAAK,CAAC,gBAAgB;YACtB,CAAC,CAAC;QACJ,MAAM,UAAU,GACd,KAAK,CAAC,iBAAiB;YACvB,KAAK,CAAC,aAAa;YACnB,KAAK,CAAC,oBAAoB;YAC1B,CAAC,CAAC;QACJ,MAAM,SAAS,GACb,KAAK,CAAC,uBAAuB;YAC7B,KAAK,CAAC,yBAAyB,EAAE,gBAAgB;YACjD,CAAC,CAAC;QACJ,OAAO;YACL,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,aAAa;YACvC,UAAU;YACV,MAAM;YACN,SAAS;SACV,CAAC;IACJ,CAAC;IAED,IAAI,MAAM,CAAC,iBAAiB,KAAK,SAAS,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QAC9E,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,iBAAiB,IAAI,CAAC;YACrC,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,CAAC;YAClC,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;SACb,CAAC;IACJ,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,UAAU,CAAC,MAAc,EAAE,IAAY,EAAE,IAAY;IAC5D,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,eAAe,CAAC,KAAa;IACpC,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAC/B,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IACrC,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,MAAM,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC5D,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IACzB,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACzD,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,KAAiB,EAAE,IAAY,EAAE,UAAmB;IAK5E,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACtC,MAAM,MAAM,GAAG,UAAU;QACvB,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YAC7E,UAAU,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBAC9E,SAAS,CAAC,CAAC;IACjB,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;IACrF,MAAM,MAAM,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,gBAAgB,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC7G,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,KAAK,GAAG,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;IACrD,IAAI,UAAU,GAAG,EAAE,CAAC;IAEpB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QAChC,UAAU,EAAE,CAAC;QACb,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAAE,SAAS;QAC3B,IAAI,MAAkB,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;YACV,SAAS;QACX,CAAC;QAED,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YACvB,IAAI,MAAM,CAAC,IAAI,KAAK,cAAc,IAAI,MAAM,CAAC,OAAO,EAAE,KAAK;gBAAE,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;YAC1F,IAAI,MAAM,CAAC,IAAI,KAAK,eAAe,IAAI,MAAM,CAAC,OAAO,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;gBACvE,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,IAAI,KAAK,aAAa;gBACjD,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,gBAAgB;gBACxC,CAAC,CAAC,IAAI,CAAC;YACT,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC;gBAC9B,QAAQ,EAAE,OAAO;gBACjB,MAAM;gBACN,QAAQ,EAAE,QAAQ;gBAClB,KAAK;gBACL,QAAQ,EAAE,WAAW;gBACrB,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,CAAC;gBACpC,YAAY,EAAE,IAAI,CAAC,mBAAmB,IAAI,CAAC;gBAC3C,gBAAgB,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC;gBACzC,eAAe,EAAE,IAAI,CAAC,uBAAuB,IAAI,CAAC;gBAClD,aAAa,EAAE,aAAa;gBAC5B,iBAAiB,EAAE,EAAE;gBACrB,SAAS;gBACT,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC;gBAChD,UAAU,EAAE,UAAU,CAAC,UAAU,CAAC;gBAClC,gBAAgB,EAAE,gBAAgB,CAAC,UAAU,CAAC;gBAC9C,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC;YACH,IAAI,MAAM,GAAG,CAAC;gBAAE,QAAQ,EAAE,CAAC;;gBAAM,OAAO,EAAE,CAAC;YAC3C,SAAS;QACX,CAAC;QAED,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxB,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM;gBAAE,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9E,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK;gBAAE,SAAS;YACrC,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,UAAU,KAAK,CAAC;gBAAE,SAAS;YAC9D,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC;gBAC9B,QAAQ,EAAE,aAAa;gBACvB,MAAM;gBACN,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,KAAK,IAAI,QAAQ;gBACxC,QAAQ,EAAE,WAAW;gBACrB,YAAY,EAAE,KAAK,CAAC,MAAM;gBAC1B,YAAY,EAAE,KAAK,CAAC,MAAM;gBAC1B,gBAAgB,EAAE,KAAK,CAAC,UAAU;gBAClC,eAAe,EAAE,KAAK,CAAC,SAAS;gBAChC,aAAa,EAAE,aAAa;gBAC5B,iBAAiB,EAAE,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;gBACrE,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,SAAS;gBACxC,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC;gBAChD,UAAU,EAAE,UAAU,CAAC,UAAU,CAAC;gBAClC,gBAAgB,EAAE,gBAAgB,CAAC,UAAU,CAAC;gBAC9C,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC;YACH,IAAI,MAAM,GAAG,CAAC;gBAAE,QAAQ,EAAE,CAAC;;gBAAM,OAAO,EAAE,CAAC;YAC3C,SAAS;QACX,CAAC;QAED,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,UAAU,KAAK,CAAC;YAAE,SAAS;QAC9D,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAChG,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC;YAC9B,QAAQ,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,QAAQ,IAAI,UAAU,IAAI,UAAU;YACpE,MAAM;YACN,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,MAAM,IAAI,UAAU,IAAI,UAAU;YACtE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,QAAQ,EAAE,KAAK,IAAI,SAAS;YAC1D,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,IAAI,QAAQ;YACxD,YAAY,EAAE,KAAK,CAAC,MAAM;YAC1B,YAAY,EAAE,KAAK,CAAC,MAAM;YAC1B,gBAAgB,EAAE,KAAK,CAAC,UAAU;YAClC,eAAe,EAAE,KAAK,CAAC,SAAS;YAChC,aAAa,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;YAC1C,iBAAiB,EAAE,WAAW,CAAC,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;YACxF,SAAS,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,SAAS,IAAI,IAAI;YACxD,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC;YAChD,UAAU,EAAE,UAAU,CAAC,aAAa,CAAC;YACrC,gBAAgB,EAAE,gBAAgB,CAAC,aAAa,CAAC;YACjD,SAAS,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,SAAS;SACjD,CAAC,CAAC;QACH,IAAI,MAAM,GAAG,CAAC;YAAE,QAAQ,EAAE,CAAC;;YAAM,OAAO,EAAE,CAAC;IAC7C,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AACvC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,KAAiB,EACjB,QAAkB,EAAE,EACpB,KAAK,GAAG,KAAK,EACb,qBAA+B,kBAAkB,EAAE;IAEnD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC;IACvF,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IAC3E,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAiB;QAC3B,UAAU,EAAE,KAAK,CAAC,MAAM;QACxB,SAAS,EAAE,CAAC;QACZ,aAAa,EAAE,CAAC;QAChB,YAAY,EAAE,CAAC;QACf,MAAM,EAAE,EAAE;QACV,OAAO,EAAE,EAAE;KACZ,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC5B,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAAE,SAAS;YACxF,MAAM,UAAU,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,CAAC,aAAa,IAAI,UAAU,CAAC,QAAQ,CAAC;YAC5C,MAAM,CAAC,YAAY,IAAI,UAAU,CAAC,OAAO,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;YACnG,KAAK,CAAC,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QACpE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC3F,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}
@@ -0,0 +1,3 @@
1
+ #!/usr/bin/env node
2
+ export {};
3
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":""}
package/dist/index.js ADDED
@@ -0,0 +1,31 @@
1
+ #!/usr/bin/env node
2
+ import { loadConfig } from "./config.js";
3
+ import { TokenStore } from "./db/store.js";
4
+ import { ProxyServer } from "./proxy/server.js";
5
+ import { DashboardServer } from "./dashboard/server.js";
6
+ const args = process.argv.slice(2);
7
+ const isDashboardOnly = args.includes("--dashboard-only");
8
+ const isProxyOnly = args.includes("--proxy-only");
9
+ const config = loadConfig();
10
+ const store = new TokenStore(config.dbPath);
11
+ if (isDashboardOnly) {
12
+ const dash = new DashboardServer(store);
13
+ dash.start(config.dashboardPort);
14
+ process.on("SIGINT", () => { dash.stop(); store.close(); process.exit(0); });
15
+ process.on("SIGTERM", () => { dash.stop(); store.close(); process.exit(0); });
16
+ }
17
+ else if (isProxyOnly) {
18
+ const proxy = new ProxyServer(store);
19
+ proxy.start(config.proxyPort);
20
+ process.on("SIGINT", () => { proxy.stop(); store.close(); process.exit(0); });
21
+ process.on("SIGTERM", () => { proxy.stop(); store.close(); process.exit(0); });
22
+ }
23
+ else {
24
+ const proxy = new ProxyServer(store);
25
+ const dash = new DashboardServer(store);
26
+ proxy.start(config.proxyPort);
27
+ dash.start(config.dashboardPort);
28
+ process.on("SIGINT", () => { proxy.stop(); dash.stop(); store.close(); process.exit(0); });
29
+ process.on("SIGTERM", () => { proxy.stop(); dash.stop(); store.close(); process.exit(0); });
30
+ }
31
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AACnC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;AAC1D,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;AAElD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;AAC5B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AAE5C,IAAI,eAAe,EAAE,CAAC;IACpB,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;IACxC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IACjC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAChF,CAAC;KAAM,IAAI,WAAW,EAAE,CAAC;IACvB,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IACrC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAC9B,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACjF,CAAC;KAAM,CAAC;IACN,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;IAExC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAC9B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAEjC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3F,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9F,CAAC"}
@@ -0,0 +1,11 @@
1
+ import type { TokenStore } from "../db/store.js";
2
+ export declare class ProxyServer {
3
+ private proxy;
4
+ private store;
5
+ private toolName;
6
+ constructor(store: TokenStore, toolName?: string);
7
+ start(port: number): void;
8
+ stop(): void;
9
+ private processCall;
10
+ }
11
+ //# sourceMappingURL=server.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/proxy/server.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAoBjD,qBAAa,WAAW;IACtB,OAAO,CAAC,KAAK,CAAe;IAC5B,OAAO,CAAC,KAAK,CAAa;IAC1B,OAAO,CAAC,QAAQ,CAAS;gBAEb,KAAK,EAAE,UAAU,EAAE,QAAQ,GAAE,MAAiB;IAK1D,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAwDzB,IAAI,IAAI,IAAI;IAIZ,OAAO,CAAC,WAAW;CAgFpB"}
@@ -0,0 +1,153 @@
1
+ import { createHash } from "crypto";
2
+ import { Proxy } from "http-mitm-proxy";
3
+ import { AI_ENDPOINTS } from "../config.js";
4
+ function normalizePrompt(value) {
5
+ return value.toLowerCase().replace(/\s+/g, " ").trim();
6
+ }
7
+ function promptHash(value) {
8
+ const normalized = normalizePrompt(value);
9
+ if (!normalized)
10
+ return null;
11
+ return createHash("sha1").update(normalized).digest("hex");
12
+ }
13
+ function promptFamilyHash(value) {
14
+ const normalized = normalizePrompt(value);
15
+ if (!normalized)
16
+ return null;
17
+ const family = normalized.split(" ").slice(0, 40).join(" ");
18
+ if (!family)
19
+ return null;
20
+ return createHash("sha1").update(family).digest("hex");
21
+ }
22
+ export class ProxyServer {
23
+ proxy = new Proxy();
24
+ store;
25
+ toolName;
26
+ constructor(store, toolName = "system") {
27
+ this.store = store;
28
+ this.toolName = toolName;
29
+ }
30
+ start(port) {
31
+ this.proxy.onError((_ctx, err) => {
32
+ if (err?.code !== "ECONNRESET") {
33
+ console.error("[proxy] Error:", err?.message);
34
+ }
35
+ });
36
+ this.proxy.onRequest((ctx, callback) => {
37
+ const host = ctx.clientToProxyRequest.headers.host || "";
38
+ const aiEndpoint = AI_ENDPOINTS.find((e) => host.includes(e.host));
39
+ if (!aiEndpoint)
40
+ return callback();
41
+ const req = ctx.clientToProxyRequest;
42
+ const startTime = Date.now();
43
+ const chunks = [];
44
+ req.on("data", (chunk) => chunks.push(chunk));
45
+ req.on("end", () => {
46
+ ctx.userData = {
47
+ requestBody: Buffer.concat(chunks).toString(),
48
+ startTime,
49
+ aiEndpoint,
50
+ };
51
+ });
52
+ callback();
53
+ });
54
+ this.proxy.onResponse((ctx, callback) => {
55
+ const ud = ctx.userData;
56
+ if (ud?.requestBody && ud.aiEndpoint) {
57
+ const res = ctx.serverToProxyResponse;
58
+ if (!res)
59
+ return callback();
60
+ const respChunks = [];
61
+ res.on("data", (chunk) => respChunks.push(chunk));
62
+ res.on("end", () => {
63
+ const responseBody = Buffer.concat(respChunks).toString();
64
+ this.processCall(ud.requestBody, responseBody, ud.aiEndpoint);
65
+ });
66
+ }
67
+ callback();
68
+ });
69
+ this.proxy.listen({ port, host: "127.0.0.1", sslCaDir: undefined });
70
+ console.log(`[proxy] MITM proxy listening on port ${port}`);
71
+ }
72
+ stop() {
73
+ this.proxy.close();
74
+ }
75
+ processCall(requestBody, responseBody, endpoint) {
76
+ try {
77
+ const req = JSON.parse(requestBody);
78
+ const model = req.model || "unknown";
79
+ let promptTokens = 0;
80
+ let completionTokens = 0;
81
+ let promptPreview = "";
82
+ let completionPreview = "";
83
+ let promptText = "";
84
+ if (responseBody) {
85
+ const res = JSON.parse(responseBody);
86
+ // OpenAI format
87
+ if (res.usage) {
88
+ promptTokens =
89
+ res.usage.prompt_tokens ??
90
+ res.usage.input_tokens ??
91
+ 0;
92
+ completionTokens =
93
+ res.usage.completion_tokens ??
94
+ res.usage.output_tokens ??
95
+ 0;
96
+ }
97
+ // Google Gemini format
98
+ else if (res.usageMetadata) {
99
+ promptTokens = res.usageMetadata.promptTokenCount ?? 0;
100
+ completionTokens = res.usageMetadata.candidatesTokenCount ?? 0;
101
+ }
102
+ // Build completion preview
103
+ if (res.content?.length) {
104
+ const text = res.content
105
+ .map((c) => (c.type === "text" ? c.text : ""))
106
+ .filter(Boolean)
107
+ .join(" ");
108
+ completionPreview = text.slice(0, 200);
109
+ }
110
+ else if (res.choices?.length) {
111
+ completionPreview = (res.choices[0]?.message?.content ?? "").slice(0, 200);
112
+ }
113
+ else if (res.candidates?.length) {
114
+ completionPreview = JSON.stringify(res.candidates[0]).slice(0, 200);
115
+ }
116
+ }
117
+ // Build prompt preview
118
+ if (req.messages?.length) {
119
+ const last = req.messages[req.messages.length - 1];
120
+ promptText = typeof last.content === "string" ? last.content : JSON.stringify(last.content ?? last);
121
+ promptPreview = promptText.slice(0, 200);
122
+ }
123
+ else if (req.prompt) {
124
+ promptText = String(req.prompt);
125
+ promptPreview = promptText.slice(0, 200);
126
+ }
127
+ else if (req.contents) {
128
+ promptText = JSON.stringify(req.contents);
129
+ promptPreview = promptText.slice(0, 200);
130
+ }
131
+ if (promptTokens > 0 || completionTokens > 0) {
132
+ this.store.insertCall({
133
+ toolName: this.toolName,
134
+ source: "proxy",
135
+ provider: endpoint.name,
136
+ model,
137
+ endpoint: endpoint.name,
138
+ promptTokens,
139
+ completionTokens,
140
+ promptPreview,
141
+ completionPreview,
142
+ sessionId: null,
143
+ promptHash: promptHash(promptText),
144
+ promptFamilyHash: promptFamilyHash(promptText),
145
+ });
146
+ }
147
+ }
148
+ catch {
149
+ // Parsing errors expected for streaming / non-JSON
150
+ }
151
+ }
152
+ }
153
+ //# sourceMappingURL=server.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/proxy/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAG5C,SAAS,eAAe,CAAC,KAAa;IACpC,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAC/B,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IACrC,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,MAAM,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC5D,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IACzB,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,OAAO,WAAW;IACd,KAAK,GAAG,IAAI,KAAK,EAAE,CAAC;IACpB,KAAK,CAAa;IAClB,QAAQ,CAAS;IAEzB,YAAY,KAAiB,EAAE,WAAmB,QAAQ;QACxD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,IAAY;QAChB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAC/B,IAAK,GAAW,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxC,OAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE;YACrC,MAAM,IAAI,GAAG,GAAG,CAAC,oBAAoB,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;YACzD,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YAEnE,IAAI,CAAC,UAAU;gBAAE,OAAO,QAAQ,EAAE,CAAC;YAEnC,MAAM,GAAG,GAAG,GAAG,CAAC,oBAAoB,CAAC;YACrC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAa,EAAE,CAAC;YAE5B,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACtD,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBAChB,GAAW,CAAC,QAAQ,GAAG;oBACtB,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;oBAC7C,SAAS;oBACT,UAAU;iBACX,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE;YACtC,MAAM,EAAE,GAAI,GAAW,CAAC,QAKX,CAAC;YAEd,IAAI,EAAE,EAAE,WAAW,IAAI,EAAE,CAAC,UAAU,EAAE,CAAC;gBACrC,MAAM,GAAG,GAAG,GAAG,CAAC,qBAAqB,CAAC;gBACtC,IAAI,CAAC,GAAG;oBAAE,OAAO,QAAQ,EAAE,CAAC;gBAE5B,MAAM,UAAU,GAAa,EAAE,CAAC;gBAChC,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC1D,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACjB,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,CAAC;oBAC1D,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,WAAY,EAAE,YAAY,EAAE,EAAE,CAAC,UAAW,CAAC,CAAC;gBAClE,CAAC,CAAC,CAAC;YACL,CAAC;YAED,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;QACpE,OAAO,CAAC,GAAG,CAAC,wCAAwC,IAAI,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,IAAI;QACF,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEO,WAAW,CACjB,WAAmB,EACnB,YAAoB,EACpB,QAAwC;QAExC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,SAAS,CAAC;YACrC,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,IAAI,gBAAgB,GAAG,CAAC,CAAC;YACzB,IAAI,aAAa,GAAG,EAAE,CAAC;YACvB,IAAI,iBAAiB,GAAG,EAAE,CAAC;YAC3B,IAAI,UAAU,GAAG,EAAE,CAAC;YAEpB,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBACrC,gBAAgB;gBAChB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBACd,YAAY;wBACV,GAAG,CAAC,KAAK,CAAC,aAAa;4BACvB,GAAG,CAAC,KAAK,CAAC,YAAY;4BACtB,CAAC,CAAC;oBACJ,gBAAgB;wBACd,GAAG,CAAC,KAAK,CAAC,iBAAiB;4BAC3B,GAAG,CAAC,KAAK,CAAC,aAAa;4BACvB,CAAC,CAAC;gBACN,CAAC;gBACD,uBAAuB;qBAClB,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;oBAC3B,YAAY,GAAG,GAAG,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC,CAAC;oBACvD,gBAAgB,GAAG,GAAG,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC,CAAC;gBACjE,CAAC;gBAED,2BAA2B;gBAC3B,IAAI,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;oBACxB,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO;yBACrB,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;yBAClD,MAAM,CAAC,OAAO,CAAC;yBACf,IAAI,CAAC,GAAG,CAAC,CAAC;oBACb,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBACzC,CAAC;qBAAM,IAAI,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;oBAC/B,iBAAiB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBAC7E,CAAC;qBAAM,IAAI,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;oBAClC,iBAAiB,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YAED,uBAAuB;YACvB,IAAI,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBACnD,UAAU,GAAG,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC;gBACpG,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC3C,CAAC;iBAAM,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;gBACtB,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAChC,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC3C,CAAC;iBAAM,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACxB,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAC1C,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC3C,CAAC;YAED,IAAI,YAAY,GAAG,CAAC,IAAI,gBAAgB,GAAG,CAAC,EAAE,CAAC;gBAC7C,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;oBACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,KAAK;oBACL,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,YAAY;oBACZ,gBAAgB;oBAChB,aAAa;oBACb,iBAAiB;oBACjB,SAAS,EAAE,IAAI;oBACf,UAAU,EAAE,UAAU,CAAC,UAAU,CAAC;oBAClC,gBAAgB,EAAE,gBAAgB,CAAC,UAAU,CAAC;iBAC/C,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,mDAAmD;QACrD,CAAC;IACH,CAAC;CACF"}
package/package.json ADDED
@@ -0,0 +1,51 @@
1
+ {
2
+ "name": "tokenlens-cli",
3
+ "version": "1.0.0",
4
+ "description": "AI token usage monitor that captures, analyzes, and reduces token consumption across coding agents and model providers.",
5
+ "type": "module",
6
+ "files": [
7
+ "dist",
8
+ "README.md",
9
+ "patches"
10
+ ],
11
+ "bin": {
12
+ "tokenlens": "dist/cli/commands.js",
13
+ "tl": "dist/cli/commands.js",
14
+ "tg": "dist/cli/commands.js"
15
+ },
16
+ "scripts": {
17
+ "dev": "tsx src/cli/commands.ts",
18
+ "cli": "tsx src/cli/commands.ts",
19
+ "dashboard": "tsx src/index.ts --dashboard-only",
20
+ "proxy": "tsx src/index.ts --proxy-only",
21
+ "build": "tsc",
22
+ "start": "node dist/cli/commands.js start",
23
+ "prepare": "npm run build",
24
+ "prepack": "npm run build",
25
+ "postinstall": "patch-package"
26
+ },
27
+ "keywords": [
28
+ "ai",
29
+ "tokens",
30
+ "usage",
31
+ "cli",
32
+ "observability",
33
+ "prompting"
34
+ ],
35
+ "dependencies": {
36
+ "better-sqlite3": "^11.7.0",
37
+ "chalk": "^5.4.1",
38
+ "cli-table3": "^0.6.5",
39
+ "commander": "^13.1.0",
40
+ "express": "^4.21.0",
41
+ "http-mitm-proxy": "^1.1.0"
42
+ },
43
+ "devDependencies": {
44
+ "@types/better-sqlite3": "^7.6.12",
45
+ "@types/express": "^5.0.0",
46
+ "@types/node": "^22.10.0",
47
+ "patch-package": "^8.0.1",
48
+ "tsx": "^4.19.0",
49
+ "typescript": "^5.7.0"
50
+ }
51
+ }
@@ -0,0 +1,13 @@
1
+ diff --git a/node_modules/http-mitm-proxy/dist/lib/proxy.js b/node_modules/http-mitm-proxy/dist/lib/proxy.js
2
+ index e896595..e94934d 100644
3
+ --- a/node_modules/http-mitm-proxy/dist/lib/proxy.js
4
+ +++ b/node_modules/http-mitm-proxy/dist/lib/proxy.js
5
+ @@ -361,7 +361,7 @@ class Proxy {
6
+ function makeConnection(port) {
7
+ const conn = net_1.default.connect({
8
+ port,
9
+ - host: "0.0.0.0",
10
+ + host: "127.0.0.1",
11
+ allowHalfOpen: true,
12
+ }, () => {
13
+ const connectKey = `${conn.localPort}:${conn.remotePort}`;