@hashimkarim/usagestat-linux-x64-gnu 2.0.0-alpha.1 → 2.0.0-alpha.3

package/plugins/factory/plugin.js

@@ -313,16 +313,14 @@
     }
   }
 
-  function fetchUsage(ctx, accessToken) {
+  function fetchUsage(ctx, accessToken, cookie) {
+    const headers = { Accept: "application/json", "Content-Type": "application/json", "User-Agent": "usagestat" }
+    if (accessToken) headers.Authorization = "Bearer " + accessToken
+    if (cookie) headers.Cookie = cookie
     var resp = ctx.util.request({
       method: "POST",
       url: USAGE_URL,
-      headers: {
-        Authorization: "Bearer " + accessToken,
-        "Content-Type": "application/json",
-        Accept: "application/json",
-        "User-Agent": "OpenUsage",
-      },
+      headers,
       bodyText: JSON.stringify({ useCache: true }),
       timeoutMs: 10000,
     })
@@ -331,18 +329,14 @@
       resp = ctx.util.request({
         method: "GET",
         url: USAGE_URL,
-        headers: {
-          Authorization: "Bearer " + accessToken,
-          Accept: "application/json",
-          "User-Agent": "OpenUsage",
-        },
+        headers,
         timeoutMs: 10000,
       })
     }
     return resp
   }
 
