@tokz/cli 0.2.3 → 0.2.5

package/dist/statusline-EXFMQYFF.js

@@ -0,0 +1,166 @@
+#!/usr/bin/env node
+import {
+  buildBlocks,
+  burnRate,
+  fmtMs
+} from "./chunk-6Y6MKFHZ.js";
+import {
+  compact,
+  costUsd,
+  dayKey,
+  findTranscripts,
+  parseTranscript,
+  shortModel,
+  usd
+} from "./chunk-Z5W5QH2Z.js";
+
+// src/statusline.ts
+import { mkdir, readFile, stat, writeFile } from "fs/promises";
+import { homedir } from "os";
+import { join } from "path";
+import pc from "picocolors";
+var CONTEXT_WINDOW = 2e5;
+var LOOKBACK_MS = 6 * 60 * 60 * 1e3;
+var BURN_MODERATE = 2e3;
+var BURN_HIGH = 5e3;
+async function lastContextTokens(transcriptPath) {
+  try {
+    const lines = (await readFile(transcriptPath, "utf8")).split("\n");
+    for (let i = lines.length - 1; i >= 0; i--) {
+      if (!lines[i].includes('"usage"')) continue;
+      try {
+        const j = JSON.parse(lines[i]);
+        const u = j?.message?.usage;
+        if (u && typeof u.input_tokens === "number") {
+          return u.input_tokens + (u.cache_read_input_tokens ?? 0) + (u.cache_creation_input_tokens ?? 0);
+        }
+      } catch {
+      }
+    }
+  } catch {
+  }
+  return void 0;
+}
+async function statusline(input, now = Date.now(), home, opts = {}) {
+  const costSource = opts.costSource ?? "auto";
+  const files = await findTranscripts(void 0, home);
+  const recent = [];
+  await Promise.all(
+    files.map(async (f) => {
+      try {
+        if ((await stat(f)).mtimeMs >= now - Math.max(LOOKBACK_MS, msSinceMidnight(now) + 6e4))
+          recent.push(f);
+      } catch {
+      }
+    })
+  );
+  const seen = /* @__PURE__ */ new Map();
+  const seenTools = /* @__PURE__ */ new Set();
+  const events = [];
+  const sessions = await Promise.all(recent.map((f) => parseTranscript(f, seen, seenTools, events)));
+  const today = dayKey(now);
+  let todayCost = 0;
+  let calcSessionCost;
+  for (const s of sessions) {
+    for (const [model, u] of Object.entries(s.dailyUsage[today] ?? {})) todayCost += costUsd(u, model).total;
+    if (input.transcript_path && s.file === input.transcript_path) {
+      calcSessionCost = Object.entries(s.usageByModel).reduce((sum, [m, u]) => sum + costUsd(u, m).total, 0);
+    }
+  }
+  const ccCost = input.cost?.total_cost_usd;
+  const blocks = buildBlocks(events, { now });
+  const active = blocks.find((b) => b.active);
+  const rate = active ? burnRate(active, now) : void 0;
+  const modelName = input.model?.display_name ?? shortModel(input.model?.id ?? "?");
+  const effort = input.effort?.level;
+  const parts = [`\u{1F916} ${modelName}${effort ? ` (${effort})` : ""}`];
+  parts.push(`\u{1F4B0} ${moneySegment(costSource, ccCost, calcSessionCost, todayCost, active, now)}`);
+  if (rate) {
+    const tpm = rate.tokensPerMinute;
+    const label = `\u{1F525} ${usd(rate.costPerHour)}/hr`;
+    parts.push(tpm > BURN_HIGH ? pc.red(label) : tpm > BURN_MODERATE ? pc.yellow(label) : pc.green(label));
+  }
+  const ctx = await contextTokens(input);
+  if (ctx !== void 0) {
+    const window = input.context_window?.context_window_size ?? CONTEXT_WINDOW;
+    const p = Math.round(ctx / window * 100);
+    const colored = p >= 80 ? pc.red(`${p}%`) : p >= 50 ? pc.yellow(`${p}%`) : pc.green(`${p}%`);
+    parts.push(`\u{1F9E0} ${compact(ctx)} (${colored})`);
+  }
+  return parts.join(pc.dim(" | "));
+}
+function moneySegment(source, ccCost, calcCost, todayCost, active, now) {
+  let session;
+  if (source === "both") {
+    session = `(${usd(ccCost ?? 0)} cc / ${usd(calcCost ?? 0)} calc) session`;
+  } else {
+    const pick = source === "cc" ? ccCost : source === "calc" ? calcCost : ccCost ?? calcCost;
+    session = `${usd(pick ?? 0)} session`;
+  }
+  const block = active ? `${usd(active.costUsd)} block (${fmtMs(active.end - now)} left)` : "No active block";
+  return `${session} / ${usd(todayCost)} today / ${block}`;
+}
+async function contextTokens(input) {
+  const cw = input.context_window;
+  if (cw && typeof cw.total_input_tokens === "number") return cw.total_input_tokens;
+  return input.transcript_path ? lastContextTokens(input.transcript_path) : void 0;
+}
+function msSinceMidnight(now) {
+  const day = dayKey(now);
+  const midnight = Date.parse(`${day}T00:00:00Z`);
+  const diff = now - midnight;
+  return Math.min(Math.max(diff, 0), 36 * 60 * 60 * 1e3);
+}
+async function readStdin() {
+  let data = "";
+  for await (const chunk of process.stdin) data += chunk;
+  return data;
+}
+var STATUSLINE_COMMAND = "npx -y @tokz/cli statusline";
+function settingsPath(home = homedir()) {
+  return join(home, ".claude", "settings.json");
+}
+async function readSettings(file) {
+  let raw;
+  try {
+    raw = await readFile(file, "utf8");
+  } catch {
+    return {};
+  }
+  const parsed = JSON.parse(raw);
+  if (!parsed || typeof parsed !== "object" || Array.isArray(parsed))
+    throw new Error(`${file} is not a JSON object`);
+  return parsed;
+}
+async function enableStatusline(home) {
+  const file = settingsPath(home);
+  const settings = await readSettings(file);
+  const prev = settings.statusLine;
+  if (prev?.command === STATUSLINE_COMMAND) return `Already enabled in ${file}.`;
+  settings.statusLine = { type: "command", command: STATUSLINE_COMMAND };
+  await mkdir(join(file, ".."), { recursive: true });
+  await writeFile(file, `${JSON.stringify(settings, null, 2)}
+`);
+  const note = prev?.command ? ` (replaced previous statusLine: ${prev.command})` : "";
+  return `Statusline enabled \u2014 Claude Code will run "${STATUSLINE_COMMAND}".${note}
+Restart Claude Code (or start a new session) to see it.`;
+}
+async function disableStatusline(home) {
+  const file = settingsPath(home);
+  const settings = await readSettings(file);
+  const prev = settings.statusLine;
+  if (!prev) return `No statusLine configured in ${file}; nothing to do.`;
+  if (prev.command !== void 0 && !prev.command.includes("tokz") && !prev.command.includes("@tokz/cli"))
+    return `statusLine in ${file} is "${prev.command}" \u2014 not tokz, leaving it alone.`;
+  delete settings.statusLine;
+  await writeFile(file, `${JSON.stringify(settings, null, 2)}
+`);
+  return `Statusline disabled \u2014 removed from ${file}.`;
+}
+export {
+  STATUSLINE_COMMAND,
+  disableStatusline,
+  enableStatusline,
+  readStdin,
+  statusline
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tokz/cli",
-  "version": "0.2.3",
+  "version": "0.2.5",
   "description": "Audit where your coding agent's context window and API dollars go.",
   "type": "module",
   "license": "MIT",