dsh-model 0.0.1 → 0.15.0

package/dist/daemon/service.js

@@ -0,0 +1,33 @@
+// daemon/service.ts — 确保守护进程（bridge 服务）在跑。setup 总会装它：来源开关、登录、统计都靠它
+import { BRIDGE_DEFAULT_PORT, ensureBridgeConfig, loadBridgeConfig } from '../bridge/runtime.js';
+import { DshModelError } from '../errors.js';
+import { L } from '../i18n.js';
+import { bridgeHealthy } from '../integrations/workbuddy.js';
+import { serviceFor } from '../service/index.js';
+import { ok, skip } from '../util/output.js';
+import { findFreePort, isPortFree } from '../util/port.js';
+export async function ensureDaemon(ctx, all, opts = {}) {
+    let port = all.config.bridge?.port || BRIDGE_DEFAULT_PORT;
+    const existing = await loadBridgeConfig(ctx.paths.home);
+    const ours = existing && (await bridgeHealthy(existing.port, existing.secret));
+    if (!ours && !(await isPortFree(port)))
+        port = await findFreePort(port + 1);
+    all.config.bridge = { ...all.config.bridge, port };
+    const cfg = await ensureBridgeConfig(ctx.paths.home, port, ctx.owner);
+    if (ctx.serviceDisabled) {
+        skip(L('已跳过守护进程服务（DSH_MODEL_SERVICE=none）', 'Skipped daemon service (DSH_MODEL_SERVICE=none)'));
+        return;
+    }
+    const svc = serviceFor(ctx, 'bridge', all.config.proxy);
+    await svc.install();
+    if (opts.restart !== false)
+        await svc.restart();
+    all.state.bridgeService = { kind: svc.kind, file: svc.spec.file, label: svc.spec.label };
+    const deadline = Date.now() + 15_000;
+    while (Date.now() < deadline && !(await bridgeHealthy(cfg.port, cfg.secret)))
+        await new Promise((r) => setTimeout(r, 300));
+    if (!(await bridgeHealthy(cfg.port, cfg.secret))) {
+        throw new DshModelError('daemon_unhealthy', L(`守护进程没有在 127.0.0.1:${cfg.port} 上启动`, `Daemon did not come up on 127.0.0.1:${cfg.port}`), L('查看日志：dsh-model logs --bridge', 'Check logs: dsh-model logs --bridge'));
+    }
+    ok(L(`守护进程已在 127.0.0.1:${cfg.port} 运行（来源开关、登录、统计）`, `Daemon running on 127.0.0.1:${cfg.port} (sources, logins, stats)`));
+}

package/dist/daemon/stats.js