-  function probe(ctx) {
+  function fetchOAuth(ctx) {
     const authState = loadAuth(ctx)
     if (!authState) {
       ctx.host.log.error("probe failed: not logged in")
@@ -412,7 +406,10 @@
     }
 
     ctx.host.log.info("usage fetch succeeded")
+    return resp
+  }
 
+  function buildResult(ctx, resp, source) {
     const data = ctx.util.tryParseJson(resp.bodyText)
     if (data === null) {
       throw "Usage response invalid. Try again later."
@@ -480,7 +477,34 @@
       lines.push(ctx.line.badge({ label: "Status", text: "No usage data", color: "#a3a3a3" }))
     }
 
-    return { plan: plan, lines: lines }
+    return { plan: plan, lines: lines, source }
+  }
+
+  function probe(ctx) {
+    const configured = ctx.provider || {}
+    const mode = ctx.sourceMode || "auto"
+    const key = String(configured.apiKey || ctx.host.env.get("FACTORY_API_KEY") || "").trim()
+    const cookie = String(configured.cookieHeader || ctx.host.env.get("FACTORY_COOKIE") || ctx.host.env.get("DROID_COOKIE") || "").trim().replace(/^cookie:\s*/i, "")
+    if (mode === "api" && !key) throw "Set FACTORY_API_KEY or provider apiKey."
+    if (mode === "web" && !cookie) throw "Set FACTORY_COOKIE or provider cookieHeader."
+    if (key && (mode === "auto" || mode === "api")) {
+      try {
+        const resp = fetchUsage(ctx, key)
+        if (ctx.util.isAuthStatus(resp.status)) throw "Factory API key was rejected."
+        if (resp.status < 200 || resp.status >= 300) throw "Factory API returned HTTP " + resp.status + "."
+        return buildResult(ctx, resp, "api")
+      } catch (error) {
+        if (mode === "api" || (!cookie && !loadAuth(ctx))) throw error
+        ctx.host.log.warn("Factory API unavailable; trying the configured session")
+      }
+    }
+    if (cookie && (mode === "auto" || mode === "web")) {
+      const resp = fetchUsage(ctx, null, cookie)
+      if (ctx.util.isAuthStatus(resp.status)) throw "Factory web session expired."
+      if (resp.status < 200 || resp.status >= 300) throw "Factory web usage returned HTTP " + resp.status + "."
+      return buildResult(ctx, resp, "web")
+    }
+    return buildResult(ctx, fetchOAuth(ctx), "oauth")
   }
 
   globalThis.__openusage_plugin = { id: "factory", probe }

package/plugins/factory/plugin.json

@@ -6,7 +6,8 @@
   "entry": "plugin.js",
   "icon": "icon.svg",
   "brandColor": "#020202",
-  "supportedModes": ["oauth", "local"],
+  "supportedModes": ["api", "oauth", "local", "web"],
+  "envVars": ["FACTORY_API_KEY", "FACTORY_COOKIE", "DROID_COOKIE"],
   "autoMode": "oauth",
   "lines": [
     { "type": "progress", "label": "Standard", "scope": "overview", "primaryOrder": 1 },

package/plugins/grok/plugin.js

@@ -1,6 +1,6 @@
 (function () {
   const AUTH_PATH = "~/.grok/auth.json";
-  const BILLING_URL = "https://cli-chat-proxy.grok.com/v1/billing";
+  const BILLING_URL = "https://cli-chat-proxy.grok.com/v1/billing?format=credits";
   const SETTINGS_URL = "https://cli-chat-proxy.grok.com/v1/settings";
   const REFRESH_URL = "https://auth.x.ai/oauth2/token";
   const DEFAULT_CLIENT_ID = "b1a00492-073a-47ea-816f-4c329264a828";
@@ -8,6 +8,11 @@
   const AUTH_REFRESH_BUFFER_MS = 5 * 60 * 1000;
   const LOGIN_HINT = "Grok auth expired. Run `grok login` again.";
 
+  function authPath(ctx) {
+    const home = ctx.host.env.get("GROK_HOME");
+    return typeof home === "string" && home.trim() ? home.trim().replace(/\/+$/, "") + "/auth.json" : AUTH_PATH;
+  }
+
   function readJson(ctx, path) {
     if (!ctx.host.fs.exists(path)) return null;
     try {
@@ -129,7 +134,7 @@
       entry.expires_at = new Date(expiresAtMs).toISOString();
 
       try {
-        ctx.host.fs.writeText(AUTH_PATH, JSON.stringify(auth, null, 2));
+        ctx.host.fs.writeText(authPath(ctx), JSON.stringify(auth, null, 2));
         ctx.host.log.info("Grok auth refresh succeeded, token persisted");
       } catch (e) {
         ctx.host.log.warn("Grok auth refresh succeeded but failed to save auth: " + String(e));
@@ -144,7 +149,7 @@
   }
 
   function loadAuth(ctx) {
-    const auth = readJson(ctx, AUTH_PATH);
+    const auth = readJson(ctx, authPath(ctx));
     if (!auth || typeof auth !== "object") {
       throw "Grok not logged in. Run `grok login`.";
     }
@@ -179,6 +184,7 @@
 
   function unitsValue(obj) {
     if (!obj || typeof obj !== "object") return null;
+    if ((typeof obj.val !== "number" && typeof obj.val !== "string") || String(obj.val).trim() === "") return null;
     const n = Number(obj.val);
     return Number.isFinite(n) ? n : null;
   }
@@ -235,7 +241,7 @@
           Accept: "application/json",
           "User-Agent": "OpenUsage",
         },
-        timeoutMs: 10000,
+        timeoutMs: 1500,
       });
       if (resp.status < 200 || resp.status >= 300) return null;
       const data = ctx.util.tryParseJson(resp.bodyText);
@@ -262,6 +268,25 @@
       throw "Grok billing response changed.";
     }
 
+    if (config.currentPeriod) {
+      const period = config.currentPeriod;
+      const start = ctx.util.parseDateMs(period.start);
+      const end = ctx.util.parseDateMs(period.end);
+      const now = nowMs(ctx);
+      const knownType = ["USAGE_PERIOD_TYPE_WEEKLY", "USAGE_PERIOD_TYPE_MONTHLY"].includes(period.type);
+      if (!knownType || !Number.isFinite(start) || !Number.isFinite(end) || end <= start) throw "Grok billing period is invalid.";
+      const hasPercent = Object.prototype.hasOwnProperty.call(config, "creditUsagePercent");
+      let percent = hasPercent ? unitsValue({ val: config.creditUsagePercent }) : 0;
+      if (percent === null || percent < 0 || (!hasPercent && !(start <= now && now < end))) throw "Grok billing usage is unavailable for this period.";
+      const cap = config.onDemandCap === undefined ? 0 : config.onDemandCap && config.onDemandCap.val === undefined ? 0 : unitsValue(config.onDemandCap);
+      if (cap === null || cap < 0) throw "Grok on-demand cap is invalid.";
+      return { plan: fetchPlanName(ctx, auth.token), source: "api", lines: [
+        ctx.line.progress({ label: period.type === "USAGE_PERIOD_TYPE_WEEKLY" ? "Weekly limit" : "Monthly limit", used: clampPercent(percent), limit: 100,
+          format: { kind: "percent" }, resetsAt: ctx.util.toIso(end), periodDurationMs: end - start }),
+        ctx.line.badge({ label: "Pay as you go", text: cap > 0 ? String(cap) + " cap" : "Disabled", color: cap > 0 ? "#22c55e" : "#a3a3a3" }),
+      ] };
+    }
+
     const usedUnits = unitsValue(config.used);
     const limitUnits = unitsValue(config.monthlyLimit);
     const onDemandCapUnits = unitsValue(config.onDemandCap);
@@ -290,7 +315,7 @@
       }),
     ];
 
