Merge pull request #2 from risenforces/feat/local-enums

Add local enums

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[] = [
@@ -49,13 +49,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),
     })
   }
@@ -237,3 +236,21 @@ export const generateBarrelFile = (
     })
   )
 }
+
+export const generateEnumsFile = (
+  enums: DMMF.DatamodelEnum[],
+  enumsFile: SourceFile
+) => {
+  for (const { name, values } of enums) {
+    const members = values.map(({ name: memberName }) => {
+      return { name: memberName, value: memberName }
+    })
+
+    enumsFile
+      .addEnum({
+        name,
+        members,
+      })
+      .setIsExported(true)
+  }
+}

src/index.ts

@@ -2,7 +2,11 @@ import { generatorHandler } from '@prisma/generator-helper'
 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'
 
 const { version } = require('../package.json')
 
@@ -18,6 +22,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
@@ -68,6 +73,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/__snapshots__/driver.test.ts.snap

@@ -181,7 +181,7 @@ export * from \\"./spreadsheet\\"
 
 exports[`Functional Tests Imports: imports - document 1`] = `
 "import * as z from \\"nestjs-zod/z\\"
-import { Status } from \\"../prisma/.client\\"
+import { Status } from \\"./enums\\"
 
 export const DocumentModel = z.object({
   id: z.string(),

src/test/functional/driver.test.ts

@@ -4,7 +4,11 @@ import { readFile } from 'fs-extra'
 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)
 
@@ -68,6 +72,22 @@ const ftForDir = (dir: string) => async () => {
 
   expect(actualIndexContents).toMatchSnapshot(`${dir} - Index file`)
 
+  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/regressions.test.ts

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