dsh-plugin-image-workbench 1.1.5 → 1.1.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-plugin-image-workbench",
-  "version": "1.1.5",
+  "version": "1.1.6",
   "type": "module",
   "description": "Bluxe手绘工坊 · 生图工作台（图文 / 摄影）：DSH 侧栏图像生成工作台，默认使用内置本地账号池，亦支持任意 OpenAI 兼容接口（Base URL / API Key / 模型）；可自定义工作台名称与图片存储目录。",
   "main": "./host.js",
@@ -31,7 +31,8 @@
     "server/requirements.txt",
     "server/static",
     "vendor",
-    "scripts/vendor-snapshot.py"
+    "scripts/vendor-snapshot.py",
+    "scripts/fetch-style-images.mjs"
   ],
   "engines": {
     "node": ">=20"

package/scripts/fetch-style-images.mjs

@@ -0,0 +1,104 @@
+/**
+ * 样图下载器（风格 / 图型 / 主题色三类预览图）。
+ *
+ * 为什么要有这个文件：repo_images/ 里的 574 张样例图不进 npm 包（包体积会从
+ * 几百 KB 涨到十几 MB，而且图片本来就该放 CDN）。但工作台后端的 /img/{path}
+ * 在文件缺失时会返回一张渐变 SVG 占位图，于是全新安装打开「配色库 / 风格库」
+ * 全是色块 —— 看起来像装坏了。安装包里必须自带一次「把样图拉下来」的动作，
+ * 光靠 README 让用户手跑 `node install.mjs` 是不够的。
+ *
+ * 约定：
+ *   - 由 host.js 在插件启动时后台调用（不阻塞工作台启动），失败只写日志；
+ *   - install.mjs 也复用它，安装完顺手补齐，两处逻辑只有这一份；
+ *   - 幂等：已存在且非空的文件直接跳过，重跑只补缺失项；
+ *   - 只接受 image/* 响应，避免把 Gitee 的 302 登录页 HTML 写成 .webp 坏图。
+ */
+import fs from 'node:fs';
+import path from 'node:path';
+
+/** 样图源，按「CDN 单文件 → Gitee 镜像 → GitHub raw」顺序尝试。 */
+export const IMAGE_SOURCES = [
+  'https://img.handraw-style.com',
+  'https://gitee.com/yang02010/handraw-style/raw/master',
+  'https://raw.githubusercontent.com/yang0/handraw-style/main',
+];
+
+/** catalog.json 里 img 字段形如 images/individual/FA/FA-001.webp，落到 repo_images/ 时去掉 images/ 前缀。 */
+export function localPathOf(repoImages, rel) {
+  return path.join(repoImages, String(rel).replace(/^images\//, ''));
+}
+
+/** 从 catalog.json 收集需要下载的样图相对路径（去重）。 */
+export function collectRels(serverDir) {
+  const file = path.join(serverDir, 'catalog.json');
+  const rels = [];
+  const cat = JSON.parse(fs.readFileSync(file, 'utf8'));
+  for (const key of ['styles', 'layouts', 'colors']) {
+    for (const item of cat[key] || []) if (item && item.img) rels.push(item.img);
+  }
+  return [...new Set(rels)];
+}
+
+/**
+ * 下载缺失样图。
+ * @param {{serverDir:string, log?:Function, concurrency?:number, timeoutMs?:number, sources?:string[]}} opts
+ * @returns {Promise<{ok:boolean, total:number, done:number, skip:number, fail:number, error?:string}>}
+ */
+export async function fetchStyleImages(opts = {}) {
+  const serverDir = opts.serverDir;
+  const log = opts.log || (() => {});
+  const concurrency = Math.max(1, Math.min(24, opts.concurrency || 12));
+  const timeoutMs = opts.timeoutMs || 15000;
+  const sources = opts.sources || IMAGE_SOURCES;
+  if (!serverDir) return { ok: false, total: 0, done: 0, skip: 0, fail: 0, error: '缺少 serverDir' };
+
+  const repoImages = path.join(serverDir, 'repo_images');
+  let rels = [];
+  try {
+    rels = collectRels(serverDir);
+  } catch (e) {
+    return { ok: false, total: 0, done: 0, skip: 0, fail: 0,
+             error: '读取 catalog.json 失败：' + (e && e.message || e) };
+  }
+  if (!rels.length) return { ok: true, total: 0, done: 0, skip: 0, fail: 0 };
+
+  fs.mkdirSync(repoImages, { recursive: true });
+  const exists = p => { try { return fs.statSync(p).size > 0; } catch { return false; } };
+
+  const missing = rels.filter(rel => !exists(localPathOf(repoImages, rel)));
+  const skip = rels.length - missing.length;
+  if (!missing.length) return { ok: true, total: rels.length, done: 0, skip, fail: 0 };
+
+  log(`正在下载缺失样图 ${missing.length} 张（共 ${rels.length} 张，已存在 ${skip} 张）…`);
+  let done = 0, fail = 0;
+
+  const getOne = async rel => {
+    const dst = localPathOf(repoImages, rel);
+    fs.mkdirSync(path.dirname(dst), { recursive: true });
+    for (const base of sources) {
+      const url = base.replace(/\/$/, '') + '/' + String(rel).replace(/^\/+/, '');
+      try {
+        const ac = new AbortController();
+        const timer = setTimeout(() => ac.abort(), timeoutMs);
+        const res = await fetch(url, { signal: ac.signal, redirect: 'follow' });
+        clearTimeout(timer);
+        if (!res.ok) continue;
+        const type = (res.headers.get('content-type') || '').toLowerCase();
+        if (!type.startsWith('image/')) continue;
+        const buf = Buffer.from(await res.arrayBuffer());
+        if (buf.length < 200) continue;
+        fs.writeFileSync(dst, buf);
+        done += 1;
+        return;
+      } catch { /* 换下一个源 */ }
+    }
+    fail += 1;
+  };
+
+  for (let i = 0; i < missing.length; i += concurrency) {
+    await Promise.all(missing.slice(i, i + concurrency).map(getOne));
+  }
+
+  log(`样图下载结束：已下 ${done} 张，跳过 ${skip} 张，失败 ${fail} 张`);
+  return { ok: fail === 0, total: rels.length, done, skip, fail };
+}

package/scripts/vendor-snapshot.py

@@ -1,154 +1,154 @@
-"""把本机一份 ChatGPT2API 号池源码，做成插件自带的 vendor 快照。
-
-用途：工作台插件发布到别的机器时，不再要求对方先装任何第三方号池插件。
-这个脚本只在「升级号池源码」时手动跑一次，日常运行完全不依赖它。
-
-用法：
-    python vendor-snapshot.py [源号池 home 目录]
-
-默认源：~/.dsh/chatgpt2api-image
-输出：  <插件目录>/vendor/chatgpt2api/
-
-严格排除：账号库 data/、虚拟环境 .venv/、日志、备份、缓存、明文密钥与代理配置。
-config.json 会被清洗成 config.default.json（auth-key / proxy / 云存储凭据一律置空）。
-"""
-from __future__ import annotations
-
-import json
-import os
-import shutil
-import sys
-from pathlib import Path
-
-PLUGIN_DIR = Path(__file__).resolve().parent.parent
-OUT = PLUGIN_DIR / "vendor" / "chatgpt2api"
-
-# 只带运行需要的代码与静态资源，不带文档 / 测试 / 容器配置
-DIRS = ["api", "services", "utils", "web", "assets", "scripts"]
-FILES = ["main.py", "pyproject.toml", "uv.lock", ".python-version",
-         ".env.example", "LICENSE", "README.md", "VERSION"]
-
-EXCLUDE_NAMES = {"__pycache__", ".git", ".codex", ".pytest_cache", ".mypy_cache"}
-EXCLUDE_SUFFIX = (".pyc", ".pyo", ".log", ".db", ".db-shm", ".db-wal", ".enc")
-
-
-def keep(p: Path) -> bool:
-    n = p.name
-    if n in EXCLUDE_NAMES or n.startswith(".dsh-"):
-        return False
-    if n.endswith(EXCLUDE_SUFFIX) or ".bak" in n or n.endswith(".tmp"):
-        return False
-    return True
-
-
-def clean_config(raw: dict) -> dict:
-    """出厂默认配置：任何凭据、代理、本机状态都不带进插件包。"""
-    d = json.loads(json.dumps(raw))          # 深拷贝
-    d["auth-key"] = ""
-    d["proxy"] = ""
-    pr = d.get("proxy_runtime") or {}
-    pr["enabled"] = False
-    pr["proxy_url"] = ""
-    pr["resource_proxy_url"] = ""
-    cl = pr.get("clearance") or {}
-    cl["cf_cookies"] = ""
-    cl["cf_clearance"] = ""
-    d["backup"] = {**(d.get("backup") or {}),
-                   "enabled": False, "access_key_id": "",
-                   "secret_access_key": "", "passphrase": ""}
-    d["ai_review"] = {**(d.get("ai_review") or {}),
-                      "enabled": False, "base_url": "", "api_key": ""}
-    img = d.get("image_storage") or {}
-    img.update({"enabled": False, "webdav_username": "", "webdav_password": ""})
-    return d
-
-
-def main() -> int:
-    home = Path(sys.argv[1]) if len(sys.argv) > 1 else (
-        Path(os.environ.get("DSH_HOME", Path.home() / ".dsh")) / "chatgpt2api-image")
-    src = home / "source"
-    if not (src / "main.py").is_file():
-        print("找不到号池源码：" + str(src))
-        return 1
-    if OUT.resolve() == src.resolve():
-        print("源目录不能是 vendor 输出目录")
-        return 1
-
-    if OUT.exists():
-        shutil.rmtree(OUT)
-    (OUT / "source").mkdir(parents=True)
-    (OUT / "runtime").mkdir(parents=True)
-
-    copied = 0
-    for d in DIRS:
-        s = src / d
-        if not s.is_dir():
-            continue
-        shutil.copytree(s, OUT / "source" / d, ignore=lambda _, names: [
-            n for n in names if not keep(Path(n))], dirs_exist_ok=True)
-        copied += sum(1 for _ in (OUT / "source" / d).rglob("*") if _.is_file())
-    for f in FILES:
-        s = src / f
-        if s.is_file() and keep(s):
-            shutil.copy2(s, OUT / "source" / f)
-            copied += 1
-
-    cfg = src / "config.json"
-    if cfg.is_file():
-        try:
-            data = clean_config(json.loads(cfg.read_text(encoding="utf-8")))
-            (OUT / "source" / "config.default.json").write_text(
-                json.dumps(data, ensure_ascii=False, indent=2) + "\n", encoding="utf-8")
-            copied += 1
-        except Exception as e:
-            print("警告：config.json 清洗失败，未生成默认配置：" + str(e))
-
-    uv = ""
-    for c in (home / "runtime" / "uv.exe", home / "runtime" / "uv",
-              shutil.which("uv") or ""):
-        if c and Path(c).is_file():
-            uv = str(c)
-            break
-    if uv:
-        shutil.copy2(uv, OUT / "runtime" / Path(uv).name)
-        copied += 1
-    else:
-        print("警告：没找到 uv.exe，全新机器初始化时会退回系统 Python + pip")
-
-    note = """# 内置号池运行时
-
-本目录是 ChatGPT2API 号池服务端的**代码快照**，随「Bluxe手绘工坊 · 生图工作台」插件一起分发，
-安装后由工作台后端（pool_service.py）释放到本机数据目录并自建虚拟环境。
-
-* 后端上游：[basketikun/chatgpt2api](https://github.com/basketikun/chatgpt2api)，`source/VERSION` 与 `SOURCE.json` 记录快照版本；DSH 插件封装：[yang0/chatgpt2api-dsh](https://github.com/yang0/chatgpt2api-dsh)。
-* 许可：MIT License，Copyright (c) 2026 kunkun，LICENSE 原文见 `source/LICENSE`。本目录仅快照后端源码，不含 DSH 插件外壳与任何账号数据。
-* 已包含本工作台需要的两处补丁：round-robin 游标持久化、限流号排除；首次初始化时还会幂等复打一次。
-* **不含**任何账号、token、密钥、代理配置或图片数据。`source/config.default.json` 是清洗后的出厂默认配置。
-* 升级方式：改完本机号池源码后，重跑 `scripts/vendor-snapshot.py`，再重新发布插件。
-
-分工：工作台插件负责「拉起并守护」号池进程；本目录只提供「能跑起来的代码」。
-两者都不需要安装任何第三方 DSH 插件。
-"""
-    (OUT / "README.md").write_text(note, encoding="utf-8")
-    (OUT / "SOURCE.json").write_text(json.dumps({
-        "name": "chatgpt2api", "license": "MIT",
-        "version": (src / "VERSION").read_text(encoding="utf-8").strip()
-        if (src / "VERSION").is_file() else "",
-        "python": (src / ".python-version").read_text(encoding="utf-8").strip()
-        if (src / ".python-version").is_file() else "",
-        "files": copied,
-        "version_source": "source/VERSION",
-        "copyright": "Copyright (c) 2026 kunkun",
-        "upstream_backend": "https://github.com/basketikun/chatgpt2api",
-        "upstream_dsh_plugin": "https://github.com/yang0/chatgpt2api-dsh",
-    }, ensure_ascii=False, indent=2) + "\n", encoding="utf-8")
-
-    size = sum(f.stat().st_size for f in OUT.rglob("*") if f.is_file())
-    print("vendor 已生成：" + str(OUT))
-    print("文件数 %d，体积 %.1f MB" % (copied, size / 1024 / 1024))
-    print("含 uv 运行时：" + ("是" if uv else "否"))
-    return 0
-
-
-if __name__ == "__main__":
-    raise SystemExit(main())
+"""把本机一份 ChatGPT2API 号池源码，做成插件自带的 vendor 快照。
+
+用途：工作台插件发布到别的机器时，不再要求对方先装任何第三方号池插件。
+这个脚本只在「升级号池源码」时手动跑一次，日常运行完全不依赖它。
+
+用法：
+    python vendor-snapshot.py [源号池 home 目录]
+
+默认源：~/.dsh/chatgpt2api-image
+输出：  <插件目录>/vendor/chatgpt2api/
+
+严格排除：账号库 data/、虚拟环境 .venv/、日志、备份、缓存、明文密钥与代理配置。
+config.json 会被清洗成 config.default.json（auth-key / proxy / 云存储凭据一律置空）。
+"""
+from __future__ import annotations
+
+import json
+import os
+import shutil
+import sys
+from pathlib import Path
+
+PLUGIN_DIR = Path(__file__).resolve().parent.parent
+OUT = PLUGIN_DIR / "vendor" / "chatgpt2api"
+
+# 只带运行需要的代码与静态资源，不带文档 / 测试 / 容器配置
+DIRS = ["api", "services", "utils", "web", "assets", "scripts"]
+FILES = ["main.py", "pyproject.toml", "uv.lock", ".python-version",
+         ".env.example", "LICENSE", "README.md", "VERSION"]
+
+EXCLUDE_NAMES = {"__pycache__", ".git", ".codex", ".pytest_cache", ".mypy_cache"}
+EXCLUDE_SUFFIX = (".pyc", ".pyo", ".log", ".db", ".db-shm", ".db-wal", ".enc")
+
+
+def keep(p: Path) -> bool:
+    n = p.name
+    if n in EXCLUDE_NAMES or n.startswith(".dsh-"):
+        return False
+    if n.endswith(EXCLUDE_SUFFIX) or ".bak" in n or n.endswith(".tmp"):
+        return False
+    return True
+
+
+def clean_config(raw: dict) -> dict:
+    """出厂默认配置：任何凭据、代理、本机状态都不带进插件包。"""
+    d = json.loads(json.dumps(raw))          # 深拷贝
+    d["auth-key"] = ""
+    d["proxy"] = ""
+    pr = d.get("proxy_runtime") or {}
+    pr["enabled"] = False
+    pr["proxy_url"] = ""
+    pr["resource_proxy_url"] = ""
+    cl = pr.get("clearance") or {}
+    cl["cf_cookies"] = ""
+    cl["cf_clearance"] = ""
+    d["backup"] = {**(d.get("backup") or {}),
+                   "enabled": False, "access_key_id": "",
+                   "secret_access_key": "", "passphrase": ""}
+    d["ai_review"] = {**(d.get("ai_review") or {}),
+                      "enabled": False, "base_url": "", "api_key": ""}
+    img = d.get("image_storage") or {}
+    img.update({"enabled": False, "webdav_username": "", "webdav_password": ""})
+    return d
+
+
+def main() -> int:
+    home = Path(sys.argv[1]) if len(sys.argv) > 1 else (
+        Path(os.environ.get("DSH_HOME", Path.home() / ".dsh")) / "chatgpt2api-image")
+    src = home / "source"
+    if not (src / "main.py").is_file():
+        print("找不到号池源码：" + str(src))
+        return 1
+    if OUT.resolve() == src.resolve():
+        print("源目录不能是 vendor 输出目录")
+        return 1
+
+    if OUT.exists():
+        shutil.rmtree(OUT)
+    (OUT / "source").mkdir(parents=True)
+    (OUT / "runtime").mkdir(parents=True)
+
+    copied = 0
+    for d in DIRS:
+        s = src / d
+        if not s.is_dir():
+            continue
+        shutil.copytree(s, OUT / "source" / d, ignore=lambda _, names: [
+            n for n in names if not keep(Path(n))], dirs_exist_ok=True)
+        copied += sum(1 for _ in (OUT / "source" / d).rglob("*") if _.is_file())
+    for f in FILES:
+        s = src / f
+        if s.is_file() and keep(s):
+            shutil.copy2(s, OUT / "source" / f)
+            copied += 1
+
+    cfg = src / "config.json"
+    if cfg.is_file():
+        try:
+            data = clean_config(json.loads(cfg.read_text(encoding="utf-8")))
+            (OUT / "source" / "config.default.json").write_text(
+                json.dumps(data, ensure_ascii=False, indent=2) + "\n", encoding="utf-8")
+            copied += 1
+        except Exception as e:
+            print("警告：config.json 清洗失败，未生成默认配置：" + str(e))
+
+    uv = ""
+    for c in (home / "runtime" / "uv.exe", home / "runtime" / "uv",
+              shutil.which("uv") or ""):
+        if c and Path(c).is_file():
+            uv = str(c)
+            break
+    if uv:
+        shutil.copy2(uv, OUT / "runtime" / Path(uv).name)
+        copied += 1
+    else:
+        print("警告：没找到 uv.exe，全新机器初始化时会退回系统 Python + pip")
+
+    note = """# 内置号池运行时
+
+本目录是 ChatGPT2API 号池服务端的**代码快照**，随「Bluxe手绘工坊 · 生图工作台」插件一起分发，
+安装后由工作台后端（pool_service.py）释放到本机数据目录并自建虚拟环境。
+
+* 后端上游：[basketikun/chatgpt2api](https://github.com/basketikun/chatgpt2api)，`source/VERSION` 与 `SOURCE.json` 记录快照版本；DSH 插件封装：[yang0/chatgpt2api-dsh](https://github.com/yang0/chatgpt2api-dsh)。
+* 许可：MIT License，Copyright (c) 2026 kunkun，LICENSE 原文见 `source/LICENSE`。本目录仅快照后端源码，不含 DSH 插件外壳与任何账号数据。
+* 已包含本工作台需要的两处补丁：round-robin 游标持久化、限流号排除；首次初始化时还会幂等复打一次。
+* **不含**任何账号、token、密钥、代理配置或图片数据。`source/config.default.json` 是清洗后的出厂默认配置。
+* 升级方式：改完本机号池源码后，重跑 `scripts/vendor-snapshot.py`，再重新发布插件。
+
+分工：工作台插件负责「拉起并守护」号池进程；本目录只提供「能跑起来的代码」。
+两者都不需要安装任何第三方 DSH 插件。
+"""
+    (OUT / "README.md").write_text(note, encoding="utf-8")
+    (OUT / "SOURCE.json").write_text(json.dumps({
+        "name": "chatgpt2api", "license": "MIT",
+        "version": (src / "VERSION").read_text(encoding="utf-8").strip()
+        if (src / "VERSION").is_file() else "",
+        "python": (src / ".python-version").read_text(encoding="utf-8").strip()
+        if (src / ".python-version").is_file() else "",
+        "files": copied,
+        "version_source": "source/VERSION",
+        "copyright": "Copyright (c) 2026 kunkun",
+        "upstream_backend": "https://github.com/basketikun/chatgpt2api",
+        "upstream_dsh_plugin": "https://github.com/yang0/chatgpt2api-dsh",
+    }, ensure_ascii=False, indent=2) + "\n", encoding="utf-8")
+
+    size = sum(f.stat().st_size for f in OUT.rglob("*") if f.is_file())
+    print("vendor 已生成：" + str(OUT))
+    print("文件数 %d，体积 %.1f MB" % (copied, size / 1024 / 1024))
+    print("含 uv 运行时：" + ("是" if uv else "否"))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

package/server/access.py

@@ -1,26 +1,26 @@
-"""本地版访问控制：单机 127.0.0.1 单人使用，不需要验证码审批。
-
-保留与 VPS 版相同的接口签名，让 app.py 一行都不用改就能加载。
-VPS 原件已备份为 access.py.vps_original（其顶层 import fcntl 在 Windows 不存在）。
-"""
-from __future__ import annotations
-
-SESSION_TTL = 7 * 86400
-LOCAL_MODE = True
-
-
-def check_session(token: str) -> bool:
-    """本地版没有会话概念，永远放行。"""
-    return True
-
-
-def request_code(email: str, ip: str, ua: str = "") -> dict:
-    return {"ok": True, "msg": "本地版无需申请，请直接返回工作台"}
-
-
-def verify_code(email: str, code: str, ip: str) -> dict:
-    return {"ok": True, "msg": "本地版无需验证码", "token": "local"}
-
-
-def master_login(password: str, ip: str) -> dict:
-    return {"ok": True, "msg": "本地版无需口令", "token": "local"}
+"""本地版访问控制：单机 127.0.0.1 单人使用，不需要验证码审批。
+
+保留与 VPS 版相同的接口签名，让 app.py 一行都不用改就能加载。
+VPS 原件已备份为 access.py.vps_original（其顶层 import fcntl 在 Windows 不存在）。
+"""
+from __future__ import annotations
+
+SESSION_TTL = 7 * 86400
+LOCAL_MODE = True
+
+
+def check_session(token: str) -> bool:
+    """本地版没有会话概念，永远放行。"""
+    return True
+
+
+def request_code(email: str, ip: str, ua: str = "") -> dict:
+    return {"ok": True, "msg": "本地版无需申请，请直接返回工作台"}
+
+
+def verify_code(email: str, code: str, ip: str) -> dict:
+    return {"ok": True, "msg": "本地版无需验证码", "token": "local"}
+
+
+def master_login(password: str, ip: str) -> dict:
+    return {"ok": True, "msg": "本地版无需口令", "token": "local"}