@arvoretech/db-diagram-mcp 1.0.0

package/dist/parser.test.js

@@ -0,0 +1,110 @@
+import { describe, it, expect } from "vitest";
+import { parseDDL, extractInlineForeignKeys } from "./parser.js";
+const sampleDDL = `
+CREATE TABLE users (
+  id INT NOT NULL AUTO_INCREMENT,
+  email VARCHAR(255) NOT NULL,
+  name VARCHAR(100),
+  created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+  PRIMARY KEY (id),
+  UNIQUE KEY (email)
+);
+
+CREATE TABLE posts (
+  id INT NOT NULL AUTO_INCREMENT,
+  user_id INT NOT NULL,
+  title VARCHAR(255) NOT NULL,
+  body TEXT,
+  published_at TIMESTAMP,
+  PRIMARY KEY (id),
+  FOREIGN KEY (user_id) REFERENCES users(id)
+);
+
+CREATE TABLE comments (
+  id INT NOT NULL AUTO_INCREMENT,
+  post_id INT NOT NULL,
+  user_id INT NOT NULL,
+  content TEXT NOT NULL,
+  PRIMARY KEY (id),
+  FOREIGN KEY (post_id) REFERENCES posts(id),
+  FOREIGN KEY (user_id) REFERENCES users(id)
+);
+
+CREATE TABLE tags (
+  id INT NOT NULL AUTO_INCREMENT,
+  name VARCHAR(50) NOT NULL UNIQUE,
+  PRIMARY KEY (id)
+);
+
+CREATE TABLE post_tags (
+  post_id INT NOT NULL,
+  tag_id INT NOT NULL,
+  PRIMARY KEY (post_id, tag_id),
+  FOREIGN KEY (post_id) REFERENCES posts(id),
+  FOREIGN KEY (tag_id) REFERENCES tags(id)
+);
+`;
+describe("parseDDL", () => {
+    it("parses all tables", () => {
+        const schema = parseDDL(sampleDDL);
+        expect(schema.tables).toHaveLength(5);
+        expect(schema.tables.map((t) => t.name)).toEqual([
+            "users",
+            "posts",
+            "comments",
+            "tags",
+            "post_tags",
+        ]);
+    });
+    it("parses columns correctly", () => {
+        const schema = parseDDL(sampleDDL);
+        const users = schema.tables.find((t) => t.name === "users");
+        expect(users.columns).toHaveLength(4);
+        const idCol = users.columns.find((c) => c.name === "id");
+        expect(idCol.primaryKey).toBe(true);
+        expect(idCol.nullable).toBe(false);
+        const emailCol = users.columns.find((c) => c.name === "email");
+        expect(emailCol.nullable).toBe(false);
+    });
+    it("parses foreign keys", () => {
+        const schema = parseDDL(sampleDDL);
+        const posts = schema.tables.find((t) => t.name === "posts");
+        expect(posts.foreignKeys).toHaveLength(1);
+        expect(posts.foreignKeys[0]).toEqual({
+            columns: ["user_id"],
+            referencedTable: "users",
+            referencedColumns: ["id"],
+        });
+    });
+    it("parses composite primary keys", () => {
+        const schema = parseDDL(sampleDDL);
+        const postTags = schema.tables.find((t) => t.name === "post_tags");
+        const pkCols = postTags.columns.filter((c) => c.primaryKey);
+        expect(pkCols.map((c) => c.name)).toEqual(["post_id", "tag_id"]);
+    });
+    it("parses multiple foreign keys on one table", () => {
+        const schema = parseDDL(sampleDDL);
+        const comments = schema.tables.find((t) => t.name === "comments");
+        expect(comments.foreignKeys).toHaveLength(2);
+    });
+    it("handles PostgreSQL inline references", () => {
+        const pgDDL = `
+CREATE TABLE orders (
+  id SERIAL PRIMARY KEY,
+  customer_id INT NOT NULL REFERENCES customers(id),
+  total DECIMAL(10,2)
+);
+
+CREATE TABLE customers (
+  id SERIAL PRIMARY KEY,
+  name VARCHAR(100) NOT NULL
+);
+`;
+        let schema = parseDDL(pgDDL);
+        schema = extractInlineForeignKeys(pgDDL, schema);
+        const orders = schema.tables.find((t) => t.name === "orders");
+        expect(orders.foreignKeys).toHaveLength(1);
+        expect(orders.foreignKeys[0].referencedTable).toBe("customers");
+    });
+});
+//# sourceMappingURL=parser.test.js.map

