@genart-dev/mcp-server 0.4.2 → 0.4.4

package/dist/index.js

@@ -2743,7 +2743,8 @@ function isDirectComponent(name, components) {
 }
 
 // src/tools/capture.ts
-import { writeFile as writeFile6 } from "fs/promises";
+import { mkdir, writeFile as writeFile6 } from "fs/promises";
+import { dirname as dirname6, join as join3 } from "path";
 import {
   createDefaultRegistry as createDefaultRegistry2
 } from "@genart-dev/core";
@@ -2848,8 +2849,9 @@ function generateSketchHtml(sketch, opts) {
   }
   return adapter.generateStandaloneHTML(effective);
 }
-function derivePreviewPath(sketchPath) {
-  return sketchPath.replace(/\.genart$/, ".png");
+function deriveSnapshotPath(sketchPath, sketchId, seed) {
+  const wsDir = dirname6(sketchPath);
+  return join3(wsDir, "snapshots", `${sketchId}-${seed}-preview.png`);
 }
 async function captureScreenshot(state, input) {
   state.requireWorkspace();
@@ -2882,11 +2884,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");
@@ -2907,8 +2910,10 @@ async function buildScreenshotMetadata(state, multi, info) {
     previewPath: info.previewPath
   };
   if (!state.remoteMode) {
+    await mkdir(dirname6(info.previewPath), { recursive: true });
     await writeFile6(info.previewPath, multi.previewPng);
     metadata.savedPreviewTo = info.previewPath;
+    metadata.previewWritten = true;
   }
   return metadata;
 }
@@ -2937,11 +2942,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({
@@ -3301,7 +3307,7 @@ function gatherRelevantSkills(aspects) {
 
 // src/tools/series.ts
 import { writeFile as writeFile7 } from "fs/promises";
-import { basename as basename8, dirname as dirname6, resolve as resolve3 } from "path";
+import { basename as basename8, dirname as dirname7, resolve as resolve3 } from "path";
 import {
   serializeGenart as serializeGenart5,
   serializeWorkspace as serializeWorkspace4
@@ -3581,7 +3587,7 @@ async function promoteSketch(state, input) {
     ...input.agent ? { agent: input.agent } : {},
     ...input.model ? { model: input.model } : {}
   };
-  const sourceDir = dirname6(source.path);
+  const sourceDir = dirname7(source.path);
   const newPath = resolve3(sourceDir, `${newId}.genart`);
   const json = serializeGenart5(promotedDef);
   if (!state.remoteMode) {
@@ -3651,8 +3657,8 @@ function countBy(items) {
 }
 
 // src/tools/reference.ts
-import { copyFile, mkdir, readFile as readFile5 } from "fs/promises";
-import { basename as basename9, dirname as dirname7, extname, resolve as resolve4 } from "path";
+import { copyFile, mkdir as mkdir2, readFile as readFile5 } from "fs/promises";
+import { basename as basename9, dirname as dirname8, extname, resolve as resolve4 } from "path";
 import {
   serializeGenart as serializeGenart6,
   serializeWorkspace as serializeWorkspace5
@@ -3707,9 +3713,9 @@ async function addReference(state, input) {
       `Invalid reference type: '${refType}'. Valid types: ${VALID_REFERENCE_TYPES.join(", ")}`
     );
   }
-  const workspaceDir = dirname7(state.workspacePath);
+  const workspaceDir = dirname8(state.workspacePath);
   const refsDir = resolve4(workspaceDir, "references");
-  await mkdir(refsDir, { recursive: true });
+  await mkdir2(refsDir, { recursive: true });
   const ext = extname(input.image);
   const destFilename = `${id}${ext}`;
   const destPath = resolve4(refsDir, destFilename);
@@ -3797,7 +3803,7 @@ 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 = dirname7(state.workspacePath);
+  const workspaceDir = dirname8(state.workspacePath);
   const imagePath = resolve4(workspaceDir, ref.path);
   let previewJpegBase64;
   try {
@@ -3957,7 +3963,7 @@ async function extractPalette(state, input) {
     input.sketchId
   );
   const count = input.count ?? 6;
-  const workspaceDir = dirname7(state.workspacePath);
+  const workspaceDir = dirname8(state.workspacePath);
   const imagePath = resolve4(workspaceDir, ref.path);
   let previewJpegBase64;
   try {
@@ -4034,7 +4040,7 @@ function findReference(state, referenceId, seriesId, sketchId) {
 // src/tools/export.ts
 import { createWriteStream } from "fs";
 import { stat as stat4, writeFile as writeFile9 } from "fs/promises";
-import { dirname as dirname8 } from "path";
+import { dirname as dirname9 } from "path";
 import archiver from "archiver";
 import {
   createDefaultRegistry as createDefaultRegistry3,
@@ -4042,7 +4048,7 @@ import {
 } from "@genart-dev/core";
 var registry4 = createDefaultRegistry3();
 async function validateOutputPath(outputPath) {
-  const parentDir = dirname8(outputPath);
+  const parentDir = dirname9(outputPath);
   try {
     const s = await stat4(parentDir);
     if (!s.isDirectory()) {
@@ -5312,20 +5318,24 @@ async function initializePluginRegistry() {
   await registry5.register(tracePlugin);
   return registry5;
 }
-function createServer(state) {
+function createServer(state, options) {
+  const captureOnly = options?.captureOnly ?? false;
   const server = new 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);
@@ -5342,7 +5352,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);
@@ -5491,11 +5503,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: 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 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));
@@ -6052,7 +6084,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: z2.enum(["selected", "sketch"]).optional().describe("What to capture (default: selected)"),
       sketchId: z2.string().optional().describe("Required when target is 'sketch'"),
@@ -6714,14 +6746,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 StdioServerTransport();
   await server.connect(transport);
   console.error("[genart-mcp] Server connected and ready");