@genart-dev/mcp-server 0.4.1 → 0.4.3

package/dist/index.js

@@ -2743,32 +2743,19 @@ 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";
 
 // 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;
-}
+import puppeteer from "puppeteer";
 var browserInstance = null;
 async function getBrowser() {
   if (browserInstance && browserInstance.connected) {
     return browserInstance;
   }
-  const puppeteer = await loadPuppeteer();
   browserInstance = await puppeteer.launch({
     headless: true,
     executablePath: process.env.PUPPETEER_EXECUTABLE_PATH || void 0,
@@ -2862,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();
@@ -2896,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");
@@ -2921,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;
 }
@@ -2951,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({
@@ -3315,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
@@ -3595,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) {
@@ -3665,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
@@ -3721,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);
@@ -3811,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 {
@@ -3971,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 {
@@ -4048,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,
@@ -4056,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()) {
@@ -5326,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);
@@ -5356,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);
@@ -6066,7 +6064,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'"),
@@ -6079,6 +6077,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) },
@@ -6086,6 +6085,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));
       }
     }
@@ -6726,14 +6726,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");