Add option to add js extensions to generated ts files

README.md

@@ -146,6 +146,10 @@ npm install -g yarn
 
       imports                  = null // (default) will import the referenced file in generated schemas to be used via imports.someExportedVariable
 
+      // https://devblogs.microsoft.com/typescript/announcing-typescript-4-7-beta/#ecmascript-module-support-in-node-js
+      includeJSExtension       = true // includes the .js extension on relative import/exports to comply with ESM in TS syntax requirements ()
+      // includeJSExtension    = false // (default) leaves relative import/exports extensionless
+
       // https://www.prisma.io/docs/concepts/components/prisma-client/working-with-fields/working-with-json-fields#filtering-by-null-values
       prismaJsonNullability    = true // (default) uses prisma's scheme for JSON field nullability
       // prismaJsonNullability = false // allows null assignment to optional JSON fields

src/config.ts

@@ -8,6 +8,7 @@ export const configSchema = z.object({
 	modelCase: z.enum(['PascalCase', 'camelCase']).default('PascalCase'),
 	useDecimalJs: configBoolean.default('false'),
 	imports: z.string().optional(),
+	includeJSExtension: configBoolean.default('false'),
 	prismaJsonNullability: configBoolean.default('true'),
 })
 

src/generator.ts

@@ -63,7 +63,7 @@ export const writeImportsForModel = (
 		if (filteredFields.length > 0) {
 			importList.push({
 				kind: StructureKind.ImportDeclaration,
-				moduleSpecifier: './index',
+				moduleSpecifier: `./index${jsExt(config)}`,
 				namedImports: Array.from(
 					new Set(
 						filteredFields.flatMap((f) => [
@@ -239,10 +239,12 @@ export const populateModelFile = (
 		generateRelatedSchemaForModel(model, sourceFile, config, prismaOptions)
 }
 
-export const generateBarrelFile = (models: DMMF.Model[], indexFile: SourceFile) => {
+export const generateBarrelFile = (models: DMMF.Model[], indexFile: SourceFile, config: Config) => {
 	models.forEach((model) =>
 		indexFile.addExportDeclaration({
-			moduleSpecifier: `./${model.name.toLowerCase()}`,
+			moduleSpecifier: `./${model.name.toLowerCase()}${jsExt(config)}`
 		})
 	)
 }
+
+const jsExt = (config: Config) => config.includeJSExtension ? '.js' : ''

src/index.ts

@@ -45,7 +45,7 @@ generatorHandler({
 			{ overwrite: true }
 		)
 
-		generateBarrelFile(models, indexFile)
+		generateBarrelFile(models, indexFile, config)
 
 		indexFile.formatText({
 			indentSize: 2,

src/test/functional/driver.test.ts

@@ -45,7 +45,7 @@ const ftForDir = (dir: string) => async () => {
 
 	const indexFile = project.createSourceFile(`${outputPath}/index.ts`, {}, { overwrite: true })
 
-	generateBarrelFile(dmmf.datamodel.models, indexFile)
+	generateBarrelFile(dmmf.datamodel.models, indexFile, config)
 
 	indexFile.formatText({
 		indentSize: 2,
@@ -105,6 +105,7 @@ describe('Functional Tests', () => {
 	test.concurrent('Config', ftForDir('config'))
 	test.concurrent('Docs', ftForDir('docs'))
 	test.concurrent('Different Client Path', ftForDir('different-client-path'))
+	test.concurrent('JS Extension', ftForDir('js-extension'))
 	test.concurrent('Recursive Schema', ftForDir('recursive'))
 	test.concurrent('relationModel = false', ftForDir('relation-false'))
 	test.concurrent('Relation - 1 to 1', ftForDir('relation-1to1'))

src/test/functional/js-extension/expected/document.ts

@@ -0,0 +1,10 @@
+import * as z from "zod"
+
+export const DocumentModel = z.object({
+  id: z.string(),
+  filename: z.string(),
+  author: z.string(),
+  contents: z.string(),
+  created: z.date(),
+  updated: z.date(),
+})

src/test/functional/js-extension/expected/index.ts

@@ -0,0 +1 @@
+export * from "./document.js"

src/test/functional/js-extension/prisma/.client/index.ts

@@ -0,0 +1,12 @@
+/**
+ * Model Document
+ *
+ */
+export type Document = {
+	id: string
+	filename: string
+	author: string
+	contents: string
+	created: Date
+	updated: Date
+}

src/test/functional/js-extension/prisma/schema.prisma

@@ -0,0 +1,28 @@
+// This is your Prisma schema file,
+// learn more about it in the docs: https://pris.ly/d/prisma-schema
+
+datasource db {
+  provider = "postgresql"
+  url      = ""
+}
+
+generator client {
+  provider = "prisma-client-js"
+  output   = ".client"
+}
+
+generator zod {
+  provider           = "zod-prisma"
+  output             = "../actual/"
+	includeJSExtension = "true"
+}
+
+model Document {
+  id       String @id @default(cuid())
+  filename String @unique
+  author   String
+  contents String
+
+  created DateTime @default(now())
+  updated DateTime @default(now())
+}