Merge pull request #24 from JackCme/feature/swagger-decorator-relation-type

Fix/swagger decorator relation type

Author: omar-dulaimi

src/helpers.ts

@@ -200,6 +200,10 @@ export const getSwaggerDecoratorByFieldType = (field: PrismaDMMF.Field) => {
     args.push(`enum: Object.values(${field.type})`);
   }
 
+  if (field.relationName) {
+    args.push(`type: () => ${field.type}`);
+  }
+
   return {
     name: 'ApiProperty',
     arguments: args.length > 0 ? [`{ ${args.join(', ')} }`] : [],

tests/relation-splitting.test.ts

@@ -80,7 +80,9 @@ describe('Relation Splitting Generation', () => {
     expect(userRelations).toContain('import { Post } from "./"');
 
     // Should have decorators for relations
-    expect(userRelations).toContain('@ApiProperty({ isArray: true })');
+    expect(userRelations).toContain(
+      '@ApiProperty({ isArray: true, type: () => Post })',
+    );
   });
 
   it('should generate combined User class extending base', () => {

tests/swagger-generation.test.ts

@@ -32,6 +32,7 @@ describe('Swagger Generation', () => {
     expect(userModel).toContain('type: "string"');
     expect(userModel).toContain('required: false');
     expect(userModel).toContain('isArray: true');
+    expect(userModel).toContain('type: () => Post');
   });
 
   it('should generate Post model with correct Swagger decorators', () => {