@dickpy/dsh-imagegen 1.0.6 → 1.0.9

package/lib/index.js

@@ -4,6 +4,7 @@ import { randomUUID } from "node:crypto";
 import { promises } from "node:fs";
 import { homedir } from "node:os";
 import path from "node:path";
+import { fileURLToPath } from "node:url";
 import { spawn } from "node:child_process";
 //#region src/protocol.ts
 /**
@@ -14,7 +15,7 @@ import { spawn } from "node:child_process";
 /** Settings namespace this plugin owns (host settings seam + bridge). */
 const IMAGEGEN_SETTINGS_NAMESPACE = "dsh-imagegen";
 /** Published package version shared by the host updater and the client UI. */
-const PLUGIN_VERSION = "1.0.6";
+const PLUGIN_VERSION = "1.0.9";
 /** Same-origin route family (loopback-only, mirroring the dsh-ssh fence). */
 const SETTINGS_API = {
 	describe: "/api/dsh-imagegen/settings/describe",
@@ -40,6 +41,17 @@ const HISTORY_API = {
 	clear: "/api/dsh-imagegen/history/clear",
 	image: "/api/dsh-imagegen/history/image"
 };
+/**
+* Same-origin route family for the bundled prompt-template library
+* (awesome-gpt-image-2 mirror). The case list ships inside the package and is
+* served by the host; reference images are proxied through the `image` prefix
+* route and cached on disk so repeated views never hit the network again.
+*/
+const TEMPLATES_API = {
+	list: "/api/dsh-imagegen/templates/list",
+	refresh: "/api/dsh-imagegen/templates/refresh",
+	image: "/api/dsh-imagegen/templates/image"
+};
 //#endregion
 //#region src/engine.ts
 /** A generation failure with a user-presentable message. */
@@ -285,7 +297,7 @@ function extensionOf(mime) {
 	}
 }
 /** MIME type for a stored image file name (image route responses). */
