@magpie-community/opencode-kiro-auth 0.1.0 → 0.1.2

package/index.mjs

@@ -266,13 +266,18 @@ function regionOf(profile, signedIn) {
   return String(signedIn ?? "").startsWith("eu-") ? "eu-central-1" : "us-east-1"
 }
 
+// STATUS_TEXT is Go's http.StatusText for the answers Kiro gives.
+const STATUS_TEXT = { 400: "Bad Request", 401: "Unauthorized", 403: "Forbidden", 404: "Not Found", 408: "Request Timeout",
+  409: "Conflict", 413: "Request Entity Too Large", 429: "Too Many Requests", 500: "Internal Server Error", 502: "Bad Gateway",
+  503: "Service Unavailable", 504: "Gateway Timeout" }
+
 class KiroError extends Error {
   constructor(status, body) {
     let msg = ""
     try {
       msg = JSON.parse(body)?.message ?? ""
     } catch {}
-    super(msg ? `Kiro: ${msg} (${status})` : `Kiro: HTTP ${status}`)
+    super(msg ? `Kiro: ${msg} (${status})` : `Kiro: ${STATUS_TEXT[status] ?? ""}`)
     this.status = status
   }
 }
@@ -333,6 +338,38 @@ function usageLimits(a, email) {
 const planName = (title) =>
   String(title ?? "").toLowerCase().split(/\s+/).filter(Boolean).map((w) => (w === "kiro" ? "Kiro" : w[0].toUpperCase() + w.slice(1))).join(" ")
 
+// goG is Go's %g of x: an exponent from a million up.
+function goG(x) {
+  const [m, e] = x.toExponential().split("e")
+  const exp = Number(e)
+  if (exp < -4 || exp >= 6) return `${m}e${exp < 0 ? "-" : "+"}${String(Math.abs(exp)).padStart(2, "0")}`
+  return String(x)
+}
+
+// usageOf is the allowance Get-Usage-Limits tells, as magpie's built-in
+// said it: the account's email, the credits of each allowance, the month's and a free trial's
+// while it runs.
+function usageOf(l) {
+  const at = (secs) => (secs > 0 ? new Date(Math.trunc(secs) * 1000).toISOString() : undefined)
+  const count = (used, limit) => used.toFixed(2).replace(/0+$/, "").replace(/\.$/, "") + " / " + goG(limit)
+  const num = (x) => (typeof x === "number" && Number.isFinite(x) ? x : 0)
+  const windows = []
+  for (const u of l?.usageBreakdownList ?? []) {
+    const t = u?.freeTrialInfo
+    if (t && String(t.freeTrialStatus ?? "").toUpperCase() === "ACTIVE" && num(t.usageLimitWithPrecision) > 0) {
+      const used = num(t.currentUsageWithPrecision), limit = num(t.usageLimitWithPrecision)
+      windows.push({ name: "Free trial", used: (100 * used) / limit, resetsAt: at(num(t.freeTrialExpiry)), display: count(used, limit) })
+    }
+    const used = num(u?.currentUsageWithPrecision), limit = num(u?.usageLimitWithPrecision)
+    if (limit <= 0) continue
+    const reset = num(u?.nextDateReset) || num(l?.nextDateReset)
+    windows.push({ name: u?.displayNamePlural || u?.displayName || "Credits", used: (100 * used) / limit, resetsAt: at(reset),
+      display: count(used, limit), span: 30 * 24 * 3600 })
+  }
+  const email = l?.userInfo?.email
+  return { plan: planName(l?.subscriptionInfo?.subscriptionTitle), ...(typeof email === "string" && email ? { user: email } : {}), windows }
+}
+
 // ---- the account, as the plugin holds it -------------------------------------------
 
 // credOf is the sign-in a stored auth names: a key, one signed in here, or
@@ -374,7 +411,11 @@ function account(client) {
       const read = await credOf(auth)
       if (!read) {
         held = null
-        throw new Error(auth?.source ? "Kiro isn't signed in; sign in with `kiro-cli login` or the Kiro IDE" : "Kiro isn't signed in; sign in again")
+        throw new Error(
+          auth?.source
+            ? "Kiro isn't signed in; add the Kiro subscription in magpie, sign in with `kiro-cli login` or the Kiro IDE, or save a Kiro API key on the provider"
+            : "this Kiro account's sign-in is gone; add it again in magpie",
+        )
       }
       let c = held?.id === id ? held.c : null
       if (!c || read.access !== c.access || !fresh(c) || stale) {
@@ -1134,7 +1175,7 @@ async function browserSignIn() {
     const q = url.searchParams
     if (!waiting) return html(200, page(false, "This sign-in is over", "Start it again."))
     const fail = (msg) => {
-      finish({ type: "failed" })
+      finish({ type: "failed", error: msg })
       html(200, page(false, "Sign-in didn't finish", msg))
     }
     if (q.get("error")) return fail(q.get("error_description") || q.get("error"))
@@ -1189,7 +1230,7 @@ async function browserSignIn() {
     html(200, page(true, "You're signed in", `${user} is signed in. You can close this tab.`))
   })
 
-  const timer = setTimeout(() => finish({ type: "failed" }), SIGN_IN_TIMEOUT)
+  const timer = setTimeout(() => finish({ type: "failed", error: "the sign-in timed out" }), SIGN_IN_TIMEOUT)
   done.then(() => {
     clearTimeout(timer)
     setTimeout(() => server.close(), 10_000).unref?.()
@@ -1214,7 +1255,7 @@ async function ownSignIn() {
     callback: async () => {
       const cli = await readCLI()
       const c = cli ?? (await readIDE())
-      if (!c) return { type: "failed" }
+      if (!c) return { type: "failed", error: "neither kiro-cli nor the Kiro IDE is signed in" }
       const who = fresh(c) ? await whoIs(c) : { email: "", plan: "" }
       return { type: "success", provider: ID, refresh: "", access: "", expires: 0, source: cli ? "kiro-cli" : "kiro-ide",
         accountId: who.email || (cli ? "kiro-cli's account" : "Kiro IDE's account"), ...(who.plan ? { plan: who.plan } : {}) }
@@ -1255,6 +1296,16 @@ export async function KiroAuthPlugin({ client } = {}) {
           },
         }
       },
+      // the account's credits, as magpie's built-in showed them
+      async usage(getAuth) {
+        const auth = await getAuth()
+        if (!(auth?.type === "api" && auth.key) && auth?.type !== "oauth") return { error: "not signed in" }
+        try {
+          return usageOf(await usageLimits(await creds(auth), true))
+        } catch (e) {
+          return { error: e.message }
+        }
+      },
       methods: [
         { type: "oauth", label: "Kiro (Google, GitHub, AWS Builder ID, IAM Identity Center)", authorize: browserSignIn },
         { type: "oauth", label: "Kiro CLI's or Kiro IDE's sign-in", authorize: ownSignIn },
@@ -1289,4 +1340,4 @@ export async function KiroAuthPlugin({ client } = {}) {
 }
 
 // for tests
-export const _internal = { buildKiro, events, reply, failure, toolID, thinking, readCLI, readIDE, regionOf, planName, frames }
+export const _internal = { usageOf, buildKiro, events, reply, failure, toolID, thinking, readCLI, readIDE, regionOf, planName, frames }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@magpie-community/opencode-kiro-auth",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Kiro subscriptions (Free, Pro, Pro+, Power) in OpenCode and magpie",
   "type": "module",
   "main": "./index.mjs",