@@ -0,0 +1,126 @@
+// daemon/stats.ts — 用量统计（设计 §14.3）
+//
+// 守护进程每 5 秒从引擎管理接口取走用量记录（引擎只保留 10 分钟），在这里累计。
+// 只保留最近 24 小时的精简记录（最多 50k 条），按 key / 来源 / 模型给出 1 小时与 24 小时两个窗口：
+// 请求数、失败数、成功率、平均延迟、输出速度（output_tokens / 延迟）、最后使用与最后失败时间。
+// 记录里的客户端 key 只存 key 名（由 keys.json 映射），不存 key 本身。
+import { join } from 'node:path';
+import { atomicWrite, readJson } from '../util/fs.js';
+const DAY_MS = 24 * 3600 * 1000;
+const MAX_RECORDS = 50_000;
+function summarize(recs) {
+    if (!recs.length)
+        return { requests: 0, failures: 0, successRate: null, avgLatencyMs: null, tokensPerSec: null, lastUsedAt: null, lastFailureAt: null };
+    let failures = 0;
+    let latSum = 0;
+    let okLat = 0;
+    let outSum = 0;
+    let last = 0;
+    let lastFail = 0;
+    for (const r of recs) {
+        latSum += r.l;
+        if (r.t > last)
+            last = r.t;
+        if (r.f) {
+            failures++;
+            if (r.t > lastFail)
+                lastFail = r.t;
+        }
+        else if (r.o > 0) {
+            // 没报 token 数的请求不参与速度计算，免得把平均速度拉低
+            okLat += r.l;
+            outSum += r.o;
+        }
+    }
+    return {
+        requests: recs.length,
+        failures,
+        successRate: (recs.length - failures) / recs.length,
+        avgLatencyMs: Math.round(latSum / recs.length),
+        tokensPerSec: okLat > 0 && outSum > 0 ? Math.round((outSum / (okLat / 1000)) * 10) / 10 : null,
+        lastUsedAt: last ? new Date(last).toISOString() : null,
+        lastFailureAt: lastFail ? new Date(lastFail).toISOString() : null,
+    };
+}
+/** 模型名前缀 → 来源：workbuddy/x → workbuddy；无前缀按引擎给的 provider */
+export function sourceOf(rec) {
+    const alias = rec.alias || rec.model || '';
+    if (alias.includes('/'))
+        return alias.split('/')[0];
+    return rec.provider || 'unknown';
+}
+export class Stats {
+    file;
+    owner;
+    recs = [];
+    dirty = false;
+    constructor(file, owner) {
+        this.file = file;
+        this.owner = owner;
+    }
+    static path(home) {
+        return join(home, 'stats.json');
+    }
+    async load() {
+        const saved = await readJson(this.file).catch(() => null);
+        this.recs = (saved?.recs ?? []).filter((r) => Date.now() - r.t < DAY_MS);
+    }
+    /** 收进一批用量记录；keyName 把客户端 key 映射成名字（未知的记为 unknown） */
+    add(batch, keyName) {
+        for (const r of batch) {
+            const t = Date.parse(r.timestamp) || Date.now();
+            this.recs.push({
+                t,
+                k: keyName(r.api_key),
+                p: sourceOf(r),
+                m: r.alias || r.model || 'unknown',
+                l: Math.max(0, Number(r.latency_ms) || 0),
+                // 推理模型的 token 大多算在 reasoning_tokens 里（引擎分开记）：速度要把两者都算上
+                o: Math.max(0, (Number(r.tokens?.output_tokens) || 0) + (Number(r.tokens?.reasoning_tokens) || 0)),
+                f: r.failed ? 1 : 0,
+            });
+        }
+        if (batch.length)
+            this.dirty = true;
+        this.prune();
+    }
+    prune() {
+        const cutoff = Date.now() - DAY_MS;
+        if (this.recs.length && this.recs[0].t < cutoff)
+            this.recs = this.recs.filter((r) => r.t >= cutoff);
+        if (this.recs.length > MAX_RECORDS)
+            this.recs = this.recs.slice(-MAX_RECORDS);
+    }
+    async save() {
+        if (!this.dirty)
+            return;
+        this.dirty = false;
+        await atomicWrite(this.file, JSON.stringify({ recs: this.recs }), { owner: this.owner });
+    }
+    snapshot(now = Date.now()) {
+        this.prune();
+        const h1 = now - 3600 * 1000;
+        const group = (key) => {
+            const m = new Map();
+            for (const r of this.recs) {
+                const k = key(r);
+                let a = m.get(k);
+                if (!a)
+                    m.set(k, (a = []));
+                a.push(r);
+            }
+            return m;
+        };
+        const win = (recs) => ({ h1: summarize(recs.filter((r) => r.t >= h1)), d1: summarize(recs) });
+        const byKey = {};
+        for (const [k, rs] of group((r) => r.k))
+            byKey[k] = win(rs);
+        const bySource = {};
+        for (const [k, rs] of group((r) => r.p))
+            bySource[k] = win(rs);
+        const byModel = {};
+        for (const [k, rs] of group((r) => r.m))
+            byModel[k] = { source: rs[0].p, ...win(rs) };
+        return { updatedAt: new Date(now).toISOString(), byKey, bySource, byModel };
+    }
+}

package/dist/daemon/usage.js

