dsh-shlr-bilitoolkit 0.0.0-stage → 0.1.0

package/dist/client.js

@@ -0,0 +1,325 @@
+/** BiliClient：统一维护 Session、Cookie、请求细节和账号状态。
+
+内置基础节流与退避，目标是让请求节奏接近正常浏览，降低触发 B 站风控的概率：
+
+- 请求之间保持最小间隔（默认 0.6 秒，带随机抖动）
+- 写操作之间额外保持间隔（默认 1 秒）
+- GET 遇到网络错误或 -799「请求过于频繁」时指数退避重试
+- 写操作绝不自动重试，避免重复评论 / 重复私信
+- HTTP 412 与业务码 -412 视为风控信号，立即停止，不做重试
+*/
+import axios from "axios";
+import { existsSync } from "node:fs";
+import { mkdir, readFile, writeFile } from "node:fs/promises";
+import * as path from "node:path";
+import * as endpoints from "./endpoints";
+import { BiliApiError, CaptchaRequired, CookieError } from "./errors";
+import { VideoAPI } from "./apis/video";
+import { SearchAPI } from "./apis/search";
+import { SpaceAPI } from "./apis/space";
+import { CommentAPI } from "./apis/comment";
+import { MessageAPI } from "./apis/message";
+import { NotifyAPI } from "./apis/notify";
+import { InteractAPI } from "./apis/interact";
+import { RelationAPI } from "./apis/relation";
+import { DynamicAPI } from "./apis/dynamic";
+import { ArticleAPI } from "./apis/article";
+import { ProfileAPI } from "./apis/profile";
+import { QrCodeLogin } from "./apis/auth";
+import { WbiSigner } from "./wbi";
+import { JsonCache } from "./cache";
+import { sleep } from "./util";
+export const USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 " +
+    "(KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36";
+export const DEFAULT_HEADERS = {
+    "User-Agent": USER_AGENT,
+    Referer: "https://www.bilibili.com/",
+};
+const RETRYABLE_CODES = new Set([-799]);
+const RISK_CONTROL_CODES = new Set([-412]);
+const RISK_CONTROL_HTTP = new Set([412]);
+export class BiliClient {
+    http;
+    timeout; // 毫秒
+    cookieFile;
+    minInterval; // 毫秒
+    jitter; // 毫秒
+    maxRetries;
+    writeInterval; // 毫秒
+    video;
+    search;
+    space;
+    comment;
+    message;
+    notify;
+    interact;
+    relation;
+    dynamic;
+    article;
+    profile;
+    auth;
+    wbi;
+    cache;
+    lastRequestAt = 0;
+    lastWriteAt = 0;
+    cookies = {};
+    _account = undefined;
+    constructor(opts = {}) {
+        this.timeout = (opts.timeout ?? 10) * 1000;
+        this.cookieFile = opts.cookieFile ?? "bili_cookies.json";
+        this.minInterval = (opts.minInterval ?? 0.6) * 1000;
+        this.jitter = (opts.jitter ?? 0.4) * 1000;
+        this.maxRetries = Math.max(0, opts.maxRetries ?? 2);
+        this.writeInterval = (opts.writeInterval ?? 1.0) * 1000;
+        this.http = axios.create({ timeout: this.timeout, headers: { ...DEFAULT_HEADERS } });
+        this.wbi = new WbiSigner(this);
+        this.video = new VideoAPI(this);
+        this.search = new SearchAPI(this);
+        this.space = new SpaceAPI(this);
+        this.comment = new CommentAPI(this);
+        this.message = new MessageAPI(this);
+        this.notify = new NotifyAPI(this);
+        this.interact = new InteractAPI(this);
+        this.relation = new RelationAPI(this);
+        this.dynamic = new DynamicAPI(this);
+        this.article = new ArticleAPI(this);
+        this.profile = new ProfileAPI(this);
+        this.auth = new QrCodeLogin(this);
+        this.cache = new JsonCache(opts.cacheFile ?? process.env.BILI_CACHE_FILE ?? "bili_cache.json");
+    }
+    // ------------------------------------------------------------------ 请求
+    async get(url, params, headers, wbi = false) {
+        if (wbi)
+            params = await this.wbi.sign(params);
+        return this.request("GET", url, { params, headers });
+    }
+    async post(url, data, headers, wbi = false, params, jsonBody = false) {
+        if (wbi && data && typeof data === "object") {
+            data = await this.wbi.sign(data);
+        }
+        return this.request("POST", url, { data, params, headers, jsonBody });
+    }
+    async getSigned(url, params, headers) {
+        for (let attempt = 0; attempt < 2; attempt++) {
+            try {
+                return await this.get(url, await this.wbi.sign(params), headers);
+            }
+            catch (exc) {
+                if (attempt === 0 &&
+                    exc instanceof BiliApiError &&
+                    (exc.code === -403 || exc.code === -352)) {
+                    await this.wbi.refresh();
+                    continue;
+                }
+                throw exc;
+            }
+        }
+        throw new BiliApiError("WBI 签名请求失败");
+    }
+    async request(method, url, extra) {
+        // B 站写接口要求表单编码；jsonBody=true 时才按 JSON 发送（如动态点赞）
+        const { jsonBody, data: rawData, headers: rawHeaders, ...rest } = extra;
+        let data = rawData;
+        let headers = rawHeaders;
+        if (!jsonBody &&
+            data != null &&
+            typeof data === "object" &&
+            !(data instanceof URLSearchParams)) {
+            const form = new URLSearchParams();
+            for (const [k, v] of Object.entries(data)) {
+                form.append(k, v == null ? "" : String(v));
+            }
+            data = form;
+            headers = {
+                ...(headers ?? {}),
+                "Content-Type": "application/x-www-form-urlencoded",
+            };
+        }
+        const config = {
+            method: method,
+            url,
+            ...rest,
+            data,
+            headers,
+        };
+        const write = method.toUpperCase() !== "GET";
+        const attempts = write ? 0 : this.maxRetries;
+        for (let attempt = 0; attempt <= attempts; attempt++) {
+            await this.throttle(write);
+            try {
+                const response = await this.http.request(config);
+                // 像 requests 的 cookie jar 一样，自动吸收响应里的 Set-Cookie
+                // （登录接口正是通过 Set-Cookie 下发 SESSDATA / bili_jct）。
+                const setCookie = response.headers["set-cookie"];
+                if (setCookie) {
+                    this.updateCookiesFromHeader(Array.isArray(setCookie) ? setCookie : [String(setCookie)]);
+                }
+                this.rememberRequest(write);
+                let payload = response.data;
+                if (typeof payload === "string") {
+                    try {
+                        payload = JSON.parse(payload);
+                    }
+                    catch {
+                        throw new BiliApiError("接口返回的不是合法 JSON");
+                    }
+                }
+                if (!payload || typeof payload !== "object") {
+                    throw new BiliApiError("接口返回格式异常");
+                }
+                const code = payload.code;
+                if (code === 0)
+                    return payload;
+                if (RISK_CONTROL_CODES.has(code)) {
+                    throw new BiliApiError(`触发 B 站风控（${code}），已停止请求，请过段时间再试`, code, payload.data);
+                }
+                if (RETRYABLE_CODES.has(code) && attempt < attempts) {
+                    await this.backoff(attempt);
+                    continue;
+                }
+                const data = payload.data;
+                const gaData = (data ?? {}).ga_data ?? {};
+                const decisions = gaData.decisions ?? [];
+                if (decisions.length) {
+                    throw new CaptchaRequired(`B 站要求人机验证（${decisions.join(", ")}），该操作需要先在网页端完成验证`, code, data, decisions);
+                }
+                throw new BiliApiError(payload.message ?? "接口返回错误", code, data);
+            }
+            catch (exc) {
+                if (exc instanceof BiliApiError)
+                    throw exc;
+                if (axios.isAxiosError(exc)) {
+                    const status = exc.response?.status;
+                    if (status && RISK_CONTROL_HTTP.has(status)) {
+                        throw new BiliApiError(`触发 B 站风控（HTTP ${status}），已停止请求`, undefined, undefined);
+                    }
+                    if (attempt < attempts) {
+                        this.rememberRequest(write);
+                        await this.backoff(attempt);
+                        continue;
+                    }
+                    throw new BiliApiError(`请求失败: ${exc.message}`);
+                }
+                throw exc;
+            }
+        }
+        throw new BiliApiError("请求重试次数已用完");
+    }
+    // ------------------------------------------------------------------ 节流
+    async throttle(write) {
+        const now = Date.now();
+        let wait = 0;
+        if (this.minInterval > 0) {
+            const gap = this.minInterval + Math.random() * Math.max(0, this.jitter);
+            wait = Math.max(wait, gap - (now - this.lastRequestAt));
+        }
+        if (write && this.writeInterval > 0) {
+            wait = Math.max(wait, this.writeInterval - (now - this.lastWriteAt));
+        }
+        if (wait > 0)
+            await sleep(wait);
+    }
+    rememberRequest(write) {
+        this.lastRequestAt = Date.now();
+        if (write)
+            this.lastWriteAt = this.lastRequestAt;
+    }
+    async backoff(attempt) {
+        await sleep(Math.min(8000, 2 ** attempt + Math.random() * 500));
+    }
+    // ------------------------------------------------------------------ Cookie
+    applyCookieHeader() {
+        const cookie = Object.entries(this.cookies)
+            .map(([k, v]) => `${k}=${v}`)
+            .join("; ");
+        this.http.defaults.headers.common["Cookie"] = cookie;
+    }
+    async loadCookies() {
+        if (!existsSync(this.cookieFile)) {
+            throw new CookieError(`未找到 Cookie 文件 [${this.cookieFile}]，请先完成登录`);
+        }
+        let text;
+        try {
+            text = await readFile(this.cookieFile, "utf-8");
+        }
+        catch (exc) {
+            throw new CookieError(`Cookie 文件读取失败: ${exc.message}`);
+        }
+        let dict;
+        try {
+            dict = JSON.parse(text);
+        }
+        catch (exc) {
+            throw new CookieError(`Cookie 文件格式错误: ${exc.message}`);
+        }
+        if (!dict || typeof dict !== "object" || Object.keys(dict).length === 0) {
+            throw new CookieError(`Cookie 文件 [${this.cookieFile}] 内容为空`);
+        }
+        this.cookies = dict;
+        this.applyCookieHeader();
+        this._account = undefined;
+        return this.cookieFile;
+    }
+    async saveCookies() {
+        await mkdir(path.dirname(this.cookieFile), { recursive: true });
+        await writeFile(this.cookieFile, JSON.stringify(this.cookies, null, 4), "utf-8");
+        return this.cookieFile;
+    }
+    /** 清空本地缓存（WBI 密钥、bvid→aid 等），下次请求会自动重新拉取。 */
+    clearCache() {
+        this.cache.clear();
+    }
+    /** 供登录流程（未来实现）把 Set-Cookie 写入本地状态。 */
+    updateCookiesFromHeader(setCookie) {
+        for (const sc of setCookie) {
+            const [pair] = sc.split(";");
+            const idx = pair.indexOf("=");
+            if (idx > 0)
+                this.cookies[pair.slice(0, idx).trim()] = pair.slice(idx + 1).trim();
+        }
+        this.applyCookieHeader();
+    }
+    get isLoggedIn() {
+        return !!this.cookies["SESSDATA"];
+    }
+    get csrf() {
+        const token = this.cookies["bili_jct"];
+        if (!token)
+            throw new CookieError("缺少 bili_jct，无法执行写操作，请重新登录");
+        return token;
+    }
+    // ------------------------------------------------------------------ 账号
+    async fetchAccount() {
+        try {
+            const payload = await this.get(endpoints.NAV);
+            const data = payload?.data ?? {};
+            return data.isLogin ? data : null;
+        }
+        catch (exc) {
+            if (exc instanceof BiliApiError && exc.code === -101)
+                return null;
+            throw exc;
+        }
+    }
+    async getAccount() {
+        if (this._account === undefined)
+            this._account = await this.fetchAccount();
+        return this._account;
+    }
+    get username() {
+        if (this._account && typeof this._account === "object" && this._account.uname) {
+            return this._account.uname;
+        }
+        return null;
+    }
+    async checkLogin() {
+        return (await this.fetchAccount()) !== null;
+    }
+    async requireLogin() {
+        const account = await this.fetchAccount();
+        if (!account)
+            throw new CookieError("未登录或 Cookie 已过期，请重新登录");
+        this._account = account;
+        return account;
+    }
+}

