@magpie-community/opencode-trae-auth 0.1.3 → 0.1.5

package/README.md

@@ -107,3 +107,22 @@ magpie's usage card shows the account's credits:
   unlimited) and what the pack has used.
 - **What it shows:** credits left as the balance, and the share used as a
   window.
+
+## Daily check-in
+
+Trae CN gives credits for a daily check-in (每日签到). magpie can press it
+once a day for each account (Settings, or the switch on the usage card).
+It asks Trae CN's own pages through
+this plugin's fetch, which sends them as the account: its Cloud-IDE-JWT
+(renewed first when near its end) and its device id.
+
+- **Status:** `POST api.trae.cn/trae/api/v2/ug/checkin_credits/status`,
+  body `{}`: `enable`, `checked_in`, `credits`.
+- **Claim:** `POST api.trae.cn/trae/api/v2/ug/checkin_credits/claim`,
+  body `{}`, only while it is on and today's isn't in.
+- **What comes back:** Trae's answer as it is. A 401 or code 1001 marks
+  the account for a new sign-in. Code 9095 means this device has checked
+  in today; nothing sends another device id to get round it.
+
+Only `api.trae.cn`'s `/trae/api/` pages are sent this way; any other URL
+that isn't a chat request is still refused (400). Needs 0.1.4 or later.

package/index.mjs

@@ -6,6 +6,9 @@
 // agent's /api/agent/v3/llm_utils_chat, which answers in its own SSE
 // events. Worked out from two open-source relays (wangqi233/trae2api and
 // autumnsentiment/Trae2api-cn); not run against a real account here.
+// The account's fetch also sends Trae CN's own pages on api.trae.cn
+// (/trae/api/…) as the account, for magpie's daily check-in (每日签到:
+// /trae/api/v2/ug/checkin_credits/status, then /claim).
 import { randomBytes, randomUUID, randomInt } from "node:crypto"
 import { createServer, STATUS_CODES } from "node:http"
 
@@ -132,6 +135,18 @@ function apiOf(a) {
   return /mchost\.guru|trae-api-/i.test(h) ? h : HOSTS.api
 }
 
