omnirush 0.3.3 → 0.4.1

package/assets/extensions/omnirush/usage.ts

@@ -151,6 +151,83 @@ export function formatGrantLines({
   return lines;
 }
 
+/**
+ * Exact ints for /usage, rendered with both the compact human form and the
+ * full digits (the full number is what support asks for).
+ */
+function tokenLine(label: string, value: unknown): string | null {
+  const n = intOr(value);
+  if (n === null) return null;
+  return `${label}: ${formatTokenCount(n)} (${n.toLocaleString("en-US")})`;
+}
+
+/**
+ * The detailed /usage block from a /device/me payload: grant date, token
+ * limit, used/reserved/remaining. Every field is optional server-side
+ * (early in a grant day only grant_date may be present), so lines appear
+ * only when the data exists.
+ */
+export function formatUsageDetails({ me }: { me: any }): string[] {
+  const u = me?.usage ?? null;
+  if (!u || typeof u !== "object") return ["no usage reported by the manager yet"];
+  const lines: string[] = [];
+  if (u.grant_date) lines.push(`grant date: ${u.grant_date}`);
+  let counters = 0;
+  for (const [label, key] of [
+    ["token limit", "token_limit"],
+    ["used", "used_tokens"],
+    ["reserved", "reserved_tokens"],
+    ["remaining", "remaining_tokens"],
+  ] as const) {
+    const line = tokenLine(label, u[key]);
+    if (line) {
+      lines.push(line);
+      counters += 1;
+    }
+  }
+  if (counters === 0) {
+    lines.push("no token counters reported yet (they appear with the first settled request)");
+  }
+  return lines;
+}
+
+/**
+ * One-line /usage summary, e.g.
+ * "997.2M of 1.0B tokens remaining (used 2.8M) — 2026-09-23".
+ * Falls back to whatever subset is present; empty string with no data.
+ */
+export function formatUsageSummary(me: any): string {
+  const u = me?.usage ?? null;
+  if (!u || typeof u !== "object") return "";
+  const limit = intOr(u.token_limit);
+  const used = intOr(u.used_tokens);
+  const reserved = intOr(u.reserved_tokens);
+  const remaining = intOr(u.remaining_tokens);
+  const parts: string[] = [];
+  if (remaining !== null && limit !== null) {
+    parts.push(`${formatTokenCount(remaining)} of ${formatTokenCount(limit)} tokens remaining`);
+  } else if (used !== null) {
+    parts.push(`${formatTokenCount(used)} tokens used`);
+  }
+  if (reserved !== null && reserved > 0) parts.push(`${formatTokenCount(reserved)} reserved`);
+  if (u.grant_date) parts.push(String(u.grant_date));
+  return parts.join(" — ");
+}
+
+/** Device list lines for /usage and /status (no credentials in me payloads). */
+export function formatDeviceLines(me: any): string[] {
+  const devices = Array.isArray(me?.devices) ? me.devices : [];
+  if (devices.length === 0) return [];
+  const lines = devices.map((device: any, index: number) => {
+    const name = typeof device?.device_name === "string" ? device.device_name : "device";
+    const platform = typeof device?.platform === "string" ? ` (${device.platform})` : "";
+    const seen = typeof device?.last_seen_at === "string" ? ` last seen ${device.last_seen_at}` : "";
+    const revoked = device?.revoked_at ? " [revoked]" : "";
+    return `  ${index + 1}. ${name}${platform}${seen}${revoked}`;
+  });
+  return [`devices signed in: ${devices.length}`, ...lines];
+}
+
 /** One-line footer summary for the TUI status area, or null with no data. */
 export function formatGrantStatusLine(
   usage: Record<string, string> | null,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "omnirush",
-  "version": "0.3.3",
+  "version": "0.4.1",
   "description": "Omnirush \u2014 free daily tokens for the most powerful coding model on earth.",
   "license": "Apache-2.0",
   "type": "module",
@@ -13,6 +13,7 @@
   "scripts": {
     "test": "node --test",
     "test:bun": "if [ -x .runtime/bin/bun ]; then .runtime/bin/bun test test/; else bun test test/; fi",
+    "build:mcp": "node scripts/build-mcp-bundle.js",
     "postinstall": "node scripts/patch-pi-branding.js && node scripts/postinstall.js"
   },
   "dependencies": {
@@ -32,5 +33,9 @@
     "assets",
     "scripts",
     "README.md"
-  ]
+  ],
+  "devDependencies": {
+    "@modelcontextprotocol/sdk": "^1.30.0",
+    "esbuild": "^0.28.2"
+  }
 }

