dsh-composer-ux 0.2.0 → 0.3.0

package/lib/index.js

@@ -807,7 +807,15 @@ var MENU_FIELDS = [
 var MENU_NATIVE_FIELD = "menuNative";
 var QUICK_PROMPTS_FIELD = "quickPrompts";
 var OPTIMIZER_TIER_FIELD = "optimizerTier";
-var QUICK_TEXT_MAX = 4e3;
+var QUICK_PROMPT_MAX = 60;
+var QUICK_CATEGORY_MAX = 20;
+var QUICK_CATEGORY_NAME_MAX = 40;
+var QUICK_LABEL_MAX = 40;
+var QUICK_TEXT_MAX = 2e5;
+var OPTIMIZE_TEXT_MAX = 8e3;
+var QUICK_PROMPTS_API_PATH = "/composer-ux/prompts";
+var QUICK_BOOK_VERSION = 2;
+var DEFAULT_CATEGORY_NAME = "\u9ED8\u8BA4";
 var OPTIMIZE_OUTPUT_MAX = 12e3;
 var OPTIMIZER_API_PATH = "/composer-ux/optimize";
 var DEFAULT_OPTIMIZER_TIER = "advanced";
@@ -822,6 +830,99 @@ var DEFAULT_QUICK_PROMPTS = [
   { id: "builtin-8", label: "\u5199\u6D4B\u8BD5", prompt: "\u8BF7\u4E3A\u4E0B\u9762\u7684\u4EE3\u7801\u7F16\u5199\u5355\u5143\u6D4B\u8BD5\u3002", always: false },
   { id: "builtin-9", label: "\u4EE3\u7801\u5BA1\u67E5", prompt: "\u8BF7\u5BF9\u4E0B\u9762\u7684\u4EE3\u7801\u8FDB\u884C\u4EE3\u7801\u5BA1\u67E5\uFF0C\u6307\u51FA\u95EE\u9898\u5E76\u7ED9\u51FA\u6539\u8FDB\u5EFA\u8BAE\u3002", always: false }
 ];
+function newQuickPromptId() {
+  return `qp-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
+}
+function newQuickCategoryId() {
+  return `qc-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
+}
+function defaultQuickBook() {
+  return {
+    version: QUICK_BOOK_VERSION,
+    categories: [{ id: "cat-default", name: DEFAULT_CATEGORY_NAME, prompts: DEFAULT_QUICK_PROMPTS }]
+  };
+}
+function asPromptRow(value) {
+  if (typeof value !== "object" || value === null) return void 0;
+  const row = value;
+  const either = (first, second) => {
+    const picked = typeof first === "string" && first !== "" ? first : second;
+    return typeof picked === "string" ? picked : "";
+  };
+  const prompt = either(row.prompt, row.text).slice(0, QUICK_TEXT_MAX).trim();
+  if (prompt === "") return void 0;
+  const label = either(row.label, row.title).slice(0, QUICK_LABEL_MAX).trim();
+  const id = typeof row.id === "string" && row.id !== "" ? row.id.slice(0, 64) : newQuickPromptId();
+  const order = typeof row.order === "number" && Number.isFinite(row.order) ? Math.max(1, Math.round(row.order)) : void 0;
+  return {
+    prompt: { id, label: label === "" ? prompt.slice(0, 12) : label, prompt, always: row.always === true || row.autoSend === true },
+    order
+  };
+}
+function toPromptList(values) {
+  const rows = [];
+  for (const value of values) {
+    const row = asPromptRow(value);
+    if (row !== void 0) rows.push({ ...row, index: rows.length });
+  }
+  rows.sort((a, b) => {
+    const left = a.order ?? Number.MAX_SAFE_INTEGER;
+    const right = b.order ?? Number.MAX_SAFE_INTEGER;
+    return left === right ? a.index - b.index : left - right;
+  });
+  const seen = /* @__PURE__ */ new Set();
+  const out = [];
+  for (const row of rows) {
+    if (out.length >= QUICK_PROMPT_MAX) break;
+    if (seen.has(row.prompt.id)) continue;
+    seen.add(row.prompt.id);
+    out.push(row.prompt);
+  }
+  return out;
+}
+function sanitizeBook(value) {
+  if (typeof value !== "object" || value === null) return void 0;
+  const source = value;
+  const flat = Array.isArray(value) ? value : Array.isArray(source.prompts) ? source.prompts : void 0;
+  if (flat !== void 0) {
+    const prompts = toPromptList(flat);
+    return {
+      version: QUICK_BOOK_VERSION,
+      categories: prompts.length === 0 ? [] : [{ id: "cat-default", name: DEFAULT_CATEGORY_NAME, prompts }]
+    };
+  }
+  if (!Array.isArray(source.categories)) return void 0;
+  const seen = /* @__PURE__ */ new Set();
+  const categories = [];
+  for (const raw of source.categories) {
+    if (categories.length >= QUICK_CATEGORY_MAX) break;
+    if (typeof raw !== "object" || raw === null) continue;
+    const row = raw;
+    const prompts = toPromptList(Array.isArray(row.prompts) ? row.prompts : []);
+    const wanted = typeof row.id === "string" && row.id !== "" ? row.id.slice(0, 64) : newQuickCategoryId();
+    const id = seen.has(wanted) ? newQuickCategoryId() : wanted;
+    seen.add(id);
+    const name2 = (typeof row.name === "string" ? row.name : "").slice(0, QUICK_CATEGORY_NAME_MAX).trim();
+    categories.push({ id, name: name2 === "" ? DEFAULT_CATEGORY_NAME : name2, prompts });
+  }
+  return { version: QUICK_BOOK_VERSION, categories };
+}
+function bookToFile(book) {
+  return {
+    version: QUICK_BOOK_VERSION,
+    categories: book.categories.map((category) => ({
+      id: category.id,
+      name: category.name,
+      prompts: category.prompts.map((prompt, index) => ({
+        id: prompt.id,
+        title: prompt.label,
+        text: prompt.prompt,
+        autoSend: prompt.always,
+        order: index + 1
+      }))
+    }))
+  };
+}
 var HEADER_ENABLED_FIELD = "headerEnabled";
 var HEADER_NAME_FIELD = "headerName";
 var HEADER_VALUE_FIELD = "headerValue";
@@ -967,6 +1068,141 @@ function buildOptimizeUser(original) {
   ].join("\n");
 }
 
