@bacnh85/pi-sub 0.1.38 → 0.1.40

package/CHANGELOG.md

@@ -1,5 +1,36 @@
 # Changelog
 
+## 0.1.40 (2026-09-14)
+
+### Tests
+
+- Added unit test coverage for the three untested parsers: the OmniRoute
+  om-usage free-text report (`parseOmniUsageText` — four windows, section
+  switching, out-of-range percentages, disabled/no-cache text), the Command
+  Code `/alpha/billing/credits` window mapper (used/cap → remaining%, epoch-ms
+  resetAt, over-cap clamp, bad-window bail), and the `.env.local` parser
+  (extracted as exported `parseEnvText`; `export ` prefix, quoted values,
+  comments, CRLF, `#`-in-value). No behavior changes.
+  One edge case differs from the inline loop it replaces: duplicate keys in a
+  single .env file now resolve last-wins (dotenv convention) instead of
+  first-wins.
+
+## 0.1.39 (2026-09-12)
+
+### Fixed
+
+- **Guarded footer render against an uninitialized theme proxy** (pi-budget
+  parity): `renderSubscriptionLine` dereferenced `ctx.ui.theme.fg` unguarded —
+  if the theme isn't ready yet the throw escapes as a rejected promise and can
+  exit pi (the same unhandledRejection class 0.1.37/0.1.38 fixed elsewhere).
+  The footer is now best-effort: skipped when the theme isn't available.
+- **Finite-cost guard on `message_end` accumulation** (pi-budget parity): a
+  string or NaN `cost.total` previously hit `+=` directly — a string cost
+  concatenated onto the accumulator and garbled every subsequent footer.
+  Costs are now coerced with `Number()` and only finite positive values
+  accumulate.
+- README intro: Router (pi-router) listed among supported providers.
+
 ## 0.1.38 (2026-09-07)
 
 ### Fixed

package/README.md

@@ -2,7 +2,7 @@
 
 Pi extension that shows subscription usage for the currently selected supported model provider.
 
- Supports OpenAI Codex (`openai-codex`) with live usage windows from ChatGPT's usage endpoint, OpenCode Go (`opencode-go`) with session cost tracking, and Z.ai GLM Coding Plan — both the international (`zai`) and China (`zai-coding-cn`, `open.bigmodel.cn`) endpoints — with quota monitoring. Also tracks Command Code (`commandcode`) 5-hour/weekly windows and monthly credit balance. Displays a subscription footer status after Pi's built-in status/token usage line.
+ Supports OpenAI Codex (`openai-codex`) with live usage windows from ChatGPT's usage endpoint, OpenCode Go (`opencode-go`) with session cost tracking, and Z.ai GLM Coding Plan — both the international (`zai`) and China (`zai-coding-cn`, `open.bigmodel.cn`) endpoints — with quota monitoring. Also tracks Router (pi-router, `router` provider) with response-speed tracking and optional OmniRoute quota windows, and Command Code (`commandcode`) 5-hour/weekly windows and monthly credit balance. Displays a subscription footer status after Pi's built-in status/token usage line.
 
 ## Install
 

package/extensions/index.ts

@@ -4,6 +4,22 @@ import fs from "node:fs";
 import os from "node:os";
 import path from "node:path";
 