-function mimeOfFile(file) {
+function mimeOfFile$1(file) {
 	switch (path.extname(file).toLowerCase()) {
 		case ".jpg":
 		case ".jpeg": return "image/jpeg";
@@ -430,11 +442,280 @@ async function readHistoryImage(file) {
 	try {
 		return {
 			data: await promises.readFile(path.join(IMAGES_DIR, file)),
-			mime: mimeOfFile(file)
+			mime: mimeOfFile$1(file)
+		};
+	} catch {
+		return;
+	}
+}
+//#endregion
+//#region src/templates-store.ts
+/**
+* Prompt-template library store (awesome-gpt-image-2 mirror).
+*
+* The case list ships as a bundled snapshot (src/templates/cases.json, inside
+* the npm package) so the library works offline out of the box; a successful
+* manual refresh writes a runtime copy under ~/.dsh/dsh-imagegen/templates/
+* which then takes precedence. Reference images are not bundled (441 files,
+* ≈100 MB) — they are fetched from the vibeui.top mirror on demand, cached on
+* disk under ~/.dsh/dsh-imagegen/template-images/, and served from there on
+* every later view.
+*
+* Framework-free (node:fs only) so the route layer and tests can drive it
+* directly.
+*/
+/** Upstream mirror the library refreshes from (vibeui.top static mirror). */
+const SOURCE_URL = "https://vibeui.top/extra/awesome-gpt-image-2/data/cases.json";
+/** Remote image directory (file names come from the case list). */
+const IMAGE_BASE_URL = "https://vibeui.top/extra/awesome-gpt-image-2/data/images/";
+/** Category label map mirrored from vibeui.top's site.js (zh display names). */
+const CATEGORY_ZH = {
+	"Architecture & Spaces": "建筑与空间",
+	"Brand & Logos": "品牌与标志",
+	"Characters & People": "人物与角色",
+	"Charts & Infographics": "图表与信息可视化",
+	"Documents & Publishing": "文档与出版物",
+	"History & Classical Themes": "历史与古风题材",
+	"Illustration & Art": "插画与艺术",
+	"Other Use Cases": "其他应用场景",
+	"Photography & Realism": "摄影与写实",
+	"Posters & Typography": "海报与排版",
+	"Products & E-commerce": "商品与电商",
+	"Scenes & Storytelling": "场景与叙事",
+	"UI & Interfaces": "UI 与界面",
+	"Portraits & Fashion": "人像与时尚",
+	"Celebrities & Sports": "名人与运动",
+	"Characters & IP": "角色与 IP",
+	"Food & Beverage": "美食与饮品",
+	"Brand & Icons": "品牌与图标",
+	"Social Media & Stickers": "社媒与表情包",
+	"Infographics & Diagrams": "信息图与图解",
+	"UI & App Screens": "UI 与应用界面",
+	"Architecture & Interiors": "建筑与室内",
+	"Cinematic & Storytelling": "影视与叙事",
+	"Illustration & Comics": "插画与漫画",
+	"Historical & Fantasy": "历史与幻想",
+	"Animals & Nature": "动物与自然",
+	"Other Creative Uses": "其他创意用途"
+};
+const DATA_DIR = path.join(homedir(), ".dsh", "dsh-imagegen");
+const REFRESHED_CASES_PATH = path.join(DATA_DIR, "templates", "cases.json");
+const IMAGE_CACHE_DIR = path.join(DATA_DIR, "template-images");
+/**
+* Bundled snapshot path. The host bundle emits to lib/index.js while this
+* source file lives at src/templates-store.ts — both exactly one level below
+* the package root — so `../src/templates/cases.json` resolves to the shipped
+* snapshot in development and in the installed package alike.
+*/
+const BUNDLED_CASES_PATH = fileURLToPath(new URL("../src/templates/cases.json", import.meta.url));
+/** Budget for one upstream fetch (list refresh or one image). */
+const FETCH_TIMEOUT_MS = 6e4;
+/** Refuse to cache implausibly large "images". */
+const MAX_IMAGE_BYTES = 20 * 1024 * 1024;
+/** Strict reference-image file names this store writes and serves. */
+const IMAGE_FILE_PATTERN = /^case\d+\.(jpg|jpeg|png|webp|gif)$/i;
+/** In-memory memo of the active list (avoid re-parsing on every request). */
+let memo;
+/** Per-file in-flight downloads, so a gallery scroll never double-fetches. */
+const inflightImages = /* @__PURE__ */ new Map();
+/** Validate + normalize one raw upstream case; undefined when unusable. */
+function normalizeCase(raw) {
+	if (raw === null || typeof raw !== "object") return void 0;
+	const record = raw;
+	const id = Number(record.id);
+	const title = typeof record.title === "string" ? record.title.trim() : "";
+	const prompt = typeof record.prompt === "string" ? record.prompt.trim() : "";
+	if (!Number.isInteger(id) || title === "" || prompt === "") return void 0;
+	const category = typeof record.category === "string" ? record.category : "";
+	const image = imageFileOf(typeof record.image === "string" ? record.image : "");
+	return {
+		id,
+		title,
+		prompt,
+		category,
+		categoryZh: CATEGORY_ZH[category] ?? category,
+		styles: Array.isArray(record.styles) ? record.styles.map(String) : [],
+		scenes: Array.isArray(record.scenes) ? record.scenes.map(String) : [],
+		sourceLabel: typeof record.sourceLabel === "string" ? record.sourceLabel : "",
+		sourceUrl: typeof record.sourceUrl === "string" ? record.sourceUrl : "",
+		githubUrl: typeof record.githubUrl === "string" ? record.githubUrl : "",
+		image,
+		featured: record.featured === true
+	};
+}
+/** Extract the bare file name from an upstream image path. */
+function imageFileOf(value) {
+	const name = value.replace(/^\/+/, "").split("/").pop() ?? "";
+	return IMAGE_FILE_PATTERN.test(name) ? name : "";
+}
+/** Parse a snapshot payload (bundled, refreshed cache, or fresh download). */
+function parseSnapshot(payload) {
+	if (payload === null || typeof payload !== "object") return void 0;
+	const snapshot = payload;
+	if (!Array.isArray(snapshot.cases)) return void 0;
+	const cases = [];
+	for (const raw of snapshot.cases) {
+		const normalized = normalizeCase(raw);
+		if (normalized !== void 0) cases.push(normalized);
+	}
+	if (cases.length === 0) return void 0;
+	cases.sort((a, b) => b.id - a.id);
+	return {
+		cases,
+		repository: typeof snapshot.repository === "string" && snapshot.repository !== "" ? snapshot.repository : "freestylefly/awesome-gpt-image-2",
+		fetchedAt: typeof snapshot.fetchedAt === "string" && snapshot.fetchedAt !== "" ? snapshot.fetchedAt : ""
+	};
+}
+/** Read + parse a snapshot file; undefined when missing/corrupt. */
+async function readSnapshotFile(file) {
+	try {
+		return parseSnapshot(JSON.parse(await promises.readFile(file, "utf8")));
+	} catch {
+		return;
+	}
+}
+/**
+* The active template list: the refreshed runtime copy wins, the bundled
+* snapshot is the always-available fallback. Memoized; a successful refresh
+* replaces the memo.
+*/
+async function listTemplates() {
+	if (memo !== void 0) return memo;
+	const refreshed = await readSnapshotFile(REFRESHED_CASES_PATH);
+	if (refreshed !== void 0) {
+		memo = {
+			...refreshed,
+			total: refreshed.cases.length,
+			origin: "refreshed"
+		};
+		return memo;
+	}
+	const bundled = await readSnapshotFile(BUNDLED_CASES_PATH);
+	if (bundled !== void 0) {
+		memo = {
+			...bundled,
+			total: bundled.cases.length,
+			origin: "bundled"
 		};
+		return memo;
+	}
+	memo = {
+		cases: [],
+		total: 0,
+		origin: "bundled",
+		repository: "freestylefly/awesome-gpt-image-2",
+		fetchedAt: ""
+	};
+	return memo;
+}
+/**
+* Re-download the case list from the upstream mirror and persist it as the
+* runtime copy. Throws with a user-presentable message on failure; the
+* previous list (refreshed or bundled) stays active.
+*/
+async function refreshTemplates() {
+	let response;
+	try {
+		response = await fetch(SOURCE_URL, { signal: AbortSignal.timeout(FETCH_TIMEOUT_MS) });
+	} catch (error) {
+		throw new Error(`无法连接模板库源站：${error instanceof Error ? error.message : String(error)}`);
+	}
+	if (!response.ok) throw new Error(`模板库源站拒绝请求（HTTP ${response.status}）`);
+	let payload;
+	try {
+		payload = await response.json();
+	} catch {
+		throw new Error("模板库源站返回了非 JSON 响应");
+	}
+	const parsed = parseSnapshot(payload);
+	if (parsed === void 0) throw new Error("模板库源站数据格式无效");
+	const fetchedAt = (/* @__PURE__ */ new Date()).toISOString();
+	const snapshot = {
+		repository: parsed.repository,
+		sourceUrl: SOURCE_URL,
+		fetchedAt,
+		totalCases: parsed.cases.length,
+		cases: parsed.cases
+	};
+	await promises.mkdir(path.dirname(REFRESHED_CASES_PATH), { recursive: true });
+	const tmp = `${REFRESHED_CASES_PATH}.tmp-${process.pid}`;
+	await promises.writeFile(tmp, JSON.stringify(snapshot), "utf8");
+	await promises.rename(tmp, REFRESHED_CASES_PATH);
+	memo = {
+		cases: parsed.cases,
+		total: parsed.cases.length,
+		origin: "refreshed",
+		repository: parsed.repository,
+		fetchedAt
+	};
+	return {
+		total: parsed.cases.length,
+		fetchedAt
+	};
+}
+/** MIME type for a cached reference-image file name. */
+function mimeOfFile(file) {
+	switch (path.extname(file).toLowerCase()) {
+		case ".jpg":
+		case ".jpeg": return "image/jpeg";
+		case ".webp": return "image/webp";
+		case ".gif": return "image/gif";
+		default: return "image/png";
+	}
+}
+/** Download one reference image into the disk cache; undefined on failure. */
+async function fetchTemplateImage(file) {
+	let response;
+	try {
+		response = await fetch(`${IMAGE_BASE_URL}${encodeURIComponent(file)}`, { signal: AbortSignal.timeout(FETCH_TIMEOUT_MS) });
 	} catch {
 		return;
 	}
+	if (!response.ok) return void 0;
+	if (Number(response.headers.get("content-length") ?? 0) > MAX_IMAGE_BYTES) return void 0;
+	const data = Buffer.from(await response.arrayBuffer());
+	if (data.byteLength === 0 || data.byteLength > MAX_IMAGE_BYTES) return void 0;
+	const mime = mimeOfFile(file);
+	try {
+		await promises.mkdir(IMAGE_CACHE_DIR, { recursive: true });
+		const tmp = path.join(IMAGE_CACHE_DIR, `${file}.tmp-${process.pid}`);
+		await promises.writeFile(tmp, data);
+		await promises.rename(tmp, path.join(IMAGE_CACHE_DIR, file));
+	} catch {}
+	return {
+		data,
+		mime
+	};
+}
+/**
+* Read one reference image for the template library. Cache hit → disk; miss →
+* fetch from the upstream mirror, cache, and serve. Only file names present in
+* the active case list are served, so the route can never act as an open
+* proxy. Undefined when the name is unknown or the fetch failed.
+*/
+async function readTemplateImage(file) {
+	if (!IMAGE_FILE_PATTERN.test(file) || file.includes("..")) return void 0;
+	if (!(await listTemplates()).cases.some((entry) => entry.image === file)) return void 0;
+	const cached = path.join(IMAGE_CACHE_DIR, file);
+	try {
+		return {
+			data: await promises.readFile(cached),
+			mime: mimeOfFile(file)
+		};
+	} catch {}
+	const inflight = inflightImages.get(file);
+	if (inflight !== void 0) return inflight;
+	const pending = fetchTemplateImage(file);
+	inflightImages.set(file, pending);
+	try {
+		return await pending;
+	} finally {
+		inflightImages.delete(file);
+	}
+}
+/** Drop the in-memory list memo (tests). */
+function clearTemplateMemo() {
+	memo = void 0;
 }
 //#endregion
 //#region src/updater.ts
@@ -683,6 +964,11 @@ function makeRoutes(deps) {
 		clear: clearHistory,
 		readImage: readHistoryImage
 	};
+	const templates = deps.templates ?? {
+		list: listTemplates,
+		refresh: refreshTemplates,
+		readImage: readTemplateImage
+	};
 	const guard = (req, res, method) => {
 		if (!isLoopbackRequest(req)) {
 			writeJson(res, 403, { error: "forbidden: loopback-only" });
@@ -1028,6 +1314,74 @@ function makeRoutes(deps) {
 				});
 				res.end(found.data);
 			}
+		},
+		{
+			kind: "exact",
+			path: TEMPLATES_API.list,
+			handler: async (req, res) => {
+				if (!guard(req, res, "POST")) return;
+				try {
+					writeJson(res, 200, {
+						ok: true,
+						...await templates.list()
+					});
+				} catch (error) {
+					writeJson(res, 200, {
+						ok: false,
+						code: "templates-failed",
+						message: messageOf(error)
+					});
+				}
+			}
+		},
+		{
+			kind: "exact",
+			path: TEMPLATES_API.refresh,
+			handler: async (req, res) => {
+				if (!guard(req, res, "POST")) return;
+				try {
+					writeJson(res, 200, {
+						ok: true,
+						...await templates.refresh()
+					});
+				} catch (error) {
+					writeJson(res, 200, {
+						ok: false,
+						code: "templates-refresh-failed",
+						message: messageOf(error)
+					});
+				}
+			}
+		},
+		{
+			kind: "prefix",
+			path: TEMPLATES_API.image,
+			handler: async (req, res) => {
+				if (!isLoopbackRequest(req)) {
+					writeJson(res, 403, { error: "forbidden: loopback-only" });
+					return;
+				}
+				if (req.method !== "GET") {
+					writeJson(res, 405, { error: `method not allowed: ${req.method}` });
+					return;
+				}
+				const file = imageFileFrom(req.url, TEMPLATES_API.image);
+				if (file === void 0) {
+					writeJson(res, 404, { error: "not found" });
+					return;
+				}
+				const found = await templates.readImage(file);
+				if (found === void 0) {
+					writeJson(res, 404, { error: "not found" });
+					return;
+				}
+				res.writeHead(200, {
+					"content-type": found.mime,
+					"content-length": found.data.length,
+					"cache-control": "private, max-age=86400"
+				});
+				res.end(found.data);
+			}
 		}
 	];
 }