package/dist/parser.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parser.test.js","sourceRoot":"","sources":["../src/parser.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AAEjE,MAAM,SAAS,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2CjB,CAAC;AAEF,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAC3B,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,OAAO;YACP,OAAO;YACP,UAAU;YACV,MAAM;YACN,WAAW;SACZ,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAE,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEtC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAE,CAAC;QAC1D,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEnC,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAE,CAAC;QAChE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;QAC7B,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAE,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACnC,OAAO,EAAE,CAAC,SAAS,CAAC;YACpB,eAAe,EAAE,OAAO;YACxB,iBAAiB,EAAE,CAAC,IAAI,CAAC;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAE,CAAC;QACpE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAE,CAAC;QACnE,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,KAAK,GAAG;;;;;;;;;;;CAWjB,CAAC;QACE,IAAI,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC7B,MAAM,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAE,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/server.d.ts

@@ -0,0 +1,8 @@
+export declare class DbDiagramMCPServer {
+    private server;
+    constructor();
+    private setupTools;
+    start(): Promise<void>;
+    setupGracefulShutdown(): void;
+}
+//# sourceMappingURL=server.d.ts.map

package/dist/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAmBA,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,MAAM,CAAY;;IAU1B,OAAO,CAAC,UAAU;IAqHZ,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAM5B,qBAAqB,IAAI,IAAI;CAS9B"}

package/dist/server.js

