captable/prisma/enum-generator.cjs

/* eslint-disable */
// @ts-nocheck

/// credit : https://github.com/calcom/cal.com/blob/main/packages/prisma/enum-generator.ts

const { generatorHandler } = require("@prisma/generator-helper");
const fs = require("node:fs/promises");
const path = require("node:path");

const header = `// This file was generated by a custom prisma generator, do not edit manually.\n`;

generatorHandler({
  onManifest() {
    return {
      defaultOutput: "./enums.ts",
      prettyName: "Prisma Enum Generator",
    };
  },
  async onGenerate(options) {
    const enums = options.dmmf.datamodel.enums;

    const output = enums.map((e) => {
      let enumString = `export const ${e.name} = {\n`;
      e.values.forEach(({ name: value }) => {
        enumString += `  ${value}: "${value}",\n`;
      });
      enumString += `} as const;\n\n`;
      enumString += `export type ${e.name} = (typeof ${e.name})[keyof typeof ${e.name}];\n`;

      return enumString;
    });

    const outputFile = options.generator.output;
    if (!outputFile?.value) {
      throw new Error("No output file specified");
    }

    const outputPath = path.resolve(outputFile.value);
    await fs.mkdir(path.dirname(outputPath), { recursive: true });
    await fs.writeFile(outputPath, header + output.join("\n"), "utf-8");
  },
});