@@ -1051,7 +1405,7 @@ const DEFAULT_ANNOUNCE = true;
 /** Order of the announcement section within the tool-guidance band. */
 const SECTION_ORDER = 150;
 /** Model-facing announcement: plugin presence, capabilities, and limits. */
-const IMAGEGEN_GUIDANCE = "本机已安装 dsh-imagegen 插件（DSH AI 生图）：侧边栏「AI 生图」入口。能力：对接 OpenAI 兼容图像生成 API（模型 gpt-image-2），支持文生图（/images/generations）与图生图（/images/edits，上传参考图）；API 地址与密钥在 GUI「设置 → 插件 → 可配置」中配置，密钥仅存于本机设置文档；生成请求由本地宿主代理转发，结果以 base64 返回面板，可预览与下载。限制：生成消耗上游 API 额度；图片内容由上游模型生成，可能不符合预期或包含不适宜内容；api_key 以明文存储在设置文档中；参考图会发送至所配置的 API 服务。用户提到「生图 / 绘画 / 生成图片 / gpt-image-2 / 文生图 / 图生图」时即指本插件，请据此协作。";
+const IMAGEGEN_GUIDANCE = "本机已安装 dsh-imagegen 插件（DSH AI 生图）：侧边栏「AI 生图」入口。能力：对接 OpenAI 兼容图像生成 API（模型 gpt-image-2），支持文生图（/images/generations）与图生图（/images/edits，上传参考图）；API 地址与密钥在 GUI「设置 → 插件 → 可配置」中配置，密钥仅存于本机设置文档；生成请求由本地宿主代理转发，结果以 base64 返回面板，可预览与下载。内置「提示词模板库」（面板提示词框左下角「模板库」按钮）：打包 awesome-gpt-image-2 的数百条 gpt-image-2 提示词案例（含中文标题、分类、参考图，可搜索/按分类筛选），参考图经宿主代理按需缓存到本地；用户可一键把模板提示词填入提示词框再生成。限制：生成消耗上游 API 额度；图片内容由上游模型生成，可能不符合预期或包含不适宜内容；api_key 以明文存储在设置文档中；参考图会发送至所配置的 API 服务；模板库在线刷新与参考图首次加载需要访问 vibeui.top。用户提到「生图 / 绘画 / 生成图片 / gpt-image-2 / 文生图 / 图生图 / 提示词模板」时即指本插件，请据此协作。";
 /**
 * Mount the settings section, routes, and announcement.
 * @param ctx - host plugin context carrying webServer/systemPrompt.
@@ -1110,4 +1464,4 @@ function apply(ctx, config) {
 	sync();
 }
 //#endregion
-export { CURRENT_VERSION, Config, IMAGEGEN_GUIDANCE, ImageGenError, ImageGenSettingsNamespace, apply, checkForUpdate, clearUpdateCache, compareVersions, generateImage, inject, installUpdate, makeRoutes, name, profileFromProcess };
+export { CURRENT_VERSION, Config, IMAGEGEN_GUIDANCE, ImageGenError, ImageGenSettingsNamespace, apply, checkForUpdate, clearTemplateMemo, clearUpdateCache, compareVersions, generateImage, inject, installUpdate, listTemplates, makeRoutes, name, profileFromProcess, readTemplateImage, refreshTemplates };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@dickpy/dsh-imagegen",
   "description": "AI 生图 (image generation) plugin for the dsh web GUI: text-to-image and image-to-image through a configurable OpenAI-compatible endpoint (gpt-image-2 / gpt-image-1 / dall-e-3), with a settings card for api_url / api_key and a sidebar entry opening a split-pane generation studio.",
-  "version": "1.0.6",
+  "version": "1.0.9",
   "type": "module",
   "main": "lib/index.js",
   "exports": {
@@ -27,15 +27,15 @@
   },
   "devDependencies": {
     "@deepseek-ai/cordis": "^4.0.1",
-    "@deepseek-ai/dsh-client-connection": "^0.1.0-rc.6",
-    "@deepseek-ai/dsh-client-locale": "^0.1.0-rc.6",
-    "@deepseek-ai/dsh-client-runtime": "^0.1.0-rc.6",
-    "@deepseek-ai/dsh-client-ui-primitives": "^0.1.0-rc.6",
-    "@deepseek-ai/dsh-client-ui-settings": "^0.1.0-rc.6",
-    "@deepseek-ai/dsh-client-ui-slots": "^0.1.0-rc.6",
-    "@deepseek-ai/dsh-host-webserver": "^0.1.0-rc.6",
-    "@deepseek-ai/dsh-settings": "^0.1.0-rc.6",
-    "@deepseek-ai/dsh-system-prompt": "^0.1.0-rc.6",
+    "@deepseek-ai/dsh-client-connection": "0.1.0-rc.7",
+    "@deepseek-ai/dsh-client-locale": "0.1.0-rc.7",
+    "@deepseek-ai/dsh-client-runtime": "0.1.0-rc.7",
+    "@deepseek-ai/dsh-client-ui-primitives": "0.1.0-rc.7",
+    "@deepseek-ai/dsh-client-ui-settings": "0.1.0-rc.7",
+    "@deepseek-ai/dsh-client-ui-slots": "0.1.0-rc.7",
+    "@deepseek-ai/dsh-host-webserver": "0.1.0-rc.7",
+    "@deepseek-ai/dsh-settings": "0.1.0-rc.7",
+    "@deepseek-ai/dsh-system-prompt": "0.1.0-rc.7",
     "@types/node": "^22.20.0",
     "@types/react": "~18.3.1",
     "@types/react-dom": "^18.3.5",
@@ -62,6 +62,7 @@
   "scripts": {
     "build": "tsdown",
     "watch": "tsdown --watch",
-    "typecheck": "tsc --noEmit"
+    "typecheck": "tsc --noEmit",
+    "fetch-templates": "node scripts/fetch-templates.mjs"
   }
-}
+}

package/src/client/ImageGenPanel.tsx

@@ -13,6 +13,7 @@ import { createPortal } from 'react-dom'
 import { Button, Pill } from '@deepseek-ai/dsh-client-ui-primitives'
 import type { ImageGenApi } from './api.ts'
 import { errorMessage, tt } from './helpers.ts'
+import { TemplateLibrary } from './TemplateLibrary.tsx'
 import type { GeneratedImage, GenerateMode, GenerateRequest, HistoryEntry, HistoryImageRef, UpdateInfo } from '../protocol.ts'
 import type { ImageGenConfig, ImageGenScope } from './settings-scope.ts'
 import css from './panel.module.css'
@@ -146,6 +147,7 @@ export function ImageGenPanel(props: {
   const [updating, setUpdating] = useState(false)
   const [updateMessage, setUpdateMessage] = useState<string | null>(null)
   const [updateResult, setUpdateResult] = useState<'success' | 'failed' | null>(null)
+  const [libraryOpen, setLibraryOpen] = useState(false)
   const fileInput = useRef<HTMLInputElement>(null)
   const previewStage = useRef<HTMLDivElement>(null)
   const elapsed = useElapsed(generating, startedAt)
@@ -508,6 +510,15 @@ export function ImageGenPanel(props: {
                 onChange={(event) => { setPrompt(event.target.value) }}
               />
               <div className={css.promptFooter}>
+                <button
+                  type="button"
+                  className={css.templatesButton}
+                  title={tt('templates.title')}
+                  onClick={() => { setLibraryOpen(true) }}
+                >
+                  <svg viewBox="0 0 16 16" width="12" height="12" fill="none" stroke="currentColor" strokeWidth="1.4" strokeLinecap="round" strokeLinejoin="round" aria-hidden="true"><path d="M2.5 3.5h11M2.5 8h11M2.5 12.5h7"/></svg>
+                  {tt('templates.open')}
+                </button>
                 <span className={css.promptCount}>{tt('prompt.count', { count: prompt.length })}</span>
               </div>
             </section>
@@ -741,6 +752,20 @@ export function ImageGenPanel(props: {
         </aside>
       </div>
 
+      {/* ------------------------------------------------ template library */}
+      {libraryOpen ? (
+        <TemplateLibrary
+          api={api}
+          onClose={() => { setLibraryOpen(false) }}
+          onUse={(text) => {
+            setMode('text')
+            setPrompt(text)
+            setError(null)
+            setLibraryOpen(false)
+          }}
+        />
+      ) : null}
+
       {/* -------------------------------------------------- preview overlay */}
       {preview !== null && previewImage !== null
         ? createPortal(