tinker-token-usage 0.1.1 → 0.1.3

package/dist/preload/index.mjs

@@ -1,13 +1,13 @@
-import { contextBridge as H } from "electron";
-import { loadDailyUsageData as P, loadSessionData as X } from "ccusage/data-loader";
-import { stat as G, readFile as J } from "node:fs/promises";
+import { contextBridge as B } from "electron";
+import { loadDailyUsageData as H, loadSessionData as P } from "ccusage/data-loader";
+import { stat as X, readFile as G } from "node:fs/promises";
 import * as g from "node:path";
-import * as K from "node:os";
-const V = "CODEX_HOME", W = g.join(K.homedir(), ".codex"), Y = "sessions";
+import * as J from "node:os";
+const K = "CODEX_HOME", V = g.join(J.homedir(), ".codex"), W = "sessions";
 function y(e) {
   return typeof e == "number" && Number.isFinite(e) ? e : 0;
 }
-function z(e) {
+function F(e) {
   if (e == null || typeof e != "object")
     return null;
   const t = e, o = y(t.input_tokens), l = y(
@@ -21,7 +21,7 @@ function z(e) {
     total_tokens: u > 0 ? u : o + s
   };
 }
-function $(e, t) {
+function Y(e, t) {
   return {
     input_tokens: Math.max(
       e.input_tokens - ((t == null ? void 0 : t.input_tokens) ?? 0),
@@ -45,7 +45,7 @@ function $(e, t) {
     )
   };
 }
-function q(e) {
+function $(e) {
   const t = e.total_tokens > 0 ? e.total_tokens : e.input_tokens + e.output_tokens, o = Math.min(e.cached_input_tokens, e.input_tokens);
   return {
     inputTokens: e.input_tokens,
@@ -55,7 +55,7 @@ function q(e) {
     totalTokens: t
   };
 }
-function L(e) {
+function A(e) {
   if (e == null || typeof e != "object")
     return;
   const t = e;
@@ -78,7 +78,7 @@ function L(e) {
       return o.model.trim();
   }
 }
-async function Q(e) {
+async function q(e) {
   try {
     const { glob: t } = await import("./index-CYePjqCs.mjs");
     return await t("**/*.jsonl", {
@@ -89,63 +89,65 @@ async function Q(e) {
     return [];
   }
 }
-async function Z() {
+async function Q() {
   var u;
-  const e = (u = process.env[V]) == null ? void 0 : u.trim(), t = e != null && e !== "" ? g.resolve(e) : W, o = g.join(t, Y);
+  const e = (u = process.env[K]) == null ? void 0 : u.trim(), t = e != null && e !== "" ? g.resolve(e) : V, o = g.join(t, W);
   try {
-    if (!(await G(o)).isDirectory())
+    if (!(await X(o)).isDirectory())
       return [];
   } catch {
     return [];
   }
-  const l = await Q(o), s = [], r = "gpt-5";
+  const l = await q(o), s = [], r = "gpt-5";
   for (const d of l) {
     const n = g.relative(o, d).split(g.sep).join("/").replace(/\.jsonl$/i, "");
     try {
-      const i = await J(d, "utf8");
-      let k = null, p, f = !1;
-      const S = i.split(/\r?\n/);
-      for (const m of S) {
-        const C = m.trim();
+      const i = await G(d, "utf8");
+      let k = null, p, m = !1;
+      const M = i.split(/\r?\n/);
+      for (const T of M) {
+        const C = T.trim();
         if (C === "") continue;
-        let T;
+        let _;
         try {
-          T = JSON.parse(C);
+          _ = JSON.parse(C);
         } catch {
           continue;
         }
-        if (T == null || typeof T != "object") continue;
-        const j = T, O = j.type, _ = j.payload, E = j.timestamp;
-        if (O === "turn_context") {
-          if (_ != null && typeof _ == "object") {
-            const A = L(_);
-            A != null && (p = A, f = !1);
+        if (_ == null || typeof _ != "object") continue;
+        const S = _, w = S.type, h = S.payload, O = S.timestamp;
+        if (w === "turn_context") {
+          if (h != null && typeof h == "object") {
+            const R = A(h);
+            R != null && (p = R, m = !1);
           }
           continue;
         }
-        if (O !== "event_msg" || _ == null || typeof _ != "object") continue;
-        const U = _;
-        if (U.type !== "token_count" || typeof E != "string") continue;
-        const b = U.info;
+        if (w !== "event_msg" || h == null || typeof h != "object") continue;
+        const j = h;
+        if (j.type !== "token_count" || typeof O != "string") continue;
+        const b = j.info;
         if (b == null || typeof b != "object") continue;
-        const F = b, N = z(F.last_token_usage), D = z(F.total_token_usage);
-        let I = N;
-        if (I == null && D != null && (I = $(D, k)), D != null && (k = D), I == null) continue;
-        const h = q(I);
-        if (h.inputTokens === 0 && h.cachedInputTokens === 0 && h.outputTokens === 0 && h.reasoningOutputTokens === 0)
+        const E = b, z = F(E.last_token_usage), D = F(E.total_token_usage);
+        let I = z;
+        if (I == null && D != null && (I = Y(D, k)), D != null && (k = D), I == null) continue;
+        const f = $(I);
+        if (f.inputTokens === 0 && f.cachedInputTokens === 0 && f.outputTokens === 0 && f.reasoningOutputTokens === 0)
           continue;
-        const B = Object.assign({}, U, { info: b }), M = L(B);
-        let w = !1;
-        M != null && (p = M, f = !1);
-        let x = M ?? p;
-        x == null ? (x = r, w = !0, p = x, f = !0) : M == null && f && (w = !0);
-        const R = {
+        const L = Object.assign({}, j, { info: b }), U = A(L);
+        U != null && (p = U, m = !1);
+        let x = U ?? p;
+        x == null && (x = r, p = x, m = !0);
+        const N = {
           sessionId: n,
-          timestamp: E,
+          timestamp: O,
           model: x,
-          ...h
+          inputTokens: f.inputTokens,
+          cachedInputTokens: f.cachedInputTokens,
+          outputTokens: f.outputTokens,
+          totalTokens: f.totalTokens
         };
-        w && (R.isFallbackModel = !0), s.push(R);
+        s.push(N);
       }
     } catch (i) {
       console.debug("Failed to read Codex session file", i);
@@ -156,31 +158,31 @@ async function Z() {
     (d, c) => new Date(d.timestamp).getTime() - new Date(c.timestamp).getTime()
   ), s;
 }
-const v = {
+const Z = {
   getUsage: async (e = "claude-code") => {
     try {
-      return e === "codex" ? await et() : await tt();
+      return e === "codex" ? await tt() : await v();
     } catch (t) {
       throw console.error("Failed to get token usage:", t), t;
     }
   }
 };
-async function tt() {
-  const e = await P({}), t = await X({}), o = /* @__PURE__ */ new Map();
+async function v() {
+  const e = await H({}), t = await P({}), o = /* @__PURE__ */ new Map();
   for (const n of t) {
     const i = n.lastActivity.split("T")[0];
     o.has(i) || o.set(i, /* @__PURE__ */ new Set()), o.get(i).add(n.sessionId);
   }
   let l = 0, s = 0, r = 0, u = 0;
   const d = /* @__PURE__ */ new Set(), c = e.map((n) => {
-    const i = n.inputTokens || 0, k = n.outputTokens || 0, p = n.cacheCreationTokens || 0, f = n.cacheReadTokens || 0, S = i + k + p + f, m = o.get(n.date), C = m ? m.size : 0;
-    return l += i, s += k, r += p, u += f, m && m.forEach((T) => d.add(T)), {
+    const i = n.inputTokens || 0, k = n.outputTokens || 0, p = n.cacheCreationTokens || 0, m = n.cacheReadTokens || 0, M = i + k + p + m, T = o.get(n.date), C = T ? T.size : 0;
+    return l += i, s += k, r += p, u += m, T && T.forEach((_) => d.add(_)), {
       date: n.date,
       inputTokens: i,
       outputTokens: k,
       cacheCreationTokens: p,
-      cacheReadTokens: f,
-      totalTokens: S,
+      cacheReadTokens: m,
+      totalTokens: M,
       sessionCount: C,
       modelsUsed: n.modelsUsed || []
     };
@@ -195,11 +197,10 @@ async function tt() {
       sessionCount: d.size
     },
     byDay: c.sort((n, i) => n.date.localeCompare(i.date))
-    // Sort by date ascending
   };
 }
-async function et() {
-  const e = await Z(), t = /* @__PURE__ */ new Map();
+async function tt() {
+  const e = await Q(), t = /* @__PURE__ */ new Map();
   for (const c of e) {
     const a = c.timestamp.split("T")[0];
     t.has(a) || t.set(a, {
@@ -222,12 +223,9 @@ async function et() {
     return o += n, l += a.outputTokens, s += i, a.sessions.forEach((k) => r.add(k)), {
       date: c,
       inputTokens: n,
-      // Non-cached input tokens (to match Claude Code)
       outputTokens: a.outputTokens,
       cacheCreationTokens: 0,
-      // Codex doesn't track cache creation separately
       cacheReadTokens: i,
-      // Cached input tokens
       totalTokens: a.totalTokens,
       sessionCount: a.sessions.size,
       modelsUsed: Array.from(a.models)
@@ -236,15 +234,13 @@ async function et() {
   return {
     total: {
       inputTokens: o,
-      // Non-cached input tokens
       outputTokens: l,
       cacheCreationTokens: 0,
       cacheReadTokens: s,
-      // Cached input tokens
       totalTokens: o + l + s,
       sessionCount: r.size
     },
     byDay: u
   };
 }
-H.exposeInMainWorld("tokenUsage", v);
+B.exposeInMainWorld("tokenUsage", Z);