@icogenie/mcp 0.1.1 → 0.3.0

package/README.md

@@ -64,7 +64,7 @@ generate_icon({
 })
 ```
 
-**Returns:** `{ generationId, preview, creditsUsed, creditsRemaining, metadata }`
+**Returns:** `{ sessionId, preview, previews, creditsRemaining, sessionData, suggestions }`
 
 ### regenerate_icon
 Regenerate a specific icon variation with a custom refinement prompt.
@@ -73,12 +73,15 @@ Regenerate a specific icon variation with a custom refinement prompt.
 
 ```
 regenerate_icon({
-  generationId: "abc123",  // or bundleId for bundles
-  index: 0,                // which variation (0-based)
+  sessionId: "abc123",       // for single icons
+  bundleId: "xyz789",        // for bundles (use one or the other)
+  index: 0,                  // which variation (0-based)
   prompt: "Make it more 3D"  // optional refinement
 })
 ```
 
+**Returns:** `{ success, index, preview, creditsRemaining }`
+
 ### check_credits
 Check your current credit balance.
 
@@ -132,7 +135,7 @@ generate_bundle({
 })
 ```
 
-**Returns:** `{ bundleId, icons: [{ name, preview }], credits }`
+**Returns:** `{ bundleId, iconCount, icons: [{ name, description, preview }], pricing, creditsUsed, creditsRemaining }`
 
 ## Example Workflow
 
