@genart-dev/mcp-server 0.4.3 → 0.4.5

package/dist/index.js

@@ -2893,7 +2893,9 @@ async function captureScreenshot(state, input) {
       previewPath
     });
     const previewJpegBase64 = Buffer.from(multi.inlineJpeg).toString("base64");
-    return { metadata, previewJpegBase64 };
+    const previewDataUri = `data:image/jpeg;base64,${previewJpegBase64}`;
+    metadata.previewDataUri = previewDataUri;
+    return { metadata, previewJpegBase64, previewDataUri };
   } catch (e) {
     const msg = e instanceof Error ? e.message : String(e);
     throw new Error(`Renderer error for '${sketchId}': ${msg}`);
@@ -5503,11 +5505,34 @@ 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: z2.string().optional().describe("Path to workspace to add sketch to after creation"),
       agent: z2.string().optional().describe("Your CLI agent name (e.g. 'claude-code', 'codex-cli', 'gemini-cli', 'opencode', 'kiro')"),
-      model: z2.string().optional().describe("Your AI model identifier (e.g. 'claude-opus-4-6', 'gpt-4o', 'gemini-2.5-pro')")
+      model: z2.string().optional().describe("Your AI model identifier (e.g. 'claude-opus-4-6', 'gpt-4o', 'gemini-2.5-pro')"),
+      capture: z2.boolean().optional().describe("When true, automatically capture a screenshot after creation. The result includes previewDataUri for rendering. IMPORTANT: When capture is true, you MUST show the preview to the user by creating an HTML artifact with an <img> tag using the previewDataUri from the capture metadata. Include savedPreviewTo as caption for the full-res file.")
     },
     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));
@@ -6064,7 +6089,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 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.",
+    "Capture a screenshot of a sketch. Returns metadata (including previewDataUri) as text + a small inline JPEG image. In local mode, writes a full-res PNG to snapshots/<sketchId>-<seed>-preview.png. IMPORTANT: After every capture, you MUST show the preview to the user by creating an HTML artifact containing an <img> tag using the previewDataUri from the metadata. Include the savedPreviewTo path as a caption so the user can find the full-res file.",
     {
       target: z2.enum(["selected", "sketch"]).optional().describe("What to capture (default: selected)"),
       sketchId: z2.string().optional().describe("Required when target is 'sketch'"),
@@ -6092,7 +6117,7 @@ function registerCaptureTools(server, state) {
   );
   server.tool(
     "capture_batch",
-    "Capture screenshots of multiple sketches in parallel. Returns metadata as text + inline JPEG images for visual review.",
+    "Capture screenshots of multiple sketches in parallel. Returns per-sketch metadata (including previewDataUri) + inline JPEG images. IMPORTANT: After batch capture, you MUST show all previews to the user by creating an HTML artifact with <img> tags using each item's previewDataUri. Include sketch titles and savedPreviewTo paths.",
     {
       sketchIds: z2.array(z2.string()).optional().describe("IDs of sketches to capture (default: all)"),
       width: z2.number().optional().describe("Override width for all captures"),