tokmon 0.9.2 → 0.11.0

package/README.md

@@ -73,6 +73,7 @@ Then just run `tokmon`. Press `q` to quit.
 - **Today / This Week / This Month** — cost and token summaries
 - **Burn rate** — current $/hr
 - **Rate Limits** — real-time session (5h), weekly (7d), and Sonnet utilization with reset countdowns, fetched from Anthropic's OAuth API
+- **Peak / Off-Peak badge** — shown in the header, fetched from [promoclock.co](https://promoclock.co) (peak hours drain session limits faster)
 
 ### Table
 

package/dist/cli.js

@@ -8,7 +8,7 @@ import { MouseProvider } from "@zenobius/ink-mouse";
 import { readFile, writeFile, mkdir } from "fs/promises";
 import { join } from "path";
 import { homedir } from "os";
-var DEFAULTS = { interval: 2, billingInterval: 5, clearScreen: true };
+var DEFAULTS = { interval: 2, billingInterval: 5, clearScreen: true, timezone: null };
 function configDir() {
   if (process.platform === "win32") {
     return join(process.env.APPDATA ?? join(homedir(), "AppData", "Roaming"), "tokmon");
@@ -45,6 +45,117 @@ import { createReadStream } from "fs";
 import { createInterface } from "readline";
 import { join as join2 } from "path";
 import { homedir as homedir2 } from "os";
+
+// src/tz.ts
+function systemTimezone() {
+  try {
+    return Intl.DateTimeFormat().resolvedOptions().timeZone || "UTC";
+  } catch {
+    return "UTC";
+  }
+}
+function isValidTimezone(tz) {
+  try {
+    new Intl.DateTimeFormat("en-CA", { timeZone: tz });
+    return true;
+  } catch {
+    return false;
+  }
+}
+function resolveTimezone(cfg) {
+  if (!cfg) return systemTimezone();
+  return isValidTimezone(cfg) ? cfg : systemTimezone();
+}
+var dayFmtCache = /* @__PURE__ */ new Map();
+function dayFmt(tz) {
+  let f = dayFmtCache.get(tz);
+  if (!f) {
+    f = new Intl.DateTimeFormat("en-CA", {
+      timeZone: tz,
+      year: "numeric",
+      month: "2-digit",
+      day: "2-digit"
+    });
+    dayFmtCache.set(tz, f);
+  }
+  return f;
+}
+function dayKey(ts, tz) {
+  return dayFmt(tz).format(new Date(ts));
+}
+function monthKey(ts, tz) {
+  return dayKey(ts, tz).slice(0, 7);
+}
+var partsFmtCache = /* @__PURE__ */ new Map();
+function partsFmt(tz) {
+  let f = partsFmtCache.get(tz);
+  if (!f) {
+    f = new Intl.DateTimeFormat("en-US", {
+      timeZone: tz,
+      hourCycle: "h23",
+      year: "numeric",
+      month: "2-digit",
+      day: "2-digit",
+      hour: "2-digit",
+      minute: "2-digit",
+      second: "2-digit",
+      weekday: "short"
+    });
+    partsFmtCache.set(tz, f);
+  }
+  return f;
+}
+var WEEKDAY_MAP = {
+  Sun: 0,
+  Mon: 1,
+  Tue: 2,
+  Wed: 3,
+  Thu: 4,
+  Fri: 5,
+  Sat: 6
+};
+function tzParts(ts, tz) {
+  const parts = partsFmt(tz).formatToParts(new Date(ts));
+  const get = (t) => parts.find((p) => p.type === t)?.value ?? "";
+  return {
+    y: Number(get("year")),
+    m: Number(get("month")),
+    d: Number(get("day")),
+    hh: Number(get("hour")),
+    mm: Number(get("minute")),
+    ss: Number(get("second")),
+    weekday: WEEKDAY_MAP[get("weekday")] ?? 0
+  };
+}
+function instantFromTz(y, m, d, hh, mm, ss, tz) {
+  const guess = Date.UTC(y, m - 1, d, hh, mm, ss);
+  const r = tzParts(guess, tz);
+  const rendered = Date.UTC(r.y, r.m - 1, r.d, r.hh, r.mm, r.ss);
+  const offset = rendered - guess;
+  return guess - offset;
+}
+function startOfDay(ts, tz) {
+  const p = tzParts(ts, tz);
+  return instantFromTz(p.y, p.m, p.d, 0, 0, 0, tz);
+}
+function startOfMonth(ts, tz) {
+  const p = tzParts(ts, tz);
+  return instantFromTz(p.y, p.m, 1, 0, 0, 0, tz);
+}
+function startOfWeek(ts, tz) {
+  const p = tzParts(ts, tz);
+  const offset = p.weekday === 0 ? 6 : p.weekday - 1;
+  return instantFromTz(p.y, p.m, p.d - offset, 0, 0, 0, tz);
+}
+function monthsAgoStart(ts, months, tz) {
+  const p = tzParts(ts, tz);
+  return instantFromTz(p.y, p.m - months, 1, 0, 0, 0, tz);
+}
+function weekKey(ts, tz) {
+  return dayKey(startOfWeek(ts, tz), tz);
+}
+
+// src/data.ts
 var PRICING = {
   "claude-opus-4": { i: 5e-6, o: 25e-6, cc: 625e-8, cr: 5e-7 },
   "claude-sonnet-4": { i: 3e-6, o: 15e-6, cc: 375e-8, cr: 3e-7 },
@@ -207,24 +318,11 @@ function groupBy(entries, keyFn) {
   }
   return rows.sort((a, b) => a.label.localeCompare(b.label));
 }
-function isoWeekLabel(ts) {
-  const d = new Date(ts);
-  const day = d.getDay();
-  const mondayOffset = day === 0 ? 6 : day - 1;
-  const monday = new Date(d);
-  monday.setDate(d.getDate() - mondayOffset);
-  return monday.toISOString().slice(0, 10);
-}
-function monthLabel(ts) {
-  return new Date(ts).toISOString().slice(0, 7);
-}
-async function fetchDashboard() {
-  const d = /* @__PURE__ */ new Date();
-  const monthStart = new Date(d.getFullYear(), d.getMonth(), 1).getTime();
-  const todayStart = new Date(d.getFullYear(), d.getMonth(), d.getDate()).getTime();
-  const weekDay = d.getDay();
-  const weekStart = new Date(d.getFullYear(), d.getMonth(), d.getDate() - (weekDay === 0 ? 6 : weekDay - 1)).getTime();
+async function fetchDashboard(tz) {
   const now = Date.now();
+  const monthStart = startOfMonth(now, tz);
+  const todayStart = startOfDay(now, tz);
+  const weekStart = startOfWeek(now, tz);
   const entries = await loadEntries(monthStart);
   const todayEntries = entries.filter((e) => e.ts >= todayStart);
   let burnRate = 0;
@@ -245,14 +343,13 @@ async function fetchDashboard() {
     burnRate
   };
 }
-async function fetchTable() {
-  const d = /* @__PURE__ */ new Date();
-  const lookback = new Date(d.getFullYear(), d.getMonth() - 6, 1).getTime();
+async function fetchTable(tz) {
+  const lookback = monthsAgoStart(Date.now(), 6, tz);
   const entries = await loadEntries(lookback);
   return {
-    daily: groupBy(entries, (e) => new Date(e.ts).toISOString().slice(0, 10)),
-    weekly: groupBy(entries, (e) => isoWeekLabel(e.ts)),
-    monthly: groupBy(entries, (e) => monthLabel(e.ts))
+    daily: groupBy(entries, (e) => dayKey(e.ts, tz)),
+    weekly: groupBy(entries, (e) => weekKey(e.ts, tz)),
+    monthly: groupBy(entries, (e) => monthKey(e.ts, tz))
   };
 }
 
@@ -297,10 +394,18 @@ async function getAccessToken() {
   }
   return readCredentialsFile();
 }
-var EMPTY = { session: null, weekly: null, sonnet: null, extraUsage: null, error: null };
+var EMPTY = { session: null, weekly: null, sonnet: null, extraUsage: null, peak: null, error: null };
 async function fetchBilling() {
   const token = await getAccessToken();
   if (!token) return { ...EMPTY, error: "No OAuth token \u2014 run claude and log in" };
+  const [usageRes, peak] = await Promise.all([
+    fetchUsage(token),
+    fetchPeakStatus()
+  ]);
+  if ("error" in usageRes) return { ...EMPTY, peak, error: usageRes.error };
+  return { ...usageRes.data, peak, error: null };
+}
+async function fetchUsage(token) {
   try {
     const res = await fetch("https://api.anthropic.com/api/oauth/usage", {
       headers: {
@@ -310,31 +415,54 @@ async function fetchBilling() {
       },
       signal: AbortSignal.timeout(1e4)
     });
-    if (res.status === 429) return { ...EMPTY, error: "Rate limited \u2014 retrying next poll" };
-    if (res.status === 401) return { ...EMPTY, error: "Token expired \u2014 restart Claude Code" };
-    if (!res.ok) return { ...EMPTY, error: `API ${res.status}` };
+    if (res.status === 429) return { error: "Rate limited \u2014 retrying next poll" };
+    if (res.status === 401) return { error: "Token expired \u2014 restart Claude Code" };
+    if (!res.ok) return { error: `API ${res.status}` };
     const data = await res.json();
     return {
-      session: data.five_hour ? {
-        utilization: data.five_hour.utilization,
-        resetsAt: formatReset(data.five_hour.resets_at)
-      } : null,
-      weekly: data.seven_day ? {
-        utilization: data.seven_day.utilization,
-        resetsAt: formatReset(data.seven_day.resets_at)
-      } : null,
-      sonnet: data.seven_day_sonnet ? {
-        utilization: data.seven_day_sonnet.utilization,
-        resetsAt: formatReset(data.seven_day_sonnet.resets_at)
-      } : null,
-      extraUsage: data.extra_usage?.is_enabled ? {
-        limit: data.extra_usage.monthly_limit / 100,
-        used: data.extra_usage.used_credits / 100
-      } : null,
-      error: null
+      data: {
+        session: data.five_hour ? {
+          utilization: data.five_hour.utilization,
+          resetsAt: formatReset(data.five_hour.resets_at)
+        } : null,
+        weekly: data.seven_day ? {
+          utilization: data.seven_day.utilization,
+          resetsAt: formatReset(data.seven_day.resets_at)
+        } : null,
+        sonnet: data.seven_day_sonnet ? {
+          utilization: data.seven_day_sonnet.utilization,
+          resetsAt: formatReset(data.seven_day_sonnet.resets_at)
+        } : null,
+        extraUsage: data.extra_usage?.is_enabled ? {
+          limit: data.extra_usage.monthly_limit / 100,
+          used: data.extra_usage.used_credits / 100
+        } : null
+      }
     };
   } catch {
-    return { ...EMPTY, error: "Network error" };
+    return { error: "Network error" };
+  }
+}
+async function fetchPeakStatus() {
+  try {
+    const res = await fetch("https://promoclock.co/api/status", {
+      headers: { "Accept": "application/json", "User-Agent": "tokmon" },
+      signal: AbortSignal.timeout(3e3)
+    });
+    if (!res.ok) return null;
+    const data = await res.json();
+    let state;
+    if (data.isPeak === true || data.status === "peak") state = "peak";
+    else if (data.isWeekend === true || data.status === "weekend") state = "weekend";
+    else if (data.isOffPeak === true || data.status === "off_peak" || data.status === "off-peak") state = "off-peak";
+    else return null;
+    return {
+      state,
+      label: state === "peak" ? "Peak" : state === "weekend" ? "Weekend" : "Off-Peak",
+      minutesUntilChange: typeof data.minutesUntilChange === "number" ? data.minutesUntilChange : null
+    };
+  } catch {
+    return null;
   }
 }
 function formatReset(iso) {
@@ -362,11 +490,12 @@ function tokens(value) {
   if (value >= 1e3) return `${(value / 1e3).toFixed(1)}K`;
   return String(value);
 }
-function time(date) {
+function time(date, tz) {
   return date.toLocaleTimeString(void 0, {
     hour: "2-digit",
     minute: "2-digit",
-    second: "2-digit"
+    second: "2-digit",
+    timeZone: tz
   });
 }
 var SHORT_MONTHS = ["", "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"];
@@ -387,7 +516,8 @@ var VIEWS = ["Daily", "Weekly", "Monthly"];
 var SORTS = ["date \u2191", "date \u2193", "cost \u2191", "cost \u2193"];
 var SPINNER_FRAMES = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
 var MONTHS = ["", "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"];
-var DEFAULT_CONFIG = { interval: 2, billingInterval: 5, clearScreen: true };
+var DEFAULT_CONFIG = { interval: 2, billingInterval: 5, clearScreen: true, timezone: null };
+var SETTINGS_ROWS = 4;
 var IS_TTY = process.stdin.isTTY === true;
 function App({ interval: cliInterval }) {
   const [dashboard, setDashboard] = useState(null);
@@ -400,10 +530,12 @@ function App({ interval: cliInterval }) {
   const [view, setView] = useState(0);
   const [cursor, setCursor] = useState(0);
   const [expanded, setExpanded] = useState(-1);
-  const [sort, setSort] = useState(0);
+  const [sort, setSort] = useState(1);
   const [showSettings, setShowSettings] = useState(false);
   const [config2, setConfig] = useState(null);
   const [settingsCursor, setSettingsCursor] = useState(0);
+  const [tzEdit, setTzEdit] = useState(null);
+  const [tzError, setTzError] = useState(null);
   const tableLoadedOnce = useRef(false);
   const { stdout } = useStdout();
   const { exit } = useApp();
@@ -411,6 +543,7 @@ function App({ interval: cliInterval }) {
   const cols = stdout?.columns ?? 80;
   const interval2 = cliInterval ?? (config2?.interval ?? 2) * 1e3;
   const cfg = config2 ?? DEFAULT_CONFIG;
+  const tz = resolveTimezone(cfg.timezone);
   useEffect(() => {
     loadConfig().then((c) => {
       if (cliInterval) c = { ...c, interval: cliInterval / 1e3 };
@@ -421,7 +554,7 @@ function App({ interval: cliInterval }) {
     let active = true;
     const load = async () => {
       try {
-        const result = await fetchDashboard();
+        const result = await fetchDashboard(tz);
         if (active) {
           setDashboard(result);
           setError(null);
@@ -437,7 +570,7 @@ function App({ interval: cliInterval }) {
       active = false;
       clearInterval(id);
     };
-  }, [interval2]);
+  }, [interval2, tz]);
   const billingMs = cfg.billingInterval * 6e4;
   useEffect(() => {
     let active = true;
@@ -452,12 +585,16 @@ function App({ interval: cliInterval }) {
       clearInterval(id);
     };
   }, [billingMs]);
+  useEffect(() => {
+    tableLoadedOnce.current = false;
+    setTable(null);
+  }, [tz]);
   useEffect(() => {
     if (tab !== 1) return;
     if (tableLoadedOnce.current && table) return;
     let active = true;
     setTableLoading(true);
-    fetchTable().then((result) => {
+    fetchTable(tz).then((result) => {
       if (active) {
         setTable(result);
         setTableLoading(false);
@@ -469,13 +606,13 @@ function App({ interval: cliInterval }) {
     return () => {
       active = false;
     };
-  }, [tab]);
+  }, [tab, tz]);
   useEffect(() => {
     if (tab !== 1 || !tableLoadedOnce.current) return;
     let active = true;
     const id = setInterval(async () => {
       try {
-        const result = await fetchTable();
+        const result = await fetchTable(tz);
         if (active) setTable(result);
       } catch {
       }
@@ -484,7 +621,7 @@ function App({ interval: cliInterval }) {
       active = false;
       clearInterval(id);
     };
-  }, [tab, interval2]);
+  }, [tab, interval2, tz]);
   const resetView = useCallback(() => {
     setCursor(0);
     setExpanded(-1);
@@ -504,10 +641,47 @@ function App({ interval: cliInterval }) {
     };
   }, [tab]);
   useInput((input, key) => {
+    if (showSettings && tzEdit !== null) {
+      if (key.escape) {
+        setTzEdit(null);
+        setTzError(null);
+        return;
+      }
+      if (key.return) {
+        const val = tzEdit.trim();
+        if (val === "" || val.toLowerCase() === "system") {
+          updateConfig((c) => ({ ...c, timezone: null }));
+          setTzEdit(null);
+          setTzError(null);
+        } else if (isValidTimezone(val)) {
+          updateConfig((c) => ({ ...c, timezone: val }));
+          setTzEdit(null);
+          setTzError(null);
+        } else {
+          setTzError(`Invalid: ${val}`);
+        }
+        return;
+      }
+      if (key.backspace || key.delete) {
+        setTzEdit((s) => (s ?? "").slice(0, -1));
+        setTzError(null);
+        return;
+      }
+      if (input && !key.ctrl && !key.meta) {
+        setTzEdit((s) => (s ?? "") + input);
+        setTzError(null);
+        return;
+      }
+      return;
+    }
+    if (input === "q") {
+      exit();
+      return;
+    }
     if (showSettings) {
       if (key.escape || input === "s") setShowSettings(false);
       if (key.upArrow) setSettingsCursor((c) => Math.max(0, c - 1));
-      if (key.downArrow) setSettingsCursor((c) => Math.min(2, c + 1));
+      if (key.downArrow) setSettingsCursor((c) => Math.min(SETTINGS_ROWS - 1, c + 1));
       if (settingsCursor === 0) {
         if (key.leftArrow) updateConfig((c) => ({ ...c, interval: Math.max(1, c.interval - 1) }));
         if (key.rightArrow) updateConfig((c) => ({ ...c, interval: c.interval + 1 }));
@@ -519,10 +693,15 @@ function App({ interval: cliInterval }) {
       if (settingsCursor === 2 && (key.leftArrow || key.rightArrow || key.return)) {
         updateConfig((c) => ({ ...c, clearScreen: !c.clearScreen }));
       }
-      return;
-    }
-    if (input === "q") {
-      exit();
+      if (settingsCursor === 3) {
+        if (key.return) {
+          setTzEdit(cfg.timezone ?? "");
+          setTzError(null);
+        }
+        if (key.leftArrow || key.rightArrow) {
+          updateConfig((c) => ({ ...c, timezone: c.timezone === null ? systemTimezone() : null }));
+        }
+      }
       return;
     }
     if (input === "s") {
@@ -631,9 +810,15 @@ function App({ interval: cliInterval }) {
           "s"
         ] })
       ] }),
-      /* @__PURE__ */ jsx(Text, { dimColor: true, children: time(updated) })
+      /* @__PURE__ */ jsxs(Box, { children: [
+        billing?.peak && /* @__PURE__ */ jsxs(Fragment, { children: [
+          /* @__PURE__ */ jsx(PeakBadge, { peak: billing.peak }),
+          /* @__PURE__ */ jsx(Text, { dimColor: true, children: "  \xB7  " })
+        ] }),
+        /* @__PURE__ */ jsx(Text, { dimColor: true, children: time(updated, tz) })
+      ] })
     ] }),
-    showSettings ? /* @__PURE__ */ jsx(SettingsView, { config: cfg, cursor: settingsCursor }) : /* @__PURE__ */ jsxs(Fragment, { children: [
+    showSettings ? /* @__PURE__ */ jsx(SettingsView, { config: cfg, cursor: settingsCursor, tzEdit, tzError, resolvedTz: tz }) : /* @__PURE__ */ jsxs(Fragment, { children: [
       /* @__PURE__ */ jsxs(Box, { marginTop: 1, children: [
         /* @__PURE__ */ jsx(TabBar, { tabs: TABS, active: tab, onSelect: (i) => {
           setTab(i);
@@ -716,7 +901,9 @@ function sortRows(rows, sortIdx) {
       return sorted;
   }
 }
-function SettingsView({ config: config2, cursor }) {
+function SettingsView({ config: config2, cursor, tzEdit, tzError, resolvedTz }) {
+  const editing = tzEdit !== null;
+  const tzDisplay = config2.timezone === null ? `System (${resolvedTz})` : config2.timezone;
   return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", marginTop: 1, children: [
     /* @__PURE__ */ jsx(Text, { bold: true, children: "Settings" }),
     /* @__PURE__ */ jsx(Text, { dimColor: true, children: configLocation() }),
@@ -767,8 +954,25 @@ function SettingsView({ config: config2, cursor }) {
       /* @__PURE__ */ jsx(Text, { children: "Clear screen        " }),
       /* @__PURE__ */ jsx(Text, { bold: true, color: config2.clearScreen ? "green" : "red", children: config2.clearScreen ? "on" : "off" })
     ] }),
+    /* @__PURE__ */ jsxs(Box, { children: [
+      /* @__PURE__ */ jsxs(Text, { color: cursor === 3 ? "green" : void 0, children: [
+        cursor === 3 ? "\u25B8" : " ",
+        " "
+      ] }),
+      /* @__PURE__ */ jsx(Text, { children: "Timezone            " }),
+      editing ? /* @__PURE__ */ jsxs(Fragment, { children: [
+        /* @__PURE__ */ jsx(Text, { dimColor: true, children: "[" }),
+        /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: tzEdit }),
+        /* @__PURE__ */ jsx(Text, { color: "cyan", children: "_" }),
+        /* @__PURE__ */ jsx(Text, { dimColor: true, children: "]" })
+      ] }) : /* @__PURE__ */ jsx(Text, { bold: true, color: "yellow", children: tzDisplay })
+    ] }),
+    cursor === 3 && tzError && /* @__PURE__ */ jsxs(Text, { color: "red", children: [
+      "  ",
+      tzError
+    ] }),
     /* @__PURE__ */ jsx(Box, { height: 1 }),
-    /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2191\u2193 select  \u2190\u2192 adjust  s/Esc close" })
+    editing ? /* @__PURE__ */ jsx(Text, { dimColor: true, children: "type IANA name (e.g. Europe/London) \xB7 empty = System \xB7 Enter save \xB7 Esc cancel" }) : /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2191\u2193 select  \u2190\u2192 adjust  Enter edit tz  s/Esc close" })
   ] });
 }
 function DashboardView({ data, billing }) {
@@ -836,6 +1040,24 @@ function DashboardView({ data, billing }) {
     )
   ] });
 }
+function PeakBadge({ peak }) {
+  const color = peak.state === "peak" ? "red" : "green";
+  return /* @__PURE__ */ jsxs(Box, { children: [
+    /* @__PURE__ */ jsx(Text, { color, children: "\u25CF " }),
+    /* @__PURE__ */ jsx(Text, { bold: true, color, children: peak.label }),
+    peak.minutesUntilChange !== null && peak.minutesUntilChange > 0 && /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
+      " (",
+      fmtMinutes(peak.minutesUntilChange),
+      ")"
+    ] })
+  ] });
+}
+function fmtMinutes(mins) {
+  if (mins < 60) return `${mins}m`;
+  const h = Math.floor(mins / 60);
+  const m = mins % 60;
+  return m === 0 ? `${h}h` : `${h}h ${m}m`;
+}
 function LimitBar({ label, pct, resets }) {
   const width = 30;
   const filled = Math.round(pct / 100 * width);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tokmon",
-  "version": "0.9.2",
+  "version": "0.11.0",
   "description": "Terminal dashboard for Claude Code usage and costs",
   "type": "module",
   "bin": {