fix: Emit metadata and enabled emitSingle cause TDZ issue

closes: #16

Author: unlight

Taskfile

@@ -22,10 +22,13 @@ compatibilty_check() {
 
     # emitSingle and emitCompiled
     sed -i 's/emitSingle = false/emitSingle = true/g' prisma/schema.prisma
-    rm -rf src/@generated && npm run prisma:g && npx tsc --noEmit --skipLibCheck --experimentalDecorators src/@generated/index.ts
+    rm -rf src/@generated && npm run prisma:g
+    npx tsc --noEmit --skipLibCheck --experimentalDecorators src/@generated/index.ts
+    npx ts-node src/@generated/index.ts
 
     sed -i 's/emitCompiled = false/emitCompiled = true/g' prisma/schema.prisma
-    rm -rf src/@generated && npm run prisma:g && node src/@generated/index.js
+    rm -rf src/@generated && npm run prisma:g
+    node src/@generated/index.js
 
     sed -i 's/emitCompiled = true/emitCompiled = false/g' prisma/schema.prisma
     sed -i 's/emitSingle = true/emitSingle = false/g' prisma/schema.prisma

src/generate.spec.ts

@@ -8,6 +8,7 @@ import {
     SourceFile,
 } from 'ts-morph';
 
+import { User } from './@generated/user/user.model';
 import { generate } from './generate';
 import { generateFileName } from './helpers/generate-file-name';
 import {
@@ -1459,7 +1460,7 @@ describe('emit single', () => {
         }
         model Post {
           id     Int   @id
-          User   User? @relation(fields: [userId], references: [id])
+          user   User? @relation(fields: [userId], references: [id])
           userId Int?
         }
     `;
@@ -1492,6 +1493,28 @@ describe('emit single', () => {
             expect(sourceText).toMatch(/export class Post {/);
         });
 
+        it('should use InstanceType trick to avoid tdz', () => {
+            const type = sourceFile
+                .getClass('Post')
+                ?.getProperty('user')
+                ?.getStructure().type;
+            expect(type).toEqual('InstanceType<typeof User>');
+        });
+
+        it('type for all properties should use InstanceType trick to avoid tdz', () => {
+            const types = sourceFile
+                .getClasses()
+                .flatMap(c => c.getProperties().map(p => ({ c, p })))
+                .map(({ c, p }) => ({ c, p, type: p.getType() }))
+                .filter(({ type }) => type.isClass())
+                .map(({ c, p }) => ({
+                    class: c.getName(),
+                    property: p.getStructure().name,
+                    type: p.getStructure().type,
+                }));
+            expect(types).toHaveLength(0);
+        });
+
         // it('^', () => console.log(sourceFile.getText()));
     });
 

src/generate.ts

@@ -7,6 +7,7 @@ import { Project, QuoteKind } from 'ts-morph';
 import { argsType } from './handlers/args-type';
 import { combineScalarFilters } from './handlers/combine-scalar-filters';
 import { createAggregateInput } from './handlers/create-aggregate-input';
+import { emitSingle } from './handlers/emit-single';
 import { beforeGenerateFiles, generateFiles } from './handlers/generate-files';
 import { inputType } from './handlers/input-type';
 import { modelData } from './handlers/model-data';
@@ -81,13 +82,18 @@ export async function generate(
             quoteKind: QuoteKind.Single,
         },
     });
+
     if (!skipAddOutputSourceFiles) {
-        project.addSourceFilesAtPaths(`${generatorOutputValue}/**/*.ts`);
+        project.addSourceFilesAtPaths([
+            `${generatorOutputValue}/**/*.ts`,
+            `!${generatorOutputValue}/**/*.d.ts`,
+        ]);
     }
 
     config.combineScalarFilters && combineScalarFilters(eventEmitter);
     config.noAtomicOperations && noAtomicOperations(eventEmitter);
     config.reExport !== ReExport.None && reExport(eventEmitter);
+    config.emitSingle && emitSingle(eventEmitter);
 
     const models = new Map<string, Model>();
     const modelNames: string[] = [];

src/handlers/emit-single.ts

@@ -0,0 +1,18 @@
+import AwaitEventEmitter from 'await-event-emitter';
+import { PropertyDeclarationStructure } from 'ts-morph';
+
+import { DMMF } from '../types';
+
+export function emitSingle(emitter: AwaitEventEmitter) {
+    emitter.on('ClassProperty', classProperty);
+}
+
+function classProperty(
+    property: PropertyDeclarationStructure,
+    eventArguments: { location: DMMF.FieldLocation; isList: boolean },
+) {
+    const { location, isList } = eventArguments;
+    if (['inputObjectTypes', 'outputObjectTypes'].includes(location) && !isList) {
+        property.type = `InstanceType<typeof ${String(property.type)}>`;
+    }
+}

src/handlers/generate-files.ts

@@ -1,7 +1,6 @@
 import {
     ClassDeclarationStructure,
     ImportSpecifierStructure,
-    ScriptKind,
     StatementStructures,
     StructureKind,
 } from 'ts-morph';

src/handlers/input-type.ts

@@ -122,6 +122,8 @@ export function inputType(
                 }),
             ],
         });
+
+        eventEmitter.emitSync('ClassProperty', property, { location, isList });
     }
 
     sourceFile.set({

src/handlers/model-output-type.ts

@@ -17,7 +17,14 @@ import { propertyStructure } from '../helpers/property-structure';
 import { EventArguments, OutputType } from '../types';
 
 export function modelOutputType(outputType: OutputType, args: EventArguments) {
-    const { getSourceFile, models, config, modelFields, fieldSettings } = args;
+    const {
+        getSourceFile,
+        models,
+        config,
+        modelFields,
+        fieldSettings,
+        eventEmitter,
+    } = args;
     const model = models.get(outputType.name);
     assert(model, `Cannot find model by name ${outputType.name}`);
     const sourceFile = getSourceFile({
@@ -149,6 +156,8 @@ export function modelOutputType(outputType: OutputType, args: EventArguments) {
                 ],
             });
         }
+
+        eventEmitter.emitSync('ClassProperty', property, { location, isList });
     }
 
     const hasExportDeclaration = (sourceFileStructure.statements as StatementStructures[]).some(

src/handlers/output-type.ts

@@ -129,6 +129,8 @@ export function outputType(outputType: OutputType, args: EventArguments) {
                 ],
             });
         }
+
+        eventEmitter.emitSync('ClassProperty', property, { location, isList });
     }
 
     sourceFile.set({

src/testing/compatibility.ts

@@ -1,4 +1,5 @@
 import { Prisma, PrismaClient } from '@prisma/client';
+import * as P from '@prisma/client';
 import { Decimal } from '@prisma/client/runtime';
 
 import { DummyCreateInput } from '../@generated/dummy/dummy-create.input';
@@ -9,6 +10,7 @@ import { StringFilter } from '../@generated/prisma/string-filter.input';
 import { AggregateUserArgs } from '../@generated/user/aggregate-user.args';
 import { FindManyUserArgs } from '../@generated/user/find-many-user.args';
 import { GroupByUserArgs } from '../@generated/user/group-by-user.args';
+import { User } from '../@generated/user/user.model';
 import { UserCreateInput } from '../@generated/user/user-create.input';
 import { UserCreateWithoutArticlesInput } from '../@generated/user/user-create-without-articles.input';
 import { UserCreateWithoutCommentsInput } from '../@generated/user/user-create-without-comments.input';
@@ -119,12 +121,6 @@ const $prisma = new PrismaClient();
     let p: Prisma.DummyCreateInput['json'] = {};
     p = x;
 }
-// {
-//     // todo: get property type should return ['Decimal', 'number', 'string']; where Decimal is import from
-//     const x: DummyCreateInput['decimal'] = new Decimal(1);
-//     let p: Prisma.DummyCreateInput['decimal'];
-//     p = x;
-// }
 {
     const x: UserCreateInput = {
         email: '',
@@ -138,3 +134,24 @@ const $prisma = new PrismaClient();
     };
     p = x;
 }
+// {
+//     // incompatible
+//     let x: User = {
+//         id: '',
+//         email: '',
+//         name: '',
+//         password: '',
+//     };
+//     let p: P.User = {
+//         id: '',
+//         email: '',
+//         name: '',
+//         password: '',
+//         bio: null,
+//         image: null,
+//         countComments: null,
+//         rating: null,
+//         role: null,
+//     };
+//     x = p;
+// }