-    return { plan: fetchPlanName(ctx, auth.token), lines };
+    return { plan: fetchPlanName(ctx, auth.token), source: "api", lines };
   }
 
   globalThis.__openusage_plugin = { id: "grok", probe };

package/plugins/grok/plugin.json

@@ -13,7 +13,9 @@
   "autoMode": "cli",
   "webUrl": "https://grok.com",
   "lines": [
-    { "type": "progress", "label": "Credits used", "scope": "overview", "primaryOrder": 1 },
+    { "type": "progress", "label": "Weekly limit", "scope": "overview", "primaryOrder": 1 },
+    { "type": "progress", "label": "Monthly limit", "scope": "overview" },
+    { "type": "progress", "label": "Credits used", "scope": "overview" },
     { "type": "badge", "label": "Pay as you go", "scope": "detail" }
   ]
 }

package/plugins/ibmbob/icon.svg

@@ -0,0 +1,23 @@
+<svg id="Bob" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 32 32">
+  <defs>
+    <linearGradient id="helmetShade" x1="0" y1="0" x2="1" y2="0">
+      <stop offset=".134" stop-opacity="0"/>
+      <stop offset=".873"/>
+    </linearGradient>
+    <linearGradient id="bobGradient" x1="0" y1="32" x2="32" y2="0" gradientUnits="userSpaceOnUse">
+      <stop offset=".1" stop-color="#a16efb"/>
+      <stop offset=".9" stop-color="#0e61fa"/>
+    </linearGradient>
+    <mask id="helmetMask" x="0" y="0" width="32" height="32" maskUnits="userSpaceOnUse">
+      <path d="M29.718 15.25a13.67 13.67 0 0 0-9.991-12.476A1.747 1.747 0 0 0 18 1.25h-4c-.887 0-1.615.667-1.727 1.524A13.67 13.67 0 0 0 2.282 15.25H0v1.5h32v-1.5h-2.282zM12.25 4.337V13h1.5V3a.25.25 0 0 1 .25-.25h4a.25.25 0 0 1 .25.25v10h1.5V4.337a12.183 12.183 0 0 1 8.468 10.913H3.782A12.183 12.183 0 0 1 12.25 4.337z" fill="#fff"/>
+      <path d="M21 7v7H11V7h10z" fill="url(#helmetShade)"/>
+    </mask>
+  </defs>
+  <g mask="url(#helmetMask)">
+    <path fill="url(#bobGradient)" d="M0 0h32v32H0z"/>
+  </g>
+  <path d="M24 30.75H8a4.756 4.756 0 0 1-4.69-4H3c-.965 0-1.75-.785-1.75-1.75v-6h1.5v6c0 .138.112.25.25.25h1.75V26A3.254 3.254 0 0 0 8 29.25h16A3.254 3.254 0 0 0 27.25 26v-.75H29a.25.25 0 0 0 .25-.25v-6h1.5v6c0 .965-.785 1.75-1.75 1.75h-.31a4.756 4.756 0 0 1-4.69 4z" fill="#001d6c"/>
+  <circle cx="22" cy="21" r="2" fill="#001d6c"/>
+  <circle cx="10" cy="21" r="2" fill="#001d6c"/>
+  <path d="M16 27.75a5.695 5.695 0 0 1-3.451-1.15.75.75 0 0 1 .903-1.199c1.476 1.111 3.617 1.111 5.099 0a.75.75 0 1 1 .901 1.199 5.696 5.696 0 0 1-3.45 1.15z" fill="#001d6c"/>
+</svg>

package/plugins/ibmbob/plugin.js

