@modudraft/mcp 0.4.0 → 0.5.0

package/README.md

@@ -104,6 +104,9 @@ Copy the key — you'll only see it once.
 | `add_sequence_participant` | Register a named participant |
 | `clear_sequence` | Remove all sequence messages |
 | `auto_sequence` | Derive sequence messages from the architecture edges |
+| `add_sequence_fragment` | Wrap messages in a loop / alt / opt / par fragment |
+| `update_sequence_fragment` | Change a fragment's kind, condition, or message list |
+| `delete_sequence_fragment` | Remove a fragment |
 
 ### DB schema (ER diagram)
 | Tool | Description |
@@ -121,6 +124,28 @@ Copy the key — you'll only see it once.
 | `delete_api_endpoint` | Remove an endpoint |
 | `export_openapi` | Export schema as OpenAPI 3.0 JSON |
 
+### Cloud diagram
+| Tool | Description |
+|---|---|
+| `add_cloud_resource` | Add a cloud resource (cluster, VPC, service, etc.) |
+| `update_cloud_resource` | Rename or change the type of a cloud resource |
+| `delete_cloud_resource` | Remove a cloud resource |
+
+### Network diagram
+| Tool | Description |
+|---|---|
+| `add_network_device` | Add a network device (router, switch, firewall, etc.) |
+| `update_network_device` | Rename or change the type of a network device |
+| `delete_network_device` | Remove a network device |
+
+### Data flow diagram
+| Tool | Description |
+|---|---|
+| `add_data_flow_stage` | Add a pipeline stage (source, transform, sink, etc.) |
+| `update_data_flow_stage` | Rename or change the type of a data flow stage |
+| `delete_data_flow_stage` | Remove a data flow stage |
+| `reorder_data_flow_stages` | Set the order of all stages in a data flow diagram |
+
 ### Discovery (static, no API call)
 | Tool | Description |
 |---|---|
@@ -140,6 +165,12 @@ Copy the key — you'll only see it once.
 
 > "What metadata is missing from the diagram nodes? Fill in the hosts and ports based on our .env file."
 
+> "Add a data flow diagram showing how events move from our Kafka topics through a Flink transform into the data warehouse."
+
+> "Draw the network topology — edge router, core switch, two firewalls, and the DMZ segment."
+
+> "Wrap the retry loop in the checkout sequence in an alt fragment with condition 'payment fails'."
+
 ---
 
 ## How it works

package/dist/index.js

@@ -25149,6 +25149,108 @@ var ALL_TOOLS = [
       return { ok: true, messagesCreated: count };
     }
   },
+  // ── Sequence fragments ───────────────────────────────────────────────────
+  {
+    name: "add_sequence_fragment",
+    description: 'Wrap a group of sequence messages in a fragment box (loop, alt, opt, or par). Use message_ids to list the message IDs that fall inside the fragment. kind: "loop" for repetition, "alt" for conditional branches, "opt" for optional, "par" for parallel.',
+    inputSchema: {
+      type: "object",
+      properties: {
+        diagram_id: { type: "string" },
+        kind: {
+          type: "string",
+          enum: ["loop", "alt", "opt", "par"],
+          description: "Fragment kind"
+        },
+        condition: {
+          type: "string",
+          description: 'Guard condition label shown on the fragment (e.g. "valid credentials", "for each item")'
+        },
+        message_ids: {
+          type: "array",
+          items: { type: "string" },
+          description: "Ordered list of sequence message IDs to include in this fragment"
+        }
+      },
+      required: ["diagram_id", "kind", "condition", "message_ids"]
+    },
+    async handler(args) {
+      const id = diagramId(args);
+      let frag;
+      await mutateDiagram(id, (entry) => {
+        if (!Array.isArray(entry.sequenceFragments)) entry.sequenceFragments = [];
+        const f = {
+          id: genId(),
+          kind: str(args.kind, "opt"),
+          condition: str(args.condition),
+          messageIds: Array.isArray(args.message_ids) ? args.message_ids : []
+        };
+        entry.sequenceFragments.push(f);
+        frag = f;
+      });
+      return frag;
+    }
+  },
+  {
+    name: "update_sequence_fragment",
+    description: "Update an existing sequence fragment \u2014 change its kind, condition label, or the set of enclosed message IDs.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        diagram_id: { type: "string" },
+        fragment_id: { type: "string", description: "ID of the fragment to update" },
+        kind: { type: "string", enum: ["loop", "alt", "opt", "par"] },
+        condition: { type: "string" },
+        message_ids: { type: "array", items: { type: "string" } }
+      },
+      required: ["diagram_id", "fragment_id"]
+    },
+    async handler(args) {
+      const id = diagramId(args);
+      let found = false;
+      await mutateDiagram(id, (entry) => {
+        if (!Array.isArray(entry.sequenceFragments)) return;
+        entry.sequenceFragments = entry.sequenceFragments.map((f) => {
+          const frag = f;
+          if (frag.id !== str(args.fragment_id)) return frag;
+          found = true;
+          return {
+            ...frag,
+            ...args.kind !== void 0 && { kind: str(args.kind) },
+            ...args.condition !== void 0 && { condition: str(args.condition) },
+            ...Array.isArray(args.message_ids) && { messageIds: args.message_ids }
+          };
+        });
+      });
+      if (!found) throw new Error(`Fragment ${args.fragment_id} not found`);
+      return { ok: true };
+    }
+  },
+  {
+    name: "delete_sequence_fragment",
+    description: "Remove a sequence fragment box. The enclosed messages remain; only the fragment wrapper is deleted.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        diagram_id: { type: "string" },
+        fragment_id: { type: "string" }
+      },
+      required: ["diagram_id", "fragment_id"]
+    },
+    async handler(args) {
+      const id = diagramId(args);
+      let removed = false;
+      await mutateDiagram(id, (entry) => {
+        if (!Array.isArray(entry.sequenceFragments)) return;
+        const before = entry.sequenceFragments.length;
+        entry.sequenceFragments = entry.sequenceFragments.filter(
+          (f) => f.id !== str(args.fragment_id)
+        );
+        removed = entry.sequenceFragments.length < before;
+      });
+      return { ok: removed };
+    }
+  },
   // ── DB schema (ER diagram) ───────────────────────────────────────────────
   {
     name: "add_db_table",
@@ -25728,14 +25830,53 @@ var ALL_TOOLS = [
       return { ok: true, id: updated.id };
     }
   },
