@nonsoo/prisma-mermaid 0.1.2 → 0.2.0

package/README.md

@@ -64,6 +64,32 @@ await generateMermaidERD({
 });
 ```
 
+## Styling Diagrams
+
+This library ships with sensible default styles for Mermaid ERD and Class diagrams. However, if you need more control, both `generateMermaidClass` and `generateMermaidERD` accept a configuration object that lets you customize the final diagram output.
+
+The configuration `type` is exported as `MermaidDiagramConfig`. You can import it directly and pass the config object to the generator functions.
+
+```ts
+import {
+  type MermaidDiagramConfig,
+  generateMermaidERD,
+} from "@nonsoo/prisma-mermaid";
+
+const config = {
+  type: "mermaid-erd",
+  config: {
+    layout: "elk",
+  },
+} satisfies MermaidDiagramConfig;
+
+await generateMermaidERD({
+  schemaPath: "./prisma/schema.prisma",
+  output: "./diagrams/erdDiagram.mmd",
+  config,
+});
+```
+
 ## Purpose
 
 Documentation should evolve alongside the code it describes. Diagrams-as-code tools such as Mermaid make it easier for teams to maintain clear, accurate diagrams as their systems grow and change. However, creating these diagrams manually — especially for database schemas — still introduces friction and the risk of diagrams falling out of sync with the system.

package/build/index.cjs

@@ -41,13 +41,18 @@ var import_node_fs = require("fs");
 var import_node_path = __toESM(require("path"), 1);
 
 // src/constants/mermaid.ts
+var DEFAULT_BASE_NODE_SPACING = 100;
+var DEFAULT_BASE_EDGE_SPACING = 150;
 var mermaidERDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
+  panZoom: true,
   themeVariables: {
     fontSize: "20px",
     fontFamily: "Arial",
     padding: "12px",
-    lineHeight: "1.4"
+    lineHeight: 1.4
   },
   flowchart: {
     nodeSpacing: 80,
@@ -57,9 +62,12 @@ var mermaidERDiagramConfig = {
 };
 var mermaidClassDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
+  panZoom: true,
   themeVariables: {
     fontFamily: "Arial",
-    lineHeight: "1.4"
+    lineHeight: 1.4
   },
   flowchart: {
     nodeSpacing: 300,
@@ -70,10 +78,9 @@ var mermaidClassDiagramConfig = {
     hideEmptyMembersBox: true
   }
 };
-var DEFAULT_BASE_NODE_SPACING = 100;
-var DEFAULT_BASE_EDGE_SPACING = 150;
 
 // src/utils/mermaid.ts
+var import_js_yaml = require("js-yaml");
 var generateDiagramSpacing = ({
   baseEdge,
   baseNode,
@@ -102,8 +109,17 @@ var generateMermaidConfig = (config, models) => {
       nodeSpacing
     };
   }
-  const json = JSON.stringify(config, null, 2);
-  return `%%{init: ${json}}%%
+  const { title, ...rest } = config;
+  const diagramConfig = {
+    config: rest,
+    ...title ? { title } : {}
+  };
+  const yamlString = (0, import_js_yaml.dump)(diagramConfig, {
+    indent: 2,
+    lineWidth: 1e3
+  });
+  return `---
+${yamlString}---
 `;
 };
 
@@ -169,7 +185,8 @@ var { getDMMF } = import_internals.default;
 var generateDiagram = async ({
   outputPath,
   schemaPath,
-  generatorPrismaDocument
+  generatorPrismaDocument,
+  config
 }) => {
   const outputDir = outputPath ? import_node_path.default.resolve(outputPath) : import_node_path.default.join(`${process.cwd()}/src/generated/diagrams`);
   try {
@@ -178,18 +195,23 @@ var generateDiagram = async ({
     });
     const models = prismaDocument.datamodel.models;
     const enums = prismaDocument.datamodel.enums;
+    const userGeneratedConfig = config?.type === "mermaid-class" ? config?.config : {};
+    const diagramConfig = {
+      ...mermaidClassDiagramConfig,
+      ...userGeneratedConfig
+    };
     const mermaidLines = [
+      generateMermaidConfig(diagramConfig, models),
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid Class Diagram.  Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidClassDiagramConfig, models),
       "classDiagram"
     ];
     const relationships = {};
     models.forEach((model) => {
-      mermaidLines.push(`class ${model.name} {`);
+      mermaidLines.push(`	class ${model.name} {`);
       model.fields.forEach((field) => {
-        mermaidLines.push(`  ${field.type} ${field.name}`);
+        mermaidLines.push(`		${field.type} ${field.name}`);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -202,21 +224,21 @@ var generateDiagram = async ({
           });
         }
       });
-      mermaidLines.push("}");
+      mermaidLines.push("	}");
     });
     enums.forEach((enumDef) => {
-      mermaidLines.push(`class ${enumDef.name} {`);
+      mermaidLines.push(`	class ${enumDef.name} {`);
       enumDef.values.forEach((val) => {
-        mermaidLines.push(`  <<enumeration>> ${val.name}`);
+        mermaidLines.push(`		<<enumeration>> ${val.name}`);
       });
-      mermaidLines.push("}");
+      mermaidLines.push("	}");
     });
     const relationLines = generateRelationships({ relationships });
     const output = mermaidLines.concat(relationLines).join("\n");
     (0, import_node_fs.mkdirSync)(outputDir, { recursive: true });
     const outFile = import_node_path.default.join(outputDir, "mermaidClassDiagram.mmd");
     (0, import_node_fs.writeFileSync)(outFile, output, "utf-8");