@@ -0,0 +1,117 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { parseDDL, extractInlineForeignKeys } from "./parser.js";
+import { generateErd, generateDomainMap, explainTable, traceFlow } from "./mermaid.js";
+import { visualize } from "./visualizer.js";
+import { GenerateErdParamsSchema, GenerateDomainMapParamsSchema, ExplainTableParamsSchema, TraceFlowParamsSchema, VisualizeParamsSchema, } from "./types.js";
+export class DbDiagramMCPServer {
+    server;
+    constructor() {
+        this.server = new McpServer({
+            name: "db-diagram-mcp-server",
+            version: "1.0.0",
+        });
+        this.setupTools();
+    }
+    setupTools() {
+        this.server.registerTool("generate_erd", {
+            title: "Generate ERD",
+            description: "Generate a Mermaid ER diagram from SQL DDL. Optionally filter to specific tables.",
+            inputSchema: {
+                ddl: GenerateErdParamsSchema.shape.ddl,
+                tables: GenerateErdParamsSchema.shape.tables,
+                title: GenerateErdParamsSchema.shape.title,
+            },
+        }, async (params) => {
+            const { ddl, tables, title } = params;
+            let schema = parseDDL(ddl);
+            schema = extractInlineForeignKeys(ddl, schema);
+            const diagram = generateErd(schema, { tables, title });
+            return { content: [{ type: "text", text: diagram }] };
+        });
+        this.server.registerTool("generate_domain_map", {
+            title: "Generate Domain Map",
+            description: "Generate a Mermaid diagram showing all tables related to an entry table within N hops.",
+            inputSchema: {
+                ddl: GenerateDomainMapParamsSchema.shape.ddl,
+                entryTable: GenerateDomainMapParamsSchema.shape.entryTable,
+                depth: GenerateDomainMapParamsSchema.shape.depth,
+            },
+        }, async (params) => {
+            const { ddl, entryTable, depth } = params;
+            let schema = parseDDL(ddl);
+            schema = extractInlineForeignKeys(ddl, schema);
+            const diagram = generateDomainMap(schema, entryTable, depth ?? 3);
+            return { content: [{ type: "text", text: diagram }] };
+        });
+        this.server.registerTool("explain_table", {
+            title: "Explain Table",
+            description: "Show a table's columns, relationships (incoming and outgoing), and a mini ER diagram of its neighborhood.",
+            inputSchema: {
+                ddl: ExplainTableParamsSchema.shape.ddl,
+                table: ExplainTableParamsSchema.shape.table,
+            },
+        }, async (params) => {
+            const { ddl, table } = params;
+            let schema = parseDDL(ddl);
+            schema = extractInlineForeignKeys(ddl, schema);
+            const explanation = explainTable(schema, table);
+            return { content: [{ type: "text", text: explanation }] };
+        });
+        this.server.registerTool("trace_flow", {
+            title: "Trace Flow",
+            description: "Find relationship paths between two tables and generate a Mermaid diagram of the connecting tables.",
+            inputSchema: {
+                ddl: TraceFlowParamsSchema.shape.ddl,
+                from: TraceFlowParamsSchema.shape.from,
+                to: TraceFlowParamsSchema.shape.to,
+            },
+        }, async (params) => {
+            const { ddl, from, to } = params;
+            let schema = parseDDL(ddl);
+            schema = extractInlineForeignKeys(ddl, schema);
+            const flow = traceFlow(schema, from, to);
+            return { content: [{ type: "text", text: flow }] };
+        });
+        this.server.registerTool("visualize", {
+            title: "Visualize Diagram",
+            description: "Render a Mermaid ER diagram in the browser. Accepts DDL (generates ERD) or raw Mermaid code.",
+            inputSchema: {
+                ddl: VisualizeParamsSchema.shape.ddl,
+                mermaid: VisualizeParamsSchema.shape.mermaid,
+                tables: VisualizeParamsSchema.shape.tables,
+                title: VisualizeParamsSchema.shape.title,
+            },
+        }, async (params) => {
+            const { ddl, mermaid: rawMermaid, tables, title } = params;
+            let mermaidCode;
+            if (rawMermaid) {
+                mermaidCode = rawMermaid;
+            }
+            else if (ddl) {
+                let schema = parseDDL(ddl);
+                schema = extractInlineForeignKeys(ddl, schema);
+                mermaidCode = generateErd(schema, { tables, title });
+            }
+            else {
+                return { content: [{ type: "text", text: "Provide either 'ddl' or 'mermaid' parameter." }] };
+            }
+            const url = await visualize(mermaidCode);
+            return { content: [{ type: "text", text: `Diagram opened at ${url} (auto-closes in 5 minutes)` }] };
+        });
+    }
+    async start() {
+        const transport = new StdioServerTransport();
+        await this.server.connect(transport);
+        console.error("DB Diagram MCP Server started successfully");
+    }
+    setupGracefulShutdown() {
+        const shutdown = async (signal) => {
+            console.error(`Received ${signal}, shutting down gracefully...`);
+            process.exit(0);
+        };
+        process.on("SIGINT", () => shutdown("SIGINT"));
+        process.on("SIGTERM", () => shutdown("SIGTERM"));
+    }
+}
+//# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,QAAQ,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACjE,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACvF,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EACL,uBAAuB,EACvB,6BAA6B,EAC7B,wBAAwB,EACxB,qBAAqB,EACrB,qBAAqB,GAOtB,MAAM,YAAY,CAAC;AAEpB,MAAM,OAAO,kBAAkB;IACrB,MAAM,CAAY;IAE1B;QACE,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC1B,IAAI,EAAE,uBAAuB;YAC7B,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC,MAAM,CAAC,YAAY,CACtB,cAAc,EACd;YACE,KAAK,EAAE,cAAc;YACrB,WAAW,EACT,mFAAmF;YACrF,WAAW,EAAE;gBACX,GAAG,EAAE,uBAAuB,CAAC,KAAK,CAAC,GAAG;gBACtC,MAAM,EAAE,uBAAuB,CAAC,KAAK,CAAC,MAAM;gBAC5C,KAAK,EAAE,uBAAuB,CAAC,KAAK,CAAC,KAAK;aAC3C;SACF,EACD,KAAK,EAAE,MAAM,EAA0B,EAAE;YACvC,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAA2B,CAAC;YAC3D,IAAI,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC3B,MAAM,GAAG,wBAAwB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC/C,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YACvD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QACxD,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,YAAY,CACtB,qBAAqB,EACrB;YACE,KAAK,EAAE,qBAAqB;YAC5B,WAAW,EACT,wFAAwF;YAC1F,WAAW,EAAE;gBACX,GAAG,EAAE,6BAA6B,CAAC,KAAK,CAAC,GAAG;gBAC5C,UAAU,EAAE,6BAA6B,CAAC,KAAK,CAAC,UAAU;gBAC1D,KAAK,EAAE,6BAA6B,CAAC,KAAK,CAAC,KAAK;aACjD;SACF,EACD,KAAK,EAAE,MAAM,EAA0B,EAAE;YACvC,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAiC,CAAC;YACrE,IAAI,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC3B,MAAM,GAAG,wBAAwB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC/C,MAAM,OAAO,GAAG,iBAAiB,CAAC,MAAM,EAAE,UAAU,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;YAClE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QACxD,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,YAAY,CACtB,eAAe,EACf;YACE,KAAK,EAAE,eAAe;YACtB,WAAW,EACT,2GAA2G;YAC7G,WAAW,EAAE;gBACX,GAAG,EAAE,wBAAwB,CAAC,KAAK,CAAC,GAAG;gBACvC,KAAK,EAAE,wBAAwB,CAAC,KAAK,CAAC,KAAK;aAC5C;SACF,EACD,KAAK,EAAE,MAAM,EAA0B,EAAE;YACvC,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,MAA4B,CAAC;YACpD,IAAI,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC3B,MAAM,GAAG,wBAAwB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC/C,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAChD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;QAC5D,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,YAAY,CACtB,YAAY,EACZ;YACE,KAAK,EAAE,YAAY;YACnB,WAAW,EACT,qGAAqG;YACvG,WAAW,EAAE;gBACX,GAAG,EAAE,qBAAqB,CAAC,KAAK,CAAC,GAAG;gBACpC,IAAI,EAAE,qBAAqB,CAAC,KAAK,CAAC,IAAI;gBACtC,EAAE,EAAE,qBAAqB,CAAC,KAAK,CAAC,EAAE;aACnC;SACF,EACD,KAAK,EAAE,MAAM,EAA0B,EAAE;YACvC,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,MAAyB,CAAC;YACpD,IAAI,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC3B,MAAM,GAAG,wBAAwB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC/C,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YACzC,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QACrD,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,YAAY,CACtB,WAAW,EACX;YACE,KAAK,EAAE,mBAAmB;YAC1B,WAAW,EACT,8FAA8F;YAChG,WAAW,EAAE;gBACX,GAAG,EAAE,qBAAqB,CAAC,KAAK,CAAC,GAAG;gBACpC,OAAO,EAAE,qBAAqB,CAAC,KAAK,CAAC,OAAO;gBAC5C,MAAM,EAAE,qBAAqB,CAAC,KAAK,CAAC,MAAM;gBAC1C,KAAK,EAAE,qBAAqB,CAAC,KAAK,CAAC,KAAK;aACzC;SACF,EACD,KAAK,EAAE,MAAM,EAA0B,EAAE;YACvC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAyB,CAAC;YAE9E,IAAI,WAAmB,CAAC;YACxB,IAAI,UAAU,EAAE,CAAC;gBACf,WAAW,GAAG,UAAU,CAAC;YAC3B,CAAC;iBAAM,IAAI,GAAG,EAAE,CAAC;gBACf,IAAI,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;gBAC3B,MAAM,GAAG,wBAAwB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;gBAC/C,WAAW,GAAG,WAAW,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YACvD,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8CAA8C,EAAE,CAAC,EAAE,CAAC;YAC/F,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,CAAC;YACzC,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,GAAG,6BAA6B,EAAE,CAAC,EAAE,CAAC;QACtG,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC7C,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrC,OAAO,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAC9D,CAAC;IAED,qBAAqB;QACnB,MAAM,QAAQ,GAAG,KAAK,EAAE,MAAc,EAAiB,EAAE;YACvD,OAAO,CAAC,KAAK,CAAC,YAAY,MAAM,+BAA+B,CAAC,CAAC;YACjE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC;QAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC/C,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IACnD,CAAC;CACF"}

