fix(hide field): Self relation field

close: unlight#103

Author: unlight

src/handlers/input-type.ts

@@ -151,8 +151,9 @@ export function inputType(
             if (
                 graphqlImport.specifier &&
                 !importDeclarations.has(graphqlImport.name) &&
-                ((graphqlImport.name !== inputType.name && !shouldHideField) ||
-                    (shouldHideField && referenceName === graphqlImport.name))
+                graphqlImport.name !== inputType.name
+                // ((graphqlImport.name !== inputType.name && !shouldHideField) ||
+                //     (shouldHideField && referenceName === graphqlImport.name))
             ) {
                 importDeclarations.set(graphqlImport.name, {
                     namedImports: [{ name: graphqlImport.name }],

src/test/helpers.ts

@@ -28,13 +28,20 @@ export function getPropertyStructure(sourceFile: SourceFile, name: string) {
 
 export function testSourceFile(args: {
     project: Project;
-    file: string;
+    file?: string;
+    class?: string;
     property?: string;
 }) {
-    const { project, file, property } = args;
-    const sourceFile = project.getSourceFileOrThrow(s =>
-        s.getFilePath().endsWith(file),
-    );
+    const { project, file, property, class: className } = args;
+    let sourceFile: SourceFile;
+    if (file) {
+        sourceFile = project.getSourceFileOrThrow(s => s.getFilePath().endsWith(file));
+    } else if (className) {
+        sourceFile = project.getSourceFileOrThrow(s => Boolean(s.getClass(className)));
+    } else {
+        throw new TypeError('file or class must be provided');
+    }
+
     const importDeclarations = sourceFile
         .getImportDeclarations()
         .map(d => d.getStructure());

src/test/hide-field.spec.ts

@@ -323,3 +323,36 @@ describe.skip('hide enum', () => {
         }
     });
 });
+
+describe('hide with self reference', () => {
+    before(async () => {
+        ({ project } = await testGenerate({
+            schema: `model User {
+                  id     Int    @id
+                  parentId Int
+                  /// @HideField({ output: false, input: true })
+                  parent User   @relation("UserToUser", fields: [parentId], references: [id])
+                  /// @HideField({ output: false, input: true })
+                  user   User[] @relation("UserToUser")
+                }`,
+        }));
+    });
+
+    it('smoke', () => {
+        const files = project.getSourceFiles().map(s => s.getBaseName());
+        expect(files).toBeTruthy();
+    });
+
+    it('order by with relation self import', () => {
+        const s = testSourceFile({
+            project,
+            class: 'UserOrderByWithRelationAndSearchRelevanceInput',
+        });
+
+        expect(s.namedImports).not.toContainEqual(
+            expect.objectContaining({
+                name: 'UserOrderByWithRelationAndSearchRelevanceInput',
+            }),
+        );
+    });
+});