Feature: local enums def file

src/generator.ts

@@ -15,7 +15,7 @@ export const writeImportsForModel = (
 	model: DMMF.Model,
 	sourceFile: SourceFile,
 	config: Config,
-	{ schemaPath, outputPath, clientPath }: PrismaOptions
+	{ schemaPath, outputPath }: PrismaOptions
 ) => {
 	const { relatedModelName } = useModelNames(config)
 	const importList: ImportDeclarationStructure[] = [
@@ -46,13 +46,12 @@ export const writeImportsForModel = (
 
 	const enumFields = model.fields.filter((f) => f.kind === 'enum')
 	const relationFields = model.fields.filter((f) => f.kind === 'object')
-	const relativePath = path.relative(outputPath, clientPath)
 
 	if (enumFields.length > 0) {
 		importList.push({
 			kind: StructureKind.ImportDeclaration,
 			isTypeOnly: enumFields.length === 0,
-			moduleSpecifier: dotSlash(relativePath),
+			moduleSpecifier: dotSlash('enums'),
 			namedImports: enumFields.map((f) => f.type),
 		})
 	}
@@ -246,3 +245,17 @@ export const generateBarrelFile = (models: DMMF.Model[], indexFile: SourceFile)
 		})
 	)
 }
+
+export const generateEnumsFile = (enums: DMMF.DatamodelEnum[], enumsFile: SourceFile) => {
+	enums.forEach(({ name, values }) => {
+		const members = values.map(({ name: memberName }) => {
+			return { name: memberName, value: memberName }
+		})
+		enumsFile
+			.addEnum({
+				name,
+				members,
+			})
+			.setIsExported(true)
+	})
+}

src/index.ts

@@ -4,7 +4,7 @@ import { version } from '../package.json'
 import { generatorHandler } from '@prisma/generator-helper'
 import { SemicolonPreference } from 'typescript'
 import { configSchema, PrismaOptions } from './config'
-import { populateModelFile, generateBarrelFile } from './generator'
+import { populateModelFile, generateBarrelFile, generateEnumsFile } from './generator'
 import { Project } from 'ts-morph'
 
 generatorHandler({
@@ -19,6 +19,7 @@ generatorHandler({
 		const project = new Project()
 
 		const models = options.dmmf.datamodel.models
+		const enums = options.dmmf.datamodel.enums
 
 		const { schemaPath } = options
 		const outputPath = options.generator.output!.value
@@ -69,6 +70,22 @@ generatorHandler({
 			})
 		})
 
+		if (enums.length > 0) {
+			const enumsFile = project.createSourceFile(
+				`${outputPath}/enums.ts`,
+				{},
+				{ overwrite: true }
+			)
+
+			generateEnumsFile(enums, enumsFile)
+
+			enumsFile.formatText({
+				indentSize: 2,
+				convertTabsToSpaces: true,
+				semicolons: SemicolonPreference.Remove,
+			})
+		}
+
 		return project.save()
 	},
 })

src/test/functional/driver.test.ts

@@ -6,7 +6,7 @@ import path from 'path'
 import { Project } from 'ts-morph'
 import { SemicolonPreference } from 'typescript'
 import { configSchema, PrismaOptions } from '../../config'
-import { populateModelFile, generateBarrelFile } from '../../generator'
+import { populateModelFile, generateBarrelFile, generateEnumsFile } from '../../generator'
 
 jest.setTimeout(10000)
 
@@ -65,6 +65,22 @@ const ftForDir = (dir: string) => async () => {
 
 	expect(actualIndexContents).toStrictEqual(expectedIndexContents)
 
+	if (dmmf.datamodel.enums.length > 0) {
+		const enumsFile = project.createSourceFile(
+			`${actualDir}/enums.ts`,
+			{},
+			{ overwrite: true }
+		)
+
+		generateEnumsFile(dmmf.datamodel.enums, enumsFile)
+
+		enumsFile.formatText({
+			indentSize: 2,
+			convertTabsToSpaces: true,
+			semicolons: SemicolonPreference.Remove,
+		})
+	}
+
 	await Promise.all(
 		dmmf.datamodel.models.map(async (model) => {
 			const sourceFile = project.createSourceFile(

src/test/functional/imports/expected/document.ts

@@ -1,5 +1,5 @@
 import * as z from "zod"
-import { Status } from "../prisma/.client"
+import { Status } from "./enums"
 
 export const DocumentModel = z.object({
   id: z.string(),

src/test/functional/imports/expected/enums.ts

@@ -0,0 +1,5 @@
+export enum Status {
+  draft = "draft",
+  live = "live",
+  archived = "archived"
+}

src/test/regressions.test.ts

@@ -35,7 +35,7 @@ describe('Regression Tests', () => {
 		writeImportsForModel(model, testFile, config, prismaOptions)
 
 		expect(testFile.print()).toBe(
-			'import * as z from "zod";\nimport { UserType } from "@prisma/client";\n'
+			'import * as z from "zod";\nimport { UserType } from "./enums";\n'
 		)
 	})
 })