package/dist/types.d.ts

@@ -0,0 +1,100 @@
+import { z } from "zod";
+export interface Column {
+    name: string;
+    type: string;
+    nullable: boolean;
+    primaryKey: boolean;
+    unique: boolean;
+    defaultValue?: string;
+}
+export interface ForeignKey {
+    columns: string[];
+    referencedTable: string;
+    referencedColumns: string[];
+}
+export interface Table {
+    name: string;
+    columns: Column[];
+    foreignKeys: ForeignKey[];
+}
+export interface Schema {
+    tables: Table[];
+}
+export declare const GenerateErdParamsSchema: z.ZodObject<{
+    ddl: z.ZodString;
+    tables: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    title: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    ddl: string;
+    tables?: string[] | undefined;
+    title?: string | undefined;
+}, {
+    ddl: string;
+    tables?: string[] | undefined;
+    title?: string | undefined;
+}>;
+export type GenerateErdParams = z.infer<typeof GenerateErdParamsSchema>;
+export declare const GenerateDomainMapParamsSchema: z.ZodObject<{
+    ddl: z.ZodString;
+    entryTable: z.ZodString;
+    depth: z.ZodDefault<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    ddl: string;
+    entryTable: string;
+    depth: number;
+}, {
+    ddl: string;
+    entryTable: string;
+    depth?: number | undefined;
+}>;
+export type GenerateDomainMapParams = z.infer<typeof GenerateDomainMapParamsSchema>;
+export declare const ExplainTableParamsSchema: z.ZodObject<{
+    ddl: z.ZodString;
+    table: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    ddl: string;
+    table: string;
+}, {
+    ddl: string;
+    table: string;
+}>;
+export type ExplainTableParams = z.infer<typeof ExplainTableParamsSchema>;
+export declare const TraceFlowParamsSchema: z.ZodObject<{
+    ddl: z.ZodString;
+    from: z.ZodString;
+    to: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    ddl: string;
+    from: string;
+    to: string;
+}, {
+    ddl: string;
+    from: string;
+    to: string;
+}>;
+export type TraceFlowParams = z.infer<typeof TraceFlowParamsSchema>;
+export declare const VisualizeParamsSchema: z.ZodObject<{
+    ddl: z.ZodOptional<z.ZodString>;
+    mermaid: z.ZodOptional<z.ZodString>;
+    tables: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    title: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    ddl?: string | undefined;
+    tables?: string[] | undefined;
+    title?: string | undefined;
+    mermaid?: string | undefined;
+}, {
+    ddl?: string | undefined;
+    tables?: string[] | undefined;
+    title?: string | undefined;
+    mermaid?: string | undefined;
+}>;
+export type VisualizeParams = z.infer<typeof VisualizeParamsSchema>;
+export interface McpToolResult {
+    [x: string]: unknown;
+    content: Array<{
+        type: "text";
+        text: string;
+    }>;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,WAAW,MAAM;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,OAAO,CAAC;IAClB,UAAU,EAAE,OAAO,CAAC;IACpB,MAAM,EAAE,OAAO,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,eAAe,EAAE,MAAM,CAAC;IACxB,iBAAiB,EAAE,MAAM,EAAE,CAAC;CAC7B;AAED,MAAM,WAAW,KAAK;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,WAAW,EAAE,UAAU,EAAE,CAAC;CAC3B;AAED,MAAM,WAAW,MAAM;IACrB,MAAM,EAAE,KAAK,EAAE,CAAC;CACjB;AAED,eAAO,MAAM,uBAAuB;;;;;;;;;;;;EAOlC,CAAC;AAEH,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAExE,eAAO,MAAM,6BAA6B;;;;;;;;;;;;EAWxC,CAAC;AAEH,MAAM,MAAM,uBAAuB,GAAG,CAAC,CAAC,KAAK,CAC3C,OAAO,6BAA6B,CACrC,CAAC;AAEF,eAAO,MAAM,wBAAwB;;;;;;;;;EAGnC,CAAC;AAEH,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAE1E,eAAO,MAAM,qBAAqB;;;;;;;;;;;;EAIhC,CAAC;AAEH,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAEpE,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;EAKhC,CAAC;AAEH,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAIpE,MAAM,WAAW,aAAa;IAC5B,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;IACrB,OAAO,EAAE,KAAK,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;KACd,CAAC,CAAC;CACJ"}

package/dist/types.js

@@ -0,0 +1,37 @@
+import { z } from "zod";
+export const GenerateErdParamsSchema = z.object({
+    ddl: z.string().min(1).describe("SQL DDL statements (CREATE TABLE)"),
+    tables: z
+        .array(z.string())
+        .optional()
+        .describe("Filter to specific tables (omit for all)"),
+    title: z.string().optional().describe("Diagram title"),
+});
+export const GenerateDomainMapParamsSchema = z.object({
+    ddl: z.string().min(1).describe("SQL DDL statements (CREATE TABLE)"),
+    entryTable: z
+        .string()
+        .describe("Starting table to trace relationships from"),
+    depth: z
+        .number()
+        .min(1)
+        .max(10)
+        .default(3)
+        .describe("How many relationship hops to follow (default: 3)"),
+});
+export const ExplainTableParamsSchema = z.object({
+    ddl: z.string().min(1).describe("SQL DDL statements (CREATE TABLE)"),
+    table: z.string().describe("Table name to explain"),
+});
+export const TraceFlowParamsSchema = z.object({
+    ddl: z.string().min(1).describe("SQL DDL statements (CREATE TABLE)"),
+    from: z.string().describe("Source table"),
+    to: z.string().describe("Target table"),
+});
+export const VisualizeParamsSchema = z.object({
+    ddl: z.string().optional().describe("SQL DDL statements (CREATE TABLE). Provide either ddl or mermaid."),
+    mermaid: z.string().optional().describe("Raw Mermaid diagram code to render directly."),
+    tables: z.array(z.string()).optional().describe("Filter to specific tables (only when using ddl)"),
+    title: z.string().optional().describe("Diagram title"),
+});
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AA2BxB,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,mCAAmC,CAAC;IACpE,MAAM,EAAE,CAAC;SACN,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,EAAE;SACV,QAAQ,CAAC,0CAA0C,CAAC;IACvD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;CACvD,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC;IACpD,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,mCAAmC,CAAC;IACpE,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,QAAQ,CAAC,4CAA4C,CAAC;IACzD,KAAK,EAAE,CAAC;SACL,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,OAAO,CAAC,CAAC,CAAC;SACV,QAAQ,CAAC,mDAAmD,CAAC;CACjE,CAAC,CAAC;AAMH,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/C,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,mCAAmC,CAAC;IACpE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;CACpD,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,mCAAmC,CAAC;IACpE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;IACzC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;CACxC,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mEAAmE,CAAC;IACxG,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8C,CAAC;IACvF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iDAAiD,CAAC;IAClG,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;CACvD,CAAC,CAAC"}

package/dist/visualizer.d.ts

@@ -0,0 +1,2 @@
+export declare function visualize(mermaidCode: string): Promise<string>;
+//# sourceMappingURL=visualizer.d.ts.map

package/dist/visualizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"visualizer.d.ts","sourceRoot":"","sources":["../src/visualizer.ts"],"names":[],"mappings":"AAIA,wBAAgB,SAAS,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CA2B9D"}

