@eraserlabs/eraser-mcp 0.1.1 → 0.3.1-next.0

package/dist/tools.js

@@ -1,415 +1,68 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isSingleDiagramTool = exports.singleDiagramTools = exports.mcpToolMap = exports.isMcpToolName = exports.mcpTools = exports.DiagramTypes = exports.directionSettings = exports.typefaceSettings = exports.styleModeSettings = exports.colorModeSettings = void 0;
-const zod_1 = require("zod");
-/**
- * Settings duplicated from @eraserlabs/shared/modules/diagram-parser/settings
- * We avoid importing to keep eraser-mcp buildable independently.
- */
-exports.colorModeSettings = ['pastel', 'bold', 'outline'];
-exports.styleModeSettings = ['plain', 'shadow', 'watercolor'];
-exports.typefaceSettings = ['rough', 'clean', 'mono'];
-exports.directionSettings = ['up', 'down', 'left', 'right'];
-/**
- * The diagram types supported by the MCP tools.
- * Duplicated from DiagramTypes enum, excluding 'custom-diagram'.
- */
-var DiagramTypes;
-(function (DiagramTypes) {
-    DiagramTypes["SD"] = "sequence-diagram";
-    DiagramTypes["ERD"] = "entity-relationship-diagram";
-    DiagramTypes["CAD"] = "cloud-architecture-diagram";
-    DiagramTypes["FLOW"] = "flowchart-diagram";
-    DiagramTypes["BPMN"] = "bpmn-diagram";
-})(DiagramTypes = exports.DiagramTypes || (exports.DiagramTypes = {}));
-const diagramElementSchema = zod_1.z.object({
-    type: zod_1.z.literal('diagram'),
-    diagramType: zod_1.z.nativeEnum(DiagramTypes),
-    code: zod_1.z.string(),
-    x: zod_1.z.number().optional(),
-    y: zod_1.z.number().optional(),
-});
-const renderOptionsSchema = zod_1.z
-    .object({
-    padding: zod_1.z.number().optional(),
-    scale: zod_1.z.union([zod_1.z.literal(1), zod_1.z.literal(2), zod_1.z.literal(3)]).optional(),
-    background: zod_1.z.boolean().optional(),
-    theme: zod_1.z.enum(['light', 'dark']).optional(),
-    format: zod_1.z.enum(['png', 'jpeg']).optional(),
-    selection: zod_1.z.array(zod_1.z.string()).optional(),
-    ignoreElements: zod_1.z.array(zod_1.z.string()).optional(),
-    limitToSelection: zod_1.z.boolean().optional(),
-    customIcons: zod_1.z.array(zod_1.z.unknown()).optional(),
-    typeface: zod_1.z.enum(exports.typefaceSettings).optional(),
-    colorMode: zod_1.z.enum(exports.colorModeSettings).optional(),
-    styleMode: zod_1.z.enum(exports.styleModeSettings).optional(),
-    direction: zod_1.z.enum(exports.directionSettings).optional(),
-    title: zod_1.z.string().optional(),
-})
-    .passthrough();
-const renderPromptSchema = renderOptionsSchema
-    .extend({
-    text: zod_1.z.string(),
-    returnFile: zod_1.z.boolean().optional(),
-    diagramType: zod_1.z.string().optional(),
-    mode: zod_1.z.string().optional(),
-    priorRequestId: zod_1.z.string().optional(),
-    attachments: zod_1.z.array(zod_1.z.unknown()).optional(),
-    contextId: zod_1.z.string().optional(),
-    git: zod_1.z.unknown().optional(),
-    fileOptions: zod_1.z
-        .object({
-        create: zod_1.z.boolean().optional(),
-        linkAccess: zod_1.z
-            .enum([
-            'no-link-access',
-            'anyone-with-link-can-edit',
-            'publicly-viewable',
-            'publicly-editable',
-        ])
-            .optional(),
-    })
-        .optional(),
-})
-    .passthrough();
-const renderElementsSchema = renderOptionsSchema
-    .extend({
-    elements: zod_1.z.array(diagramElementSchema),
-    returnFile: zod_1.z.boolean().optional(),
-    fileName: zod_1.z.string().optional(),
-    teamId: zod_1.z.string().optional(),
-    returnElements: zod_1.z.boolean().optional(),
-    skipCache: zod_1.z.boolean().optional(),
-})
-    .passthrough();
-// Schema for individual diagram type tools
-const singleDiagramSchema = zod_1.z.object({
-    code: zod_1.z.string(),
-    theme: zod_1.z.enum(['light', 'dark']).optional(),
-    colorMode: zod_1.z.enum(exports.colorModeSettings).optional(),
-    styleMode: zod_1.z.enum(exports.styleModeSettings).optional(),
-    typeface: zod_1.z.enum(exports.typefaceSettings).optional(),
-    background: zod_1.z.boolean().optional(),
-});
-const renderPromptJsonSchema = {
-    type: 'object',
-    additionalProperties: true,
-    required: ['text'],
-    properties: {
-        text: { type: 'string' },
-        returnFile: { type: 'boolean' },
-        diagramType: { type: 'string' },
-        mode: { type: 'string' },
-        priorRequestId: { type: 'string' },
-        attachments: { type: 'array', items: {} },
-        contextId: { type: 'string' },
-        git: {},
-        fileOptions: {
-            type: 'object',
-            additionalProperties: false,
-            properties: {
-                create: { type: 'boolean' },
-                linkAccess: {
-                    type: 'string',
-                    enum: [
-                        'no-link-access',
-                        'anyone-with-link-can-edit',
-                        'publicly-viewable',
-                        'publicly-editable',
-                    ],
-                },
-            },
-        },
-        padding: { type: 'number' },
-        scale: { type: 'number', enum: [1, 2, 3] },
-        background: { type: 'boolean' },
-        theme: { type: 'string', enum: ['light', 'dark'] },
-        format: { type: 'string', enum: ['png', 'jpeg'] },
-        selection: { type: 'array', items: { type: 'string' } },
-        typeface: { type: 'string', enum: [...exports.typefaceSettings] },
-        colorMode: { type: 'string', enum: [...exports.colorModeSettings] },
-        styleMode: { type: 'string', enum: [...exports.styleModeSettings] },
-        direction: { type: 'string', enum: [...exports.directionSettings] },
-        title: { type: 'string' },
-    },
-};
-const renderElementsJsonSchema = {
-    type: 'object',
-    additionalProperties: true,
-    required: ['elements'],
-    properties: {
-        elements: {
-            type: 'array',
-            items: {
-                type: 'object',
-                required: ['type', 'diagramType', 'code'],
-                properties: {
-                    type: { type: 'string', const: 'diagram' },
-                    diagramType: {
-                        type: 'string',
-                        enum: Object.values(DiagramTypes),
-                    },
-                    code: { type: 'string' },
-                },
-            },
-        },
-        fileName: { type: 'string' },
-        teamId: { type: 'string' },
-        returnElements: { type: 'boolean' },
-        skipCache: { type: 'boolean' },
-        padding: { type: 'number' },
-        scale: { type: 'number', enum: [1, 2, 3] },
-        background: { type: 'boolean' },
-        theme: { type: 'string', enum: ['light', 'dark'] },
-        format: { type: 'string', enum: ['png', 'jpeg'] },
-        typeface: { type: 'string', enum: [...exports.typefaceSettings] },
-        colorMode: { type: 'string', enum: [...exports.colorModeSettings] },
-        styleMode: { type: 'string', enum: [...exports.styleModeSettings] },
-        direction: { type: 'string', enum: [...exports.directionSettings] },
-    },
-};
-const singleDiagramJsonSchema = {
-    type: 'object',
-    required: ['code'],
-    properties: {
-        code: { type: 'string', description: 'The diagram code in Eraser syntax' },
-        theme: { type: 'string', enum: ['light', 'dark'] },
-        colorMode: { type: 'string', enum: [...exports.colorModeSettings] },
-        styleMode: { type: 'string', enum: [...exports.styleModeSettings] },
-        typeface: { type: 'string', enum: [...exports.typefaceSettings] },
-        background: { type: 'boolean', description: 'Whether to include a solid background' },
-    },
-};
-// Diagram type descriptions with syntax examples
-const SEQUENCE_DIAGRAM_DESCRIPTION = `Render a sequence diagram. Use Eraser's sequence diagram syntax.
-
-Example syntax:
-\`\`\`
-title Authentication Flow
-
-// Define actors with icons
-User [icon: user]
-Browser [icon: globe]
-Server [icon: server]
-Database [icon: database]
-
-// Messages between actors
-User > Browser: Enter credentials
-Browser > Server: POST /login
-Server > Database: Query user
-Database > Server: User data
-Server > Browser: JWT token
-Browser > User: Login success
-
-// Control flow
-alt [label: "Valid credentials"] {
-  Server > Browser: 200 OK
-}
-else [label: "Invalid"] {
-  Server > Browser: 401 Unauthorized
-}
-
-// Activation bars
-activate Server
-Server > Database: Query
-deactivate Server
-
-// Loops
-loop [label: "Retry 3 times"] {
-  Browser > Server: Request
-}
-\`\`\``;
-const ERD_DESCRIPTION = `Render an entity-relationship diagram. Use Eraser's ERD syntax.
-
-Example syntax:
-\`\`\`
-title E-commerce Database
-
-// Define tables with columns
-users [icon: user, color: blue] {
-  id int pk
-  email string
-  name string
-  created_at timestamp
-}
-
-orders [icon: shopping-cart, color: green] {
-  id int pk
-  user_id int
-  total decimal
-  status string
-  created_at timestamp
-}
-
-products [icon: box, color: orange] {
-  id int pk
-  name string
-  price decimal
-  stock int
-}
-
-order_items [icon: list] {
-  order_id int pk
-  product_id int pk
-  quantity int
-  price decimal
-}
-
-// Relationships
-users.id < orders.user_id
-orders.id < order_items.order_id
-products.id < order_items.product_id
-\`\`\``;
-const CLOUD_ARCHITECTURE_DESCRIPTION = `Render a cloud architecture diagram. Use Eraser's cloud architecture syntax.
-
-Example syntax:
-\`\`\`
-title AWS Microservices Architecture
-
-// Groups with cloud provider icons
-AWS Cloud [icon: aws] {
-  VPC [icon: aws-vpc] {
-    Public Subnet {
-      ALB [icon: aws-elb]
-      NAT Gateway [icon: aws-nat-gateway]
-    }
-    Private Subnet {
-      ECS Cluster [icon: aws-ecs] {
-        API Service [icon: aws-lambda]
-        Worker Service [icon: aws-lambda]
-      }
-      RDS [icon: aws-rds]
-      ElastiCache [icon: aws-elasticache]
-    }
-  }
-  S3 [icon: aws-s3]
-  CloudFront [icon: aws-cloudfront]
-}
-
-Users [icon: users]
-
-// Connections
-Users > CloudFront
-CloudFront > ALB
-ALB > API Service
-API Service > RDS
-API Service > ElastiCache
-Worker Service > S3
-\`\`\``;
-const FLOWCHART_DESCRIPTION = `Render a flowchart diagram. Use Eraser's flowchart syntax.
-
-Example syntax:
-\`\`\`
-title User Registration Flow
-
-// Nodes with shapes and icons
-Start [shape: oval, icon: play]
-Enter Details [icon: edit]
-Valid Email? [shape: diamond, icon: help-circle]
-Send Verification [icon: mail]
-Email Verified? [shape: diamond]
-Create Account [icon: user-plus, color: green]
-Show Error [icon: alert-triangle, color: red]
-End [shape: oval, icon: check]
-
-// Groups
-Validation [color: blue] {
-  Check Password Strength [icon: lock]
-  Password OK? [shape: diamond]
-}
-
-// Connections with labels
-Start > Enter Details
-Enter Details > Valid Email?
-Valid Email? > Send Verification: Yes
-Valid Email? > Show Error: No
-Send Verification > Email Verified?
-Email Verified? > Create Account: Yes
-Email Verified? > Show Error: No
-Create Account > End
-Show Error > Enter Details
-\`\`\``;
-const BPMN_DESCRIPTION = `Render a BPMN (Business Process Model and Notation) diagram. Use Eraser's BPMN syntax.
-
-Example syntax:
-\`\`\`
-title Order Fulfillment Process
-
-// Swimlanes (pools)
-Customer [color: blue] {
-  Place Order [type: event, icon: shopping-cart]
-  Receive Confirmation [type: event, icon: mail]
-  Receive Package [type: event, icon: package]
-}
-
-Sales [color: green] {
-  Process Order [icon: clipboard]
-  Check Inventory [icon: database]
-  In Stock? [type: gateway, icon: help-circle]
-  Create Backorder [icon: clock]
-  Confirm Order [icon: check]
-}
-
-Warehouse [color: orange] {
-  Pick Items [icon: box]
-  Pack Order [icon: package]
-  Ship Order [icon: truck]
-}
-
-// Flow connections (use --> for message flows between pools)
-Place Order --> Process Order: Order details
-Process Order > Check Inventory
-Check Inventory > In Stock?
-In Stock? > Confirm Order: Yes
-In Stock? > Create Backorder: No
-Confirm Order --> Receive Confirmation: Confirmation email
-Confirm Order > Pick Items
-Pick Items > Pack Order
-Pack Order > Ship Order
-Ship Order --> Receive Package: Delivery
-\`\`\``;
+exports.DiagramTypes = exports.isSingleDiagramTool = exports.singleDiagramTools = exports.mcpToolMap = exports.isMcpToolName = exports.mcpTools = void 0;
+// Schemas
+const shared_1 = require("./schemas/shared");
+Object.defineProperty(exports, "DiagramTypes", { enumerable: true, get: function () { return shared_1.DiagramTypes; } });
+const createFile_1 = require("./schemas/createFile");
+const renderPrompt_1 = require("./schemas/renderPrompt");
+const renderElements_1 = require("./schemas/renderElements");
+const singleDiagram_1 = require("./schemas/singleDiagram");
+// Descriptions
+const tools_1 = require("./descriptions/tools");
+const sequenceDiagram_1 = require("./descriptions/sequenceDiagram");
+const erd_1 = require("./descriptions/erd");
+const cloudArchitecture_1 = require("./descriptions/cloudArchitecture");
+const flowchart_1 = require("./descriptions/flowchart");
+const bpmn_1 = require("./descriptions/bpmn");
 exports.mcpTools = [
+    {
+        name: 'createFile',
+        description: tools_1.CREATE_FILE_DESCRIPTION,
+        schema: createFile_1.createFileSchema,
+        jsonSchema: (0, shared_1.toJsonSchema)(createFile_1.createFileSchema),
+    },
     {
         name: 'renderPrompt',
-        description: 'Generate a diagram using AI from a natural language prompt, existing code, infrastructure configuration, or other diagram languages. Best for when you want AI to create the diagram code for you.',
-        schema: renderPromptSchema,
-        jsonSchema: renderPromptJsonSchema,
+        description: tools_1.RENDER_PROMPT_DESCRIPTION,
+        schema: renderPrompt_1.renderPromptSchema,
+        jsonSchema: (0, shared_1.toJsonSchema)(renderPrompt_1.renderPromptSchema),
     },
     {
         name: 'renderElements',
-        description: 'Render multiple diagram elements. Advanced use case for rendering multiple diagrams at once.',
-        schema: renderElementsSchema,
-        jsonSchema: renderElementsJsonSchema,
+        description: tools_1.RENDER_ELEMENTS_DESCRIPTION,
+        schema: renderElements_1.renderElementsSchema,
+        jsonSchema: (0, shared_1.toJsonSchema)(renderElements_1.renderElementsSchema),
     },
     {
         name: 'renderSequenceDiagram',
-        description: SEQUENCE_DIAGRAM_DESCRIPTION,
-        schema: singleDiagramSchema,
-        jsonSchema: singleDiagramJsonSchema,
+        description: sequenceDiagram_1.SEQUENCE_DIAGRAM_DESCRIPTION,
+        schema: singleDiagram_1.singleDiagramSchema,
+        jsonSchema: (0, shared_1.toJsonSchema)(singleDiagram_1.singleDiagramSchema),
     },
     {
         name: 'renderEntityRelationshipDiagram',
-        description: ERD_DESCRIPTION,
-        schema: singleDiagramSchema,
-        jsonSchema: singleDiagramJsonSchema,
+        description: erd_1.ERD_DESCRIPTION,
+        schema: singleDiagram_1.singleDiagramSchema,
+        jsonSchema: (0, shared_1.toJsonSchema)(singleDiagram_1.singleDiagramSchema),
     },
     {
         name: 'renderCloudArchitectureDiagram',
-        description: CLOUD_ARCHITECTURE_DESCRIPTION,
-        schema: singleDiagramSchema,
-        jsonSchema: singleDiagramJsonSchema,
+        description: cloudArchitecture_1.CLOUD_ARCHITECTURE_DESCRIPTION,
+        schema: singleDiagram_1.singleDiagramSchema,
+        jsonSchema: (0, shared_1.toJsonSchema)(singleDiagram_1.singleDiagramSchema),
     },
     {
         name: 'renderFlowchart',
-        description: FLOWCHART_DESCRIPTION,
-        schema: singleDiagramSchema,
-        jsonSchema: singleDiagramJsonSchema,
+        description: flowchart_1.FLOWCHART_DESCRIPTION,
+        schema: singleDiagram_1.singleDiagramSchema,
+        jsonSchema: (0, shared_1.toJsonSchema)(singleDiagram_1.singleDiagramSchema),
     },
     {
         name: 'renderBpmnDiagram',
-        description: BPMN_DESCRIPTION,
-        schema: singleDiagramSchema,
-        jsonSchema: singleDiagramJsonSchema,
+        description: bpmn_1.BPMN_DESCRIPTION,
+        schema: singleDiagram_1.singleDiagramSchema,
+        jsonSchema: (0, shared_1.toJsonSchema)(singleDiagram_1.singleDiagramSchema),
     },
 ];
 function isMcpToolName(name) {
@@ -419,11 +72,11 @@ exports.isMcpToolName = isMcpToolName;
 exports.mcpToolMap = new Map(exports.mcpTools.map((tool) => [tool.name, tool]));
 // Mapping from single diagram tool names to their diagram types
 exports.singleDiagramTools = {
-    renderSequenceDiagram: DiagramTypes.SD,
-    renderEntityRelationshipDiagram: DiagramTypes.ERD,
-    renderCloudArchitectureDiagram: DiagramTypes.CAD,
-    renderFlowchart: DiagramTypes.FLOW,
-    renderBpmnDiagram: DiagramTypes.BPMN,
+    renderSequenceDiagram: shared_1.DiagramTypes.SD,
+    renderEntityRelationshipDiagram: shared_1.DiagramTypes.ERD,
+    renderCloudArchitectureDiagram: shared_1.DiagramTypes.CAD,
+    renderFlowchart: shared_1.DiagramTypes.FLOW,
+    renderBpmnDiagram: shared_1.DiagramTypes.BPMN,
 };
 function isSingleDiagramTool(name) {
     return name in exports.singleDiagramTools;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eraserlabs/eraser-mcp",
-  "version": "0.1.1",
+  "version": "0.3.1-next.0",
   "description": "MCP server for generating diagrams with Eraser.io",
   "private": false,
   "main": "dist/index.js",
@@ -29,7 +29,8 @@
   },
   "license": "MIT",
   "dependencies": {
-    "zod": "^3.25.76"
+    "zod": "^3.25.76",
+    "zod-to-json-schema": "^3.24.5"
   },
   "devDependencies": {
     "@types/node": "^20.0.0",