@genart-dev/mcp-server 0.4.2 → 0.4.4

package/dist/index.cjs

@@ -2744,6 +2744,7 @@ function isDirectComponent(name, components) {
 
 // src/tools/capture.ts
 var import_promises8 = require("fs/promises");
+var import_path9 = require("path");
 var import_core9 = require("@genart-dev/core");
 
 // src/capture/headless.ts
@@ -2846,8 +2847,9 @@ function generateSketchHtml(sketch, opts) {
   }
   return adapter.generateStandaloneHTML(effective);
 }
-function derivePreviewPath(sketchPath) {
-  return sketchPath.replace(/\.genart$/, ".png");
+function deriveSnapshotPath(sketchPath, sketchId, seed) {
+  const wsDir = (0, import_path9.dirname)(sketchPath);
+  return (0, import_path9.join)(wsDir, "snapshots", `${sketchId}-${seed}-preview.png`);
 }
 async function captureScreenshot(state, input) {
   state.requireWorkspace();
@@ -2880,11 +2882,12 @@ async function captureScreenshot(state, input) {
       height,
       inlineSize
     });
-    const previewPath = derivePreviewPath(loaded.path);
+    const effectiveSeed = input.seed ?? sketch.state.seed;
+    const previewPath = deriveSnapshotPath(loaded.path, sketchId, effectiveSeed);
     const metadata = await buildScreenshotMetadata(state, multi, {
       target,
       sketchId,
-      seed: input.seed ?? sketch.state.seed,
+      seed: effectiveSeed,
       previewPath
     });
     const previewJpegBase64 = Buffer.from(multi.inlineJpeg).toString("base64");
@@ -2905,8 +2908,10 @@ async function buildScreenshotMetadata(state, multi, info) {
     previewPath: info.previewPath
   };
   if (!state.remoteMode) {
+    await (0, import_promises8.mkdir)((0, import_path9.dirname)(info.previewPath), { recursive: true });
     await (0, import_promises8.writeFile)(info.previewPath, multi.previewPng);
     metadata.savedPreviewTo = info.previewPath;
+    metadata.previewWritten = true;
   }
   return metadata;
 }
@@ -2935,11 +2940,12 @@ async function captureBatch(state, input) {
         height,
         inlineSize
       });