package/dist/visualizer.js

@@ -0,0 +1,106 @@
+import { createServer } from "node:http";
+import { exec } from "node:child_process";
+import { platform } from "node:os";
+export function visualize(mermaidCode) {
+    return new Promise((resolve, reject) => {
+        const html = buildHtml(mermaidCode);
+        const server = createServer((req, res) => {
+            res.writeHead(200, { "Content-Type": "text/html" });
+            res.end(html);
+        });
+        server.listen(0, "127.0.0.1", () => {
+            const address = server.address();
+            if (!address || typeof address === "string") {
+                reject(new Error("Failed to get server address"));
+                return;
+            }
+            const url = `http://127.0.0.1:${address.port}`;
+            openBrowser(url);
+            setTimeout(() => {
+                server.close();
+            }, 300_000);
+            resolve(url);
+        });
+        server.on("error", reject);
+    });
+}
+function openBrowser(url) {
+    const cmd = platform() === "darwin"
+        ? `open "${url}"`
+        : platform() === "win32"
+            ? `start "${url}"`
+            : `xdg-open "${url}"`;
+    exec(cmd, () => { });
+}
+function buildHtml(mermaidCode) {
+    const escaped = mermaidCode.replace(/</g, "&lt;").replace(/>/g, "&gt;");
+    return `<!DOCTYPE html>
+<html lang="en">
+<head>
+<meta charset="UTF-8">
+<title>DB Diagram</title>
+<style>
+  * { margin: 0; padding: 0; box-sizing: border-box; }
+  html, body { width: 100%; height: 100%; overflow: hidden; background: #1a1a2e; }
+  #viewport { width: 100%; height: 100%; cursor: grab; overflow: hidden; }
+  #viewport.dragging { cursor: grabbing; }
+  #diagram { display: inline-block; padding: 2rem; transform-origin: 0 0; }
+</style>
+</head>
+<body>
+<div id="viewport">
+  <div id="diagram">
+    <pre class="mermaid">${escaped}</pre>
+  </div>
+</div>
+<script type="module">
+  import mermaid from 'https://cdn.jsdelivr.net/npm/mermaid@11/dist/mermaid.esm.min.mjs';
+  mermaid.initialize({ startOnLoad: true, theme: 'dark', er: { useMaxWidth: false, layoutDirection: 'LR' } });
+
+  const viewport = document.getElementById('viewport');
+  const diagram = document.getElementById('diagram');
+  let isPanning = false;
+  let startX = 0, startY = 0;
+  let offsetX = 0, offsetY = 0;
+  let scale = 1;
+
+  function applyTransform() {
+    diagram.style.transform = \`translate(\${offsetX}px, \${offsetY}px) scale(\${scale})\`;
+  }
+
+  viewport.addEventListener('mousedown', (e) => {
+    isPanning = true;
+    startX = e.clientX - offsetX;
+    startY = e.clientY - offsetY;
+    viewport.classList.add('dragging');
+  });
+
+  window.addEventListener('mousemove', (e) => {
+    if (!isPanning) return;
+    offsetX = e.clientX - startX;
+    offsetY = e.clientY - startY;
+    applyTransform();
+  });
+
+  window.addEventListener('mouseup', () => {
+    isPanning = false;
+    viewport.classList.remove('dragging');
+  });
+
+  viewport.addEventListener('wheel', (e) => {
+    e.preventDefault();
+    const delta = e.deltaY > 0 ? 0.9 : 1.1;
+    const newScale = Math.min(Math.max(scale * delta, 0.1), 5);
+    const rect = viewport.getBoundingClientRect();
+    const mx = e.clientX - rect.left;
+    const my = e.clientY - rect.top;
+    offsetX -= (mx - offsetX) * (newScale / scale - 1);
+    offsetY -= (my - offsetY) * (newScale / scale - 1);
+    scale = newScale;
+    applyTransform();
+  }, { passive: false });
+</script>
+</body>
+</html>`;
+}
+//# sourceMappingURL=visualizer.js.map