+// src/quick-store.ts
+import { mkdir, open, readFile, rename, unlink } from "node:fs/promises";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
+var QUICK_STORE_FILE = "quick-prompts.json";
+var LOCK_RETRY_MS = 50;
+var LOCK_RETRY_MAX = 40;
+var LOCK_STALE_MS = 3e4;
+function dshHome() {
+  const fromEnv = process.env.DSH_HOME?.trim();
+  return fromEnv !== void 0 && fromEnv !== "" ? fromEnv : join(homedir(), ".dsh");
+}
+function quickStorePath() {
+  return join(dshHome(), QUICK_STORE_FILE);
+}
+function errorText(error) {
+  return error instanceof Error ? error.message : String(error);
+}
+function hasCode(error, code) {
+  return typeof error === "object" && error !== null && error.code === code;
+}
+function delay(ms) {
+  return new Promise((resolve2) => {
+    setTimeout(resolve2, ms);
+  });
+}
+async function quarantine(file) {
+  const stamp = (/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-");
+  const target = `${file}.bad-${stamp}`;
+  try {
+    await rename(file, target);
+    return target;
+  } catch {
+    return void 0;
+  }
+}
+async function readQuickBook(file = quickStorePath()) {
+  let raw;
+  try {
+    raw = await readFile(file, "utf8");
+  } catch (error) {
+    if (hasCode(error, "ENOENT")) return { kind: "missing" };
+    return { kind: "broken", error: `\u8BFB\u53D6\u5931\u8D25\uFF1A${errorText(error)}` };
+  }
+  if (raw.trim() === "") return { kind: "broken", error: "\u6587\u4EF6\u662F\u7A7A\u7684", quarantined: await quarantine(file) };
+  let parsed;
+  try {
+    parsed = JSON.parse(raw);
+  } catch (error) {
+    return { kind: "broken", error: `\u4E0D\u662F\u5408\u6CD5 JSON\uFF1A${errorText(error)}`, quarantined: await quarantine(file) };
+  }
+  const book = sanitizeBook(parsed);
+  if (book === void 0) {
+    return {
+      kind: "broken",
+      error: "\u7ED3\u6784\u8BA4\u4E0D\u51FA\uFF08\u65E2\u4E0D\u662F categories \u4E5F\u4E0D\u662F\u5E73\u94FA prompts\uFF09",
+      quarantined: await quarantine(file)
+    };
+  }
+  return { kind: "ok", book };
+}
+async function acquireLock(file) {
+  const lock = `${file}.lock`;
+  for (let attempt = 0; attempt < LOCK_RETRY_MAX; attempt += 1) {
+    try {
+      const handle = await open(lock, "wx", 384);
+      await handle.close();
+      return async () => {
+        try {
+          await unlink(lock);
+        } catch {
+        }
+      };
+    } catch (error) {
+      if (!hasCode(error, "EEXIST")) throw error;
+      let ageMs;
+      try {
+        const info = await open(lock, "r");
+        try {
+          const stat = await info.stat();
+          ageMs = Date.now() - stat.mtimeMs;
+        } finally {
+          await info.close();
+        }
+      } catch {
+      }
+      if (ageMs !== void 0 && ageMs > LOCK_STALE_MS) {
+        throw new Error(
+          `\u5199\u5165\u9501 ${lock} \u5DF2\u5B58\u5728 ${Math.round(ageMs / 1e3)} \u79D2\uFF0C\u50CF\u662F\u4E0A\u6B21\u5199\u5165\u5D29\u6E83\u7559\u4E0B\u7684\uFF1B\u786E\u8BA4\u6CA1\u6709\u6B63\u5728\u8FD0\u884C\u7684 DSH \u540E\u5220\u9664\u5B83\u518D\u91CD\u8BD5`
+        );
+      }
+      await delay(LOCK_RETRY_MS);
+    }
+  }
+  throw new Error(`\u53E6\u4E00\u4E2A\u8FDB\u7A0B\u6B63\u5728\u5199\u5165 ${file}\uFF08\u7B49\u5F85 ${String(LOCK_RETRY_MS * LOCK_RETRY_MAX / 1e3)} \u79D2\u4ECD\u672A\u62FF\u5230\u9501\uFF09`);
+}
+async function writeQuickBook(book, file = quickStorePath()) {
+  await mkdir(dirname(file), { recursive: true });
+  const release = await acquireLock(file);
+  try {
+    await writeLocked(book, file);
+  } finally {
+    await release();
+  }
+}
+async function writeLocked(book, file) {
+  const safe = sanitizeBook(book) ?? { version: QUICK_BOOK_VERSION, categories: [] };
+  const tmp = `${file}.${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}.tmp`;
+  const handle = await open(tmp, "w", 384);
+  try {
+    await handle.writeFile(`${JSON.stringify(bookToFile(safe), null, 2)}
+`, "utf8");
+    await handle.sync();
+  } finally {
+    await handle.close();
+  }
+  await rename(tmp, file);
+}
+async function ensureQuickBook(seed, file = quickStorePath()) {
+  const current = await readQuickBook(file);
+  if (current.kind !== "missing") return current;
+  await mkdir(dirname(file), { recursive: true });
+  const release = await acquireLock(file);
+  try {
+    const again = await readQuickBook(file);
+    if (again.kind !== "missing") return again;
+    const migrated = seed === void 0 ? void 0 : sanitizeBook({ prompts: seed });
+    const book = migrated !== void 0 && migrated.categories.length > 0 ? migrated : defaultQuickBook();
+    await writeLocked(book, file);
+    return { kind: "ok", book };
+  } finally {
+    await release();
+  }
+}
+
 // src/host.ts
 var name = "composer-ux";
 var PANEL_MIN = 560;
@@ -1013,7 +1249,7 @@ function pickRoutes(providers, listed) {
   }
   return listed.filter((key) => Object.prototype.hasOwnProperty.call(providers, key));
 }
