openusage 0.1.0

package/plugins/gemini/icon.svg

@@ -0,0 +1,4 @@
+<svg width="100" height="100" viewBox="0 0 192 192" fill="none" xmlns="http://www.w3.org/2000/svg">
+  <path d="M96 8c3.6 0 6.8 2.4 7.7 5.9 5.4 21.2 19.2 35 40.4 40.4 3.5.9 5.9 4.1 5.9 7.7s-2.4 6.8-5.9 7.7c-21.2 5.4-35 19.2-40.4 40.4-.9 3.5-4.1 5.9-7.7 5.9s-6.8-2.4-7.7-5.9c-5.4-21.2-19.2-35-40.4-40.4C44.4 68.8 42 65.6 42 62s2.4-6.8 5.9-7.7c21.2-5.4 35-19.2 40.4-40.4C89.2 10.4 92.4 8 96 8z" fill="currentColor"/>
+  <path d="M160 88c1.8 0 3.4 1.2 3.8 2.9 2.6 10.2 9.3 16.9 19.5 19.5 1.7.4 2.9 2 2.9 3.8s-1.2 3.4-2.9 3.8c-10.2 2.6-16.9 9.3-19.5 19.5-.4 1.7-2 2.9-3.8 2.9s-3.4-1.2-3.8-2.9c-2.6-10.2-9.3-16.9-19.5-19.5-1.7-.4-2.9-2-2.9-3.8s1.2-3.4 2.9-3.8c10.2-2.6 16.9-9.3 19.5-19.5.4-1.7 2-2.9 3.8-2.9z" fill="currentColor"/>
+</svg>

package/plugins/gemini/plugin.js