-    console.log(`Mermaid Class Diagram written to: ${outFile}`);
+    console.log(`Mermaid Class Diagram generated at: ${outFile}`);
     return outFile;
   } catch (e) {
     console.error("Failed to generate Mermaid Class Diagram.", e);
@@ -240,14 +262,8 @@ var generateCardinality2 = ({
   }
   return isRequired ? "||" : "o|";
 };
-var getKeyConstraints = (isId, fieldName, foreignKeys, nativeTypes) => {
+var getKeyConstraints = (isId, fieldName, foreignKeys) => {
   if (isId) return "PK";
-  if (nativeTypes) {
-    const allNativeTypes = nativeTypes.flatMap((nativeType) => nativeType);
-    if (!isId && allNativeTypes.includes("UniqueIdentifier")) {
-      return "FK";
-    }
-  }
   if (!isId && foreignKeys.has(fieldName)) return "FK";
   return "";
 };
@@ -302,13 +318,32 @@ var generateRelationships2 = ({
   }
   return relationLines;
 };
+var validateForeignKeys = ({
+  foreignKeys,
+  foreignKeyLocation,
+  mermaidLines
+}) => {
+  if (foreignKeys.size > 0) {
+    for (const key of foreignKeys) {
+      const keyIndexMermaidLines = foreignKeyLocation.get(key);
+      if (!keyIndexMermaidLines) continue;
+      const currentLine = mermaidLines[keyIndexMermaidLines];
+      if (!currentLine) continue;
+      const lineArray = currentLine.split(" ");
+      lineArray[2] = "FK";
+      const finalLine = lineArray.join(" ");
+      mermaidLines[keyIndexMermaidLines] = finalLine;
+    }
+  }
+};
 
 // src/lib/MermaidERD/prismaMermaidErd.ts
 var { getDMMF: getDMMF2 } = import_internals2.default;
 var generateDiagram2 = async ({
   outputPath,
   schemaPath,
-  generatorPrismaDocument
+  generatorPrismaDocument,
+  config
 }) => {
   const outputDir = outputPath ? import_node_path2.default.resolve(outputPath) : import_node_path2.default.join(`${process.cwd()}/src/generated/diagrams`);
   try {
@@ -317,17 +352,22 @@ var generateDiagram2 = async ({
     });
     const schemaModels = prismaDocument.datamodel.models;
     const schemaEnums = prismaDocument.datamodel.enums;
-    console.dir(schemaEnums, { depth: null });
+    const userGeneratedConfig = config?.type === "mermaid-erd" ? config?.config : {};
+    const diagramConfig = {
+      ...mermaidERDiagramConfig,
+      ...userGeneratedConfig
+    };
     const mermaidLines = [
+      generateMermaidConfig(diagramConfig, schemaModels),
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid ER Diagram. Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidERDiagramConfig, schemaModels),
       "erDiagram"
     ];
     const relationships = {};
     schemaModels.forEach((model) => {
       mermaidLines.push(`	${model.name} {`);
+      const foreignKeyLocation = /* @__PURE__ */ new Map();
       const foreignKeys = /* @__PURE__ */ new Set();
       model.fields.forEach((field) => {
         if (field.relationFromFields && field.relationFromFields.length > 0) {
@@ -339,10 +379,10 @@ var generateDiagram2 = async ({
           `		${field.type} ${field.name} ${getKeyConstraints(
             field.isId,
             field.name,
-            foreignKeys,
-            field.nativeType
+            foreignKeys
           )} ${getOptionalitySymbol(field.isRequired)}`
         );
+        foreignKeyLocation.set(field.name, mermaidLines.length - 1);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -355,6 +395,11 @@ var generateDiagram2 = async ({
           });
         }
       });
+      validateForeignKeys({
+        foreignKeyLocation,
+        foreignKeys,
+        mermaidLines
+      });
       mermaidLines.push(`	}`);
     });
     schemaEnums.forEach((enumDef) => {

package/build/index.d.cts

@@ -1,7 +1,52 @@
 import { DMMF } from '@prisma/generator-helper';
 
+type ConfigTheme = "default" | "redux-dark" | "forest" | "neutral" | "mc" | "base" | "redux" | "redux-dark";
+type ConfigLook = "classic" | "handDrawn" | "neo";
+type ConfigLayout = "dagre" | "elk";
+type ConfigThemeVariables = {
+    fontSize: `${number}px`;
+    fontFamily: string;
+    padding: `${number}px`;
+    lineHeight: number;
+    nodeSpacing: number;
+    edgeSpacing: number;
+};
+type ConfigFlowchart = {
+    nodeSpacing: number;
+    rankSpacing: number;
+    htmlLabels: boolean;
+};
+type MermaidERDiagramConfig = {
+    type: Exclude<PrismaGeneratorsKeys, "mermaid-class">;
+    config: Partial<{
+        theme: ConfigTheme;
+        layout: ConfigLayout;
+        look: ConfigLook;
+        panZoom: boolean;
+        themeVariables: Partial<ConfigThemeVariables>;
+        flowchart: Partial<ConfigFlowchart>;
+        title: string;
+    }>;
+};
+type MermaidClassDiagramConfig = {
+    type: Exclude<PrismaGeneratorsKeys, "mermaid-erd">;
+    config: Partial<{
+        theme: ConfigTheme;
+        layout: ConfigLayout;
+        look: ConfigLook;
+        themeVariables: Partial<ConfigThemeVariables>;
+        flowchart: Partial<ConfigFlowchart>;
+        panZoom: boolean;
+        class: Partial<{
+            hideEmptyMembersBox: boolean;
+        }>;
+        title: string;
+    }>;
+};
+type MermaidDiagramConfig = MermaidERDiagramConfig | MermaidClassDiagramConfig;
 type GenerateDiagramOptions = {
     generatorPrismaDocument?: DMMF.Document;
+    config?: MermaidDiagramConfig;
     schemaPath: string;
     outputPath: string | undefined;
 };
@@ -44,7 +89,7 @@ type PrismaGeneratorsKeys = "mermaid-erd" | "mermaid-class";
  * or, if `outputPath` is omitted:
  *   `<projectRoot>/src/generated/diagrams/mermaidClassDiagram.mmd`
  */
-declare const generateDiagram$1: ({ outputPath, schemaPath, generatorPrismaDocument, }: GenerateDiagramOptions) => Promise<string>;
+declare const generateDiagram$1: ({ outputPath, schemaPath, generatorPrismaDocument, config, }: GenerateDiagramOptions) => Promise<string>;
 
 /**
  * Generates a Mermaid ERD (Entity-Relationship Diagram) from a Prisma schema.
@@ -58,6 +103,6 @@ declare const generateDiagram$1: ({ outputPath, schemaPath, generatorPrismaDocum
  * or, if no output path is provided:
  *   `<projectRoot>/src/generated/diagrams/mermaidErdDiagram.mmd`
  */
-declare const generateDiagram: ({ outputPath, schemaPath, generatorPrismaDocument, }: GenerateDiagramOptions) => Promise<string>;
+declare const generateDiagram: ({ outputPath, schemaPath, generatorPrismaDocument, config, }: GenerateDiagramOptions) => Promise<string>;
 
-export { type ClassCardinality, type ERDCardinality, type GenerateCardinality, type GenerateCardinalityOptions, type GenerateDiagram, type GenerateDiagramOptions, type GenerateDiagramSpacingOptions, type GenerateRelationshipOptions, type GenerateRelationships, type PrismaGeneratorsKeys, type Relationships, generateDiagram$1 as generateMermaidClass, generateDiagram as generateMermaidERD };
+export { type ClassCardinality, type ERDCardinality, type GenerateCardinality, type GenerateCardinalityOptions, type GenerateDiagram, type GenerateDiagramOptions, type GenerateDiagramSpacingOptions, type GenerateRelationshipOptions, type GenerateRelationships, type MermaidClassDiagramConfig, type MermaidDiagramConfig, type MermaidERDiagramConfig, type PrismaGeneratorsKeys, type Relationships, generateDiagram$1 as generateMermaidClass, generateDiagram as generateMermaidERD };

package/build/index.d.ts

@@ -1,7 +1,52 @@
 import { DMMF } from '@prisma/generator-helper';
 
+type ConfigTheme = "default" | "redux-dark" | "forest" | "neutral" | "mc" | "base" | "redux" | "redux-dark";
+type ConfigLook = "classic" | "handDrawn" | "neo";
+type ConfigLayout = "dagre" | "elk";
+type ConfigThemeVariables = {
+    fontSize: `${number}px`;
+    fontFamily: string;
+    padding: `${number}px`;
+    lineHeight: number;
+    nodeSpacing: number;
+    edgeSpacing: number;
+};
+type ConfigFlowchart = {
+    nodeSpacing: number;
+    rankSpacing: number;
+    htmlLabels: boolean;
+};
+type MermaidERDiagramConfig = {
+    type: Exclude<PrismaGeneratorsKeys, "mermaid-class">;
+    config: Partial<{
+        theme: ConfigTheme;
+        layout: ConfigLayout;
+        look: ConfigLook;
+        panZoom: boolean;
+        themeVariables: Partial<ConfigThemeVariables>;
+        flowchart: Partial<ConfigFlowchart>;
+        title: string;
+    }>;
+};
+type MermaidClassDiagramConfig = {
+    type: Exclude<PrismaGeneratorsKeys, "mermaid-erd">;
+    config: Partial<{
+        theme: ConfigTheme;
+        layout: ConfigLayout;
+        look: ConfigLook;
+        themeVariables: Partial<ConfigThemeVariables>;
+        flowchart: Partial<ConfigFlowchart>;
+        panZoom: boolean;
+        class: Partial<{
+            hideEmptyMembersBox: boolean;
+        }>;
+        title: string;
+    }>;
+};
+type MermaidDiagramConfig = MermaidERDiagramConfig | MermaidClassDiagramConfig;
 type GenerateDiagramOptions = {
     generatorPrismaDocument?: DMMF.Document;
+    config?: MermaidDiagramConfig;
     schemaPath: string;
     outputPath: string | undefined;
 };
@@ -44,7 +89,7 @@ type PrismaGeneratorsKeys = "mermaid-erd" | "mermaid-class";
  * or, if `outputPath` is omitted:
  *   `<projectRoot>/src/generated/diagrams/mermaidClassDiagram.mmd`
  */
-declare const generateDiagram$1: ({ outputPath, schemaPath, generatorPrismaDocument, }: GenerateDiagramOptions) => Promise<string>;
+declare const generateDiagram$1: ({ outputPath, schemaPath, generatorPrismaDocument, config, }: GenerateDiagramOptions) => Promise<string>;
 
 /**
  * Generates a Mermaid ERD (Entity-Relationship Diagram) from a Prisma schema.
@@ -58,6 +103,6 @@ declare const generateDiagram$1: ({ outputPath, schemaPath, generatorPrismaDocum
  * or, if no output path is provided:
  *   `<projectRoot>/src/generated/diagrams/mermaidErdDiagram.mmd`
  */
-declare const generateDiagram: ({ outputPath, schemaPath, generatorPrismaDocument, }: GenerateDiagramOptions) => Promise<string>;
+declare const generateDiagram: ({ outputPath, schemaPath, generatorPrismaDocument, config, }: GenerateDiagramOptions) => Promise<string>;
 
-export { type ClassCardinality, type ERDCardinality, type GenerateCardinality, type GenerateCardinalityOptions, type GenerateDiagram, type GenerateDiagramOptions, type GenerateDiagramSpacingOptions, type GenerateRelationshipOptions, type GenerateRelationships, type PrismaGeneratorsKeys, type Relationships, generateDiagram$1 as generateMermaidClass, generateDiagram as generateMermaidERD };
+export { type ClassCardinality, type ERDCardinality, type GenerateCardinality, type GenerateCardinalityOptions, type GenerateDiagram, type GenerateDiagramOptions, type GenerateDiagramSpacingOptions, type GenerateRelationshipOptions, type GenerateRelationships, type MermaidClassDiagramConfig, type MermaidDiagramConfig, type MermaidERDiagramConfig, type PrismaGeneratorsKeys, type Relationships, generateDiagram$1 as generateMermaidClass, generateDiagram as generateMermaidERD };

package/build/index.js

@@ -4,13 +4,18 @@ import { readFileSync, writeFileSync, mkdirSync } from "fs";
 import path from "path";
 
 // src/constants/mermaid.ts
+var DEFAULT_BASE_NODE_SPACING = 100;
+var DEFAULT_BASE_EDGE_SPACING = 150;
 var mermaidERDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
+  panZoom: true,
   themeVariables: {
     fontSize: "20px",
     fontFamily: "Arial",
     padding: "12px",
-    lineHeight: "1.4"
+    lineHeight: 1.4
   },
   flowchart: {
     nodeSpacing: 80,
@@ -20,9 +25,12 @@ var mermaidERDiagramConfig = {
 };
 var mermaidClassDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
+  panZoom: true,
   themeVariables: {
     fontFamily: "Arial",
-    lineHeight: "1.4"
+    lineHeight: 1.4
   },
   flowchart: {
     nodeSpacing: 300,
@@ -33,10 +41,9 @@ var mermaidClassDiagramConfig = {
     hideEmptyMembersBox: true
   }
 };
-var DEFAULT_BASE_NODE_SPACING = 100;
-var DEFAULT_BASE_EDGE_SPACING = 150;
 
 // src/utils/mermaid.ts
+import { dump } from "js-yaml";
 var generateDiagramSpacing = ({
   baseEdge,
   baseNode,
@@ -65,8 +72,17 @@ var generateMermaidConfig = (config, models) => {
       nodeSpacing
     };
   }
-  const json = JSON.stringify(config, null, 2);
-  return `%%{init: ${json}}%%
+  const { title, ...rest } = config;
+  const diagramConfig = {
+    config: rest,
+    ...title ? { title } : {}
+  };
+  const yamlString = dump(diagramConfig, {
+    indent: 2,
+    lineWidth: 1e3
+  });
+  return `---
+${yamlString}---
 `;
 };
 
@@ -132,7 +148,8 @@ var { getDMMF } = pkg;
 var generateDiagram = async ({
   outputPath,
   schemaPath,
-  generatorPrismaDocument
+  generatorPrismaDocument,
+  config
 }) => {
   const outputDir = outputPath ? path.resolve(outputPath) : path.join(`${process.cwd()}/src/generated/diagrams`);
   try {
@@ -141,18 +158,23 @@ var generateDiagram = async ({
     });
     const models = prismaDocument.datamodel.models;
     const enums = prismaDocument.datamodel.enums;
+    const userGeneratedConfig = config?.type === "mermaid-class" ? config?.config : {};
+    const diagramConfig = {
+      ...mermaidClassDiagramConfig,
+      ...userGeneratedConfig
+    };
     const mermaidLines = [
+      generateMermaidConfig(diagramConfig, models),
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid Class Diagram.  Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidClassDiagramConfig, models),
       "classDiagram"
     ];
     const relationships = {};
     models.forEach((model) => {
-      mermaidLines.push(`class ${model.name} {`);
+      mermaidLines.push(`	class ${model.name} {`);
       model.fields.forEach((field) => {
-        mermaidLines.push(`  ${field.type} ${field.name}`);
+        mermaidLines.push(`		${field.type} ${field.name}`);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -165,21 +187,21 @@ var generateDiagram = async ({
           });
         }
       });
-      mermaidLines.push("}");
+      mermaidLines.push("	}");
     });
     enums.forEach((enumDef) => {
-      mermaidLines.push(`class ${enumDef.name} {`);
+      mermaidLines.push(`	class ${enumDef.name} {`);
       enumDef.values.forEach((val) => {
-        mermaidLines.push(`  <<enumeration>> ${val.name}`);
+        mermaidLines.push(`		<<enumeration>> ${val.name}`);
       });
-      mermaidLines.push("}");
+      mermaidLines.push("	}");
     });
     const relationLines = generateRelationships({ relationships });
     const output = mermaidLines.concat(relationLines).join("\n");
     mkdirSync(outputDir, { recursive: true });
     const outFile = path.join(outputDir, "mermaidClassDiagram.mmd");
     writeFileSync(outFile, output, "utf-8");
-    console.log(`Mermaid Class Diagram written to: ${outFile}`);
+    console.log(`Mermaid Class Diagram generated at: ${outFile}`);
     return outFile;
   } catch (e) {
     console.error("Failed to generate Mermaid Class Diagram.", e);
@@ -203,14 +225,8 @@ var generateCardinality2 = ({
   }
   return isRequired ? "||" : "o|";
 };
-var getKeyConstraints = (isId, fieldName, foreignKeys, nativeTypes) => {
+var getKeyConstraints = (isId, fieldName, foreignKeys) => {
   if (isId) return "PK";
-  if (nativeTypes) {
-    const allNativeTypes = nativeTypes.flatMap((nativeType) => nativeType);
-    if (!isId && allNativeTypes.includes("UniqueIdentifier")) {
-      return "FK";
-    }
-  }
   if (!isId && foreignKeys.has(fieldName)) return "FK";
   return "";
 };
@@ -265,13 +281,32 @@ var generateRelationships2 = ({
   }
   return relationLines;
 };
+var validateForeignKeys = ({
+  foreignKeys,
+  foreignKeyLocation,
+  mermaidLines
+}) => {
+  if (foreignKeys.size > 0) {
+    for (const key of foreignKeys) {
+      const keyIndexMermaidLines = foreignKeyLocation.get(key);
+      if (!keyIndexMermaidLines) continue;
+      const currentLine = mermaidLines[keyIndexMermaidLines];
+      if (!currentLine) continue;
+      const lineArray = currentLine.split(" ");
+      lineArray[2] = "FK";
+      const finalLine = lineArray.join(" ");
+      mermaidLines[keyIndexMermaidLines] = finalLine;
+    }
+  }
+};
 
 // src/lib/MermaidERD/prismaMermaidErd.ts
 var { getDMMF: getDMMF2 } = pkg2;
 var generateDiagram2 = async ({
   outputPath,
   schemaPath,
-  generatorPrismaDocument
+  generatorPrismaDocument,
+  config
 }) => {
   const outputDir = outputPath ? path2.resolve(outputPath) : path2.join(`${process.cwd()}/src/generated/diagrams`);
   try {
@@ -280,17 +315,22 @@ var generateDiagram2 = async ({
     });
     const schemaModels = prismaDocument.datamodel.models;
     const schemaEnums = prismaDocument.datamodel.enums;
-    console.dir(schemaEnums, { depth: null });
+    const userGeneratedConfig = config?.type === "mermaid-erd" ? config?.config : {};
+    const diagramConfig = {
+      ...mermaidERDiagramConfig,
+      ...userGeneratedConfig
+    };
     const mermaidLines = [
+      generateMermaidConfig(diagramConfig, schemaModels),
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid ER Diagram. Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidERDiagramConfig, schemaModels),
       "erDiagram"
     ];
     const relationships = {};
     schemaModels.forEach((model) => {
       mermaidLines.push(`	${model.name} {`);
+      const foreignKeyLocation = /* @__PURE__ */ new Map();
       const foreignKeys = /* @__PURE__ */ new Set();
       model.fields.forEach((field) => {
         if (field.relationFromFields && field.relationFromFields.length > 0) {
@@ -302,10 +342,10 @@ var generateDiagram2 = async ({
           `		${field.type} ${field.name} ${getKeyConstraints(
             field.isId,
             field.name,
-            foreignKeys,
-            field.nativeType
+            foreignKeys
           )} ${getOptionalitySymbol(field.isRequired)}`
         );
+        foreignKeyLocation.set(field.name, mermaidLines.length - 1);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -318,6 +358,11 @@ var generateDiagram2 = async ({
           });
         }
       });
+      validateForeignKeys({
+        foreignKeyLocation,
+        foreignKeys,
+        mermaidLines
+      });
       mermaidLines.push(`	}`);
     });
     schemaEnums.forEach((enumDef) => {

package/build/lib/PrismaMermaidGenerators/bin.cjs

@@ -1382,13 +1382,18 @@ var import_node_fs = require("fs");
 var import_node_path = __toESM(require("path"), 1);
 
 // src/constants/mermaid.ts
+var DEFAULT_BASE_NODE_SPACING = 100;
+var DEFAULT_BASE_EDGE_SPACING = 150;
 var mermaidERDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
+  panZoom: true,
   themeVariables: {
     fontSize: "20px",
     fontFamily: "Arial",
     padding: "12px",
-    lineHeight: "1.4"
+    lineHeight: 1.4
   },
   flowchart: {
     nodeSpacing: 80,
@@ -1398,9 +1403,12 @@ var mermaidERDiagramConfig = {
 };
 var mermaidClassDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
+  panZoom: true,
   themeVariables: {
     fontFamily: "Arial",
-    lineHeight: "1.4"
+    lineHeight: 1.4
   },
   flowchart: {
     nodeSpacing: 300,
@@ -1411,10 +1419,9 @@ var mermaidClassDiagramConfig = {
     hideEmptyMembersBox: true
   }
 };
-var DEFAULT_BASE_NODE_SPACING = 100;
-var DEFAULT_BASE_EDGE_SPACING = 150;
 
 // src/utils/mermaid.ts
+var import_js_yaml = require("js-yaml");
 var generateDiagramSpacing = ({
   baseEdge,
   baseNode,
@@ -1443,8 +1450,17 @@ var generateMermaidConfig = (config, models) => {
       nodeSpacing
     };
   }
-  const json = JSON.stringify(config, null, 2);
-  return `%%{init: ${json}}%%
+  const { title, ...rest } = config;
+  const diagramConfig = {
+    config: rest,
+    ...title ? { title } : {}
+  };
+  const yamlString = (0, import_js_yaml.dump)(diagramConfig, {
+    indent: 2,
+    lineWidth: 1e3
+  });
+  return `---
+${yamlString}---
 `;
 };
 
@@ -1510,7 +1526,8 @@ var { getDMMF } = import_internals.default;
 var generateDiagram = async ({
   outputPath,
   schemaPath,
-  generatorPrismaDocument
+  generatorPrismaDocument,
+  config
 }) => {
   const outputDir = outputPath ? import_node_path.default.resolve(outputPath) : import_node_path.default.join(`${process.cwd()}/src/generated/diagrams`);
   try {
@@ -1519,18 +1536,23 @@ var generateDiagram = async ({
     });
     const models = prismaDocument.datamodel.models;
     const enums = prismaDocument.datamodel.enums;
+    const userGeneratedConfig = config?.type === "mermaid-class" ? config?.config : {};
+    const diagramConfig = {
+      ...mermaidClassDiagramConfig,
+      ...userGeneratedConfig
+    };
     const mermaidLines = [
+      generateMermaidConfig(diagramConfig, models),
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid Class Diagram.  Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidClassDiagramConfig, models),
       "classDiagram"
     ];
     const relationships = {};
     models.forEach((model) => {
-      mermaidLines.push(`class ${model.name} {`);
+      mermaidLines.push(`	class ${model.name} {`);
       model.fields.forEach((field) => {
-        mermaidLines.push(`  ${field.type} ${field.name}`);
+        mermaidLines.push(`		${field.type} ${field.name}`);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -1543,21 +1565,21 @@ var generateDiagram = async ({
           });
         }
       });
-      mermaidLines.push("}");
+      mermaidLines.push("	}");
     });
     enums.forEach((enumDef) => {
-      mermaidLines.push(`class ${enumDef.name} {`);
+      mermaidLines.push(`	class ${enumDef.name} {`);
       enumDef.values.forEach((val) => {
-        mermaidLines.push(`  <<enumeration>> ${val.name}`);
+        mermaidLines.push(`		<<enumeration>> ${val.name}`);
       });
-      mermaidLines.push("}");
+      mermaidLines.push("	}");
     });
     const relationLines = generateRelationships({ relationships });
     const output = mermaidLines.concat(relationLines).join("\n");
     (0, import_node_fs.mkdirSync)(outputDir, { recursive: true });
     const outFile = import_node_path.default.join(outputDir, "mermaidClassDiagram.mmd");
     (0, import_node_fs.writeFileSync)(outFile, output, "utf-8");
-    console.log(`Mermaid Class Diagram written to: ${outFile}`);
+    console.log(`Mermaid Class Diagram generated at: ${outFile}`);
     return outFile;
   } catch (e) {
     console.error("Failed to generate Mermaid Class Diagram.", e);
@@ -1581,14 +1603,8 @@ var generateCardinality2 = ({
   }
   return isRequired ? "||" : "o|";
 };
-var getKeyConstraints = (isId, fieldName, foreignKeys, nativeTypes) => {
+var getKeyConstraints = (isId, fieldName, foreignKeys) => {
   if (isId) return "PK";
-  if (nativeTypes) {
-    const allNativeTypes = nativeTypes.flatMap((nativeType) => nativeType);
-    if (!isId && allNativeTypes.includes("UniqueIdentifier")) {
-      return "FK";
-    }
-  }
   if (!isId && foreignKeys.has(fieldName)) return "FK";
   return "";
 };
@@ -1643,13 +1659,32 @@ var generateRelationships2 = ({
   }
   return relationLines;
 };
+var validateForeignKeys = ({
+  foreignKeys,
+  foreignKeyLocation,
+  mermaidLines
+}) => {
+  if (foreignKeys.size > 0) {
+    for (const key of foreignKeys) {
+      const keyIndexMermaidLines = foreignKeyLocation.get(key);
+      if (!keyIndexMermaidLines) continue;
+      const currentLine = mermaidLines[keyIndexMermaidLines];
+      if (!currentLine) continue;
+      const lineArray = currentLine.split(" ");
+      lineArray[2] = "FK";
+      const finalLine = lineArray.join(" ");
+      mermaidLines[keyIndexMermaidLines] = finalLine;
+    }
+  }
+};
 
 // src/lib/MermaidERD/prismaMermaidErd.ts
 var { getDMMF: getDMMF2 } = import_internals2.default;
 var generateDiagram2 = async ({
   outputPath,
   schemaPath,
-  generatorPrismaDocument
+  generatorPrismaDocument,
+  config
 }) => {
   const outputDir = outputPath ? import_node_path2.default.resolve(outputPath) : import_node_path2.default.join(`${process.cwd()}/src/generated/diagrams`);
   try {
@@ -1658,17 +1693,22 @@ var generateDiagram2 = async ({
     });
     const schemaModels = prismaDocument.datamodel.models;
     const schemaEnums = prismaDocument.datamodel.enums;
-    console.dir(schemaEnums, { depth: null });
+    const userGeneratedConfig = config?.type === "mermaid-erd" ? config?.config : {};
+    const diagramConfig = {
+      ...mermaidERDiagramConfig,
+      ...userGeneratedConfig
+    };
     const mermaidLines = [
+      generateMermaidConfig(diagramConfig, schemaModels),
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid ER Diagram. Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidERDiagramConfig, schemaModels),
       "erDiagram"
     ];
     const relationships = {};
     schemaModels.forEach((model) => {
       mermaidLines.push(`	${model.name} {`);
+      const foreignKeyLocation = /* @__PURE__ */ new Map();
       const foreignKeys = /* @__PURE__ */ new Set();
       model.fields.forEach((field) => {
         if (field.relationFromFields && field.relationFromFields.length > 0) {
@@ -1680,10 +1720,10 @@ var generateDiagram2 = async ({
           `		${field.type} ${field.name} ${getKeyConstraints(
             field.isId,
             field.name,
-            foreignKeys,
-            field.nativeType
+            foreignKeys
           )} ${getOptionalitySymbol(field.isRequired)}`
         );
+        foreignKeyLocation.set(field.name, mermaidLines.length - 1);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -1696,6 +1736,11 @@ var generateDiagram2 = async ({
           });
         }
       });
+      validateForeignKeys({
+        foreignKeyLocation,
+        foreignKeys,
+        mermaidLines
+      });
       mermaidLines.push(`	}`);
     });
     schemaEnums.forEach((enumDef) => {
@@ -1727,7 +1772,7 @@ var prismaGenerators = /* @__PURE__ */ new Map([
 // package.json
 var package_default = {
   name: "@nonsoo/prisma-mermaid",
-  version: "0.1.2",
+  version: "0.2.0",
   description: "A Prisma generator that generates Mermaid Class or ER diagrams from your Prisma schema.",
   main: "build/index.js",
   bin: {
@@ -1777,6 +1822,7 @@ var package_default = {
   devDependencies: {
     "@changesets/cli": "^2.29.7",
     "@eslint/js": "^9.39.1",
+    "@types/js-yaml": "^4.0.9",
     "@types/node": "^24.10.1",
     eslint: "^9.39.1",
     "eslint-config-prettier": "^10.1.8",
@@ -1792,7 +1838,8 @@ var package_default = {
     "@prisma/client": "^6.0.0 || ^7.0.0"
   },
   dependencies: {
-    "@prisma/internals": "^7.0.0"
+    "@prisma/internals": "^7.0.0",
+    "js-yaml": "^4.1.1"
   }
 };
 
@@ -1814,12 +1861,12 @@ var package_default = {
       return;
     }
     if (Array.isArray(formats)) {
-      for (const format of formats) {
+      for await (const format of formats) {
         const mermaidGenerator2 = prismaGenerators.get(
           format
         );
         if (!mermaidGenerator2) continue;
-        mermaidGenerator2({
+        await mermaidGenerator2({
           schemaPath,
           outputPath: outputDir,
           generatorPrismaDocument
@@ -1831,7 +1878,7 @@ var package_default = {
       formats
     );
     if (!mermaidGenerator) return;
-    mermaidGenerator({
+    await mermaidGenerator({
       schemaPath,
       outputPath: outputDir,
       generatorPrismaDocument

package/build/lib/PrismaMermaidGenerators/bin.js

@@ -1387,13 +1387,18 @@ import { readFileSync, writeFileSync, mkdirSync } from "fs";
 import path from "path";
 
 // src/constants/mermaid.ts
+var DEFAULT_BASE_NODE_SPACING = 100;
+var DEFAULT_BASE_EDGE_SPACING = 150;
 var mermaidERDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
+  panZoom: true,
   themeVariables: {
     fontSize: "20px",
     fontFamily: "Arial",
     padding: "12px",
-    lineHeight: "1.4"
+    lineHeight: 1.4
   },
   flowchart: {
     nodeSpacing: 80,
@@ -1403,9 +1408,12 @@ var mermaidERDiagramConfig = {
 };
 var mermaidClassDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
+  panZoom: true,
   themeVariables: {
     fontFamily: "Arial",
-    lineHeight: "1.4"
+    lineHeight: 1.4
   },
   flowchart: {
     nodeSpacing: 300,
@@ -1416,10 +1424,9 @@ var mermaidClassDiagramConfig = {
     hideEmptyMembersBox: true
   }
 };
-var DEFAULT_BASE_NODE_SPACING = 100;
-var DEFAULT_BASE_EDGE_SPACING = 150;
 
 // src/utils/mermaid.ts
+import { dump } from "js-yaml";
 var generateDiagramSpacing = ({
   baseEdge,
   baseNode,
@@ -1448,8 +1455,17 @@ var generateMermaidConfig = (config, models) => {
       nodeSpacing
     };
   }
-  const json = JSON.stringify(config, null, 2);
-  return `%%{init: ${json}}%%
+  const { title, ...rest } = config;
+  const diagramConfig = {
+    config: rest,
+    ...title ? { title } : {}
+  };
+  const yamlString = dump(diagramConfig, {
+    indent: 2,
+    lineWidth: 1e3
+  });
+  return `---
+${yamlString}---
 `;
 };
 
@@ -1515,7 +1531,8 @@ var { getDMMF } = pkg;
 var generateDiagram = async ({
   outputPath,
   schemaPath,
-  generatorPrismaDocument
+  generatorPrismaDocument,
+  config
 }) => {
   const outputDir = outputPath ? path.resolve(outputPath) : path.join(`${process.cwd()}/src/generated/diagrams`);
   try {
@@ -1524,18 +1541,23 @@ var generateDiagram = async ({
     });
     const models = prismaDocument.datamodel.models;
     const enums = prismaDocument.datamodel.enums;
+    const userGeneratedConfig = config?.type === "mermaid-class" ? config?.config : {};
+    const diagramConfig = {
+      ...mermaidClassDiagramConfig,
+      ...userGeneratedConfig
+    };
     const mermaidLines = [
+      generateMermaidConfig(diagramConfig, models),
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid Class Diagram.  Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidClassDiagramConfig, models),
       "classDiagram"
     ];
     const relationships = {};
     models.forEach((model) => {
-      mermaidLines.push(`class ${model.name} {`);
+      mermaidLines.push(`	class ${model.name} {`);
       model.fields.forEach((field) => {
-        mermaidLines.push(`  ${field.type} ${field.name}`);
+        mermaidLines.push(`		${field.type} ${field.name}`);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -1548,21 +1570,21 @@ var generateDiagram = async ({
           });
         }
       });
-      mermaidLines.push("}");
+      mermaidLines.push("	}");
     });
     enums.forEach((enumDef) => {
-      mermaidLines.push(`class ${enumDef.name} {`);
+      mermaidLines.push(`	class ${enumDef.name} {`);
       enumDef.values.forEach((val) => {
-        mermaidLines.push(`  <<enumeration>> ${val.name}`);
+        mermaidLines.push(`		<<enumeration>> ${val.name}`);
       });
-      mermaidLines.push("}");
+      mermaidLines.push("	}");
     });
     const relationLines = generateRelationships({ relationships });
     const output = mermaidLines.concat(relationLines).join("\n");
     mkdirSync(outputDir, { recursive: true });
     const outFile = path.join(outputDir, "mermaidClassDiagram.mmd");
     writeFileSync(outFile, output, "utf-8");
-    console.log(`Mermaid Class Diagram written to: ${outFile}`);
+    console.log(`Mermaid Class Diagram generated at: ${outFile}`);
     return outFile;
   } catch (e) {
     console.error("Failed to generate Mermaid Class Diagram.", e);
@@ -1586,14 +1608,8 @@ var generateCardinality2 = ({
   }
   return isRequired ? "||" : "o|";
 };
-var getKeyConstraints = (isId, fieldName, foreignKeys, nativeTypes) => {
+var getKeyConstraints = (isId, fieldName, foreignKeys) => {
   if (isId) return "PK";
-  if (nativeTypes) {
-    const allNativeTypes = nativeTypes.flatMap((nativeType) => nativeType);
-    if (!isId && allNativeTypes.includes("UniqueIdentifier")) {
-      return "FK";
-    }
-  }
   if (!isId && foreignKeys.has(fieldName)) return "FK";
   return "";
 };
@@ -1648,13 +1664,32 @@ var generateRelationships2 = ({
   }
   return relationLines;
 };
+var validateForeignKeys = ({
+  foreignKeys,
+  foreignKeyLocation,
+  mermaidLines
+}) => {
+  if (foreignKeys.size > 0) {
+    for (const key of foreignKeys) {
+      const keyIndexMermaidLines = foreignKeyLocation.get(key);
+      if (!keyIndexMermaidLines) continue;
+      const currentLine = mermaidLines[keyIndexMermaidLines];
+      if (!currentLine) continue;
+      const lineArray = currentLine.split(" ");
+      lineArray[2] = "FK";
+      const finalLine = lineArray.join(" ");
+      mermaidLines[keyIndexMermaidLines] = finalLine;
+    }
+  }
+};
 
 // src/lib/MermaidERD/prismaMermaidErd.ts
 var { getDMMF: getDMMF2 } = pkg2;
 var generateDiagram2 = async ({
   outputPath,
   schemaPath,
-  generatorPrismaDocument
+  generatorPrismaDocument,
+  config
 }) => {
   const outputDir = outputPath ? path2.resolve(outputPath) : path2.join(`${process.cwd()}/src/generated/diagrams`);
   try {
@@ -1663,17 +1698,22 @@ var generateDiagram2 = async ({
     });
     const schemaModels = prismaDocument.datamodel.models;
     const schemaEnums = prismaDocument.datamodel.enums;
-    console.dir(schemaEnums, { depth: null });
+    const userGeneratedConfig = config?.type === "mermaid-erd" ? config?.config : {};
+    const diagramConfig = {
+      ...mermaidERDiagramConfig,
+      ...userGeneratedConfig
+    };
     const mermaidLines = [
+      generateMermaidConfig(diagramConfig, schemaModels),
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid ER Diagram. Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidERDiagramConfig, schemaModels),
       "erDiagram"
     ];
     const relationships = {};
     schemaModels.forEach((model) => {
       mermaidLines.push(`	${model.name} {`);
+      const foreignKeyLocation = /* @__PURE__ */ new Map();
       const foreignKeys = /* @__PURE__ */ new Set();
       model.fields.forEach((field) => {
         if (field.relationFromFields && field.relationFromFields.length > 0) {
@@ -1685,10 +1725,10 @@ var generateDiagram2 = async ({
           `		${field.type} ${field.name} ${getKeyConstraints(
             field.isId,
             field.name,
-            foreignKeys,
-            field.nativeType
+            foreignKeys
           )} ${getOptionalitySymbol(field.isRequired)}`
         );
+        foreignKeyLocation.set(field.name, mermaidLines.length - 1);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -1701,6 +1741,11 @@ var generateDiagram2 = async ({
           });
         }
       });
+      validateForeignKeys({
+        foreignKeyLocation,
+        foreignKeys,
+        mermaidLines
+      });
       mermaidLines.push(`	}`);
     });
     schemaEnums.forEach((enumDef) => {
@@ -1732,7 +1777,7 @@ var prismaGenerators = /* @__PURE__ */ new Map([
 // package.json
 var package_default = {
   name: "@nonsoo/prisma-mermaid",
-  version: "0.1.2",
+  version: "0.2.0",
   description: "A Prisma generator that generates Mermaid Class or ER diagrams from your Prisma schema.",
   main: "build/index.js",
   bin: {
@@ -1782,6 +1827,7 @@ var package_default = {
   devDependencies: {
     "@changesets/cli": "^2.29.7",
     "@eslint/js": "^9.39.1",
+    "@types/js-yaml": "^4.0.9",
     "@types/node": "^24.10.1",
     eslint: "^9.39.1",
     "eslint-config-prettier": "^10.1.8",
@@ -1797,7 +1843,8 @@ var package_default = {
     "@prisma/client": "^6.0.0 || ^7.0.0"
   },
   dependencies: {
-    "@prisma/internals": "^7.0.0"
+    "@prisma/internals": "^7.0.0",
+    "js-yaml": "^4.1.1"
   }
 };
 
@@ -1819,12 +1866,12 @@ var package_default = {
       return;
     }
     if (Array.isArray(formats)) {
-      for (const format of formats) {
+      for await (const format of formats) {
         const mermaidGenerator2 = prismaGenerators.get(
           format
         );
         if (!mermaidGenerator2) continue;
-        mermaidGenerator2({
+        await mermaidGenerator2({
           schemaPath,
           outputPath: outputDir,
           generatorPrismaDocument
@@ -1836,7 +1883,7 @@ var package_default = {
       formats
     );
     if (!mermaidGenerator) return;
-    mermaidGenerator({
+    await mermaidGenerator({
       schemaPath,
       outputPath: outputDir,
       generatorPrismaDocument

package/package.json

@@ -1,10 +1,10 @@
 {
   "name": "@nonsoo/prisma-mermaid",
-  "version": "0.1.2",
+  "version": "0.2.0",
   "description": "A Prisma generator that generates Mermaid Class or ER diagrams from your Prisma schema.",
   "main": "build/index.js",
   "bin": {
-    "prisma-mermaid": "./build/lib/PrismaMermaidGenerators/bin.cjs"
+    "prisma-mermaid": "build/lib/PrismaMermaidGenerators/bin.cjs"
   },
   "types": "build/lib/PrismaMermaidGenerators/index.d.js",
   "files": [
@@ -34,7 +34,7 @@
   "type": "module",
   "repository": {
     "type": "git",
-    "url": "https://github.com/nonsoo/prisma-mermaid.git"
+    "url": "git+https://github.com/nonsoo/prisma-mermaid.git"
   },
   "engines": {
     "node": ">=22.0.0"
@@ -50,6 +50,7 @@
   "devDependencies": {
     "@changesets/cli": "^2.29.7",
     "@eslint/js": "^9.39.1",
+    "@types/js-yaml": "^4.0.9",
     "@types/node": "^24.10.1",
     "eslint": "^9.39.1",
     "eslint-config-prettier": "^10.1.8",
@@ -65,6 +66,7 @@
     "@prisma/client": "^6.0.0 || ^7.0.0"
   },
   "dependencies": {
-    "@prisma/internals": "^7.0.0"
+    "@prisma/internals": "^7.0.0",
+    "js-yaml": "^4.1.1"
   }
 }