@nonsoo/prisma-mermaid 0.1.2 → 0.1.3

package/README.md

@@ -64,6 +64,35 @@ 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: {
+    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,8 +41,12 @@ 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",
   themeVariables: {
     fontSize: "20px",
     fontFamily: "Arial",
@@ -57,6 +61,8 @@ var mermaidERDiagramConfig = {
 };
 var mermaidClassDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
   themeVariables: {
     fontFamily: "Arial",
     lineHeight: "1.4"
@@ -70,8 +76,6 @@ var mermaidClassDiagramConfig = {
     hideEmptyMembersBox: true
   }
 };
-var DEFAULT_BASE_NODE_SPACING = 100;
-var DEFAULT_BASE_EDGE_SPACING = 150;
 
 // src/utils/mermaid.ts
 var generateDiagramSpacing = ({
@@ -169,7 +173,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,11 +183,16 @@ 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 = [
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid Class Diagram.  Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidClassDiagramConfig, models),
+      generateMermaidConfig(diagramConfig, models),
       "classDiagram"
     ];
     const relationships = {};
@@ -302,13 +312,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 +346,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 = [
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid ER Diagram. Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidERDiagramConfig, schemaModels),
+      generateMermaidConfig(diagramConfig, 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) {
@@ -343,6 +377,7 @@ var generateDiagram2 = async ({
             field.nativeType
           )} ${getOptionalitySymbol(field.isRequired)}`
         );
+        foreignKeyLocation.set(field.name, mermaidLines.length - 1);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -355,6 +390,11 @@ var generateDiagram2 = async ({
           });
         }
       });
+      validateForeignKeys({
+        foreignKeyLocation,
+        foreignKeys,
+        mermaidLines
+      });
       mermaidLines.push(`	}`);
     });
     schemaEnums.forEach((enumDef) => {

package/build/index.d.cts

@@ -1,7 +1,48 @@
 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: string;
+    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;
+        themeVariables: Partial<ConfigThemeVariables>;
+        flowchart: Partial<ConfigFlowchart>;
+    }>;
+};
+type MermaidClassDiagramConfig = {
+    type: Exclude<PrismaGeneratorsKeys, "mermaid-erd">;
+    config: Partial<{
+        theme: ConfigTheme;
+        layout: ConfigLayout;
+        look: ConfigLook;
+        themeVariables: Partial<ConfigThemeVariables>;
+        flowchart: Partial<ConfigFlowchart>;
+        class: Partial<{
+            hideEmptyMembersBox: boolean;
+        }>;
+    }>;
+};
+type MermaidDiagramConfig = MermaidERDiagramConfig | MermaidClassDiagramConfig;
 type GenerateDiagramOptions = {
     generatorPrismaDocument?: DMMF.Document;
+    config?: MermaidDiagramConfig;
     schemaPath: string;
     outputPath: string | undefined;
 };
@@ -44,7 +85,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 +99,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,48 @@
 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: string;
+    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;
+        themeVariables: Partial<ConfigThemeVariables>;
+        flowchart: Partial<ConfigFlowchart>;
+    }>;
+};
+type MermaidClassDiagramConfig = {
+    type: Exclude<PrismaGeneratorsKeys, "mermaid-erd">;
+    config: Partial<{
+        theme: ConfigTheme;
+        layout: ConfigLayout;
+        look: ConfigLook;
+        themeVariables: Partial<ConfigThemeVariables>;
+        flowchart: Partial<ConfigFlowchart>;
+        class: Partial<{
+            hideEmptyMembersBox: boolean;
+        }>;
+    }>;
+};
+type MermaidDiagramConfig = MermaidERDiagramConfig | MermaidClassDiagramConfig;
 type GenerateDiagramOptions = {
     generatorPrismaDocument?: DMMF.Document;
+    config?: MermaidDiagramConfig;
     schemaPath: string;
     outputPath: string | undefined;
 };
@@ -44,7 +85,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 +99,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,8 +4,12 @@ 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",
   themeVariables: {
     fontSize: "20px",
     fontFamily: "Arial",
@@ -20,6 +24,8 @@ var mermaidERDiagramConfig = {
 };
 var mermaidClassDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
   themeVariables: {
     fontFamily: "Arial",
     lineHeight: "1.4"
@@ -33,8 +39,6 @@ var mermaidClassDiagramConfig = {
     hideEmptyMembersBox: true
   }
 };
-var DEFAULT_BASE_NODE_SPACING = 100;
-var DEFAULT_BASE_EDGE_SPACING = 150;
 
 // src/utils/mermaid.ts
 var generateDiagramSpacing = ({
@@ -132,7 +136,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,11 +146,16 @@ 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 = [
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid Class Diagram.  Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidClassDiagramConfig, models),
+      generateMermaidConfig(diagramConfig, models),
       "classDiagram"
     ];
     const relationships = {};
@@ -265,13 +275,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 +309,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 = [
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid ER Diagram. Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidERDiagramConfig, schemaModels),
+      generateMermaidConfig(diagramConfig, 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) {
@@ -306,6 +340,7 @@ var generateDiagram2 = async ({
             field.nativeType
           )} ${getOptionalitySymbol(field.isRequired)}`
         );
+        foreignKeyLocation.set(field.name, mermaidLines.length - 1);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -318,6 +353,11 @@ var generateDiagram2 = async ({
           });
         }
       });