@@ -0,0 +1,413 @@
+(function () {
+  const SETTINGS_PATH = "~/.gemini/settings.json"
+  const CREDS_PATH = "~/.gemini/oauth_creds.json"
+  const OAUTH2_JS_PATHS = [
+    "~/.bun/install/global/node_modules/@google/gemini-cli-core/dist/src/code_assist/oauth2.js",
+    "~/.npm-global/lib/node_modules/@google/gemini-cli-core/dist/src/code_assist/oauth2.js",
+    "~/.nvm/versions/node/current/lib/node_modules/@google/gemini-cli-core/dist/src/code_assist/oauth2.js",
+    "/opt/homebrew/opt/gemini-cli/libexec/lib/node_modules/@google/gemini-cli/node_modules/@google/gemini-cli-core/dist/src/code_assist/oauth2.js",
+    "/usr/local/opt/gemini-cli/libexec/lib/node_modules/@google/gemini-cli/node_modules/@google/gemini-cli-core/dist/src/code_assist/oauth2.js",
+  ]
+
+  // Dynamic discovery for pnpm global installs (path varies by pnpm version)
+  function discoverPnpmOauth2Paths(ctx) {
+    const pnpmGlobal = "~/.local/share/pnpm/global"
+    if (!ctx.host.fs.exists(pnpmGlobal)) return []
+    try {
+      const entries = ctx.host.fs.listDir(pnpmGlobal)
+      const paths = []
+      for (let i = 0; i < entries.length; i += 1) {
+        const versionDir = pnpmGlobal + "/" + entries[i]
+        const pnpmDir = versionDir + "/.pnpm"
+        if (!ctx.host.fs.exists(pnpmDir)) continue
+        try {
+          const pnpmEntries = ctx.host.fs.listDir(pnpmDir)
+          for (let j = 0; j < pnpmEntries.length; j += 1) {
+            if (pnpmEntries[j].indexOf("@google+gemini-cli-core") === 0) {
+              const candidate = pnpmDir + "/" + pnpmEntries[j] +
+                "/node_modules/@google/gemini-cli-core/dist/src/code_assist/oauth2.js"
+              paths.push(candidate)
+            }
+          }
+        } catch (e) { /* ignore */ }
+      }
+      return paths
+    } catch (e) {
+      return []
+    }
+  }
+
+  const LOAD_CODE_ASSIST_URL = "https://cloudcode-pa.googleapis.com/v1internal:loadCodeAssist"
+  const QUOTA_URL = "https://cloudcode-pa.googleapis.com/v1internal:retrieveUserQuota"
+  const PROJECTS_URL = "https://cloudresourcemanager.googleapis.com/v1/projects"
+  const TOKEN_URL = "https://oauth2.googleapis.com/token"
+  const REFRESH_BUFFER_MS = 5 * 60 * 1000
+
+  const IDE_METADATA = {
+    ideType: "IDE_UNSPECIFIED",
+    platform: "PLATFORM_UNSPECIFIED",
+    pluginType: "GEMINI",
+    duetProject: "default",
+  }
+
+  function loadSettings(ctx) {
+    if (!ctx.host.fs.exists(SETTINGS_PATH)) return null
+    try {
+      return ctx.util.tryParseJson(ctx.host.fs.readText(SETTINGS_PATH))
+    } catch (e) {
+      ctx.host.log.warn("failed reading settings: " + String(e))
+      return null
+    }
+  }
+
+  function assertSupportedAuthType(ctx) {
+    const settings = loadSettings(ctx)
+    const authType =
+      settings && typeof settings.authType === "string" ? settings.authType.trim().toLowerCase() : null
+
+    if (!authType || authType === "oauth-personal") return
+    if (authType === "api-key") {
+      throw "Gemini auth type api-key is not supported by this plugin yet."
+    }
+    if (authType === "vertex-ai") {
+      throw "Gemini auth type vertex-ai is not supported by this plugin yet."
+    }
+    throw "Gemini unsupported auth type: " + authType
+  }
+
+  function loadOauthCreds(ctx) {
+    if (!ctx.host.fs.exists(CREDS_PATH)) return null
+    try {
+      const parsed = ctx.util.tryParseJson(ctx.host.fs.readText(CREDS_PATH))
+      if (!parsed || typeof parsed !== "object") return null
+      if (!parsed.access_token && !parsed.refresh_token) return null
+      return parsed
+    } catch (e) {
+      ctx.host.log.warn("failed reading creds: " + String(e))
+      return null
+    }
+  }
+
+  function saveOauthCreds(ctx, creds) {
+    try {
+      ctx.host.fs.writeText(CREDS_PATH, JSON.stringify(creds, null, 2))
+    } catch (e) {
+      ctx.host.log.warn("failed persisting creds: " + String(e))
+    }
+  }
+
+  function parseOauthClientCreds(text) {
+    if (!text || typeof text !== "string") return null
+    const idMatch = text.match(/OAUTH_CLIENT_ID\s*=\s*['"]([^'"]+)['"]/)
+    const secretMatch = text.match(/OAUTH_CLIENT_SECRET\s*=\s*['"]([^'"]+)['"]/)
+    if (!idMatch || !secretMatch) return null
+    return { clientId: idMatch[1], clientSecret: secretMatch[1] }
+  }
+
+  function loadOauthClientCreds(ctx) {
+    const allPaths = OAUTH2_JS_PATHS.concat(discoverPnpmOauth2Paths(ctx))
+    for (let i = 0; i < allPaths.length; i += 1) {
+      const path = allPaths[i]
+      if (!ctx.host.fs.exists(path)) continue
+      try {
+        const parsed = parseOauthClientCreds(ctx.host.fs.readText(path))
+        if (parsed) return parsed
+      } catch (e) {
+        ctx.host.log.warn("failed reading oauth2.js: " + String(e))
+      }
+    }
+    return null
+  }
+
+  function readNumber(value) {
+    const n = Number(value)
+    return Number.isFinite(n) ? n : null
+  }
+
+  function decodeIdToken(ctx, token) {
+    if (typeof token !== "string" || !token) return null
+    try {
+      const payload = ctx.jwt.decodePayload(token)
+      return payload && typeof payload === "object" ? payload : null
+    } catch {
+      return null
+    }
+  }
+
+  function needsRefresh(creds) {
+    if (!creds.access_token) return true
+    const expiry = readNumber(creds.expiry_date)
+    if (expiry === null) return false
+    const expiryMs = expiry > 10_000_000_000 ? expiry : expiry * 1000
+    return Date.now() + REFRESH_BUFFER_MS >= expiryMs
+  }
+
+  function refreshToken(ctx, creds) {
+    if (!creds.refresh_token) return null
+    const clientCreds = loadOauthClientCreds(ctx)
+    if (!clientCreds) return null
+
+    let resp
+    try {
+      resp = ctx.util.request({
+        method: "POST",
+        url: TOKEN_URL,
+        headers: { "Content-Type": "application/x-www-form-urlencoded" },
+        bodyText:
+          "client_id=" +
+          encodeURIComponent(clientCreds.clientId) +
+          "&client_secret=" +
+          encodeURIComponent(clientCreds.clientSecret) +
+          "&refresh_token=" +
+          encodeURIComponent(creds.refresh_token) +
+          "&grant_type=refresh_token",
+        timeoutMs: 15000,
+      })
+    } catch (e) {
+      ctx.host.log.warn("refresh request failed: " + String(e))
+      return null
+    }
+
+    if (ctx.util.isAuthStatus(resp.status)) {
+      throw "Gemini session expired. Run `gemini auth login` to authenticate."
+    }
+    if (resp.status < 200 || resp.status >= 300) return null
+
+    const data = ctx.util.tryParseJson(resp.bodyText)
+    if (!data || typeof data.access_token !== "string" || !data.access_token) return null
+
+    creds.access_token = data.access_token
+    if (typeof data.id_token === "string" && data.id_token) creds.id_token = data.id_token
+    if (typeof data.refresh_token === "string" && data.refresh_token) creds.refresh_token = data.refresh_token
+    if (typeof data.expires_in === "number") {
+      creds.expiry_date = Date.now() + data.expires_in * 1000
+    }
+
+    saveOauthCreds(ctx, creds)
+    return creds.access_token
+  }
+
+  function postJson(ctx, url, accessToken, body) {
+    return ctx.util.request({
+      method: "POST",
+      url,
+      headers: {
+        Authorization: "Bearer " + accessToken,
+        Accept: "application/json",
+        "Content-Type": "application/json",
+      },
+      bodyText: JSON.stringify(body || {}),
+      timeoutMs: 10000,
+    })
+  }
+
+  function readFirstStringDeep(value, keys) {
+    if (!value || typeof value !== "object") return null
+
+    for (let i = 0; i < keys.length; i += 1) {
+      const v = value[keys[i]]
+      if (typeof v === "string" && v.trim()) return v.trim()
+    }
+
+    const nested = Object.values(value)
+    for (let i = 0; i < nested.length; i += 1) {
+      const found = readFirstStringDeep(nested[i], keys)
+      if (found) return found
+    }
+    return null
+  }
+
+  function mapTierToPlan(tier, idTokenPayload) {
+    if (!tier) return null
+    const normalized = String(tier).trim().toLowerCase()
+    if (normalized === "standard-tier") return "Paid"
+    if (normalized === "legacy-tier") return "Legacy"
+    if (normalized === "free-tier") return idTokenPayload && idTokenPayload.hd ? "Workspace" : "Free"
+    return null
+  }
+
+  function discoverProjectId(ctx, accessToken, loadCodeAssistData) {
+    const fromLoadCodeAssist = readFirstStringDeep(loadCodeAssistData, ["cloudaicompanionProject"])
+    if (fromLoadCodeAssist) return fromLoadCodeAssist
+
+    let projectsResp
+    try {
+      projectsResp = ctx.util.request({
+        method: "GET",
+        url: PROJECTS_URL,
+        headers: { Authorization: "Bearer " + accessToken, Accept: "application/json" },
+        timeoutMs: 10000,
+      })
+    } catch (e) {
+      ctx.host.log.warn("project discovery failed: " + String(e))
+      return null
+    }
+
+    if (projectsResp.status < 200 || projectsResp.status >= 300) return null
+    const projectsData = ctx.util.tryParseJson(projectsResp.bodyText)
+    const projects = projectsData && Array.isArray(projectsData.projects) ? projectsData.projects : []
+    for (let i = 0; i < projects.length; i += 1) {
+      const project = projects[i]
+      const projectId = project && typeof project.projectId === "string" ? project.projectId : null
+      if (!projectId) continue
+      if (projectId.indexOf("gen-lang-client") === 0) return projectId
+      const labels = project && project.labels && typeof project.labels === "object" ? project.labels : null
+      if (labels && labels["generative-language"] !== undefined) return projectId
+    }
+    return null
+  }
+
+  function collectQuotaBuckets(value, out) {
+    if (Array.isArray(value)) {
+      for (let i = 0; i < value.length; i += 1) collectQuotaBuckets(value[i], out)
+      return
+    }
+    if (!value || typeof value !== "object") return
+
+    if (typeof value.remainingFraction === "number") {
+      const modelId =
+        typeof value.modelId === "string"
+          ? value.modelId
+          : typeof value.model_id === "string"
+            ? value.model_id
+            : "unknown"
+      out.push({
+        modelId,
+        remainingFraction: value.remainingFraction,
+        resetTime: value.resetTime || value.reset_time || null,
+      })
+    }
+
+    const nested = Object.values(value)
+    for (let i = 0; i < nested.length; i += 1) collectQuotaBuckets(nested[i], out)
+  }
+
+  function pickLowestRemainingBucket(buckets) {
+    let best = null
+    for (let i = 0; i < buckets.length; i += 1) {
+      const bucket = buckets[i]
+      if (!Number.isFinite(bucket.remainingFraction)) continue
+      if (!best || bucket.remainingFraction < best.remainingFraction) best = bucket
+    }
+    return best
+  }
+
+  function toUsageLine(ctx, label, bucket) {
+    const clampedRemaining = Math.max(0, Math.min(1, Number(bucket.remainingFraction)))
+    const used = Math.round((1 - clampedRemaining) * 100)
+    const resetsAt = ctx.util.toIso(bucket.resetTime)
+    const opts = {
+      label,
+      used,
+      limit: 100,
+      format: { kind: "percent" },
+    }
+    if (resetsAt) opts.resetsAt = resetsAt
+    return ctx.line.progress(opts)
+  }
+
+  function parseQuotaLines(ctx, quotaData) {
+    const buckets = []
+    collectQuotaBuckets(quotaData, buckets)
+    if (!buckets.length) return []
+
+    const proBuckets = []
+    const flashBuckets = []
+    for (let i = 0; i < buckets.length; i += 1) {
+      const bucket = buckets[i]
+      const lower = String(bucket.modelId || "").toLowerCase()
+      if (lower.indexOf("gemini") !== -1 && lower.indexOf("pro") !== -1) {
+        proBuckets.push(bucket)
+      } else if (lower.indexOf("gemini") !== -1 && lower.indexOf("flash") !== -1) {
+        flashBuckets.push(bucket)
+      }
+    }
+
+    const lines = []
+    const pro = pickLowestRemainingBucket(proBuckets)
+    if (pro) lines.push(toUsageLine(ctx, "Pro", pro))
+    const flash = pickLowestRemainingBucket(flashBuckets)
+    if (flash) lines.push(toUsageLine(ctx, "Flash", flash))
+    return lines
+  }
+
+  function fetchLoadCodeAssist(ctx, accessToken, creds) {
+    let currentToken = accessToken
+    const resp = ctx.util.retryOnceOnAuth({
+      request: function (token) {
+        return postJson(ctx, LOAD_CODE_ASSIST_URL, token || currentToken, { metadata: IDE_METADATA })
+      },
+      refresh: function () {
+        const refreshed = refreshToken(ctx, creds)
+        if (refreshed) currentToken = refreshed
+        return refreshed
+      },
+    })
+
+    if (ctx.util.isAuthStatus(resp.status)) {
+      throw "Gemini session expired. Run `gemini auth login` to authenticate."
+    }
+    if (resp.status < 200 || resp.status >= 300) return { data: null, accessToken: currentToken }
+    return { data: ctx.util.tryParseJson(resp.bodyText), accessToken: currentToken }
+  }
+
+  function fetchQuotaWithRetry(ctx, accessToken, creds, projectId) {
+    let currentToken = accessToken
+    const resp = ctx.util.retryOnceOnAuth({
+      request: function (token) {
+        const body = projectId ? { project: projectId } : {}
+        return postJson(ctx, QUOTA_URL, token || currentToken, body)
+      },
+      refresh: function () {
+        const refreshed = refreshToken(ctx, creds)
+        if (refreshed) currentToken = refreshed
+        return refreshed
+      },
+    })
+
+    if (ctx.util.isAuthStatus(resp.status)) {
+      throw "Gemini session expired. Run `gemini auth login` to authenticate."
+    }
+    if (resp.status < 200 || resp.status >= 300) {
+      throw "Gemini quota request failed (HTTP " + String(resp.status) + "). Try again later."
+    }
+    return resp
+  }
+
+  function probe(ctx) {
+    assertSupportedAuthType(ctx)
+
+    const creds = loadOauthCreds(ctx)
+    if (!creds) throw "Not logged in. Run `gemini auth login` to authenticate."
+
+    let accessToken = creds.access_token
+    if (needsRefresh(creds)) {
+      const refreshed = refreshToken(ctx, creds)
+      if (refreshed) accessToken = refreshed
+      else if (!accessToken) throw "Not logged in. Run `gemini auth login` to authenticate."
+    }
+
+    const idTokenPayload = decodeIdToken(ctx, creds.id_token)
+    const loadCodeAssistResult = fetchLoadCodeAssist(ctx, accessToken, creds)
+    accessToken = loadCodeAssistResult.accessToken
+
+    const tier = readFirstStringDeep(loadCodeAssistResult.data, ["tier", "userTier", "subscriptionTier"])
+    const plan = mapTierToPlan(tier, idTokenPayload)
+
+    const projectId = discoverProjectId(ctx, accessToken, loadCodeAssistResult.data)
+    const quotaResp = fetchQuotaWithRetry(ctx, accessToken, creds, projectId)
+    const quotaData = ctx.util.tryParseJson(quotaResp.bodyText)
+    if (!quotaData || typeof quotaData !== "object") {
+      throw "Gemini quota response invalid. Try again later."
+    }
+
+    const lines = parseQuotaLines(ctx, quotaData)
+    const email = idTokenPayload && typeof idTokenPayload.email === "string" ? idTokenPayload.email : null
+    if (email) lines.push(ctx.line.text({ label: "Account", value: email }))
+    if (!lines.length) lines.push(ctx.line.badge({ label: "Status", text: "No usage data", color: "#a3a3a3" }))
+
+    return { plan: plan || undefined, lines }
+  }
+
+  globalThis.__openusage_plugin = { id: "gemini", probe }
+})()

package/plugins/gemini/plugin.json

@@ -0,0 +1,20 @@
+{
+  "schemaVersion": 1,
+  "id": "gemini",
+  "name": "Gemini",
+  "version": "0.0.1",
+  "entry": "plugin.js",
+  "icon": "icon.svg",
+  "brandColor": "#4285F4",
+  "cli": {
+    "category": "cli",
+    "binaryName": "gemini",
+    "installCmd": "npm install -g @google/gemini-cli",
+    "loginCmd": "gemini auth login"
+  },
+  "lines": [
+    { "type": "progress", "label": "Pro", "scope": "overview", "primaryOrder": 1 },
+    { "type": "progress", "label": "Flash", "scope": "overview", "primaryOrder": 2 },
+    { "type": "text", "label": "Account", "scope": "detail" }
+  ]
+}