opencode-usage-total 0.1.0 → 0.1.2

package/{usage-total.ts → dist/chunk-XNCJ32GU.js}

@@ -1,14 +1,12 @@
-import type { Plugin } from "@opencode-ai/plugin"
-
-export const UsageTotalPlugin: Plugin = async ({ client }) => {
+// usage-total.ts
+var UsageTotalPlugin = async ({ client }) => {
   await client.app.log({
     body: {
       service: "usage-total",
       level: "info",
-      message: "usage-total plugin loaded",
-    },
-  })
-
+      message: "usage-total plugin loaded"
+    }
+  });
   return {
     event: async ({ event }) => {
       if (event.type === "session.created") {
@@ -16,21 +14,24 @@ export const UsageTotalPlugin: Plugin = async ({ client }) => {
           body: {
             service: "usage-total",
             level: "info",
-            message: `session.created [${event.properties.info.id}]`,
-          },
-        })
+            message: `session.created [${event.properties.info.id}]`
+          }
+        });
       }
-
       if (event.type === "message.updated") {
-        const msg = event.properties.info
+        const msg = event.properties.info;
         void client.app.log({
           body: {
             service: "usage-total",
             level: "info",
-            message: `message.updated [${msg.id}] role=${msg.role}`,
-          },
-        })
+            message: `message.updated [${msg.id}] role=${msg.role}`
+          }
+        });
       }
-    },
-  }
-}
+    }
+  };
+};
+
+export {
+  UsageTotalPlugin
+};

package/dist/index.js

