@fangyb/ahchat-bridge 0.1.32 → 0.1.34

package/dist/seedanceMcpCli.cjs

@@ -3258,8 +3258,8 @@ var require_utils = __commonJS({
       }
       return ind;
     }
-    function removeDotSegments(path) {
-      let input = path;
+    function removeDotSegments(path2) {
+      let input = path2;
       const output = [];
       let nextSlash = -1;
       let len = 0;
@@ -3512,8 +3512,8 @@ var require_schemes = __commonJS({
         wsComponent.secure = void 0;
       }
       if (wsComponent.resourceName) {
-        const [path, query] = wsComponent.resourceName.split("?");
-        wsComponent.path = path && path !== "/" ? path : void 0;
+        const [path2, query] = wsComponent.resourceName.split("?");
+        wsComponent.path = path2 && path2 !== "/" ? path2 : void 0;
         wsComponent.query = query;
         wsComponent.resourceName = void 0;
       }
@@ -6952,12 +6952,12 @@ var require_dist = __commonJS({
         throw new Error(`Unknown format "${name}"`);
       return f;
     };
-    function addFormats(ajv, list, fs, exportName) {
+    function addFormats(ajv, list, fs2, exportName) {
       var _a3;
       var _b;
       (_a3 = (_b = ajv.opts.code).formats) !== null && _a3 !== void 0 ? _a3 : _b.formats = (0, codegen_1._)`require("ajv-formats/dist/formats").${exportName}`;
       for (const f of list)
-        ajv.addFormat(f, fs[f]);
+        ajv.addFormat(f, fs2[f]);
     }
     module2.exports = exports2 = formatsPlugin;
     Object.defineProperty(exports2, "__esModule", { value: true });
@@ -7350,8 +7350,8 @@ function getErrorMap() {
 // ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v3/helpers/parseUtil.js
 init_cjs_shims();
 var makeIssue = (params) => {
-  const { data, path, errorMaps, issueData } = params;
-  const fullPath = [...path, ...issueData.path || []];
+  const { data, path: path2, errorMaps, issueData } = params;
+  const fullPath = [...path2, ...issueData.path || []];
   const fullIssue = {
     ...issueData,
     path: fullPath
@@ -7470,11 +7470,11 @@ var errorUtil;
 
 // ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v3/types.js
 var ParseInputLazyPath = class {
-  constructor(parent, value, path, key) {
+  constructor(parent, value, path2, key) {
     this._cachedPath = [];
     this.parent = parent;
     this.data = value;
-    this._path = path;
+    this._path = path2;
     this._key = key;
   }
   get path() {
@@ -11406,10 +11406,10 @@ function mergeDefs(...defs) {
 function cloneDef(schema) {
   return mergeDefs(schema._zod.def);
 }
-function getElementAtPath(obj, path) {
-  if (!path)
+function getElementAtPath(obj, path2) {
+  if (!path2)
     return obj;
-  return path.reduce((acc, key) => acc?.[key], obj);
+  return path2.reduce((acc, key) => acc?.[key], obj);
 }
 function promiseAllObject(promisesObj) {
   const keys = Object.keys(promisesObj);
@@ -11818,11 +11818,11 @@ function explicitlyAborted(x, startIndex = 0) {
   }
   return false;
 }
-function prefixIssues(path, issues) {
+function prefixIssues(path2, issues) {
   return issues.map((iss) => {
     var _a3;
     (_a3 = iss).path ?? (_a3.path = []);
-    iss.path.unshift(path);
+    iss.path.unshift(path2);
     return iss;
   });
 }
@@ -11969,16 +11969,16 @@ function flattenError(error51, mapper = (issue2) => issue2.message) {
 }
 function formatError(error51, mapper = (issue2) => issue2.message) {
   const fieldErrors = { _errors: [] };
-  const processError = (error52, path = []) => {
+  const processError = (error52, path2 = []) => {
     for (const issue2 of error52.issues) {
       if (issue2.code === "invalid_union" && issue2.errors.length) {
-        issue2.errors.map((issues) => processError({ issues }, [...path, ...issue2.path]));
+        issue2.errors.map((issues) => processError({ issues }, [...path2, ...issue2.path]));
       } else if (issue2.code === "invalid_key") {
-        processError({ issues: issue2.issues }, [...path, ...issue2.path]);
+        processError({ issues: issue2.issues }, [...path2, ...issue2.path]);
       } else if (issue2.code === "invalid_element") {
-        processError({ issues: issue2.issues }, [...path, ...issue2.path]);
+        processError({ issues: issue2.issues }, [...path2, ...issue2.path]);
       } else {
-        const fullpath = [...path, ...issue2.path];
+        const fullpath = [...path2, ...issue2.path];
         if (fullpath.length === 0) {
           fieldErrors._errors.push(mapper(issue2));
         } else {
@@ -12005,17 +12005,17 @@ function formatError(error51, mapper = (issue2) => issue2.message) {
 }
 function treeifyError(error51, mapper = (issue2) => issue2.message) {
   const result = { errors: [] };
-  const processError = (error52, path = []) => {
+  const processError = (error52, path2 = []) => {
     var _a3, _b;
     for (const issue2 of error52.issues) {
       if (issue2.code === "invalid_union" && issue2.errors.length) {
-        issue2.errors.map((issues) => processError({ issues }, [...path, ...issue2.path]));
+        issue2.errors.map((issues) => processError({ issues }, [...path2, ...issue2.path]));
       } else if (issue2.code === "invalid_key") {
-        processError({ issues: issue2.issues }, [...path, ...issue2.path]);
+        processError({ issues: issue2.issues }, [...path2, ...issue2.path]);
       } else if (issue2.code === "invalid_element") {
-        processError({ issues: issue2.issues }, [...path, ...issue2.path]);
+        processError({ issues: issue2.issues }, [...path2, ...issue2.path]);
       } else {
-        const fullpath = [...path, ...issue2.path];
+        const fullpath = [...path2, ...issue2.path];
         if (fullpath.length === 0) {
           result.errors.push(mapper(issue2));
           continue;
@@ -12047,8 +12047,8 @@ function treeifyError(error51, mapper = (issue2) => issue2.message) {
 }
 function toDotPath(_path) {
   const segs = [];
-  const path = _path.map((seg) => typeof seg === "object" ? seg.key : seg);
-  for (const seg of path) {
+  const path2 = _path.map((seg) => typeof seg === "object" ? seg.key : seg);
+  for (const seg of path2) {
     if (typeof seg === "number")
       segs.push(`[${seg}]`);
     else if (typeof seg === "symbol")
@@ -25262,13 +25262,13 @@ function resolveRef(ref, ctx) {
   if (!ref.startsWith("#")) {
     throw new Error("External $ref is not supported, only local refs (#/...) are allowed");
   }
-  const path = ref.slice(1).split("/").filter(Boolean);
-  if (path.length === 0) {
+  const path2 = ref.slice(1).split("/").filter(Boolean);
+  if (path2.length === 0) {
     return ctx.rootSchema;
   }
   const defsKey = ctx.version === "draft-2020-12" ? "$defs" : "definitions";
-  if (path[0] === defsKey) {
-    const key = path[1];
+  if (path2[0] === defsKey) {
+    const key = path2[1];
     if (!key || !ctx.defs[key]) {
       throw new Error(`Reference not found: ${ref}`);
     }
@@ -31231,6 +31231,63 @@ var StdioServerTransport = class {
   }
 };
 
+// src/seedanceMcpCli.ts
+var import_promises = __toESM(require("fs/promises"), 1);
+var import_node_path = __toESM(require("path"), 1);
+
+// src/officialMcpQuota.ts
+init_cjs_shims();
+var OfficialMcpQuotaError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "OfficialMcpQuotaError";
+  }
+};
+async function consumeOfficialMcpDailyQuota(serverName, toolName) {
+  const serverApiUrl = process.env.AHCHAT_SERVER_API_URL?.trim();
+  const bridgeToken = process.env.AHCHAT_BRIDGE_TOKEN?.trim();
+  if (!serverApiUrl || !bridgeToken) return;
+  let response;
+  try {
+    response = await fetch(`${serverApiUrl.replace(/\/+$/, "")}/api/mcp/usage/consume`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "X-AHChat-Bridge-Token": bridgeToken
+      },
+      body: JSON.stringify({ serverName, toolName })
+    });
+  } catch {
+    throw new OfficialMcpQuotaError("\u6682\u65F6\u65E0\u6CD5\u6821\u9A8C\u5B98\u65B9\u5A92\u4F53 MCP \u6BCF\u65E5\u989D\u5EA6\uFF0C\u8BF7\u7A0D\u540E\u518D\u8BD5\u3002");
+  }
+  if (response.status === 404) return;
+  const body = await readQuotaResponse(response);
+  if (response.status === 429 || body.allowed === false) {
+    throw new OfficialMcpQuotaError(formatQuotaExceededMessage(serverName, body));
+  }
+  if (!response.ok) {
+    throw new OfficialMcpQuotaError(body.message || "\u5B98\u65B9\u5A92\u4F53 MCP \u6BCF\u65E5\u989D\u5EA6\u6821\u9A8C\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u518D\u8BD5\u3002");
+  }
+}
+async function readQuotaResponse(response) {
+  try {
+    const value = await response.json();
+    if (value && typeof value === "object" && !Array.isArray(value)) {
+      return value;
+    }
+  } catch {
+  }
+  return {};
+}
+function formatQuotaExceededMessage(serverName, body) {
+  if (body.message) return body.message;
+  const label = serverName === "seedance" ? "\u751F\u89C6\u9891" : serverName === "seedream" ? "\u751F\u56FE" : "\u5B98\u65B9\u5A92\u4F53";
+  if (typeof body.used === "number" && typeof body.limit === "number") {
+    return `\u4ECA\u5929\u7684${label}\u989D\u5EA6\u5DF2\u7528\u5B8C\uFF08${body.used}/${body.limit}\uFF09\u3002\u8BF7\u660E\u5929\u518D\u8BD5\uFF0C\u6216\u8054\u7CFB\u8FD0\u8425\u8C03\u6574\u989D\u5EA6\u3002`;
+  }
+  return `\u4ECA\u5929\u7684${label}\u989D\u5EA6\u5DF2\u7528\u5B8C\u3002\u8BF7\u660E\u5929\u518D\u8BD5\uFF0C\u6216\u8054\u7CFB\u8FD0\u8425\u8C03\u6574\u989D\u5EA6\u3002`;
+}
+
 // src/seedanceMcpCli.ts
 var SERVER_NAME = "ahchat-seedance-mcp";
 var SERVER_VERSION = "0.1.0";
@@ -31292,7 +31349,7 @@ function createServer() {
   const server = new McpServer(
     { name: SERVER_NAME, version: SERVER_VERSION },
     {
-      instructions: "AHChat official Seedance video generation MCP. Call seedance_create_task to submit a job, then seedance_check_task to poll the task_id. Generated URLs can expire; show or download them promptly."
+      instructions: "AHChat official Seedance video generation MCP. Each user request should create exactly one video task. Call seedance_create_task once to submit a job, then seedance_check_task only when progress is requested. Generated URLs can expire; show or download them promptly."
     }
   );
   server.registerTool(
@@ -31315,7 +31372,7 @@ function createServer() {
     "seedance_create_task",
     {
       title: "Create Seedance video generation task",
-      description: "Submit a Seedance video generation task to the Volcengine Ark API and return the task_id immediately. Poll seedance_check_task afterwards.",
+      description: "Submit exactly one Seedance video generation task to the Volcengine Ark API and return the task_id immediately. In AHChat, do not submit alternate tasks, loop, or sleep afterwards; the media task card tracks status.",
       inputSchema: createTaskInputShape,
       annotations: {
         readOnlyHint: false,
@@ -31328,6 +31385,7 @@ function createServer() {
       if (validationError) return errorResult(validationError);
       try {
         const normalized = normalizeCreateTaskInput(input);
+        await consumeOfficialMcpDailyQuota("seedance", "seedance_create_task");
         const result = await createSeedanceTask(normalized);
         const payload = {
           state: "submitted",
@@ -31363,12 +31421,16 @@ function createServer() {
     async (input) => {
       try {
         const parsed = checkTaskInputSchema.parse(input);
-        const result = await checkSeedanceTask(parsed.task_id);
+        const result = await withDownloadedSeedanceVideo(await checkSeedanceTask(parsed.task_id));
         const payload = {
           task_id: result.id,
           status: result.status,
           video_url: result.videoUrl,
           last_frame_url: result.lastFrameUrl,
+          local_path: result.localPath,
+          file_name: result.fileName,
+          mime_type: result.mimeType,
+          download_error: result.downloadError,
           fail_reason: result.failReason
         };
         return {
@@ -31495,12 +31557,139 @@ async function checkSeedanceTask(taskId) {
     raw: obj
   };
 }
+async function withDownloadedSeedanceVideo(result) {
+  if (!result.videoUrl) return result;
+  try {
+    const downloaded = await downloadGeneratedVideoToWorkdir(result.videoUrl, result.id);
+    return {
+      ...result,
+      localPath: downloaded.localPath,
+      fileName: downloaded.fileName,
+      mimeType: downloaded.mimeType
+    };
+  } catch (error51) {
+    return {
+      ...result,
+      downloadError: error51 instanceof Error ? error51.message : String(error51)
+    };
+  }
+}
+async function downloadGeneratedVideoToWorkdir(url2, taskId) {
+  if (!/^https?:\/\//i.test(url2)) {
+    throw new Error("generated video is not a downloadable URL");
+  }
+  const existingFileName = generatedVideoFileName(taskId, videoExtensionFromUrl(url2) ?? ".mp4");
+  const existingPath = import_node_path.default.join(process.cwd(), existingFileName);
+  if (await fileExists(existingPath)) {
+    return {
+      localPath: existingPath,
+      fileName: existingFileName,
+      mimeType: videoMimeTypeFromPath(existingPath) ?? "video/mp4"
+    };
+  }
+  const response = await fetch(url2, { signal: AbortSignal.timeout(18e4) });
+  if (!response.ok) {
+    throw new Error(`download failed with HTTP ${response.status}`);
+  }
+  const buffer = Buffer.from(await response.arrayBuffer());
+  const mimeType = videoMimeTypeFromContentType(response.headers.get("content-type")) ?? videoMimeTypeFromUrl(url2) ?? "video/mp4";
+  const extension = videoExtensionForMimeType(mimeType);
+  const fileName = generatedVideoFileName(taskId, extension);
+  const filePath = import_node_path.default.join(process.cwd(), fileName);
+  await import_promises.default.mkdir(process.cwd(), { recursive: true });
+  try {
+    await import_promises.default.writeFile(filePath, buffer, { flag: "wx" });
+  } catch (error51) {
+    if (!isFileExistsError(error51)) throw error51;
+  }
+  return { localPath: filePath, fileName, mimeType };
+}
+function generatedVideoFileName(taskId, extension) {
+  const safeTaskId = taskId.replace(/[^A-Za-z0-9._-]+/g, "_").slice(0, 96) || "task";
+  return `seedance-${safeTaskId}${extension}`;
+}
+async function fileExists(filePath) {
+  try {
+    await import_promises.default.access(filePath);
+    return true;
+  } catch {
+    return false;
+  }
+}
+function videoMimeTypeFromContentType(contentType) {
+  if (!contentType) return null;
+  return normalizeVideoMimeType(contentType.split(";")[0]?.trim() ?? "");
+}
+function videoMimeTypeFromUrl(url2) {
+  try {
+    return videoMimeTypeFromPath(new URL(url2).pathname);
+  } catch {
+    return videoMimeTypeFromPath(url2);
+  }
+}
+function videoExtensionFromUrl(url2) {
+  try {
+    return videoExtensionFromPath(new URL(url2).pathname);
+  } catch {
+    return videoExtensionFromPath(url2);
+  }
+}
+function videoMimeTypeFromPath(filePath) {
+  switch (import_node_path.default.extname(filePath).toLowerCase()) {
+    case ".mov":
+      return "video/quicktime";
+    case ".webm":
+      return "video/webm";
+    case ".m4v":
+      return "video/x-m4v";
+    case ".mp4":
+      return "video/mp4";
+    default:
+      return null;
+  }
+}
+function videoExtensionFromPath(filePath) {
+  switch (import_node_path.default.extname(filePath).toLowerCase()) {
+    case ".mov":
+    case ".webm":
+    case ".m4v":
+    case ".mp4":
+      return import_node_path.default.extname(filePath).toLowerCase();
+    default:
+      return null;
+  }
+}
+function normalizeVideoMimeType(mimeType) {
+  const normalized = mimeType.toLowerCase();
+  if (normalized === "video/mp4" || normalized === "video/quicktime" || normalized === "video/webm" || normalized === "video/x-m4v") {
+    return normalized;
+  }
+  return null;
+}
+function videoExtensionForMimeType(mimeType) {
+  switch (mimeType) {
+    case "video/quicktime":
+      return ".mov";
+    case "video/webm":
+      return ".webm";
+    case "video/x-m4v":
+      return ".m4v";
+    default:
+      return ".mp4";
+  }
+}
+function isFileExistsError(error51) {
+  return Boolean(error51 && typeof error51 === "object" && error51.code === "EEXIST");
+}
 async function parseJsonSafe(response) {
   const text = await response.text();
   if (!text) return void 0;
   try {
     return JSON.parse(text);
-  } catch {
+  } catch (error51) {
+    writeStderr(
+      `[${SERVER_NAME}] ARK API returned a non-JSON response body: ${error51 instanceof Error ? error51.message : String(error51)}`
+    );
     return text;
   }
 }
@@ -31579,22 +31768,27 @@ function errorResult(message) {
 function usageGuide() {
   return [
     "AHChat Seedance workflow:",
-    "1. Call seedance_create_task with prompt and optional reference media URLs.",
-    "2. Wait 30-90 seconds, then call seedance_check_task with the returned task_id.",
-    "3. When status is succeeded, use video_url in the chat result. Signed URLs may expire.",
+    "1. Call seedance_create_task with prompt and optional reference media URLs. Each user request creates exactly one video task.",
+    "2. After submission, stop manual waiting. AHChat will track the task in the visible media card.",
+    "3. Only call seedance_check_task once when the user explicitly asks for progress or when diagnostics are needed.",
+    "4. When status is succeeded, keep the text reply short and let the media card show preview, download, copy, and regenerate actions. Signed URLs may expire.",
     "",
     "Common defaults: model=doubao-seedance-2-0-260128, duration=5, ratio=16:9, resolution=720p, generate_audio=true.",
     "For vertical short video, set ratio=9:16. Reference media URLs must be reachable by Volcengine Ark."
   ].join("\n");
 }
+function writeStderr(message) {
+  process.stderr.write(`${message}
+`);
+}
 async function main() {
   const server = createServer();
   const transport = new StdioServerTransport();
   await server.connect(transport);
-  console.error(`${SERVER_NAME} v${SERVER_VERSION} running on stdio`);
+  writeStderr(`${SERVER_NAME} v${SERVER_VERSION} running on stdio`);
 }
 main().catch((error51) => {
   const message = error51 instanceof Error ? error51.stack ?? error51.message : String(error51);
-  console.error(`Fatal error in ${SERVER_NAME}: ${message}`);
+  writeStderr(`Fatal error in ${SERVER_NAME}: ${message}`);
   process.exit(1);
 });