+/** Parse .env-style text into KEY→VALUE entries: `export ` prefix allowed,
+ *  single/double quotes stripped, comment/blank/non-assignment lines ignored.
+ *  No inline-comment stripping (a `#` in the value stays part of the value).
+ *  Exported for tests. */
+export function parseEnvText(text: string): Record<string, string> {
+  const out: Record<string, string> = {};
+  for (const line of text.split(/\r?\n/)) {
+    const m = line.match(/^\s*(?:export\s+)?([A-Za-z_][A-Za-z0-9_]*)\s*=\s*(.*)\s*$/);
+    if (!m) continue;
+    let v = m[2].trim();
+    if ((v.startsWith('"') && v.endsWith('"')) || (v.startsWith("'") && v.endsWith("'"))) v = v.slice(1, -1);
+    out[m[1]] = v;
+  }
+  return out;
+}
+
 /** Pi config dirs + .env.local/.env discovery (pi-munin convention, stdlib parse). */
 function loadEnvFiles(): void {
   const dirs = process.env.PI_CODING_AGENT_DIR
@@ -14,12 +30,8 @@ function loadEnvFiles(): void {
   for (const file of candidates) {
     try {
       const text = fs.readFileSync(file, "utf8");
-      for (const line of text.split(/\r?\n/)) {
-        const m = line.match(/^\s*(?:export\s+)?([A-Za-z_][A-Za-z0-9_]*)\s*=\s*(.*)\s*$/);
-        if (!m) continue;
-        let v = m[2].trim();
-        if ((v.startsWith('"') && v.endsWith('"')) || (v.startsWith("'") && v.endsWith("'"))) v = v.slice(1, -1);
-        if (process.env[m[1]] === undefined) process.env[m[1]] = v;
+      for (const [key, value] of Object.entries(parseEnvText(text))) {
+        if (process.env[key] === undefined) process.env[key] = value;
       }
     } catch { /* optional file */ }
   }
@@ -437,7 +449,8 @@ export function parseOmniUsageText(text: string): {
   return out;
 }
 
-// ponytail: runnable self-check (pi-sub has no test runner — pack gate only)
+// ponytail: runnable self-check (pack gate; extensions/test covers the same
+// parser paths plus the adapters the self-check doesn't)
 if (process.env.PI_SUB_SELF_CHECK === "1") {
   const sample = [
     "Personal quota", "Daily", "80% left", "⏱ reset in 15h 0m", "",
@@ -850,7 +863,7 @@ interface CommandCodeCreditsApiResponse {
 
 /** Map a Command Code USD window (used/cap in dollars, resetAt in ms) into
  *  the shared UsageWindow shape (remaining%, reset labels). */
-function commandCodeWindowToUsageWindow(window: CommandCodeWindowApi | undefined): UsageWindow | undefined {
+export function commandCodeWindowToUsageWindow(window: CommandCodeWindowApi | undefined): UsageWindow | undefined {
   if (!window || typeof window.used !== "number" || typeof window.cap !== "number" || window.cap <= 0) return undefined;
   const usedPct = Math.round((window.used / window.cap) * 100);
   const percent = Math.min(100, usedPct);
@@ -1103,6 +1116,9 @@ export function renderSubscriptionLine(state: State): void {
   const ctx = state.ctx;
   if (!ctx) return;
   const theme = ctx.ui.theme;
+  // pi-budget parity: the theme proxy may not be initialized yet — dereferencing
+  // theme.fg throws (unhandledRejection → pi exits). Best-effort footer: skip.
+  if (!theme?.fg) return;
   if (!state.adapter) {
     // Unsupported provider (e.g. Ollama): still show the last response speed.
     ctx.ui.setStatus(STATUS_KEY, state.lastTokPerSec !== undefined ? theme.fg("dim", `${state.lastTokPerSec} tok/s`) : undefined);
@@ -1340,7 +1356,11 @@ export default function (pi: ExtensionAPI) {
 
   pi.on("message_end", async (event, _ctx) => {
     if (event.message.role === "assistant") {
-      state.cumulativeCost += (event.message.usage as any)?.cost?.total ?? 0;
+      // pi-budget parity: coerce + finite guard so a string/NaN cost.total can
+      // never poison the accumulator (string concat garbles every subsequent
+      // footer).
+      const cost = Number((event.message.usage as any)?.cost?.total);
+      if (Number.isFinite(cost) && cost > 0) state.cumulativeCost += cost;
       if (state.responseStartTime) {
         // usage.output already includes reasoning tokens (Pi SDK contract) —
         // this is total tok/s in both thinking and normal mode.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bacnh85/pi-sub",
-  "version": "0.1.38",
+  "version": "0.1.40",
   "description": "Pi extension showing subscription usage for supported model providers.",
   "type": "module",
   "license": "MIT",