@@ -0,0 +1,213 @@
+// daemon/usage.ts — 订阅用量（设计 §14.6）
+//
+// 用各来源已有的登录令牌，主动去调它们自己的额度接口。思路与接口来自 steipete 的 CodexBar（MIT，docs/<provider>.md）：
+//   Codex   GET chatgpt.com/backend-api/wham/usage            rate_limit.primary_window / secondary_window
+//   Claude  GET api.anthropic.com/api/oauth/usage             five_hour / seven_day / seven_day_opus（anthropic-beta: oauth-2025-04-20）
+//   Kimi    GET api.kimi.com/coding/v1/usages                 usage（总额度）+ limits[].window（如 5 小时）
+//   Grok    GET cli-chat-proxy.grok.com/v1/billing?format=credits（返回格式未公开，宽松解析）
+//   WorkBuddy 移植来的 fetchCredits（剩余积分）
+// 令牌来源：引擎 auth-dir 里的凭据文件（引擎会自动续期并写回），WorkBuddy 用 bridge 的凭据库。
+// 一律只读：不刷新、不改写任何凭据文件。查询失败只在那一行显示"用量暂不可用"。
+import { readFile } from 'node:fs/promises';
+import { join } from 'node:path';
+const TIMEOUT_MS = 15_000;
+/** 这几家暂不支持（见设计 §14.6） */
+export const USAGE_UNSUPPORTED = new Set(['antigravity', 'devin', 'meta', 'opencode']);
+async function getJson(url, headers) {
+    const res = await fetch(url, { headers: { Accept: 'application/json', ...headers }, signal: AbortSignal.timeout(TIMEOUT_MS) });
+    const text = await res.text();
+    if (!res.ok)
+        throw new Error(`HTTP ${res.status}${text ? `: ${text.slice(0, 120)}` : ''}`);
+    return text ? JSON.parse(text) : {};
+}
+const num = (v) => {
+    const n = typeof v === 'string' ? Number(v) : v;
+    return typeof n === 'number' && Number.isFinite(n) ? n : undefined;
+};
+const str = (v) => (typeof v === 'string' && v ? v : undefined);
+const obj = (v) => (v && typeof v === 'object' && !Array.isArray(v) ? v : undefined);
+/** 秒 / 毫秒时间戳或 ISO 字符串 → ISO */
+function toIso(v) {
+    if (typeof v === 'string' && v) {
+        const t = Date.parse(v);
+        return Number.isNaN(t) ? undefined : new Date(t).toISOString();
+    }
+    const n = num(v);
+    if (n === undefined)
+        return undefined;
+    return new Date(n < 1e12 ? n * 1000 : n).toISOString();
+}
+/** 读引擎凭据文件里的访问令牌（字段名各家不同，宽松取） */
+async function readEngineToken(ctx, cred) {
+    const doc = JSON.parse(await readFile(join(ctx.paths.auth, cred.name), 'utf8'));
+    const nested = obj(doc.token) ?? obj(doc.tokens) ?? {};
+    const token = str(doc.access_token) ?? str(doc.accessToken) ?? str(nested.access_token) ?? str(nested.accessToken);
+    if (!token)
+        throw new Error('no access token in credential file');
+    return { token, doc };
+}
+// —— 各家解析（导出给测试）——
+export function parseCodexUsage(body) {
+    const b = obj(body) ?? {};
+    const rl = obj(b.rate_limit) ?? obj(b.rateLimit) ?? {};
+    const windows = [];
+    const win = (w, id, fallbackLabel) => {
+        const o = obj(w);
+        if (!o)
+            return;
+        const secs = num(o.limit_window_seconds) ?? num(o.window_seconds);
+        const label = secs ? humanWindow(secs) : fallbackLabel;
+        const resetAt = toIso(o.reset_at) ?? (num(o.reset_after_seconds) !== undefined ? new Date(Date.now() + num(o.reset_after_seconds) * 1000).toISOString() : undefined);
+        windows.push({ id, label, ...(num(o.used_percent) !== undefined ? { usedPercent: num(o.used_percent) } : {}), ...(resetAt ? { resetAt } : {}) });
+    };
+    win(rl.primary_window, 'primary', '5h');
+    win(rl.secondary_window, 'secondary', '7d');
+    const c = obj(b.credits);
+    const balance = num(c?.balance);
+    return {
+        ...(str(b.plan_type) ? { plan: str(b.plan_type) } : {}),
+        windows,
+        ...(c && (balance !== undefined || c.unlimited === true) ? { credits: { remaining: balance ?? 0, ...(c.unlimited === true ? { unlimited: true } : {}), label: 'credits' } } : {}),
+    };
+}
+export function parseClaudeUsage(body) {
+    const b = obj(body) ?? {};
+    const windows = [];
+    const add = (key, id, label) => {
+        const o = obj(b[key]);
+        if (!o)
+            return;
+        const used = num(o.utilization);
+        if (used === undefined && !o.resets_at)
+            return;
+        const resetAt = toIso(o.resets_at);
+        windows.push({ id, label, ...(used !== undefined ? { usedPercent: used } : {}), ...(resetAt ? { resetAt } : {}) });
+    };
+    add('five_hour', '5h', '5h');
+    add('seven_day', '7d', '7d');
+    add('seven_day_opus', 'opus-7d', 'Opus 7d');
+    add('seven_day_sonnet', 'sonnet-7d', 'Sonnet 7d');
+    return { windows };
+}
+export function parseKimiUsage(body) {
+    const b = obj(body) ?? {};
+    const windows = [];
+    const fromDetail = (d, id, label) => {
+        const o = obj(d);
+        if (!o)
+            return;
+        const limit = num(o.limit);
+        const used = num(o.used) ?? (limit !== undefined && num(o.remaining) !== undefined ? limit - num(o.remaining) : undefined);
+        const resetAt = toIso(o.resetTime);
+        windows.push({
+            id,
+            label,
+            ...(used !== undefined ? { used } : {}),
+            ...(limit !== undefined ? { limit } : {}),
+            ...(used !== undefined && limit ? { usedPercent: Math.round((used / limit) * 1000) / 10 } : {}),
+            ...(resetAt ? { resetAt } : {}),
+        });
+    };
+    for (const l of (Array.isArray(b.limits) ? b.limits : [])) {
+        const o = obj(l) ?? {};
+        const w = obj(o.window) ?? {};
+        const dur = num(w.duration) ?? 0;
+        const unit = String(w.timeUnit ?? '');
+        const secs = unit.includes('MINUTE') ? dur * 60 : unit.includes('HOUR') ? dur * 3600 : unit.includes('DAY') ? dur * 86400 : dur;
+        const label = secs ? humanWindow(secs) : 'limit';
+        fromDetail(o.detail, `w${secs}`, label);
+    }
+    fromDetail(b.usage, 'total', 'quota');
+    return { windows };
+}
+export function parseGrokBilling(body) {
+    const b = obj(body) ?? {};
+    const c = obj(b.credits) ?? b;
+    const remaining = num(c.remaining) ?? num(c.balance) ?? num(c.remainingCredits) ?? num(c.available);
+    const total = num(c.total) ?? num(c.limit) ?? num(c.totalCredits);
+    return { windows: [], ...(remaining !== undefined ? { credits: { remaining, ...(total !== undefined ? { total } : {}), label: 'credits' } } : {}) };
+}
+function humanWindow(secs) {
+    if (secs % 86400 === 0)
+        return `${secs / 86400}d`;
+    if (secs % 3600 === 0)
+        return `${secs / 3600}h`;
+    return `${Math.round(secs / 60)}m`;
+}
+// —— 取用量 ——
+export async function fetchEngineUsage(ctx, def, cred) {
+    const { token, doc } = await readEngineToken(ctx, cred);
+    if (def.id === 'codex') {
+        const account = str(doc.account_id) ?? str(doc.accountId) ?? str(obj(doc.tokens)?.account_id);
+        const body = await getJson('https://chatgpt.com/backend-api/wham/usage', {
+            Authorization: `Bearer ${token}`,
+            ...(account ? { 'ChatGPT-Account-Id': account } : {}),
+            'User-Agent': 'codex_cli_rs',
+        });
+        return parseCodexUsage(body);
+    }
+    if (def.id === 'claude') {
+        const body = await getJson('https://api.anthropic.com/api/oauth/usage', { Authorization: `Bearer ${token}`, 'anthropic-beta': 'oauth-2025-04-20' });
+        return parseClaudeUsage(body);
+    }
+    if (def.id === 'kimi') {
+        const r = parseKimiUsage(await getJson('https://api.kimi.com/coding/v1/usages', { Authorization: `Bearer ${token}` }));
+        // 没有 Kimi Code 订阅时接口返回 {}（实测；同一账号调模型得到 403 access_terminated）
+        return r.windows.length ? r : { windows: [], noAccess: true, plan: 'none' };
+    }
+    if (def.id === 'xai') {
+        return parseGrokBilling(await getJson('https://cli-chat-proxy.grok.com/v1/billing?format=credits', { Authorization: `Bearer ${token}` }));
+    }
+    return { windows: [], unsupported: true };
+}
+export async function fetchWorkbuddyUsage(rt) {
+    const credential = await rt.wbStore.resolve();
+    const credits = await rt.wbClient.fetchCredits(credential);
+    const unlimited = credits.accounts.some((a) => a.unlimited);
+    const size = credits.accounts.reduce((s, a) => s + (Number.isFinite(a.size) ? a.size : 0), 0);
+    return {
+        windows: [],
+        credits: { remaining: credits.total, ...(size > 0 ? { total: size } : {}), ...(unlimited ? { unlimited: true } : {}), label: 'credits' },
+        ...(credits.accounts.length ? { plan: packageSummary(credits.accounts.map((a) => a.packageName)) } : {}),
+    };
+}
+/** 套餐名去重计数："个人版 + 加量包 ×38"（账号下常有几十个同名加量包，实测） */
+export function packageSummary(names) {
+    const counts = new Map();
+    for (const n of names)
+        if (n)
+            counts.set(n, (counts.get(n) ?? 0) + 1);
+    return [...counts].map(([n, c]) => (c > 1 ? `${n} ×${c}` : n)).join(' + ');
+}
+/** 实测一次能不能调：发一条极短的请求（经引擎，用 dsh 的 key）。
+ *  被拒（402/403、insufficient_quota、payment_required、access_terminated）→ noAccess；
+ *  其他失败（网络、限流、上游 5xx）不下结论，抛错由调用方保留上次结果。 */
+export async function probeEngineSource(port, key, model) {
+    const res = await fetch(`http://127.0.0.1:${port}/v1/chat/completions`, {
+        method: 'POST',
+        headers: { Authorization: `Bearer ${key}`, 'Content-Type': 'application/json' },
+        body: JSON.stringify({ model, messages: [{ role: 'user', content: 'ok' }], max_tokens: 1, stream: false }),
+        signal: AbortSignal.timeout(60_000),
+    });
+    const text = await res.text();
+    if (res.ok)
+        return { ok: true };
+    return classifyProbeFailure(res.status, text);
+}
+export function classifyProbeFailure(status, text) {
+    const denied = status === 402 || status === 403 || /insufficient_quota|payment_required|permission_denied|access_terminated|subscription/i.test(text);
+    let msg = text;
+    try {
+        msg = String(JSON.parse(text).error?.message ?? text);
+    }
+    catch {
+        // 不是 JSON
+    }
+    if (denied)
+        return { ok: false, reason: msg.slice(0, 160) };
+    throw new Error(`HTTP ${status}: ${msg.slice(0, 160)}`);
+}
+/** 守护进程把用量写到这里，同步模型时（任何进程）据此隐藏没有订阅的来源 */
+export function usageFilePath(home) {
+    return join(home, 'usage.json');
+}

