dodraw-mcp-server 0.1.7 → 0.1.8

package/dist/src/create_class_diagram_example.js

@@ -1,21 +1,19 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const diagramTools_1 = require("./tools/diagramTools");
+import { handleToolCall } from './tools/diagramTools.js';
 async function createClassDiagram() {
     const FILE_PATH = "class_diagram_example.3duml";
     console.log(`Creating class diagram example at ${FILE_PATH}...`);
     // 1. Create Diagram
     try {
-        await (0, diagramTools_1.handleToolCall)("create_new_diagram", { filePath: FILE_PATH });
+        await handleToolCall("create_new_diagram", { filePath: FILE_PATH });
     }
     catch (e) {
         console.log("File might exist, continuing...");
     }
     // 2. Add Layer
-    await (0, diagramTools_1.handleToolCall)("add_layer", { filePath: FILE_PATH, name: "Classes" });
+    await handleToolCall("add_layer", { filePath: FILE_PATH, name: "Classes" });
     // 3. Create Classes
     // Abstract Class: GraphicObject
-    await (0, diagramTools_1.handleToolCall)("create_uml_class", {
+    await handleToolCall("create_uml_class", {
         filePath: FILE_PATH,
         className: "GraphicObject",
         attributes: ["x: int", "y: int"],
@@ -24,7 +22,7 @@ async function createClassDiagram() {
         id: "GraphicObject"
     });
     // Subclass: Circle
-    await (0, diagramTools_1.handleToolCall)("create_uml_class", {
+    await handleToolCall("create_uml_class", {
         filePath: FILE_PATH,
         className: "Circle",
         attributes: ["radius: float"],
@@ -33,7 +31,7 @@ async function createClassDiagram() {
         id: "Circle"
     });
     // Subclass: Rectangle
-    await (0, diagramTools_1.handleToolCall)("create_uml_class", {
+    await handleToolCall("create_uml_class", {
         filePath: FILE_PATH,
         className: "Rectangle",
         attributes: ["width: float", "height: float"],
@@ -42,7 +40,7 @@ async function createClassDiagram() {
         id: "Rectangle"
     });
     // Container: Group
-    await (0, diagramTools_1.handleToolCall)("create_uml_class", {
+    await handleToolCall("create_uml_class", {
         filePath: FILE_PATH,
         className: "Group",
         attributes: [],
@@ -51,7 +49,7 @@ async function createClassDiagram() {
         id: "Group"
     });
     // Client: Canvas
-    await (0, diagramTools_1.handleToolCall)("create_uml_class", {
+    await handleToolCall("create_uml_class", {
         filePath: FILE_PATH,
         className: "Canvas",
         attributes: ["width: int", "height: int", "backgroundColor: string"],
@@ -61,7 +59,7 @@ async function createClassDiagram() {
     });
     // 4. Create Edges (Relationships)
     // Inheritance: Circle -> GraphicObject
-    await (0, diagramTools_1.handleToolCall)("add_edge", {
+    await handleToolCall("add_edge", {
         filePath: FILE_PATH,
         sourceId: "Circle",
         targetId: "GraphicObject",
@@ -80,7 +78,7 @@ async function createClassDiagram() {
         id: "edge_circle_extends"
     });
     // Inheritance: Rectangle -> GraphicObject
-    await (0, diagramTools_1.handleToolCall)("add_edge", {
+    await handleToolCall("add_edge", {
         filePath: FILE_PATH,
         sourceId: "Rectangle",
         targetId: "GraphicObject",
@@ -99,7 +97,7 @@ async function createClassDiagram() {
     // The Diamond is on the Group end.
     // Source: Group. Target: GraphicObject.
     // Diamond at Source. Arrow/None at Target.
-    await (0, diagramTools_1.handleToolCall)("add_edge", {
+    await handleToolCall("add_edge", {
         filePath: FILE_PATH,
         sourceId: "Group",
         targetId: "GraphicObject",
@@ -116,7 +114,7 @@ async function createClassDiagram() {
     // Let's say Canvas has a list of Shapes.
     // Canvas (8, -4) -> GraphicObject (0, -4).
     // Canvas Left (3) -> GraphicObject Right (1).
-    await (0, diagramTools_1.handleToolCall)("add_edge", {
+    await handleToolCall("add_edge", {
         filePath: FILE_PATH,
         sourceId: "Canvas",
         targetId: "GraphicObject",

package/dist/src/create_termination_reference.js

@@ -1,18 +1,16 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const diagramTools_1 = require("./tools/diagramTools");
+import { handleToolCall } from './tools/diagramTools.js';
 async function createReference() {
     const FILE_PATH = "edge_terminations_test.3duml";
     console.log(`Creating diagram at ${FILE_PATH}...`);
     // 1. Create Diagram
     try {
-        await (0, diagramTools_1.handleToolCall)("create_new_diagram", { filePath: FILE_PATH });
+        await handleToolCall("create_new_diagram", { filePath: FILE_PATH });
     }
     catch (e) {
         console.log("File might exist, continuing...");
     }
     // 2. Add Layer
-    await (0, diagramTools_1.handleToolCall)("add_layer", { filePath: FILE_PATH, name: "Terminations" });
+    await handleToolCall("add_layer", { filePath: FILE_PATH, name: "Terminations" });
     // 3. Define Types
     const types = [
         { id: 'none', label: 'None', description: 'No visual marker' },
@@ -37,7 +35,7 @@ async function createReference() {
         const sourceId = `src_${t.id}`;
         const targetId = `tgt_${t.id}`;
         // Start Node
-        await (0, diagramTools_1.handleToolCall)("add_node", {
+        await handleToolCall("add_node", {
             filePath: FILE_PATH,
             label: `Start`,
             id: sourceId,
@@ -46,7 +44,7 @@ async function createReference() {
             color: "#eeeeee"
         });
         // End Node
-        await (0, diagramTools_1.handleToolCall)("add_node", {
+        await handleToolCall("add_node", {
             filePath: FILE_PATH,
             label: `End`,
             id: targetId,
@@ -55,7 +53,7 @@ async function createReference() {
             color: "#eeeeee"
         });
         // Description Note (Left side)
-        await (0, diagramTools_1.handleToolCall)("add_node", {
+        await handleToolCall("add_node", {
             filePath: FILE_PATH,
             label: `${t.label}\n${t.description}`,
             shape: 'note',
@@ -65,7 +63,7 @@ async function createReference() {
             id: `note_${t.id}`
         });
         // Edge
-        await (0, diagramTools_1.handleToolCall)("add_edge", {
+        await handleToolCall("add_edge", {
             filePath: FILE_PATH,
             sourceId: sourceId,
             targetId: targetId,

package/dist/src/index.js

@@ -1,11 +1,9 @@
 #!/usr/bin/env node
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/server/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/server/stdio.js");
-const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
-const diagramTools_js_1 = require("./tools/diagramTools.js");
-const server = new index_js_1.Server({
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+import { toolDefinitions, handleToolCall } from "./tools/diagramTools.js";
+const server = new Server({
     name: "dodraw-mcp-server",
     version: "0.1.0",
 }, {
@@ -13,14 +11,14 @@ const server = new index_js_1.Server({
         tools: {},
     },
 });
-server.setRequestHandler(types_js_1.ListToolsRequestSchema, async () => {
+server.setRequestHandler(ListToolsRequestSchema, async () => {
     return {
-        tools: diagramTools_js_1.toolDefinitions,
+        tools: toolDefinitions,
     };
 });
-server.setRequestHandler(types_js_1.CallToolRequestSchema, async (request) => {
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
     try {
-        return await (0, diagramTools_js_1.handleToolCall)(request.params.name, request.params.arguments);
+        return await handleToolCall(request.params.name, request.params.arguments);
     }
     catch (error) {
         return {
@@ -35,7 +33,7 @@ server.setRequestHandler(types_js_1.CallToolRequestSchema, async (request) => {
     }
 });
 async function main() {
-    const transport = new stdio_js_1.StdioServerTransport();
+    const transport = new StdioServerTransport();
     await server.connect(transport);
     console.error("DoDraw MCP Server running on stdio");
 }

package/dist/src/tools/diagramTools.js

@@ -1,10 +1,6 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.toolDefinitions = void 0;
-exports.handleToolCall = handleToolCall;
-const fileHandler_1 = require("../utils/fileHandler");
-const crypto_1 = require("crypto");
-exports.toolDefinitions = [
+import { readDiagramFile, saveDiagramFile } from "../utils/fileHandler.js";
+import { randomUUID } from "crypto";
+export const toolDefinitions = [
     {
         name: "create_new_diagram",
         description: "Create a new empty DoDraw diagram file",
@@ -175,7 +171,7 @@ exports.toolDefinitions = [
     },
     {
         name: "create_uml_class",
-        description: "Create a UML Class node. Use this for the FIRST class. For connected classes, use 'add_directional_node' to ensure correct relative positioning.",
+        description: "Create a UML Class node. Supports absolute positioning (x,y) OR relative positioning (referenceNodeId, direction). Can also auto-connect.",
         inputSchema: {
             type: "object",
             properties: {
@@ -183,9 +179,19 @@ exports.toolDefinitions = [
                 className: { type: "string" },
                 attributes: { type: "array", items: { type: "string" }, description: "List of attributes e.g. ['- id: int', '+ name: string']" },
                 methods: { type: "array", items: { type: "string" }, description: "List of methods e.g. ['+ getName(): string', '+ save(): void']" },
-                x: { type: "number", description: "Optional X position" },
-                y: { type: "number", description: "Optional Y position (mapped to Z in 3DUML)" },
-                id: { type: "string", description: "Optional explicit ID" }
+                id: { type: "string", description: "Optional explicit ID" },
+                referenceNodeId: { type: "string", description: "Optional. ID of an existing node to place this new class relative to." },
+                direction: { type: "string", enum: ["UP", "DOWN", "LEFT", "RIGHT"], description: "Optional. Direction to place relative to referenceNodeId." },
+                connection: {
+                    type: "object",
+                    description: "Optional. If provided, creates an edge connecting the reference node to this new class.",
+                    properties: {
+                        type: { type: "string", enum: ["association", "inheritance", "implementation", "dependency", "aggregation", "composition"], description: "Relationship type" },
+                        label: { type: "string" },
+                        direction: { type: "string", enum: ["source-to-target", "target-to-source", "bi-directional"], description: "Edge direction flow" }
+                    },
+                    required: ["type"]
+                }
             },
             required: ["filePath", "className"]
         }
@@ -244,11 +250,11 @@ exports.toolDefinitions = [
         }
     }
 ];
-async function handleToolCall(name, args) {
+export async function handleToolCall(name, args) {
     switch (name) {
         case "create_new_diagram": {
             // ... (existing code)
-            const defaultLayerId = (0, crypto_1.randomUUID)();
+            const defaultLayerId = randomUUID();
             const initialState = {
                 nodes: [],
                 edges: [],
@@ -262,12 +268,12 @@ async function handleToolCall(name, args) {
                 gridSnappingEnabled: true,
                 gridSize: 0.5
             };
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, initialState);
+            await saveDiagramFile(args.filePath, initialState);
             return { content: [{ type: "text", text: `Created new diagram at ${args.filePath}` }] };
         }
         case "read_diagram_structure": {
             // ... (existing code)
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
+            const state = await readDiagramFile(args.filePath);
             return {
                 content: [{
                         type: "text",
@@ -284,7 +290,7 @@ async function handleToolCall(name, args) {
         }
         case "add_node": {
             // ... (existing code)
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
+            const state = await readDiagramFile(args.filePath);
             // Auto-placement Logic
             let x = 0;
             let z = 0;
@@ -310,7 +316,7 @@ async function handleToolCall(name, args) {
                 }
             }
             const newNode = {
-                id: args.id || (0, crypto_1.randomUUID)(),
+                id: args.id || randomUUID(),
                 x: x,
                 y: 0.05,
                 z: z,
@@ -331,16 +337,16 @@ async function handleToolCall(name, args) {
             };
             console.error(`DEBUG: Adding node. Args: x=${args.x}, y=${args.y}. Computed: x=${newNode.x}, z=${newNode.z}`);
             state.nodes.push(newNode);
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, state);
+            await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Added node ${newNode.id} at (${newNode.x}, ${newNode.z})` }] };
         }
         case "add_directional_node": {
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
+            const state = await readDiagramFile(args.filePath);
             const sourceNode = state.nodes.find(n => n.id === args.sourceNodeId);
             if (!sourceNode) {
                 throw new Error(`Source node with ID ${args.sourceNodeId} not found.`);
             }
-            const newNodeId = (0, crypto_1.randomUUID)();
+            const newNodeId = randomUUID();
             // 1. Calculate base position with tighter spacing (matching App interaction)
             const INITIAL_SPACING = 2.0;
             let dirX = 0;
@@ -438,7 +444,7 @@ async function handleToolCall(name, args) {
                 textAlignHorizontal: 'center'
             };
             const newEdge = {
-                id: (0, crypto_1.randomUUID)(),
+                id: randomUUID(),
                 sourceId: sourceNode.id,
                 targetId: newNodeId,
                 sourcePointIndex: sourcePoint,
@@ -453,11 +459,11 @@ async function handleToolCall(name, args) {
             };
             state.nodes.push(newNode);
             state.edges.push(newEdge);
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, state);
+            await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Added node '${args.label}' (ID: ${newNodeId}) to the ${args.direction} of '${sourceNode.label}' and connected them.` }] };
         }
         case "add_edge": {
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
+            const state = await readDiagramFile(args.filePath);
             // Verify nodes exist
             const sourceExists = state.nodes.find(n => n.id === args.sourceId);
             const targetExists = state.nodes.find(n => n.id === args.targetId);
@@ -466,7 +472,7 @@ async function handleToolCall(name, args) {
             if (!targetExists)
                 throw new Error(`Target node ${args.targetId} not found`);
             const newEdge = {
-                id: args.id || (0, crypto_1.randomUUID)(),
+                id: args.id || randomUUID(),
                 sourceId: args.sourceId,
                 targetId: args.targetId,
                 sourcePointIndex: Number(args.sourcePointIndex),
@@ -482,11 +488,11 @@ async function handleToolCall(name, args) {
                 terminationEnd: args.terminationEnd
             };
             state.edges.push(newEdge);
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, state);
+            await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Added edge ${newEdge.id}` }] };
         }
         case "update_node": {
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
+            const state = await readDiagramFile(args.filePath);
             const nodeIndex = state.nodes.findIndex(n => n.id === args.nodeId);
             if (nodeIndex === -1) {
                 throw new Error(`Node ${args.nodeId} not found`);
@@ -503,12 +509,12 @@ async function handleToolCall(name, args) {
                 state.nodes[nodeIndex].height = Number(args.height);
             if (args.backgroundColor)
                 state.nodes[nodeIndex].backgroundColor = args.backgroundColor;
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, state);
+            await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Updated node ${args.nodeId}` }] };
         }
         case "add_layer": {
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
-            const newLayerId = (0, crypto_1.randomUUID)();
+            const state = await readDiagramFile(args.filePath);
+            const newLayerId = randomUUID();
             // Auto-positioning logic
             let position = { x: 0, y: 0, z: 0 };
             if (args.position) {
@@ -547,12 +553,12 @@ async function handleToolCall(name, args) {
                 visible: args.visible !== false
             };
             state.layers.push(newLayer);
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, state);
+            await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Added layer '${args.name}' (ID: ${newLayerId}) at Y=${position.y}` }] };
         }
         case "create_uml_class": {
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
-            const newNodeId = args.id || (0, crypto_1.randomUUID)();
+            const state = await readDiagramFile(args.filePath);
+            const newNodeId = args.id || randomUUID();
             // Standard size for class
             const width = 3;
             // dynamic height? 
@@ -560,10 +566,64 @@ async function handleToolCall(name, args) {
             const methodCount = (args.methods || []).length;
             const estimatedHeight = 0.8 + (attrCount + methodCount) * 0.4;
             const height = Math.max(2, estimatedHeight);
-            // Auto-placement logic if X/Y not provided
-            let x = args.x !== undefined ? Number(args.x) : 0;
-            let z = args.y !== undefined ? Number(args.y) : 0;
-            if (args.x === undefined && args.y === undefined && state.nodes.length > 0) {
+            // Auto-placement logic (defaults to 0,0 for first node)
+            let x = 0;
+            let z = 0;
+            // Relative Placement Logic
+            if (args.referenceNodeId && args.direction) {
+                const sourceNode = state.nodes.find(n => n.id === args.referenceNodeId);
+                if (!sourceNode)
+                    throw new Error(`Reference node ${args.referenceNodeId} not found`);
+                const INITIAL_SPACING = 2.0;
+                let dirX = 0;
+                let dirZ = 0;
+                const srcW = Number(sourceNode.width) || 2;
+                const srcH = Number(sourceNode.height) || 1.5;
+                switch (args.direction) {
+                    case "RIGHT":
+                        dirX = 1;
+                        break;
+                    case "LEFT":
+                        dirX = -1;
+                        break;
+                    case "DOWN":
+                        dirZ = 1;
+                        break;
+                    case "UP":
+                        dirZ = -1;
+                        break;
+                }
+                x = sourceNode.x + (dirX * (srcW + INITIAL_SPACING));
+                z = sourceNode.z + (dirZ * (srcH + INITIAL_SPACING));
+                // Collision Avoidance reuse
+                const COLLISION_SPACING = 0.5;
+                const shiftStepX = srcW + COLLISION_SPACING;
+                const shiftStepZ = srcH + COLLISION_SPACING;
+                const newW = width;
+                const newH = height;
+                const checkCollision = (cx, cz) => {
+                    return state.nodes.some(n => {
+                        if (n.layerId !== sourceNode.layerId)
+                            return false;
+                        const dx = Math.abs(n.x - cx);
+                        const dz = Math.abs(n.z - cz);
+                        const otherW = Number(n.width) || 2;
+                        const otherH = Number(n.height) || 1.5;
+                        const combinedHalfWidth = (otherW + newW) / 2;
+                        const combinedHalfHeight = (otherH + newH) / 2;
+                        return dx < combinedHalfWidth && dz < combinedHalfHeight;
+                    });
+                };
+                let iterations = 0;
+                while (checkCollision(x, z) && iterations < 50) {
+                    iterations++;
+                    if (dirX !== 0)
+                        z += shiftStepZ;
+                    else
+                        x += shiftStepX;
+                }
+            }
+            else if (state.nodes.length > 0) {
                 const lastNode = state.nodes[state.nodes.length - 1];
                 x = lastNode.x + 5;
                 z = lastNode.z;
@@ -585,12 +645,83 @@ async function handleToolCall(name, args) {
                 }
             };
             state.nodes.push(newNode);
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, state);
+            // Auto-Connect logic
+            if (args.connection && args.referenceNodeId) {
+                let termStart = 'none';
+                let termEnd = 'none';
+                let borderStyle = 'solid';
+                // UML Mapping
+                switch (args.connection.type) {
+                    case 'inheritance':
+                        termEnd = 'triangle-empty';
+                        break;
+                    case 'implementation':
+                        termEnd = 'triangle-empty';
+                        borderStyle = 'dashed';
+                        break;
+                    case 'dependency':
+                        termEnd = 'arrow';
+                        borderStyle = 'dashed';
+                        break;
+                    case 'association':
+                        termEnd = 'arrow';
+                        break;
+                    case 'aggregation':
+                        termStart = 'diamond-empty';
+                        break;
+                    case 'composition':
+                        termStart = 'diamond-filled';
+                        break;
+                }
+                // Points? Simple defaults based on direction
+                let sp = 0;
+                let tp = 0;
+                switch (args.direction) {
+                    case "RIGHT":
+                        sp = 1;
+                        tp = 3;
+                        break;
+                    case "LEFT":
+                        sp = 3;
+                        tp = 1;
+                        break;
+                    case "DOWN":
+                        sp = 2;
+                        tp = 0;
+                        break;
+                    case "UP":
+                        sp = 0;
+                        tp = 2;
+                        break;
+                    default:
+                        sp = 2;
+                        tp = 0;
+                        break;
+                }
+                const newEdge = {
+                    id: randomUUID(),
+                    sourceId: args.referenceNodeId,
+                    targetId: newNodeId,
+                    sourcePointIndex: sp,
+                    targetPointIndex: tp,
+                    label: args.connection.label,
+                    style: 'line',
+                    routingType: 'orthogonal',
+                    color: '#000000',
+                    thickness: 0.01,
+                    fontSize: 20,
+                    borderStyle: borderStyle,
+                    terminationStart: termStart,
+                    terminationEnd: termEnd
+                };
+                state.edges.push(newEdge);
+            }
+            await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Created UML Class '${args.className}'` }] };
         }
         case "create_db_table": {
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
-            const newNodeId = args.id || (0, crypto_1.randomUUID)();
+            const state = await readDiagramFile(args.filePath);
+            const newNodeId = args.id || randomUUID();
             const cols = args.columns || [];
             // Header + items
             const estimatedHeight = 0.8 + cols.length * 0.4;
@@ -720,7 +851,7 @@ async function handleToolCall(name, args) {
                         break;
                 }
                 const newEdge = {
-                    id: (0, crypto_1.randomUUID)(),
+                    id: randomUUID(),
                     sourceId: args.referenceNodeId,
                     targetId: newNodeId,
                     sourcePointIndex: sp,
@@ -737,11 +868,11 @@ async function handleToolCall(name, args) {
                 };
                 state.edges.push(newEdge);
             }
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, state);
+            await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Created DB Table '${args.tableName}'` }] };
         }
         case "connect_entities": {
-            const state = await (0, fileHandler_1.readDiagramFile)(args.filePath);
+            const state = await readDiagramFile(args.filePath);
             // Find nodes by Label (tolerant search?)
             const sourceNode = state.nodes.find(n => n.label === args.sourceName);
             const targetNode = state.nodes.find(n => n.label === args.targetName);
@@ -792,7 +923,7 @@ async function handleToolCall(name, args) {
                     break;
             }
             const newEdge = {
-                id: (0, crypto_1.randomUUID)(),
+                id: randomUUID(),
                 sourceId: sourceNode.id,
                 targetId: targetNode.id,
                 sourcePointIndex: 2, // Bottom of source
@@ -808,7 +939,7 @@ async function handleToolCall(name, args) {
                 terminationEnd: terminationEnd
             };
             state.edges.push(newEdge);
-            await (0, fileHandler_1.saveDiagramFile)(args.filePath, state);
+            await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Connected '${args.sourceName}' to '${args.targetName}' via ${args.relationType}` }] };
         }
         default:

package/dist/src/types.js

@@ -1,2 +1 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
+export {};

package/dist/src/utils/fileHandler.js

@@ -1,16 +1,9 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.readDiagramFile = readDiagramFile;
-exports.saveDiagramFile = saveDiagramFile;
-const promises_1 = __importDefault(require("fs/promises"));
-const jszip_1 = __importDefault(require("jszip"));
-async function readDiagramFile(filePath) {
+import fs from 'fs/promises';
+import JSZip from 'jszip';
+export async function readDiagramFile(filePath) {
     try {
-        const fileContent = await promises_1.default.readFile(filePath);
-        const zip = await jszip_1.default.loadAsync(fileContent);
+        const fileContent = await fs.readFile(filePath);
+        const zip = await JSZip.loadAsync(fileContent);
         const diagramFile = zip.file("diagram.json");
         if (!diagramFile) {
             throw new Error("Invalid .DoDraw file: diagram.json missing");
@@ -25,17 +18,17 @@ async function readDiagramFile(filePath) {
         throw error;
     }
 }
-async function saveDiagramFile(filePath, state) {
+export async function saveDiagramFile(filePath, state) {
     let zip;
     // Try to load existing zip to preserve other files (e.g. images)
     try {
-        const fileContent = await promises_1.default.readFile(filePath);
-        zip = await jszip_1.default.loadAsync(fileContent);
+        const fileContent = await fs.readFile(filePath);
+        zip = await JSZip.loadAsync(fileContent);
     }
     catch (error) {
         if (error.code === 'ENOENT') {
             // Create new zip if file doesn't exist
-            zip = new jszip_1.default();
+            zip = new JSZip();
         }
         else {
             throw error;
@@ -49,5 +42,5 @@ async function saveDiagramFile(filePath, state) {
         mimeType: "application/zip",
         compression: "DEFLATE"
     });
-    await promises_1.default.writeFile(filePath, content);
+    await fs.writeFile(filePath, content);
 }

package/dist/test/test-auto-placement.js

@@ -1,23 +1,18 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const path_1 = __importDefault(require("path"));
-const process_1 = __importDefault(require("process"));
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import path from "path";
+import process from "process";
 async function main() {
     console.log("Connecting to server...");
-    const transport = new stdio_js_1.StdioClientTransport({
+    const transport = new StdioClientTransport({
         command: "node",
-        args: [path_1.default.join(process_1.default.cwd(), "dist", "src", "index.js")],
+        args: [path.join(process.cwd(), "dist", "src", "index.js")],
     });
-    const client = new index_js_1.Client({ name: "dodraw-test-client", version: "1.0.0" }, { capabilities: {} });
+    const client = new Client({ name: "dodraw-test-client", version: "1.0.0" }, { capabilities: {} });
     await client.connect(transport);
     console.log("Connected.");
     try {
-        const filePath = path_1.default.join(process_1.default.cwd(), "test_output", "test_autoplacement.3duml");
+        const filePath = path.join(process.cwd(), "test_output", "test_autoplacement.3duml");
         await client.callTool({ name: "create_new_diagram", arguments: { filePath } });
         // 1. Add First Node (auto) -> Should be 0,0
         console.log("Adding Node A (auto)...");

package/dist/test/test-client.js

@@ -1,19 +1,14 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const path_1 = __importDefault(require("path"));
-const fs_1 = __importDefault(require("fs"));
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import path from "path";
+import fs from "fs";
 async function main() {
-    const serverPath = path_1.default.join(process.cwd(), "dist/src/index.js");
-    const transport = new stdio_js_1.StdioClientTransport({
+    const serverPath = path.join(process.cwd(), "dist/src/index.js");
+    const transport = new StdioClientTransport({
         command: "node",
         args: [serverPath]
     });
-    const client = new index_js_1.Client({
+    const client = new Client({
         name: "test-client",
         version: "1.0.0",
     }, {
@@ -26,14 +21,14 @@ async function main() {
     const tools = await client.listTools();
     console.log("Tools found:", tools.tools.map(t => t.name).join(", "));
     // Create diagram
-    const testDir = path_1.default.join(__dirname, "../../test_output");
-    if (!fs_1.default.existsSync(testDir)) {
-        fs_1.default.mkdirSync(testDir);
+    const testDir = path.join(__dirname, "../../test_output");
+    if (!fs.existsSync(testDir)) {
+        fs.mkdirSync(testDir);
     }
-    const filePath = path_1.default.join(testDir, "test_diagram.3duml");
+    const filePath = path.join(testDir, "test_diagram.3duml");
     // Clean up previous run
-    if (fs_1.default.existsSync(filePath)) {
-        fs_1.default.unlinkSync(filePath);
+    if (fs.existsSync(filePath)) {
+        fs.unlinkSync(filePath);
     }
     console.log(`Creating diagram at ${filePath}...`);
     await client.callTool({

package/dist/test/test-collision.js

@@ -1,22 +1,17 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const path_1 = __importDefault(require("path"));
-const process_1 = __importDefault(require("process"));
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import path from "path";
+import process from "process";
 async function main() {
     console.log("Connecting...");
-    const transport = new stdio_js_1.StdioClientTransport({
+    const transport = new StdioClientTransport({
         command: "node",
-        args: [path_1.default.join(process_1.default.cwd(), "dist", "src", "index.js")],
+        args: [path.join(process.cwd(), "dist", "src", "index.js")],
     });
-    const client = new index_js_1.Client({ name: "test-collision", version: "1.0" }, { capabilities: {} });
+    const client = new Client({ name: "test-collision", version: "1.0" }, { capabilities: {} });
     await client.connect(transport);
     try {
-        const filePath = path_1.default.join(process_1.default.cwd(), "test_output", "test_collision.3duml");
+        const filePath = path.join(process.cwd(), "test_output", "test_collision.3duml");
         await client.callTool({ name: "create_new_diagram", arguments: { filePath } });
         // 1. Add Root Node
         console.log("Adding Root Node...");
@@ -75,7 +70,7 @@ async function main() {
     }
     catch (e) {
         console.error("FAIL:", e);
-        process_1.default.exit(1);
+        process.exit(1);
     }
     finally {
         await client.close();

package/dist/test/test-constraints.js

@@ -1,22 +1,17 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const path_1 = __importDefault(require("path"));
-const process_1 = __importDefault(require("process"));
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import path from "path";
+import process from "process";
 async function main() {
     console.log("Connecting...");
-    const transport = new stdio_js_1.StdioClientTransport({
+    const transport = new StdioClientTransport({
         command: "node",
-        args: [path_1.default.join(process_1.default.cwd(), "dist", "src", "index.js")],
+        args: [path.join(process.cwd(), "dist", "src", "index.js")],
     });
-    const client = new index_js_1.Client({ name: "test-constraints", version: "1.0" }, { capabilities: {} });
+    const client = new Client({ name: "test-constraints", version: "1.0" }, { capabilities: {} });
     await client.connect(transport);
     try {
-        const filePath = path_1.default.join(process_1.default.cwd(), "test_output", "test_constraints.3duml");
+        const filePath = path.join(process.cwd(), "test_output", "test_constraints.3duml");
         await client.callTool({ name: "create_new_diagram", arguments: { filePath } });
         // 1. Test Layer Auto-Stacking
         console.log("Adding Layer 2 (auto)...");

package/dist/test/test-debug.js

@@ -1,22 +1,17 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const path_1 = __importDefault(require("path"));
-const process_1 = __importDefault(require("process"));
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import path from "path";
+import process from "process";
 async function main() {
     console.log("Connecting...");
-    const transport = new stdio_js_1.StdioClientTransport({
+    const transport = new StdioClientTransport({
         command: "node",
-        args: [path_1.default.join(process_1.default.cwd(), "dist", "src", "index.js")],
+        args: [path.join(process.cwd(), "dist", "src", "index.js")],
     });
-    const client = new index_js_1.Client({ name: "debug", version: "1.0" }, { capabilities: {} });
+    const client = new Client({ name: "debug", version: "1.0" }, { capabilities: {} });
     await client.connect(transport);
     try {
-        const filePath = path_1.default.join(process_1.default.cwd(), "test_output", "test_debug.3duml");
+        const filePath = path.join(process.cwd(), "test_output", "test_debug.3duml");
         await client.callTool({ name: "create_new_diagram", arguments: { filePath } });
         console.log("Adding Node C...");
         await client.callTool({

package/dist/test/test-directional.js

@@ -1,22 +1,17 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const path_1 = __importDefault(require("path"));
-const process_1 = __importDefault(require("process"));
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import path from "path";
+import process from "process";
 async function main() {
     console.log("Connecting...");
-    const transport = new stdio_js_1.StdioClientTransport({
+    const transport = new StdioClientTransport({
         command: "node",
-        args: [path_1.default.join(process_1.default.cwd(), "dist", "src", "index.js")],
+        args: [path.join(process.cwd(), "dist", "src", "index.js")],
     });
-    const client = new index_js_1.Client({ name: "test-directional", version: "1.0" }, { capabilities: {} });
+    const client = new Client({ name: "test-directional", version: "1.0" }, { capabilities: {} });
     await client.connect(transport);
     try {
-        const filePath = path_1.default.join(process_1.default.cwd(), "test_output", "test_directional.3duml");
+        const filePath = path.join(process.cwd(), "test_output", "test_directional.3duml");
         await client.callTool({ name: "create_new_diagram", arguments: { filePath } });
         // 1. Add Root Node
         console.log("Adding Root Node...");

package/dist/test/test-layer-support.js

@@ -1,23 +1,18 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const path_1 = __importDefault(require("path"));
-const process_1 = __importDefault(require("process"));
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import path from "path";
+import process from "process";
 async function main() {
     console.log("Connecting to server...");
-    const transport = new stdio_js_1.StdioClientTransport({
+    const transport = new StdioClientTransport({
         command: "node",
-        args: [path_1.default.join(process_1.default.cwd(), "dist", "src", "index.js")],
+        args: [path.join(process.cwd(), "dist", "src", "index.js")],
     });
-    const client = new index_js_1.Client({ name: "dodraw-test-client", version: "1.0.0" }, { capabilities: {} });
+    const client = new Client({ name: "dodraw-test-client", version: "1.0.0" }, { capabilities: {} });
     await client.connect(transport);
     console.log("Connected.");
     try {
-        const filePath = path_1.default.join(process_1.default.cwd(), "test_output", "test_layers.3duml");
+        const filePath = path.join(process.cwd(), "test_output", "test_layers.3duml");
         // 1. Create Diagram
         await client.callTool({ name: "create_new_diagram", arguments: { filePath } });
         console.log("Created diagram.");

package/dist/test/test-refined-spacing.js

@@ -1,22 +1,17 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
-const path_1 = __importDefault(require("path"));
-const process_1 = __importDefault(require("process"));
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import path from "path";
+import process from "process";
 async function main() {
     console.log("Connecting...");
-    const transport = new stdio_js_1.StdioClientTransport({
+    const transport = new StdioClientTransport({
         command: "node",
-        args: [path_1.default.join(process_1.default.cwd(), "dist", "src", "index.js")],
+        args: [path.join(process.cwd(), "dist", "src", "index.js")],
     });
-    const client = new index_js_1.Client({ name: "test-refined", version: "1.0" }, { capabilities: {} });
+    const client = new Client({ name: "test-refined", version: "1.0" }, { capabilities: {} });
     await client.connect(transport);
     try {
-        const filePath = path_1.default.join(process_1.default.cwd(), "test_output", "test_refined_spacing.3duml");
+        const filePath = path.join(process.cwd(), "test_output", "test_refined_spacing.3duml");
         await client.callTool({ name: "create_new_diagram", arguments: { filePath } });
         // Default Layer 1 is at Y=0
         // 1. Add Layer 2 (auto) -> Should be -2

package/dist/test/verify_types.js

@@ -1,15 +1,10 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const promises_1 = __importDefault(require("fs/promises"));
-const jszip_1 = __importDefault(require("jszip"));
+import fs from 'fs/promises';
+import JSZip from 'jszip';
 async function checkFile(filePath, label) {
     try {
         console.log(`Checking ${label}: ${filePath}`);
-        const content = await promises_1.default.readFile(filePath);
-        const zip = await jszip_1.default.loadAsync(content);
+        const content = await fs.readFile(filePath);
+        const zip = await JSZip.loadAsync(content);
         const jsonStr = await zip.file("diagram.json")?.async("string");
         if (!jsonStr)
             throw new Error("No diagram.json");

package/package.json

@@ -1,7 +1,8 @@
 {
   "name": "dodraw-mcp-server",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "MCP server for DoDraw",
+  "type": "module",
   "main": "dist/src/index.js",
   "bin": {
     "dodraw-mcp-server": "dist/src/index.js"

package/src/create_class_diagram_example.ts

@@ -1,5 +1,5 @@
 
-import { handleToolCall } from './tools/diagramTools';
+import { handleToolCall } from './tools/diagramTools.js';
 
 async function createClassDiagram() {
     const FILE_PATH = "class_diagram_example.3duml";

package/src/create_termination_reference.ts

@@ -1,5 +1,5 @@
 
-import { handleToolCall } from './tools/diagramTools';
+import { handleToolCall } from './tools/diagramTools.js';
 
 interface TerminationType {
     id: string;

package/src/tools/diagramTools.ts

@@ -1,7 +1,7 @@
 
 import { CallToolRequestSchema, ListToolsRequestSchema, Tool } from "@modelcontextprotocol/sdk/types.js";
-import { readDiagramFile, saveDiagramFile } from "../utils/fileHandler";
-import { DiagramState, NodeData, EdgeData } from "../types";
+import { readDiagramFile, saveDiagramFile } from "../utils/fileHandler.js";
+import { DiagramState, NodeData, EdgeData } from "../types.js";
 import { randomUUID } from "crypto";
 
 export const toolDefinitions: Tool[] = [
@@ -175,7 +175,7 @@ export const toolDefinitions: Tool[] = [
   },
   {
     name: "create_uml_class",
-    description: "Create a UML Class node. Use this for the FIRST class. For connected classes, use 'add_directional_node' to ensure correct relative positioning.",
+    description: "Create a UML Class node. Supports absolute positioning (x,y) OR relative positioning (referenceNodeId, direction). Can also auto-connect.",
     inputSchema: {
       type: "object",
       properties: {
@@ -183,9 +183,19 @@ export const toolDefinitions: Tool[] = [
         className: { type: "string" },
         attributes: { type: "array", items: { type: "string" }, description: "List of attributes e.g. ['- id: int', '+ name: string']" },
         methods: { type: "array", items: { type: "string" }, description: "List of methods e.g. ['+ getName(): string', '+ save(): void']" },
-        x: { type: "number", description: "Optional X position" },
-        y: { type: "number", description: "Optional Y position (mapped to Z in 3DUML)" },
-        id: { type: "string", description: "Optional explicit ID" }
+        id: { type: "string", description: "Optional explicit ID" },
+        referenceNodeId: { type: "string", description: "Optional. ID of an existing node to place this new class relative to." },
+        direction: { type: "string", enum: ["UP", "DOWN", "LEFT", "RIGHT"], description: "Optional. Direction to place relative to referenceNodeId." },
+        connection: { 
+             type: "object",
+             description: "Optional. If provided, creates an edge connecting the reference node to this new class.",
+             properties: {
+                 type: { type: "string", enum: ["association", "inheritance", "implementation", "dependency", "aggregation", "composition"], description: "Relationship type" },
+                 label: { type: "string" },
+                 direction: { type: "string", enum: ["source-to-target", "target-to-source", "bi-directional"], description: "Edge direction flow" }
+             },
+             required: ["type"]
+        }
       },
       required: ["filePath", "className"]
     }
@@ -578,11 +588,58 @@ export async function handleToolCall(name: string, args: any): Promise<any> {
             const estimatedHeight = 0.8 + (attrCount + methodCount) * 0.4; 
             const height = Math.max(2, estimatedHeight);
 
-            // Auto-placement logic if X/Y not provided
-            let x = args.x !== undefined ? Number(args.x) : 0;
-            let z = args.y !== undefined ? Number(args.y) : 0;
+            // Auto-placement logic (defaults to 0,0 for first node)
+            let x = 0;
+            let z = 0;
             
-            if (args.x === undefined && args.y === undefined && state.nodes.length > 0) {
+            // Relative Placement Logic
+            if (args.referenceNodeId && args.direction) {
+                const sourceNode = state.nodes.find(n => n.id === args.referenceNodeId);
+                if (!sourceNode) throw new Error(`Reference node ${args.referenceNodeId} not found`);
+
+                const INITIAL_SPACING = 2.0;
+                let dirX = 0;
+                let dirZ = 0;
+                const srcW = Number(sourceNode.width) || 2;
+                const srcH = Number(sourceNode.height) || 1.5;
+
+                switch (args.direction) {
+                    case "RIGHT": dirX = 1; break;
+                    case "LEFT": dirX = -1; break;
+                    case "DOWN": dirZ = 1; break;
+                    case "UP": dirZ = -1; break;
+                }
+
+                x = sourceNode.x + (dirX * (srcW + INITIAL_SPACING));
+                z = sourceNode.z + (dirZ * (srcH + INITIAL_SPACING));
+
+                // Collision Avoidance reuse
+                const COLLISION_SPACING = 0.5;
+                const shiftStepX = srcW + COLLISION_SPACING;
+                const shiftStepZ = srcH + COLLISION_SPACING;
+                const newW = width;
+                const newH = height;
+
+                const checkCollision = (cx: number, cz: number) => {
+                    return state.nodes.some(n => {
+                        if (n.layerId !== sourceNode.layerId) return false;
+                        const dx = Math.abs(n.x - cx);
+                        const dz = Math.abs(n.z - cz);
+                        const otherW = Number(n.width) || 2;
+                        const otherH = Number(n.height) || 1.5;
+                        const combinedHalfWidth = (otherW + newW) / 2;
+                        const combinedHalfHeight = (otherH + newH) / 2;
+                        return dx < combinedHalfWidth && dz < combinedHalfHeight;
+                    });
+                };
+
+                let iterations = 0;
+                 while (checkCollision(x, z) && iterations < 50) {
+                    iterations++;
+                    if (dirX !== 0) z += shiftStepZ;
+                    else x += shiftStepX;
+                }
+            } else if (state.nodes.length > 0) {
                  const lastNode = state.nodes[state.nodes.length - 1];
                  x = lastNode.x + 5;
                  z = lastNode.z;
@@ -606,6 +663,53 @@ export async function handleToolCall(name: string, args: any): Promise<any> {
             };
             
             state.nodes.push(newNode);
+
+            // Auto-Connect logic
+            if (args.connection && args.referenceNodeId) {
+                let termStart = 'none';
+                let termEnd = 'none';
+                let borderStyle = 'solid';
+
+                // UML Mapping
+                switch (args.connection.type) {
+                    case 'inheritance': termEnd = 'triangle-empty'; break;
+                    case 'implementation': termEnd = 'triangle-empty'; borderStyle = 'dashed'; break;
+                    case 'dependency': termEnd = 'arrow'; borderStyle = 'dashed'; break;
+                    case 'association': termEnd = 'arrow'; break;
+                    case 'aggregation': termStart = 'diamond-empty'; break;
+                    case 'composition': termStart = 'diamond-filled'; break;
+                }
+
+                // Points? Simple defaults based on direction
+                let sp = 0; 
+                let tp = 0;
+                switch(args.direction) {
+                    case "RIGHT": sp=1; tp=3; break;
+                    case "LEFT": sp=3; tp=1; break;
+                    case "DOWN": sp=2; tp=0; break;
+                    case "UP": sp=0; tp=2; break;
+                    default: sp=2; tp=0; break; 
+                }
+
+                const newEdge: EdgeData = {
+                    id: randomUUID(),
+                    sourceId: args.referenceNodeId,
+                    targetId: newNodeId,
+                    sourcePointIndex: sp,
+                    targetPointIndex: tp,
+                    label: args.connection.label,
+                    style: 'line',
+                    routingType: 'orthogonal',
+                    color: '#000000',
+                    thickness: 0.01,
+                    fontSize: 20,
+                    borderStyle: borderStyle as any,
+                    terminationStart: termStart as any,
+                    terminationEnd: termEnd as any
+                };
+                state.edges.push(newEdge);
+            }
+
             await saveDiagramFile(args.filePath, state);
             return { content: [{ type: "text", text: `Created UML Class '${args.className}'` }] };
         }

package/src/utils/fileHandler.ts

@@ -1,7 +1,7 @@
 
 import fs from 'fs/promises';
 import JSZip from 'jszip';
-import { DiagramState } from '../types';
+import { DiagramState } from '../types.js';
 
 export async function readDiagramFile(filePath: string): Promise<DiagramState> {
   try {

package/tsconfig.json

@@ -1,7 +1,8 @@
 {
   "compilerOptions": {
-    "target": "ES2020",
-    "module": "commonjs",
+    "target": "ES2022",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
     "outDir": "./dist",
     "rootDir": ".",
     "strict": true,