@@ -145,19 +148,19 @@ generate_bundle({
 2. **Generate a single icon:**
    ```
    generate_icon({ prompt: "notification bell icon", style: "outline" })
-   → { generationId: "abc123", preview: "...", creditsRemaining: 49 }
+   → { sessionId: "abc123", preview: "...", creditsRemaining: 49 }
    ```
 
 3. **Refine if needed:**
    ```
-   regenerate_icon({ generationId: "abc123", index: 0, prompt: "Add a dot indicator" })
+   regenerate_icon({ sessionId: "abc123", index: 0, prompt: "Add a dot indicator" })
    → { preview: "...", creditsRemaining: 48 }
    ```
 
 4. **Download final package:**
    ```
    download_icon({ generationId: "abc123", outputPath: "./bell-icon.zip" })
-   → { savedTo: "./bell-icon.zip", creditsRemaining: 43 }
+   → { savedTo: "./bell-icon.zip" }
    ```
 
 ## Bundle Workflow

package/dist/index.js

@@ -5,6 +5,7 @@ import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js"
 
 // src/server.ts
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { CREDIT_CONFIG } from "@icogenie/core";
 
 // src/tools/generate.ts
 import { z } from "zod";
@@ -162,15 +163,32 @@ function readReferenceImage(filePath) {
   const mimeType = ext === ".png" ? "image/png" : ext === ".webp" ? "image/webp" : "image/jpeg";
   return { data: buffer.toString("base64"), mimeType };
 }
-async function request2(endpoint, body, requireAuth = true) {
-  const url = `${getApiUrl()}/api/cli${endpoint}`;
+async function request2(endpoint, options = {}, requireAuth = true) {
+  const { method = "POST", body, params } = options;
+  let url = `${getApiUrl()}/api${endpoint}`;
+  if (params && Object.keys(params).length > 0) {
+    url += `?${new URLSearchParams(params).toString()}`;
+  }
   const makeRequest = async (sessionToken) => {
-    const requestBody = sessionToken ? { ...body, sessionToken } : body;
+    const headers = {};
+    if (sessionToken) {
+      headers["Authorization"] = `Bearer ${sessionToken}`;
+    }
+    if (body) {
+      headers["Content-Type"] = "application/json";
+    }
     const response = await fetch(url, {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify(requestBody)
+      method,
+      headers,
+      ...body && { body: JSON.stringify(body) }
     });
+    const contentType = response.headers.get("content-type");
+    if (contentType?.includes("application/zip")) {
+      if (!response.ok) {
+        throw new ApiError("Download failed", response.status);
+      }
+      return response;
+    }
     const data = await response.json();
     if (response.status === 401 && requireAuth) {
       const newCreds = await handleAuthError();
@@ -192,51 +210,66 @@ async function request2(endpoint, body, requireAuth = true) {
   return makeRequest();
 }
 async function generate(options) {
-  return request2("/generate", {
-    prompt: options.prompt,
-    variations: options.variations || 1,
-    style: options.style || "solid",
-    ...options.referenceImage && { referenceImage: options.referenceImage }
+  return request2("/generate-preview", {
+    body: {
+      prompt: options.prompt,
+      variations: options.variations || 1,
+      style: options.style || "solid",
+      ...options.referenceImage && { referenceImage: options.referenceImage }
+    }
   });
 }
 async function regenerate(options) {
-  return request2("/regenerate", {
-    generationId: options.generationId,
-    bundleId: options.bundleId,
-    index: options.index,
-    prompt: options.prompt
+  return request2("/regenerate-icon", {
+    body: {
+      sessionId: options.sessionId,
+      bundleId: options.bundleId,
+      index: options.index,
+      prompt: options.prompt
+    }
   });
 }
 async function getCredits() {
-  return request2("/credits", {});
+  return request2("/auth/session", { method: "GET" });
 }
 async function download(options) {
-  return request2("/download", {
-    generationId: options.generationId,
-    bundleId: options.bundleId
+  const params = {};
+  if (options.generationId) params.generation_id = options.generationId;
+  if (options.bundleId) params.bundle_id = options.bundleId;
+  const response = await request2("/download", {
+    method: "GET",
+    params
   });
+  const contentDisposition = response.headers.get("content-disposition");
+  const filenameMatch = contentDisposition?.match(/filename="?([^";\n]+)"?/);
+  const filename = filenameMatch?.[1] || `icogenie-${(options.generationId || options.bundleId || "export").slice(0, 8)}.zip`;
+  return { response, filename };
 }
 async function normalizeBundle(options) {
-  return request2("/normalize", {
-    description: options.description,
-    targetCount: options.targetCount,
-    style: options.style
+  return request2("/normalize-bundle", {
+    body: {
+      description: options.description,
+      targetCount: options.targetCount,
+      style: options.style
+    }
   });
 }
 async function generateBundle(options) {
-  return request2("/bundle", {
-    description: options.description,
-    targetCount: options.targetCount,
-    icons: options.icons,
-    style: options.style || "solid",
-    ...options.referenceImage && { referenceImage: options.referenceImage }
+  const bundleId = options.bundleId || crypto.randomUUID();
+  return request2("/generate-bundle-preview", {
+    body: {
+      bundleId,
+      icons: options.icons,
+      style: options.style || "solid",
+      ...options.referenceImage && { referenceImage: options.referenceImage }
+    }
   });
 }
 
 // src/tools/generate.ts
 var generateIconSchema = {
   prompt: z.string().describe("Description of the icon to generate"),
-  style: z.enum(["solid", "outline"]).default("solid").describe("Icon style"),
+  style: z.enum(["solid", "outline", "illustration"]).default("solid").describe("Icon style: solid (filled shapes), outline (stroked lines), or illustration (colorful, detailed)"),
   variations: z.union([z.literal(1), z.literal(2), z.literal(4)]).default(1).describe("Number of variations (1, 2, or 4)"),
   referenceImagePath: z.string().optional().describe("Local file path to reference image for style extraction"),
   referenceImage: z.object({
@@ -258,37 +291,38 @@ async function generateIcon(args) {
     referenceImage: refImage
   });
   return {
-    generationId: result.generationId,
+    sessionId: result.sessionId,
     preview: result.preview,
-    creditsUsed: result.creditsUsed,
-    creditsRemaining: result.creditsRemaining,
-    metadata: result.metadata
+    previews: result.previews,
+    creditsRemaining: result.credits,
+    sessionData: result.sessionData,
+    suggestions: result.suggestions
   };
 }
 
 // src/tools/regenerate.ts
 import { z as z2 } from "zod";
 var regenerateIconSchema = {
-  generationId: z2.string().optional().describe("For single icon variations"),
+  sessionId: z2.string().optional().describe("For single icon variations"),
   bundleId: z2.string().optional().describe("For bundle icons"),
   index: z2.number().describe("Which variation/icon to regenerate (0-based)"),
   prompt: z2.string().optional().describe("Custom refinement prompt")
 };
 async function regenerateIcon(args) {
-  if (!args.generationId && !args.bundleId) {
-    throw new Error("Must provide either generationId or bundleId");
+  if (!args.sessionId && !args.bundleId) {
+    throw new Error("Must provide either sessionId or bundleId");
   }
   const result = await regenerate({
-    generationId: args.generationId,
+    sessionId: args.sessionId,
     bundleId: args.bundleId,
     index: args.index,
     prompt: args.prompt
   });
   return {
     success: result.success,
+    index: result.index,
     preview: result.preview,
-    creditsUsed: result.creditsUsed,
-    creditsRemaining: result.creditsRemaining
+    creditsRemaining: result.credits
   };
 }
 
@@ -315,27 +349,23 @@ async function downloadIcon(args) {
   if (!args.generationId && !args.bundleId) {
     throw new Error("Must provide either generationId or bundleId");
   }
-  const result = await download({
+  const { response, filename } = await download({
     generationId: args.generationId,
     bundleId: args.bundleId
   });
-  const response = {
-    success: result.success,
-    creditsUsed: result.creditsUsed,
-    creditsRemaining: result.creditsRemaining,
-    iconCount: result.iconCount
+  const arrayBuffer = await response.arrayBuffer();
+  const zipBuffer = Buffer.from(arrayBuffer);
+  const result = {
+    success: true,
+    filename
   };
   if (args.outputPath) {
-    const buffer = Buffer.from(result.bundle.data, "base64");
-    writeFileSync(args.outputPath, buffer);
-    response.savedTo = args.outputPath;
+    writeFileSync(args.outputPath, zipBuffer);
+    result.savedTo = args.outputPath;
   } else {
-    response.bundle = {
-      data: result.bundle.data,
-      filename: result.bundle.filename
-    };
+    result.bundleData = zipBuffer.toString("base64");
   }
-  return response;
+  return result;
 }
 
 // src/tools/normalize-bundle.ts
@@ -343,7 +373,7 @@ import { z as z4 } from "zod";
 var normalizeBundleSchema = {
   description: z4.string().describe("Description of the icon bundle to create"),
   targetCount: z4.number().optional().describe("Target number of icons (2-20)"),
-  style: z4.enum(["solid", "outline"]).optional().describe("Preferred icon style")
+  style: z4.enum(["solid", "outline", "illustration"]).optional().describe("Preferred icon style")
 };
 async function normalizeBundleTool(args) {
   const result = await normalizeBundle({
@@ -371,7 +401,7 @@ var generateBundleSchema = {
   ).optional().describe("Icon list from normalize_bundle or custom"),
   description: z5.string().optional().describe("Alternative: describe the bundle to auto-generate icon list"),
   targetCount: z5.number().optional().describe("Target icon count when using description"),
-  style: z5.enum(["solid", "outline"]).default("solid").describe("Icon style"),
+  style: z5.enum(["solid", "outline", "illustration"]).default("solid").describe("Icon style: solid (filled shapes), outline (stroked lines), or illustration (colorful, detailed)"),
   referenceImagePath: z5.string().optional().describe("Local file path to reference image"),
   referenceImage: z5.object({
     data: z5.string(),
@@ -379,8 +409,8 @@ var generateBundleSchema = {
   }).optional().describe("Base64-encoded reference image")
 };
 async function generateBundleTool(args) {
-  if (!args.icons && !args.description) {
-    throw new Error("Must provide either icons array or description");
+  if (!args.icons) {
+    throw new Error("Must provide icons array (use normalize_bundle first to get icon list)");
   }
   let refImage;
   if (args.referenceImagePath) {
@@ -390,21 +420,20 @@ async function generateBundleTool(args) {
   }
   const result = await generateBundle({
     icons: args.icons,
-    description: args.description,
-    targetCount: args.targetCount,
     style: args.style || "solid",
     referenceImage: refImage
   });
   return {
     bundleId: result.bundleId,
-    iconCount: result.iconCount,
-    icons: result.icons.map((icon) => ({
+    iconCount: result.previews.length,
+    icons: result.previews.map((icon) => ({
       name: icon.name,
       description: icon.description,
       preview: icon.preview
     })),
-    credits: result.credits,
-    nextStep: result.nextStep
+    pricing: result.pricing,
+    creditsUsed: result.creditsUsed,
+    creditsRemaining: result.credits
   };
 }
 
@@ -418,7 +447,7 @@ function createServer() {
     "generate_icon",
     {
       title: "Generate Icon",
-      description: "Generate an AI-powered icon preview from a text description. Costs 1 credit. Returns a generationId for download.",
+      description: `Generate an AI-powered icon preview from a text description. Costs ${CREDIT_CONFIG.costs.preview} credit. Returns a generationId for download. After generation, save the preview image to the user's working directory and display it.`,
       inputSchema: generateIconSchema
     },
     async (args) => {
@@ -433,7 +462,7 @@ function createServer() {
     "regenerate_icon",
     {
       title: "Regenerate Icon",
-      description: "Regenerate a specific icon variation with an optional custom prompt. Costs 1 credit. Use with generationId (single) or bundleId (bundle).",
+      description: `Regenerate a specific icon variation with an optional custom prompt. Costs ${CREDIT_CONFIG.costs.preview} credit. Use with generationId (single) or bundleId (bundle). After regeneration, save and display the new preview.`,
       inputSchema: regenerateIconSchema
     },
     async (args) => {
@@ -463,7 +492,7 @@ function createServer() {
     "download_icon",
     {
       title: "Download Icon",
-      description: "Download the final SVG + PNG package for an icon. Costs 5 credits (single) or 4 credits/icon (bundle). Provide outputPath to save to file.",
+      description: `Download the final SVG + PNG package for an icon. Costs ${CREDIT_CONFIG.costs.vectorDownload} credits (single) or ${CREDIT_CONFIG.costs.bundleDownload} credits/icon (bundle). Provide outputPath to save to file. Always save to the user's working directory with a descriptive filename.`,
       inputSchema: downloadIconSchema
     },
     async (args) => {
@@ -493,7 +522,7 @@ function createServer() {
     "generate_bundle",
     {
       title: "Generate Bundle",
-      description: "Generate a bundle of icons from an icon list. Costs 1 credit per icon. Use normalize_bundle first to plan, or provide icons directly.",
+      description: `Generate a bundle of icons from an icon list. Costs ${CREDIT_CONFIG.costs.bundlePreview} credit per icon. Use normalize_bundle first to plan, or provide icons directly. After generation, save preview images to the user's working directory and display them.`,
       inputSchema: generateBundleSchema
     },
     async (args) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@icogenie/mcp",
-  "version": "0.1.1",
+  "version": "0.3.0",
   "description": "MCP server for IcoGenie - Enable AI agents to generate icons programmatically",
   "type": "module",
   "bin": {
@@ -18,7 +18,8 @@
     "@modelcontextprotocol/sdk": "^1.12.0",
     "conf": "^12.0.0",
     "open": "^10.0.0",
-    "zod": "^3.23.0"
+    "zod": "^3.23.0",
+    "@icogenie/core": "0.1.0"
   },
   "devDependencies": {
     "@types/node": "^20.0.0",