-      const previewPath = derivePreviewPath(loaded.path);
+      const effectiveSeed = input.seed ?? sketch.state.seed;
+      const previewPath = deriveSnapshotPath(loaded.path, id, effectiveSeed);
       const itemMetadata = await buildScreenshotMetadata(state, multi, {
         target: "sketch",
         sketchId: id,
-        seed: input.seed ?? sketch.state.seed,
+        seed: effectiveSeed,
         previewPath
       });
       items.push({
@@ -3299,7 +3305,7 @@ function gatherRelevantSkills(aspects) {
 
 // src/tools/series.ts
 var import_promises9 = require("fs/promises");
-var import_path9 = require("path");
+var import_path10 = require("path");
 var import_core11 = require("@genart-dev/core");
 function now6() {
   return (/* @__PURE__ */ new Date()).toISOString();
@@ -3462,7 +3468,7 @@ async function seriesSummary(state, input) {
   for (const file of series.sketchFiles) {
     let found = false;
     for (const [id, loaded] of state.sketches) {
-      if ((0, import_path9.basename)(loaded.path) === file) {
+      if ((0, import_path10.basename)(loaded.path) === file) {
         const def = loaded.definition;
         sketchInfos.push({
           id,
@@ -3576,18 +3582,18 @@ async function promoteSketch(state, input) {
     ...input.agent ? { agent: input.agent } : {},
     ...input.model ? { model: input.model } : {}
   };
-  const sourceDir = (0, import_path9.dirname)(source.path);
-  const newPath = (0, import_path9.resolve)(sourceDir, `${newId}.genart`);
+  const sourceDir = (0, import_path10.dirname)(source.path);
+  const newPath = (0, import_path10.resolve)(sourceDir, `${newId}.genart`);
   const json = (0, import_core11.serializeGenart)(promotedDef);
   if (!state.remoteMode) {
     await (0, import_promises9.writeFile)(newPath, json, "utf-8");
   }
   state.sketches.set(newId, { definition: promotedDef, path: newPath });
   const sourceRef = ws.sketches.find(
-    (s) => s.file === (0, import_path9.basename)(source.path)
+    (s) => s.file === (0, import_path10.basename)(source.path)
   );
   const position = sourceRef ? { x: sourceRef.position.x, y: sourceRef.position.y + sourceDef.canvas.height + 200 } : { x: 0, y: 0 };
-  const file = (0, import_path9.basename)(newPath);
+  const file = (0, import_path10.basename)(newPath);
   state.workspace = {
     ...ws,
     modified: ts,
@@ -3647,7 +3653,7 @@ function countBy(items) {
 
 // src/tools/reference.ts
 var import_promises10 = require("fs/promises");
-var import_path10 = require("path");
+var import_path11 = require("path");
 var import_core12 = require("@genart-dev/core");
 var import_promises11 = require("fs/promises");
 function now7() {
@@ -3672,7 +3678,7 @@ var IMAGE_EXTENSIONS = /* @__PURE__ */ new Set([
   ".svg"
 ]);
 function isImageFile(path) {
-  return IMAGE_EXTENSIONS.has((0, import_path10.extname)(path).toLowerCase());
+  return IMAGE_EXTENSIONS.has((0, import_path11.extname)(path).toLowerCase());
 }
 var VALID_REFERENCE_TYPES = [
   "image",
@@ -3688,7 +3694,7 @@ async function addReference(state, input) {
       `Not a recognized image file: ${input.image}. Supported: ${[...IMAGE_EXTENSIONS].join(", ")}`
     );
   }
-  const id = input.id ?? (0, import_path10.basename)(input.image, (0, import_path10.extname)(input.image)).toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-|-$/g, "");
+  const id = input.id ?? (0, import_path11.basename)(input.image, (0, import_path11.extname)(input.image)).toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-|-$/g, "");
   if (!id) {
     throw new Error("Could not derive a valid ID from the image filename");
   }
@@ -3699,15 +3705,15 @@ async function addReference(state, input) {
       `Invalid reference type: '${refType}'. Valid types: ${VALID_REFERENCE_TYPES.join(", ")}`
     );
   }
-  const workspaceDir = (0, import_path10.dirname)(state.workspacePath);
-  const refsDir = (0, import_path10.resolve)(workspaceDir, "references");
+  const workspaceDir = (0, import_path11.dirname)(state.workspacePath);
+  const refsDir = (0, import_path11.resolve)(workspaceDir, "references");
   await (0, import_promises10.mkdir)(refsDir, { recursive: true });
-  const ext = (0, import_path10.extname)(input.image);
+  const ext = (0, import_path11.extname)(input.image);
   const destFilename = `${id}${ext}`;
-  const destPath = (0, import_path10.resolve)(refsDir, destFilename);
+  const destPath = (0, import_path11.resolve)(refsDir, destFilename);
   const relativePath = `references/${destFilename}`;
   if (!state.remoteMode) {
-    await (0, import_promises10.copyFile)((0, import_path10.resolve)(input.image), destPath);
+    await (0, import_promises10.copyFile)((0, import_path11.resolve)(input.image), destPath);
   }
   const ref = {
     id,
@@ -3789,12 +3795,12 @@ async function addReference(state, input) {
 async function analyzeReference(state, input) {
   state.requireWorkspace();
   const { ref, location } = findReference(state, input.referenceId, input.seriesId, input.sketchId);
-  const workspaceDir = (0, import_path10.dirname)(state.workspacePath);
-  const imagePath = (0, import_path10.resolve)(workspaceDir, ref.path);
+  const workspaceDir = (0, import_path11.dirname)(state.workspacePath);
+  const imagePath = (0, import_path11.resolve)(workspaceDir, ref.path);
   let previewJpegBase64;
   try {
     const imageBuffer = await (0, import_promises10.readFile)(imagePath);
-    const ext = (0, import_path10.extname)(ref.path).toLowerCase();
+    const ext = (0, import_path11.extname)(ref.path).toLowerCase();
     const mimeMap = {
       ".png": "image/png",
       ".jpg": "image/jpeg",
@@ -3949,8 +3955,8 @@ async function extractPalette(state, input) {
     input.sketchId
   );
   const count = input.count ?? 6;
-  const workspaceDir = (0, import_path10.dirname)(state.workspacePath);
-  const imagePath = (0, import_path10.resolve)(workspaceDir, ref.path);
+  const workspaceDir = (0, import_path11.dirname)(state.workspacePath);
+  const imagePath = (0, import_path11.resolve)(workspaceDir, ref.path);
   let previewJpegBase64;
   try {
     const imageBuffer = await (0, import_promises10.readFile)(imagePath);
@@ -4026,12 +4032,12 @@ function findReference(state, referenceId, seriesId, sketchId) {
 // src/tools/export.ts
 var import_fs = require("fs");
 var import_promises12 = require("fs/promises");
-var import_path11 = require("path");
+var import_path12 = require("path");
 var import_archiver = __toESM(require("archiver"), 1);
 var import_core13 = require("@genart-dev/core");
 var registry4 = (0, import_core13.createDefaultRegistry)();
 async function validateOutputPath(outputPath) {
-  const parentDir = (0, import_path11.dirname)(outputPath);
+  const parentDir = (0, import_path12.dirname)(outputPath);
   try {
     const s = await (0, import_promises12.stat)(parentDir);
     if (!s.isDirectory()) {
@@ -5297,20 +5303,24 @@ async function initializePluginRegistry() {
   await registry5.register(import_plugin_trace.default);
   return registry5;
 }
-function createServer(state) {
+function createServer(state, options) {
+  const captureOnly = options?.captureOnly ?? false;
   const server = new import_mcp.McpServer(
     {
-      name: "@genart/mcp-server",
+      name: captureOnly ? "@genart/mcp-capture" : "@genart/mcp-server",
       version: "0.4.0"
     },
     {
       capabilities: {
         tools: {},
-        resources: {},
-        prompts: {}
+        ...!captureOnly && { resources: {}, prompts: {} }
       }
     }
   );
+  if (captureOnly) {
+    registerCaptureTools(server, state);
+    return server;
+  }
   const registryReady = initializePluginRegistry().then((registry5) => {
     state.pluginRegistry = registry5;
     registerPluginMcpTools(server, registry5, state);
@@ -5327,7 +5337,9 @@ function createServer(state) {
   registerSnapshotTools(server, state);
   registerKnowledgeTools(server, state);
   registerDesignTools(server, state);
-  registerCaptureTools(server, state);
+  if (!state.remoteMode) {
+    registerCaptureTools(server, state);
+  }
   registerCritiqueTools(server, state);
   registerSeriesTools(server, state);
   registerReferenceTools(server, state);
@@ -5476,11 +5488,31 @@ function registerSketchTools(server, state) {
       ).optional().describe('Component dependencies. Use list_components to see available. Keys are component names, values are semver ranges (e.g. "^1.0.0") or objects with version/code/exports.'),
       addToWorkspace: import_zod2.z.string().optional().describe("Path to workspace to add sketch to after creation"),
       agent: import_zod2.z.string().optional().describe("Your CLI agent name (e.g. 'claude-code', 'codex-cli', 'gemini-cli', 'opencode', 'kiro')"),
-      model: import_zod2.z.string().optional().describe("Your AI model identifier (e.g. 'claude-opus-4-6', 'gpt-4o', 'gemini-2.5-pro')")
+      model: import_zod2.z.string().optional().describe("Your AI model identifier (e.g. 'claude-opus-4-6', 'gpt-4o', 'gemini-2.5-pro')"),
+      capture: import_zod2.z.boolean().optional().describe("When true, automatically capture a screenshot after creation and return it inline (avoids a separate capture_screenshot call)")
     },
     async (args) => {
       try {
         const result = await createSketch(state, args);
+        if (args.capture && !state.remoteMode) {
+          try {
+            const captureResult = await captureScreenshot(state, {
+              target: "sketch",
+              sketchId: args.id
+            });
+            return {
+              content: [
+                { type: "text", text: JSON.stringify({ ...result, capture: captureResult.metadata }, null, 2) },
+                { type: "image", data: captureResult.previewJpegBase64, mimeType: "image/jpeg" }
+              ]
+            };
+          } catch (captureErr) {
+            return jsonResult({
+              ...result,
+              captureError: captureErr instanceof Error ? captureErr.message : String(captureErr)
+            });
+          }
+        }
         return jsonResult(result);
       } catch (e) {
         return toolError(e instanceof Error ? e.message : String(e));
@@ -6037,7 +6069,7 @@ function registerSnapshotTools(server, state) {
 function registerCaptureTools(server, state) {
   server.tool(
     "capture_screenshot",
-    "Capture a screenshot of a sketch. Returns metadata as text + a small inline JPEG image for visual review. In remote mode, metadata includes previewFileContent (base64 PNG) to Write locally.",
+    "Capture a screenshot of a sketch. Returns metadata as text + a small inline JPEG image for visual review. In local mode, writes a full-res PNG to snapshots/<sketchId>-<seed>-preview.png next to the workspace. The savedPreviewTo path in metadata points to the file on disk.",
     {
       target: import_zod2.z.enum(["selected", "sketch"]).optional().describe("What to capture (default: selected)"),
       sketchId: import_zod2.z.string().optional().describe("Required when target is 'sketch'"),
@@ -6699,14 +6731,17 @@ function registerKnowledgeTools(server, state) {
 
 // src/index.ts
 async function main() {
-  const sidecar = isSidecarMode();
-  if (sidecar) {
+  const captureOnly = process.argv.includes("--capture-only");
+  const sidecar = !captureOnly && isSidecarMode();
+  if (captureOnly) {
+    console.error("[genart-mcp] Starting in capture-only mode");
+  } else if (sidecar) {
     console.error("[genart-mcp] Starting in sidecar mode");
   } else {
     console.error("[genart-mcp] Starting in stdio mode");
   }
   const state = new EditorState();
-  const server = createServer(state);
+  const server = createServer(state, { captureOnly });
   const transport = new import_stdio.StdioServerTransport();
   await server.connect(transport);
   console.error("[genart-mcp] Server connected and ready");