@genart-dev/mcp-server 0.4.1 → 0.4.3

package/dist/index.cjs

@@ -2744,30 +2744,17 @@ 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
-var cachedModule = null;
-async function loadPuppeteer() {
-  if (!cachedModule) {
-    try {
-      const mod = await import("puppeteer");
-      cachedModule = mod.default ?? mod;
-    } catch {
-      throw new Error(
-        "Puppeteer is required for screenshot capture. Install it with: npm install puppeteer"
-      );
-    }
-  }
-  return cachedModule;
-}
+var import_puppeteer = __toESM(require("puppeteer"), 1);
 var browserInstance = null;
 async function getBrowser() {
   if (browserInstance && browserInstance.connected) {
     return browserInstance;
   }
-  const puppeteer = await loadPuppeteer();
-  browserInstance = await puppeteer.launch({
+  browserInstance = await import_puppeteer.default.launch({
     headless: true,
     executablePath: process.env.PUPPETEER_EXECUTABLE_PATH || void 0,
     args: [
@@ -2860,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();
@@ -2894,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");
@@ -2919,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;
 }
@@ -2949,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({
@@ -3313,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();
@@ -3476,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,
@@ -3590,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,
@@ -3661,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() {
@@ -3686,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",
@@ -3702,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");
   }
@@ -3713,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,
@@ -3803,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",
@@ -3963,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);
@@ -4040,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()) {
@@ -5311,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);
@@ -5341,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);
@@ -6051,7 +6049,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'"),
@@ -6064,6 +6062,7 @@ function registerCaptureTools(server, state) {
     async (args) => {
       try {
         const result = await captureScreenshot(state, args);
+        console.error(`[capture_screenshot] jpeg base64 length: ${result.previewJpegBase64.length}, metadata: ${JSON.stringify(result.metadata)}`);
         return {
           content: [
             { type: "text", text: JSON.stringify(result.metadata, null, 2) },
@@ -6071,6 +6070,7 @@ function registerCaptureTools(server, state) {
           ]
         };
       } catch (e) {
+        console.error(`[capture_screenshot] error: ${e instanceof Error ? e.message : String(e)}`);
         return toolError(e instanceof Error ? e.message : String(e));
       }
     }
@@ -6711,14 +6711,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");