@magpie-community/opencode-workbuddy-auth 0.1.4 → 0.1.6

package/index.mjs

@@ -86,10 +86,11 @@ const api = (site) => site.endpoint + "/v2"
 // ---- WorkBuddy's API ------------------------------------------------------
 
 class WBError extends Error {
-  constructor(message, code, status) {
+  constructor(message, code, status, said) {
     super(message)
     this.code = code
     this.status = status
+    this.said = said // as magpie's built-in (wbCall) worded it
   }
 }
 
@@ -112,11 +113,13 @@ async function call(site, method, path, headers = {}, body) {
   try {
     env = JSON.parse(text)
   } catch {}
+  const code = Number.parseInt(env.code, 10) || 0
   if (!res.ok) {
     const why = env.msg || env.message || env.error?.message || text.slice(0, 200) || res.statusText
-    throw new WBError(`${site.name}: ${why} (HTTP ${res.status}${env.code ? ", code " + env.code : ""})`, env.code, res.status)
+    const said = code ? env.msg || `error ${code}` : env.msg || statusText(res.status)
+    throw new WBError(`${site.name}: ${why} (HTTP ${res.status}${env.code ? ", code " + env.code : ""})`, env.code, res.status, said)
   }
-  if (env.code) throw new WBError(`${site.name}: ${env.msg || "error"} (${env.code})`, env.code, res.status)
+  if (env.code) throw new WBError(`${site.name}: ${env.msg || "error"} (${env.code})`, env.code, res.status, env.msg || `error ${code}`)
   return env.data
 }
 
@@ -145,13 +148,14 @@ function stale(a) {
 
 // fresh is the sign-in with an access token that isn't about to end:
 // refreshed (and saved) when it is. A refresh that fails, or a refresh
-// token that has ended, leaves the access token there is.
+// token that has ended, leaves the access token there is. Its failures
+// are worded as the built-in's (wbFresh), "WorkBuddy" on both sites.
 async function fresh(site, client, a) {
   if (!stale(a)) return a
   const refreshable = a.refresh && !(a.refreshExpiresAt && Date.now() >= a.refreshExpiresAt)
   if (!refreshable) {
     if (a.access) return a
-    throw new Error(`this ${site.name} account is signed out; sign in again`)
+    throw new Error("this WorkBuddy account is signed out; sign in again")
   }
   try {
     const got = await call(site, "POST", "/v2/plugin/auth/token/refresh", {
@@ -159,13 +163,14 @@ async function fresh(site, client, a) {
       "X-Auth-Refresh-Source": "plugin",
       "X-Domain": domainOf(site, a),
     }, {})
-    if (!got?.accessToken) throw new Error(`${site.name} gave no refreshed token`)
+    if (!got?.accessToken) throw Object.assign(new Error("WorkBuddy gave no refreshed token"), { bare: true })
     const next = merge(a, got)
     await client?.auth?.set?.({ path: { id: site.id }, body: next })
     return next
   } catch (e) {
     if (a.access) return a
-    throw new Error(`${site.name} token refresh: ${e?.message ?? e}`)
+    if (e?.bare) throw e
+    throw new Error(`WorkBuddy token refresh: ${e?.said ?? e?.message ?? e}`)
   }
 }
 
@@ -460,18 +465,29 @@ function modelOf(site, provider, m) {
     variants: variants(m.efforts),
     // magpie's own: served at no cost to the plan's credits
     free: !!m.free,
+    // magpie's own: the credits a request costs, as a multiple (0: none)
+    rate: m.rate ?? 0,
   }
 }
 
 // liveModels asks WorkBuddy's product config for the plan's models: the
 // "cli" agent's list, each with its details. The CLI User-Agent picks
 // WorkBuddy's config (a bare WorkBuddy/<v> gets CodeBuddy IDE's, no cli).
+// creditsOf reads a model's credits, the multiple WorkBuddy's picker shows
+// by it: "x0.03", "0" or 0; "x0.00" is free. A rate it can't read (or
+// none, or an infinite one) is undefined: the model is neither free nor
+// rated. The same rule as magpie's built-in (wbCredits).
+function creditsOf(c) {
+  if (c == null || c === "") return undefined
+  const s = String(c).trim().toLowerCase().replace(/^x/, "")
+  const n = Number(s)
+  return s !== "" && Number.isFinite(n) && n >= 0 ? n : undefined
+}
+
 // freeCredits says whether a model's credits are none: "x0.00", "0" or 0.
 // A rate it can't read is not free.
 function freeCredits(c) {
-  if (c == null || c === "") return false
-  const s = String(c).trim().toLowerCase().replace(/^x/, "")
-  return s !== "" && Number.isFinite(Number(s)) && Number(s) === 0
+  return creditsOf(c) === 0
 }
 
 async function liveModels(site, a) {
@@ -495,7 +511,11 @@ async function liveModels(site, a) {
         m.context = d.maxInputTokens
         m.output = d.maxOutputTokens
         m.images = d.supportsImages === true
-        m.free = freeCredits(d.credits)
+        const r = creditsOf(d.credits)
+        if (r !== undefined) {
+          m.free = r === 0
+          m.rate = r
+        }
         const es = d.reasoning?.supportedEfforts ?? []
         if (es.length) {
           const canOff = d.reasoning?.canDisableThinking
@@ -570,20 +590,22 @@ function makePlugin(site) {
             let body = init?.body
             if (body === undefined && req) body = await req.clone().text()
             const res = await fetch(req ? req.url : input, { ...init, method: init?.method ?? req?.method, headers, body: withSystem(body) })
-            return res.status >= 400 ? explained(res) : res
+            return res.status >= 400 ? explained(res) : kept(res)
           },
         }
       },
-      // the account's credits, as magpie's built-in showed them
+      // the account's credits, as magpie's built-in showed them; each
+      // read keeps the sign-in (signIn), as the built-in's neither marked
+      // an account lapsed nor cleared it
       async usage(getAuth) {
         const auth = await getAuth()
-        if (auth?.type !== "oauth" || !(auth.access || auth.source)) return { error: "not signed in" }
+        if (auth?.type !== "oauth" || !(auth.access || auth.source)) return { error: "not signed in", signIn: "kept" }
         try {
           const a = await current(site, client, auth)
           // the meter has no /v2 prefix, on either site
-          return usageOf(await meter(site, a, "/billing/meter/get-user-resource-summary"), auth.plan)
+          return { ...usageOf(await meter(site, a, "/billing/meter/get-user-resource-summary"), auth.plan), signIn: "kept" }
         } catch (e) {
-          return { error: e?.message ?? String(e) }
+          return { error: e?.message ?? String(e), signIn: "kept" }
         }
       },
       methods: [
@@ -611,13 +633,24 @@ const REFUSED_HINT =
   "WorkBuddy refuses chats from Codex and Claude Code (their system prompt); use it from Hermes, OpenCode or Pi, or add another provider to this group"
 const REFUSED = /unapproved channel|illegal api invocation/i
 
+// kept is an answer that went through, saying the sign-in is kept: the
+// built-in never cleared a WorkBuddy account's mark, as it never set one.
+function kept(res) {
+  const headers = new Headers(res.headers)
+  headers.set("X-Magpie-Sign-In", "kept")
+  return new Response(res.body, { status: res.status, statusText: res.statusText, headers })
+}
+
 // explained is res with REFUSED_HINT added to that refusal's message.
+// Each refusal says the sign-in is kept: the built-in passed WorkBuddy's
+// 401 on and never marked a WorkBuddy account lapsed.
 async function explained(res) {
   const text = await res.text()
   const again = (b) => {
     const headers = new Headers(res.headers)
     headers.delete("content-length")
     headers.delete("content-encoding")
+    headers.set("X-Magpie-Sign-In", "kept")
     return new Response(b, { status: res.status, statusText: res.statusText, headers })
   }
   if (!REFUSED.test(text)) return again(text)
@@ -635,6 +668,7 @@ async function explained(res) {
   headers.set("content-type", "application/json")
   headers.delete("content-length")
   headers.delete("content-encoding")
+  headers.set("X-Magpie-Sign-In", "kept")
   return new Response(JSON.stringify({ error: { message: msg, type: "permission_error", code: null } }), { status: res.status, statusText: res.statusText, headers })
 }
 
@@ -642,4 +676,4 @@ export const WorkBuddyAuthPlugin = makePlugin(SITES.workbuddy)
 export const WorkBuddyAIAuthPlugin = makePlugin(SITES["workbuddy-ai"])
 
 // for tests
-export const _internal = { withSystem, usageOf, desktopHeld, explained, REFUSED_HINT, freeCredits }
+export const _internal = { withSystem, usageOf, desktopHeld, explained, REFUSED_HINT, freeCredits, creditsOf, fresh, SITES }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@magpie-community/opencode-workbuddy-auth",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "description": "OpenCode provider plugins for WorkBuddy's plans: WorkBuddy (CodeBuddy, copilot.tencent.com) and WorkBuddy AI (www.workbuddy.ai)",
   "type": "module",
   "main": "./index.mjs",