prisma-markdown/src/executable/markdown.ts

Prisma Markdown documents generator including ERD diagrams and comment descriptions

#!/usr/bin/env node
import fs from "fs";
import path from "path";

import { generatorHandler } from "@prisma/generator-helper";
import { PrismaMarkdown } from "../PrismaMarkdown";
import { IPrismaMarkdownConfig } from "../IPrismaMarkdownConfig";

const { version } = require("../../package.json");

generatorHandler({
  onManifest: () => ({
    version,
    defaultOutput: "./ERD.md",
    prettyName: "prisma-markdown",
  }),

  onGenerate: async (options) => {
    const content: string = PrismaMarkdown.write(
      options.dmmf.datamodel,
      options.generator.config as IPrismaMarkdownConfig | undefined,
    );
    const file: string = options.generator.output?.value ?? "./ERD.md";
    try {
      await fs.promises.mkdir(path.dirname(file), { recursive: true });
    } catch {}
    await fs.promises.writeFile(file, content, "utf8");
  },
});