mslxdff 0.1.199 → 0.1.203
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +3 -3
- package/bin/mslxdff.js +1 -1
- package/package.json +38 -38
- package/plugins/README.md +14 -14
- package/plugins/prefer-model.mjs.example +14 -14
- package/src/anthropic/count-tokens.js +61 -61
- package/src/anthropic/request-body.js +126 -126
- package/src/anthropic/shared.js +18 -18
- package/src/anthropic/sse-aggregate.js +56 -56
- package/src/anthropic/translate.js +287 -287
- package/src/anthropic/web-search.js +361 -361
- package/src/api-kind.js +31 -31
- package/src/autostart.js +288 -288
- package/src/bench/probe.js +1 -1
- package/src/bench/runner.js +1 -1
- package/src/bench/via-probe.js +1 -1
- package/src/bench/via.js +144 -144
- package/src/bench/workbuddy-bench.js +1 -1
- package/src/chat/config.js +11 -11
- package/src/chat/cooling.js +81 -81
- package/src/chat/direct.js +58 -58
- package/src/chat/index.js +6 -6
- package/src/chat/spinner.js +38 -38
- package/src/chat/sse.js +69 -69
- package/src/chat/stats.js +247 -247
- package/src/chat/store.js +48 -48
- package/src/chat/tool-handlers.js +67 -67
- package/src/chat/tools.js +1 -1
- package/src/chat/upstream.js +59 -59
- package/src/chat-pipeline/auto-race.js +158 -158
- package/src/chat-pipeline/empty-turn.js +147 -147
- package/src/chat-pipeline/engine.js +17 -17
- package/src/chat-pipeline/serial-trial.js +283 -283
- package/src/chat-pipeline/terminal-verdict.js +147 -147
- package/src/chooser.js +54 -54
- package/src/claude-desktop/apply.js +101 -101
- package/src/claude-desktop/context1m.js +38 -38
- package/src/claude-desktop/doctor.js +146 -146
- package/src/claude-desktop/slots.js +152 -152
- package/src/cli/bootstrap.js +1 -1
- package/src/cli/commands/claude-desktop.js +284 -284
- package/src/cli/commands/daemon.js +231 -231
- package/src/cli/commands/model/list-live.js +22 -22
- package/src/cli/commands/model/list-providers.js +76 -76
- package/src/cli/commands/model/list-render.js +77 -77
- package/src/cli/commands/model/list-sort.js +52 -52
- package/src/cli/commands/model/list-tty.js +76 -76
- package/src/cli/commands/model/list.js +166 -166
- package/src/cli/commands/model/live-models.js +24 -24
- package/src/cli/commands/model/stats.js +43 -43
- package/src/cli/commands/model/status.js +47 -47
- package/src/cli/commands/model.js +51 -51
- package/src/cli/commands/provider/add.js +108 -108
- package/src/cli/commands/provider/allowlist.js +99 -99
- package/src/cli/commands/provider/cline-free.js +99 -99
- package/src/cli/commands/provider/cline-login.js +218 -218
- package/src/cli/commands/provider/cline-quota.js +65 -65
- package/src/cli/commands/provider/codearts-login.js +45 -45
- package/src/cli/commands/provider/config.js +119 -119
- package/src/cli/commands/provider/globalqwenwork-login.js +126 -126
- package/src/cli/commands/provider/keys-guard.js +50 -50
- package/src/cli/commands/provider/keys.js +161 -161
- package/src/cli/commands/provider/qoder-checkin.js +81 -81
- package/src/cli/commands/provider/qoder-login.js +102 -102
- package/src/cli/commands/provider/raccoon-login.js +180 -180
- package/src/cli/commands/provider/raccoon-password.js +162 -162
- package/src/cli/commands/provider/raccoon-quota.js +109 -109
- package/src/cli/commands/provider/traework-checkin.js +144 -144
- package/src/cli/commands/provider/traework-login.js +164 -164
- package/src/cli/commands/provider/workbuddy-login.js +153 -153
- package/src/cli/commands/provider/zcode-login.js +77 -77
- package/src/cli/commands/provider/zcode-quota.js +55 -55
- package/src/cli/commands/provider.js +1 -1
- package/src/cli/commands/stats.js +149 -149
- package/src/cli/commands/system.js +261 -261
- package/src/cli/commands/timezone.js +47 -47
- package/src/cli/commands/use-group.js +60 -60
- package/src/cli/commands/workbuddy.js +189 -189
- package/src/cli/format.js +141 -141
- package/src/cli/group-helpers.js +70 -70
- package/src/cli/help.js +80 -80
- package/src/cli/index.js +70 -70
- package/src/cli/interactive.js +121 -121
- package/src/cli/join-core.js +99 -99
- package/src/cli/join-wizard.js +175 -175
- package/src/cli/policy.js +136 -136
- package/src/cli/provider-row.js +126 -126
- package/src/cli/util.js +24 -24
- package/src/compat.js +53 -53
- package/src/events.js +38 -38
- package/src/free-lane.js +131 -131
- package/src/free-watcher.js +63 -63
- package/src/groups.js +262 -262
- package/src/logs.js +195 -195
- package/src/metrics.js +62 -62
- package/src/model-capabilities/context-lookup.js +65 -65
- package/src/model-capabilities/merge.js +142 -142
- package/src/model-trace.js +187 -187
- package/src/models.js +1 -1
- package/src/opencode-identity.js +72 -72
- package/src/peers.js +254 -251
- package/src/plugins.js +107 -107
- package/src/providers/base.js +164 -164
- package/src/providers/classify.js +22 -22
- package/src/providers/cline/allowlist-sync.js +93 -93
- package/src/providers/cline/free-catalog.js +71 -71
- package/src/providers/cline/headers.js +37 -37
- package/src/providers/cline/index.js +130 -130
- package/src/providers/cline/models.js +150 -150
- package/src/providers/cline/usage.js +274 -274
- package/src/providers/codearts/auth-pool.js +148 -148
- package/src/providers/codearts/concurrency.js +34 -34
- package/src/providers/codearts/const.js +65 -65
- package/src/providers/codearts/dpop.js +95 -95
- package/src/providers/codearts/index.js +175 -175
- package/src/providers/codearts/login.js +214 -214
- package/src/providers/codearts/models.js +182 -182
- package/src/providers/codearts/sign.js +96 -96
- package/src/providers/codearts/sse.js +204 -204
- package/src/providers/codearts/stream.js +209 -209
- package/src/providers/codearts/sts.js +124 -124
- package/src/providers/codearts.js +1 -1
- package/src/providers/dispatcher.js +146 -146
- package/src/providers/generic.js +156 -156
- package/src/providers/globalqwenwork/account-store.js +141 -141
- package/src/providers/globalqwenwork/constants.js +73 -73
- package/src/providers/globalqwenwork/crypto.js +2 -2
- package/src/providers/globalqwenwork/http.js +2 -2
- package/src/providers/globalqwenwork/index.js +331 -331
- package/src/providers/globalqwenwork/payload.js +147 -147
- package/src/providers/globalqwenwork/rsa.js +2 -2
- package/src/providers/globalqwenwork/sse.js +2 -2
- package/src/providers/globalqwenwork/stream.js +2 -2
- package/src/providers/globalqwenwork.js +1 -1
- package/src/providers/globalworkbuddy/account-store.js +198 -198
- package/src/providers/globalworkbuddy/auth.js +97 -97
- package/src/providers/globalworkbuddy/catalog.js +138 -138
- package/src/providers/globalworkbuddy/constants.js +73 -73
- package/src/providers/globalworkbuddy/device-auth.js +120 -120
- package/src/providers/globalworkbuddy/errors.js +98 -98
- package/src/providers/globalworkbuddy/headers.js +89 -89
- package/src/providers/globalworkbuddy/models.js +96 -96
- package/src/providers/globalworkbuddy/payload.js +80 -80
- package/src/providers/index.js +5 -5
- package/src/providers/model-id.js +172 -172
- package/src/providers/opencode.js +19 -19
- package/src/providers/openrouter.js +31 -31
- package/src/providers/qoder/account-store.js +106 -106
- package/src/providers/qoder/aggregate.js +60 -60
- package/src/providers/qoder/baseprompt.json +591 -591
- package/src/providers/qoder/checkin.js +160 -160
- package/src/providers/qoder/constants.js +44 -44
- package/src/providers/qoder/encode.js +41 -41
- package/src/providers/qoder/fingerprint.js +38 -38
- package/src/providers/qoder/models.js +77 -77
- package/src/providers/qoder/oauth.js +84 -84
- package/src/providers/qoder/payload.js +162 -162
- package/src/providers/qoder/request.js +27 -27
- package/src/providers/qoder/session.js +134 -134
- package/src/providers/qoder/sticky.js +38 -38
- package/src/providers/qoder/stream.js +116 -116
- package/src/providers/qoder.js +1 -1
- package/src/providers/raccoon/account-store.js +216 -216
- package/src/providers/raccoon/auth.js +67 -67
- package/src/providers/raccoon/chat.js +228 -228
- package/src/providers/raccoon/const.js +180 -165
- package/src/providers/raccoon/credits.js +185 -175
- package/src/providers/raccoon/crypto.js +43 -0
- package/src/providers/raccoon/envelope.js +31 -31
- package/src/providers/raccoon/headers.js +45 -45
- package/src/providers/raccoon/index.js +216 -216
- package/src/providers/raccoon/login.js +279 -279
- package/src/providers/raccoon/models.js +160 -160
- package/src/providers/raccoon/password-login.js +102 -87
- package/src/providers/raccoon/qr-codewords.js +115 -115
- package/src/providers/raccoon/qr.js +236 -236
- package/src/providers/raccoon/sse.js +135 -135
- package/src/providers/responses-channel.js +120 -120
- package/src/providers/share-keys.js +53 -53
- package/src/providers/traework/account-store.js +234 -234
- package/src/providers/traework/chat.js +217 -217
- package/src/providers/traework/checkin.js +77 -77
- package/src/providers/traework/constants.js +41 -41
- package/src/providers/traework/errors.js +52 -52
- package/src/providers/traework/headers.js +79 -79
- package/src/providers/traework/index.js +132 -132
- package/src/providers/traework/market-identity.js +116 -116
- package/src/providers/traework/models.js +98 -98
- package/src/providers/traework/payload.js +86 -86
- package/src/providers/traework/sse.js +184 -184
- package/src/providers/traework/token.js +52 -52
- package/src/providers/traework.js +2 -2
- package/src/providers/workbuddy/account-store.js +186 -186
- package/src/providers/workbuddy/auth.js +158 -158
- package/src/providers/workbuddy/balance.js +86 -86
- package/src/providers/workbuddy/cat-travel.js +125 -125
- package/src/providers/workbuddy/checkin.js +75 -75
- package/src/providers/workbuddy/desktop-info.js +78 -78
- package/src/providers/workbuddy/growth-api.js +67 -67
- package/src/providers/workbuddy/growth-plans.js +63 -63
- package/src/providers/workbuddy/growth.js +181 -181
- package/src/providers/workbuddy/index.js +217 -217
- package/src/providers/workbuddy/models.js +126 -126
- package/src/providers/workbuddy/payload.js +141 -141
- package/src/providers/workbuddy/rotation-log.js +83 -83
- package/src/providers/workbuddy/sanitize-tools.js +61 -61
- package/src/providers/workbuddy/sdk-chat.js +26 -26
- package/src/providers/workbuddy-balance.js +78 -78
- package/src/providers/zcode/account-store.js +129 -129
- package/src/providers/zcode/auth.js +28 -28
- package/src/providers/zcode/chat.js +179 -179
- package/src/providers/zcode/const.js +54 -54
- package/src/providers/zcode/context-shape.js +162 -162
- package/src/providers/zcode/headers.js +93 -93
- package/src/providers/zcode/index.js +124 -124
- package/src/providers/zcode/models.js +65 -65
- package/src/providers/zcode/oauth.js +120 -120
- package/src/providers/zcode/quota.js +176 -176
- package/src/providers/zcode/sse.js +196 -196
- package/src/readline-compat.js +25 -25
- package/src/responses/translate.js +311 -311
- package/src/routes/chat/broadband-handler.js +72 -72
- package/src/routes/chat/empty-turn-report.js +31 -31
- package/src/routes/chat/exhausted-handler.js +87 -87
- package/src/routes/chat/hedge-handler.js +130 -130
- package/src/routes/chat/index.js +1 -1
- package/src/routes/chat/local-handler.js +63 -63
- package/src/routes/chat/peer-handler.js +81 -81
- package/src/routes/chat.js +3 -3
- package/src/routes/fallback.js +88 -88
- package/src/routes/groups-relay.js +214 -214
- package/src/routes/groups.js +122 -122
- package/src/routes/helpers.js +83 -83
- package/src/routes/index.js +147 -147
- package/src/routes/messages-route.js +365 -365
- package/src/routes/peers.js +23 -1
- package/src/routes/relay-queue.js +181 -181
- package/src/routes/relay.js +60 -60
- package/src/routes/responses-route.js +213 -213
- package/src/routes/stream-body.js +84 -84
- package/src/routes/stream-hold.js +337 -337
- package/src/routes/stream-scan.js +366 -366
- package/src/routes.js +3 -3
- package/src/runtime/auto-update.js +113 -113
- package/src/runtime/bootstrap.js +8 -0
- package/src/runtime/broadband-stream.js +79 -79
- package/src/runtime/broadband.js +103 -103
- package/src/runtime/group-sync.js +81 -81
- package/src/runtime/keep-warm.js +120 -0
- package/src/runtime/lifecycle-forensics.js +116 -116
- package/src/runtime/lifecycle-log.js +87 -87
- package/src/runtime/log-stamp.js +73 -73
- package/src/runtime/proc-info.js +137 -137
- package/src/runtime/qoder-checkin.js +93 -93
- package/src/runtime/raccoon-checkin.js +126 -126
- package/src/runtime/server-lifecycle.js +189 -189
- package/src/runtime/traework-checkin.js +118 -118
- package/src/runtime/workbuddy-checkin.js +120 -120
- package/src/runtime/workbuddy-growth.js +133 -133
- package/src/server.js +7 -0
- package/src/state/facade.js +58 -58
- package/src/state/memory.js +162 -162
- package/src/state/merge.js +25 -25
- package/src/state/migrations/cline-unify.js +94 -94
- package/src/state/migrations.js +33 -33
- package/src/state/persist.js +42 -42
- package/src/state/provider-config.js +142 -142
- package/src/state/schemas/group.js +31 -31
- package/src/state/schemas/model.js +94 -94
- package/src/state/schemas/peer.js +31 -31
- package/src/state/schemas/port.js +10 -10
- package/src/state/schemas/provider.js +202 -202
- package/src/state/schemas/timezone.js +70 -70
- package/src/state/schemas/token.js +48 -48
- package/src/state/schemas/use-group.js +102 -102
- package/src/state/store.js +176 -176
- package/src/state.js +1 -1
- package/src/sync-claude-desktop.js +234 -234
- package/src/sync-claude.js +152 -152
- package/src/sync-codex.js +116 -116
- package/src/sync-workbuddy.js +158 -158
- package/src/talk-full-redact.js +129 -129
- package/src/talk-full.js +131 -131
- package/src/talk-log.js +257 -257
- package/src/text-tokens.js +52 -52
- package/src/thinking-effort.js +67 -67
- package/src/time.js +106 -106
- package/src/timeline.js +26 -26
- package/src/transport/index.js +1 -1
- package/src/upstream-engine/index.js +74 -72
- package/src/upstream-engine/mode.js +15 -15
- package/src/upstream-engine/sdk/attempt.js +221 -221
- package/src/upstream-engine/sdk/chat.js +28 -28
- package/src/upstream-engine/sdk/convert.js +192 -192
- package/src/upstream-engine/sdk/diagnose.js +62 -62
- package/src/upstream-engine/sdk/dispatch.js +55 -55
- package/src/upstream-engine/sdk/responses.js +176 -176
- package/src/upstream-engine/sdk/sse.js +166 -166
- package/src/upstream-responses.js +305 -305
- package/src/upstream.js +53 -1
- package/src/usage/record.js +106 -106
- package/src/usage/report.js +195 -195
- package/src/v6pool/claim.js +157 -157
- package/src/v6pool/config.js +65 -65
- package/src/v6pool/detect.js +130 -130
- package/src/v6pool/index.js +152 -152
- package/src/v6pool/prefix.js +140 -140
- package/src/v6pool/probe.js +40 -40
- package/src/v6pool/report.js +104 -104
- package/src/v6pool/rotate.js +263 -239
- package/src/warp-pool/config.js +74 -74
- package/src/warp-pool/health.js +106 -106
- package/src/warp-pool/identity.js +141 -141
- package/src/warp-pool/index.js +358 -358
- package/src/warp-pool/keys.js +29 -29
- package/src/warp-pool/nodes.js +95 -95
- package/src/warp-pool/ports.js +160 -160
- package/src/warp-pool/register.js +113 -113
- package/src/warp-pool/report.js +232 -232
- package/src/warp-pool/revive.js +168 -168
- package/src/warp-pool/rotate.js +214 -214
- package/src/warp-pool/strategy.js +43 -43
- package/src/warp-pool/verdict.js +69 -69
- package/src/warp-pool/wireproxy.js +162 -162
- package/docs/cli_help_mini.md +0 -146
- package/src/providers/AGENTS.md +0 -49
- package/src/routes/AGENTS.md +0 -37
|
@@ -1,124 +1,124 @@
|
|
|
1
|
-
// zcode 供应商工厂:keyring 多号轮换 + 差异化冷却(auth/限流/服务端=短冷却、1005=长冷却、3007/参数=不冷却)。
|
|
2
|
-
import { compatFetch } from "../../compat.js";
|
|
3
|
-
import { envInt } from "../base.js";
|
|
4
|
-
import { createKeyRing } from "../keyring.js";
|
|
5
|
-
import { loadProviderKeys } from "../../state.js";
|
|
6
|
-
import { createModelsService } from "./models.js";
|
|
7
|
-
import { forwardZcodeChat } from "./chat.js";
|
|
8
|
-
import { fetchZcodeBalance } from "./quota.js";
|
|
9
|
-
import { listZcodeAccountDocs } from "./account-store.js";
|
|
10
|
-
|
|
11
|
-
// 冷却策略:auth/限流/服务端/网络=短冷却;quota=长冷却(按日重置);security/param/unknown=不冷却(换号无意义)
|
|
12
|
-
const COOLDOWN_BY_KIND = { auth: "short", rate_limit: "short", server: "short", network: "short", quota: "long" };
|
|
13
|
-
|
|
14
|
-
function notLoggedInResponse() {
|
|
15
|
-
return new Response(
|
|
16
|
-
JSON.stringify({ error: { message: "zcode: 无可用账号 —— 先运行 mslxdff -provider zcode login", type: "auth_error" } }),
|
|
17
|
-
{ status: 401, headers: { "Content-Type": "application/json" } },
|
|
18
|
-
);
|
|
19
|
-
}
|
|
20
|
-
|
|
21
|
-
function allCoolingResponse() {
|
|
22
|
-
return new Response(
|
|
23
|
-
JSON.stringify({ error: { message: "zcode: 账号暂不可用(冷却中,可能额度用尽或限流),请稍后重试", type: "all_cooling" } }),
|
|
24
|
-
{ status: 429, headers: { "Content-Type": "application/json", "x-mslxdff-zcode-all-cooling": "1" } },
|
|
25
|
-
);
|
|
26
|
-
}
|
|
27
|
-
|
|
28
|
-
export function createZcodeProvider({
|
|
29
|
-
id = "zcode",
|
|
30
|
-
apiKeys,
|
|
31
|
-
file,
|
|
32
|
-
fetchImpl,
|
|
33
|
-
cooldownMs = envInt("MSLXDFF_ZCODE_COOLDOWN_MS", 30_000),
|
|
34
|
-
quotaCooldownMs = envInt("MSLXDFF_ZCODE_QUOTA_COOLDOWN_MS", 3_600_000),
|
|
35
|
-
timeoutMs = Number(process.env.MSLXDFF_ZCODE_TIMEOUT_MS) || 120_000,
|
|
36
|
-
} = {}) {
|
|
37
|
-
if (!fetchImpl) fetchImpl = compatFetch;
|
|
38
|
-
const keys = (() => {
|
|
39
|
-
if (Array.isArray(apiKeys) && apiKeys.length) return [...new Set(apiKeys.map((k) => String(k).trim()).filter(Boolean))];
|
|
40
|
-
try {
|
|
41
|
-
return loadProviderKeys(id, file ? { file } : {});
|
|
42
|
-
} catch {
|
|
43
|
-
return [];
|
|
44
|
-
}
|
|
45
|
-
})();
|
|
46
|
-
const ring = createKeyRing(keys, { cooldownMs });
|
|
47
|
-
|
|
48
|
-
// JWT → deviceMid 映射(设备身份跟账号走,取自 auths/zcode-<uid>.json)
|
|
49
|
-
let midCache = null;
|
|
50
|
-
function deviceMidFor(key) {
|
|
51
|
-
if (midCache === null) {
|
|
52
|
-
try {
|
|
53
|
-
midCache = new Map(listZcodeAccountDocs().map((d) => [d.jwt, d.deviceMid]));
|
|
54
|
-
} catch {
|
|
55
|
-
midCache = new Map();
|
|
56
|
-
}
|
|
57
|
-
}
|
|
58
|
-
return midCache.get(key) || "";
|
|
59
|
-
}
|
|
60
|
-
|
|
61
|
-
const modelsSvc = createModelsService({
|
|
62
|
-
id,
|
|
63
|
-
hasAccount: () => ring.keys.length > 0,
|
|
64
|
-
// balance capabilities 探测:entitlement 名 → canonical 目录交集(失败/空 → 服务回退全量目录)
|
|
65
|
-
probe: async () => {
|
|
66
|
-
const token = ring.keys[0];
|
|
67
|
-
if (!token) return [];
|
|
68
|
-
const r = await fetchZcodeBalance({ token, deviceMid: deviceMidFor(token), fetchImpl });
|
|
69
|
-
return r.ok ? r.modelIds : [];
|
|
70
|
-
},
|
|
71
|
-
});
|
|
72
|
-
|
|
73
|
-
async function chat(body, opts = {}) {
|
|
74
|
-
const total = ring.keys.length;
|
|
75
|
-
if (!total) return notLoggedInResponse();
|
|
76
|
-
let lastRes = null;
|
|
77
|
-
for (let hop = 0; hop <= total; hop++) {
|
|
78
|
-
const key = ring.next();
|
|
79
|
-
if (!key) break;
|
|
80
|
-
const res = await forwardZcodeChat({ body, token: key, deviceMid: deviceMidFor(key), fetchImpl, timeoutMs });
|
|
81
|
-
const kind = res.headers.get("x-mslxdff-zcode-kind");
|
|
82
|
-
if (!kind) return res;
|
|
83
|
-
const mode = COOLDOWN_BY_KIND[kind] || "none";
|
|
84
|
-
if (mode === "long") ring.onError(key, quotaCooldownMs);
|
|
85
|
-
else if (mode === "short") ring.onError(key, cooldownMs);
|
|
86
|
-
lastRes = res;
|
|
87
|
-
if (mode === "none") break; // security/param/unknown:换号无意义,如实透出
|
|
88
|
-
if (ring.available() === 0) break; // 没有别的可用号
|
|
89
|
-
}
|
|
90
|
-
if (lastRes) return lastRes;
|
|
91
|
-
return allCoolingResponse();
|
|
92
|
-
}
|
|
93
|
-
|
|
94
|
-
async function listModels() {
|
|
95
|
-
return modelsSvc.listModels();
|
|
96
|
-
}
|
|
97
|
-
|
|
98
|
-
async function preheat() {
|
|
99
|
-
try {
|
|
100
|
-
const list = await modelsSvc.listModels();
|
|
101
|
-
return list.length ? { ok: true } : { ok: false, error: "no catalog" };
|
|
102
|
-
} catch (e) {
|
|
103
|
-
return { ok: false, error: String(e?.message || e).slice(0, 120) };
|
|
104
|
-
}
|
|
105
|
-
}
|
|
106
|
-
|
|
107
|
-
async function close() {}
|
|
108
|
-
|
|
109
|
-
async function chatWithKeys(body, keysOverride, opts) {
|
|
110
|
-
const tmp = createZcodeProvider({ id, apiKeys: keysOverride, file, fetchImpl, cooldownMs, quotaCooldownMs, timeoutMs });
|
|
111
|
-
return tmp.chat(body, opts);
|
|
112
|
-
}
|
|
113
|
-
|
|
114
|
-
return {
|
|
115
|
-
id,
|
|
116
|
-
chat,
|
|
117
|
-
chatWithKeys,
|
|
118
|
-
listModels,
|
|
119
|
-
preheat,
|
|
120
|
-
close,
|
|
121
|
-
keyRing: ring,
|
|
122
|
-
baseUrl: "zcode://native",
|
|
123
|
-
};
|
|
124
|
-
}
|
|
1
|
+
// zcode 供应商工厂:keyring 多号轮换 + 差异化冷却(auth/限流/服务端=短冷却、1005=长冷却、3007/参数=不冷却)。
|
|
2
|
+
import { compatFetch } from "../../compat.js";
|
|
3
|
+
import { envInt } from "../base.js";
|
|
4
|
+
import { createKeyRing } from "../keyring.js";
|
|
5
|
+
import { loadProviderKeys } from "../../state.js";
|
|
6
|
+
import { createModelsService } from "./models.js";
|
|
7
|
+
import { forwardZcodeChat } from "./chat.js";
|
|
8
|
+
import { fetchZcodeBalance } from "./quota.js";
|
|
9
|
+
import { listZcodeAccountDocs } from "./account-store.js";
|
|
10
|
+
|
|
11
|
+
// 冷却策略:auth/限流/服务端/网络=短冷却;quota=长冷却(按日重置);security/param/unknown=不冷却(换号无意义)
|
|
12
|
+
const COOLDOWN_BY_KIND = { auth: "short", rate_limit: "short", server: "short", network: "short", quota: "long" };
|
|
13
|
+
|
|
14
|
+
function notLoggedInResponse() {
|
|
15
|
+
return new Response(
|
|
16
|
+
JSON.stringify({ error: { message: "zcode: 无可用账号 —— 先运行 mslxdff -provider zcode login", type: "auth_error" } }),
|
|
17
|
+
{ status: 401, headers: { "Content-Type": "application/json" } },
|
|
18
|
+
);
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
function allCoolingResponse() {
|
|
22
|
+
return new Response(
|
|
23
|
+
JSON.stringify({ error: { message: "zcode: 账号暂不可用(冷却中,可能额度用尽或限流),请稍后重试", type: "all_cooling" } }),
|
|
24
|
+
{ status: 429, headers: { "Content-Type": "application/json", "x-mslxdff-zcode-all-cooling": "1" } },
|
|
25
|
+
);
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
export function createZcodeProvider({
|
|
29
|
+
id = "zcode",
|
|
30
|
+
apiKeys,
|
|
31
|
+
file,
|
|
32
|
+
fetchImpl,
|
|
33
|
+
cooldownMs = envInt("MSLXDFF_ZCODE_COOLDOWN_MS", 30_000),
|
|
34
|
+
quotaCooldownMs = envInt("MSLXDFF_ZCODE_QUOTA_COOLDOWN_MS", 3_600_000),
|
|
35
|
+
timeoutMs = Number(process.env.MSLXDFF_ZCODE_TIMEOUT_MS) || 120_000,
|
|
36
|
+
} = {}) {
|
|
37
|
+
if (!fetchImpl) fetchImpl = compatFetch;
|
|
38
|
+
const keys = (() => {
|
|
39
|
+
if (Array.isArray(apiKeys) && apiKeys.length) return [...new Set(apiKeys.map((k) => String(k).trim()).filter(Boolean))];
|
|
40
|
+
try {
|
|
41
|
+
return loadProviderKeys(id, file ? { file } : {});
|
|
42
|
+
} catch {
|
|
43
|
+
return [];
|
|
44
|
+
}
|
|
45
|
+
})();
|
|
46
|
+
const ring = createKeyRing(keys, { cooldownMs });
|
|
47
|
+
|
|
48
|
+
// JWT → deviceMid 映射(设备身份跟账号走,取自 auths/zcode-<uid>.json)
|
|
49
|
+
let midCache = null;
|
|
50
|
+
function deviceMidFor(key) {
|
|
51
|
+
if (midCache === null) {
|
|
52
|
+
try {
|
|
53
|
+
midCache = new Map(listZcodeAccountDocs().map((d) => [d.jwt, d.deviceMid]));
|
|
54
|
+
} catch {
|
|
55
|
+
midCache = new Map();
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
return midCache.get(key) || "";
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
const modelsSvc = createModelsService({
|
|
62
|
+
id,
|
|
63
|
+
hasAccount: () => ring.keys.length > 0,
|
|
64
|
+
// balance capabilities 探测:entitlement 名 → canonical 目录交集(失败/空 → 服务回退全量目录)
|
|
65
|
+
probe: async () => {
|
|
66
|
+
const token = ring.keys[0];
|
|
67
|
+
if (!token) return [];
|
|
68
|
+
const r = await fetchZcodeBalance({ token, deviceMid: deviceMidFor(token), fetchImpl });
|
|
69
|
+
return r.ok ? r.modelIds : [];
|
|
70
|
+
},
|
|
71
|
+
});
|
|
72
|
+
|
|
73
|
+
async function chat(body, opts = {}) {
|
|
74
|
+
const total = ring.keys.length;
|
|
75
|
+
if (!total) return notLoggedInResponse();
|
|
76
|
+
let lastRes = null;
|
|
77
|
+
for (let hop = 0; hop <= total; hop++) {
|
|
78
|
+
const key = ring.next();
|
|
79
|
+
if (!key) break;
|
|
80
|
+
const res = await forwardZcodeChat({ body, token: key, deviceMid: deviceMidFor(key), fetchImpl, timeoutMs });
|
|
81
|
+
const kind = res.headers.get("x-mslxdff-zcode-kind");
|
|
82
|
+
if (!kind) return res;
|
|
83
|
+
const mode = COOLDOWN_BY_KIND[kind] || "none";
|
|
84
|
+
if (mode === "long") ring.onError(key, quotaCooldownMs);
|
|
85
|
+
else if (mode === "short") ring.onError(key, cooldownMs);
|
|
86
|
+
lastRes = res;
|
|
87
|
+
if (mode === "none") break; // security/param/unknown:换号无意义,如实透出
|
|
88
|
+
if (ring.available() === 0) break; // 没有别的可用号
|
|
89
|
+
}
|
|
90
|
+
if (lastRes) return lastRes;
|
|
91
|
+
return allCoolingResponse();
|
|
92
|
+
}
|
|
93
|
+
|
|
94
|
+
async function listModels() {
|
|
95
|
+
return modelsSvc.listModels();
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
async function preheat() {
|
|
99
|
+
try {
|
|
100
|
+
const list = await modelsSvc.listModels();
|
|
101
|
+
return list.length ? { ok: true } : { ok: false, error: "no catalog" };
|
|
102
|
+
} catch (e) {
|
|
103
|
+
return { ok: false, error: String(e?.message || e).slice(0, 120) };
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
async function close() {}
|
|
108
|
+
|
|
109
|
+
async function chatWithKeys(body, keysOverride, opts) {
|
|
110
|
+
const tmp = createZcodeProvider({ id, apiKeys: keysOverride, file, fetchImpl, cooldownMs, quotaCooldownMs, timeoutMs });
|
|
111
|
+
return tmp.chat(body, opts);
|
|
112
|
+
}
|
|
113
|
+
|
|
114
|
+
return {
|
|
115
|
+
id,
|
|
116
|
+
chat,
|
|
117
|
+
chatWithKeys,
|
|
118
|
+
listModels,
|
|
119
|
+
preheat,
|
|
120
|
+
close,
|
|
121
|
+
keyRing: ring,
|
|
122
|
+
baseUrl: "zcode://native",
|
|
123
|
+
};
|
|
124
|
+
}
|
|
@@ -1,65 +1,65 @@
|
|
|
1
|
-
// zcode 模型目录:内置 Start/Coding Plan 目录 + allowlist 只增不减补齐 + 目录服务(可选按额度 capabilities 过滤)。
|
|
2
|
-
import { joinModelId } from "../model-id.js";
|
|
3
|
-
import { loadProviderAllowedModels, saveProviderAllowedModels } from "../../state.js";
|
|
4
|
-
import { ZCODE_MODEL_CATALOG } from "./const.js";
|
|
5
|
-
|
|
6
|
-
const CREATED = 1753600000;
|
|
7
|
-
const CACHE_TTL_MS = 10 * 60 * 1000;
|
|
8
|
-
|
|
9
|
-
// 登录后开箱可用:内置目录只增不减并入 allowlist(对齐 cline allowlist-sync 先例,避免显式请求 403)。
|
|
10
|
-
export function seedZcodeAllowlist({ ids = ZCODE_MODEL_CATALOG, file } = {}) {
|
|
11
|
-
const opts = file ? { file } : {};
|
|
12
|
-
const cur = loadProviderAllowedModels("zcode", opts);
|
|
13
|
-
const added = ids.filter((m) => !cur.includes(m));
|
|
14
|
-
if (added.length) saveProviderAllowedModels("zcode", [...cur, ...added], opts);
|
|
15
|
-
return { allowed: [...cur, ...added], added };
|
|
16
|
-
}
|
|
17
|
-
|
|
18
|
-
// 对外目录条目(免费池:价格 0.00;id 带 provider 前缀,对齐 joinModelId 契约)。
|
|
19
|
-
export function buildZcodeModelList({ id = "zcode", ids = ZCODE_MODEL_CATALOG } = {}) {
|
|
20
|
-
return ids.map((raw) => ({
|
|
21
|
-
id: joinModelId(id, raw),
|
|
22
|
-
object: "model",
|
|
23
|
-
created: CREATED,
|
|
24
|
-
owned_by: "zcode",
|
|
25
|
-
name: raw,
|
|
26
|
-
price: "0.00",
|
|
27
|
-
enable: true,
|
|
28
|
-
free: true,
|
|
29
|
-
mark: "*", // 可用标注:免费额度内,登录后可直接调用
|
|
30
|
-
}));
|
|
31
|
-
}
|
|
32
|
-
|
|
33
|
-
// 目录服务:静态目录 + 可选 probe()(额度 capabilities 的 live id 列表)过滤;10 分钟缓存。
|
|
34
|
-
// probe 缺失/失败/为空 → 返回全量目录(宁可多列不可误删)。
|
|
35
|
-
export function createModelsService({ id = "zcode", probe, hasAccount = () => true, clock = Date.now } = {}) {
|
|
36
|
-
let cache = null;
|
|
37
|
-
let fetchedAt = 0;
|
|
38
|
-
|
|
39
|
-
async function listModels() {
|
|
40
|
-
if (!hasAccount()) return []; // 未登录:空目录(不假装有额度)
|
|
41
|
-
const now = clock();
|
|
42
|
-
if (cache && now - fetchedAt < CACHE_TTL_MS) return cache;
|
|
43
|
-
let ids = ZCODE_MODEL_CATALOG;
|
|
44
|
-
if (typeof probe === "function") {
|
|
45
|
-
try {
|
|
46
|
-
const live = await probe();
|
|
47
|
-
if (Array.isArray(live) && live.length) {
|
|
48
|
-
const low = new Set(live.map((m) => String(m).toLowerCase()));
|
|
49
|
-
const filtered = ZCODE_MODEL_CATALOG.filter((m) => low.has(m.toLowerCase()));
|
|
50
|
-
ids = filtered.length ? filtered : ZCODE_MODEL_CATALOG;
|
|
51
|
-
}
|
|
52
|
-
} catch {}
|
|
53
|
-
}
|
|
54
|
-
cache = buildZcodeModelList({ id, ids });
|
|
55
|
-
fetchedAt = now;
|
|
56
|
-
return cache;
|
|
57
|
-
}
|
|
58
|
-
|
|
59
|
-
function clearCache() {
|
|
60
|
-
cache = null;
|
|
61
|
-
fetchedAt = 0;
|
|
62
|
-
}
|
|
63
|
-
|
|
64
|
-
return { listModels, clearCache, _cache: () => cache };
|
|
65
|
-
}
|
|
1
|
+
// zcode 模型目录:内置 Start/Coding Plan 目录 + allowlist 只增不减补齐 + 目录服务(可选按额度 capabilities 过滤)。
|
|
2
|
+
import { joinModelId } from "../model-id.js";
|
|
3
|
+
import { loadProviderAllowedModels, saveProviderAllowedModels } from "../../state.js";
|
|
4
|
+
import { ZCODE_MODEL_CATALOG } from "./const.js";
|
|
5
|
+
|
|
6
|
+
const CREATED = 1753600000;
|
|
7
|
+
const CACHE_TTL_MS = 10 * 60 * 1000;
|
|
8
|
+
|
|
9
|
+
// 登录后开箱可用:内置目录只增不减并入 allowlist(对齐 cline allowlist-sync 先例,避免显式请求 403)。
|
|
10
|
+
export function seedZcodeAllowlist({ ids = ZCODE_MODEL_CATALOG, file } = {}) {
|
|
11
|
+
const opts = file ? { file } : {};
|
|
12
|
+
const cur = loadProviderAllowedModels("zcode", opts);
|
|
13
|
+
const added = ids.filter((m) => !cur.includes(m));
|
|
14
|
+
if (added.length) saveProviderAllowedModels("zcode", [...cur, ...added], opts);
|
|
15
|
+
return { allowed: [...cur, ...added], added };
|
|
16
|
+
}
|
|
17
|
+
|
|
18
|
+
// 对外目录条目(免费池:价格 0.00;id 带 provider 前缀,对齐 joinModelId 契约)。
|
|
19
|
+
export function buildZcodeModelList({ id = "zcode", ids = ZCODE_MODEL_CATALOG } = {}) {
|
|
20
|
+
return ids.map((raw) => ({
|
|
21
|
+
id: joinModelId(id, raw),
|
|
22
|
+
object: "model",
|
|
23
|
+
created: CREATED,
|
|
24
|
+
owned_by: "zcode",
|
|
25
|
+
name: raw,
|
|
26
|
+
price: "0.00",
|
|
27
|
+
enable: true,
|
|
28
|
+
free: true,
|
|
29
|
+
mark: "*", // 可用标注:免费额度内,登录后可直接调用
|
|
30
|
+
}));
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
// 目录服务:静态目录 + 可选 probe()(额度 capabilities 的 live id 列表)过滤;10 分钟缓存。
|
|
34
|
+
// probe 缺失/失败/为空 → 返回全量目录(宁可多列不可误删)。
|
|
35
|
+
export function createModelsService({ id = "zcode", probe, hasAccount = () => true, clock = Date.now } = {}) {
|
|
36
|
+
let cache = null;
|
|
37
|
+
let fetchedAt = 0;
|
|
38
|
+
|
|
39
|
+
async function listModels() {
|
|
40
|
+
if (!hasAccount()) return []; // 未登录:空目录(不假装有额度)
|
|
41
|
+
const now = clock();
|
|
42
|
+
if (cache && now - fetchedAt < CACHE_TTL_MS) return cache;
|
|
43
|
+
let ids = ZCODE_MODEL_CATALOG;
|
|
44
|
+
if (typeof probe === "function") {
|
|
45
|
+
try {
|
|
46
|
+
const live = await probe();
|
|
47
|
+
if (Array.isArray(live) && live.length) {
|
|
48
|
+
const low = new Set(live.map((m) => String(m).toLowerCase()));
|
|
49
|
+
const filtered = ZCODE_MODEL_CATALOG.filter((m) => low.has(m.toLowerCase()));
|
|
50
|
+
ids = filtered.length ? filtered : ZCODE_MODEL_CATALOG;
|
|
51
|
+
}
|
|
52
|
+
} catch {}
|
|
53
|
+
}
|
|
54
|
+
cache = buildZcodeModelList({ id, ids });
|
|
55
|
+
fetchedAt = now;
|
|
56
|
+
return cache;
|
|
57
|
+
}
|
|
58
|
+
|
|
59
|
+
function clearCache() {
|
|
60
|
+
cache = null;
|
|
61
|
+
fetchedAt = 0;
|
|
62
|
+
}
|
|
63
|
+
|
|
64
|
+
return { listModels, clearCache, _cache: () => cache };
|
|
65
|
+
}
|
|
@@ -1,120 +1,120 @@
|
|
|
1
|
-
// zcode CLI 轮询 OAuth:POST cli/init → 浏览器授权 → GET cli/poll/{flow_id} 拿 zcode JWT。
|
|
2
|
-
// 契约对齐 zcode-switch oauth.rs(init 自带 poll_token 鉴权;服务端返回的 poll_token 优先;3004=流程过期)。
|
|
3
|
-
import { randomBytes } from "node:crypto";
|
|
4
|
-
import { timeoutSignal } from "../../compat.js";
|
|
5
|
-
import { sleep } from "../base.js";
|
|
6
|
-
import { ZCODE_CLI_INIT_URL, zcodeCliPollUrl, zcodeAppVersion } from "./const.js";
|
|
7
|
-
import { buildZcodeHeaders } from "./headers.js";
|
|
8
|
-
|
|
9
|
-
const INIT_TIMEOUT_MS = 20_000;
|
|
10
|
-
const POLL_TIMEOUT_MS = 15_000;
|
|
11
|
-
const DEFAULT_FLOW_TTL_MS = 300_000;
|
|
12
|
-
|
|
13
|
-
export class ZcodeOAuthError extends Error {
|
|
14
|
-
constructor(message, code = 0) {
|
|
15
|
-
super(message);
|
|
16
|
-
this.name = "ZcodeOAuthError";
|
|
17
|
-
this.code = code;
|
|
18
|
-
}
|
|
19
|
-
}
|
|
20
|
-
|
|
21
|
-
function parseExpiresAtMs(v) {
|
|
22
|
-
const n = Number(v);
|
|
23
|
-
return Number.isFinite(n) && n > 0 ? n * 1000 : 0;
|
|
24
|
-
}
|
|
25
|
-
|
|
26
|
-
export async function initZcodeFlow({ provider = "zai", fetchImpl, deviceMid, version } = {}) {
|
|
27
|
-
const pollToken = randomBytes(32).toString("hex");
|
|
28
|
-
const headers = {
|
|
29
|
-
...buildZcodeHeaders({ token: pollToken, deviceMid, version: version || zcodeAppVersion() }),
|
|
30
|
-
"Content-Type": "application/json",
|
|
31
|
-
};
|
|
32
|
-
const res = await fetchImpl(ZCODE_CLI_INIT_URL, {
|
|
33
|
-
method: "POST",
|
|
34
|
-
headers,
|
|
35
|
-
body: JSON.stringify({ provider }),
|
|
36
|
-
signal: timeoutSignal(INIT_TIMEOUT_MS),
|
|
37
|
-
});
|
|
38
|
-
const payload = await res.json().catch(() => null);
|
|
39
|
-
if (!res.ok || payload?.code !== 0) {
|
|
40
|
-
const msg = String(payload?.msg || "").trim();
|
|
41
|
-
throw new ZcodeOAuthError(`zcode 登录初始化失败:HTTP ${res.status}${msg ? ` ${msg}` : ""}`, Number(payload?.code) || 0);
|
|
42
|
-
}
|
|
43
|
-
const data = payload?.data || {};
|
|
44
|
-
const flowId = String(data.flow_id || "").trim();
|
|
45
|
-
const authorizeUrl = String(data.authorize_url || "").trim();
|
|
46
|
-
if (!flowId || !authorizeUrl) throw new ZcodeOAuthError("zcode 登录初始化响应缺少 flow_id/authorize_url");
|
|
47
|
-
let parsed;
|
|
48
|
-
try { parsed = new URL(authorizeUrl); } catch { throw new ZcodeOAuthError("zcode authorize_url 非法"); }
|
|
49
|
-
if (parsed.protocol !== "https:") throw new ZcodeOAuthError("zcode authorize_url 必须 https");
|
|
50
|
-
const state = parsed.searchParams.get("state") || "";
|
|
51
|
-
if (!state) throw new ZcodeOAuthError("zcode authorize_url 缺少 state");
|
|
52
|
-
return {
|
|
53
|
-
provider,
|
|
54
|
-
flowId,
|
|
55
|
-
authorizeUrl,
|
|
56
|
-
state,
|
|
57
|
-
deviceMid: String(deviceMid || ""),
|
|
58
|
-
pollUrl: zcodeCliPollUrl(flowId),
|
|
59
|
-
pollToken: String(data.poll_token || "").trim() || pollToken,
|
|
60
|
-
expiresAtMs: parseExpiresAtMs(data.expires_at) || Date.now() + DEFAULT_FLOW_TTL_MS,
|
|
61
|
-
pollIntervalMs: Math.max(1000, Number(data.poll_interval_sec) * 1000 || 3000),
|
|
62
|
-
};
|
|
63
|
-
}
|
|
64
|
-
|
|
65
|
-
// 单次轮询:pending 继续、ready 返回凭据;3004/failed 抛终态;网络抖动按 pending(不判死)。
|
|
66
|
-
export async function pollZcodeFlowOnce({ pollUrl, pollToken, deviceMid, version, fetchImpl } = {}) {
|
|
67
|
-
let res;
|
|
68
|
-
try {
|
|
69
|
-
res = await fetchImpl(pollUrl, {
|
|
70
|
-
headers: buildZcodeHeaders({ token: pollToken, deviceMid, version: version || zcodeAppVersion() }),
|
|
71
|
-
signal: timeoutSignal(POLL_TIMEOUT_MS),
|
|
72
|
-
});
|
|
73
|
-
} catch {
|
|
74
|
-
return { status: "pending" };
|
|
75
|
-
}
|
|
76
|
-
if (res.status >= 400) {
|
|
77
|
-
if (res.status === 408 || res.status === 429) return { status: "pending" };
|
|
78
|
-
const body = await res.json().catch(() => null);
|
|
79
|
-
const code = Number(body?.code) || 0;
|
|
80
|
-
if (code === 3004) throw new ZcodeOAuthError("授权已过期,请重新运行 login", 3004);
|
|
81
|
-
const msg = String(body?.msg || "").trim();
|
|
82
|
-
throw new ZcodeOAuthError(`zcode 登录轮询失败:HTTP ${res.status}${msg ? ` ${msg}` : ""}`, code);
|
|
83
|
-
}
|
|
84
|
-
const payload = await res.json().catch(() => null);
|
|
85
|
-
if (!payload || payload.code !== 0) {
|
|
86
|
-
const msg = String(payload?.msg || "响应异常").trim();
|
|
87
|
-
throw new ZcodeOAuthError(`zcode 登录轮询失败:${msg}`, Number(payload?.code) || 0);
|
|
88
|
-
}
|
|
89
|
-
const data = payload?.data || {};
|
|
90
|
-
const status = String(data.status || "").trim();
|
|
91
|
-
if (status === "pending") return { status: "pending" };
|
|
92
|
-
if (status === "failed") throw new ZcodeOAuthError("授权失败(服务端标记 failed),请重试", 0);
|
|
93
|
-
if (status === "ready") {
|
|
94
|
-
const jwt = String(data.token || "").trim();
|
|
95
|
-
const accessToken = String(data?.zai?.access_token || data?.bigmodel?.access_token || "").trim();
|
|
96
|
-
const user = {
|
|
97
|
-
userId: String(data?.user?.user_id || "").trim(),
|
|
98
|
-
name: String(data?.user?.name || "").trim(),
|
|
99
|
-
email: String(data?.user?.email || "").trim(),
|
|
100
|
-
};
|
|
101
|
-
if (!jwt || !user.userId) throw new ZcodeOAuthError("zcode 登录响应缺少 token/user_id");
|
|
102
|
-
return { status: "ready", jwt, accessToken, user };
|
|
103
|
-
}
|
|
104
|
-
throw new ZcodeOAuthError(`zcode 登录轮询未知状态:${status || "(空)"}`);
|
|
105
|
-
}
|
|
106
|
-
|
|
107
|
-
export async function waitForZcodeLogin({ flow, fetchImpl, log = () => {}, sleepFn = sleep, now = Date.now } = {}) {
|
|
108
|
-
if (!flow?.pollUrl) throw new ZcodeOAuthError("缺少轮询信息");
|
|
109
|
-
while (now() < (flow.expiresAtMs || 0)) {
|
|
110
|
-
const out = await pollZcodeFlowOnce({
|
|
111
|
-
pollUrl: flow.pollUrl,
|
|
112
|
-
pollToken: flow.pollToken,
|
|
113
|
-
deviceMid: flow.deviceMid,
|
|
114
|
-
fetchImpl,
|
|
115
|
-
});
|
|
116
|
-
if (out.status === "ready") return { ...out, provider: flow.provider || "zai" };
|
|
117
|
-
await sleepFn(flow.pollIntervalMs || 3000);
|
|
118
|
-
}
|
|
119
|
-
throw new ZcodeOAuthError("授权已过期,请重新运行 login", 3004);
|
|
120
|
-
}
|
|
1
|
+
// zcode CLI 轮询 OAuth:POST cli/init → 浏览器授权 → GET cli/poll/{flow_id} 拿 zcode JWT。
|
|
2
|
+
// 契约对齐 zcode-switch oauth.rs(init 自带 poll_token 鉴权;服务端返回的 poll_token 优先;3004=流程过期)。
|
|
3
|
+
import { randomBytes } from "node:crypto";
|
|
4
|
+
import { timeoutSignal } from "../../compat.js";
|
|
5
|
+
import { sleep } from "../base.js";
|
|
6
|
+
import { ZCODE_CLI_INIT_URL, zcodeCliPollUrl, zcodeAppVersion } from "./const.js";
|
|
7
|
+
import { buildZcodeHeaders } from "./headers.js";
|
|
8
|
+
|
|
9
|
+
const INIT_TIMEOUT_MS = 20_000;
|
|
10
|
+
const POLL_TIMEOUT_MS = 15_000;
|
|
11
|
+
const DEFAULT_FLOW_TTL_MS = 300_000;
|
|
12
|
+
|
|
13
|
+
export class ZcodeOAuthError extends Error {
|
|
14
|
+
constructor(message, code = 0) {
|
|
15
|
+
super(message);
|
|
16
|
+
this.name = "ZcodeOAuthError";
|
|
17
|
+
this.code = code;
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
function parseExpiresAtMs(v) {
|
|
22
|
+
const n = Number(v);
|
|
23
|
+
return Number.isFinite(n) && n > 0 ? n * 1000 : 0;
|
|
24
|
+
}
|
|
25
|
+
|
|
26
|
+
export async function initZcodeFlow({ provider = "zai", fetchImpl, deviceMid, version } = {}) {
|
|
27
|
+
const pollToken = randomBytes(32).toString("hex");
|
|
28
|
+
const headers = {
|
|
29
|
+
...buildZcodeHeaders({ token: pollToken, deviceMid, version: version || zcodeAppVersion() }),
|
|
30
|
+
"Content-Type": "application/json",
|
|
31
|
+
};
|
|
32
|
+
const res = await fetchImpl(ZCODE_CLI_INIT_URL, {
|
|
33
|
+
method: "POST",
|
|
34
|
+
headers,
|
|
35
|
+
body: JSON.stringify({ provider }),
|
|
36
|
+
signal: timeoutSignal(INIT_TIMEOUT_MS),
|
|
37
|
+
});
|
|
38
|
+
const payload = await res.json().catch(() => null);
|
|
39
|
+
if (!res.ok || payload?.code !== 0) {
|
|
40
|
+
const msg = String(payload?.msg || "").trim();
|
|
41
|
+
throw new ZcodeOAuthError(`zcode 登录初始化失败:HTTP ${res.status}${msg ? ` ${msg}` : ""}`, Number(payload?.code) || 0);
|
|
42
|
+
}
|
|
43
|
+
const data = payload?.data || {};
|
|
44
|
+
const flowId = String(data.flow_id || "").trim();
|
|
45
|
+
const authorizeUrl = String(data.authorize_url || "").trim();
|
|
46
|
+
if (!flowId || !authorizeUrl) throw new ZcodeOAuthError("zcode 登录初始化响应缺少 flow_id/authorize_url");
|
|
47
|
+
let parsed;
|
|
48
|
+
try { parsed = new URL(authorizeUrl); } catch { throw new ZcodeOAuthError("zcode authorize_url 非法"); }
|
|
49
|
+
if (parsed.protocol !== "https:") throw new ZcodeOAuthError("zcode authorize_url 必须 https");
|
|
50
|
+
const state = parsed.searchParams.get("state") || "";
|
|
51
|
+
if (!state) throw new ZcodeOAuthError("zcode authorize_url 缺少 state");
|
|
52
|
+
return {
|
|
53
|
+
provider,
|
|
54
|
+
flowId,
|
|
55
|
+
authorizeUrl,
|
|
56
|
+
state,
|
|
57
|
+
deviceMid: String(deviceMid || ""),
|
|
58
|
+
pollUrl: zcodeCliPollUrl(flowId),
|
|
59
|
+
pollToken: String(data.poll_token || "").trim() || pollToken,
|
|
60
|
+
expiresAtMs: parseExpiresAtMs(data.expires_at) || Date.now() + DEFAULT_FLOW_TTL_MS,
|
|
61
|
+
pollIntervalMs: Math.max(1000, Number(data.poll_interval_sec) * 1000 || 3000),
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
// 单次轮询:pending 继续、ready 返回凭据;3004/failed 抛终态;网络抖动按 pending(不判死)。
|
|
66
|
+
export async function pollZcodeFlowOnce({ pollUrl, pollToken, deviceMid, version, fetchImpl } = {}) {
|
|
67
|
+
let res;
|
|
68
|
+
try {
|
|
69
|
+
res = await fetchImpl(pollUrl, {
|
|
70
|
+
headers: buildZcodeHeaders({ token: pollToken, deviceMid, version: version || zcodeAppVersion() }),
|
|
71
|
+
signal: timeoutSignal(POLL_TIMEOUT_MS),
|
|
72
|
+
});
|
|
73
|
+
} catch {
|
|
74
|
+
return { status: "pending" };
|
|
75
|
+
}
|
|
76
|
+
if (res.status >= 400) {
|
|
77
|
+
if (res.status === 408 || res.status === 429) return { status: "pending" };
|
|
78
|
+
const body = await res.json().catch(() => null);
|
|
79
|
+
const code = Number(body?.code) || 0;
|
|
80
|
+
if (code === 3004) throw new ZcodeOAuthError("授权已过期,请重新运行 login", 3004);
|
|
81
|
+
const msg = String(body?.msg || "").trim();
|
|
82
|
+
throw new ZcodeOAuthError(`zcode 登录轮询失败:HTTP ${res.status}${msg ? ` ${msg}` : ""}`, code);
|
|
83
|
+
}
|
|
84
|
+
const payload = await res.json().catch(() => null);
|
|
85
|
+
if (!payload || payload.code !== 0) {
|
|
86
|
+
const msg = String(payload?.msg || "响应异常").trim();
|
|
87
|
+
throw new ZcodeOAuthError(`zcode 登录轮询失败:${msg}`, Number(payload?.code) || 0);
|
|
88
|
+
}
|
|
89
|
+
const data = payload?.data || {};
|
|
90
|
+
const status = String(data.status || "").trim();
|
|
91
|
+
if (status === "pending") return { status: "pending" };
|
|
92
|
+
if (status === "failed") throw new ZcodeOAuthError("授权失败(服务端标记 failed),请重试", 0);
|
|
93
|
+
if (status === "ready") {
|
|
94
|
+
const jwt = String(data.token || "").trim();
|
|
95
|
+
const accessToken = String(data?.zai?.access_token || data?.bigmodel?.access_token || "").trim();
|
|
96
|
+
const user = {
|
|
97
|
+
userId: String(data?.user?.user_id || "").trim(),
|
|
98
|
+
name: String(data?.user?.name || "").trim(),
|
|
99
|
+
email: String(data?.user?.email || "").trim(),
|
|
100
|
+
};
|
|
101
|
+
if (!jwt || !user.userId) throw new ZcodeOAuthError("zcode 登录响应缺少 token/user_id");
|
|
102
|
+
return { status: "ready", jwt, accessToken, user };
|
|
103
|
+
}
|
|
104
|
+
throw new ZcodeOAuthError(`zcode 登录轮询未知状态:${status || "(空)"}`);
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
export async function waitForZcodeLogin({ flow, fetchImpl, log = () => {}, sleepFn = sleep, now = Date.now } = {}) {
|
|
108
|
+
if (!flow?.pollUrl) throw new ZcodeOAuthError("缺少轮询信息");
|
|
109
|
+
while (now() < (flow.expiresAtMs || 0)) {
|
|
110
|
+
const out = await pollZcodeFlowOnce({
|
|
111
|
+
pollUrl: flow.pollUrl,
|
|
112
|
+
pollToken: flow.pollToken,
|
|
113
|
+
deviceMid: flow.deviceMid,
|
|
114
|
+
fetchImpl,
|
|
115
|
+
});
|
|
116
|
+
if (out.status === "ready") return { ...out, provider: flow.provider || "zai" };
|
|
117
|
+
await sleepFn(flow.pollIntervalMs || 3000);
|
|
118
|
+
}
|
|
119
|
+
throw new ZcodeOAuthError("授权已过期,请重新运行 login", 3004);
|
|
120
|
+
}
|