callman-mcp 0.5.0 → 0.6.0

package/dist/tools/diagrams.js

@@ -1,19 +1,19 @@
-import { applyDiagramOps, BPMN_GRAMMAR_VERSION, bpmnXmlFromDocument, createEmptyDocument, isMermaidFlowchart, looksLikeBpmn, looksLikeMermaid, MERMAID_GRAMMAR_VERSION, normalizeDocument, parseBpmn, parseMermaid, placeBpmnNodesOps, } from "callman-core";
+import { applyDatabaseEdits, applyDiagramOps, BPMN_GRAMMAR_VERSION, bpmnXmlFromDocument, createEmptyDocument, DATABASE_GRAMMAR_VERSION, DATABASE_LAYOUT_REQUEST, DatabaseEditError, databaseModelFromDocument, databaseModelFromSchema, databaseSchemaView, emptyDatabaseModel, isDatabaseType, isMermaidFlowchart, looksLikeBpmn, looksLikeMermaid, MERMAID_GRAMMAR_VERSION, normalizeDocument, parseBpmn, parseDatabaseModel, parseMermaid, placeBpmnNodesOps, serializeDatabaseModel, } from "callman-core";
 import { z } from "zod";
 import { CallmanApiError, ToolInputError } from "../client/errors.js";
 import { buildDiagramDocument, diagramInputSchema, toToolError } from "../domain/diagram/build.js";
-import { layoutDocument, placeNewNodes } from "../domain/diagram/layout.js";
-import { applyDiagramPatch, diagramPatchOpSchema } from "../domain/diagram/patch.js";
+import { layoutDocument, placeDatabaseNodes, placeNewNodes } from "../domain/diagram/layout.js";
+import { applyDiagramPatch, dbRelationshipInputSchema, dbTableInputSchema, diagramPatchOpSchema } from "../domain/diagram/patch.js";
 import { diagramFullView, diagramGraphView, diagramListView, diagramMermaidView, diagramSourceView } from "../domain/diagram/summarize.js";
 import { resolvedSummary } from "../domain/resolve.js";
 import { normalizeName } from "../lib/ids.js";
 import { sliceWithCursor } from "../lib/pagination.js";
-import { loadDiagrams, resolveDiagram } from "./lookups.js";
-import { confirmArg, cursorArgs, DESTRUCTIVE, dryRunArg, guarded, IDEMPOTENT_WRITE, ok, READ, refArg, requireConfirm, WRITE } from "./shared.js";
+import { diagramFolderPath, diagramFolderRows, loadDiagrams, resolveDiagram, resolveDiagramFolder, workspacesForListing, workspaceTag } from "./lookups.js";
+import { confirmArg, cursorArgs, DESTRUCTIVE, dryRunArg, guarded, IDEMPOTENT_WRITE, ok, READ, refArg, requireConfirm, workspaceArg, WRITE } from "./shared.js";
 const directionArg = z.enum(["DOWN", "RIGHT", "UP", "LEFT"]).optional().describe("Layout flow direction (default DOWN)");
 const sourceArg = z
     .object({
-    format: z.enum(["mermaid", "bpmn"]).describe('"mermaid": Mermaid text; "bpmn": a BPMN 2.0 XML file'),
+    format: z.enum(["mermaid", "bpmn", "database"]).describe('"mermaid": Mermaid text; "bpmn": a BPMN 2.0 XML file; "database": the JSON database model of a Database Diagram (get_diagram view "source" of one, or the desktop\'s schema export)'),
     text: z
         .string()
         .min(1)
@@ -22,7 +22,7 @@ const sourceArg = z
 })
     .strict()
     .optional()
-    .describe("Create the diagram from Mermaid text or BPMN 2.0 XML instead of nodes/edges");
+    .describe("Create the diagram from Mermaid text, BPMN 2.0 XML or a database model instead of nodes/edges");
 // A BPMN diagram keeps its truth in BPMN 2.0 XML: the model is derived from it.
 const bpmnSource = (text) => ({ format: "bpmn", text, diagramType: "bpmn", grammar: BPMN_GRAMMAR_VERSION });
 // Typed bpmn.* nodes → a BPMN diagram: the model becomes XML, the XML becomes
@@ -33,6 +33,83 @@ const bpmnDocumentFromModel = (model, unpositionedNodeIds, base) => {
     const document = applyDiagramOps({ ...base, nodes: [], edges: [], source: undefined }, [{ op: "setSource", source: bpmnSource(xml), resync: true }], { now: now() }).doc;
     return { document, layoutNote: unpositionedNodeIds.length > 0 ? `Written as BPMN 2.0 XML; ${unpositionedNodeIds.length} node(s) placed left to right after their predecessors.` : "Written as BPMN 2.0 XML with the positions given." };
 };
