@inferhub/usage 0.1.7 → 0.1.9

package/bin/statusline.mjs

@@ -1,4 +1,12 @@
 #!/usr/bin/env node
+/**
+ * Claude Code statusline — one compact line.
+ *
+ * Session id strategy:
+ *  - Prefer conversation-prefix hash from transcript (matches proxy sessionid.Derive)
+ *  - Also try Claude's session_id (if proxy stored user/metadata session)
+ *  - Never fall back to "latest account session" (that jumps between chats)
+ */
 import { readFileSync } from "node:fs";
 import { dirname, join } from "node:path";
 import { fileURLToPath, pathToFileURL } from "node:url";
@@ -14,35 +22,45 @@ function readStdin() {
   }
 }
 
-let session = {};
+let input = {};
 try {
-  session = JSON.parse(readStdin() || "{}");
+  input = JSON.parse(readStdin() || "{}");
 } catch {
-  session = {};
+  input = {};
 }
 
 try {
-  // Do NOT pass Claude's session_id through: InferHub derives its own conversation
-  // key server-side (hash of system+first user message / X-Session-Id). Claude's
-  // UUID almost never matches, which produced "sess 0r/0". Default API path uses
-  // the user's latest InferHub session.
+  const sessionId = client.sessionIdFromClaudeStatuslineInput({
+    session_id: input.session_id,
+    transcript_path: input.transcript_path,
+  });
+
+  // Fetch with explicit session when known; otherwise omit session block rather
+  // than showing the account's latest unrelated conversation.
   const { usage } = await client.fetchAccountUsageAuto({
     window: "day",
-    force: false,
+    sessionId: sessionId || undefined,
   });
+
+  // If we couldn't derive a session id, strip session so we don't show wrong data
+  if (!sessionId) {
+    usage.session = null;
+  } else if (usage.session && usage.session.id !== sessionId) {
+    // API returned a different session (e.g. latest) — hide it
+    usage.session = null;
+  }
+
   const contextPct =
-    typeof session?.context_window?.used_percentage === "number"
-      ? session.context_window.used_percentage
+    typeof input?.context_window?.used_percentage === "number"
+      ? input.context_window.used_percentage
       : null;
-  const model = session?.model?.display_name || session?.model?.id || null;
+  const model = input?.model?.display_name || input?.model?.id || null;
+
   process.stdout.write(
-    client.formatStatusLine(usage, {
-      contextPct,
-      model,
-    }) + "\n",
+    client.formatStatusLine(usage, { contextPct, model }) + "\n",
   );
 } catch (e) {
   const msg = e instanceof Error ? e.message : String(e);
-  process.stdout.write("InferHub · offline (" + msg.slice(0, 48) + ")\n");
+  process.stdout.write("InferHub · offline (" + msg.slice(0, 40) + ")\n");
   process.exitCode = 0;
 }

package/dist/format.d.ts

@@ -1,6 +1,8 @@
 import type { AccountUsage } from "./types.js";
 export declare function money(usdc: string | number | null | undefined, digits?: number): string;
 export declare function tokens(n: number | null | undefined): string;
+/** Compact integer for statusline: 3943 → 3.9k, 28000000 → 28.0M */
+export declare function compactCount(n: number | null | undefined): string;
 export declare function shortModel(model: string | null | undefined): string;
 /**
  * One-line Claude statusline.

package/dist/format.js

@@ -9,12 +9,19 @@ export function money(usdc, digits = 2) {
     return `$${n.toFixed(2)}`;
 }
 export function tokens(n) {
+    return compactCount(n);
+}
+/** Compact integer for statusline: 3943 → 3.9k, 28000000 → 28.0M */
+export function compactCount(n) {
     const v = Number(n ?? 0);
-    if (v >= 1_000_000_000)
+    if (!Number.isFinite(v))
+        return "0";
+    const abs = Math.abs(v);
+    if (abs >= 1_000_000_000)
         return `${(v / 1_000_000_000).toFixed(1)}B`;
-    if (v >= 1_000_000)
+    if (abs >= 1_000_000)
         return `${(v / 1_000_000).toFixed(1)}M`;
-    if (v >= 1_000)
+    if (abs >= 1_000)
         return `${(v / 1_000).toFixed(1)}k`;
     return String(Math.round(v));
 }
