mslxdff 0.1.200 → 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 -74
- 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/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 -263
- 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
package/src/providers/generic.js
CHANGED
|
@@ -1,156 +1,156 @@
|
|
|
1
|
-
import { createKeyRing } from "./keyring.js";
|
|
2
|
-
import { loadProviderKeys, loadProviderBaseUrl, loadProviderModelsPath, loadProviderChatPath } from "../state.js";
|
|
3
|
-
import { envInt, joinUrl, getUndici, createAgent, collectApiKeysGeneric, createChatRunner, createListModelsRunner, createPreheatRunner } from "./base.js";
|
|
4
|
-
import { compatFetch } from "../compat.js";
|
|
5
|
-
import crypto from "node:crypto";
|
|
6
|
-
import { genId, opencodeUa, digestIdTail } from "../opencode-identity.js";
|
|
7
|
-
import { isResponsesModel } from "../upstream-responses.js";
|
|
8
|
-
import { createResponsesChannel, resolveResponsesPath, envSlug } from "./responses-channel.js";
|
|
9
|
-
import { resolveEngineMode } from "../upstream-engine/mode.js";
|
|
10
|
-
|
|
11
|
-
const { UndiciFetch } = getUndici();
|
|
12
|
-
|
|
13
|
-
function resolveBaseUrl(id, baseUrl) {
|
|
14
|
-
if (baseUrl) return String(baseUrl).trim().replace(/\/+$/, "");
|
|
15
|
-
const env = loadProviderBaseUrl(id);
|
|
16
|
-
if (env) return env;
|
|
17
|
-
return "";
|
|
18
|
-
}
|
|
19
|
-
|
|
20
|
-
function isOpencodeHost(baseUrl) {
|
|
21
|
-
return /opencode\.ai/i.test(String(baseUrl || ""));
|
|
22
|
-
}
|
|
23
|
-
|
|
24
|
-
// Console Go (zen/go) 要求 x-opencode-session 才能路由(缺失 → 400 MissingSessionID)。
|
|
25
|
-
// 会话取值:已合规(ses_ 前缀)直接用;客户端透传的任意串做 sha1 摘要派生(稳定+合规);
|
|
26
|
-
// 无则每请求 fresh(与 -chat curl / bench 直连行为一致)。
|
|
27
|
-
function resolveGoSession(sessionId) {
|
|
28
|
-
const raw = String(sessionId || "").trim();
|
|
29
|
-
if (/^ses_/.test(raw)) return raw;
|
|
30
|
-
if (raw) {
|
|
31
|
-
try {
|
|
32
|
-
return `ses_${digestIdTail(crypto.createHash("sha1").update(raw).digest())}`;
|
|
33
|
-
} catch {
|
|
34
|
-
return genId("ses_");
|
|
35
|
-
}
|
|
36
|
-
}
|
|
37
|
-
return genId("ses_");
|
|
38
|
-
}
|
|
39
|
-
|
|
40
|
-
export function createGenericProvider({
|
|
41
|
-
id,
|
|
42
|
-
baseUrl,
|
|
43
|
-
apiKeys,
|
|
44
|
-
apiKey,
|
|
45
|
-
modelsPath,
|
|
46
|
-
chatPath,
|
|
47
|
-
responsesPath,
|
|
48
|
-
connectTimeoutMs = Number(process.env.MSLXDFF_GENERIC_TIMEOUT_MS) || 30_000,
|
|
49
|
-
cooldownMs = envInt("MSLXDFF_GENERIC_COOLDOWN_MS", 30_000),
|
|
50
|
-
retry = {
|
|
51
|
-
network: { attempts: 2, delayMs: 300 },
|
|
52
|
-
429: { attempts: 1, delayMs: 100 },
|
|
53
|
-
502: { attempts: 1, delayMs: 100 },
|
|
54
|
-
503: { attempts: 1, delayMs: 100 },
|
|
55
|
-
504: { attempts: 1, delayMs: 100 },
|
|
56
|
-
},
|
|
57
|
-
fetchImpl,
|
|
58
|
-
headers: extraHeaders,
|
|
59
|
-
mapModel,
|
|
60
|
-
noAgent = false,
|
|
61
|
-
file,
|
|
62
|
-
} = {}) {
|
|
63
|
-
if (!id) throw new Error("generic provider requires id");
|
|
64
|
-
const resolvedBase = resolveBaseUrl(id, baseUrl);
|
|
65
|
-
if (!resolvedBase) throw new Error(`generic provider ${id}: missing baseUrl`);
|
|
66
|
-
if (!fetchImpl) fetchImpl = UndiciFetch || compatFetch;
|
|
67
|
-
const resolvedModelsPath = modelsPath || loadProviderModelsPath(id, file ? { file } : {});
|
|
68
|
-
const resolvedChatPath = chatPath || loadProviderChatPath(id, file ? { file } : {});
|
|
69
|
-
const ring = createKeyRing(collectApiKeysGeneric(id, apiKeys, apiKey, loadProviderKeys), { cooldownMs });
|
|
70
|
-
const goHost = isOpencodeHost(resolvedBase);
|
|
71
|
-
|
|
72
|
-
let dispatcher = null;
|
|
73
|
-
let agent = null;
|
|
74
|
-
if (!noAgent) {
|
|
75
|
-
const a = createAgent({
|
|
76
|
-
keepAliveTimeout: envInt("MSLXDFF_GENERIC_KEEPALIVE_TIMEOUT", 30_000),
|
|
77
|
-
keepAliveMaxTimeout: envInt("MSLXDFF_GENERIC_KEEPALIVE_MAX_TIMEOUT", 60_000),
|
|
78
|
-
connections: envInt("MSLXDFF_GENERIC_KEEPALIVE_CONNECTIONS", 20),
|
|
79
|
-
});
|
|
80
|
-
agent = a.agent; dispatcher = a.dispatcher;
|
|
81
|
-
}
|
|
82
|
-
|
|
83
|
-
function buildHeaders(body, key, opts) {
|
|
84
|
-
const isStream = body?.stream !== false;
|
|
85
|
-
const h = {
|
|
86
|
-
"Content-Type": "application/json",
|
|
87
|
-
Accept: isStream ? "text/event-stream" : "*/*",
|
|
88
|
-
"User-Agent": "mslxdff",
|
|
89
|
-
};
|
|
90
|
-
if (key) h["Authorization"] = `Bearer ${key}`;
|
|
91
|
-
const out = { ...h, ...extraHeaders };
|
|
92
|
-
// ocgo(opencode.ai 域名):补 Console Go 路由所需的 opencode 身份头
|
|
93
|
-
if (goHost) {
|
|
94
|
-
out["User-Agent"] = opencodeUa();
|
|
95
|
-
out["x-opencode-client"] = "desktop";
|
|
96
|
-
out["x-opencode-session"] = resolveGoSession(opts?.sessionId);
|
|
97
|
-
out["x-opencode-request"] = genId("msg_");
|
|
98
|
-
out["x-opencode-project"] = "global";
|
|
99
|
-
}
|
|
100
|
-
return out;
|
|
101
|
-
}
|
|
102
|
-
|
|
103
|
-
// base 的 attemptOnce 只调 buildHeaders(body, key):用闭包把 opts.sessionId 带进去
|
|
104
|
-
function scopedRunner(activeRing, opts) {
|
|
105
|
-
return createChatRunner({
|
|
106
|
-
id, ring: activeRing, cooldownMs, retry, fetchImpl, dispatcher,
|
|
107
|
-
buildHeaders: (b, k) => buildHeaders(b, k, opts),
|
|
108
|
-
getUrl: () => joinUrl(resolvedBase, resolvedChatPath),
|
|
109
|
-
connectTimeoutMs,
|
|
110
|
-
});
|
|
111
|
-
}
|
|
112
|
-
|
|
113
|
-
// responses 类模型(muse-spark* 等)只在 /responses 挂载 → 单独通道,出参恒 chat 形状。
|
|
114
|
-
const responses = createResponsesChannel({
|
|
115
|
-
id,
|
|
116
|
-
url: joinUrl(resolvedBase, resolveResponsesPath(id, responsesPath)),
|
|
117
|
-
connectTimeoutMs,
|
|
118
|
-
retry,
|
|
119
|
-
cooldownMs,
|
|
120
|
-
// 供应商级 <ID>_SDK 未设置即继承全局总闸(MSLXDFF_UPSTREAM_ENGINE)
|
|
121
|
-
sdkEnabled: resolveEngineMode(process.env, `MSLXDFF_${envSlug(id)}_SDK`) === "sdk",
|
|
122
|
-
buildHeaders,
|
|
123
|
-
fetchImpl,
|
|
124
|
-
dispatcher,
|
|
125
|
-
});
|
|
126
|
-
|
|
127
|
-
async function chat(body, opts) {
|
|
128
|
-
const sourceKey = `MSLXDFF_${envSlug(id)}_KEY`;
|
|
129
|
-
if (isResponsesModel(body?.model)) return responses.run(body, ring, sourceKey, opts);
|
|
130
|
-
const { runChat } = scopedRunner(ring, opts);
|
|
131
|
-
return runChat(body, ring, sourceKey);
|
|
132
|
-
}
|
|
133
|
-
async function chatWithKeys(body, keys, opts) {
|
|
134
|
-
const tmp = createKeyRing(keys, { cooldownMs });
|
|
135
|
-
if (isResponsesModel(body?.model)) return responses.run(body, tmp, "shared provider keys", opts);
|
|
136
|
-
const { runChat } = scopedRunner(tmp, opts);
|
|
137
|
-
return runChat(body, tmp, "shared provider keys");
|
|
138
|
-
}
|
|
139
|
-
|
|
140
|
-
const { listModels } = createListModelsRunner({
|
|
141
|
-
id, ring, dispatcher, fetchImpl,
|
|
142
|
-
getUrl: () => joinUrl(resolvedBase, resolvedModelsPath),
|
|
143
|
-
mapModel,
|
|
144
|
-
});
|
|
145
|
-
|
|
146
|
-
const { preheat } = createPreheatRunner({
|
|
147
|
-
dispatcher, fetchImpl, getUrl: () => joinUrl(resolvedBase, resolvedModelsPath),
|
|
148
|
-
id, ring, loadKeys: loadProviderKeys,
|
|
149
|
-
});
|
|
150
|
-
|
|
151
|
-
async function close() {
|
|
152
|
-
if (agent && typeof agent.close === "function") { try { await agent.close(); } catch {} }
|
|
153
|
-
}
|
|
154
|
-
|
|
155
|
-
return { id, chat, chatWithKeys, listModels, preheat, close, agent, keyRing: ring, baseUrl: resolvedBase, responsesUrl: responses.url };
|
|
156
|
-
}
|
|
1
|
+
import { createKeyRing } from "./keyring.js";
|
|
2
|
+
import { loadProviderKeys, loadProviderBaseUrl, loadProviderModelsPath, loadProviderChatPath } from "../state.js";
|
|
3
|
+
import { envInt, joinUrl, getUndici, createAgent, collectApiKeysGeneric, createChatRunner, createListModelsRunner, createPreheatRunner } from "./base.js";
|
|
4
|
+
import { compatFetch } from "../compat.js";
|
|
5
|
+
import crypto from "node:crypto";
|
|
6
|
+
import { genId, opencodeUa, digestIdTail } from "../opencode-identity.js";
|
|
7
|
+
import { isResponsesModel } from "../upstream-responses.js";
|
|
8
|
+
import { createResponsesChannel, resolveResponsesPath, envSlug } from "./responses-channel.js";
|
|
9
|
+
import { resolveEngineMode } from "../upstream-engine/mode.js";
|
|
10
|
+
|
|
11
|
+
const { UndiciFetch } = getUndici();
|
|
12
|
+
|
|
13
|
+
function resolveBaseUrl(id, baseUrl) {
|
|
14
|
+
if (baseUrl) return String(baseUrl).trim().replace(/\/+$/, "");
|
|
15
|
+
const env = loadProviderBaseUrl(id);
|
|
16
|
+
if (env) return env;
|
|
17
|
+
return "";
|
|
18
|
+
}
|
|
19
|
+
|
|
20
|
+
function isOpencodeHost(baseUrl) {
|
|
21
|
+
return /opencode\.ai/i.test(String(baseUrl || ""));
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
// Console Go (zen/go) 要求 x-opencode-session 才能路由(缺失 → 400 MissingSessionID)。
|
|
25
|
+
// 会话取值:已合规(ses_ 前缀)直接用;客户端透传的任意串做 sha1 摘要派生(稳定+合规);
|
|
26
|
+
// 无则每请求 fresh(与 -chat curl / bench 直连行为一致)。
|
|
27
|
+
function resolveGoSession(sessionId) {
|
|
28
|
+
const raw = String(sessionId || "").trim();
|
|
29
|
+
if (/^ses_/.test(raw)) return raw;
|
|
30
|
+
if (raw) {
|
|
31
|
+
try {
|
|
32
|
+
return `ses_${digestIdTail(crypto.createHash("sha1").update(raw).digest())}`;
|
|
33
|
+
} catch {
|
|
34
|
+
return genId("ses_");
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
return genId("ses_");
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
export function createGenericProvider({
|
|
41
|
+
id,
|
|
42
|
+
baseUrl,
|
|
43
|
+
apiKeys,
|
|
44
|
+
apiKey,
|
|
45
|
+
modelsPath,
|
|
46
|
+
chatPath,
|
|
47
|
+
responsesPath,
|
|
48
|
+
connectTimeoutMs = Number(process.env.MSLXDFF_GENERIC_TIMEOUT_MS) || 30_000,
|
|
49
|
+
cooldownMs = envInt("MSLXDFF_GENERIC_COOLDOWN_MS", 30_000),
|
|
50
|
+
retry = {
|
|
51
|
+
network: { attempts: 2, delayMs: 300 },
|
|
52
|
+
429: { attempts: 1, delayMs: 100 },
|
|
53
|
+
502: { attempts: 1, delayMs: 100 },
|
|
54
|
+
503: { attempts: 1, delayMs: 100 },
|
|
55
|
+
504: { attempts: 1, delayMs: 100 },
|
|
56
|
+
},
|
|
57
|
+
fetchImpl,
|
|
58
|
+
headers: extraHeaders,
|
|
59
|
+
mapModel,
|
|
60
|
+
noAgent = false,
|
|
61
|
+
file,
|
|
62
|
+
} = {}) {
|
|
63
|
+
if (!id) throw new Error("generic provider requires id");
|
|
64
|
+
const resolvedBase = resolveBaseUrl(id, baseUrl);
|
|
65
|
+
if (!resolvedBase) throw new Error(`generic provider ${id}: missing baseUrl`);
|
|
66
|
+
if (!fetchImpl) fetchImpl = UndiciFetch || compatFetch;
|
|
67
|
+
const resolvedModelsPath = modelsPath || loadProviderModelsPath(id, file ? { file } : {});
|
|
68
|
+
const resolvedChatPath = chatPath || loadProviderChatPath(id, file ? { file } : {});
|
|
69
|
+
const ring = createKeyRing(collectApiKeysGeneric(id, apiKeys, apiKey, loadProviderKeys), { cooldownMs });
|
|
70
|
+
const goHost = isOpencodeHost(resolvedBase);
|
|
71
|
+
|
|
72
|
+
let dispatcher = null;
|
|
73
|
+
let agent = null;
|
|
74
|
+
if (!noAgent) {
|
|
75
|
+
const a = createAgent({
|
|
76
|
+
keepAliveTimeout: envInt("MSLXDFF_GENERIC_KEEPALIVE_TIMEOUT", 30_000),
|
|
77
|
+
keepAliveMaxTimeout: envInt("MSLXDFF_GENERIC_KEEPALIVE_MAX_TIMEOUT", 60_000),
|
|
78
|
+
connections: envInt("MSLXDFF_GENERIC_KEEPALIVE_CONNECTIONS", 20),
|
|
79
|
+
});
|
|
80
|
+
agent = a.agent; dispatcher = a.dispatcher;
|
|
81
|
+
}
|
|
82
|
+
|
|
83
|
+
function buildHeaders(body, key, opts) {
|
|
84
|
+
const isStream = body?.stream !== false;
|
|
85
|
+
const h = {
|
|
86
|
+
"Content-Type": "application/json",
|
|
87
|
+
Accept: isStream ? "text/event-stream" : "*/*",
|
|
88
|
+
"User-Agent": "mslxdff",
|
|
89
|
+
};
|
|
90
|
+
if (key) h["Authorization"] = `Bearer ${key}`;
|
|
91
|
+
const out = { ...h, ...extraHeaders };
|
|
92
|
+
// ocgo(opencode.ai 域名):补 Console Go 路由所需的 opencode 身份头
|
|
93
|
+
if (goHost) {
|
|
94
|
+
out["User-Agent"] = opencodeUa();
|
|
95
|
+
out["x-opencode-client"] = "desktop";
|
|
96
|
+
out["x-opencode-session"] = resolveGoSession(opts?.sessionId);
|
|
97
|
+
out["x-opencode-request"] = genId("msg_");
|
|
98
|
+
out["x-opencode-project"] = "global";
|
|
99
|
+
}
|
|
100
|
+
return out;
|
|
101
|
+
}
|
|
102
|
+
|
|
103
|
+
// base 的 attemptOnce 只调 buildHeaders(body, key):用闭包把 opts.sessionId 带进去
|
|
104
|
+
function scopedRunner(activeRing, opts) {
|
|
105
|
+
return createChatRunner({
|
|
106
|
+
id, ring: activeRing, cooldownMs, retry, fetchImpl, dispatcher,
|
|
107
|
+
buildHeaders: (b, k) => buildHeaders(b, k, opts),
|
|
108
|
+
getUrl: () => joinUrl(resolvedBase, resolvedChatPath),
|
|
109
|
+
connectTimeoutMs,
|
|
110
|
+
});
|
|
111
|
+
}
|
|
112
|
+
|
|
113
|
+
// responses 类模型(muse-spark* 等)只在 /responses 挂载 → 单独通道,出参恒 chat 形状。
|
|
114
|
+
const responses = createResponsesChannel({
|
|
115
|
+
id,
|
|
116
|
+
url: joinUrl(resolvedBase, resolveResponsesPath(id, responsesPath)),
|
|
117
|
+
connectTimeoutMs,
|
|
118
|
+
retry,
|
|
119
|
+
cooldownMs,
|
|
120
|
+
// 供应商级 <ID>_SDK 未设置即继承全局总闸(MSLXDFF_UPSTREAM_ENGINE)
|
|
121
|
+
sdkEnabled: resolveEngineMode(process.env, `MSLXDFF_${envSlug(id)}_SDK`) === "sdk",
|
|
122
|
+
buildHeaders,
|
|
123
|
+
fetchImpl,
|
|
124
|
+
dispatcher,
|
|
125
|
+
});
|
|
126
|
+
|
|
127
|
+
async function chat(body, opts) {
|
|
128
|
+
const sourceKey = `MSLXDFF_${envSlug(id)}_KEY`;
|
|
129
|
+
if (isResponsesModel(body?.model)) return responses.run(body, ring, sourceKey, opts);
|
|
130
|
+
const { runChat } = scopedRunner(ring, opts);
|
|
131
|
+
return runChat(body, ring, sourceKey);
|
|
132
|
+
}
|
|
133
|
+
async function chatWithKeys(body, keys, opts) {
|
|
134
|
+
const tmp = createKeyRing(keys, { cooldownMs });
|
|
135
|
+
if (isResponsesModel(body?.model)) return responses.run(body, tmp, "shared provider keys", opts);
|
|
136
|
+
const { runChat } = scopedRunner(tmp, opts);
|
|
137
|
+
return runChat(body, tmp, "shared provider keys");
|
|
138
|
+
}
|
|
139
|
+
|
|
140
|
+
const { listModels } = createListModelsRunner({
|
|
141
|
+
id, ring, dispatcher, fetchImpl,
|
|
142
|
+
getUrl: () => joinUrl(resolvedBase, resolvedModelsPath),
|
|
143
|
+
mapModel,
|
|
144
|
+
});
|
|
145
|
+
|
|
146
|
+
const { preheat } = createPreheatRunner({
|
|
147
|
+
dispatcher, fetchImpl, getUrl: () => joinUrl(resolvedBase, resolvedModelsPath),
|
|
148
|
+
id, ring, loadKeys: loadProviderKeys,
|
|
149
|
+
});
|
|
150
|
+
|
|
151
|
+
async function close() {
|
|
152
|
+
if (agent && typeof agent.close === "function") { try { await agent.close(); } catch {} }
|
|
153
|
+
}
|
|
154
|
+
|
|
155
|
+
return { id, chat, chatWithKeys, listModels, preheat, close, agent, keyRing: ring, baseUrl: resolvedBase, responsesUrl: responses.url };
|
|
156
|
+
}
|
|
@@ -1,141 +1,141 @@
|
|
|
1
|
-
// globalqwenwork 账号落盘单一源:auths/globalqwenwork-<uid>.json(0600 tmp+rename)+ state.json 双写。
|
|
2
|
-
// vendor 自 src/providers/qwenwork/account-store.js 逐字搬,改三处命名空间:
|
|
3
|
-
// ① 文件名前缀 `globalqwenwork-`(与 cn 的 `qwenwork-` **互不为前缀**,故两站号同目录也不串;
|
|
4
|
-
// 反向亦然:`qwenwork-x.json` 不 startsWith("globalqwenwork-"))
|
|
5
|
-
// ② state 键 `providerConfigs.globalqwenwork`
|
|
6
|
-
// ③ auth 目录 env 独立为 `GLOBALQWENWORK_AUTH_DIR`(不与 cn 的 QWENWORK_AUTH_DIR 共享,防两站串号)
|
|
7
|
-
// 国际站与 cn 站凭证**不互通**(实测 cn token 打 .ai userinfo → 401 INVALID_TOKEN),故必须各存各的。
|
|
8
|
-
import { writeFileSync, mkdirSync, existsSync, renameSync, unlinkSync, readdirSync, readFileSync, chmodSync } from "node:fs";
|
|
9
|
-
import { join, dirname } from "node:path";
|
|
10
|
-
import { tmpdir } from "node:os";
|
|
11
|
-
import { isTestEnv, defaultStateFile } from "../../state/store.js";
|
|
12
|
-
|
|
13
|
-
const PROVIDER_ID = "globalqwenwork";
|
|
14
|
-
const FILE_PREFIX = "globalqwenwork-";
|
|
15
|
-
|
|
16
|
-
export function authDirFor({ explicit = "", testEnv = false, stateFile } = {}) {
|
|
17
|
-
if (explicit) return explicit;
|
|
18
|
-
if (testEnv) return join(tmpdir(), "mslxdff-test-auths-globalqwenwork");
|
|
19
|
-
return join(dirname(stateFile), "auths");
|
|
20
|
-
}
|
|
21
|
-
|
|
22
|
-
export function resolveAuthDir() {
|
|
23
|
-
return authDirFor({
|
|
24
|
-
explicit: process.env.GLOBALQWENWORK_AUTH_DIR || "",
|
|
25
|
-
testEnv: isTestEnv(),
|
|
26
|
-
stateFile: defaultStateFile(),
|
|
27
|
-
});
|
|
28
|
-
}
|
|
29
|
-
|
|
30
|
-
export function resolveAuthDirs() {
|
|
31
|
-
const primary = resolveAuthDir();
|
|
32
|
-
const cwdDir = join(process.cwd(), "auths");
|
|
33
|
-
return primary === cwdDir ? [primary] : [primary, cwdDir];
|
|
34
|
-
}
|
|
35
|
-
|
|
36
|
-
export function listAccountDocs({ dirs } = {}) {
|
|
37
|
-
const out = [];
|
|
38
|
-
const seen = new Set();
|
|
39
|
-
for (const dir of dirs || resolveAuthDirs()) {
|
|
40
|
-
let files = [];
|
|
41
|
-
try {
|
|
42
|
-
if (!existsSync(dir)) continue;
|
|
43
|
-
files = readdirSync(dir).filter((f) => f.startsWith(FILE_PREFIX) && f.endsWith(".json"));
|
|
44
|
-
} catch { continue; }
|
|
45
|
-
for (const f of files) {
|
|
46
|
-
try {
|
|
47
|
-
const doc = JSON.parse(readFileSync(join(dir, f), "utf8"));
|
|
48
|
-
const uid = doc?.account?.uid;
|
|
49
|
-
if (!uid || !doc?.auth?.accessToken) continue;
|
|
50
|
-
if (seen.has(String(uid))) continue;
|
|
51
|
-
seen.add(String(uid));
|
|
52
|
-
out.push({ uid: String(uid), file: join(dir, f), dir, doc });
|
|
53
|
-
} catch {}
|
|
54
|
-
}
|
|
55
|
-
}
|
|
56
|
-
return out;
|
|
57
|
-
}
|
|
58
|
-
|
|
59
|
-
function writeAccountFile({ uid, accessToken, refreshToken = "", expiresAt = 0, name = "", email = "" }) {
|
|
60
|
-
const authDir = resolveAuthDir();
|
|
61
|
-
mkdirSync(authDir, { recursive: true });
|
|
62
|
-
const doc = {
|
|
63
|
-
account: { uid, name, email },
|
|
64
|
-
auth: { accessToken, refreshToken, expiresAt },
|
|
65
|
-
};
|
|
66
|
-
const fp = join(authDir, `${FILE_PREFIX}${uid}.json`);
|
|
67
|
-
const tmp = `${fp}.tmp`;
|
|
68
|
-
writeFileSync(tmp, JSON.stringify(doc, null, 2), { mode: 0o600 });
|
|
69
|
-
try {
|
|
70
|
-
if (existsSync(fp)) { try { chmodSync(fp, 0o600); } catch {} unlinkSync(fp); }
|
|
71
|
-
renameSync(tmp, fp);
|
|
72
|
-
} catch {
|
|
73
|
-
writeFileSync(fp, JSON.stringify(doc, null, 2), { mode: 0o600 });
|
|
74
|
-
try { unlinkSync(tmp); } catch {}
|
|
75
|
-
}
|
|
76
|
-
try { chmodSync(fp, 0o600); } catch {}
|
|
77
|
-
return fp;
|
|
78
|
-
}
|
|
79
|
-
|
|
80
|
-
// state 落盘:keys[i] 存 JSON blob({device_token, refresh_token},对齐家族形状),auths 存 uid 索引行。
|
|
81
|
-
async function persistAccounts(accounts, file) {
|
|
82
|
-
const { loadProviderConfigs } = await import("../../state.js");
|
|
83
|
-
const { saveProviderConfig } = await import("../../state.js");
|
|
84
|
-
const cfg = (loadProviderConfigs(file ? { file } : {})?.[PROVIDER_ID]) || {};
|
|
85
|
-
const keys = [...(Array.isArray(cfg.keys) ? cfg.keys : [])];
|
|
86
|
-
const auths = [...(Array.isArray(cfg.auths) ? cfg.auths : [])];
|
|
87
|
-
let updated = 0;
|
|
88
|
-
for (const acc of accounts) {
|
|
89
|
-
if (!acc?.uid || !acc?.accessToken) continue;
|
|
90
|
-
const blob = JSON.stringify({ device_token: acc.accessToken, refresh_token: acc.refreshToken || "" });
|
|
91
|
-
const row = { uid: acc.uid, name: acc.name || "", refreshToken: acc.refreshToken || "" };
|
|
92
|
-
const idx = auths.findIndex((a) => String(a?.uid) === String(acc.uid));
|
|
93
|
-
if (idx >= 0) { keys[idx] = blob; auths[idx] = row; updated += 1; }
|
|
94
|
-
else { keys.push(blob); auths.push(row); }
|
|
95
|
-
}
|
|
96
|
-
// saveProviderConfig 起手保留 allowAnyModels(已在 state 层修复,见 ADR-0033 字段保留修复)
|
|
97
|
-
saveProviderConfig(PROVIDER_ID, { baseUrl: cfg.baseUrl || "", keys, auths }, file ? { file } : {});
|
|
98
|
-
return { keys, auths, updated };
|
|
99
|
-
}
|
|
100
|
-
|
|
101
|
-
export async function saveGlobalQwenworkAccount({ uid, accessToken, refreshToken = "", expiresAt = 0, name = "", email = "", file } = {}) {
|
|
102
|
-
if (!uid || !accessToken) throw new Error("saveGlobalQwenworkAccount: 缺少 uid/accessToken");
|
|
103
|
-
const fp = writeAccountFile({ uid, accessToken, refreshToken, expiresAt, name, email });
|
|
104
|
-
const st = await persistAccounts([{ uid, accessToken, refreshToken, name }], file);
|
|
105
|
-
return { file: fp, accounts: st.keys.length, updated: st.updated > 0 };
|
|
106
|
-
}
|
|
107
|
-
|
|
108
|
-
/** 回写刷新后的 token(上游每次 refresh 会轮换 refreshToken,必须落盘)。 */
|
|
109
|
-
export async function updateGlobalQwenworkTokens({ uid, accessToken, refreshToken = "", expiresAt = 0, file } = {}) {
|
|
110
|
-
if (!uid || !accessToken) throw new Error("updateGlobalQwenworkTokens: 缺少 uid/accessToken");
|
|
111
|
-
const docs = listAccountDocs();
|
|
112
|
-
const hit = docs.find((d) => String(d.uid) === String(uid));
|
|
113
|
-
const prev = hit?.doc || {};
|
|
114
|
-
const fp = writeAccountFile({
|
|
115
|
-
uid,
|
|
116
|
-
accessToken,
|
|
117
|
-
refreshToken: refreshToken || prev?.auth?.refreshToken || "",
|
|
118
|
-
expiresAt: expiresAt || 0,
|
|
119
|
-
name: prev?.account?.name || "",
|
|
120
|
-
email: prev?.account?.email || "",
|
|
121
|
-
});
|
|
122
|
-
await persistAccounts([{ uid, accessToken, refreshToken: refreshToken || prev?.auth?.refreshToken || "", name: prev?.account?.name || "" }], file);
|
|
123
|
-
try {
|
|
124
|
-
const { invalidateSession } = await import("./cosy.js");
|
|
125
|
-
invalidateSession({ uid, accessToken: prev?.auth?.accessToken });
|
|
126
|
-
} catch {}
|
|
127
|
-
return { file: fp };
|
|
128
|
-
}
|
|
129
|
-
|
|
130
|
-
// 从 provider keys 的 blob 还原凭据(device_token/deviceToken 双形状兼容)
|
|
131
|
-
export function accountFromBlob(blob) {
|
|
132
|
-
try {
|
|
133
|
-
const j = JSON.parse(String(blob || "{}"));
|
|
134
|
-
const accessToken = String(j.device_token || j.accessToken || "");
|
|
135
|
-
const refreshToken = String(j.refresh_token || j.refreshToken || "");
|
|
136
|
-
if (!accessToken) return null;
|
|
137
|
-
return { accessToken, refreshToken };
|
|
138
|
-
} catch {
|
|
139
|
-
return null;
|
|
140
|
-
}
|
|
141
|
-
}
|
|
1
|
+
// globalqwenwork 账号落盘单一源:auths/globalqwenwork-<uid>.json(0600 tmp+rename)+ state.json 双写。
|
|
2
|
+
// vendor 自 src/providers/qwenwork/account-store.js 逐字搬,改三处命名空间:
|
|
3
|
+
// ① 文件名前缀 `globalqwenwork-`(与 cn 的 `qwenwork-` **互不为前缀**,故两站号同目录也不串;
|
|
4
|
+
// 反向亦然:`qwenwork-x.json` 不 startsWith("globalqwenwork-"))
|
|
5
|
+
// ② state 键 `providerConfigs.globalqwenwork`
|
|
6
|
+
// ③ auth 目录 env 独立为 `GLOBALQWENWORK_AUTH_DIR`(不与 cn 的 QWENWORK_AUTH_DIR 共享,防两站串号)
|
|
7
|
+
// 国际站与 cn 站凭证**不互通**(实测 cn token 打 .ai userinfo → 401 INVALID_TOKEN),故必须各存各的。
|
|
8
|
+
import { writeFileSync, mkdirSync, existsSync, renameSync, unlinkSync, readdirSync, readFileSync, chmodSync } from "node:fs";
|
|
9
|
+
import { join, dirname } from "node:path";
|
|
10
|
+
import { tmpdir } from "node:os";
|
|
11
|
+
import { isTestEnv, defaultStateFile } from "../../state/store.js";
|
|
12
|
+
|
|
13
|
+
const PROVIDER_ID = "globalqwenwork";
|
|
14
|
+
const FILE_PREFIX = "globalqwenwork-";
|
|
15
|
+
|
|
16
|
+
export function authDirFor({ explicit = "", testEnv = false, stateFile } = {}) {
|
|
17
|
+
if (explicit) return explicit;
|
|
18
|
+
if (testEnv) return join(tmpdir(), "mslxdff-test-auths-globalqwenwork");
|
|
19
|
+
return join(dirname(stateFile), "auths");
|
|
20
|
+
}
|
|
21
|
+
|
|
22
|
+
export function resolveAuthDir() {
|
|
23
|
+
return authDirFor({
|
|
24
|
+
explicit: process.env.GLOBALQWENWORK_AUTH_DIR || "",
|
|
25
|
+
testEnv: isTestEnv(),
|
|
26
|
+
stateFile: defaultStateFile(),
|
|
27
|
+
});
|
|
28
|
+
}
|
|
29
|
+
|
|
30
|
+
export function resolveAuthDirs() {
|
|
31
|
+
const primary = resolveAuthDir();
|
|
32
|
+
const cwdDir = join(process.cwd(), "auths");
|
|
33
|
+
return primary === cwdDir ? [primary] : [primary, cwdDir];
|
|
34
|
+
}
|
|
35
|
+
|
|
36
|
+
export function listAccountDocs({ dirs } = {}) {
|
|
37
|
+
const out = [];
|
|
38
|
+
const seen = new Set();
|
|
39
|
+
for (const dir of dirs || resolveAuthDirs()) {
|
|
40
|
+
let files = [];
|
|
41
|
+
try {
|
|
42
|
+
if (!existsSync(dir)) continue;
|
|
43
|
+
files = readdirSync(dir).filter((f) => f.startsWith(FILE_PREFIX) && f.endsWith(".json"));
|
|
44
|
+
} catch { continue; }
|
|
45
|
+
for (const f of files) {
|
|
46
|
+
try {
|
|
47
|
+
const doc = JSON.parse(readFileSync(join(dir, f), "utf8"));
|
|
48
|
+
const uid = doc?.account?.uid;
|
|
49
|
+
if (!uid || !doc?.auth?.accessToken) continue;
|
|
50
|
+
if (seen.has(String(uid))) continue;
|
|
51
|
+
seen.add(String(uid));
|
|
52
|
+
out.push({ uid: String(uid), file: join(dir, f), dir, doc });
|
|
53
|
+
} catch {}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
return out;
|
|
57
|
+
}
|
|
58
|
+
|
|
59
|
+
function writeAccountFile({ uid, accessToken, refreshToken = "", expiresAt = 0, name = "", email = "" }) {
|
|
60
|
+
const authDir = resolveAuthDir();
|
|
61
|
+
mkdirSync(authDir, { recursive: true });
|
|
62
|
+
const doc = {
|
|
63
|
+
account: { uid, name, email },
|
|
64
|
+
auth: { accessToken, refreshToken, expiresAt },
|
|
65
|
+
};
|
|
66
|
+
const fp = join(authDir, `${FILE_PREFIX}${uid}.json`);
|
|
67
|
+
const tmp = `${fp}.tmp`;
|
|
68
|
+
writeFileSync(tmp, JSON.stringify(doc, null, 2), { mode: 0o600 });
|
|
69
|
+
try {
|
|
70
|
+
if (existsSync(fp)) { try { chmodSync(fp, 0o600); } catch {} unlinkSync(fp); }
|
|
71
|
+
renameSync(tmp, fp);
|
|
72
|
+
} catch {
|
|
73
|
+
writeFileSync(fp, JSON.stringify(doc, null, 2), { mode: 0o600 });
|
|
74
|
+
try { unlinkSync(tmp); } catch {}
|
|
75
|
+
}
|
|
76
|
+
try { chmodSync(fp, 0o600); } catch {}
|
|
77
|
+
return fp;
|
|
78
|
+
}
|
|
79
|
+
|
|
80
|
+
// state 落盘:keys[i] 存 JSON blob({device_token, refresh_token},对齐家族形状),auths 存 uid 索引行。
|
|
81
|
+
async function persistAccounts(accounts, file) {
|
|
82
|
+
const { loadProviderConfigs } = await import("../../state.js");
|
|
83
|
+
const { saveProviderConfig } = await import("../../state.js");
|
|
84
|
+
const cfg = (loadProviderConfigs(file ? { file } : {})?.[PROVIDER_ID]) || {};
|
|
85
|
+
const keys = [...(Array.isArray(cfg.keys) ? cfg.keys : [])];
|
|
86
|
+
const auths = [...(Array.isArray(cfg.auths) ? cfg.auths : [])];
|
|
87
|
+
let updated = 0;
|
|
88
|
+
for (const acc of accounts) {
|
|
89
|
+
if (!acc?.uid || !acc?.accessToken) continue;
|
|
90
|
+
const blob = JSON.stringify({ device_token: acc.accessToken, refresh_token: acc.refreshToken || "" });
|
|
91
|
+
const row = { uid: acc.uid, name: acc.name || "", refreshToken: acc.refreshToken || "" };
|
|
92
|
+
const idx = auths.findIndex((a) => String(a?.uid) === String(acc.uid));
|
|
93
|
+
if (idx >= 0) { keys[idx] = blob; auths[idx] = row; updated += 1; }
|
|
94
|
+
else { keys.push(blob); auths.push(row); }
|
|
95
|
+
}
|
|
96
|
+
// saveProviderConfig 起手保留 allowAnyModels(已在 state 层修复,见 ADR-0033 字段保留修复)
|
|
97
|
+
saveProviderConfig(PROVIDER_ID, { baseUrl: cfg.baseUrl || "", keys, auths }, file ? { file } : {});
|
|
98
|
+
return { keys, auths, updated };
|
|
99
|
+
}
|
|
100
|
+
|
|
101
|
+
export async function saveGlobalQwenworkAccount({ uid, accessToken, refreshToken = "", expiresAt = 0, name = "", email = "", file } = {}) {
|
|
102
|
+
if (!uid || !accessToken) throw new Error("saveGlobalQwenworkAccount: 缺少 uid/accessToken");
|
|
103
|
+
const fp = writeAccountFile({ uid, accessToken, refreshToken, expiresAt, name, email });
|
|
104
|
+
const st = await persistAccounts([{ uid, accessToken, refreshToken, name }], file);
|
|
105
|
+
return { file: fp, accounts: st.keys.length, updated: st.updated > 0 };
|
|
106
|
+
}
|
|
107
|
+
|
|
108
|
+
/** 回写刷新后的 token(上游每次 refresh 会轮换 refreshToken,必须落盘)。 */
|
|
109
|
+
export async function updateGlobalQwenworkTokens({ uid, accessToken, refreshToken = "", expiresAt = 0, file } = {}) {
|
|
110
|
+
if (!uid || !accessToken) throw new Error("updateGlobalQwenworkTokens: 缺少 uid/accessToken");
|
|
111
|
+
const docs = listAccountDocs();
|
|
112
|
+
const hit = docs.find((d) => String(d.uid) === String(uid));
|
|
113
|
+
const prev = hit?.doc || {};
|
|
114
|
+
const fp = writeAccountFile({
|
|
115
|
+
uid,
|
|
116
|
+
accessToken,
|
|
117
|
+
refreshToken: refreshToken || prev?.auth?.refreshToken || "",
|
|
118
|
+
expiresAt: expiresAt || 0,
|
|
119
|
+
name: prev?.account?.name || "",
|
|
120
|
+
email: prev?.account?.email || "",
|
|
121
|
+
});
|
|
122
|
+
await persistAccounts([{ uid, accessToken, refreshToken: refreshToken || prev?.auth?.refreshToken || "", name: prev?.account?.name || "" }], file);
|
|
123
|
+
try {
|
|
124
|
+
const { invalidateSession } = await import("./cosy.js");
|
|
125
|
+
invalidateSession({ uid, accessToken: prev?.auth?.accessToken });
|
|
126
|
+
} catch {}
|
|
127
|
+
return { file: fp };
|
|
128
|
+
}
|
|
129
|
+
|
|
130
|
+
// 从 provider keys 的 blob 还原凭据(device_token/deviceToken 双形状兼容)
|
|
131
|
+
export function accountFromBlob(blob) {
|
|
132
|
+
try {
|
|
133
|
+
const j = JSON.parse(String(blob || "{}"));
|
|
134
|
+
const accessToken = String(j.device_token || j.accessToken || "");
|
|
135
|
+
const refreshToken = String(j.refresh_token || j.refreshToken || "");
|
|
136
|
+
if (!accessToken) return null;
|
|
137
|
+
return { accessToken, refreshToken };
|
|
138
|
+
} catch {
|
|
139
|
+
return null;
|
|
140
|
+
}
|
|
141
|
+
}
|