@@ -0,0 +1,287 @@
+import {
+  UsageTotalPlugin
+} from "./chunk-XNCJ32GU.js";
+
+// usage-total-tui.tsx
+import { createRoot, createSignal } from "solid-js";
+
+// package.json
+var version = "0.1.2";
+
+// usage-total-tui.tsx
+import { jsx, jsxs } from "@opentui/solid/jsx-runtime";
+var DEFAULT_AGENT = "primary";
+var UNKNOWN_ID = "?";
+function resolveRouteSessionID(api) {
+  const current = api.route.current;
+  if (current.name === "session") {
+    const sid = current.params?.sessionID;
+    return typeof sid === "string" ? sid : void 0;
+  }
+  return void 0;
+}
+function modelTokens(m) {
+  return m.tokensInput + m.tokensOutput + m.tokensReasoning;
+}
+function safeNum(value) {
+  const n = typeof value === "number" ? value : Number(value);
+  return Number.isFinite(n) ? n : 0;
+}
+function roundCost(n) {
+  return safeNum(Number(n.toFixed(6)));
+}
+function fmtTokens(n) {
+  if (!Number.isFinite(n) || n === 0) return "0";
+  if (n >= 1e6) return `${(n / 1e6).toFixed(1)}M`;
+  if (n >= 1e3) return `${(n / 1e3).toFixed(1)}k`;
+  return String(Math.round(n));
+}
+function fmtCost(n) {
+  if (!Number.isFinite(n) || n === 0) return "";
+  if (n < 0.01) return `$${n.toFixed(4)}`;
+  return `$${n.toFixed(2)}`;
+}
+var initialized = false;
+var tui = async (api) => {
+  if (initialized) {
+    api.ui?.toast?.({
+      message: "usage-total TUI already loaded; skipping re-init",
+      variant: "warning"
+    });
+    return;
+  }
+  initialized = true;
+  api.ui?.toast?.({ message: "usage-total TUI loaded", variant: "info" });
+  let unsub;
+  let keymapDispose;
+  let solidDispose;
+  let flushPending;
+  const cleanup = () => {
+    try {
+      flushPending?.();
+    } catch {
+    }
+    try {
+      unsub?.();
+    } catch {
+    }
+    try {
+      keymapDispose?.();
+    } catch {
+    }
+    try {
+      solidDispose?.();
+    } catch {
+    }
+    initialized = false;
+  };
+  try {
+    if (api.lifecycle?.onDispose) {
+      api.lifecycle.onDispose(cleanup);
+    }
+    createRoot((dispose) => {
+      solidDispose = dispose;
+      const [modelState, setModelState] = createSignal({});
+      const EXPANDED_KV_KEY = "usage-total.sidebar.expanded";
+      const [expanded, setExpanded] = createSignal(
+        api.kv?.get?.(EXPANDED_KV_KEY, true) !== false
+      );
+      const TOGGLE_CMD = "usage-total.toggle-section";
+      keymapDispose = api.keymap?.registerLayer ? api.keymap.registerLayer({
+        commands: [
+          {
+            name: TOGGLE_CMD,
+            title: "Usage: Toggle models section",
+            description: "Collapse or expand the models list in the sidebar",
+            run: () => {
+              const next = !expanded();
+              setExpanded(next);
+              api.kv?.set?.(EXPANDED_KV_KEY, next);
+            }
+          }
+        ],
+        bindings: [{ key: "alt+m", cmd: TOGGLE_CMD }]
+      }) : void 0;
+      const loadedSessions = /* @__PURE__ */ new Set();
+      function kvKey(sessionID) {
+        return `usage-total:models:${sessionID}`;
+      }
+      const SAVE_DEBOUNCE_MS = 500;
+      const dirtySessions = /* @__PURE__ */ new Set();
+      let saveTimer;
+      function flushDirtySessions() {
+        if (saveTimer) {
+          clearTimeout(saveTimer);
+          saveTimer = void 0;
+        }
+        for (const sid of dirtySessions) {
+          const models = modelState()[sid];
+          if (models && models.length > 0) {
+            api.kv?.set?.(kvKey(sid), models);
+          }
+        }
+        dirtySessions.clear();
+      }
+      flushPending = flushDirtySessions;
+      function scheduleSave(sessionID) {
+        dirtySessions.add(sessionID);
+        if (saveTimer) clearTimeout(saveTimer);
+        saveTimer = setTimeout(flushDirtySessions, SAVE_DEBOUNCE_MS);
+      }
+      function loadSession(sessionID) {
+        if (loadedSessions.has(sessionID)) return;
+        loadedSessions.add(sessionID);
+        const saved = api.kv?.get?.(kvKey(sessionID));
+        if (saved && saved.length > 0) {
+          setModelState((current) => ({ ...current, [sessionID]: saved }));
+        }
+      }
+      function upsertModel(sessionID, entry, cost, tokens) {
+        const dedupeKey = `${entry.provider}/${entry.model}/${entry.agent}`;
+        const current = modelState();
+        const sessionModels = [...current[sessionID] ?? []];
+        const existingIdx = sessionModels.findIndex(
+          (m) => `${m.provider}/${m.model}/${m.agent}` === dedupeKey
+        );
+        const safeCost = safeNum(cost);
+        const safeInput = safeNum(tokens.input);
+        const safeOutput = safeNum(tokens.output);
+        const safeReasoning = safeNum(tokens.reasoning);
+        if (existingIdx >= 0) {
+          const existing = sessionModels[existingIdx];
+          sessionModels[existingIdx] = {
+            ...existing,
+            cost: roundCost(existing.cost + safeCost),
+            tokensInput: safeNum(existing.tokensInput + safeInput),
+            tokensOutput: safeNum(existing.tokensOutput + safeOutput),
+            tokensReasoning: safeNum(existing.tokensReasoning + safeReasoning),
+            messageCount: existing.messageCount + 1
+          };
+        } else {
+          sessionModels.push({
+            ...entry,
+            cost: safeCost,
+            tokensInput: safeInput,
+            tokensOutput: safeOutput,
+            tokensReasoning: safeReasoning,
+            messageCount: 1
+          });
+        }
+        setModelState({
+          ...current,
+          [sessionID]: sessionModels
+        });
+        scheduleSave(sessionID);
+        return existingIdx < 0;
+      }
+      function trackModel(eventSessionID, entry, cost, tokens) {
+        const isNew = upsertModel(eventSessionID, entry, cost, tokens);
+        if (isNew) {
+          api.ui?.toast?.({
+            message: `${entry.agent}: ${entry.model}`,
+            variant: "success"
+          });
+        }
+        const session = api.state?.session?.get?.(eventSessionID);
+        if (session?.parentID && session.parentID !== eventSessionID) {
+          upsertModel(session.parentID, entry, cost, tokens);
+        }
+      }
+      unsub = api.event?.on?.("message.updated", (event) => {
+        const info = event?.properties?.info;
+        if (!info) return;
+        const eventSessionID = info.sessionID;
+        if (!eventSessionID) return;
+        let provider;
+        let model;
+        let agent;
+        let cost = 0;
+        let tokens = {};
+        if (info.role === "user") {
+          const mdl = info.model;
+          provider = mdl?.providerID ?? UNKNOWN_ID;
+          model = mdl?.modelID ?? UNKNOWN_ID;
+          agent = info.agent ?? DEFAULT_AGENT;
+        } else if (info.role === "assistant") {
+          provider = info.providerID ?? UNKNOWN_ID;
+          model = info.modelID ?? UNKNOWN_ID;
+          agent = info.agent ?? info.mode ?? DEFAULT_AGENT;
+          cost = safeNum(info.cost);
+          tokens = {
+            input: safeNum(info.tokens?.input),
+            output: safeNum(info.tokens?.output),
+            reasoning: safeNum(info.tokens?.reasoning)
+          };
+        } else {
+          return;
+        }
+        trackModel(eventSessionID, { provider, model, agent }, cost, tokens);
+      });
+      api.slots?.register?.({
+        order: 200,
+        slots: {
+          sidebar_content(ctx, props) {
+            const sessionID = props.session_id ?? resolveRouteSessionID(api) ?? "";
+            if (sessionID) loadSession(sessionID);
+            const models = sessionID ? modelState()[sessionID] ?? [] : [];
+            if (!sessionID || models.length === 0) {
+              return /* @__PURE__ */ jsxs("box", { flexDirection: "column", children: [
+                /* @__PURE__ */ jsx("text", { fg: ctx.theme.current.text, children: "\u{1F9E0} Models" }),
+                /* @__PURE__ */ jsx("text", { fg: ctx.theme.current.textMuted, children: sessionID ? "waiting for messages..." : "open a session to track models" })
+              ] });
+            }
+            const totalCost = models.reduce((sum, m) => sum + m.cost, 0);
+            const totalTokens = models.reduce(
+              (sum, m) => sum + modelTokens(m),
+              0
+            );
+            return /* @__PURE__ */ jsxs("box", { flexDirection: "column", children: [
+              /* @__PURE__ */ jsxs(
+                "box",
+                {
+                  flexDirection: "row",
+                  justifyContent: "space-between",
+                  children: [
+                    /* @__PURE__ */ jsxs("box", { flexDirection: "row", children: [
+                      /* @__PURE__ */ jsxs("text", { fg: ctx.theme.current.text, children: [
+                        expanded() ? "\u25BC" : "\u25B6",
+                        " \u{1F9E0} Models"
+                      ] }),
+                      /* @__PURE__ */ jsxs("text", { fg: ctx.theme.current.textMuted, children: [
+                        " ",
+                        version
+                      ] })
+                    ] }),
+                    /* @__PURE__ */ jsx("text", { fg: ctx.theme.current.text, children: totalCost > 0 ? `${fmtCost(totalCost)} \xB7 ${fmtTokens(totalTokens)}` : fmtTokens(totalTokens) })
+                  ]
+                }
+              ),
+              expanded() && models.map((m) => /* @__PURE__ */ jsxs("box", { flexDirection: "column", children: [
+                /* @__PURE__ */ jsxs("text", { fg: ctx.theme.current.text, children: [
+                  m.agent,
+                  ":"
+                ] }),
+                /* @__PURE__ */ jsxs("box", { flexDirection: "row", justifyContent: "space-between", children: [
+                  /* @__PURE__ */ jsx("text", { fg: ctx.theme.current.text, children: "  " + m.model }),
+                  /* @__PURE__ */ jsx("text", { fg: ctx.theme.current.textMuted, children: m.cost > 0 ? `${fmtCost(m.cost)}${modelTokens(m) > 0 ? ` \xB7 ${fmtTokens(modelTokens(m))}` : ""}` : modelTokens(m) > 0 ? fmtTokens(modelTokens(m)) : "-" })
+                ] })
+              ] }))
+            ] });
+          }
+        }
+      });
+    });
+  } catch (err) {
+    cleanup();
+    api.ui?.toast?.({
+      message: "usage-total TUI failed to initialize",
+      variant: "error"
+    });
+    throw err;
+  }
+};
+var usage_total_tui_default = { id: "usage-total", tui };
+export {
+  UsageTotalPlugin,
+  usage_total_tui_default as default
+};

