@bacnh85/pi-sub 0.1.20 → 0.1.21

package/README.md

@@ -61,6 +61,14 @@ The built-in `zai-coding-cn` provider targets the domestic BigModel endpoint (`o
 (Z.ai (CN) key#1a2b3c4d) R:55%/2H W:80%/3D 42 tok/s
 ```
 
+### 9router
+
+9router is a proxy/router with no upstream usage API, so the footer shows the configured endpoint and last response speed:
+
+```text
+9router (172.30.55.22:20128) 145 tok/s
+```
+
 ### Tokens per second
 
 `pi-sub` tracks each response's tokens-per-second (tok/s) speed by measuring the time from provider request to message completion against the response's output token count. The last response's speed is shown in the footer next to usage data. The `/sub` detail view shows both the last response speed and the session-wide average.

package/extensions/index.ts

@@ -1,5 +1,6 @@
 import { readStoredCredential, type ExtensionAPI, type ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { createHash } from "node:crypto";
+import fs from "node:fs";
 import os from "node:os";
 import path from "node:path";
 
@@ -15,6 +16,8 @@ const ZAI_PROVIDER = "zai";
 const ZAI_USAGE_URL = "https://api.z.ai/api/monitor/usage/quota/limit";
 const ZAI_CODING_CN_PROVIDER = "zai-coding-cn";
 const ZAI_CODING_CN_USAGE_URL = "https://open.bigmodel.cn/api/monitor/usage/quota/limit";
+const NINE_ROUTER_PROVIDER = "9router";
+const NINE_ROUTER_CONFIG_PATH = path.join(os.homedir(), ".pi", "agent", "9router-config.json");
 
 type ModelLike = { provider?: string; id?: string } | undefined;
 
@@ -110,6 +113,10 @@ function isZaiCodingCnModel(model: ModelLike): boolean {
   return (model?.provider?.toLowerCase() ?? "") === ZAI_CODING_CN_PROVIDER;
 }
 
+function isNineRouterModel(model: ModelLike): boolean {
+  return (model?.provider?.toLowerCase() ?? "") === NINE_ROUTER_PROVIDER;
+}
+
 function piAuthPath(): string {
   const configDir = process.env.PI_CODING_AGENT_DIR?.trim() || path.join(os.homedir(), ".pi", "agent");
   return path.join(configDir, "auth.json");
@@ -294,6 +301,21 @@ async function readZaiAuth(providerId: string, label: string): Promise<{ key: st
   return { key: entry.key, account: authAccountSnapshot(label, entry) };
 }
 
+// ponytail: duplicated from pi-9router — no shared config lib, 8 lines, acceptable
+function readNineRouterConfig(): { baseUrl: string; apiKey?: string } | null {
+  try {
+    const exists = (p: string) => { try { return fs.statSync(p).isFile(); } catch { return false; } };
+    if (!exists(NINE_ROUTER_CONFIG_PATH)) return null;
+    const data = JSON.parse(fs.readFileSync(NINE_ROUTER_CONFIG_PATH, "utf8")) as Record<string, unknown>;
+    const baseUrl = process.env.NINE_ROUTER_BASE_URL || (typeof data.baseUrl === "string" ? data.baseUrl : undefined);
+    const apiKey = process.env.NINE_ROUTER_API_KEY || (typeof data.apiKey === "string" ? data.apiKey : undefined);
+    if (!baseUrl || typeof baseUrl !== "string") return null;
+    return { baseUrl, apiKey };
+  } catch {
+    return null;
+  }
+}
+
 async function fetchUsageFromPiAuth(entry: PiAuthEntry, signal?: AbortSignal): Promise<UsageApiSnapshot | undefined> {
   const accountId = getCodexAccountId(entry) ?? entry.accountId;
   if (!accountId) throw new Error("Missing openai-codex OAuth entry in Pi auth");
@@ -364,6 +386,31 @@ async function fetchOpenCodeGoUsage(_signal?: AbortSignal): Promise<Subscription
   }
 }
 
+async function fetchNineRouterUsage(_signal?: AbortSignal): Promise<SubscriptionUsageSnapshot> {
+  const cfg = readNineRouterConfig();
+  const now = Date.now();
+  if (!cfg) {
+    return {
+      providerDisplayName: "9router",
+      accounts: [],
+      fetchedAt: now,
+      error: "9router not configured — run /login-9router",
+    };
+  }
+  const account: SubscriptionAccountSnapshot = {
+    id: cfg.baseUrl,
+    isActive: true,
+    accountLabel: cfg.baseUrl.replace(/^https?:\/\//, ""),
+    lastActivity: "Now",
+  };
+  return {
+    providerDisplayName: "9router",
+    accounts: [account],
+    activeAccount: account,
+    fetchedAt: now,
+  };
+}
+
 // ---------------------------------------------------------------------------
 // Z.ai adapter
 // ---------------------------------------------------------------------------
@@ -552,6 +599,7 @@ function supportedAdapter(model: ModelLike): SubscriptionProviderAdapter | undef
   if (isOpenCodeGoModel(model)) return { id: OPC_PROVIDER, displayName: "OpenCode Go", fetchUsage: fetchOpenCodeGoUsage };
   if (isZaiModel(model)) return { id: ZAI_PROVIDER, displayName: "Z.ai", ...zaiUsageAdapter(ZAI_PROVIDER, ZAI_USAGE_URL, "Z.ai") };
   if (isZaiCodingCnModel(model)) return { id: ZAI_CODING_CN_PROVIDER, displayName: "Z.ai (CN)", ...zaiUsageAdapter(ZAI_CODING_CN_PROVIDER, ZAI_CODING_CN_USAGE_URL, "Z.ai (CN)") };
+  if (isNineRouterModel(model)) return { id: NINE_ROUTER_PROVIDER, displayName: "9router", fetchUsage: fetchNineRouterUsage };
   return undefined;
 }
 

package/package.json

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