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

package/plugins/kiro/plugin.js

@@ -0,0 +1,390 @@
+(function () {
+  const STATE_KEY = "kiro.kiroAgent"
+  const LOG_FILE_NAME = "q-client.log"
+  const TOKEN_PATH = "~/.aws/sso/cache/kiro-auth-token.json"
+  const REFRESH_URL = "https://prod.us-east-1.auth.desktop.kiro.dev/refreshToken"
+  const LIVE_STALE_MS = 15 * 60 * 1000
+  const REFRESH_BUFFER_MS = 10 * 60 * 1000
+  const DEFAULT_REGION = "us-east-1"
+  const COUNT_FORMAT = { kind: "count", suffix: "credits" }
+  const LOGIN_HINT = "Open Kiro and sign in, then try again."
+  const SESSION_HINT = "Kiro session expired. Open Kiro and sign in again."
+  const DATA_HINT = "Kiro usage data unavailable. Open the Kiro account dashboard once and try again."
+
+  function appPath(ctx, suffix) {
+    const settings = (ctx.provider && ctx.provider.settings) || {}
+    const root = typeof settings.userDataDir === "string" && settings.userDataDir
+      ? settings.userDataDir
+      : ctx.host.fs.appSupportPath("Kiro")
+    if (!root) throw { code: "failed", message: "Kiro data directory unavailable. Set settings.userDataDir to the IDE user data directory." }
+    return root.replace(/[\\/]+$/, "") + "/" + suffix
+  }
+
+  function num(value) {
+    if (typeof value === "number") return Number.isFinite(value) ? value : null
+    if (typeof value !== "string" || !value.trim()) return null
+    const parsed = Number(value)
+    return Number.isFinite(parsed) ? parsed : null
+  }
+  function first() {
+    for (let i = 0; i < arguments.length; i += 1) {
+      const value = num(arguments[i])
+      if (value !== null) return value
+    }
+    return null
+  }
+  function iso(ctx, value) {
+    const normalized = ctx.util.toIso(value)
+    return typeof normalized === "string" && normalized ? normalized : null
+  }
+  function title(value) {
+    const trimmed = String(value || "").trim()
+    return trimmed
+      ? trimmed
+          .toLowerCase()
+          .split(/\s+/)
+          .map((part) => part.charAt(0).toUpperCase() + part.slice(1))
+          .join(" ")
+      : null
+  }
+  function sanitizeAuth(token) {
+    if (!token || typeof token !== "object") return null
+    if (token.provider === "Google" || token.provider === "Github") return { ...token, authMethod: "social" }
+    if (token.provider === "ExternalIdp") return { ...token, authMethod: "external_idp" }
+    if (token.provider === "Enterprise" || token.provider === "BuilderId" || token.provider === "Internal") {
+      return { ...token, authMethod: "IdC" }
+    }
+    return token
+  }
+  function readJsonFile(ctx, path, label) {
+    if (!ctx.host.fs.exists(path)) return null
+    try {
+      return ctx.util.tryParseJson(ctx.host.fs.readText(path))
+    } catch (e) {
+      ctx.host.log.warn(label + " read failed: " + String(e))
+      return null
+    }
+  }
+  function loadAuthState(ctx) {
+    const parsed = readJsonFile(ctx, TOKEN_PATH, "auth token")
+    if (!parsed || typeof parsed !== "object") return null
+    const token = sanitizeAuth(parsed)
+    return token && (token.refreshToken || token.accessToken) ? { path: TOKEN_PATH, token } : null
+  }
+  function saveAuthState(ctx, authState) {
+    try {
+      ctx.host.fs.writeText(authState.path, JSON.stringify(authState.token, null, 2))
+      return true
+    } catch (e) {
+      ctx.host.log.warn("failed to persist refreshed Kiro auth: " + String(e))
+      return false
+    }
+  }
+  function loadProfileArn(ctx, authState) {
+    const fromToken = authState && authState.token && authState.token.profileArn
+    if (typeof fromToken === "string" && fromToken) return fromToken
+    const parsed = readJsonFile(ctx, appPath(ctx, "User/globalStorage/kiro.kiroagent/profile.json"), "profile")
+    return parsed && typeof parsed.arn === "string" && parsed.arn.trim() ? parsed.arn.trim() : null
+  }
+  function regionFromArn(profileArn) {
+    const parts = String(profileArn || "").split(":")
+    return parts.length > 3 && parts[3] ? parts[3] : DEFAULT_REGION
+  }
+  function readStateValue(ctx, key) {
+    try {
+      const sql = "SELECT value FROM ItemTable WHERE key = '" + String(key).replace(/'/g, "''") + "' LIMIT 1;"
+      const rows = ctx.util.tryParseJson(ctx.host.sqlite.query(appPath(ctx, "User/globalStorage/state.vscdb"), sql))
+      return Array.isArray(rows) && rows.length && typeof rows[0].value === "string" ? rows[0].value : null
+    } catch (e) {
+      ctx.host.log.warn("Kiro sqlite read failed: " + String(e))
+      return null
+    }
+  }
+  function normalizePool(ctx, raw, config) {
+    if (!raw || typeof raw !== "object") return null
+    if (config.statusKey) {
+      const status = raw[config.statusKey]
+      if (status && !config.allowedStatuses.includes(status)) return null
+    }
+    const currentUsage = first(raw[config.preciseCurrent], raw[config.current])
+    const usageLimit = first(raw[config.preciseLimit], raw[config.limit])
+    if (currentUsage === null || usageLimit === null || usageLimit <= 0) return null
+    return {
+      currentUsage,
+      usageLimit,
+      expiryDate: iso(ctx, raw[config.expiryA] || raw[config.expiryB]),
+      displayName: typeof raw.displayName === "string" && raw.displayName.trim() ? raw.displayName.trim() : null,
+    }
+  }
+  function normalizeBreakdown(ctx, raw) {
+    if (!raw || typeof raw !== "object") return null
+    const currentUsage = first(raw.currentUsageWithPrecision, raw.currentUsage)
+    const usageLimit = first(raw.usageLimitWithPrecision, raw.usageLimit)
+    if (currentUsage === null || usageLimit === null || usageLimit <= 0) return null
+    const bonuses = Array.isArray(raw.bonuses)
+      ? raw.bonuses
+          .map((item) =>
+            normalizePool(ctx, item, {
+              current: "currentUsage",
+              preciseCurrent: null,
+              limit: "usageLimit",
+              preciseLimit: null,
+              expiryA: "expiresAt",
+              expiryB: "expiryDate",
+              statusKey: "status",
+              allowedStatuses: ["ACTIVE", "EXHAUSTED"],
+            })
+          )
+          .filter(Boolean)
+      : []
+
+    return {
+      type: typeof raw.resourceType === "string" ? raw.resourceType : raw.type,
+      currentUsage,
+      usageLimit,
+      resetDate: iso(ctx, raw.nextDateReset || raw.resetDate),
+      freeTrialUsage: normalizePool(ctx, raw.freeTrialInfo || raw.freeTrialUsage, {
+        current: "currentUsage",
+        preciseCurrent: "currentUsageWithPrecision",
+        limit: "usageLimit",
+        preciseLimit: "usageLimitWithPrecision",
+        expiryA: "freeTrialExpiry",
+        expiryB: "expiryDate",
+        statusKey: "freeTrialStatus",
+        allowedStatuses: ["ACTIVE"],
+      }),
+      bonuses,
+    }
+  }
+  function normalizeCachedState(ctx) {
+    const parsed = ctx.util.tryParseJson(readStateValue(ctx, STATE_KEY))
+    const usageState = parsed && parsed["kiro.resourceNotifications.usageState"]
+    if (!usageState || !Array.isArray(usageState.usageBreakdowns)) return null
+    const usageBreakdowns = usageState.usageBreakdowns.map((item) => normalizeBreakdown(ctx, item)).filter(Boolean)
+    return usageBreakdowns.length
+      ? { usageBreakdowns, timestampMs: first(usageState.timestamp), plan: null, overageEnabled: null }
+      : null
+  }
+  function normalizeApiSnapshot(ctx, raw, timestampMs) {
+    if (!raw || typeof raw !== "object") return null
+    return {
+      usageBreakdowns: Array.isArray(raw.usageBreakdownList)
+        ? raw.usageBreakdownList.map((item) => normalizeBreakdown(ctx, item)).filter(Boolean)
+        : [],
+      timestampMs: timestampMs !== null ? timestampMs : null,
+      plan: title(raw.subscriptionInfo && raw.subscriptionInfo.subscriptionTitle),
+      overageEnabled: raw.overageConfiguration ? raw.overageConfiguration.overageStatus === "ENABLED" : null,
+    }
+  }
+  function parseUsageLogText(ctx, text) {
+    const lines = String(text || "").split(/\r?\n/)
+    for (let i = lines.length - 1; i >= 0; i -= 1) {
+      const line = lines[i]
+      if (line.indexOf('"commandName":"GetUsageLimitsCommand"') === -1) continue
+      const jsonStart = line.indexOf("{")
+      if (jsonStart === -1) continue
+      const parsed = ctx.util.tryParseJson(line.slice(jsonStart))
+      if (!parsed || !parsed.output) continue
+      const loggedAt = line.slice(0, jsonStart).trim().split(" [")[0]
+      return normalizeApiSnapshot(ctx, parsed.output, loggedAt ? ctx.util.parseDateMs(loggedAt.replace(" ", "T")) : null)
+    }
+    return null
+  }
+  function loadLoggedState(ctx) {
+    const logsRoot = appPath(ctx, "logs")
+    let sessions = []
+    try {
+      sessions = ctx.host.fs.listDir(logsRoot).slice().sort().reverse()
+    } catch {
+      return null
+    }
+    for (let i = 0; i < sessions.length && i < 12; i += 1) {
+      const sessionRoot = logsRoot + "/" + sessions[i]
+      let windows = []
+      try {
+        windows = ctx.host.fs.listDir(sessionRoot).slice().sort().reverse()
+      } catch {
+        continue
+      }
+      for (let j = 0; j < windows.length; j += 1) {
+        const logPath = sessionRoot + "/" + windows[j] + "/exthost/kiro.kiroAgent/" + LOG_FILE_NAME
+        if (!ctx.host.fs.exists(logPath)) continue
+        try {
+          const snapshot = parseUsageLogText(ctx, ctx.host.fs.readText(logPath))
+          if (snapshot) return snapshot
+        } catch (e) {
+          ctx.host.log.warn("failed to parse Kiro usage log: " + String(e))
+        }
+      }
+    }
+    return null
+  }
+  function buildUserAgent(ctx) {
+    return "OpenUsage/" + String(ctx.app && ctx.app.version ? ctx.app.version : "0.0.0")
+  }
+  function needsRefresh(ctx, authState, nowMs) {
+    return ctx.util.needsRefreshByExpiry({
+      nowMs,
+      expiresAtMs: ctx.util.parseDateMs(authState.token && authState.token.expiresAt),
+      bufferMs: REFRESH_BUFFER_MS,
+    })
+  }
+  function buildUsageHeaders(ctx, authState, accessToken) {
+    const headers = { Authorization: "Bearer " + accessToken, Accept: "application/json", "User-Agent": buildUserAgent(ctx) }
+    if (authState.token && authState.token.authMethod === "external_idp") headers.TokenType = "EXTERNAL_IDP"
+    if (authState.token && authState.token.provider === "Internal") headers["redirect-for-internal"] = "true"
+    return headers
+  }
+  function refreshAccessToken(ctx, authState, nowMs) {
+    if (!authState.token || !authState.token.refreshToken) throw SESSION_HINT
+    const { resp, json } = ctx.util.requestJson({
+      method: "POST",
+      url: REFRESH_URL,
+      headers: { "Content-Type": "application/json", "User-Agent": buildUserAgent(ctx) },
+      bodyText: JSON.stringify({ refreshToken: authState.token.refreshToken }),
+      timeoutMs: 15000,
+    })
+    if (ctx.util.isAuthStatus(resp.status)) throw SESSION_HINT
+    if (resp.status < 200 || resp.status >= 300 || !json || typeof json.accessToken !== "string" || !json.accessToken) {
+      ctx.host.log.warn("Kiro token refresh failed: HTTP " + resp.status)
+      return null
+    }
+    const expiresIn = first(json.expiresIn, json.expires_in)
+    authState.token = sanitizeAuth({
+      ...authState.token,
+      accessToken: json.accessToken,
+      refreshToken: typeof json.refreshToken === "string" && json.refreshToken ? json.refreshToken : authState.token.refreshToken,
+      profileArn: typeof json.profileArn === "string" && json.profileArn ? json.profileArn : authState.token.profileArn,
+      expiresAt: expiresIn !== null && expiresIn > 0 ? new Date(nowMs + expiresIn * 1000).toISOString() : authState.token.expiresAt,
+    })
+    saveAuthState(ctx, authState)
+    return authState.token.accessToken
+  }
+  function fetchLiveState(ctx, authState, nowMs) {
+    const profileArn = loadProfileArn(ctx, authState)
+    if (!profileArn) return null
+    const url =
+      "https://q." +
+      regionFromArn(profileArn) +
+      ".amazonaws.com/getUsageLimits?origin=" +
+      encodeURIComponent("AI_EDITOR") +
+      "&profileArn=" +
+      encodeURIComponent(profileArn) +
+      "&resourceType=" +
+      encodeURIComponent("AGENTIC_REQUEST")
+
+    let accessToken = authState.token && authState.token.accessToken
+    if (!accessToken || needsRefresh(ctx, authState, nowMs)) {
+      const refreshed = refreshAccessToken(ctx, authState, nowMs)
+      if (refreshed) accessToken = refreshed
+    }
+    if (!accessToken) throw SESSION_HINT
+
+    const resp = ctx.util.retryOnceOnAuth({
+      request: (tokenOverride) =>
+        ctx.util.request({
+          method: "GET",
+          url,
+          headers: buildUsageHeaders(ctx, authState, tokenOverride || accessToken),
+          timeoutMs: 15000,
+        }),
+      refresh: () => refreshAccessToken(ctx, authState, nowMs),
+    })
+
+    if (ctx.util.isAuthStatus(resp.status)) throw SESSION_HINT
+    if (resp.status < 200 || resp.status >= 300) {
+      ctx.host.log.warn("Kiro live usage request failed: HTTP " + resp.status)
+      return null
+    }
+    const parsed = ctx.util.tryParseJson(resp.bodyText)
+    if (!parsed) {
+      ctx.host.log.warn("Kiro live usage response invalid JSON")
+      return null
+    }
+    return normalizeApiSnapshot(ctx, parsed, nowMs)
+  }
+  function shouldTryLive(localState, loggedState, nowMs) {
+    return !localState || !loggedState || !loggedState.plan || localState.timestampMs === null || nowMs - localState.timestampMs > LIVE_STALE_MS
+  }
+  function mergeSnapshots(localState, loggedState, liveState, nowMs) {
+    const usageSource =
+      liveState && liveState.usageBreakdowns.length
+        ? liveState
+        : localState && localState.usageBreakdowns.length
+          ? localState
+          : loggedState && loggedState.usageBreakdowns.length
+            ? loggedState
+            : null
+    if (!usageSource) return null
+    return {
+      plan: (liveState && liveState.plan) || (loggedState && loggedState.plan) || null,
+      overageEnabled:
+        liveState && liveState.overageEnabled !== null
+          ? liveState.overageEnabled
+          : loggedState && loggedState.overageEnabled !== null
+            ? loggedState.overageEnabled
+            : null,
+      usageBreakdowns: usageSource.usageBreakdowns,
+      timestampMs: usageSource.timestampMs !== null ? usageSource.timestampMs : nowMs,
+    }
+  }
+  function pickPrimaryBreakdown(usageBreakdowns) {
+    for (let i = 0; i < usageBreakdowns.length; i += 1) if (usageBreakdowns[i].type === "CREDIT") return usageBreakdowns[i]
+    return usageBreakdowns.length ? usageBreakdowns[0] : null
+  }
+  function pickBonusUsage(primary) {
+    return !primary ? null : primary.freeTrialUsage && primary.freeTrialUsage.usageLimit > 0 ? primary.freeTrialUsage : primary.bonuses && primary.bonuses.length ? primary.bonuses[0] : null
+  }
+  function formatAge(nowMs, timestampMs) {
+    if (!Number.isFinite(nowMs) || !Number.isFinite(timestampMs)) return null
+    const diffMs = Math.max(0, nowMs - timestampMs)
+    if (diffMs < 60000) return "Just now"
+    const minutes = Math.floor(diffMs / 60000)
+    if (minutes < 60) return minutes + "m ago"
+    const hours = Math.floor(minutes / 60)
+    return hours < 48 ? hours + "h ago" : Math.floor(hours / 24) + "d ago"
+  }
+  function buildOutput(ctx, snapshot, nowMs) {
+    const primary = pickPrimaryBreakdown(snapshot.usageBreakdowns)
+    if (!primary) throw DATA_HINT
+    const lines = [ctx.line.progress({ label: "Credits", used: primary.currentUsage, limit: primary.usageLimit, format: COUNT_FORMAT, resetsAt: primary.resetDate || undefined })]
+    const bonusUsage = pickBonusUsage(primary)
+    if (bonusUsage) {
+      lines.push(
+        ctx.line.progress({
+          label: "Bonus Credits",
+          used: bonusUsage.currentUsage,
+          limit: bonusUsage.usageLimit,
+          format: COUNT_FORMAT,
+          resetsAt: bonusUsage.expiryDate || undefined,
+        })
+      )
+    }
+    if (snapshot.overageEnabled !== null) lines.push(ctx.line.badge({ label: "Overages", text: snapshot.overageEnabled ? "Enabled" : "Disabled" }))
+    return { plan: snapshot.plan || undefined, lines }
+  }
+  function probe(ctx) {
+    const nowMs = ctx.util.parseDateMs(ctx.nowIso) || Date.now()
+    const authState = loadAuthState(ctx)
+    if (!authState || !authState.token || !authState.token.refreshToken) throw LOGIN_HINT
+    const localState = normalizeCachedState(ctx)
+    const loggedState = loadLoggedState(ctx)
+    let liveState = null
+    let liveError = null
+    if (shouldTryLive(localState, loggedState, nowMs)) {
+      try {
+        liveState = fetchLiveState(ctx, authState, nowMs)
+      } catch (e) {
+        liveError = e
+        ctx.host.log.warn("Kiro live fallback failed: " + String(e))
+      }
+    }
+    const snapshot = mergeSnapshots(localState, loggedState, liveState, nowMs)
+    if (!snapshot) {
+      if (typeof liveError === "string" && liveError) throw liveError
+      throw DATA_HINT
+    }
+    return buildOutput(ctx, snapshot, nowMs)
+  }
+  globalThis.__openusage_plugin = { id: "kiro", probe }
+})()

package/plugins/kiro/plugin.json

@@ -0,0 +1,16 @@
+{
+  "schemaVersion": 1,
+  "id": "kiro",
+  "name": "Kiro",
+  "version": "0.0.1",
+  "entry": "plugin.js",
+  "icon": "icon.svg",
+  "brandColor": "#C09CFF",
+  "supportedModes": ["oauth", "local"],
+  "autoMode": "local",
+  "lines": [
+    { "type": "progress", "label": "Credits", "scope": "overview", "primaryOrder": 1 },
+    { "type": "progress", "label": "Bonus Credits", "scope": "overview" },
+    { "type": "badge", "label": "Overages", "scope": "detail" }
+  ]
+}

package/plugins/litellm/icon.svg

@@ -0,0 +1,16 @@
+<svg viewBox="0 0 32 32" xmlns="http://www.w3.org/2000/svg">
+  <g fill="none" stroke="currentColor" stroke-linecap="round" stroke-linejoin="round">
+    <path d="M8.1 12.4C12.5 8.8 17.7 6.3 22.4 5.7c0-.5.5-.7 1-.8 1.8-.4 4.1-.5 5.8-.5.6 0 1 .4 1 1v19c0 .6-.4 1-1 1H10c-.7 0-1.3 0-1.9-.1V12.4Z" stroke-width="1.35"/>
+    <path d="M8.1 12.4C5 14.8 3 17 2.8 19.2c-.2 1.9 1.8 3.5 5.3 4.5V12.4Z" fill="currentColor" stroke-width="1.35"/>
+    <path d="M10.6 19.5h3.7" stroke-width="1.35"/>
+    <path d="M20.6 25.5l-1.3 2.6c.5.3 1.1.4 1.7.2.9-.2 1.6-1.2 1.7-2.8" stroke-width="1.2"/>
+    <path d="M25.3 25.5c.1 1.8 1.2 3 2.6 3s2.5-1.2 2.6-3" stroke-width="1.2"/>
+    <path d="M25.3 15.5h2.9c.5 0 .9.4.9.9v6.9c0 .5-.4.9-.9.9h-2.9c-.5 0-.9-.4-.9-.9v-6.9c0-.5.4-.9.9-.9Z" stroke-width="1.2"/>
+    <path d="M17.3 22.6h7.1m4.7 0h1.1" stroke-width="1.05"/>
+  </g>
+  <path d="M12.9 13.1c4.2-2.3 8.3-3.5 12.7-3.6.8.2 1.8 1.5 2.4 3.1-5.3.2-10.3.8-15.2 2.6-1.2.4-1.5-1.2.1-2.1Z" fill="currentColor"/>
+  <path d="M20.4 16.5h4v5.4h-6.6l2.6-5.4Z" fill="currentColor"/>
+  <path d="M29.1 16.5h1.1v5.4h-1.1v-5.4Z" fill="currentColor"/>
+  <path d="M10.9 25.5h9.5l-1.6 2.3H10c-.8 0-1.1-.7-.7-1.3l1.6-1Z" fill="currentColor"/>
+  <path d="M25.8 25.5h4c-.1 1.2-.9 2.1-2 2.1s-1.9-.9-2-2.1Z" fill="currentColor"/>
+</svg>

package/plugins/litellm/plugin.js

@@ -0,0 +1,199 @@
+(function () {
+  function env(ctx, name) {
+    try {
+      var value = ctx.host.env.get(name);
+      return typeof value === "string" && value.trim() ? value.trim() : null;
+    } catch (_) {
+      return null;
+    }
+  }
+
+  function apiKey(ctx) {
+    var configured = ctx.provider && typeof ctx.provider.apiKey === "string" ? ctx.provider.apiKey.trim() : "";
+    return configured || env(ctx, "LITELLM_API_KEY");
+  }
+
+  function setting(ctx, names) {
+    var settings = ctx.provider && ctx.provider.settings ? ctx.provider.settings : {};
+    for (var i = 0; i < names.length; i++) {
+      var value = settings[names[i]];
+      if (typeof value === "string" && value.trim()) return value.trim();
+    }
+    return null;
+  }
+
+  function baseUrl(ctx) {
+    var configured = setting(ctx, ["enterpriseHost", "baseUrl", "baseURL", "apiUrl", "apiURL"]);
+    var base = configured || env(ctx, "LITELLM_BASE_URL");
+    if (!base) throw "Missing LiteLLM base URL. Set LITELLM_BASE_URL or provider settings.enterpriseHost.";
+    base = base.trim().replace(/\/+$/, "");
+    if (base.toLowerCase().endsWith("/v1")) base = base.slice(0, -3).replace(/\/+$/, "");
+    if (!/^https?:\/\//i.test(base)) throw "LiteLLM base URL is invalid.";
+    return base;
+  }
+
+  function numberValue(value) {
+    if (typeof value === "number" && Number.isFinite(value)) return value;
+    if (typeof value === "string" && value.trim()) {
+      var parsed = Number(value.replace(/[$,]/g, ""));
+      if (Number.isFinite(parsed)) return parsed;
+    }
+    return null;
+  }
+
+  function nonEmpty(value) {
+    return typeof value === "string" && value.trim() ? value.trim() : null;
+  }
+
+  function parseDate(value) {
+    if (typeof value !== "string" || !value.trim()) return null;
+    var ms = Date.parse(value);
+    return Number.isFinite(ms) ? new Date(ms) : null;
+  }
+
+  function requestJson(ctx, base, path, key) {
+    var resp = ctx.util.request({
+      method: "GET",
+      url: base + path,
+      headers: { Authorization: "Bearer " + key, Accept: "application/json" },
+      timeoutMs: 15000,
+    });
+    if (ctx.util.isAuthStatus(resp.status)) throw "LiteLLM API key was rejected.";
+    if (resp.status < 200 || resp.status >= 300) {
+      throw "LiteLLM API request failed (HTTP " + resp.status + ").";
+    }
+    var json = ctx.util.tryParseJson(resp.bodyText);
+    if (!json) throw "LiteLLM response was not valid JSON.";
+    return json;
+  }
+
+  function keyInfo(json) {
+    var info = json && json.info ? json.info : json;
+    var userID = nonEmpty(info && (info.user_id || info.userID));
+    var teamID = nonEmpty(info && (info.team_id || info.teamID));
+    if (!userID && !teamID) throw "LiteLLM key info did not include a user_id or team_id.";
+    return {
+      userID: userID,
+      teamID: teamID,
+      keyName: nonEmpty(info.key_name || info.keyName),
+      spendUSD: numberValue(info.spend) || 0,
+      expiresAt: parseDate(info.expires),
+    };
+  }
+
+  function findTeam(teams, teamID) {
+    if (!Array.isArray(teams) || !teamID) return null;
+    for (var i = 0; i < teams.length; i++) {
+      if (String(teams[i].team_id || teams[i].teamID || "") === teamID) return teams[i];
+    }
+    return null;
+  }
+
+  function usd(value) {
+    return "$" + (Number(value) || 0).toFixed(2);
+  }
+
+  function spendDetail(spend, budget, prefix) {
+    var value = budget != null && budget > 0 ? usd(spend) + " / " + usd(budget) : usd(spend);
+    return prefix ? prefix + ": " + value : value;
+  }
+
+  function spendLabel(label) {
+    return label.replace(/\s+budget$/i, " spend");
+  }
+
+  function addBudget(lines, label, spend, budget, resetAt, detailPrefix) {
+    if (budget != null && budget > 0) {
+      var line = ctxLineProgress(label, (spend / budget) * 100);
+      if (resetAt) line.resetsAt = resetAt.toISOString();
+      line.detail = spendDetail(spend, budget, detailPrefix);
+      lines.push(line);
+      return true;
+    } else if (spend > 0) {
+      lines.push({ type: "text", label: spendLabel(label), value: spendDetail(spend, null, detailPrefix) });
+      return true;
+    }
+    return false;
+  }
+
+  function ctxLineProgress(label, percent) {
+    return {
+      type: "progress",
+      label: label,
+      used: Math.max(0, Math.min(100, percent)),
+      limit: 100,
+      format: { kind: "percent" },
+    };
+  }
+
+  function userSnapshot(ctx, base, key, info) {
+    var encoded = encodeURIComponent(info.userID);
+    var json = requestJson(ctx, base, "/user/info?user_id=" + encoded, key);
+    var user = json.user_info || json.userInfo || json;
+    var responseID = nonEmpty(user.user_id || user.userID || json.user_id || json.userID);
+    if (responseID && responseID !== info.userID) throw "LiteLLM user_id did not match /key/info.";
+    var team = findTeam(json.teams, info.teamID);
+    return {
+      userID: info.userID,
+      email: nonEmpty(user.user_email || user.userEmail || user.user_alias || user.userAlias || (user.metadata && user.metadata.preferred_username)),
+      personalSpendUSD: numberValue(user.spend) || 0,
+      personalBudgetUSD: numberValue(user.max_budget || user.maxBudget),
+      personalResetAt: parseDate(user.budget_reset_at || user.budgetResetAt),
+      team: team ? {
+        id: String(team.team_id || team.teamID),
+        alias: nonEmpty(team.team_alias || team.teamAlias),
+        spendUSD: numberValue(team.spend) || 0,
+        budgetUSD: numberValue(team.max_budget || team.maxBudget),
+        resetAt: parseDate(team.budget_reset_at || team.budgetResetAt),
+      } : null,
+      keyName: info.keyName,
+      keyExpiresAt: info.expiresAt,
+    };
+  }
+
+  function teamSnapshot(ctx, base, key, info) {
+    var encoded = encodeURIComponent(info.teamID);
+    var json = requestJson(ctx, base, "/team/info?team_id=" + encoded, key);
+    var team = json.team_info || json.teamInfo || json;
+    var responseID = nonEmpty(team.team_id || team.teamID || json.team_id || json.teamID);
+    if (responseID && responseID !== info.teamID) throw "LiteLLM team_id did not match /key/info.";
+    return {
+      userID: null,
+      email: null,
+      personalSpendUSD: 0,
+      personalBudgetUSD: null,
+      personalResetAt: null,
+      team: {
+        id: info.teamID,
+        alias: nonEmpty(team.team_alias || team.teamAlias),
+        spendUSD: numberValue(team.spend) || 0,
+        budgetUSD: numberValue(team.max_budget || team.maxBudget),
+        resetAt: parseDate(team.budget_reset_at || team.budgetResetAt),
+      },
+      keyName: info.keyName,
+      keyExpiresAt: info.expiresAt,
+    };
+  }
+
+  function linesFor(snapshot) {
+    var lines = [];
+    addBudget(lines, "Personal budget", snapshot.personalSpendUSD, snapshot.personalBudgetUSD, snapshot.personalResetAt);
+    if (snapshot.team) addBudget(lines, "Team budget", snapshot.team.spendUSD, snapshot.team.budgetUSD, snapshot.team.resetAt, snapshot.team.alias ? "Team " + snapshot.team.alias : "Team");
+    if (!lines.length) lines.push({ type: "text", label: "Spend", value: usd(snapshot.personalSpendUSD || (snapshot.team && snapshot.team.spendUSD) || 0) });
+    if (snapshot.keyName) lines.push({ type: "text", label: "Key", value: snapshot.keyName });
+    if (snapshot.keyExpiresAt) lines.push({ type: "text", label: "Key Expires", value: snapshot.keyExpiresAt.toISOString().slice(0, 10) });
+    return lines;
+  }
+
+  function probe(ctx) {
+    var key = apiKey(ctx);
+    if (!key) throw "Missing LiteLLM API key. Set LITELLM_API_KEY or provider apiKey.";
+    var base = baseUrl(ctx);
+    var info = keyInfo(requestJson(ctx, base, "/key/info", key));
+    var snapshot = info.userID ? userSnapshot(ctx, base, key, info) : teamSnapshot(ctx, base, key, info);
+    var plan = snapshot.team && snapshot.team.alias ? snapshot.team.alias : null;
+    return { displayName: "LiteLLM", source: "api", plan: plan, lines: linesFor(snapshot) };
+  }
+
+  globalThis.__openusage_plugin = { id: "litellm", probe: probe };
+})();

package/plugins/litellm/plugin.json

@@ -0,0 +1,16 @@
+{
+  "schemaVersion": 1,
+  "id": "litellm",
+  "name": "LiteLLM",
+  "version": "0.0.1",
+  "entry": "plugin.js",
+  "icon": "icon.svg",
+  "brandColor": "#111827",
+  "supportedModes": ["api"],
+  "autoMode": "api",
+  "lines": [
+    { "type": "progress", "label": "Personal budget", "scope": "overview", "primaryOrder": 1 },
+    { "type": "progress", "label": "Team budget", "scope": "overview", "primaryOrder": 2 },
+    { "type": "text", "label": "Spend", "scope": "detail" }
+  ]
+}

package/plugins/llm-proxy/icon.svg

@@ -0,0 +1 @@
+<svg fill="currentColor" fill-rule="evenodd" height="1em" style="flex:none;line-height:1" viewBox="0 0 24 24" width="1em" xmlns="http://www.w3.org/2000/svg"><title>OpenAI</title><path d="M9.205 8.658v-2.26c0-.19.072-.333.238-.428l4.543-2.616c.619-.357 1.356-.523 2.117-.523 2.854 0 4.662 2.212 4.662 4.566 0 .167 0 .357-.024.547l-4.71-2.759a.797.797 0 00-.856 0l-5.97 3.473zm10.609 8.8V12.06c0-.333-.143-.57-.429-.737l-5.97-3.473 1.95-1.118a.433.433 0 01.476 0l4.543 2.617c1.309.76 2.189 2.378 2.189 3.948 0 1.808-1.07 3.473-2.76 4.163zM7.802 12.703l-1.95-1.142c-.167-.095-.239-.238-.239-.428V5.899c0-2.545 1.95-4.472 4.591-4.472 1 0 1.927.333 2.712.928L8.23 5.067c-.285.166-.428.404-.428.737v6.898zM12 15.128l-2.795-1.57v-3.33L12 8.658l2.795 1.57v3.33L12 15.128zm1.796 7.23c-1 0-1.927-.332-2.712-.927l4.686-2.712c.285-.166.428-.404.428-.737v-6.898l1.974 1.142c.167.095.238.238.238.428v5.233c0 2.545-1.974 4.472-4.614 4.472zm-5.637-5.303l-4.544-2.617c-1.308-.761-2.188-2.378-2.188-3.948A4.482 4.482 0 014.21 6.327v5.423c0 .333.143.571.428.738l5.947 3.449-1.95 1.118a.432.432 0 01-.476 0zm-.262 3.9c-2.688 0-4.662-2.021-4.662-4.519 0-.19.024-.38.047-.57l4.686 2.71c.286.167.571.167.856 0l5.97-3.448v2.26c0 .19-.07.333-.237.428l-4.543 2.616c-.619.357-1.356.523-2.117.523zm5.899 2.83a5.947 5.947 0 005.827-4.756C22.287 18.339 24 15.84 24 13.296c0-1.665-.713-3.282-1.998-4.448.119-.5.19-.999.19-1.498 0-3.401-2.759-5.947-5.946-5.947-.642 0-1.26.095-1.88.31A5.962 5.962 0 0010.205 0a5.947 5.947 0 00-5.827 4.757C1.713 5.447 0 7.945 0 10.49c0 1.666.713 3.283 1.998 4.448-.119.5-.19 1-.19 1.499 0 3.401 2.759 5.946 5.946 5.946.642 0 1.26-.095 1.88-.309a5.96 5.96 0 004.162 1.713z"></path></svg>