package/dist/usage-total.js

@@ -0,0 +1,6 @@
+import {
+  UsageTotalPlugin
+} from "./chunk-XNCJ32GU.js";
+export {
+  UsageTotalPlugin
+};

package/package.json

@@ -1,17 +1,15 @@
 {
   "name": "opencode-usage-total",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Track model usage, tokens, and costs per agent in the OpenCode TUI sidebar",
   "type": "module",
-  "main": "index.ts",
+  "main": "dist/index.js",
   "exports": {
-    ".": "./index.ts",
-    "./runtime": "./usage-total.ts"
+    ".": "./dist/index.js",
+    "./runtime": "./dist/usage-total.js"
   },
   "files": [
-    "index.ts",
-    "usage-total.ts",
-    "usage-total-tui.tsx",
+    "dist",
     "image.png",
     "README.md",
     "README.en.md"
@@ -30,11 +28,16 @@
     "type": "git",
     "url": "git+https://github.com/AlonsoSG0/opencode-usage-total.git"
   },
+  "scripts": {
+    "build": "tsup",
+    "prepublishOnly": "npm run build"
+  },
   "devDependencies": {
     "@opencode-ai/plugin": "^1.17.8",
     "@opentui/core": "^0.4.1",
     "@opentui/keymap": "^0.4.1",
     "@opentui/solid": "^0.4.1",
-    "solid-js": "^1.9.12"
+    "solid-js": "^1.9.12",
+    "tsup": "^8.5.1"
   }
 }