@@ -25,8 +32,8 @@ export function shortModel(model) {
     return parts[parts.length - 1] || model;
 }
 /**
- * Compact metric: "4r/1.2M" free, or "$0.12/5r/80k" when billed.
- * Always includes tokens when > 0.
+ * Compact metric: "4r/1.2M" free, or "$0.41/3.9kr/355.7M" when billed.
+ * Both requests and tokens use k/M/B.
  */
 function compact(spendUsdc, requests, totalTokens) {
     const spend = Number(spendUsdc ?? 0);
@@ -38,9 +45,9 @@ function compact(spendUsdc, requests, totalTokens) {
     if (spend > 0)
         bits.push(money(spend));
     if (req > 0)
-        bits.push(`${req}r`);
+        bits.push(`${compactCount(req)}r`);
     if (tok > 0)
-        bits.push(tokens(tok));
+        bits.push(compactCount(tok));
     return bits.join("/") || "—";
 }
 /**

package/dist/format.test.js

@@ -49,11 +49,11 @@ test("formatStatusLine includes tokens compactly", () => {
         cache: { cached: false, ttl_seconds: 30 },
     };
     const line = formatStatusLine(usage, { model: "grok-4.5", contextPct: 8 });
-    assert.equal(line, "InferHub · sess 19r/6.0M · day 129r/26.4M · all $0.41/3933r/354.2M · bal $1.31 · top gpt-5.5");
-    assert.ok(line.length < 110, `too long: ${line.length} ${line}`);
-    assert.match(line, /6\.0M/);
-    assert.match(line, /26\.4M/);
+    assert.equal(line, "InferHub · sess 19r/6.0M · day 129r/26.4M · all $0.41/3.9kr/354.2M · bal $1.31 · top gpt-5.5");
+    assert.ok(line.length < 100, `too long: ${line.length} ${line}`);
+    assert.match(line, /3\.9kr/);
     assert.match(line, /354\.2M/);
+    assert.doesNotMatch(line, /3933r/);
     // empty session omitted
     const noSess = { ...usage, session: null };
     assert.doesNotMatch(formatStatusLine(noSess), /sess /);

package/dist/index.d.ts

@@ -1,3 +1,4 @@
 export type { AccountUsage, FetchUsageOptions, ResolvedAuth, UsageWindowKind, } from "./types.js";
 export { DEFAULT_BASE_URL, fetchAccountUsage, fetchAccountUsageAuto, normalizeBaseUrl, resolveAuth, usageUrl, } from "./client.js";
 export { formatReport, formatStatusLine, money, shortModel, tokens } from "./format.js";
+export { deriveConversationSessionId, messagesFromTranscript, sessionIdFromClaudeStatuslineInput, } from "./session-id.js";

package/dist/index.js

@@ -1,2 +1,3 @@
 export { DEFAULT_BASE_URL, fetchAccountUsage, fetchAccountUsageAuto, normalizeBaseUrl, resolveAuth, usageUrl, } from "./client.js";
 export { formatReport, formatStatusLine, money, shortModel, tokens } from "./format.js";
+export { deriveConversationSessionId, messagesFromTranscript, sessionIdFromClaudeStatuslineInput, } from "./session-id.js";

package/dist/session-id.d.ts

@@ -0,0 +1,19 @@
+/** Same algorithm as proxy sessionid.conversationPrefix + sha256[:16]. */
+export declare function deriveConversationSessionId(messages: Array<{
+    role?: string;
+    content?: unknown;
+    type?: string;
+}>): string | null;
+/**
+ * Best-effort extract ordered messages from a Claude transcript jsonl.
+ * Handles common shapes: {type:"user"|"assistant", message:{role,content}} etc.
+ */
+export declare function messagesFromTranscript(transcriptPath: string): Array<{
+    role?: string;
+    content?: unknown;
+    type?: string;
+}>;
+export declare function sessionIdFromClaudeStatuslineInput(input: {
+    session_id?: string;
+    transcript_path?: string;
+}): string | null;

package/dist/session-id.js

@@ -0,0 +1,103 @@
+/**
+ * Match proxy/internal/sessionid.Derive conversation-prefix hashing so the
+ * Claude statusline can query the same session_id that request_logs stores.
+ *
+ * Priority mirrors the proxy:
+ *  1. Explicit session id (Claude stdin session_id) if we ever store it server-side
+ *  2. sha256(system + first user message)[:16] hex — same as proxy
+ */
+import { createHash } from "node:crypto";
+import { existsSync, readFileSync } from "node:fs";
+function flattenContent(content) {
+    if (content == null)
+        return "";
+    if (typeof content === "string")
+        return content;
+    if (Array.isArray(content)) {
+        return content.map(flattenContent).join("");
+    }
+    if (typeof content === "object") {
+        const o = content;
+        if (typeof o.text === "string")
+            return o.text;
+        if (o.content != null)
+            return flattenContent(o.content);
+    }
+    return "";
+}
+/** Same algorithm as proxy sessionid.conversationPrefix + sha256[:16]. */
+export function deriveConversationSessionId(messages) {
+    let system = "";
+    let firstUser = "";
+    for (const m of messages) {
+        const role = (m.role || m.type || "").toLowerCase();
+        if (role === "system") {
+            const t = flattenContent(m.content);
+            if (t)
+                system += t + "\n";
+        }
+        else if (role === "user" || role === "human") {
+            firstUser = flattenContent(m.content);
+            break;
+        }
+    }
+    if (!firstUser)
+        return null;
+    const prefix = system + firstUser;
+    const sum = createHash("sha256").update(prefix, "utf8").digest("hex");
+    return sum.slice(0, 32); // 16 bytes = 32 hex chars (proxy uses sum[:16] bytes)
+}
+/**
+ * Best-effort extract ordered messages from a Claude transcript jsonl.
+ * Handles common shapes: {type:"user"|"assistant", message:{role,content}} etc.
+ */
+export function messagesFromTranscript(transcriptPath) {
+    if (!transcriptPath || !existsSync(transcriptPath))
+        return [];
+    let text;
+    try {
+        text = readFileSync(transcriptPath, "utf8");
+    }
+    catch {
+        return [];
+    }
+    const out = [];
+    for (const line of text.split(/\r?\n/)) {
+        if (!line.trim())
+            continue;
+        try {
+            const row = JSON.parse(line);
+            // Claude Code transcript variants
+            if (row.type === "user" || row.type === "assistant" || row.type === "system") {
+                const msg = row.message || row;
+                out.push({
+                    role: String(msg.role || row.type),
+                    content: msg.content ?? row.content,
+                });
+                continue;
+            }
+            if (typeof row.role === "string" && row.content != null) {
+                out.push({ role: row.role, content: row.content });
+            }
+        }
+        catch {
+            /* skip bad lines */
+        }
+    }
+    return out;
+}
+export function sessionIdFromClaudeStatuslineInput(input) {
+    // Prefer Claude UUID — proxy now stores metadata.user_id / session_id when
+    // present, so each Claude conversation keeps its own counter.
+    if (input.session_id && input.session_id.length > 8)
+        return input.session_id;
+    // Fallback: conversation-prefix hash (matches proxy sessionid.Derive when no
+    // explicit session was sent).
+    if (input.transcript_path) {
+        const msgs = messagesFromTranscript(input.transcript_path);
+        const derived = deriveConversationSessionId(msgs);
+        if (derived)
+            return derived;
+    }
+    return null;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@inferhub/usage",
-  "version": "0.1.7",
+  "version": "0.1.9",
   "description": "Shared InferHub account usage client for coding-agent plugins",
   "type": "module",
   "main": "./dist/index.js",

package/src/format.test.ts

@@ -59,12 +59,12 @@ test("formatStatusLine includes tokens compactly", () => {
   const line = formatStatusLine(usage, { model: "grok-4.5", contextPct: 8 });
   assert.equal(
     line,
-    "InferHub · sess 19r/6.0M · day 129r/26.4M · all $0.41/3933r/354.2M · bal $1.31 · top gpt-5.5",
+    "InferHub · sess 19r/6.0M · day 129r/26.4M · all $0.41/3.9kr/354.2M · bal $1.31 · top gpt-5.5",
   );
-  assert.ok(line.length < 110, `too long: ${line.length} ${line}`);
-  assert.match(line, /6\.0M/);
-  assert.match(line, /26\.4M/);
+  assert.ok(line.length < 100, `too long: ${line.length} ${line}`);
+  assert.match(line, /3\.9kr/);
   assert.match(line, /354\.2M/);
+  assert.doesNotMatch(line, /3933r/);
 
   // empty session omitted
   const noSess = { ...usage, session: null };

package/src/format.ts

@@ -9,10 +9,17 @@ export function money(usdc: string | number | null | undefined, digits = 2): str
 }
 
 export function tokens(n: number | null | undefined): string {
+  return compactCount(n);
+}
+
+/** Compact integer for statusline: 3943 → 3.9k, 28000000 → 28.0M */
+export function compactCount(n: number | null | undefined): string {
   const v = Number(n ?? 0);
-  if (v >= 1_000_000_000) return `${(v / 1_000_000_000).toFixed(1)}B`;
-  if (v >= 1_000_000) return `${(v / 1_000_000).toFixed(1)}M`;
-  if (v >= 1_000) return `${(v / 1_000).toFixed(1)}k`;
+  if (!Number.isFinite(v)) return "0";
+  const abs = Math.abs(v);
+  if (abs >= 1_000_000_000) return `${(v / 1_000_000_000).toFixed(1)}B`;
+  if (abs >= 1_000_000) return `${(v / 1_000_000).toFixed(1)}M`;
+  if (abs >= 1_000) return `${(v / 1_000).toFixed(1)}k`;
   return String(Math.round(v));
 }
 
@@ -23,8 +30,8 @@ export function shortModel(model: string | null | undefined): string {
 }
 
 /**
- * Compact metric: "4r/1.2M" free, or "$0.12/5r/80k" when billed.
- * Always includes tokens when > 0.
+ * Compact metric: "4r/1.2M" free, or "$0.41/3.9kr/355.7M" when billed.
+ * Both requests and tokens use k/M/B.
  */
 function compact(
   spendUsdc: string | number | null | undefined,
@@ -38,8 +45,8 @@ function compact(
 
   const bits: string[] = [];
   if (spend > 0) bits.push(money(spend));
-  if (req > 0) bits.push(`${req}r`);
-  if (tok > 0) bits.push(tokens(tok));
+  if (req > 0) bits.push(`${compactCount(req)}r`);
+  if (tok > 0) bits.push(compactCount(tok));
   return bits.join("/") || "—";
 }
 

package/src/index.ts

@@ -13,3 +13,8 @@ export {
   usageUrl,
 } from "./client.js";
 export { formatReport, formatStatusLine, money, shortModel, tokens } from "./format.js";
+export {
+  deriveConversationSessionId,
+  messagesFromTranscript,
+  sessionIdFromClaudeStatuslineInput,
+} from "./session-id.js";

package/src/session-id.ts

@@ -0,0 +1,106 @@
+/**
+ * Match proxy/internal/sessionid.Derive conversation-prefix hashing so the
+ * Claude statusline can query the same session_id that request_logs stores.
+ *
+ * Priority mirrors the proxy:
+ *  1. Explicit session id (Claude stdin session_id) if we ever store it server-side
+ *  2. sha256(system + first user message)[:16] hex — same as proxy
+ */
+import { createHash } from "node:crypto";
+import { existsSync, readFileSync } from "node:fs";
+
+function flattenContent(content: unknown): string {
+  if (content == null) return "";
+  if (typeof content === "string") return content;
+  if (Array.isArray(content)) {
+    return content.map(flattenContent).join("");
+  }
+  if (typeof content === "object") {
+    const o = content as { text?: unknown; content?: unknown };
+    if (typeof o.text === "string") return o.text;
+    if (o.content != null) return flattenContent(o.content);
+  }
+  return "";
+}
+
+/** Same algorithm as proxy sessionid.conversationPrefix + sha256[:16]. */
+export function deriveConversationSessionId(messages: Array<{
+  role?: string;
+  content?: unknown;
+  type?: string;
+}>): string | null {
+  let system = "";
+  let firstUser = "";
+  for (const m of messages) {
+    const role = (m.role || m.type || "").toLowerCase();
+    if (role === "system") {
+      const t = flattenContent(m.content);
+      if (t) system += t + "\n";
+    } else if (role === "user" || role === "human") {
+      firstUser = flattenContent(m.content);
+      break;
+    }
+  }
+  if (!firstUser) return null;
+  const prefix = system + firstUser;
+  const sum = createHash("sha256").update(prefix, "utf8").digest("hex");
+  return sum.slice(0, 32); // 16 bytes = 32 hex chars (proxy uses sum[:16] bytes)
+}
+
+/**
+ * Best-effort extract ordered messages from a Claude transcript jsonl.
+ * Handles common shapes: {type:"user"|"assistant", message:{role,content}} etc.
+ */
+export function messagesFromTranscript(transcriptPath: string): Array<{
+  role?: string;
+  content?: unknown;
+  type?: string;
+}> {
+  if (!transcriptPath || !existsSync(transcriptPath)) return [];
+  let text: string;
+  try {
+    text = readFileSync(transcriptPath, "utf8");
+  } catch {
+    return [];
+  }
+  const out: Array<{ role?: string; content?: unknown; type?: string }> = [];
+  for (const line of text.split(/\r?\n/)) {
+    if (!line.trim()) continue;
+    try {
+      const row = JSON.parse(line) as Record<string, unknown>;
+      // Claude Code transcript variants
+      if (row.type === "user" || row.type === "assistant" || row.type === "system") {
+        const msg = (row.message as Record<string, unknown>) || row;
+        out.push({
+          role: String(msg.role || row.type),
+          content: msg.content ?? row.content,
+        });
+        continue;
+      }
+      if (typeof row.role === "string" && row.content != null) {
+        out.push({ role: row.role, content: row.content });
+      }
+    } catch {
+      /* skip bad lines */
+    }
+  }
+  return out;
+}
+
+export function sessionIdFromClaudeStatuslineInput(input: {
+  session_id?: string;
+  transcript_path?: string;
+}): string | null {
+  // Prefer Claude UUID — proxy now stores metadata.user_id / session_id when
+  // present, so each Claude conversation keeps its own counter.
+  if (input.session_id && input.session_id.length > 8) return input.session_id;
+
+  // Fallback: conversation-prefix hash (matches proxy sessionid.Derive when no
+  // explicit session was sent).
+  if (input.transcript_path) {
+    const msgs = messagesFromTranscript(input.transcript_path);
+    const derived = deriveConversationSessionId(msgs);
+    if (derived) return derived;
+  }
+  return null;
+}