@modudraft/mcp 0.4.3 → 0.5.1

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

@@ -24311,7 +24311,9 @@ var apiEndpointSchema = external_exports.object({
 var dataFlowSchema = external_exports.object({
   id: external_exports.string().min(1),
   name: external_exports.string().default("Untitled flow"),
-  order: external_exports.number().int().min(0)
+  order: external_exports.number().int().min(0),
+  sourceToTransformNote: external_exports.string().optional(),
+  transformToSinkNote: external_exports.string().optional()
 });
 var dataFlowStageSchema = external_exports.object({
   id: external_exports.string().min(1),
@@ -24319,6 +24321,7 @@ var dataFlowStageSchema = external_exports.object({
   type: external_exports.enum(["source", "transform", "sink"]),
   tool: external_exports.string().default(""),
   label: external_exports.string().default(""),
+  description: external_exports.string().optional(),
   order: external_exports.number().int().min(0)
 });
 
@@ -25830,21 +25833,69 @@ 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")' },
+        source_to_transform_note: { type: "string", description: "Optional note shown on the arrow between source and transform columns" },
+        transform_to_sink_note: { type: "string", description: "Optional note shown on the arrow between transform and sink columns" }
+      },
+      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,
+          ...str(args.source_to_transform_note) ? { sourceToTransformNote: str(args.source_to_transform_note) } : {},
+          ...str(args.transform_to_sink_note) ? { transformToSinkNote: str(args.transform_to_sink_note) } : {}
+        };
+        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"],
           description: "Stage role: source (ingestion), transform (processing), sink (output/storage)"
         },
         label: { type: "string", description: 'Display label (e.g. "Raw Events", "Filter PII", "DW Load")' },
-        tool: { type: "string", description: 'Technology slug (e.g. "kafka", "postgresql", "spark", "dbt", "redis", "s3")' }
+        tool: { type: "string", description: 'Technology slug (e.g. "kafka", "postgresql", "spark", "dbt", "redis", "s3")' },
+        description: { type: "string", description: "Optional explanation of what this stage does, shown on the canvas card" }
       },
       required: ["diagram_id", "type", "label"]
     },
@@ -25852,12 +25903,15 @@ 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),
+          ...str(args.description) ? { description: str(args.description) } : {},
           order
         };
         entry.dataFlowStages = [...entry.dataFlowStages ?? [], stage];
@@ -25888,7 +25942,7 @@ var ALL_TOOLS = [
   },
   {
     name: "update_data_flow_stage",
-    description: "Update an existing data flow stage's type, tool, and/or label.",
+    description: "Update an existing data flow stage's type, tool, label, and/or description.",
     inputSchema: {
       type: "object",
       properties: {
@@ -25896,7 +25950,8 @@ var ALL_TOOLS = [
         stage_id: { type: "string", description: "ID of the stage to update" },
         type: { type: "string", enum: ["source", "transform", "sink"], description: "New stage role (optional)" },
         tool: { type: "string", description: "New technology slug (optional)" },
-        label: { type: "string", description: "New display label (optional)" }
+        label: { type: "string", description: "New display label (optional)" },
+        description: { type: "string", description: "Explanation of what this stage does (optional; pass empty string to clear)" }
       },
       required: ["diagram_id", "stage_id"]
     },
@@ -25910,6 +25965,7 @@ var ALL_TOOLS = [
         if (typeof args.type === "string") stage.type = args.type;
         if (typeof args.tool === "string") stage.tool = args.tool;
         if (typeof args.label === "string") stage.label = args.label;
+        if (typeof args.description === "string") stage.description = args.description || void 0;
         updated = stage;
       });
       return { ok: true, id: updated.id };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@modudraft/mcp",
-  "version": "0.4.3",
+  "version": "0.5.1",
   "description": "MCP server for Modudraft — create and edit system-design diagrams via AI (cloud edition)",
   "type": "module",
   "bin": {