@zeppeli777/tools-doubao 0.0.0-stage → 0.1.0

package/scripts/test-offline.mjs

@@ -0,0 +1,313 @@
+// 离线单测：不联网，用合成 SSE 报文验证解析器、配置解析与插件结构。
+//   node scripts/test-offline.mjs
+import assert from "node:assert/strict";
+import path from "node:path";
+import { deflateSync, crc32 } from "node:zlib";
+import { parseSseBlocks, parsePcImages, parseWebImages, normalizeRatio } from "../src/client.mjs";
+import { DoubaoError, DoubaoAuthError, DoubaoRiskError } from "../src/http.mjs";
+import { loadConfig, DEFAULTS } from "../src/config.mjs";
+import { qrPngToAscii } from "../src/qlogin.mjs";
+import { encodePng, decodePng, cropPngBottom } from "../src/png.mjs";
+import { parseExtendFrac } from "../src/watermark.mjs";
+import { DoubaoTools } from "../index.mjs";
+
+let passed = 0;
+function ok(name, fn) {
+  try {
+    fn();
+    passed++;
+    console.log(`  ✅ ${name}`);
+  } catch (err) {
+    console.error(`  ❌ ${name}\n${err.stack}`);
+    process.exitCode = 1;
+  }
+}
+
+// ————— parseSseBlocks —————
+console.log("parseSseBlocks");
+ok("基本切块与 event 名", () => {
+  const blocks = parseSseBlocks('event: SSE_ACK\ndata: {"a":1}\n\ndata: {"b":2}\n\n');
+  assert.equal(blocks.length, 2);
+  assert.equal(blocks[0].event, "SSE_ACK");
+  assert.equal(blocks[0].json.a, 1);
+  assert.equal(blocks[1].event, "message");
+  assert.equal(blocks[1].json.b, 2);
+});
+ok("多行 data: 合并", () => {
+  const blocks = parseSseBlocks('data: {"x":\ndata: 1}\n');
+  assert.equal(blocks[0].json.x, 1);
+});
+ok("非 JSON 数据块 json 为 null", () => {
+  const blocks = parseSseBlocks("data: ping\n");
+  assert.equal(blocks[0].json, null);
+});
+
+// ————— parsePcImages（samantha 2010 形态，doubao2api 真值） —————
+console.log("parsePcImages");
+// 真实响应是三层 JSON 嵌套（SSE data → event_data 字符串 → message.content 字符串），程序化构造
+const pcImageData = {
+  data: [
+    {
+      key: "img1",
+      image_ori: { url: "https://cdn/ori.jpg", width: 1024, height: 1024, format: "jpeg" },
+      image_raw: { url: "https://cdn/raw.jpg" },
+      image_thumb: { url: "https://cdn/thumb.jpg" },
+    },
+  ],
+};
+const pcBlockImage = `data: ${JSON.stringify({
+  event_type: 2001,
+  event_data: JSON.stringify({ message: { content_type: 2010, content: JSON.stringify(pcImageData), status: 1 } }),
+})}`;
+const pcBlockText = `data: ${JSON.stringify({
+  event_type: 2001,
+  event_data: JSON.stringify({ message: { content_type: 2001, content: JSON.stringify({ text: "已生成" }) } }),
+})}`;
+const pcFixture = [pcBlockImage, "", pcBlockText, ""].join("\n");
+ok("提取 image_ori 为最优档（pc 顺序）", () => {
+  const { images, text } = parsePcImages(pcFixture);
+  assert.equal(images.length, 1);
+  assert.equal(images[0].url, "https://cdn/ori.jpg");
+  assert.equal(images[0].width, 1024);
+  assert.equal(text, "已生成");
+});
+ok("event_type 2005 抛错", () => {
+  assert.throws(() => parsePcImages('data: {"event_type":2005,"event_data":"boom"}'), /boom/);
+});
+ok("JSON 错误体分类：登录失效", () => {
+  assert.throws(() => parsePcImages('{"code":710012001,"msg":"登录已过期"}'), DoubaoAuthError);
+});
+ok("JSON 错误体分类：风控", () => {
+  try {
+    parsePcImages('{"code":710022004,"msg":"verify"}');
+    assert.fail("应当抛出");
+  } catch (err) {
+    assert.ok(err instanceof DoubaoRiskError);
+    assert.match(err.hint, /风控|冷却/);
+  }
+});
+ok("gateway-error 抛登录失效", () => {
+  assert.throws(
+    () => parsePcImages('event: gateway-error\ndata: {"code":"710012001","message":"user invalid"}'),
+    DoubaoAuthError,
+  );
+});
+
+// ————— parseWebImages（chat_ability 2074 形态，web-mcp 笔记） —————
+console.log("parseWebImages");
+const webBlock = {
+  block_type: 2074,
+  content: {
+    creation_block: {
+      creations: [
+        {
+          image: {
+            image_raw: { url: "https://cdn-raw/x.jpg", width: 1440, height: 810 },
+            lwm3_url: "https://cdn-lwm/x.jpg",
+          },
+        },
+      ],
+    },
+  },
+};
+const webFixture = [
+  'event: SSE_ACK\ndata: {"ack_client_meta":{"conversation_id":"7421"}}',
+  "",
+  `event: STREAM_CHUNK\ndata: ${JSON.stringify({ patch_op: [{ patch_value: { content_block: [webBlock] } }] })}`,
+  "",
+  'event: STREAM_CHUNK\ndata: {"text":"生成中"}',
+  "",
+  'event: SSE_REPLY_END\ndata: {}',
+  "",
+].join("\n");
+ok("2074 creation 提取，image_raw 优先（web 顺序）", () => {
+  const { images, text, conversationId } = parseWebImages(webFixture);
+  assert.equal(images.length, 1);
+  assert.equal(images[0].url, "https://cdn-raw/x.jpg");
+  assert.equal(images[0].width, 1440);
+  assert.equal(conversationId, "7421");
+  assert.ok(text.includes("生成中"));
+});
+ok("无图但带订阅文案 → 付费墙提示", () => {
+  const fixture = [
+    'event: STREAM_CHUNK',
+    `data: ${JSON.stringify({ patch_op: [{ patch_value: { content_block: [{ block_type: 10000, content: { text_block: { text: "这是豆包订阅标准套餐专属能力" } } }] } }] })}`,
+    "",
+  ].join("\n");
+  const err = (() => {
+    try {
+      parseWebImages(fixture);
+      return null;
+    } catch (e) {
+      return e;
+    }
+  })();
+  assert.ok(err instanceof DoubaoError);
+  assert.match(err.message, /订阅/);
+});
+ok("深度兜底扫描命中嵌套 2074", () => {
+  const fixture = `event: FULL_MSG_NOTIFY\ndata: ${JSON.stringify({ deep: { nested: [webBlock] } })}\n\n`;
+  const { images } = parseWebImages(fixture);
+  assert.equal(images.length, 1);
+  assert.equal(images[0].url, "https://cdn-raw/x.jpg");
+});
+
+// ————— 二维码 PNG 解码 —————
+console.log("qrPngToAscii");
+function pngChunk(type, data) {
+  const len = Buffer.alloc(4);
+  len.writeUInt32BE(data.length);
+  const body = Buffer.concat([Buffer.from(type, "ascii"), data]);
+  const crc = Buffer.alloc(4);
+  crc.writeUInt32BE(crc32(body) >>> 0);
+  return Buffer.concat([len, body, crc]);
+}
+/** 自造伪二维码 PNG（8-bit 灰度、filter 0，四周 4 模块静区）。 */
+function makeQrPng(modules, scale = 8) {
+  const w = (modules + 8) * scale;
+  const ihdr = Buffer.alloc(13);
+  ihdr.writeUInt32BE(w, 0);
+  ihdr.writeUInt32BE(w, 4);
+  ihdr[8] = 8; // bit depth
+  ihdr[9] = 0; // grayscale
+  const rows = [];
+  for (let y = 0; y < w; y++) {
+    const row = Buffer.alloc(1 + w);
+    row[0] = 0; // filter: none
+    for (let x = 0; x < w; x++) {
+      const mx = Math.floor(x / scale) - 4;
+      const my = Math.floor(y / scale) - 4;
+      const inGrid = mx >= 0 && my >= 0 && mx < modules && my < modules;
+      row[1 + x] = inGrid && (mx * 7 + my * 11 + mx * my) % 3 === 0 ? 0 : 255;
+    }
+    rows.push(row);
+  }
+  return Buffer.concat([
+    Buffer.from([0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a]),
+    pngChunk("IHDR", ihdr),
+    pngChunk("IDAT", deflateSync(Buffer.concat(rows))),
+    pngChunk("IEND", Buffer.alloc(0)),
+  ]);
+}
+ok("解码自造 PNG 为 ASCII 半块画", () => {
+  const ascii = qrPngToAscii(makeQrPng(25));
+  assert.ok(ascii, "应当解码成功");
+  const lines = ascii.split("\n");
+  assert.equal(lines.length, Math.ceil(25 / 2) + 2, "13 对行 + 上下白边");
+  // (0,0) 依图案公式为暗、(0,1) 为亮 → 首个内容行第 1 个内容字符应为 "▀"
+  const content = lines[1].trim();
+  assert.ok(content.startsWith("▀"), `首字符应为 ▀，实际: ${content.slice(0, 3)}`);
+});
+ok("非 PNG 输入返回 null", () => {
+  assert.equal(qrPngToAscii(Buffer.from("not a png")), null);
+});
+
+// ————— PNG 编解码与去水印裁剪 —————
+console.log("png & watermark crop");
+ok("encodePng/decodePng 往返无损", () => {
+  const w = 16;
+  const h = 10;
+  const rgb = Buffer.alloc(w * h * 3);
+  for (let y = 0; y < h; y++) {
+    for (let x = 0; x < w; x++) {
+      const i = (y * w + x) * 3;
+      rgb[i] = x * 16;
+      rgb[i + 1] = y * 25;
+      rgb[i + 2] = 128;
+    }
+  }
+  const back = decodePng(encodePng(w, h, rgb));
+  assert.equal(back.width, w);
+  assert.equal(back.height, h);
+  assert.ok(back.data.equals(rgb));
+});
+ok("cropPngBottom 保留顶部 80%", () => {
+  const w = 16;
+  const h = 10;
+  const rgb = Buffer.alloc(w * h * 3);
+  for (let y = 0; y < h; y++) {
+    for (let x = 0; x < w; x++) {
+      const i = (y * w + x) * 3;
+      rgb[i] = x * 16;
+      rgb[i + 1] = y * 25;
+      rgb[i + 2] = 7;
+    }
+  }
+  const png = encodePng(w, h, rgb);
+  const cropped = cropPngBottom(png, 0.8);
+  assert.equal(cropped.origHeight, 10);
+  assert.equal(cropped.height, 8); // 10 × 0.8
+  const img = decodePng(cropped.buffer);
+  assert.equal(img.width, 16);
+  assert.ok(img.data.equals(rgb.subarray(0, 8 * w * 3)), "顶部 8 行像素应与原图一致");
+});
+ok("cropPngBottom 对非 PNG 抛错", () => {
+  assert.throws(() => cropPngBottom(Buffer.from("\xff\xd8\xff jpeg-ish")), /PNG/);
+});
+ok("parseExtendFrac 解析", () => {
+  assert.equal(parseExtendFrac("1/4"), 0.25);
+  assert.equal(parseExtendFrac("1/5"), 0.2);
+  assert.equal(parseExtendFrac("0.2"), 0.2);
+  assert.equal(parseExtendFrac(0.5), 0.5);
+  assert.equal(parseExtendFrac(undefined), 0.25);
+  assert.equal(parseExtendFrac("垃圾"), 0.25);
+});
+
+// ————— 杂项 —————
+console.log("misc");
+ok("normalizeRatio 白名单", () => {
+  assert.equal(normalizeRatio("16:9"), "16:9");
+  assert.equal(normalizeRatio("2:99"), "1:1");
+  assert.equal(normalizeRatio("", "auto"), "auto");
+});
+ok("loadConfig 解析 DOUBAO_ACCOUNTS", () => {
+  process.env.DOUBAO_ACCOUNTS = "主号:sessionid=abc123456789,ttwid=tt1;小号:sessionid=def123456789";
+  const cfg = loadConfig();
+  assert.equal(cfg.accounts.length, 2);
+  assert.equal(cfg.accounts[0].name, "主号");
+  assert.equal(cfg.accounts[0].ttwid, "tt1");
+  assert.equal(cfg.accounts[1].name, "小号");
+  assert.equal(cfg.model, DEFAULTS.model);
+  delete process.env.DOUBAO_ACCOUNTS;
+});
+ok("loadConfig 过滤无效账号", () => {
+  const cfg = loadConfig({ accounts: [{ name: "坏", sessionid: "x" }, { name: "好", sessionid: "valid-sessionid-123" }] });
+  assert.equal(cfg.accounts.length, 1);
+  assert.equal(cfg.accounts[0].name, "好");
+});
+
+// ————— 插件结构 —————
+console.log("plugin structure");
+ok("导出 generate_image 与 check_login 两个工具", async () => {
+  const plugin = await DoubaoTools({}, {});
+  const tools = plugin?.tool || {};
+  assert.deepEqual(Object.keys(tools).sort(), ["check_login", "generate_image"]);
+  const gi = tools.generate_image;
+  assert.match(gi.description, /豆包/);
+  assert.equal(gi.args.type, "object");
+  assert.deepEqual(gi.args.required, ["prompt"]);
+  assert.equal(gi.args.properties.ratio.enum.length, 6);
+  assert.equal(typeof gi.execute, "function");
+  assert.equal(typeof tools.check_login.execute, "function");
+});
+ok("未配置账号时 execute 返回可读错误而非抛异常", async () => {
+  // 与真实 config.json 隔离：指向空配置文件
+  const { writeFileSync, rmSync, mkdtempSync } = await import("node:fs");
+  const { tmpdir } = await import("node:os");
+  const dir = mkdtempSync(path.join(tmpdir(), "doubao-test-"));
+  const emptyCfg = path.join(dir, "config.json");
+  writeFileSync(emptyCfg, "{}");
+  const prev = process.env.DOUBAO_IMAGE_CONFIG;
+  process.env.DOUBAO_IMAGE_CONFIG = emptyCfg;
+  try {
+    const plugin = await DoubaoTools({}, {});
+    const out = await plugin.tool.generate_image.execute({ prompt: "测试" }, { directory: dir });
+    assert.match(out, /未配置豆包账号/);
+  } finally {
+    if (prev === undefined) delete process.env.DOUBAO_IMAGE_CONFIG;
+    else process.env.DOUBAO_IMAGE_CONFIG = prev;
+    rmSync(dir, { recursive: true, force: true });
+  }
+});
+
+console.log(process.exitCode ? "\n有测试失败" : `\n全部通过（${passed} 项）`);

