dsh-multi-router 0.4.1 → 0.5.1

package/lib/{chunk-GGRCX2NR.js → chunk-XV7O2LRF.js}

@@ -2,7 +2,7 @@ import {
   isFreeModel,
   reasoningModes,
   writePrivateJson
-} from "./chunk-XMXFIV5G.js";
+} from "./chunk-GAQ77CH4.js";
 
 // src/telemetry.mjs
 import { readFile } from "node:fs/promises";
@@ -446,4 +446,4 @@ export {
   matchModel,
   Router
 };
-//# sourceMappingURL=chunk-GGRCX2NR.js.map
+//# sourceMappingURL=chunk-XV7O2LRF.js.map

package/lib/{chunk-UFS7ZJKY.js → chunk-YUP77Q3A.js}

@@ -12,7 +12,7 @@ import {
 import {
   RouteError,
   abortable
-} from "./chunk-GGRCX2NR.js";
+} from "./chunk-XV7O2LRF.js";
 import {
   WorkBuddyUpstreamClient,
   modelWithCurrentPromotion,
@@ -4909,7 +4909,7 @@ function cachedModels(discover, ttl) {
   };
 }
 async function createAccountClient(account, config, attachments) {
-  let chat, discover, accountInfo;
+  let chat, discover, accountInfo, checkin;
   if (account.provider.startsWith("workbuddy")) {
     const client = new WorkBuddyUpstreamClient();
     const variant = WORKBUDDY_VARIANTS.find((v) => v.id === account.provider);
@@ -4974,6 +4974,25 @@ async function createAccountClient(account, config, attachments) {
       }
     };
     const usage = new TraeUsageClient({ credential, deviceId: async () => (await identity()).deviceId });
+    if (account.provider === "trae") {
+      const checkinUsage = new TraeUsageClient({
+        credential,
+        deviceId: async () => (await identity()).deviceId,
+        fetchImpl: async (url, options) => {
+          const response = await fetch(url, options);
+          if (!response.ok) throw Object.assign(Error("checkin_http_error"), { status: response.status });
+          if (String(url).endsWith("/status")) {
+            const body = await response.clone().json();
+            if (body.code !== void 0 && body.code !== 0 || typeof body.checked_in !== "boolean" || typeof body.did_checked_in !== "boolean") throw Object.assign(Error("checkin_status_rejected"), { status: 403 });
+          }
+          return response;
+        }
+      });
+      checkin = {
+        identity: async () => ({ account: (await credential()).userId, device: (await identity()).deviceId }),
+        claim: (signal) => claimTraeCheckin(checkinUsage, signal)
+      };
+    }
     accountInfo = async (signal) => {
       const c = await credential();
       const login = {
@@ -5022,7 +5041,8 @@ async function createAccountClient(account, config, attachments) {
     const transport = createQoderTransport({
       region,
       resolvePat: getPat,
-      resolveMachineId: () => getMachineId([`${account.credentialFile}.machine-id`]),
+      // Reuse the official CLI identity (or one persisted plugin seed) for this physical device.
+      resolveMachineId: () => getMachineId(),
       ...attachments ? { attachments } : {}
     });
     const client = new QoderUpstreamClient({
@@ -5032,6 +5052,27 @@ async function createAccountClient(account, config, attachments) {
       transport,
       ...attachments ? { attachments } : {}
     });
+    const checkinTransport = createQoderTransport({
+      region,
+      resolvePat: getPat,
+      resolveMachineId: () => getMachineId(),
+      fetch: async (url, options) => {
+        const response = await fetch(url, options);
+        if (!response.ok) {
+          const restricted = [401, 403, 429].includes(response.status);
+          throw new QoderLlmError(`checkin_http_${response.status}`, restricted ? "QUOTA" : "PROVIDER_ERROR", { status: response.status });
+        }
+        return response;
+      }
+    });
+    checkin = {
+      identity: async (signal) => ({ account: (await checkinTransport.readAccount({ signal })).profile.id }),
+      claim: async (signal) => {
+        const result = await checkinTransport.checkIn(signal);
+        if (result.status !== "error") return { status: result.status };
+        return { status: /401|403|429|rate.?limit|throttl|captcha|verify|verification|风控|验证|频繁|forbidden|unauthor|AUTH/i.test(result.message || "") ? "paused" : "uncertain" };
+      }
+    };
     accountInfo = async (signal) => {
       const result = await transport.readAccount({ force: true, signal });
       const buckets = [["\u4E2A\u4EBA\u5957\u9910", result.usage?.userQuota], ["\u9644\u52A0\u989D\u5EA6", result.usage?.addOnQuota], ["\u7EC4\u7EC7\u989D\u5EA6", result.usage?.orgResourcePackage]].filter(([, p]) => p).map(([name, p]) => ({ ...p, name }));
@@ -5052,17 +5093,33 @@ async function createAccountClient(account, config, attachments) {
   return {
     models: cachedModels(discover, config.catalogTtlMs),
     accountInfo,
+    checkin,
     async chat(input, signal) {
       const result = await chat(input, signal);
       return result.ok ? preflightStream(result.response, signal) : result;
     }
   };
 }
+async function claimTraeCheckin(usage, signal) {
+  try {
+    const status = await usage.checkinStatus(signal);
+    if (status.checkedIn) return { status: "already-claimed" };
+    if (status.didCheckedIn) return { status: "device-used" };
+    if (!status.enabled) return { status: "no-campaign" };
+    const claim = await usage.claimCheckin(signal);
+    if (claim.code === 9095) return { status: "device-used" };
+    if (!claim.claimed) return { status: "paused" };
+    return { status: "claimed", ...Number.isFinite(status.credits) ? { amount: status.credits + (status.extraCredits || 0) } : {} };
+  } catch (e) {
+    return { status: [401, 403, 429].includes(e.status) ? "paused" : "uncertain" };
+  }
+}
 
 export {
   qoderPluginDataDir,
   jsonCredential,
   cachedModels,
-  createAccountClient
+  createAccountClient,
+  claimTraeCheckin
 };
-//# sourceMappingURL=chunk-UFS7ZJKY.js.map
+//# sourceMappingURL=chunk-YUP77Q3A.js.map