feat: Generate commented class if re-export found

Author: unlight

src/example/user/user.resolver.ts

@@ -1,7 +1,7 @@
 import { Args, Info, Mutation, Query, Resolver } from '@nestjs/graphql';
 import { PrismaSelect } from '@paljs/plugins';
 import { PrismaClient } from '@prisma/client';
-import * as Prisma from '@prisma/client';
+import { GraphQLResolveInfo } from 'graphql';
 
 import { AggregateUserArgs } from '../../@generated/user/aggregate-user.args';
 import { AggregateUser } from '../../@generated/user/aggregate-user.output';
@@ -33,7 +33,7 @@ export class UserResolver {
      * Query for single user.
      */
     @Query(() => [User])
-    async users(@Args('where') where: UserWhereInput, @Info() info) {
+    async users(@Args('where') where: UserWhereInput, @Info() info: GraphQLResolveInfo) {
         const select = new PrismaSelect(info).value;
         // console.log('select', select);
         return await prisma.user.findMany({ where, ...select });
@@ -45,10 +45,7 @@ export class UserResolver {
     }
 
     @Query(() => AggregateUser)
-    async userAggregate(@Args() args: AggregateUserArgs, @Info() info) {
-        console.log('args', args);
-        const result = await prisma.user.aggregate(args as Prisma.AggregateUserArgs);
-        console.log('result', result);
-        return result;
+    userAggregate(@Args() args: AggregateUserArgs) {
+        return prisma.user.aggregate(args);
     }
 }

src/generate-input/generate-input.ts

@@ -4,7 +4,7 @@ import { DecoratorPropertyType, generateClass } from '../generate-class';
 import { generateImport, generateProjectImport } from '../generate-import';
 import { Field, generateProperty } from '../generate-property';
 import { GeneratorConfiguration, PrismaDMMF } from '../types';
-import { fieldLocationToKind, toPropertyType } from '../utils';
+import { checkExport, fieldLocationToKind, toPropertyType } from '../utils';
 import { getMatchingInputType } from './get-matching-input-type';
 
 type GenerateInputArgs = {
@@ -68,4 +68,6 @@ export function generateInput(args: GenerateInputArgs) {
             config,
         });
     }
+
+    checkExport({ name: className, sourceFile, classDeclaration });
 }

src/generate-model/generate-model.spec.ts

@@ -268,4 +268,21 @@ describe('generate models', () => {
             specifier: 'decimal.js',
         });
     });
+
+    it('generated commented class if reexport found', async () => {
+        await getResult({
+            schema: `model User {
+                id String @id
+            }`,
+            sourceFileText: `
+                export { User } from 'src/user/model'
+                export { User as UserModel } from 'src/user2/model'
+            `,
+        });
+        sourceText = sourceFile.getText();
+        const sourceClass = sourceFile.getClasses();
+        expect(sourceClass).toHaveLength(0);
+        expect(sourceText).toContain(`export { User } from 'src/user/model'`);
+        expect(sourceText).toContain(`// export class User`);
+    });
 });

src/generate-model/generate-model.ts

@@ -4,6 +4,7 @@ import { generateClass } from '../generate-class';
 import { generateImport } from '../generate-import';
 import { generateProperty } from '../generate-property';
 import { GeneratorConfiguration, PrismaDMMF } from '../types';
+import { checkExport } from '../utils';
 
 type GenerateModelArgs = {
     model: PrismaDMMF.Model;
@@ -37,4 +38,5 @@ export function generateModel(args: GenerateModelArgs) {
             config,
         });
     });
+    checkExport({ name: model.name, classDeclaration, sourceFile });
 }

src/generate-object.ts

@@ -4,6 +4,7 @@ import { generateClass } from './generate-class';
 import { generateImport } from './generate-import';
 import { generateProperty, Model } from './generate-property';
 import { GeneratorConfiguration } from './types';
+import { checkExport } from './utils';
 
 type GenerateObjectArgs = {
     sourceFile: SourceFile;
@@ -37,4 +38,6 @@ export function generateObject(args: GenerateObjectArgs) {
             config,
         });
     });
+
+    checkExport({ name: model.name, sourceFile, classDeclaration });
 }

src/utils/check-export.ts

@@ -0,0 +1,22 @@
+import { ClassDeclaration, SourceFile } from 'ts-morph';
+
+type CommentArgs = {
+    sourceFile: SourceFile;
+    name: string;
+    classDeclaration: ClassDeclaration;
+};
+
+export function checkExport(args: CommentArgs) {
+    const { sourceFile, name, classDeclaration } = args;
+    const exportDeclaration = sourceFile.getExportDeclaration((d) => {
+        return Boolean(d.getNamedExports().find((x) => x.getNameNode().getText() === name));
+    });
+    if (exportDeclaration) {
+        const commentedText = classDeclaration
+            .getText()
+            .split('\n')
+            .map((x) => `// ${x}`);
+        classDeclaration.remove();
+        sourceFile.addStatements(['\n', ...commentedText]);
+    }
+}

src/utils/index.ts

@@ -4,6 +4,8 @@ export { createConfig } from './create-config';
 import { ObjectLiteralExpression, PropertyAssignment, StructureKind } from 'ts-morph';
 
 import { PrismaDMMF, TypeRecord } from '../types';
+export { checkExport } from './check-export';
+
 export { generateFileName } from './generate-file-name';
 
 type ToGraphqlImportTypeArgs = {