@magpie-community/opencode-zcode-auth 0.1.6 → 0.1.8

package/README.md

@@ -17,7 +17,14 @@ over Anthropic's Messages API. Provider id: `zcode`.
   - If the account has no plan of its own, it looks for a seat on a team's
     plan and uses that team project's `zcode-team-api-key`.
   - Failing both, it uses ZCode's free **Start Plan**, served by zcode.z.ai
-    with ZCode's session token and ZCode's own headers.
+    with ZCode's session token.
+  - zcode.z.ai serves the Start Plan only to requests that look like the
+    ZCode client's own, and refuses others with HTTP 405 "request has been
+    blocked due to unusual activity". So a Start Plan request goes as
+    ZCode's does: ZCode's headers, ZCode's system prompt ahead of the
+    agent's, the day in a `<system-reminder>` turn before the agent's
+    turns, and the sign-in's device id in `metadata.user_id`. GLM Coding
+    Plan and team requests go as the agent sent them.
   - The Start Plan's token can't be refreshed. When it runs out, sign in
     again.
 - **ZCode app's sign-in**
@@ -47,12 +54,42 @@ Before each request, the plugin checks which plan the account is on, and
 asks again every 10 minutes:
 
 - If the key's account has a GLM Coding Plan
-  (`/api/biz/subscription/list`), requests go to that plan.
-- Otherwise they go to the Start Plan.
+  (`/api/biz/subscription/list`), its requests go to that plan —
+  except for a model the account's gift plans also serve, which goes to
+  the gift first (see below).
+- Otherwise they go to the gift plans.
 
 Every request carries the key as both `x-api-key` and
 `Authorization: Bearer`.
 
+**Spending the gift first.** An account with a GLM Coding Plan can also
+hold gift plans from ZCode activities, ones that expire on a fixed date.
+Their allowance would lapse unused if every request went to the coding
+plan, so the plugin reads the gift's balance (per the sign-in's session,
+kept for 10 minutes) and sends a request for a model the gift still has
+quota to spend to the gift first, dressed as ZCode's own. A bucket is
+judged spent only by its `remaining_units` — a missing one is unknown,
+not spent — and a model whose every bucket says none goes straight to
+the coding plan without asking. When the gift can't serve the request —
+any answer 400 or over (its 405, its 401, its 5xx), or its quota error
+(code 1113 "Insufficient balance or no resource package", code 1005
+"exceed quota limit", which it can answer inside an HTTP 200) — the
+plugin replays the request once to the coding plan, as the agent sent
+it. Only a quota answer keeps that model off the gift for a minute; a
+refusal that says nothing about quota is asked again next request. Every
+429 counts as spent, Z.ai's 1302/1303 rate limits among them. A stream's
+body is never read or buffered on the way.
+
+## The usage card
+
+An account with both plans shows both allowances: the coding plan's
+windows first, then the gift's buckets, each with its own name, models
+and end. The card's plan and term stay the coding plan's. A gift read
+that fails leaves just the coding card — magpie hides a card's windows
+behind an error, so a gift hiccup must not black out a working Coding
+Plan. A gift that goes out of date while requests use it simply stops
+showing.
+
 ## Models
 
 The models are the ones ZCode offers for the account's plan. They come

package/index.mjs

@@ -18,8 +18,9 @@
 
 import { createDecipheriv, createHash } from "node:crypto"
 import { existsSync, readFileSync, readdirSync } from "node:fs"
-import { homedir, userInfo } from "node:os"
-import { join } from "node:path"
+import { arch, homedir, release, userInfo } from "node:os"
+import { basename, join } from "node:path"
+import PROMPT from "./prompt.mjs"
 
 const PROVIDER = "zcode"
 const ZCODE = "https://zcode.z.ai"
@@ -129,48 +130,103 @@ const num = (v) => {
   return typeof n === "number" && Number.isFinite(n) ? n : undefined
 }
 