@@ -0,0 +1,61 @@
+(function () {
+  function clean(value) {
+    return typeof value === "string" && value.trim() ? value.trim() : null
+  }
+  function number(value) {
+    if (typeof value !== "number" && typeof value !== "string") return null
+    if (typeof value === "string" && !value.trim()) return null
+    const n = Number(value)
+    return Number.isFinite(n) && n >= 0 ? n : null
+  }
+  function regionalBase(domain) {
+    if (!clean(domain)) return "https://api.us-east.bob.ibm.com"
+    const host = domain.trim().toLowerCase()
+    if (!/^(?:[a-z0-9-]+\.)*bob\.ibm\.com$/.test(host)) throw "IBM Bob returned an untrusted regional host."
+    return "https://" + (host.startsWith("api.") ? host : "api." + host)
+  }
+  function request(ctx, url, headers) {
+    const result = ctx.util.requestJson({ method: "GET", url, headers, timeoutMs: 5000 })
+    if (ctx.util.isAuthStatus(result.resp.status)) throw "IBM Bob rejected the API key."
+    if (result.resp.status < 200 || result.resp.status >= 300) throw "IBM Bob API returned HTTP " + result.resp.status + "."
+    if (!result.json || typeof result.json !== "object") throw "Invalid IBM Bob response."
+    return result.json
+  }
+  function probe(ctx) {
+    const key = clean(ctx.provider && ctx.provider.apiKey) || clean(ctx.host.env.get("BOBSHELL_API_KEY"))
+    if (!key) throw "Set BOBSHELL_API_KEY or the IBM Bob provider API key."
+    const jwt = key.split(".").length === 3 && ctx.jwt.decodePayload(key)
+    const headers = { Authorization: (jwt ? "Bearer " : "Apikey ") + key, Accept: "application/json", "User-Agent": "UsageStat" }
+    const profile = request(ctx, "https://api.us-east.bob.ibm.com/admin/v1/profile", headers)
+    if (!Array.isArray(profile.instances)) throw "IBM Bob profile is missing subscriptions."
+    const lines = []
+    const plans = []
+    let count = 0
+    for (const instance of profile.instances) {
+      if (!instance || !clean(instance.user_id) || !Array.isArray(instance.teams)) continue
+      const base = regionalBase(instance.region_domain)
+      for (const team of instance.teams) {
+        if (!team || !clean(team.id)) continue
+        if (ctx.provider && ctx.provider.workspaceId && ctx.provider.workspaceId !== team.id) continue
+        if (++count > 20) throw "IBM Bob returned more than 20 teams; select a workspace."
+        const budget = request(ctx, base + "/admin/v1/teams/" + encodeURIComponent(team.id) + "/users/" + encodeURIComponent(instance.user_id),
+          Object.assign({}, headers, { "x-instance-id": instance.instance_id, "x-team-id": team.id }))
+        const used = number(budget.usage)
+        const limit = number(budget.budget_limit) === null ? number(team.budget_limit) : number(budget.budget_limit)
+        if (used === null) throw "IBM Bob did not return valid Bobcoin usage."
+        const label = clean(instance.instance_name) || clean(instance.name) || instance.instance_id
+        const teamName = clean(team.name) || team.id
+        const reset = typeof instance.refresh_at === "number" ? instance.refresh_at * 1000 : instance.refresh_at
+        if (limit !== null && limit > 0) {
+          lines.push(ctx.line.progress({ label: label + " / " + teamName, used, limit,
+            format: { kind: "count", suffix: "Bobcoins" }, resetsAt: ctx.util.toIso(reset),
+            periodDurationMs: 30 * 24 * 60 * 60 * 1000 }))
+        } else lines.push(ctx.line.text({ label: teamName, value: used + " Bobcoins used" }))
+        if (clean(instance.plan_name) && !plans.includes(instance.plan_name)) plans.push(instance.plan_name)
+      }
+    }
+    if (!lines.length) throw "No IBM Bob team subscription found."
+    return { displayName: "IBM Bob", source: "api", plan: plans.join(", ") || undefined, lines }
+  }
+  globalThis.__openusage_plugin = { id: "ibmbob", probe }
+})()

package/plugins/ibmbob/plugin.json

@@ -0,0 +1,22 @@
+{
+  "schemaVersion": 1,
+  "id": "ibmbob",
+  "name": "IBM Bob",
+  "version": "0.1.0",
+  "entry": "plugin.js",
+  "enabledByDefault": false,
+  "envVars": [
+    "BOBSHELL_API_KEY"
+  ],
+  "supportedModes": [
+    "api"
+  ],
+  "autoMode": "api",
+  "usageDashboardUrl": "https://bob.ibm.com",
+  "links": [
+    {
+      "label": "Dashboard",
+      "url": "https://bob.ibm.com"
+    }
+  ]
+}