package/scripts/test-protocol.mjs

@@ -0,0 +1,94 @@
+#!/usr/bin/env node
+// 独立协议测试脚本：不经 Magpie，直接调用豆包接口验证配置与生图链路。
+//   node scripts/test-protocol.mjs --check                     # 检查各账号登录态
+//   node scripts/test-protocol.mjs "一只柴犬在樱花树下"         # 文生图
+//   node scripts/test-protocol.mjs "改成赛博朋克风格" --ref a.png --ref b.jpg
+//   可选：--ratio 16:9  --model "Seedream 4.5"  --channel web|pc  --out 目录  --account 主号
+import path from "node:path";
+import { loadConfig, configSource } from "../src/config.mjs";
+import { generate, checkLogin } from "../src/client.mjs";
+import { removeWatermarkByExtend } from "../src/watermark.mjs";
+import { downloadImage, fileStamp } from "../src/download.mjs";
+
+function parseArgv(argv) {
+  const opts = { refs: [], prompt: "" };
+  for (let i = 0; i < argv.length; i++) {
+    const a = argv[i];
+    if (a === "--check") opts.check = true;
+    else if (a === "--ratio") opts.ratio = argv[++i];
+    else if (a === "--model") opts.model = argv[++i];
+    else if (a === "--channel") opts.channel = argv[++i];
+    else if (a === "--out") opts.out = argv[++i];
+    else if (a === "--account") opts.account = argv[++i];
+    else if (a === "--ref") opts.refs.push(argv[++i]);
+    else if (a === "--wm") opts.wm = true;
+    else if (a === "--no-hint") opts.noHint = true;
+    else if (!a.startsWith("--")) opts.prompt += (opts.prompt ? " " : "") + a;
+  }
+  return opts;
+}
+
+const args = parseArgv(process.argv.slice(2));
+const cfg = loadConfig(args.noHint ? { ratioHint: false } : {});
+if (args.channel) cfg.channel = args.channel;
+
+console.log(`配置来源: ${configSource()}；账号 ${cfg.accounts.length} 个；通道 ${cfg.channel}；默认模型 ${cfg.model}`);
+
+if (!cfg.accounts.length) {
+  console.error(
+    "未配置账号。请先创建 config.json（复制 config.example.json 并填入 sessionid），\n" +
+      "或设置环境变量 DOUBAO_ACCOUNTS=\"主号:sessionid=xxx\"。",
+  );
+  process.exit(1);
+}
+
+if (args.check) {
+  const statuses = await checkLogin(cfg);
+  for (const s of statuses) {
+    console.log(
+      `${s.live ? "✅ 存活" : "❌ 失效"}  ${s.name}${s.userName ? ` (${s.userName})` : ""}` +
+        `${s.userId ? ` user_id=${s.userId}` : ""}${s.error ? ` 错误: ${s.error}` : ""}`,
+    );
+  }
+  process.exit(statuses.some((s) => s.live) ? 0 : 2);
+}
+
+if (!args.prompt) {
+  console.error('用法: node scripts/test-protocol.mjs "提示词" [--ratio 16:9] [--model ...] [--ref 文件] [--check]');
+  process.exit(1);
+}
+
+console.log(`开始生成（prompt="${args.prompt}", ratio=${args.ratio || cfg.ratio}, model=${args.model || cfg.model}${args.refs.length ? `, 参考图 ${args.refs.length} 张` : ""}）…`);
+const started = Date.now();
+try {
+  const result = await generate(cfg, {
+    prompt: args.prompt,
+    ratio: args.ratio,
+    model: args.model,
+    refImages: args.refs,
+    account: args.account,
+  });
+  console.log(`生成完成（${((Date.now() - started) / 1000).toFixed(1)}s，通道 ${result.channel}，账号 ${result.account}）`);
+  if (result.conversationId) console.log(`会话: ${result.conversationId}`);
+  if (result.text) console.log(`附言: ${result.text.slice(0, 200)}`);
+  const outDir = args.out || path.resolve(import.meta.dirname, "..", "output");
+  for (let i = 0; i < result.images.length; i++) {
+    const img = result.images[i];
+    const { file, bytes } = await downloadImage(img.url, outDir, `doubao-${fileStamp()}-${i + 1}`);
+    console.log(`图片 ${i + 1}: ${file} (${(bytes / 1024).toFixed(0)} KB, ${img.width || "?"}x${img.height || "?"})`);
+    if (args.wm && !args.refs.length) {
+      console.log("去水印：扩图+裁剪中…");
+      const r = await removeWatermarkByExtend(cfg, {
+        file,
+        prompt: args.prompt,
+        ratio: args.ratio,
+        account: args.account,
+      });
+      console.log(`  完成：${r.width}x${r.height}（扩图前 ${r.width}x${r.origHeight}，扩图比例 ${Math.round(r.extendFrac * 100)}%）`);
+    }
+  }
+} catch (err) {
+  console.error(`生成失败: ${err.message}`);
+  if (err.hint) console.error(`建议: ${err.hint}`);
+  process.exit(1);
+}