package/dist/visualizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"visualizer.js","sourceRoot":"","sources":["../src/visualizer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAU,MAAM,WAAW,CAAC;AACjD,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC1C,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAEnC,MAAM,UAAU,SAAS,CAAC,WAAmB;IAC3C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,IAAI,GAAG,SAAS,CAAC,WAAW,CAAC,CAAC;QACpC,MAAM,MAAM,GAAW,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC/C,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,WAAW,EAAE,CAAC,CAAC;YACpD,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE;YACjC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC5C,MAAM,CAAC,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC,CAAC;gBAClD,OAAO;YACT,CAAC;YAED,MAAM,GAAG,GAAG,oBAAoB,OAAO,CAAC,IAAI,EAAE,CAAC;YAC/C,WAAW,CAAC,GAAG,CAAC,CAAC;YAEjB,UAAU,CAAC,GAAG,EAAE;gBACd,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,CAAC,EAAE,OAAO,CAAC,CAAC;YAEZ,OAAO,CAAC,GAAG,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAAC,GAAW;IAC9B,MAAM,GAAG,GACP,QAAQ,EAAE,KAAK,QAAQ;QACrB,CAAC,CAAC,SAAS,GAAG,GAAG;QACjB,CAAC,CAAC,QAAQ,EAAE,KAAK,OAAO;YACtB,CAAC,CAAC,UAAU,GAAG,GAAG;YAClB,CAAC,CAAC,aAAa,GAAG,GAAG,CAAC;IAE5B,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,SAAS,CAAC,WAAmB;IACpC,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAExE,OAAO;;;;;;;;;;;;;;;;2BAgBkB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;QAmD1B,CAAC;AACT,CAAC"}