package/dist/dsh/cli.js

@@ -0,0 +1,90 @@
+// dsh/cli.ts — 找到并调用 dsh 自己的命令行（用它的 `dsh plugin` 装 / 卸插件：带兼容性检查，会自动登记 bundles）
+import { readFile } from 'node:fs/promises';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import { VPS_ROOT } from '../context.js';
+import { DshModelError } from '../errors.js';
+import { L } from '../i18n.js';
+import { run, runInherit, which } from '../util/exec.js';
+import { exists } from '../util/fs.js';
+const MAC_APPS = ['/Applications/DeepSeek Harness.app', join(homedir(), 'Applications/DeepSeek Harness.app')];
+export async function locateDshCli(ctx) {
+    if (ctx.env.DSH_MODEL_DSH_BIN)
+        return ctx.env.DSH_MODEL_DSH_BIN;
+    if (ctx.mode === 'vps') {
+        try {
+            const env = await readFile(join(VPS_ROOT, 'state/gate.env'), 'utf8');
+            const bin = env.match(/^DSH_BIN=(.+)$/m)?.[1]?.trim().replace(/^["']|["']$/g, '');
+            if (bin && (await exists(bin)))
+                return bin;
+        }
+        catch {
+            // 往下找
+        }
+    }
+    if (ctx.platform === 'darwin') {
+        for (const app of MAC_APPS) {
+            const bin = join(app, 'Contents/Resources/runtime/cli/bin/dsh');
+            if (await exists(bin))
+                return bin;
+        }
+    }
+    const onPath = await which('dsh');
+    if (onPath)
+        return onPath;
+    throw new DshModelError('dsh_cli_not_found', L('找不到 dsh 的命令行', 'Could not find the dsh command line'), L('请确认 dsh 已安装；或设置 DSH_MODEL_DSH_BIN 指向 dsh 可执行文件', 'Make sure dsh is installed, or set DSH_MODEL_DSH_BIN to the dsh executable'));
+}
+function dshEnv(ctx, proxy) {
+    return { ...ctx.env, DSH_HOME: ctx.dshHome, ...(proxy ? { HTTPS_PROXY: proxy, HTTP_PROXY: proxy } : {}) };
+}
+/** vps 模式下 dsh 的文件归 dsh 用户，插件操作要以它的身份跑 */
+function wrap(ctx, bin, args, proxy) {
+    if (ctx.runAs && ctx.isRoot) {
+        const keep = ['DSH_HOME', ...(proxy ? ['HTTPS_PROXY', 'HTTP_PROXY'] : [])].join(',');
+        return { cmd: 'sudo', args: ['-u', ctx.runAs, '-H', `--preserve-env=${keep}`, bin, ...args] };
+    }
+    return { cmd: bin, args };
+}
+/**
+ * 这个 profile 的 node_modules 当初链接的 pnpm 仓库：沿用它。
+ * 从终端调 dsh 和 dsh 桌面版自己用的仓库位置可能不同，不沿用就报 ERR_PNPM_UNEXPECTED_STORE（Mac 上实测）
+ */
+async function profileStoreDir(ctx, profile) {
+    try {
+        const text = await readFile(join(ctx.dshHome, 'profiles', profile, 'node_modules', '.modules.yaml'), 'utf8');
+        const dir = /"?storeDir"?\s*:\s*"?([^"\n,]+)"?/.exec(text)?.[1]?.trim();
+        if (!dir)
+            return null;
+        // 记录的是带版本的子目录（…/store/v11），store-dir 要它的上一级
+        return dir.replace(/\/v\d+$/, '');
+    }
+    catch {
+        return null;
+    }
+}
+/** dsh plugin --profile <p> <pnpm-args...>，输出直接给用户看（pnpm 进度） */
+export async function dshPlugin(ctx, profile, pnpmArgs, proxy) {
+    const bin = await locateDshCli(ctx);
+    const store = await profileStoreDir(ctx, profile);
+    const w = wrap(ctx, bin, ['plugin', '--profile', profile, ...pnpmArgs, ...(store ? [`--config.store-dir=${store}`] : [])], proxy);
+    return runInherit(w.cmd, w.args, { env: dshEnv(ctx, proxy) });
+}
+/** profile 的 package.json 里有没有这个依赖（有就是用户或 dsh 已经装了） */
+export async function profileHasDependency(ctx, profile, name) {
+    try {
+        const pkg = JSON.parse(await readFile(join(ctx.dshHome, 'profiles', profile, 'package.json'), 'utf8'));
+        return pkg.dependencies?.[name] ?? null;
+    }
+    catch {
+        return null;
+    }
+}
+export async function dshCliVersion(ctx) {
+    try {
+        const r = await run(await locateDshCli(ctx), ['--version'], { timeoutMs: 20_000 });
+        return r.code === 0 ? r.stdout.trim() || null : null;
+    }
+    catch {
+        return null;
+    }
+}