-  // ── Data Flow stages ─────────────────────────────────────────────────────
+  // ── Data Flow flows + stages ──────────────────────────────────────────────
+  {
+    name: "list_data_flows",
+    description: "List all named data flows in the Data Flow tab of a diagram.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        diagram_id: { type: "string" }
+      },
+      required: ["diagram_id"]
+    },
+    async handler(args) {
+      const { data } = await fetchDiagram(diagramId(args));
+      const tab = data.diagrams.find((d) => d.type === "data-flow");
+      return tab?.dataFlows ?? [];
+    }
+  },
+  {
+    name: "add_data_flow",
+    description: "Create a new named data flow in the Data Flow tab. Returns the flow id to use when adding stages.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        diagram_id: { type: "string" },
+        name: { type: "string", description: 'Name for the flow (e.g. "Ingestion pipeline", "ETL")' }
+      },
+      required: ["diagram_id", "name"]
+    },
+    async handler(args) {
+      const id = diagramId(args);
+      let flow;
+      await mutateTypedTab(id, "data-flow", (entry) => {
+        const order = (entry.dataFlows ?? []).length;
+        flow = { id: genId(), name: str(args.name, "Untitled flow"), order };
+        entry.dataFlows = [...entry.dataFlows ?? [], flow];
+      });
+      return flow;
+    }
+  },
   {
     name: "add_data_flow_stage",
-    description: "Add a stage to the Data Flow tab. Stages have a role (source \u2192 transform \u2192 sink) and an optional tool. Stages are ordered by insertion; use reorder_data_flow_stages to rearrange.",
+    description: "Add a stage to a data flow. Stages have a role (source \u2192 transform \u2192 sink) and an optional tool. Pass flow_id to add to a specific flow (use list_data_flows or add_data_flow to get one). Stages are ordered by insertion; use reorder_data_flow_stages to rearrange.",
     inputSchema: {
       type: "object",
       properties: {
         diagram_id: { type: "string" },
+        flow_id: { type: "string", description: "ID of the data flow this stage belongs to (from add_data_flow or list_data_flows)" },
         type: {
           type: "string",
           enum: ["source", "transform", "sink"],
@@ -25750,9 +25891,11 @@ var ALL_TOOLS = [
       const id = diagramId(args);
       let stage;
       await mutateTypedTab(id, "data-flow", (entry) => {
-        const order = (entry.dataFlowStages ?? []).length;
+        const flowId = str(args.flow_id) || void 0;
+        const order = (entry.dataFlowStages ?? []).filter((s) => s.flowId === flowId).length;
         stage = {
           id: genId(),
+          ...flowId ? { flowId } : {},
           type: str(args.type, "source"),
           label: str(args.label),
           tool: str(args.tool),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@modudraft/mcp",
-  "version": "0.4.0",
+  "version": "0.5.0",
   "description": "MCP server for Modudraft — create and edit system-design diagrams via AI (cloud edition)",
   "type": "module",
   "bin": {
@@ -9,7 +9,9 @@
   "scripts": {
     "build": "tsup && node -e \"const fs=require('fs');fs.chmodSync('dist/index.js','755')\"",
     "typecheck": "tsc --noEmit",
-    "dev": "tsup --watch"
+    "dev": "tsup --watch",
+    "test": "vitest run",
+    "test:watch": "vitest"
   },
   "files": [
     "dist",
@@ -20,8 +22,9 @@
   },
   "devDependencies": {
     "@modudraft/core": "*",
-    "@types/node": "^22.0.0",
+    "@types/node": "^26.0.0",
     "tsup": "^8.5.0",
-    "typescript": "~5.8.0"
+    "typescript": "~5.8.0",
+    "vitest": "^4.0.0"
   }
 }