package/dist/dsh.d.ts

@@ -0,0 +1,25 @@
+/** dsh 插件入口（对齐官方 DeepSeek Harness / cordis 插件规范）。
+
+本文件即组合包的插件入口（`package.json` 的 main 指向它）。需要导出：
+- `name`   插件名（Cordis 内部标识）
+- `inject` 声明依赖的服务，这里注入 `tools`
+- `apply(ctx)` 入口，由 Cordis 调用；内部用 `ctx.tools.register(defineTool(...))`
+  注册每个工具。
+
+工具定义仍以 JSON-Schema 形式写在 `TOOLS` 里（便于阅读），`apply` 会转换成
+官方的 `defineTool({ name, description, parameters, output, execute })` 形式。
+*/
+import { BiliClient } from "./client";
+import { Collection, Page, PlayInfo, Subtitle, User, Video } from "./models";
+export declare const name = "bili-toolkit";
+export declare const inject: string[];
+export interface DshTool {
+    id: string;
+    description: string;
+    parameters: Record<string, unknown>;
+    handler: (args: any) => Promise<unknown>;
+}
+export declare const TOOLS: DshTool[];
+export declare function apply(ctx: any): void;
+export * from "./index";
+export { BiliClient, Video, User, Page, PlayInfo, Subtitle, Collection };