-// startPlan is the Start Plan the account has now ({name, until}), as
-// ZCode reads its balance: an "active" plan past its end is over.
+// startPlan is the gift an account can spend now ({models}), as ZCode reads
+// its balance: several plans can be active at once, and a model routes to
+// the gift when any of them has a live bucket serving it — a bucket past
+// its plan's own end, or one whose remaining_units says none, is skipped
+// rather than asked and refused. models are those live buckets' names.
 async function startPlan(jwt, device) {
   if (!jwt) throw new Error("not signed in to ZCode")
   if (jwtExpired(jwt)) throw new Error(EXPIRED)
   const b = (await call("GET", `${ZCODE}/api/v1/zcode-plan/billing/balance?app_version=${APP_VERSION}`, { auth: "Bearer " + jwt, device })) ?? {}
   const now = num(b.server_time) > 0 ? num(b.server_time) : Date.now() / 1000
-  const isStart = (x) => x.includes("start-plan") || x.includes("start plan")
-  for (const p of b.plans ?? []) {
-    if (String(p.status ?? "").trim().toLowerCase() !== "active") continue
-    const end = num(p.ends_at)
-    if (end > 0 && end <= now) continue
-    const id = String(p.plan_id ?? "").trim().toLowerCase()
-    const n = String(p.name ?? "").trim().toLowerCase()
-    if ((id || n) && !isStart(id) && !isStart(n)) continue
-    return { name: first(p.name, "Start Plan"), until: end > 0 ? end * 1000 : 0 }
+  const same = (x, p) => (x.user_plan_id && p.user_plan_id ? x.user_plan_id === p.user_plan_id : String(x.plan_id ?? "") === String(p.plan_id ?? ""))
+  const plans = Array.isArray(b.plans) ? b.plans : []
+  const balances = Array.isArray(b.balances) ? b.balances : []
+  const ms = new Set()
+  let any = false
+  let name = ""
+  for (const p of plans) {
+    const st = String(p?.status ?? "").trim().toLowerCase()
+    const end = num(p?.ends_at)
+    if (st !== "active" || (end > 0 && end <= now)) continue
+    any = true
+    if (!name) name = String(p?.name ?? "").trim()
+    for (const x of balances) {
+      if (!same(x, p)) continue
+      const exp = num(x.expires_at)
+      if (exp > 0 && exp <= now) continue
+      // a bucket spends when it has quota: only remaining_units says so, and
+      // a model whose every bucket says none drops out of models, so its
+      // requests go to the coding plan without first asking the gift
+      if ((num(x.remaining_units) ?? 1) <= 0) continue
+      for (const c of Array.isArray(x.capabilities) ? x.capabilities : []) {
+        const m = String(c ?? "").trim().replace(/^model:/, "").trim()
+        if (m) ms.add(m)
+      }
+    }
   }
-  return null
+  return any ? { name: name || "Start Plan", models: [...ms] } : null
 }
 
-// onStart says whether an account's requests go to the Start Plan: when it
-// has ZCode's token and no coding plan key, or a key whose account has no
-// coding plan. Asked again after 10 minutes (a minute when unsure).
+// plansOf says where an account's requests go and what more it can spend:
+// {start, gift}. start is what onStart said: an account with no coding plan
+// sends every request to the gift and reads no balance to do it; a team's
+// seat, a sign-in with no ZCode token and a key whose plan-list read found
+// no gift stay on the coding plan. gift is what a coding account also holds:
+// the gift plans ({models}), and its requests spend the gift for a model it
+// serves, until it says it is spent (see blocked). Asked again after 10
+// minutes (a minute when unsure — a plan-list read that failed is retried
+// sooner; a gift read that failed simply leaves gift null until the next
+// ask).
 const routes = new Map()
-async function onStart(s) {
-  if (!s.jwt || isTeam(s)) return false // a team's seat is on the team's plan
-  if (!s.key) return true
+async function plansOf(s) {
+  if (isTeam(s) || !s.jwt) return { start: false, gift: null }
+  if (!s.key) return { start: true, gift: null }
   const id = s.key + "\0" + s.jwt
   const r = routes.get(id)
-  if (r && Date.now() - r.at < r.ttl) return r.start
+  if (r && Date.now() - r.at < r.ttl) return r
   let start = false
+  let gift = null
   let sure = false
   try {
-    start = !(await plan(s))
+    if (!(await plan(s))) start = true
+    else gift = await startPlan(s.jwt, s.device).catch(() => null)
     sure = true
   } catch {
+    // the plan list could not be read: a key account takes the dual route
+    // (start stays false, the gift is kept, a minute's ttl), so a gift
+    // refusal still replays to the coding plan rather than blacking out
     try {
-      if (await startPlan(s.jwt, s.device)) start = sure = true
+      gift = await startPlan(s.jwt, s.device)
+      sure = false
     } catch {}
   }
-  routes.set(id, { start, at: Date.now(), ttl: sure ? 600_000 : 60_000 })
-  return start
+  const out = { start, gift, at: Date.now(), ttl: sure ? 600_000 : 60_000 }
+  routes.set(id, out)
+  return out
+}
+
+// blocked remembers the models a gift said it has none left for: a key+jwt+
+// model for a minute, so one turn that is refused does not ask again.
+const blocked = new Map()
+
+// giftServes: the gift's buckets name their models lowercased ("glm-5.3-
+// flash"); what is asked for arrives as the plan lists it ("GLM-5.3-Flash").
+const giftServes = (gift, m) => (gift.models ?? []).some((x) => x.toLowerCase() === m.toLowerCase())
+
+// spentUp is the gift saying it has nothing left to spend: its own no-
+// resource-package code 1113, its "exceed quota limit" 1005 — a spent gift
+// can answer a 200 with a JSON body carrying one of those codes. Every 429
+// counts as spent, Z.ai's 1302/1303 rate limits among them.
+function spentUp(text, status) {
+  if (status === 429) return true
+  let b
+  try { b = JSON.parse(text) } catch { return false }
+  if (!b || typeof b !== "object") return false
+  const code = (v) => String(v ?? "")
+  const c1 = code(b.code), c2 = code(b.error?.code)
+  return c1 === "1113" || c1 === "1005" || c2 === "1113" || c2 === "1005"
 }
 
 // ---- keys ------------------------------------------------------------------------
@@ -365,10 +421,12 @@ function startNum(v) {
   return undefined
 }
 
-// startUsage is the Start Plan's allowance: a window for each of its
-// buckets, a model's tokens for the day or for the plan's time. A plan
-// still "active" past its end is over, and its buckets are left out.
-async function startUsage(s) {
+// giftOf reads the balance and returns the gift's card fields: its plans and
+// windows. Several plans can be active at once, so every live plan's buckets
+// show, each keeping its own models and end; the plan and end put on the
+// card (for a gift-only account) are the soonest-ending live plan's. null
+// when there is no live plan to show.
+async function giftOf(s) {
   if (!s.jwt) throw new Error("not signed in to ZCode")
   if (jwtExpired(s.jwt)) throw new Error(EXPIRED)
   const b = (await call("GET", `${ZCODE}/api/v1/zcode-plan/billing/balance?app_version=${APP_VERSION}`, { auth: "Bearer " + s.jwt, device: s.device })) ?? {}
@@ -381,7 +439,7 @@ async function startUsage(s) {
     if (end > 0 && end <= now && p.status.trim().toLowerCase() === "active") p.status = "expired"
     if (p.status.trim().toLowerCase() === "expired") over.add(p.user_plan_id + "\0" + p.plan_id)
   }
-  const same = (x, p) => (x.user_plan_id && p.user_plan_id ? x.user_plan_id === p.user_plan_id : x.plan_id === p.plan_id)
+  const same = (x, p) => (x.user_plan_id && p.user_plan_id ? x.user_plan_id === p.user_plan_id : String(x.plan_id ?? "") === String(p.plan_id ?? ""))
   const balances = (Array.isArray(b.balances) ? b.balances : [])
     .map((x) => ({ ...x, user_plan_id: str(x?.user_plan_id), plan_id: str(x?.plan_id) }))
     .filter((x) => {
@@ -394,32 +452,41 @@ async function startUsage(s) {
       return keep
     })
 
+  // every live plan counts as the gift: no plan-name filter decides it,
+  // the balance says what the session can spend
   let active = null
   for (const p of plans) {
     if (p.status.trim().toLowerCase() !== "active") continue
-    const id = p.plan_id.trim().toLowerCase(), n = str(p.name).trim().toLowerCase()
-    const isStart = (x) => x.includes("start-plan") || x.includes("start plan")
-    if ((id || n) && !isStart(id) && !isStart(n)) continue
     const end = startNum(p.ends_at)
-    active = { name: first(str(p.name), "Start Plan"), until: end > 0 ? new Date(Math.trunc(end) * 1000).toISOString() : "" }
-    break
+    if (end > 0 && end <= now) continue
+    if (active && !(end > 0 && (!startNum(active.end) || end < startNum(active.end)))) continue
+    active = { name: first(str(p.name), "Start Plan"), end, until: end > 0 ? new Date(Math.trunc(end) * 1000).toISOString() : "" }
   }
-  if (!active) return { error: "this account has no GLM Coding Plan, and ZCode's Start Plan has ended or was never started" }
+  if (!active) return null
   const out = { plan: active.name, windows: [] }
   if (active.until) Object.assign(out, { until: active.until, renew: "off" })
+  const live = plans.filter((p) => {
+    if (p.status.trim().toLowerCase() !== "active") return false
+    const end = startNum(p.ends_at)
+    return !(end > 0 && end <= now)
+  })
   for (const x of balances) {
+    const owner = live.find((p) => same(x, p))
+    if (!owner) continue
+    const exp = startNum(x.expires_at)
+    if (exp > 0 && exp <= now) continue // a bucket past its own end is over
     const total = startNum(x.total_units), left = startNum(x.remaining_units)
     let used = startNum(x.used_units)
     if (total === undefined && used === undefined && left === undefined) continue
     const models = (Array.isArray(x.capabilities) ? x.capabilities : [])
       .map((c) => String(c ?? "").trim().replace(/^model:/, "").trim()).filter(Boolean)
-    const w = { name: first(str(x.show_name), models.join(", "), "Credits"), used: 0 }
+    const base = first(str(x.show_name), models.join(", "), "Credits")
+    const w = { name: base, used: 0, _plan: str(owner.name) }
     if (used === undefined) used = total !== undefined && left !== undefined ? total - left : 0
     if (total > 0) {
       w.used = (100 * used) / total
       w.display = `${compact(used)} / ${compact(total)}`
     }
-    const exp = startNum(x.expires_at)
     if (exp > 0) w.resetsAt = new Date(Math.trunc(exp) * 1000).toISOString()
     let span = 0
     for (const p of plans) {
@@ -439,6 +506,37 @@ async function startUsage(s) {
   return out
 }
 
+// startUsage is a gift-only account's card: its buckets, or the words
+// for having none to show. The card's header already names the plan, so
+// the windows keep their own names.
+const NO_START = "this account has no GLM Coding Plan, and ZCode's Start Plan has ended or was never started"
+async function startUsage(s) {
+  const g = await giftOf(s)
+  if (!g) return { error: NO_START }
+  for (const w of g.windows) delete w._plan
+  return g
+}
+
+// dualUsage is an account with both a Coding Plan and a gift: its card shows
+// both allowances, the Coding windows first and the gift's buckets after,
+// each keeping its own models and end. The gift's windows carry their plan's
+// name as a prefix so they are not mistaken for the coding plan's own. A
+// gift read that fails leaves just the Coding card: magpie hides a card's
+// windows behind an error and leaves it out of the menu bar and auto-switch,
+// so erroring here would hide a working Coding Plan over a gift hiccup.
+async function dualUsage(s) {
+  const out = await codingUsage(s)
+  try {
+    const g = await giftOf(s)
+    if (g) out.windows = [...out.windows, ...g.windows.map((w) => {
+      const p = w._plan
+      delete w._plan
+      return p && p !== w.name ? { ...w, name: p + " · " + w.name } : w
+    })]
+  } catch {}
+  return out
+}
+
 // periodOf reads an entitlement's period: "daily", "weekly", "monthly".
 function periodOf(p) {
   p = p.toLowerCase()
@@ -988,6 +1086,26 @@ const entry = (m) => ({
 })
 
 // model is a model as OpenCode's provider hands it to provider.models.
+// modelOf reads the model a Messages body asks for.
+function modelOf(text) {
+  try {
+    const m = JSON.parse(text)?.model
+    return typeof m === "string" ? m : ""
+  } catch {
+    return ""
+  }
+}
+
+// kept passes a response on as it came, the account kept: the built-in
+// never marked a ZCode account lapsed nor cleared one.
+function kept(res, text) {
+  const h = new Headers(res.headers)
+  h.delete("content-length")
+  h.delete("content-encoding")
+  h.set("X-Magpie-Sign-In", "kept")
+  return new Response(text !== undefined ? text : res.body, { status: res.status, statusText: res.statusText, headers: h })
+}
+
 function model(m, url) {
   const e = entry(m)
   return {
@@ -1014,6 +1132,128 @@ function model(m, url) {
   }
 }
 
+// ---- the Start Plan's requests ----------------------------------------------------
+// zcode.z.ai serves the Start Plan only to what looks like ZCode's own
+// request, and turns others away with 405 "request has been blocked due to
+// unusual activity" (code 3012). A Start Plan request goes as the plugin
+// the plan serves (ARNO's "Freeflow", provider zcode-start) sends it: its
+// headers (startHeaders) and its body (dress). The texts are prompt.mjs.
+
+const osVersion = () => `${process.platform} ${release()} ${arch()}`
+const osCategory = () => (process.platform === "darwin" ? "macos" : process.platform === "win32" ? "windows" : "linux")
+
+function intl() {
+  try {
+    return Intl.DateTimeFormat().resolvedOptions()
+  } catch {
+    return {}
+  }
+}
+
+// startHeaders puts ZCode's headers on a Start Plan request, jwt its key.
+function startHeaders(h, jwt) {
+  h.delete("x-api-key")
+  h.delete("authorization")
+  h.delete("X-Device-Mid")
+  h.set("Authorization", "Bearer " + jwt)
+  h.set("anthropic-version", "2023-06-01")
+  h.set("HTTP-Referer", ZCODE)
+  h.set("User-Agent", UA + " ai-sdk/anthropic/3.0.81")
+  h.set("X-ZCode-App-Version", APP_VERSION)
+  h.set("X-Title", "Z Code@cli")
+  h.set("X-Release-Channel", "production")
+  const o = intl()
+  h.set("X-Client-Language", o.locale || "unknown")
+  if (o.timeZone) h.set("X-Client-Timezone", o.timeZone)
+  h.set("X-ZCode-Agent", "glm")
+  h.set("X-Platform", platform())
+  h.set("X-Os-Category", osCategory())
+  h.set("X-Os-Version", osVersion())
+  h.set("x-request-id", uuid())
+  h.set("x-zcode-session-type", "main")
+  h.set("x-zcode-trace-id", uuid())
+}
+
+// environment is the prompt's environment section; provider and model
+// name the powered-by line, left out with no model.
+function environment(provider, model) {
+  const e = PROMPT.environment
+  const sh = process.env.SHELL || process.env.ComSpec
+  const lines = [
+    e.heading,
+    e.invokedLine,
+    `- ${e.cwdLabel}: ${process.cwd()}`,
+    `- ${e.gitLabel}: ${e.gitNo}`,
+    `- ${e.platformLabel}: ${process.platform}`,
+    `- ${e.shellLabel}: ${sh ? basename(sh) : "unknown"}`,
+    `- ${e.osVersionLabel}: ${osVersion()}`,
+  ]
+  if (model) lines.push(e.poweredByLine.replace("{provider}", provider).replace("{model}", model))
+  return lines.join("\n")
+}
+
+// today is the local date, YYYY-MM-DD.
+function today(now = new Date()) {
+  const p = (n) => String(n).padStart(2, "0")
+  return `${now.getFullYear()}-${p(now.getMonth() + 1)}-${p(now.getDate())}`
+}
+
+// contextPrefix is the user turn ZCode puts before the agent's: the day in
+// a <system-reminder>, the tags hugging the text.
+function contextPrefix() {
+  const c = PROMPT.context
+  const body = [c.intro, c.currentDateHeading + "\n" + c.currentDateLine.replace("{date}", today()), "", c.outro].join("\n")
+  return { role: "user", content: [{ type: "text", text: "<system-reminder>" + body + "</system-reminder>" }] }
+}
+
+// dress is a Messages request body as ZCode sends it: ZCode's three cached
+// system blocks, then the agent's own system text uncached; the context
+// prefix as the first turn; no cache mark in the turns but on the last
+// block of the last; none on the tools; metadata.user_id naming the
+// sign-in's device. site names the powered-by line's provider. A body that
+// isn't a Messages request goes as it is.
+function dress(text, site, device) {
+  if (typeof text !== "string" || !text.includes("messages")) return text
+  let b
+  try {
+    b = JSON.parse(text)
+  } catch {
+    return text
+  }
+  if (!b || typeof b !== "object" || !Array.isArray(b.messages)) return text
+  const cached = (t) => ({ type: "text", text: t, cache_control: { type: "ephemeral" } })
+  const provider = site === "bigmodel" ? "bigmodel-api" : "zai-api"
+  const model = typeof b.model === "string" ? b.model : ""
+  const own = []
+  if (typeof b.system === "string") {
+    if (b.system.trim()) own.push({ type: "text", text: b.system })
+  } else if (Array.isArray(b.system)) {
+    for (const x of b.system) if (x?.type === "text" && typeof x.text === "string" && x.text) own.push({ type: "text", text: x.text })
+  }
+  b.system = [
+    cached(PROMPT.cliPrefix),
+    cached(PROMPT.stableSections.join("\n\n")),
+    cached("\n\n" + [PROMPT.beforeEnvironment, environment(provider, model), PROMPT.afterEnvironment].join("\n\n")),
+    ...own,
+  ]
+  b.messages = [contextPrefix(), ...b.messages]
+  let last = null
+  for (const m of b.messages) {
+    if (!m || m.role === "system") continue
+    last = m
+    if (Array.isArray(m.content)) for (const x of m.content) if (x && typeof x === "object") delete x.cache_control
+  }
+  if (last && typeof last.content === "string") last.content = [cached(last.content)]
+  else if (last && Array.isArray(last.content) && last.content.length) {
+    const x = last.content[last.content.length - 1]
+    if (x && typeof x === "object" && !x.cache_control) x.cache_control = { type: "ephemeral" }
+  }
+  if (Array.isArray(b.tools)) for (const t of b.tools) if (t && typeof t === "object") delete t.cache_control
+  const meta = b.metadata && typeof b.metadata === "object" ? b.metadata : {}
+  b.metadata = { ...meta, user_id: JSON.stringify({ device_id: device, account_uuid: "", session_id: "" }) }
+  return JSON.stringify(b)
+}
+
 // ---- the plugin ------------------------------------------------------------------
 
 export async function ZCodeAuthPlugin({ client }) {
@@ -1045,7 +1285,7 @@ export async function ZCodeAuthPlugin({ client }) {
       async models(provider, { auth } = {}) {
         const s = stateOf(auth)
         if (!s) return provider.models
-        const start = await onStart(s).catch(() => false)
+        const start = await plansOf(s).then((r) => r.start).catch(() => false)
         const base = start ? START_BASE : s.base
         let ms = modelsOf(await zcodeConfig().catch(() => null), planID(base))
         // ZCode's config can't be had: its table stands in, and magpie keeps
@@ -1073,14 +1313,7 @@ export async function ZCodeAuthPlugin({ client }) {
             if (!s) throw new Error("not signed in to ZCode")
             let url = input instanceof Request ? input.url : String(input)
             const opts = input instanceof Request ? { method: input.method, headers: input.headers, body: input.body, signal: input.signal, duplex: "half", ...init } : { ...init }
-            const start = await onStart(s)
-            // the request goes to the plan the account is on, wherever it was made for
-            for (const b of [START_BASE, ZAI_BASE, BIGMODEL_BASE]) {
-              if (url.startsWith(b)) {
-                url = (start ? START_BASE : s.base) + url.slice(b.length)
-                break
-              }
-            }
+            const { start: alone, gift } = await plansOf(s)
             let key = s.key
             if (isTeam(s) && !key) {
               key = await teamKey(s)
@@ -1089,30 +1322,79 @@ export async function ZCodeAuthPlugin({ client }) {
                 await client?.auth?.set?.({ path: { id }, body: next }).catch?.(() => {})
               }
             }
-            const h = new Headers(opts.headers)
+            // a dual account's body is read ahead of sending, so a gift that
+            // says it is spent can spend nothing and replay what was asked for
+            let start = alone
+            let orig
+            if (!alone && gift && opts.body != null) {
+              orig = opts.body = typeof opts.body === "string" ? opts.body : await new Response(opts.body).text()
+              delete opts.duplex
+              const m = modelOf(orig)
+              // a ZCode sign-in past its end reaches no gift: the coding plan
+              // takes the request, as it did before the gift was known
+              start = !!m && !jwtExpired(s.jwt) && giftServes(gift, m) && Date.now() >= (blocked.get(s.key + "\0" + s.jwt + "\0" + m) ?? 0)
+            }
+            const swap = (base) => {
+              for (const b of [START_BASE, ZAI_BASE, BIGMODEL_BASE]) {
+                if (url.startsWith(b)) {
+                  url = base + url.slice(b.length)
+                  break
+                }
+              }
+            }
+            let h = new Headers(opts.headers)
             if (start) {
               if (jwtExpired(s.jwt)) throw new Error(EXPIRED)
-              key = s.jwt
-              // ZCode names itself on a Start Plan request
-              h.set("User-Agent", UA)
-              h.set("X-ZCode-App-Version", APP_VERSION)
-              h.set("X-Title", "Z Code@electron")
-              h.set("HTTP-Referer", ZCODE)
-              h.set("X-Platform", platform())
-              h.set("X-Device-Mid", s.device)
+              if (!s.jwt) throw new Error("ZCode's sign-in has no key; sign in again")
+              swap(START_BASE)
+              // the Start Plan is served only to what looks like ZCode's own request
+              startHeaders(h, s.jwt)
+              if (opts.body != null) {
+                const text = orig ?? (typeof opts.body === "string" ? opts.body : await new Response(opts.body).text())
+                opts.body = dress(text, s.site, s.device)
+                delete opts.duplex
+                h.delete("content-length")
+              }
+            } else {
+              if (!key) throw new Error("ZCode's sign-in has no key; sign in again")
+              swap(s.base)
+              h.delete("authorization")
+              h.set("x-api-key", key)
+              h.set("Authorization", "Bearer " + key)
+            }
+            let res = await fetch(url, { ...opts, headers: h })
+            // a dual account's gift answer that can't serve the request goes
+            // to its coding plan instead, as the agent sent it: any status
+            // 400 or over (its 405, its 401, its 5xx — a paid plan is waiting,
+            // and a gift that keeps refusing would otherwise black out the
+            // account), and its quota answers inside a 200. Only a quota
+            // answer keeps that model off the gift for a minute: a refusal
+            // that says nothing about quota is asked again next request.
+            // Every 429 counts as spent, Z.ai's 1302/1303 rate limits among
+            // them — replaying one to the coding plan is the good answer, and
+            // the minute's block keeps a rate limit from being knocked twice.
+            // A stream's body is never read here.
+            if (start && !alone) {
+              const json = (res.headers.get("content-type") ?? "").includes("application/json")
+              if (res.status >= 400 || json) {
+                const text = await res.text()
+                const spent = spentUp(text, res.status)
+                if (!spent && res.status < 400) return kept(res, text)
+                if (spent) {
+                  for (const [k, until] of blocked) if (Date.now() >= until) blocked.delete(k)
+                  blocked.set(s.key + "\0" + s.jwt + "\0" + modelOf(orig), Date.now() + 60_000)
+                }
+                swap(s.base)
+                h = new Headers(opts.headers)
+                h.delete("authorization")
+                h.set("x-api-key", key)
+                h.set("Authorization", "Bearer " + key)
+                res = await fetch(url, { ...opts, body: orig, headers: h })
+              }
             }
-            if (!key) throw new Error("ZCode's sign-in has no key; sign in again")
-            h.delete("authorization")
-            h.set("x-api-key", key)
-            h.set("Authorization", "Bearer " + key)
-            const res = await fetch(url, { ...opts, headers: h })
             // the plan's answer goes on as it came, the account kept: the
             // built-in never marked a ZCode account lapsed nor cleared one
-            const kept = new Headers(res.headers)
-            kept.delete("content-length")
-            kept.delete("content-encoding")
-            kept.set("X-Magpie-Sign-In", "kept")
-            return new Response(res.body, { status: res.status, statusText: res.statusText, headers: kept })
+            return kept(res)
           },
         }
       },
@@ -1144,7 +1426,9 @@ export async function ZCodeAuthPlugin({ client }) {
             }
             return saved(await teamUsage(s, key))
           }
-          if (await onStart(s)) return saved(await startUsage(s))
+          const { start, gift } = await plansOf(s)
+          if (start) return saved(await startUsage(s))
+          if (gift) return saved(await dualUsage(s))
           return saved(await codingUsage(s))
         } catch (e) {
           return saved({ error: e?.message ?? String(e) })
@@ -1175,4 +1459,4 @@ export async function ZCodeAuthPlugin({ client }) {
 }
 
 // for tests
-export const _internal = { entry, limitWindows, termOf, startUsage, routes, teamKeys, ownSignIn, stateOf }
+export const _internal = { entry, limitWindows, termOf, startUsage, routes, blocked, plansOf, giftOf, dualUsage, spentUp, giftServes, modelOf, teamKeys, ownSignIn, stateOf, dress, PROMPT }

package/package.json

@@ -1,11 +1,11 @@
 {
   "name": "@magpie-community/opencode-zcode-auth",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "OpenCode provider plugin: signs in to Z.ai's or BigModel's (智谱) GLM Coding Plan as ZCode does, including team seats and ZCode's Start Plan",
   "type": "module",
   "main": "./index.mjs",
   "exports": "./index.mjs",
-  "files": ["index.mjs", "README.md"],
+  "files": ["index.mjs", "prompt.mjs", "README.md"],
   "keywords": ["opencode", "opencode-plugin", "magpie", "zcode", "z.ai", "bigmodel", "glm", "coding-plan"],
   "license": "MIT",
   "repository": {

package/prompt.mjs

@@ -0,0 +1,34 @@
+// ZCode's own prompt texts, which zcode.z.ai looks for on a Start Plan
+// request, byte for byte as the plugin the Start Plan serves (ARNO's
+// "Freeflow", provider zcode-start) sends them. Taken from magpie's
+// internal/provider/zcode_prompt.json (901fd87e), ZCode 3.14.3's texts.
+// SHA-256, first 8 bytes: cliPrefix 46dd360a22c87a92, stableSections[0]
+// 3f21ff9a88a03a76, [1] 39a1e86c93452c01, beforeEnvironment
+// bbdc66b399d297ff, afterEnvironment 1732b7f098a925d7 (start.test.mjs
+// checks them).
+export default {
+  "cliPrefix": "You are ZCode, an interactive coding agent",
+  "stableSections": [
+    "\nYou are an interactive ZCode agent that helps users with software engineering tasks.\n\nIMPORTANT: Assist with authorized security testing, defensive security, CTF challenges, and educational contexts. Refuse requests for destructive techniques, DoS attacks, mass targeting, supply chain compromise, or detection evasion for malicious purposes. Dual-use security tools (C2 frameworks, credential testing, exploit development) require clear authorization context: pentesting engagements, CTF competitions, security research, or defensive use cases.\n\n# Harness\n- Text you output outside of tool use is displayed to the user as Github-flavored markdown in a terminal.\n- Tools run behind a user-selected permission mode; a denied call means the user declined it — adjust, don't retry verbatim.\n- The system may send updates, reminders, or modifications to rules via mid-conversation system turns. These are system-controlled, unlike function results. Hooks may intercept tool calls; treat hook output as user feedback.\n- Prefer the dedicated file/search tools over shell commands when one fits. Independent tool calls can run in parallel in one response.\n- Reference code as `file_path:line_number` — it's clickable.",
+    "# ZCode Desktop Context\n\n### Files & URLs\n- Return local web URLs as Markdown links (e.g., [label](http://127.0.0.1:8080)).\n- File should be an absolute path or include the workspace folder segment so it can be resolved relative to the workspace.\n- Unless otherwise specified, return local file references as Markdown links (e.g., [name.md](/absolute/path/to/name.md)).\n\n### Inline Code Comments\n- Use the ::code-comment{...} directive when you need to attach feedback directly to specific code lines.\n- Emit one directive per inline comment; emit none when there are no actionable inline comments.\n- Required attributes: title (short label), body (one-paragraph explanation), file (path to the file).\n- Optional attributes: start, end (1-based line numbers), priority (0-3).\n- file should be an absolute path or include the workspace folder segment so it can be resolved relative to the workspace.\n- Keep line ranges tight; end defaults to start.\n- Example: ::code-comment{title=\"[P2] Off-by-one\" body=\"Loop iterates past the end when length is 0.\" file=\"/path/to/foo.ts\" start=10 end=11 priority=2}"
+  ],
+  "beforeEnvironment": "# Communicating with the user\n\nYour text output is what the user reads; they usually can't see your thinking or the raw tool results. Write it for a teammate who stepped away and is catching up, not for a log file: they don't know the codenames or shorthand you created along the way, and they didn't watch your process unfold. Before your first tool call, say in a sentence what you're about to do; while working, give brief updates when you find something load-bearing or change direction.\n\nText you write between tool calls may not be shown to the user. Everything the user needs from this turn — answers, summaries, findings, conclusions, deliverables — must be in the final text message of your turn, with no tool calls after it. Keep text between tool calls to brief status notes. If something important appeared only mid-turn or in your thinking, restate it in that final message.\n\nLead with the outcome. Your first sentence after finishing should answer \"what happened\" or \"what did you find\" — the thing the user would ask for if they said \"just give me the TLDR.\" Supporting detail and reasoning come after, for readers who want them.\n\nBeing readable and being concise are different things, and readable matters more. If the user has to reread your summary or ask you to explain, any time saved by brevity is gone. The way to keep output short is to be selective about what you include (drop details that don't change what the reader would do next), not to compress the writing into fragments, abbreviations, arrow chains like `A → B → fails`, or jargon. What you do include, write in complete sentences with the technical terms spelled out. Don't make the reader cross-reference labels or numbering you invented earlier; say what you mean in place.\n\nMatch the response to the question: a simple question gets a direct answer in prose, not headers and sections. Use tables only for short enumerable facts, with explanations in the surrounding prose rather than the cells. Calibrate to the user — a bit tighter for an expert, more explanatory for someone newer.\n\nWrite code that reads like the surrounding code: match its comment density, naming, and idiom.\nOnly write a code comment to state a constraint the code itself can't show — never to say where it came from, what the next line does, or why your change is correct; that's you talking to the reviewer, not the next reader, and it's noise the moment the PR merges.\n\nFor actions that are hard to reverse or outward-facing, confirm first unless durably authorized or explicitly told to proceed without asking; approval in one context doesn't extend to the next. Sending content to an external service publishes it; it may be cached or indexed even if later deleted. Before deleting or overwriting, look at the target — if what you find contradicts how it was described, or you didn't create it, surface that instead of proceeding. Report outcomes faithfully: if tests fail, say so with the output; if a step was skipped, say that; when something is done and verified, state it plainly without hedging.",
+  "afterEnvironment": "# Context management\nWhen the conversation grows long, some or all of the current context is summarized; the summary, along with any remaining unsummarized context, is provided in the next context window so work can continue — you don't need to wrap up early or hand off mid-task.\n\nWhen you have enough information to act, act. Do not re-derive facts already established in the conversation, re-litigate a decision the user has already made, or narrate options you will not pursue. If you are weighing a choice, give a recommendation, not an exhaustive survey\n\nYou are operating autonomously. The user is not watching in real time and cannot answer questions mid-task, so asking 'Want me to…?' or 'Shall I…?' will block the work. For reversible actions that follow from the original request, proceed without asking. Stop only for destructive actions or genuine scope changes the user must decide. Offering follow-ups after the task is done is fine; asking permission before doing the work is not.\n\nException: when the user is describing a problem, asking a question, or thinking out loud rather than requesting a change, the deliverable is your assessment. Report your findings and stop. Don't apply a fix until they ask for one.\n\nBefore ending your turn, check your last paragraph. If it is a plan, an analysis, a question, a list of next steps, or a promise about work you have not done ('I'll…', 'let me know when…'), do that work now with tool calls. That includes retrying after errors and gathering missing information yourself. Do not stop because the context or session is long. End your turn only when the task is complete or you are blocked on input only the user can provide.\n\nBefore running a command that changes system state — restarts, deletes, config edits — check that the evidence actually supports that specific action. A signal that pattern-matches to a known failure may have a different cause.",
+  "environment": {
+    "heading": "# Environment",
+    "invokedLine": "You have been invoked in the following environment:",
+    "cwdLabel": "Primary working directory",
+    "gitLabel": "Is a git repository",
+    "gitNo": "no",
+    "platformLabel": "Platform",
+    "shellLabel": "Shell",
+    "osVersionLabel": "OS Version",
+    "poweredByLine": "- You are powered by the model named {provider}/{model}."
+  },
+  "context": {
+    "intro": "As you answer the user's questions, you can use the following context:",
+    "outro": "      IMPORTANT: this context may or may not be relevant to your tasks. You should not respond to this context unless it is highly relevant to your task.",
+    "currentDateHeading": "# currentDate",
+    "currentDateLine": "Today's date is {date}."
+  }
+}