package/src/accounts.mjs

@@ -0,0 +1,58 @@
+// 多账号轮换与登录态检测。
+import { BASE_URL, baseHeaders, DoubaoError } from "./http.mjs";
+
+let cursor = 0;
+
+/**
+ * 生成尝试顺序：preferred（如指定）最优先，然后从轮换游标开始铺满全部账号。
+ * 同一进程内游标自增，多次调用自动分摊额度。
+ */
+export function accountOrder(accounts, preferred) {
+  if (!accounts.length) {
+    throw new DoubaoError(
+      "未配置豆包账号。",
+      "把 doubao.com 网页版 Cookie 里的 sessionid 写入插件目录 config.json（参考 config.example.json），或设置环境变量 DOUBAO_ACCOUNTS=\"主号:sessionid=xxx\"。",
+    );
+  }
+  const rest = accounts.slice();
+  const ordered = [];
+  if (preferred) {
+    const idx = rest.findIndex((a) => a.name === preferred || a.sessionid === preferred);
+    if (idx >= 0) ordered.push(...rest.splice(idx, 1));
+  }
+  for (let i = 0; i < rest.length; i++) {
+    ordered.push(rest[(cursor + i) % rest.length]);
+  }
+  cursor += 1;
+  return ordered;
+}
+
+/** 检测单个账号登录态：/passport/account/info/v2 返回 user_id 即存活。 */
+export async function checkAccount(account) {
+  try {
+    const resp = await fetch(`${BASE_URL}/passport/account/info/v2`, {
+      method: "POST",
+      headers: baseHeaders(account, { "Content-Type": "application/x-www-form-urlencoded" }),
+      body: "",
+      signal: AbortSignal.timeout(15_000),
+    });
+    const text = await resp.text();
+    let userId = "";
+    let userName = "";
+    try {
+      const body = JSON.parse(text);
+      const data = body.data || body;
+      userId = String(data.user_id ?? "");
+      userName = String(data.name ?? data.user_name ?? "");
+    } catch {
+      // 非 JSON 响应视为失效
+    }
+    return { name: account.name || "(未命名)", live: Boolean(userId), userId, userName };
+  } catch (err) {
+    return { name: account.name || "(未命名)", live: false, error: err.message };
+  }
+}
+
+export async function checkAllAccounts(accounts) {
+  return Promise.all(accounts.map(checkAccount));
+}