+// A database diagram keeps its truth in the JSON database model.
+const databaseSource = (text) => ({ format: "database", text, diagramType: "database", grammar: DATABASE_GRAMMAR_VERSION, valid: true });
+const DATABASE_DIALECTS = ["postgres", "mysql", "mssql", "oracle", "mongo", "generic"];
+const toolErrorFromDatabaseEdit = (error) => {
+    if (error instanceof DatabaseEditError)
+        throw new ToolInputError(error.code, error.message);
+    throw error;
+};
+// A database model → a database diagram: the model becomes the source, the
+// tables are derived from it and laid out (unrelated tables packed in rows).
+const databaseDocumentFromModel = async (model, base, keepPositions = false) => {
+    const applied = applyDiagramOps({ ...base, ...(keepPositions ? {} : { nodes: [], edges: [] }), source: undefined }, [{ op: "setSource", source: databaseSource(serializeDatabaseModel(model)), resync: true }], { now: now() });
+    let document = applied.doc;
+    if (keepPositions) {
+        const fresh = applied.added.nodes.filter((id) => document.nodes.some((node) => node.id === id));
+        if (fresh.length > 0)
+            document = placeDatabaseNodes(document, fresh);
+        return { document, layoutNote: fresh.length > 0 ? `${fresh.length} new table(s) placed to the right of the existing ones.` : "Positions kept." };
+    }
+    if (document.nodes.length === 0)
+        return { document, layoutNote: "An empty database diagram." };
+    const laid = await layoutDocument(document, { request: DATABASE_LAYOUT_REQUEST });
+    return { document: laid.document, layoutNote: laid.warning ?? `${document.nodes.length} table(s) laid out with ${laid.engine} (relationships left to right).` };
+};
+// Tables + relationships as the create tool takes them → a model.
+const databaseModelFromInput = (tables, relationships, dialect) => {
+    const edits = [
+        ...tables.map((table) => ({ kind: "addTable", table: { ...table, ...(table.columns ? { columns: table.columns.map((column) => ({ ...column, defaultValue: column.defaultValue ?? undefined, comment: column.comment ?? undefined })) } : {}) } })),
+        ...relationships.map((relationship) => ({
+            kind: "addRelationship",
+            ...(relationship.id ? { id: relationship.id } : {}),
+            source: relationship.source,
+            target: relationship.target,
+            ...(relationship.cardinality ? { cardinality: relationship.cardinality } : {}),
+            ...(relationship.kind ? { relationshipKind: relationship.kind } : {}),
+            ...(relationship.name ? { name: relationship.name } : {}),
+            ...(relationship.label ? { label: relationship.label } : {}),
+        })),
+    ];
+    try {
+        return applyDatabaseEdits(emptyDatabaseModel(dialect), edits).model;
+    }
+    catch (error) {
+        return toolErrorFromDatabaseEdit(error);
+    }
+};
+// A database connection by id or name, in one workspace or across all.
+const findDatabaseConnection = async (ctx, workspace, ref) => {
+    const wanted = ref.trim().toLowerCase();
+    const workspaces = await workspacesForListing(ctx, workspace);
+    const matches = [];
+    for (const ws of workspaces) {
+        for (const connection of await ctx.api.listConnections(ws.id, "database")) {
+            if (connection.id === ref.trim() || connection.name.trim().toLowerCase() === wanted)
+                matches.push({ ws, connection });
+        }
+    }
+    if (matches.length === 0)
+        throw new ToolInputError("CONNECTION_NOT_FOUND", `No database connection "${ref}"${workspace ? ` in workspace ${workspace}` : " in your workspaces"}. list_connections shows them (type: database).`);
+    const exact = matches.filter((match) => match.connection.id === ref.trim());
+    if (exact.length === 1)
+        return exact[0];
+    if (matches.length > 1) {
+        throw new ToolInputError("AMBIGUOUS_CONNECTION", `"${ref}" matches ${matches.length} database connections; pass its id or a \`workspace\`.`, {
+            candidates: matches.map((match) => ({ id: match.connection.id, name: match.connection.name, workspace: workspaceTag(match.ws) })),
+        });
+    }
+    return matches[0];
+};
+const schemaQuery = (options) => ({
+    ...(options.schemas && options.schemas.length > 0 ? { schemas: options.schemas.join(",") } : {}),
+    ...(options.tables && options.tables.length > 0 ? { tables: options.tables.join(",") } : {}),
+    ...(options.includeViews !== undefined ? { includeViews: String(options.includeViews) } : {}),
+    ...(options.objectsOnly ? { objectsOnly: "true" } : {}),
+    ...(options.maxTables !== undefined ? { maxTables: options.maxTables } : {}),
+});
+const PROVIDER_LABEL = { postgresql: "PostgreSQL", mysql: "MySQL", mssql: "SQL Server", oracle: "Oracle", mongodb: "MongoDB" };
 // Every document entering a tool goes through the shared gate (legacy fields
 // stripped, ids deduped, parents healed) — the same one the studio applies.
 const loadDocument = (diagram) => {
@@ -55,6 +132,21 @@ const assertEditableStructure = (doc) => {
 // nodes/edges (positions come from layout); anything else is kept as
 // source only (the desktop previews it, tools read it back).
 const documentFromSource = async (title, text, existing, direction, format = "mermaid") => {
+    if (format === "database") {
+        const parsed = parseDatabaseModel(text);
+        if (!parsed.ok)
+            throw new ToolInputError("DIAGRAM_SOURCE_INVALID", "The text is not a readable database model.", { errors: parsed.errors.slice(0, 20) });
+        const base = existing ?? createEmptyDocument(title);
+        const keep = Boolean(existing && existing.source?.format === "database");
+        let built;
+        try {
+            built = await databaseDocumentFromModel(parsed.model, keep ? existing : { ...base, nodes: [], edges: [] }, keep);
+        }
+        catch (error) {
+            return toToolError(error);
+        }
+        return { document: built.document, layoutNote: built.layoutNote, warnings: parsed.warnings };
+    }
     if (format === "bpmn") {
         if (!looksLikeBpmn(text))
             throw new ToolInputError("DIAGRAM_SOURCE_INVALID", "The text is not a BPMN 2.0 document (no <definitions> root).");
@@ -124,39 +216,59 @@ const isConflict = (error) => error instanceof CallmanApiError && error.status =
 export const registerDiagramTools = (server, ctx) => {
     server.registerTool("list_diagrams", {
         title: "List diagrams",
-        description: "List your system-design diagrams and the ones shared with you (owner, node/edge counts, whether you can edit). Diagrams belong to users and are shared into workspaces — not workspace-scoped.",
-        inputSchema: { query: z.string().max(200).optional(), ...cursorArgs },
+        description: "List your system-design diagrams and the ones shared with you (owner, node/edge counts, whether you can edit), with the folder tree of your library: `folders` are your own folders as paths (\"Architecture/Payments\"), each diagram carries its `folder` path (null = the top level; always null for a diagram shared with you — folders are private to their owner). Diagrams belong to users and are shared into workspaces — not workspace-scoped.",
+        inputSchema: {
+            query: z.string().max(200).optional(),
+            folder: z.string().max(300).optional().describe('Only the diagrams in this folder of your library (id, name or path such as "Architecture/Payments"); "/" = the top level only'),
+            ...cursorArgs,
+        },
         annotations: READ,
-    }, guarded(async ({ query, cursor, limit }) => {
-        const all = await loadDiagrams(ctx);
+    }, guarded(async ({ query, folder, cursor, limit }) => {
+        const [all, rows] = await Promise.all([loadDiagrams(ctx), diagramFolderRows(ctx)]);
         const needle = query?.trim().toLowerCase();
-        const filtered = needle ? all.filter((diagram) => diagram.name.toLowerCase().includes(needle)) : all;
-        const page = sliceWithCursor(filtered, cursor, limit, needle ?? "");
-        return ok({ diagrams: page.items.map(diagramListView), total: page.total, nextCursor: page.nextCursor });
+        const scope = folder?.trim() === "/" ? { item: null, annotation: undefined } : folder ? await resolveDiagramFolder(ctx, folder) : undefined;
+        const filtered = all.filter((diagram) => {
+            if (needle && !diagram.name.toLowerCase().includes(needle))
+                return false;
+            if (scope === undefined)
+                return true;
+            return (diagram.folderId ?? null) === (scope.item?.id ?? null);
+        });
+        const page = sliceWithCursor(filtered, cursor, limit, `${needle ?? ""}|${scope?.item?.id ?? (scope ? "/" : "")}`);
+        return ok({
+            diagrams: page.items.map((diagram) => ({ ...diagramListView(diagram), folder: diagramFolderPath(rows, diagram.folderId) })),
+            folders: rows.map((row) => ({ id: row.id, name: row.name, parentId: row.parentId, path: row.path })),
+            total: page.total,
+            nextCursor: page.nextCursor,
+            ...(scope?.item && scope.annotation ? { resolved: [resolvedSummary(folder, scope)] } : {}),
+        });
     }));
     server.registerTool("get_diagram", {
         title: "Get a diagram",
-        description: "Read a diagram: `summary` (counts, owner, sharing), `graph` (default — nodes with type/label/parent/style and edges with semantic/color/routing/sides, in the same shape the write tools accept: enough to describe, restyle or edit it), `mermaid` (the diagram rendered as Mermaid flowchart text), `source` (the Mermaid text or BPMN 2.0 XML a source-backed diagram keeps) or `full` (the complete stored DiagramDocument with positions, sizes and routes, up to 512 KB).",
-        inputSchema: { diagram: refArg("Diagram"), view: z.enum(["summary", "graph", "mermaid", "source", "full"]).default("graph") },
+        description: "Read a diagram: `summary` (counts, owner, sharing), `graph` (default — nodes with type/label/parent/style and edges with semantic/color/routing/sides, in the same shape the write tools accept: enough to describe, restyle or edit it), `mermaid` (the diagram rendered as Mermaid flowchart text), `source` (the Mermaid text, BPMN 2.0 XML or JSON database model a source-backed diagram keeps), `schema` (a database diagram's tables with columns, keys and relationships as table.column pairs — read it before database ops) or `full` (the complete stored DiagramDocument with positions, sizes and routes, up to 512 KB).",
+        inputSchema: { diagram: refArg("Diagram"), view: z.enum(["summary", "graph", "mermaid", "source", "schema", "full"]).default("graph") },
         annotations: READ,
     }, guarded(async ({ diagram, view }) => {
         const resolved = await resolveDiagram(ctx, diagram);
         const doc = loadDocument(resolved.item);
         const summary = diagramListView(resolved.item);
+        const summaryWithFolder = { ...summary, folder: diagramFolderPath(await diagramFolderRows(ctx), resolved.item.folderId) };
         const body = view === "summary"
-            ? summary
+            ? summaryWithFolder
             : view === "graph"
                 ? { ...summary, ...diagramGraphView(doc) }
                 : view === "mermaid"
                     ? { ...summary, ...diagramMermaidView(doc) }
                     : view === "source"
                         ? { ...summary, ...diagramSourceView(doc) }
-                        : { ...summary, ...diagramFullView(doc) };
+                        : view === "schema"
+                            ? { ...summary, schema: databaseSchemaView(doc) }
+                            : { ...summary, ...diagramFullView(doc) };
         return ok({ diagram: body, ...(resolved.annotation ? { resolved: [resolvedSummary(diagram, resolved)] } : {}) });
     }));
     server.registerTool("create_diagram", {
-        title: "Create a system-design diagram",
-        description: "Create a diagram from typed nodes and edges — or from Mermaid text / BPMN 2.0 XML (`source`). A BPMN process (`kind: \"bpmn\"`, or nodes typed bpmn.*) is stored as BPMN 2.0 XML the desktop opens in its BPMN editor. Node `type` must come from callman://diagram/components (e.g. net.service, net.api-gateway, db.postgres, mq.kafka, aws.lambda, flow.process); unknown types are rejected with suggestions unless strictTypes: false. Group nodes inside a container node — frame, pool.horizontal or pool.vertical — with `parent`. Style nodes with `style` (fill / stroke / textColor as CSS colors, strokeWidth, dashed, radius, bold, fontSize, opacity…) and edges with `semantic` (sync default, async, fallback, bus), an optional `color`, `routing` and fixed `sourceSide`/`targetSide`. Layout is automatic when nodes have no positions. Read callman://docs/diagram-schema for every field. Keep it readable: 8–25 nodes, short labels.",
+        title: "Create a system-design, process or database diagram",
+        description: "Create a diagram from typed nodes and edges — or from Mermaid text / BPMN 2.0 XML / a database model (`source`). A BPMN process (`kind: \"bpmn\"`, or nodes typed bpmn.*) is stored as BPMN 2.0 XML the desktop opens in its BPMN editor. A DATABASE diagram (entity-relationship: `kind: \"database\"`, `tables` + `relationships`, nodes typed db.table/db.view/db.collection with the columns in data: { db: { columns } }, or `fromConnection` to read a real database's schema) is stored as the JSON database model the desktop opens on its ERD canvas — tables with columns and keys, relationships with cardinality, laid out left to right. Node `type` must come from callman://diagram/components (e.g. net.service, net.api-gateway, db.postgres, mq.kafka, aws.lambda, flow.process); unknown types are rejected with suggestions unless strictTypes: false. Group nodes inside a container node — frame, pool.horizontal or pool.vertical — with `parent`. Style nodes with `style` (fill / stroke / textColor as CSS colors, strokeWidth, dashed, radius, bold, fontSize, opacity…) and edges with `semantic` (sync default, async, fallback, bus), an optional `color`, `routing` and fixed `sourceSide`/`targetSide`. Layout is automatic when nodes have no positions. Read callman://docs/diagram-schema for every field. Keep it readable: 8–25 nodes, short labels.",
         inputSchema: {
             name: z.string().min(1).max(200).describe("Diagram name (library entry)"),
             title: diagramInputSchema.shape.title.optional().describe("Diagram title (defaults to the name)"),
@@ -164,28 +276,76 @@ export const registerDiagramTools = (server, ctx) => {
             edges: diagramInputSchema.shape.edges.optional(),
             canvas: diagramInputSchema.shape.canvas,
             source: sourceArg,
-            kind: z.enum(["canvas", "bpmn"]).optional().describe('"bpmn": model a business process — nodes use bpmn.* types and the diagram is stored as BPMN 2.0 XML (default: canvas; inferred as bpmn when every node type starts with "bpmn.")'),
+            kind: z.enum(["canvas", "bpmn", "database", "mermaid"]).optional().describe('"bpmn": model a business process — nodes use bpmn.* types and the diagram is stored as BPMN 2.0 XML; "database": an entity-relationship diagram stored as the JSON database model; "mermaid": a Mermaid diagram — give `source: { format: "mermaid", text }` (default: canvas; inferred from the node types, from `source`, from `tables` and from `fromConnection`)'),
+            tables: z.array(dbTableInputSchema).max(500).optional().describe("Database diagram: the tables (name, schema?, kind?, columns with dataType / primaryKey / nullable / unique / defaultValue / comment)"),
+            relationships: z.array(dbRelationshipInputSchema).max(2000).optional().describe('Database diagram: relationships between the tables — source is the child (foreign-key) side as "table.column", target the parent; cardinality source:target (default N:1)'),
+            dialect: z.enum(DATABASE_DIALECTS).optional().describe("Database diagram: which database the types are written for"),
+            fromConnection: z
+                .object({
+                connection: z.string().min(1).max(200).describe("A database connection by id or name (list_connections, type: database)"),
+                schemas: z.array(z.string().min(1).max(120)).max(50).optional().describe("Schemas to read (default: every user schema; MySQL: the connection's database; Oracle: the current schema)"),
+                tables: z.array(z.string().min(1).max(240)).max(2000).optional().describe('Only these tables ("schema.table" or "table"); default: all'),
+                includeViews: z.boolean().optional(),
+            })
+                .strict()
+                .optional()
+                .describe("Database diagram: read a real database's schema through one of the workspace's connections and draw it — tables, columns, keys and the relationships the database declares"),
+            workspace: workspaceArg,
+            folder: z.string().max(300).optional().describe('File it in a folder of your library (id, name or path such as "Architecture/Payments"; create one with create_diagram_folder). Default: the top level'),
             autoLayout: z.boolean().optional().describe("Force layout even when positions were given"),
             direction: directionArg,
             strictTypes: z.boolean().optional(),
             dryRun: dryRunArg,
         },
         annotations: WRITE,
-    }, guarded(async ({ name, title, nodes, edges, canvas, source, kind, autoLayout, direction, strictTypes, dryRun }) => {
-        if (!nodes && !source)
-            throw new ToolInputError("NOTHING_TO_CREATE", "Pass nodes (+ edges) or source { format: \"mermaid\" | \"bpmn\", text }.");
-        if (nodes && source)
-            throw new ToolInputError("CONFLICTING_INPUT", "Pass either nodes or source, not both.");
+    }, guarded(async ({ name, title, nodes, edges, canvas, source, kind, tables, relationships, dialect, fromConnection, workspace, folder, autoLayout, direction, strictTypes, dryRun }) => {
+        const targetFolder = folder ? await resolveDiagramFolder(ctx, folder) : null;
+        const inputs = [nodes ? "nodes" : null, source ? "source" : null, tables ? "tables" : null, fromConnection ? "fromConnection" : null].filter(Boolean);
+        if (inputs.length === 0)
+            throw new ToolInputError("NOTHING_TO_CREATE", 'Pass nodes (+ edges), source { format: "mermaid" | "bpmn" | "database", text }, tables (+ relationships) or fromConnection.');
+        if (inputs.length > 1)
+            throw new ToolInputError("CONFLICTING_INPUT", `Pass one of nodes, source, tables or fromConnection — not ${inputs.join(" + ")}.`);
+        if (relationships && !tables)
+            throw new ToolInputError("CONFLICTING_INPUT", "relationships go with tables.");
+        if (kind === "mermaid" && source?.format !== "mermaid") {
+            throw new ToolInputError("MERMAID_NEEDS_SOURCE", 'A Mermaid diagram is created from source: { format: "mermaid", text } — a flowchart becomes nodes, any other Mermaid type is stored as its source.');
+        }
         let document;
         let layoutNote;
         let warnings = [];
         const bpmnKind = kind === "bpmn" || (kind === undefined && Boolean(nodes && nodes.length > 0 && nodes.every((node) => node.type.startsWith("bpmn."))));
+        const databaseKind = kind === "database" || Boolean(tables) || Boolean(fromConnection) || (kind === undefined && Boolean(nodes && nodes.length > 0 && nodes.every((node) => isDatabaseType(node.type))));
+        let connectionNote = {};
         if (source) {
             const fromSource = await documentFromSource(title ?? name, source.text, null, direction, source.format);
             document = canvas ? { ...fromSource.document, canvas: { ...fromSource.document.canvas, ...canvas } } : fromSource.document;
             layoutNote = fromSource.layoutNote;
             warnings = fromSource.warnings;
         }
+        else if (fromConnection) {
+            const found = await findDatabaseConnection(ctx, workspace, fromConnection.connection);
+            const schema = await ctx.api.getConnectionSchema(found.ws.id, found.connection.id, schemaQuery(fromConnection));
+            const model = databaseModelFromSchema(schema, { connection: { connectionId: found.connection.id, name: found.connection.name, provider: found.connection.provider } });
+            const built = await databaseDocumentFromModel(model, createEmptyDocument(title ?? name));
+            document = built.document;
+            layoutNote = built.layoutNote;
+            warnings = [...schema.warnings, ...(schema.truncated ? ["The database has more tables than the cap; the list was cut."] : [])];
+            connectionNote = { connection: { id: found.connection.id, name: found.connection.name, provider: PROVIDER_LABEL[found.connection.provider] ?? found.connection.provider, workspace: workspaceTag(found.ws) }, tables: model.entities.length, relationships: model.relationships.length };
+        }
+        else if (tables) {
+            const model = databaseModelFromInput(tables, relationships ?? [], dialect);
+            const built = await databaseDocumentFromModel(model, createEmptyDocument(title ?? name));
+            document = built.document;
+            layoutNote = built.layoutNote;
+        }
+        else if (databaseKind) {
+            const built = buildDiagramDocument({ title: title ?? name, nodes: nodes ?? [], edges: edges ?? [], canvas }, { strictTypes, kind: "database" });
+            const model = databaseModelFromDocument({ ...built.document, source: undefined }, dialect);
+            const database = await databaseDocumentFromModel(model, createEmptyDocument(title ?? name));
+            document = database.document;
+            layoutNote = database.layoutNote;
+            warnings = built.warnings;
+        }
         else if (bpmnKind) {
             const built = buildDiagramDocument({ title: title ?? name, nodes: nodes ?? [], edges: edges ?? [], canvas }, { strictTypes, kind: "bpmn" });
             const bpmn = bpmnDocumentFromModel(built.document, built.unpositionedNodeIds, createEmptyDocument(title ?? name));
@@ -203,21 +363,26 @@ export const registerDiagramTools = (server, ctx) => {
                 layoutNote = laid.warning ?? `Laid out with ${laid.engine}.`;
             }
         }
+        if (document.source?.format === "database" && canvas && !source)
+            document = { ...document, canvas: { ...document.canvas, ...canvas } };
+        const view = document.source?.format === "database" ? { ...diagramGraphView(document), schema: databaseSchemaView(document) } : diagramGraphView(document);
         if (dryRun) {
-            return ok({ dryRun: true, diagram: { name, ...diagramGraphView(document) }, warnings, ...(layoutNote ? { layout: layoutNote } : {}) });
+            return ok({ dryRun: true, diagram: { name, folder: targetFolder?.item.path ?? null, ...view }, warnings, ...connectionNote, ...(layoutNote ? { layout: layoutNote } : {}) });
         }
-        const created = await ctx.api.createDiagram({ name: name.trim(), document });
+        const created = await ctx.api.createDiagram({ name: name.trim(), document, ...(targetFolder ? { folderId: targetFolder.item.id } : {}) });
         ctx.invalidate("diagrams");
         return ok({
-            diagram: { ...diagramListView(created), ...diagramGraphView(document) },
+            diagram: { ...diagramListView(created), folder: targetFolder?.item.path ?? null, ...view },
             warnings,
+            ...connectionNote,
             ...(layoutNote ? { layout: layoutNote } : {}),
-            note: "Saved. Open it in the desktop app (Diagrams → Refresh). Edit with patch_diagram; share with share_diagram.",
+            ...(targetFolder?.annotation ? { resolved: [resolvedSummary(folder, targetFolder)] } : {}),
+            note: `Saved${targetFolder ? ` in ${targetFolder.item.path}` : ""}. Open it in the desktop app (Diagrams → Refresh). Edit with patch_diagram; share with share_diagram.`,
         });
     }));
     server.registerTool("update_diagram", {
         title: "Replace a diagram's content",
-        description: "Replace a diagram's whole document (nodes + edges, same fields as create_diagram), replace or attach its Mermaid text or BPMN 2.0 XML `source`, and/or rename it, retitle it or change its canvas — keeping its identity and sharing. Passing nodes replaces ALL nodes and edges (omitted edges are removed). Passing `source` re-derives nodes and edges from the text (existing ones keep their place; new ones are placed). For small edits — including restyling — prefer patch_diagram.",
+        description: "Replace a diagram's whole document (nodes + edges, same fields as create_diagram), replace or attach its Mermaid text, BPMN 2.0 XML or database-model `source`, rename it, retitle it, change its canvas and/or move it to a folder of your library (`folder`; null = the top level; only the owner can move) — keeping its identity and sharing. Passing nodes replaces ALL nodes and edges (omitted edges are removed). Passing `source` re-derives nodes and edges from the text (existing ones keep their place; new ones are placed). For small edits — including restyling — prefer patch_diagram.",
         inputSchema: {
             diagram: refArg("Diagram"),
             name: z.string().min(1).max(200).optional(),
@@ -226,17 +391,19 @@ export const registerDiagramTools = (server, ctx) => {
             edges: diagramInputSchema.shape.edges.optional(),
             canvas: diagramInputSchema.shape.canvas,
             source: sourceArg,
+            folder: z.string().max(300).nullable().optional().describe('Move the diagram into this folder of your library (id, name or path); null = the top level'),
             autoLayout: z.boolean().optional(),
             direction: directionArg,
             strictTypes: z.boolean().optional(),
         },
         annotations: IDEMPOTENT_WRITE,
-    }, guarded(async ({ diagram, name, title, nodes, edges, canvas, source, autoLayout, direction, strictTypes }) => {
-        if (name === undefined && nodes === undefined && title === undefined && canvas === undefined && source === undefined) {
-            throw new ToolInputError("NOTHING_TO_UPDATE", "Pass name, title, canvas, source and/or nodes (+ edges).");
+    }, guarded(async ({ diagram, name, title, nodes, edges, canvas, source, folder, autoLayout, direction, strictTypes }) => {
+        if (name === undefined && nodes === undefined && title === undefined && canvas === undefined && source === undefined && folder === undefined) {
+            throw new ToolInputError("NOTHING_TO_UPDATE", "Pass name, title, canvas, source, folder and/or nodes (+ edges).");
         }
         if (nodes && source)
             throw new ToolInputError("CONFLICTING_INPUT", "Pass either nodes or source, not both.");
+        const targetFolder = folder ? await resolveDiagramFolder(ctx, folder) : null;
         const resolved = await resolveDiagram(ctx, diagram);
         if (!resolved.item.canEdit)
             throw new ToolInputError("DIAGRAM_READ_ONLY", "You have read-only access to this diagram.");
@@ -244,7 +411,19 @@ export const registerDiagramTools = (server, ctx) => {
         let document;
         let layoutNote;
         let warnings = [];
-        if (nodes && existing.source?.format === "bpmn") {
+        if (nodes && existing.source?.format === "database") {
+            // A database diagram stays one: the tables given become the new model.
+            const built = buildDiagramDocument({ title: title ?? existing.meta.title, nodes, edges: edges ?? [], canvas: canvas ?? existing.canvas }, { strictTypes, existingId: existing.id, createdAt: existing.meta.createdAt, kind: "database" });
+            const previous = parseDatabaseModel(existing.source.text).model;
+            const model = databaseModelFromDocument({ ...built.document, source: undefined }, previous.dialect);
+            if (previous.connection)
+                model.connection = previous.connection;
+            const database = await databaseDocumentFromModel(model, existing, true);
+            document = database.document;
+            layoutNote = database.layoutNote;
+            warnings = [...built.warnings, "The database model was rewritten from the tables given; tables not in the input are gone. Prefer patch_diagram (addTable, addColumn, addRelationship …) for edits."];
+        }
+        else if (nodes && existing.source?.format === "bpmn") {
             // A BPMN diagram stays BPMN: the new model is written as fresh XML.
             const built = buildDiagramDocument({ title: title ?? existing.meta.title, nodes, edges: edges ?? [], canvas: canvas ?? existing.canvas }, { strictTypes, existingId: existing.id, createdAt: existing.meta.createdAt, kind: "bpmn" });
             const bpmn = bpmnDocumentFromModel(built.document, built.unpositionedNodeIds, existing);
@@ -261,7 +440,7 @@ export const registerDiagramTools = (server, ctx) => {
                 warnings.push(`edges was omitted, so the ${existing.edges.length} existing edge(s) were removed. Pass edges to keep connections, or use patch_diagram for partial edits.`);
             }
             if (existing.source)
-                warnings.push("The diagram's Mermaid source was detached: nodes and edges were replaced from the input.");
+                warnings.push(`The diagram's ${existing.source.format === "mermaid" ? "Mermaid" : existing.source.format} source was detached: nodes and edges were replaced from the input.`);
             if (autoLayout || built.unpositionedNodeIds.length > 0) {
                 const laid = await layoutDocument(document, { direction });
                 document = laid.document;
@@ -291,13 +470,20 @@ export const registerDiagramTools = (server, ctx) => {
         const updated = await ctx.api.updateDiagram(resolved.item.id, {
             ...(name !== undefined ? { name: name.trim() } : {}),
             ...(document ? { document, ...(typeof existing.meta.revision === "number" ? { baseRevision: existing.meta.revision } : {}) } : {}),
+            ...(folder !== undefined ? { folderId: targetFolder?.item.id ?? null } : {}),
         });
         ctx.invalidate("diagrams");
-        return ok({ diagram: { ...diagramListView(updated), ...diagramGraphView(document ?? existing) }, warnings, ...(layoutNote ? { layout: layoutNote } : {}), resolved: [resolvedSummary(diagram, resolved)] });
+        const rows = await diagramFolderRows(ctx);
+        return ok({
+            diagram: { ...diagramListView(updated), folder: diagramFolderPath(rows, updated.folderId), ...diagramGraphView(document ?? existing) },
+            warnings,
+            ...(layoutNote ? { layout: layoutNote } : {}),
+            resolved: [resolvedSummary(diagram, resolved), ...(targetFolder?.annotation ? [resolvedSummary(folder, targetFolder)] : [])],
+        });
     }));
     server.registerTool("patch_diagram", {
         title: "Edit a diagram incrementally",
-        description: "Apply operations to a diagram without resending it: addNode, updateNode, removeNode (drops its edges, lifts its children out), addEdge, updateEdge, removeEdge, setTitle, moveNode, reparent, setEdgeRoute (routing, fixed sides, bends), reorder (stacking), setCanvas, group (wrap nodes in a container), ungroup, relayout (a hint: new | all). updateNode `changes` MERGE: style and data keys you pass change, a key set to null is removed. updateEdge can change semantic, color, label, arrows, sides and ends. Moving a node re-routes its edges. `relayout`: none | new (place only new nodes, default) | all (re-run automatic layout). Call get_diagram (view: graph) first to see node ids and current styles. Ops are documented in callman://docs/diagram-schema.",
+        description: "Apply operations to a diagram without resending it: addNode, updateNode, removeNode (drops its edges, lifts its children out), addEdge, updateEdge, removeEdge, setTitle, moveNode, reparent, setEdgeRoute (routing, fixed sides, bends), reorder (stacking), setCanvas, group (wrap nodes in a container), ungroup, relayout (a hint: new | all). On a DATABASE diagram use the schema ops — addTable, renameTable, removeTable, setTableComment, addColumn, alterColumn, removeColumn, setPrimaryKey, addRelationship (source = child \"table.column\", target = parent), updateRelationship, removeRelationship, hideTable, showTable — read get_diagram view schema first. updateNode `changes` MERGE: style and data keys you pass change, a key set to null is removed. updateEdge can change semantic, color, label, arrows, sides and ends. Moving a node re-routes its edges. `relayout`: none | new (place only new nodes, default) | all (re-run automatic layout). Call get_diagram (view: graph) first to see node ids and current styles. Ops are documented in callman://docs/diagram-schema.",
         inputSchema: {
             diagram: refArg("Diagram"),
             ops: z.array(diagramPatchOpSchema).min(1).max(200),
@@ -321,15 +507,21 @@ export const registerDiagramTools = (server, ctx) => {
             const wantsAll = relayout === "all" || patched.relayout?.scope === "all";
             const layoutDirection = direction ?? patched.relayout?.direction;
             const bpmn = document.source?.format === "bpmn";
+            const database = document.source?.format === "database";
             if (wantsAll) {
-                // A process reads left to right; the BPMN XML follows the moves.
-                const laid = await layoutDocument(document, { direction: layoutDirection ?? (bpmn ? "RIGHT" : undefined) });
+                // A process reads left to right; the BPMN XML follows the moves. An
+                // ERD packs unrelated tables into rows.
+                const laid = await layoutDocument(document, { direction: layoutDirection ?? (bpmn || database ? "RIGHT" : undefined), ...(database ? { request: DATABASE_LAYOUT_REQUEST } : {}) });
                 document = laid.document;
                 layoutNote = laid.warning ?? `Laid out with ${laid.engine}.`;
             }
             else if (relayout === "new" && patched.addedNodeIds.length > 0) {
                 const ids = patched.addedNodeIds.filter((id) => document.nodes.some((node) => node.id === id));
-                if (bpmn) {
+                if (database) {
+                    document = placeDatabaseNodes(document, ids);
+                    layoutNote = `Placed ${ids.length} new table(s) to the right of the existing ones.`;
+                }
+                else if (bpmn) {
                     document = applyDiagramOps(document, placeBpmnNodesOps(document, ids), { now: now() }).doc;
                     layoutNote = `Placed ${ids.length} new node(s) after their predecessors.`;
                 }
@@ -358,7 +550,7 @@ export const registerDiagramTools = (server, ctx) => {
         ctx.invalidate("diagrams");
         const { updated, document, patched, layoutNote } = outcome;
         return ok({
-            diagram: { ...diagramListView(updated), ...diagramGraphView(document) },
+            diagram: { ...diagramListView(updated), ...diagramGraphView(document), ...(document.source?.format === "database" ? { schema: databaseSchemaView(document) } : {}) },
             applied: ops.length,
             added: patched.addedNodeIds,
             removedNodes: patched.removedNodeIds,
@@ -370,6 +562,103 @@ export const registerDiagramTools = (server, ctx) => {
             resolved: [resolvedSummary(diagram, resolved)],
         });
     }));
+    server.registerTool("introspect_database", {
+        title: "Read a database's schema through a connection",
+        description: "Read the database behind one of the workspace's database connections (list_connections, type: database): schemas, tables and views (MongoDB: collections with fields inferred from a sample), columns with types, nullability, defaults and comments, primary and unique keys, foreign keys with referential actions, indexes. Metadata only — never credentials, never data. Use `objectsOnly` to list names first, then read the tables you need; `create_diagram` with `fromConnection` draws the result as a Database Diagram in one step. Runs on the Callman backend, which must be able to reach the database.",
+        inputSchema: {
+            connection: z.string().min(1).max(200).describe("Connection id or name"),
+            workspace: workspaceArg,
+            schemas: z.array(z.string().min(1).max(120)).max(50).optional(),
+            tables: z.array(z.string().min(1).max(240)).max(2000).optional().describe('"schema.table" or "table"'),
+            includeViews: z.boolean().optional(),
+            objectsOnly: z.boolean().optional().describe("Names only (no columns / keys) — for a first look at a large database"),
+            maxTables: z.number().int().min(1).max(20000).optional(),
+        },
+        annotations: READ,
+    }, guarded(async ({ connection, workspace, schemas, tables, includeViews, objectsOnly, maxTables }) => {
+        const found = await findDatabaseConnection(ctx, workspace, connection);
+        const schema = await ctx.api.getConnectionSchema(found.ws.id, found.connection.id, schemaQuery({ schemas, tables, includeViews, objectsOnly, maxTables }));
+        return ok({
+            connection: { id: found.connection.id, name: found.connection.name, provider: PROVIDER_LABEL[found.connection.provider] ?? found.connection.provider, workspace: workspaceTag(found.ws) },
+            schema,
+            ...(schema.truncated ? { note: "The database has more tables than the cap; pass schemas/tables to narrow the read or raise maxTables." } : {}),
+        });
+    }));
+    // ── library folders (private to the token's owner) ──────────────────
+    const folderView = (row) => ({ id: row.id, name: row.name, parentId: row.parentId, path: row.path });
+    server.registerTool("create_diagram_folder", {
+        title: "Create a diagram folder",
+        description: "Create a folder in your diagram library, optionally inside a parent folder (id, name or path). Folders are yours alone (nobody else sees them; a diagram you share shows at the top level for others), nest up to 8 levels, and sibling names are unique — an existing folder with that name is returned instead of a duplicate. File diagrams with create_diagram / update_diagram `folder`.",
+        inputSchema: {
+            name: z.string().min(1).max(120),
+            parent: z.string().max(300).optional().describe("Parent folder id, name or path; omit for the top level"),
+        },
+        annotations: WRITE,
+    }, guarded(async ({ name, parent }) => {
+        const parentFolder = parent ? await resolveDiagramFolder(ctx, parent) : null;
+        const rows = await diagramFolderRows(ctx);
+        const trimmed = name.trim();
+        const sibling = rows.find((row) => row.parentId === (parentFolder?.item.id ?? null) && normalizeName(row.name) === normalizeName(trimmed));
+        if (sibling) {
+            return ok({ folder: folderView(sibling), existed: true, note: "A folder with that name already exists there — returned instead of creating a duplicate." });
+        }
+        const created = await ctx.api.createDiagramFolder({ name: trimmed, parentId: parentFolder?.item.id ?? null });
+        ctx.invalidate("diagram-folders");
+        const path = parentFolder ? `${parentFolder.item.path}/${created.name}` : created.name;
+        return ok({
+            folder: { id: created.id, name: created.name, parentId: created.parentId, path },
+            ...(parentFolder?.annotation ? { resolved: [resolvedSummary(parent, parentFolder)] } : {}),
+        });
+    }));
+    server.registerTool("update_diagram_folder", {
+        title: "Rename or move a diagram folder",
+        description: "Rename a folder of your diagram library and/or move it under another folder (parent: null = the top level). A folder cannot be moved into itself or one of its subfolders.",
+        inputSchema: {
+            folder: refArg("Diagram folder", "(or path)"),
+            name: z.string().min(1).max(120).optional(),
+            parent: z.string().max(300).nullable().optional().describe("New parent folder (id, name or path); null = the top level"),
+        },
+        annotations: IDEMPOTENT_WRITE,
+    }, guarded(async ({ folder, name, parent }) => {
+        if (name === undefined && parent === undefined)
+            throw new ToolInputError("NOTHING_TO_UPDATE", "Pass name and/or parent.");
+        const target = await resolveDiagramFolder(ctx, folder);
+        const parentFolder = parent ? await resolveDiagramFolder(ctx, parent) : null;
+        if (parentFolder && (parentFolder.item.id === target.item.id || parentFolder.item.path.startsWith(`${target.item.path}/`))) {
+            throw new ToolInputError("FOLDER_CYCLE", "A folder cannot be moved into itself or one of its subfolders.");
+        }
+        const updated = await ctx.api.updateDiagramFolder(target.item.id, {
+            ...(name !== undefined ? { name: name.trim() } : {}),
+            ...(parent !== undefined ? { parentId: parentFolder?.item.id ?? null } : {}),
+        });
+        ctx.invalidate("diagram-folders");
+        const rows = await diagramFolderRows(ctx);
+        const row = rows.find((entry) => entry.id === updated.id);
+        return ok({
+            folder: row ? folderView(row) : { id: updated.id, name: updated.name, parentId: updated.parentId, path: updated.name },
+            resolved: [resolvedSummary(folder, target), ...(parentFolder?.annotation ? [resolvedSummary(parent, parentFolder)] : [])],
+        });
+    }));
+    server.registerTool("delete_diagram_folder", {
+        title: "Delete a diagram folder",
+        description: "Delete a folder of your diagram library and its subfolders. No diagram is deleted: everything inside moves to the deleted folder's parent. Confirm with the user first.",
+        inputSchema: { folder: refArg("Diagram folder", "(or path)"), confirm: confirmArg },
+        annotations: DESTRUCTIVE,
+    }, guarded(async ({ folder, confirm }) => {
+        requireConfirm(confirm, "a diagram folder");
+        const target = await resolveDiagramFolder(ctx, folder, { exact: true });
+        const result = await ctx.api.deleteDiagramFolder(target.item.id);
+        ctx.invalidate("diagram-folders");
+        ctx.invalidate("diagrams");
+        const rows = await diagramFolderRows(ctx);
+        return ok({
+            deleted: folderView(target.item),
+            deletedFolderCount: result.deletedFolderIds.length,
+            movedDiagramCount: result.movedDiagramCount,
+            movedTo: diagramFolderPath(rows, result.movedToFolderId) ?? "the top level",
+            resolved: [resolvedSummary(folder, target)],
+        });
+    }));
     server.registerTool("delete_diagram", {
         title: "Delete a diagram",
         description: "Delete a diagram you own. Permanent — confirm with the user first.",