package/package.json

@@ -0,0 +1,43 @@
+{
+  "name": "@arvoretech/db-diagram-mcp",
+  "version": "1.0.0",
+  "description": "MCP Server that generates Mermaid ER diagrams from SQL DDL",
+  "main": "dist/index.js",
+  "type": "module",
+  "publishConfig": {
+    "access": "public"
+  },
+  "bin": {
+    "db-diagram-mcp": "./dist/index.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsx src/index.ts",
+    "start": "node dist/index.js",
+    "test": "vitest",
+    "test:cov": "vitest --coverage",
+    "lint": "eslint src/**/*.ts",
+    "lint:fix": "eslint src/**/*.ts --fix"
+  },
+  "keywords": [
+    "mcp",
+    "database",
+    "diagram",
+    "erd",
+    "mermaid",
+    "ddl",
+    "ai",
+    "arvore"
+  ],
+  "author": "Arvore",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/arvoreeducacao/arvore-mcp-servers.git",
+    "directory": "packages/db-diagram"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "~1.22.0",
+    "zod": "^3.22.4"
+  }
+}

package/src/index.ts

@@ -0,0 +1,23 @@
+#!/usr/bin/env node
+
+import { DbDiagramMCPServer } from "./server.js";
+
+try {
+  const server = new DbDiagramMCPServer();
+  server.setupGracefulShutdown();
+  await server.start();
+} catch (error) {
+  console.error("Failed to start DB Diagram MCP Server:", error);
+  process.exit(1);
+}
+
+export { DbDiagramMCPServer } from "./server.js";
+export { parseDDL, extractInlineForeignKeys } from "./parser.js";
+export {
+  generateErd,
+  generateDomainMap,
+  explainTable,
+  traceFlow,
+} from "./mermaid.js";
+export { visualize } from "./visualizer.js";
+export * from "./types.js";