package/scripts/build-mcp-bundle.js

@@ -0,0 +1,47 @@
+#!/usr/bin/env node
+// Bundle the MCP client SDK (@modelcontextprotocol/sdk) into a single
+// self-contained ESM file committed at
+// assets/extensions/omnirush/mcp-client.bundle.js.
+//
+// Why a committed bundle: the omnirush extension directory is copied to
+// ~/.pi/agent/extensions/omnirush/ and its files can only import
+// siblings — the jiti loader that runs pi extensions has no npm
+// resolution there, and the published npm package ships no node_modules.
+// The bundle ships in the repo, so CI and `npm pack` never need the
+// toolchain; regenerate only when bumping the SDK:
+//
+//   npm run build:mcp
+//
+// The output is deterministic for a given SDK version (esbuild, fixed
+// entry, no code splitting) so diffs stay reviewable.
+
+import { build } from "esbuild";
+import path from "node:path";
+import { fileURLToPath } from "node:url";
+
+const root = path.resolve(path.dirname(fileURLToPath(import.meta.url)), "..");
+const entry = path.join(root, "dev", "mcp-bundle-entry.js");
+const outfile = path.join(root, "assets", "extensions", "omnirush", "mcp-client.bundle.js");
+
+await build({
+  entryPoints: [entry],
+  outfile,
+  bundle: true,
+  platform: "node",
+  format: "esm",
+  target: "node18",
+  // Node builtins stay external (real `node:*` imports in the output);
+  // everything else — including cross-spawn — is inlined.
+  external: ["node:*"],
+  minify: false,
+  sourcemap: false,
+  legalComments: "none",
+  logLevel: "info",
+  banner: {
+    js: [
+      "// GENERATED FILE — do not edit by hand.",
+      "// Regenerate with: npm run build:mcp",
+      "// Bundled from @modelcontextprotocol/sdk (devDependency) via esbuild.",
+    ].join("\n"),
+  },
+});

package/scripts/patch-pi-branding.js

@@ -93,6 +93,33 @@ export const REPLACEMENTS = [
     from: "Update pi only (self works as alias to pi)",
     to: "Update the agent only (self works as an alias)",
   },
+  // Belt-and-braces against the pi.dev update banner (Foxtran's 0.73.1
+  // report: a stale core showed "Update Available — pi.dev/changelog"
+  // even though PI_OFFLINE should suppress the check). PI_OFFLINE=1 in
+  // the spawn env stops the check; these rewrites make the banner
+  // UNRENDERABLE even where PI_OFFLINE is absent (direct pi spawns,
+  // older cores): showNewVersionNotification returns before its body
+  // (the parameter is renamed so the replacement stays idempotent and
+  // never re-matches), and the pi.dev changelog URL is blanked. The
+  // parameter spelling differs between the standalone dist (readable)
+  // and the minified runtime bundle chunk, hence the two variants; a
+  // future core that renames the parameter just reports "missing"
+  // (fail-soft) and we would re-derive the target.
+  {
+    label: "TUI update banner (standalone dist)",
+    from: "showNewVersionNotification(release) {",
+    to: "showNewVersionNotification(_release) { return;",
+  },
+  {
+    label: "TUI update banner (runtime bundle chunk)",
+    from: "showNewVersionNotification(release2){",
+    to: "showNewVersionNotification(_r){return;",
+  },
+  {
+    label: "pi.dev changelog URL",
+    from: '"https://pi.dev/changelog"',
+    to: '""',
+  },
 ];
 
 function findPiRoot(startDir) {

package/src/bin.js

@@ -79,7 +79,7 @@ function managerOrigin() {
   );
 }
 