-function errorText(error) {
+function errorText2(error) {
   if (error instanceof Error && error.message !== "") return error.message;
   return String(error);
 }
@@ -1078,7 +1314,7 @@ async function mirrorHeader(settings) {
       await settings.mutate(LLM_NAMESPACE, providerOps);
     } catch (error) {
       failed = true;
-      status = `\u5199\u5165\u5931\u8D25\uFF1A${errorText(error)}`;
+      status = `\u5199\u5165\u5931\u8D25\uFF1A${errorText2(error)}`;
     }
   }
   const nextAppliedName = !failed && applied.length > 0 ? name2 : "";
@@ -1160,7 +1396,7 @@ function apply(ctx) {
   ctx.inject(["webServer", "llm"], (optCtx) => {
     const LLM_TIMEOUT_MS = 18e4;
     const BODY_MAX_BYTES = 1e6;
-    const TEXT_MAX = QUICK_TEXT_MAX * 2;
+    const TEXT_MAX = OPTIMIZE_TEXT_MAX;
     const sendJson = (res, code, payload) => {
       res.writeHead(code, { "content-type": "application/json; charset=utf-8" });
       res.end(JSON.stringify(payload));
@@ -1197,7 +1433,7 @@ function apply(ctx) {
       try {
         payload = objectOf(JSON.parse(await readBody(req))) ?? {};
       } catch (error) {
-        sendJson(res, 400, { ok: false, error: `\u8BF7\u6C42\u4F53\u4E0D\u662F\u5408\u6CD5 JSON\uFF1A${errorText(error)}` });
+        sendJson(res, 400, { ok: false, error: `\u8BF7\u6C42\u4F53\u4E0D\u662F\u5408\u6CD5 JSON\uFF1A${errorText2(error)}` });
         return;
       }
       const text = textOf(payload.text).trim();
@@ -1248,7 +1484,7 @@ function apply(ctx) {
           }
         }
       } catch (error) {
-        failure = errorText(error);
+        failure = errorText2(error);
       } finally {
         clearTimeout(timer);
       }
@@ -1274,6 +1510,85 @@ function apply(ctx) {
       handler: handle
     }), "composer-ux: prompt optimizer route");
   });