+      validateForeignKeys({
+        foreignKeyLocation,
+        foreignKeys,
+        mermaidLines
+      });
       mermaidLines.push(`	}`);
     });
     schemaEnums.forEach((enumDef) => {

package/build/lib/PrismaMermaidGenerators/bin.cjs

@@ -1382,8 +1382,12 @@ 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",
   themeVariables: {
     fontSize: "20px",
     fontFamily: "Arial",
@@ -1398,6 +1402,8 @@ var mermaidERDiagramConfig = {
 };
 var mermaidClassDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
   themeVariables: {
     fontFamily: "Arial",
     lineHeight: "1.4"
@@ -1411,8 +1417,6 @@ var mermaidClassDiagramConfig = {
     hideEmptyMembersBox: true
   }
 };
-var DEFAULT_BASE_NODE_SPACING = 100;
-var DEFAULT_BASE_EDGE_SPACING = 150;
 
 // src/utils/mermaid.ts
 var generateDiagramSpacing = ({
@@ -1510,7 +1514,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,11 +1524,16 @@ 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 = [
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid Class Diagram.  Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidClassDiagramConfig, models),
+      generateMermaidConfig(diagramConfig, models),
       "classDiagram"
     ];
     const relationships = {};
@@ -1643,13 +1653,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 +1687,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 = [
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid ER Diagram. Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidERDiagramConfig, schemaModels),
+      generateMermaidConfig(diagramConfig, 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) {
@@ -1684,6 +1718,7 @@ var generateDiagram2 = async ({
             field.nativeType
           )} ${getOptionalitySymbol(field.isRequired)}`
         );
+        foreignKeyLocation.set(field.name, mermaidLines.length - 1);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -1696,6 +1731,11 @@ var generateDiagram2 = async ({
           });
         }
       });
+      validateForeignKeys({
+        foreignKeyLocation,
+        foreignKeys,
+        mermaidLines
+      });
       mermaidLines.push(`	}`);
     });
     schemaEnums.forEach((enumDef) => {
@@ -1727,7 +1767,7 @@ var prismaGenerators = /* @__PURE__ */ new Map([
 // package.json
 var package_default = {
   name: "@nonsoo/prisma-mermaid",
-  version: "0.1.2",
+  version: "0.1.3",
   description: "A Prisma generator that generates Mermaid Class or ER diagrams from your Prisma schema.",
   main: "build/index.js",
   bin: {

package/build/lib/PrismaMermaidGenerators/bin.js

@@ -1387,8 +1387,12 @@ 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",
   themeVariables: {
     fontSize: "20px",
     fontFamily: "Arial",
@@ -1403,6 +1407,8 @@ var mermaidERDiagramConfig = {
 };
 var mermaidClassDiagramConfig = {
   theme: "neutral",
+  layout: "dagre",
+  look: "classic",
   themeVariables: {
     fontFamily: "Arial",
     lineHeight: "1.4"
@@ -1416,8 +1422,6 @@ var mermaidClassDiagramConfig = {
     hideEmptyMembersBox: true
   }
 };
-var DEFAULT_BASE_NODE_SPACING = 100;
-var DEFAULT_BASE_EDGE_SPACING = 150;
 
 // src/utils/mermaid.ts
 var generateDiagramSpacing = ({
@@ -1515,7 +1519,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,11 +1529,16 @@ 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 = [
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid Class Diagram.  Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidClassDiagramConfig, models),
+      generateMermaidConfig(diagramConfig, models),
       "classDiagram"
     ];
     const relationships = {};
@@ -1648,13 +1658,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 +1692,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 = [
       "%% --------------------------------------------",
       "%% Auto-generated Mermaid ER Diagram. Do Not Edit Directly.",
       "%% --------------------------------------------\n",
-      generateMermaidConfig(mermaidERDiagramConfig, schemaModels),
+      generateMermaidConfig(diagramConfig, 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) {
@@ -1689,6 +1723,7 @@ var generateDiagram2 = async ({
             field.nativeType
           )} ${getOptionalitySymbol(field.isRequired)}`
         );
+        foreignKeyLocation.set(field.name, mermaidLines.length - 1);
         if (field.relationName) {
           if (!relationships[field.relationName]) {
             relationships[field.relationName] = [];
@@ -1701,6 +1736,11 @@ var generateDiagram2 = async ({
           });
         }
       });
+      validateForeignKeys({
+        foreignKeyLocation,
+        foreignKeys,
+        mermaidLines
+      });
       mermaidLines.push(`	}`);
     });
     schemaEnums.forEach((enumDef) => {
@@ -1732,7 +1772,7 @@ var prismaGenerators = /* @__PURE__ */ new Map([
 // package.json
 var package_default = {
   name: "@nonsoo/prisma-mermaid",
-  version: "0.1.2",
+  version: "0.1.3",
   description: "A Prisma generator that generates Mermaid Class or ER diagrams from your Prisma schema.",
   main: "build/index.js",
   bin: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nonsoo/prisma-mermaid",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "A Prisma generator that generates Mermaid Class or ER diagrams from your Prisma schema.",
   "main": "build/index.js",
   "bin": {