-function piEntry() {
+function piPackage() {
   // The pi package is Bun-first and its "exports" map blocks plain Node
   // resolution, so locate it by path: walk up from our own package root
   // to the nearest node_modules containing it.
@@ -88,13 +88,22 @@ function piEntry() {
   for (let i = 0; i < 6; i++) {
     const pkgPath = path.join(dir, "node_modules", PKG, "package.json");
     if (fs.existsSync(pkgPath)) {
-      const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf8"));
-      if (pkg.bin?.pi) return path.join(dir, "node_modules", PKG, pkg.bin.pi);
+      try {
+        return { dir: path.dirname(pkgPath), pkg: JSON.parse(fs.readFileSync(pkgPath, "utf8")) };
+      } catch {
+        return null;
+      }
     }
     const parent = path.dirname(dir);
     if (parent === dir) break;
     dir = parent;
   }
+  return null;
+}
+
+function piEntry() {
+  const found = piPackage();
+  if (found?.pkg?.bin?.pi) return path.join(found.dir, found.pkg.bin.pi);
   throw new Error("could not locate the pinned pi CLI (@earendil-works/pi-coding-agent)");
 }
 
@@ -421,6 +430,24 @@ async function cmdDoctor() {
   }
   ok("agent config", true, PI_MODELS_JSON);
   ok("extensions", true, PI_EXT_DIR);
+  // Agent core version: a stale core predates the omnirush provider and
+  // branding and shows pi.dev "Update Available" banners (Foxtran's
+  // 0.73.1 report). Advisory WARN with the fix; exit code unaffected.
+  {
+    const found = piPackage();
+    const coreVersion = String(found?.pkg?.version ?? "");
+    const own = ownVersion();
+    if (!coreVersion) {
+      console.log(`WARN agent core — pinned agent package not found (broken install). Fix: omnirush update`);
+    } else if (coreVersion === own) {
+      ok("agent core", true, `v${coreVersion}`);
+    } else {
+      console.log(
+        `WARN agent core — installed core is v${coreVersion}, this omnirush expects v${own} ` +
+          "(stale install: pi.dev update banners, missing omnirush provider). Fix: omnirush update",
+      );
+    }
+  }
   // Search tools: bundled fd/rg (or system ones) keep the agent off its
   // GitHub download path, which fails on restricted networks.
   const toolsRoot = path.join(PI_AGENT_DIR, "bin");
@@ -465,6 +492,14 @@ function requireAuth() {
   process.exit(1);
 }
 
+function ownVersion() {
+  try {
+    return JSON.parse(fs.readFileSync(path.join(__dirname, "..", "package.json"), "utf8")).version;
+  } catch {
+    return "unknown";
+  }
+}
+
 function childEnv(auth) {
   const gateway = agentGatewayUrl(auth);
   const origin = (process.env.OMNIRUSH_ORIGIN || "").trim().replace(/\/+$/, "") ||
@@ -475,12 +510,19 @@ function childEnv(auth) {
     OMNIRUSH_ORIGIN: origin,
     OMNIRUSH_GATEWAY_URL: gateway,
     OMNIRUSH_DIR: OMNI_DIR,
+    // /status shows the CLI version; extensions have no package.json of
+    // their own in ~/.pi, so the launcher hands it over.
+    OMNIRUSH_VERSION: ownVersion(),
     // The agent resolves its config dir from <APP>_CODING_AGENT_DIR. The
     // branding patch renames the app to omnirush, so set the omnirush
     // spelling; the pi spelling stays for unpatched cores.
     OMNIRUSH_CODING_AGENT_DIR: PI_AGENT_DIR,
     PI_CODING_AGENT_DIR: PI_AGENT_DIR,
     // The agent core is a fork/rebrand: never phone home to pi.dev.
+    // PI_OFFLINE kills startup update checks (the "Update Available —
+    // pi.dev/changelog" banner), package updates, and install/update
+    // telemetry; PI_TELEMETRY stays for cores that predate PI_OFFLINE.
+    PI_OFFLINE: "1",
     PI_TELEMETRY: "0",
   };
 }
@@ -527,7 +569,7 @@ if (cmd === "login") await cmdLogin(argv.slice(1));
 else if (cmd === "logout") await cmdLogout();
 else if (cmd === "whoami") await cmdWhoami();
 else if (cmd === "doctor") await cmdDoctor();
-else if (cmd === "--version" || cmd === "-v") console.log("omnirush 0.3.3");
+else if (cmd === "--version" || cmd === "-v") console.log(`omnirush ${ownVersion()}`);
 else if (cmd === "collect") await cmdCollect(argv.slice(1));
 else if (cmd === undefined || cmd === "run") await cmdRun();
 else {