package/index.ts

@@ -1,2 +0,0 @@
-export { default } from "./usage-total-tui"
-export { UsageTotalPlugin } from "./usage-total"

package/usage-total-tui.tsx

@@ -1,316 +0,0 @@
-// @ts-nocheck
-/** @jsxImportSource @opentui/solid */
-import type { TuiPlugin } from "@opencode-ai/plugin/tui"
-import { createRoot, createSignal } from "solid-js"
-
-// ---- Constants ----
-const DEFAULT_AGENT = "primary"
-const UNKNOWN_ID = "?"
-
-interface ModelEntry {
-  provider: string
-  model: string
-  agent: string
-  cost: number
-  tokensInput: number
-  tokensOutput: number
-  tokensReasoning: number
-  messageCount: number
-}
-
-type ModelEntryKey = Omit<
-  ModelEntry,
-  "cost" | "tokensInput" | "tokensOutput" | "tokensReasoning" | "messageCount"
->
-
-// ---- Helpers ----
-function resolveRouteSessionID(api: any): string | undefined {
-  return (
-    api.route.current.name === "session" &&
-    typeof api.route.current.params?.sessionID === "string"
-      ? api.route.current.params.sessionID
-      : undefined
-  )
-}
-
-function modelTokens(m: ModelEntry): number {
-  return m.tokensInput + m.tokensOutput + m.tokensReasoning
-}
-
-function safeNum(value: unknown): number {
-  const n = typeof value === "number" ? value : Number(value)
-  return Number.isFinite(n) ? n : 0
-}
-
-function fmtTokens(n: number): string {
-  if (!Number.isFinite(n) || n === 0) return "0"
-  if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`
-  if (n >= 1_000) return `${(n / 1_000).toFixed(1)}k`
-  return String(Math.round(n))
-}
-
-function fmtCost(n: number): string {
-  if (!Number.isFinite(n) || n === 0) return ""
-  if (n < 0.01) return `$${n.toFixed(4)}`
-  return `$${n.toFixed(2)}`
-}
-
-// ---- Plugin ----
-const tui: TuiPlugin = async (api) => {
-  api.ui.toast({ message: "usage-total TUI loaded", variant: "info" })
-
-  createRoot((dispose) => {
-    const [modelState, setModelState] = createSignal<
-      Record<string, ModelEntry[]>
-    >({})
-
-    // Collapse/expand toggle persisted via KV
-    const EXPANDED_KV_KEY = "usage-total.sidebar.expanded"
-    const [expanded, setExpanded] = createSignal(
-      api.kv.get(EXPANDED_KV_KEY, true) !== false,
-    )
-
-    // Register keyboard shortcut to toggle section
-    const TOGGLE_CMD = "usage-total.toggle-section"
-    const keymapDispose = api.keymap?.registerLayer
-      ? api.keymap.registerLayer({
-          commands: [
-            {
-              name: TOGGLE_CMD,
-              title: "Usage: Toggle models section",
-              description: "Collapse or expand the models list in the sidebar",
-              run: () => {
-                const next = !expanded()
-                setExpanded(next)
-                api.kv.set(EXPANDED_KV_KEY, next)
-              },
-            },
-          ],
-          bindings: [{ key: "alt+m", cmd: TOGGLE_CMD }],
-        })
-      : undefined
-
-    // KV persistence – loadedSessions grows with visited sessions,
-    // bounded by total session count (not a true leak).
-    const loadedSessions = new Set<string>()
-
-    function kvKey(sessionID: string) {
-      return `usage-total:models:${sessionID}`
-    }
-
-    function saveSession(sessionID: string) {
-      const models = modelState()[sessionID]
-      if (models && models.length > 0) {
-        api.kv.set(kvKey(sessionID), models)
-      }
-    }
-
-    function loadSession(sessionID: string) {
-      if (loadedSessions.has(sessionID)) return
-      loadedSessions.add(sessionID)
-      const saved = api.kv.get<ModelEntry[]>(kvKey(sessionID))
-      if (saved && saved.length > 0) {
-        setModelState((current) => ({ ...current, [sessionID]: saved }))
-      }
-    }
-
-    function upsertModel(
-      sessionID: string,
-      entry: ModelEntryKey,
-      cost: number,
-      tokens: { input?: number; output?: number; reasoning?: number },
-    ) {
-      const dedupeKey = `${entry.provider}/${entry.model}/${entry.agent}`
-      const current = modelState()
-      const sessionModels = [...(current[sessionID] ?? [])]
-      const existingIdx = sessionModels.findIndex(
-        (m) => `${m.provider}/${m.model}/${m.agent}` === dedupeKey,
-      )
-
-      // Guard against NaN/Infinity that would corrupt accumulators and KV
-      const safeCost = safeNum(cost)
-      const safeInput = safeNum(tokens.input)
-      const safeOutput = safeNum(tokens.output)
-      const safeReasoning = safeNum(tokens.reasoning)
-
-      if (existingIdx >= 0) {
-        const existing = sessionModels[existingIdx]
-        sessionModels[existingIdx] = {
-          ...existing,
-          cost: safeNum(existing.cost + safeCost),
-          tokensInput: safeNum(existing.tokensInput + safeInput),
-          tokensOutput: safeNum(existing.tokensOutput + safeOutput),
-          tokensReasoning: safeNum(existing.tokensReasoning + safeReasoning),
-          messageCount: existing.messageCount + 1,
-        }
-      } else {
-        sessionModels.push({
-          ...entry,
-          cost: safeCost,
-          tokensInput: safeInput,
-          tokensOutput: safeOutput,
-          tokensReasoning: safeReasoning,
-          messageCount: 1,
-        })
-      }
-
-      setModelState({
-        ...current,
-        [sessionID]: sessionModels,
-      })
-
-      saveSession(sessionID)
-
-      return existingIdx < 0
-    }
-
-    function trackModel(
-      eventSessionID: string,
-      entry: ModelEntryKey,
-      cost: number,
-      tokens: { input?: number; output?: number; reasoning?: number },
-    ) {
-      const isNew = upsertModel(eventSessionID, entry, cost, tokens)
-      if (isNew) {
-        api.ui.toast({
-          message: `${entry.agent}: ${entry.model}`,
-          variant: "success",
-        })
-      }
-
-      // Attribute sub-agent models to the parent session so they
-      // appear in the main session sidebar.
-      const session = api.state.session.get(eventSessionID)
-      if (session?.parentID && session.parentID !== eventSessionID) {
-        upsertModel(session.parentID, entry, cost, tokens)
-      }
-    }
-
-    const unsub = api.event.on("message.updated", (event) => {
-      const info = event?.properties?.info
-      if (!info) return
-
-      const eventSessionID = info.sessionID
-      if (!eventSessionID) return
-
-      let provider: string
-      let model: string
-      let agent: string
-      let cost = 0
-      let tokens: { input?: number; output?: number; reasoning?: number } = {}
-
-      if (info.role === "user") {
-        const mdl = info.model
-        provider = mdl?.providerID ?? UNKNOWN_ID
-        model = mdl?.modelID ?? UNKNOWN_ID
-        agent = info.agent ?? DEFAULT_AGENT
-      } else if (info.role === "assistant") {
-        provider = info.providerID ?? UNKNOWN_ID
-        model = info.modelID ?? UNKNOWN_ID
-        agent = info.agent ?? info.mode ?? DEFAULT_AGENT
-        cost = safeNum(info.cost)
-        tokens = {
-          input: safeNum(info.tokens?.input),
-          output: safeNum(info.tokens?.output),
-          reasoning: safeNum(info.tokens?.reasoning),
-        }
-      } else {
-        return
-      }
-
-      trackModel(eventSessionID, { provider, model, agent }, cost, tokens)
-    })
-
-    api.slots.register({
-      id: "usage-total",
-      order: 200,
-      slots: {
-        sidebar_content(ctx) {
-          const sessionID =
-            ctx.session_id ?? resolveRouteSessionID(api) ?? ""
-          if (sessionID) loadSession(sessionID)
-          const models = sessionID ? (modelState()[sessionID] ?? []) : []
-
-          if (!sessionID || models.length === 0) {
-            return (
-              <box
-                flexDirection="column"
-                padding={{ left: 1, right: 1, top: 1 }}
-              >
-                <text fg={ctx.theme.current.text} bold>
-                  🧠 Models
-                </text>
-                <text fg={ctx.theme.current.textMuted}>
-                  {sessionID
-                    ? "waiting for messages..."
-                    : "open a session to track models"}
-                </text>
-              </box>
-            )
-          }
-
-          const totalCost = models.reduce((sum, m) => sum + m.cost, 0)
-          const totalTokens = models.reduce(
-            (sum, m) => sum + modelTokens(m),
-            0,
-          )
-
-          return (
-            <box
-              flexDirection="column"
-              padding={{ left: 1, right: 1, top: 1 }}
-            >
-              <box
-                flexDirection="row"
-                justifyContent="space-between"
-              >
-                <box flexDirection="row">
-                  <text
-                    fg={ctx.theme.current.text}
-                    bold
-                  >
-                    {expanded() ? "▼" : "▶"} 🧠 Models
-                  </text>
-                  <text fg={ctx.theme.current.textMuted}> 0.1.0</text>
-                </box>
-                <text fg={ctx.theme.current.text}>
-                  {totalCost > 0
-                    ? `${fmtCost(totalCost)} · ${fmtTokens(totalTokens)}`
-                    : fmtTokens(totalTokens)}
-                </text>
-              </box>
-              {expanded() &&
-                models.map((m, i) => (
-                <box key={i} flexDirection="column">
-                  <text fg={ctx.theme.current.text}>
-                    {m.agent}:
-                  </text>
-                  <box flexDirection="row" justifyContent="space-between">
-                    <text fg={ctx.theme.current.text}>
-                      {"  " + m.model}
-                    </text>
-                    <text fg={ctx.theme.current.textMuted}>
-                      {m.cost > 0
-                        ? `${fmtCost(m.cost)}${modelTokens(m) > 0 ? ` · ${fmtTokens(modelTokens(m))}` : ""}`
-                        : modelTokens(m) > 0
-                          ? fmtTokens(modelTokens(m))
-                          : "-"}
-                    </text>
-                  </box>
-                </box>
-              ))}
-            </box>
-          )
-        },
-      },
-    })
-
-    api.lifecycle.onDispose(() => {
-      unsub()
-      keymapDispose?.()
-      dispose()
-    })
-  })
-}
-
-export default { id: "usage-total", tui }