+  ctx.inject(["webServer"], (storeCtx) => {
+    const BODY_MAX_BYTES = 8e6;
+    const sendJson = (res, code, payload) => {
+      res.writeHead(code, { "content-type": "application/json; charset=utf-8" });
+      res.end(JSON.stringify(payload));
+    };
+    const readBody = async (req) => {
+      const chunks = [];
+      let total = 0;
+      for await (const chunk of req) {
+        const buf = Buffer.isBuffer(chunk) ? chunk : Buffer.from(String(chunk));
+        total += buf.length;
+        if (total > BODY_MAX_BYTES) throw new Error("\u8BF7\u6C42\u4F53\u8FC7\u5927");
+        chunks.push(buf);
+      }
+      return Buffer.concat(chunks).toString("utf8");
+    };
+    const legacyPrompts = () => {
+      try {
+        const settings = storeCtx.get("settings");
+        const row = objectOf(settings?.get(NAMESPACE));
+        const list = row?.[QUICK_PROMPTS_FIELD];
+        return Array.isArray(list) ? list : void 0;
+      } catch {
+        return void 0;
+      }
+    };
+    const handle = async (req, res) => {
+      const method = (req.method ?? "GET").toUpperCase();
+      if (method !== "GET" && method !== "POST") {
+        sendJson(res, 405, { ok: false, error: "\u53EA\u63A5\u53D7 GET / POST" });
+        return;
+      }
+      const file = quickStorePath();
+      const outcome = await ensureQuickBook(legacyPrompts(), file);
+      if (outcome.kind === "broken") {
+        sendJson(res, 200, {
+          ok: false,
+          file,
+          error: `\u5FEB\u6377\u6307\u4EE4\u6587\u4EF6\u8BFB\u4E0D\u4E86\uFF1A${outcome.error}`,
+          quarantined: outcome.quarantined ?? ""
+        });
+        return;
+      }
+      if (method === "GET") {
+        sendJson(res, 200, { ok: true, file, book: outcome.book });
+        return;
+      }
+      let payload;
+      try {
+        payload = objectOf(JSON.parse(await readBody(req))) ?? {};
+      } catch (error) {
+        sendJson(res, 400, { ok: false, error: `\u8BF7\u6C42\u4F53\u4E0D\u662F\u5408\u6CD5 JSON\uFF1A${errorText2(error)}` });
+        return;
+      }
+      const candidate = payload.book !== void 0 ? payload.book : payload;
+      const next = sanitizeBook(candidate);
+      if (next === void 0) {
+        sendJson(res, 400, {
+          ok: false,
+          error: "\u63D0\u4EA4\u7684\u7ED3\u6784\u8BA4\u4E0D\u51FA\uFF08\u671F\u671B { book: { categories: [...] } } \u6216 { categories: [...] }\uFF09"
+        });
+        return;
+      }
+      try {
+        await writeQuickBook(next, file);
+      } catch (error) {
+        sendJson(res, 500, { ok: false, error: `\u5199\u5165\u5931\u8D25\uFF1A${errorText2(error)}` });
+        return;
+      }
+      const verified = await readQuickBook(file);
+      sendJson(res, 200, { ok: true, file, book: verified.kind === "ok" ? verified.book : next });
+    };
+    storeCtx.effect(() => storeCtx.webServer.register({
+      kind: "exact",
+      path: QUICK_PROMPTS_API_PATH,
+      handler: handle
+    }), "composer-ux: quick prompt store route");
+  });
 }
 export {
   apply,