specific-ai-cli 0.0.0-stage → 0.1.1

package/dist/platforms/bilibili/web/client.js

@@ -0,0 +1,311 @@
+import { SpecificAIError } from '../../../core/errors.js';
+import { parseJson } from '../../../core/http.js';
+import { compactJson, jsonLoads } from '../../../core/py.js';
+import * as rand from '../../../core/rand.js';
+import { authError, httpClient, isGuest, scope } from '../../../core/toolkit.js';
+import * as api from './api.js';
+import { fingerCore, fingerPayload } from './gaia.js';
+import { API, BROWSER, COOKIE_DOMAIN, COOKIE_ORDER, headers, PROFILE } from './profile.js';
+import { bLsid, correspondPath, dmImgParams, encWbi, liveBuvid, mixinKey, murmur3Hex, sid, TICKET_KEY_ID, ticketHexSign, uuidInfoc } from './sign.js';
+/**
+ * B 站 web 端的会话（上游 builder/auth.py 的 BiliAuth + utils/http_util.py）：
+ * 设备 cookie、bili_ticket、WBI 密钥的生命周期，以及带风控退避重试的请求。
+ */
+/** 触发重试的业务错误码与 HTTP 状态（上游 RETRYABLE_CODES / RETRYABLE_STATUS）。 */
+const RETRYABLE_CODES = new Set([-352, -799, -509, -412]);
+const RETRYABLE_STATUS = new Set([412, 429, 502, 503, 504]);
+/** WBI 密钥缓存半小时。 */
+const MIXIN_TTL = 1800_000;
+/** bili_ticket 提前 1 小时换新。 */
+const TICKET_SKEW = 3600;
+export class Bili {
+    ctx;
+    http;
+    jar;
+    constructor(ctx) {
+        this.ctx = ctx;
+        this.http = httpClient(ctx, { browser: BROWSER, os: 'windows', cookieOrder: COOKIE_ORDER });
+        this.jar = this.http.jar;
+    }
+    get csrf() {
+        return this.jar.get('bili_jct') ?? '';
+    }
+    get mid() {
+        return this.jar.get('DedeUserID') ?? '';
+    }
+    get isLogin() {
+        return Boolean(this.jar.get('SESSDATA'));
+    }
+    setCookie(name, value) {
+        this.jar.set(name, value, COOKIE_DOMAIN);
+    }
+    setDefault(name, value) {
+        if (!this.jar.has(name))
+            this.setCookie(name, value());
+    }
+    // ---------------------------------------------------------------- 设备初始化
+    /**
+     * 每条命令开始时调用，相当于上游的 `from_cookie()` / `anonymous()`：
+     * - 没有设备 cookie 的游客：完整跑一遍匿名设备初始化（spi → bili_ticket → 指纹上报），结果缓存在 guest.json；
+     * - 登录用户缺设备 cookie 时补齐；
+     * - 会话级的 sid / PVID / b_lsid 每次重新生成（b_lsid）或补齐。
+     */
+    async init() {
+        if (isGuest(this.ctx) && !this.jar.has('buvid3')) {
+            await this.anonymous();
+            return;
+        }
+        if (!this.jar.has('buvid3')) {
+            for (const [name, value] of await this.deviceCookies())
+                this.setDefault(name, () => value);
+        }
+        fillSessionCookies((name) => this.jar.has(name), (name, value) => this.setCookie(name, value));
+        if (isGuest(this.ctx) && Number(this.jar.get('bili_ticket_expires') ?? 0) - TICKET_SKEW < rand.nowSeconds()) {
+            await this.refreshTicket();
+        }
+    }
+    async anonymous() {
+        this.jar.cookies.splice(0);
+        for (const [name, value] of await this.deviceCookies())
+            this.setCookie(name, value);
+        await this.refreshTicket();
+        await this.reportFinger();
+    }
+    /** 服务端只下发 buvid3 / buvid4，其余由浏览器 JS 本地生成（上游 build_device_cookies）。 */
+    async deviceCookies() {
+        const spi = await this.get(`${API}/x/frontend/finger/spi`, { headers: headers('GET').get() });
+        return [
+            ['buvid3', spi.b_3],
+            ['b_nut', String(rand.nowSeconds())],
+            ['_uuid', uuidInfoc()],
+            ['home_feed_column', '5'],
+            ['buvid4', spi.b_4],
+            ['browser_resolution', PROFILE.browserResolution],
+            ['buvid_fp', murmur3Hex(compactJson(fingerCore()))],
+            ['LIVE_BUVID', liveBuvid()],
+            ['theme-tip-show', 'SHOWED'],
+            ['CURRENT_FNVAL', '4048'],
+            ['sid', sid()],
+            ['PVID', '1'],
+            ['ogv_device_support_dolby', '0'],
+            ['ogv_device_support_hdr', '0'],
+            ['b_lsid', bLsid()],
+        ];
+    }
+    /** 换取 bili_ticket，响应里顺带的 WBI 密钥一并缓存（上游 refresh_ticket）。 */
+    async refreshTicket() {
+        const ts = rand.nowSeconds();
+        const data = await this.post(`${API}/bapis/bilibili.api.ticket.v1.Ticket/GenWebTicket`, {
+            headers: headers('POST').get(),
+            query: [
+                ['key_id', TICKET_KEY_ID],
+                ['hexsign', ticketHexSign(ts)],
+                ['context[ts]', ts],
+                ['csrf', this.csrf],
+            ],
+        });
+        const expires = Number(data.created_at ?? rand.nowSeconds()) + Number(data.ttl ?? 259200);
+        this.setCookie('bili_ticket', data.ticket);
+        this.setCookie('bili_ticket_expires', String(expires));
+        if (data.nav?.img && data.nav.sub)
+            this.saveMixin(mixinKey(data.nav.img, data.nav.sub));
+    }
+    /** 明文指纹上报 ExClimbWuzhi。失败只记日志：它只影响风控评分，不影响功能。 */
+    async reportFinger(referer = 'https://www.bilibili.com/') {
+        const h = headers('POST', { accept: '*/*' }).referer(referer).set('content-type', 'application/json;charset=UTF-8');
+        try {
+            const payload = compactJson(fingerPayload(this.jar.get('_uuid') ?? '', referer));
+            await this.post(`${API}/x/internal/gaia-gateway/ExClimbWuzhi`, { headers: h.get(), json: { payload } }, { raw: true });
+        }
+        catch (err) {
+            this.ctx.log.debug(`ExClimbWuzhi 上报失败：${err.message}`);
+        }
+    }
+    // ---------------------------------------------------------------- WBI
+    saveMixin(key) {
+        this.ctx.credential.extra.wbi = { key, at: rand.now() };
+    }
+    /** 当前的 mixin_key：缓存半小时，过期后从 nav 重新派生。 */
+    async mixinKey() {
+        const cached = this.ctx.credential.extra.wbi;
+        if (cached?.key && rand.now() - cached.at < MIXIN_TTL)
+            return cached.key;
+        const nav = await this.json({ url: `${API}/x/web-interface/nav`, headers: headers('GET').get() });
+        const img = nav.data?.wbi_img;
+        if (!img?.img_url)
+            throw new SpecificAIError('UPSTREAM', 'nav 没有返回 wbi_img，无法签名', { detail: { code: nav.code } });
+        const key = mixinKey(img.img_url, img.sub_url);
+        this.saveMixin(key);
+        return key;
+    }
+    /** 签名后的 query：原参数顺序 + w_rid + wts。 */
+    async wbi(params) {
+        return encWbi(params, await this.mixinKey());
+    }
+    /** 带 dm_img_* 指纹再签名，空间类接口需要。 */
+    async wbiDm(params, sampleCount = 0) {
+        return this.wbi([...params, ...dmImgParams(sampleCount)]);
+    }
+    // ---------------------------------------------------------------- 请求
+    /** 带风控退避重试的原始请求（上游 http_util.request）。 */
+    request(req, retries = 3) {
+        return retrying(() => this.http.request(req), this.ctx.signal, retries);
+    }
+    /** 请求并解析 JSON，不检查业务码。 */
+    async json(req) {
+        const res = await this.request(req);
+        if (res.status === 412 || res.status === 429) {
+            throw new SpecificAIError('RISK_CONTROL', `B 站拒绝了请求（HTTP ${res.status}），请稍后再试`, { detail: { kind: 'rate_limit', status: res.status } });
+        }
+        return parseJson(res);
+    }
+    /** 请求二进制内容（弹幕 protobuf 等）：HTTP 状态不是 200、或者返回的是 JSON 错误时报错，不把错误页当数据解。 */
+    async bytes(req) {
+        const res = await this.request(req);
+        if (res.status === 412 || res.status === 429) {
+            throw new SpecificAIError('RISK_CONTROL', `B 站拒绝了请求（HTTP ${res.status}），请稍后再试`, { detail: { kind: 'rate_limit', status: res.status } });
+        }
+        if (res.status !== 200)
+            throw new SpecificAIError('UPSTREAM', `B 站返回 HTTP ${res.status}`, { detail: { status: res.status } });
+        if ((res.headers.get('content-type') ?? '').includes('json')) {
+            const body = await parseJson(res);
+            check(this.ctx, body);
+            throw new SpecificAIError('UPSTREAM', 'B 站返回了 JSON，不是二进制数据', { detail: { code: body?.code } });
+        }
+        return new Uint8Array(await res.arrayBuffer());
+    }
+    /** 请求并检查业务码，返回 data。raw 为 true 时返回整个 JSON。 */
+    async call(req, options = {}) {
+        const body = await this.json(req);
+        check(this.ctx, body);
+        return (options.raw ? body : body.data);
+    }
+    get(url, req = {}, options) {
+        return this.call({ ...req, url, method: 'GET' }, options);
+    }
+    post(url, req = {}, options) {
+        return this.call({ ...req, url, method: 'POST' }, options);
+    }
+    /** 写操作前检查登录态（registry 已保证不是游客，这里防 cookie 不全）。 */
+    requireLogin() {
+        if (!this.isLogin || !this.csrf)
+            throw authError(this.ctx, '当前账号缺少 SESSDATA / bili_jct，请重新登录');
+    }
+}
+/**
+ * 会话级 cookie（上游 BiliAuth.from_cookie）：sid、PVID 缺了才补，b_lsid 每次重新生成。
+ * 上游 `cookie.setdefault('sid', gen_sid())` 总会先算出 gen_sid()，随机数照样消耗。
+ */
+export function fillSessionCookies(has, set) {
+    const newSid = sid();
+    if (!has('sid'))
+        set('sid', newSid);
+    if (!has('PVID'))
+        set('PVID', '1');
+    set('b_lsid', bLsid());
+}
+/**
+ * 风控退避重试（上游 http_util.request）：HTTP 200 但业务码在 RETRYABLE_CODES 里、或 HTTP 状态在 RETRYABLE_STATUS 里时，
+ * 依次等 1、2、4 秒再发，最多再发 retries 次，返回最后一次的响应。B 站接口和极验共用。
+ */
+export async function retrying(send, signal, retries = 3) {
+    let res;
+    for (let attempt = 0; attempt <= retries; attempt++) {
+        res = await send();
+        if (res.status === 200) {
+            const code = await peekCode(res);
+            if (code === undefined || !RETRYABLE_CODES.has(code))
+                return res;
+        }
+        else if (!RETRYABLE_STATUS.has(res.status))
+            return res;
+        if (attempt < retries)
+            await rand.sleep(1000 * 2 ** attempt, signal);
+    }
+    return res;
+}
+async function peekCode(res) {
+    if (!(res.headers.get('content-type') ?? '').includes('json'))
+        return undefined;
+    try {
+        const body = jsonLoads(await res.clone().text());
+        return typeof body?.code === 'number' ? body.code : undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+/** 请求过于频繁（-509、-799）和按 IP 限流的 -412（request was banned，与 HTTP 412 同属限流）；其余风控码（-352 风控校验失败）为 blocked。 */
+const RATE_LIMIT_CODES = new Set([-509, -799, -412]);
+/** 业务码映射成 specific-ai 的错误（AGENTS 6.4）。 */
+export function check(ctx, body) {
+    const code = body?.code;
+    if (code === 0 || code === undefined)
+        return;
+    const message = String(body.message ?? body.msg ?? '');
+    if (code === -101 || code === -111 || code === 86095)
+        throw authError(ctx, message || undefined);
+    if (RETRYABLE_CODES.has(code)) {
+        const kind = RATE_LIMIT_CODES.has(code) ? 'rate_limit' : 'blocked';
+        throw new SpecificAIError('RISK_CONTROL', `B 站风控拦截：${message || code}`, { detail: { kind, code, message } });
+    }
+    if (code === 340022 || code === 10031 || code === -625) {
+        throw new SpecificAIError('RISK_CONTROL', `需要人机验证：${message || code}`, { detail: { kind: 'captcha', code, message } });
+    }
+    throw new SpecificAIError('UPSTREAM', message || `B 站返回错误 ${code}`, { detail: { code, message } });
+}
+/**
+ * Cookie 续期（上游 refresh_cookies）：每天最多检查一次 cookie/info，需要时换新的 SESSDATA 并确认。
+ * 失败只记警告，当前 cookie 往往还能用一阵。响应里的 Set-Cookie 由 HttpClient 并进 cookie 罐，命令结束时由 core 落盘。
+ */
+export async function refreshCookies(b) {
+    const tokens = scope(b.ctx.credential).tokens;
+    const old = tokens.refresh_token;
+    if (!old || !b.isLogin)
+        return;
+    const checked = Number(b.ctx.credential.extra.refresh_checked_at ?? 0);
+    if (rand.now() - checked < 86_400_000)
+        return;
+    b.ctx.credential.extra.refresh_checked_at = rand.now();
+    try {
+        const info = await api.cookieInfo(b);
+        if (info.code !== 0)
+            throw new Error(`查 cookie/info 失败：${info.code} ${info.message ?? ''}`);
+        if (!info.data?.refresh)
+            return;
+        // 用服务端给的毫秒时间戳，避开本地时钟偏移
+        const csrf = await api.refreshCsrf(b, correspondPath(info.data.timestamp || rand.now()));
+        if (!csrf)
+            throw new Error('correspond 页面没有返回 refresh_csrf（路径过期或登录态失效）');
+        const before = b.jar.get('SESSDATA');
+        const r = await api.cookieRefresh(b, csrf, old);
+        if (r.code !== 0)
+            throw new Error(`换 Cookie 失败：${r.code} ${r.message ?? ''}`);
+        if (b.jar.get('SESSDATA') === before)
+            throw new Error('换 Cookie 成功，但响应没有带新的 SESSDATA');
+        // 旧 refresh_token 已经用掉：响应里没有新的就删掉，之后不再尝试续期
+        if (r.data?.refresh_token)
+            tokens.refresh_token = r.data.refresh_token;
+        else {
+            delete tokens.refresh_token;
+            b.ctx.log.warn('续期响应没有新的 refresh_token，之后不能自动续期；过期后请重新登录');
+        }
+        // 两个参数分属新旧两代：csrf 用刷新后的 bili_jct，refresh_token 用刷新前的
+        const confirm = await api.confirmRefresh(b, old);
+        if (confirm.code !== 0) {
+            b.ctx.log.warn(`登录态已续期，但确认更新失败：${confirm.code} ${confirm.message ?? ''}（新 Cookie 可用，旧会话没有失效）`);
+            return;
+        }
+        b.ctx.log.info('登录态已自动续期');
+    }
+    catch (err) {
+        b.ctx.log.warn(`登录态续期失败：${err.message}`);
+    }
+}
+/** 建立会话，完成设备初始化；登录用户顺带做每日一次的续期检查。 */
+export async function bili(ctx) {
+    const b = new Bili(ctx);
+    await b.init();
+    await refreshCookies(b);
+    return b;
+}