+// ownPage: url is one of Trae CN's own JSON pages on api.trae.cn
+// (/trae/api/…: the daily check-in, credits), which the account's fetch
+// sends as the account rather than as a chat
+function ownPage(url) {
+  try {
+    const u = new URL(url)
+    return u.origin === new URL(HOSTS.auth).origin && u.pathname.startsWith("/trae/api/")
+  } catch {
+    return false
+  }
+}
+
 // ---- errors --------------------------------------------------------------------
 
 class Expired extends Error {
@@ -819,6 +834,32 @@ export const TraeCNAuthPlugin = async ({ client }) => {
     return { status: res.status, text, v: parseJSON(text) }
   }
 
+  // ownRequest sends a request for one of Trae CN's own JSON pages as the
+  // account, as its IDE does: magpie asks the daily check-in
+  // (/trae/api/v2/ug/checkin_credits/status, then /claim) through the
+  // account's fetch, as it does WorkBuddy's. Trae's answer comes back as it
+  // is; a token it turned away (401, or code 1001 in a 200) marks the
+  // sign-in, as usage does.
+  const ownRequest = async (getAuth, a0, url, method, body, signal) => {
+    let a
+    try {
+      a = await fresh(getAuth)
+    } catch (e) {
+      if (e instanceof Expired) return errorResponse(401, `${a0.name || a0.uid}: ${e.message}`, "expired")
+      throw e
+    }
+    const res = await fetch(url, {
+      method,
+      headers: ideHeaders(a, { Accept: "application/json" }),
+      body: method === "GET" || method === "HEAD" ? undefined : typeof body === "string" ? body : "{}",
+      signal: signal ?? AbortSignal.timeout(20_000),
+    })
+    const text = await res.text()
+    const lapsed = res.status === 401 || lapsedCode(errorOf(text).code)
+    const headers = { "content-type": res.headers.get("content-type") ?? "application/json", "X-Magpie-Sign-In": lapsed ? "expired" : a.renewed ? "renewed" : "kept" }
+    return new Response(text, { status: res.status, headers })
+  }
+
   const usage = async (getAuth) => {
     let a
     try {
@@ -889,7 +930,8 @@ export const TraeCNAuthPlugin = async ({ client }) => {
   // The lists come from batch_get_detail_param, as TRAE SOLO CN asks them;
   // when that gives none, from get_detail_param, one function at a time.
   // A model several functions list is asked through the first whose entry
-  // names a __dev model (one Trae serves), else the first.
+  // names a __dev model (one Trae serves) and isn't misnamed, else the
+  // first that names one, else the first.
   const liveModels = async (a) => {
     let lists = null
     try {
@@ -907,13 +949,35 @@ export const TraeCNAuthPlugin = async ({ client }) => {
         if (g.status === "fulfilled") lists[FUNCTIONS[i]] = g.value
       })
     }
+    // a function can give a model another's name: chat_v3 calls
+    // deepseek-v4.1-flash "DeepSeek-V4-Flash 正式版", as it does
+    // DeepSeek-V4-Flash-Official, where SOLO's lists call it
+    // DeepSeek-V4.1-Flash (yetone/magpie#681) — Trae's usage page then books
+    // it under that name too. Such an entry is misnamed: its name is another
+    // model's in the same list, and other lists name it otherwise.
+    const nameOf = (m) => String(m.display_config?.display_name || m.display_name || m.display_model_name || m.config_name)
+    const names = new Map() // id → the names its lists give it
+    for (const fn of FUNCTIONS) {
+      for (const m of lists[fn] ?? []) {
+        if (!chatModel(m)) continue
+        const id = String(m.config_name)
+        names.set(id, (names.get(id) ?? new Set()).add(nameOf(m)))
+      }
+    }
+    const misnamed = (fn, m) => {
+      const id = String(m.config_name), name = nameOf(m)
+      return names.get(id).size > 1 && (lists[fn] ?? []).some((o) => chatModel(o) && String(o.config_name) !== id && nameOf(o) === name)
+    }
+    // a function's entry is worth more when it names a __dev model, then
+    // when it isn't misnamed; the first of the best is taken
+    const worth = (fn, m) => (devModel(m) ? 2 : 0) + (misnamed(fn, m) ? 0 : 1)
     const out = new Map()
     for (const fn of FUNCTIONS) {
       for (const m of lists[fn] ?? []) {
         if (!chatModel(m)) continue
         const id = String(m.config_name)
         const was = out.get(id)
-        if (!was || (!devModel(was.m) && devModel(m))) out.set(id, { m, fn })
+        if (!was || worth(fn, m) > was.worth) out.set(id, { m, fn, worth: worth(fn, m) })
       }
     }
     for (const [id, { m, fn }] of out) {
@@ -981,6 +1045,8 @@ export const TraeCNAuthPlugin = async ({ client }) => {
             let body = init.body ?? (r0 ? await r0.text() : undefined)
             if (body instanceof ArrayBuffer) body = new TextDecoder().decode(body)
             else if (ArrayBuffer.isView(body)) body = new TextDecoder().decode(new Uint8Array(body.buffer, body.byteOffset, body.byteLength))
+            const url = r0 ? r0.url : input instanceof URL ? input.href : String(input)
+            if (ownPage(url)) return ownRequest(getAuth, a0, url, init.method ?? r0?.method ?? "POST", body, init.signal ?? r0?.signal)
             const req = parseJSON(body)
             if (!Array.isArray(req.messages)) return errorResponse(400, "Trae CN: only chat completions are served")
             let a

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@magpie-community/opencode-trae-auth",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "OpenCode provider plugin for Trae CN (trae.cn): your Trae CN account's models, free tier included, with Trae's own browser sign-in",
   "type": "module",
   "main": "./index.mjs",