fix: Get model name for CompoundUniqueInput

close: #53

Author: unlight

@generated/user/user-email-name-compound-unique.input.ts

@@ -0,0 +1,15 @@
+import { Field } from '@nestjs/graphql';
+import { InputType } from '@nestjs/graphql';
+import * as Scalars from 'graphql-scalars';
+import * as Validator from 'class-validator';
+
+@InputType()
+export class UserEmailNameCompoundUniqueInput {
+    @Field(() => Scalars.GraphQLEmailAddress, { nullable: false })
+    email!: string;
+
+    @Field(() => String, { nullable: false })
+    @Validator.MinLength(3)
+    @Validator.MaxLength(50)
+    name!: string;
+}

@generated/user/user-where-unique.input.ts

@@ -2,6 +2,7 @@ import { Field } from '@nestjs/graphql';
 import { InputType } from '@nestjs/graphql';
 import * as Scalars from 'graphql-scalars';
 import * as Validator from 'class-validator';
+import { UserEmailNameCompoundUniqueInput } from './user-email-name-compound-unique.input';
 
 @InputType()
 export class UserWhereUniqueInput {
@@ -15,4 +16,7 @@ export class UserWhereUniqueInput {
     @Validator.MinLength(3)
     @Validator.MaxLength(50)
     name?: string;
+
+    @Field(() => UserEmailNameCompoundUniqueInput, { nullable: true })
+    email_name?: UserEmailNameCompoundUniqueInput;
 }

prisma/schema.prisma

@@ -57,6 +57,8 @@ model User {
   rating           Float?
   role             Role?
   profile          Profile?
+
+  @@unique([email, name])
 }
 
 model Tag {

src/generate.ts

@@ -112,7 +112,7 @@ export async function generate(
         eventEmitter,
         typeNames: new Set<string>(),
         enums: mapKeys(datamodel.enums, x => x.name),
-        getModelName: createGetModelName(modelNames),
+        getModelName,
         removeTypes,
     };
 

src/helpers/get-model-name.ts

@@ -1,4 +1,4 @@
-import { memoize } from 'lodash';
+import { first, memoize } from 'lodash';
 
 export function createGetModelName(modelNames: string[]) {
     return memoize(tryGetName);
@@ -35,6 +35,16 @@ function getModelName(args: {
             return test;
         }
     }
+
+    // test for {Model}{UniqueName}CompoundUniqueInput
+    if (name.slice(-19) === 'CompoundUniqueInput') {
+        const test = name.slice(0, -19);
+        const models = modelNames
+            .filter(x => test.startsWith(x))
+            .sort((a, b) => b.length - a.length);
+        return first(models);
+    }
+
     // test for {Model}Count
     if (name.slice(-5) === 'Count') {
         const test = name.slice(0, -5);

src/helpers/object-settings.ts

@@ -1,6 +1,7 @@
 import JSON5 from 'json5';
 import { isObject, merge, omit, trim } from 'lodash';
 import outmatch from 'outmatch';
+import { PlainObject } from 'simplytyped';
 
 import { GeneratorConfiguration } from '../types';
 
@@ -53,7 +54,7 @@ export class ObjectSettings extends Array<ObjectSetting> {
         const resultArguments: any[] = [objectTypeOptions];
         const objectType = this.find(s => s.kind === 'ObjectType');
         if (objectType && isObject(objectType.arguments)) {
-            const name = objectType.arguments.name;
+            const name = (objectType.arguments as PlainObject).name;
             merge(objectTypeOptions, omit(objectType.arguments, 'name'));
             if (name) {
                 resultArguments.unshift(name);

src/test/generate.spec.ts

@@ -34,9 +34,9 @@ const t = (name: string) =>
     getPropertyStructure(sourceFile, name)?.decorators?.find(d => d.name === 'Field')
         ?.arguments?.[0];
 const setSourceFile = (name: string) => {
-    sourceFile = project.getSourceFile(s => s.getFilePath().endsWith(name))!;
-    classFile = sourceFile.getClass(() => true)!;
+    sourceFile = project.getSourceFileOrThrow(s => s.getFilePath().endsWith(name));
     sourceText = sourceFile.getText();
+    classFile = sourceFile.getClass(() => true)!;
     importDeclarations = sourceFile.getImportDeclarations().map(d => d.getStructure());
     imports = importDeclarations.flatMap(d =>
         (d.namedImports as ImportSpecifierStructure[]).map(x => ({
@@ -2051,3 +2051,32 @@ describe('object model options', () => {
         });
     });
 });
+
+describe('compound index', () => {
+    it('user unique input compound', async () => {
+        ({ project, sourceFiles } = await testGenerate({
+            schema: `
+                model User {
+                  id    Int    @id
+                  /// @Validator.MinLength(3)
+                  name String
+                  /// @PropertyType({ name: 'G.Email', from: 'graphql-type-email' })
+                  email String?
+
+                  @@unique([email, name])
+                }
+                model Us {
+                  id    Int    @id
+                }
+                `,
+            options: [
+                `outputFilePattern = "{name}.{type}.ts"`,
+                `fields_Validator_from = "class-validator"`,
+                `fields_Validator_input = true`,
+            ],
+        }));
+        setSourceFile('user-email-name-compound-unique.input.ts');
+        const minLength = classFile.getProperty('name')?.getDecorator('MinLength');
+        expect(minLength?.getText()).toEqual('@Validator.MinLength(3)');
+    });
+});