From aa7ead3e8c2608f83c9f3c87c3b158163a74b991 Mon Sep 17 00:00:00 2001
From: "gitstart-app[bot]"
 <57568882+gitstart-app[bot]@users.noreply.github.com>
Date: Fri, 1 Mar 2024 16:39:40 +0100
Subject: [PATCH] TWNTY-3942 - Enable Attachments on Custom Objects (#4253)

* Enable Attachments on Custom Objects

Co-authored-by: v1b3m <vibenjamin6@gmail.com>

* Revert changes to the client

Co-authored-by: v1b3m <vibenjamin6@gmail.com>

---------

Co-authored-by: gitstart-twenty <gitstart-twenty@users.noreply.github.com>
Co-authored-by: v1b3m <vibenjamin6@gmail.com>
---
 .../object-metadata.service.ts                | 124 ++++++++++++++++++
 1 file changed, 124 insertions(+)

diff --git a/packages/twenty-server/src/metadata/object-metadata/object-metadata.service.ts b/packages/twenty-server/src/metadata/object-metadata/object-metadata.service.ts
index 56c692b4a..e18056139 100644
--- a/packages/twenty-server/src/metadata/object-metadata/object-metadata.service.ts
+++ b/packages/twenty-server/src/metadata/object-metadata/object-metadata.service.ts
@@ -320,6 +320,11 @@ export class ObjectMetadataService extends TypeOrmQueryService<ObjectMetadataEnt
       createdObjectMetadata,
     );
 
+    const { attachmentObjectMetadata } = await this.createAttachmentRelation(
+      objectMetadataInput.workspaceId,
+      createdObjectMetadata,
+    );
+
     await this.workspaceMigrationService.createCustomMigration(
       generateMigrationName(`create-${createdObjectMetadata.nameSingular}`),
       createdObjectMetadata.workspaceId,
@@ -360,6 +365,37 @@ export class ObjectMetadataService extends TypeOrmQueryService<ObjectMetadataEnt
             },
           ],
         },
+        // Add attachment relation
+        {
+          name: computeObjectTargetTable(attachmentObjectMetadata),
+          action: 'alter',
+          columns: [
+            {
+              action: WorkspaceMigrationColumnActionType.CREATE,
+              columnName: `${computeObjectTargetTable(
+                createdObjectMetadata,
+              )}Id`,
+              columnType: 'uuid',
+              isNullable: true,
+            } satisfies WorkspaceMigrationColumnCreate,
+          ],
+        },
+        {
+          name: computeObjectTargetTable(attachmentObjectMetadata),
+          action: 'alter',
+          columns: [
+            {
+              action: WorkspaceMigrationColumnActionType.CREATE_FOREIGN_KEY,
+              columnName: `${computeObjectTargetTable(
+                createdObjectMetadata,
+              )}Id`,
+              referencedTableName: computeObjectTargetTable(
+                createdObjectMetadata,
+              ),
+              referencedTableColumnName: 'id',
+            },
+          ],
+        },
         // Add favorite relation
         {
           name: computeObjectTargetTable(favoriteObjectMetadata),
@@ -621,6 +657,94 @@ export class ObjectMetadataService extends TypeOrmQueryService<ObjectMetadataEnt
     return { activityTargetObjectMetadata };
   }
 
+  private async createAttachmentRelation(
+    workspaceId: string,
+    createdObjectMetadata: ObjectMetadataEntity,
+  ) {
+    const attachmentObjectMetadata =
+      await this.objectMetadataRepository.findOneByOrFail({
+        nameSingular: 'attachment',
+        workspaceId: workspaceId,
+      });
+
+    const attachmentRelationFieldMetadata =
+      await this.fieldMetadataRepository.save([
+        // FROM
+        {
+          objectMetadataId: createdObjectMetadata.id,
+          workspaceId: workspaceId,
+          isCustom: true,
+          isActive: true,
+          type: FieldMetadataType.RELATION,
+          name: 'attachments',
+          label: 'Attachments',
+          targetColumnMap: {},
+          description: `Attachments tied to the ${createdObjectMetadata.labelSingular}`,
+          icon: 'IconFileImport',
+          isNullable: true,
+        },
+        // TO
+        {
+          objectMetadataId: attachmentObjectMetadata.id,
+          workspaceId: workspaceId,
+          isCustom: true,
+          isActive: true,
+          type: FieldMetadataType.RELATION,
+          name: createdObjectMetadata.nameSingular,
+          label: createdObjectMetadata.labelSingular,
+          targetColumnMap: {},
+          description: `Attachment ${createdObjectMetadata.labelSingular}`,
+          icon: 'IconBuildingSkyscraper',
+          isNullable: true,
+        },
+        // Foreign key
+        {
+          objectMetadataId: attachmentObjectMetadata.id,
+          workspaceId: workspaceId,
+          isCustom: true,
+          isActive: true,
+          type: FieldMetadataType.UUID,
+          name: `${createdObjectMetadata.nameSingular}Id`,
+          label: `${createdObjectMetadata.labelSingular} ID (foreign key)`,
+          targetColumnMap: {
+            value: `${computeObjectTargetTable(createdObjectMetadata)}Id`,
+          },
+          description: `Attachment ${createdObjectMetadata.labelSingular} id foreign key`,
+          icon: undefined,
+          isNullable: true,
+          isSystem: true,
+          defaultValue: undefined,
+        },
+      ]);
+
+    const attachmentRelationFieldMetadataMap =
+      attachmentRelationFieldMetadata.reduce(
+        (acc, fieldMetadata: FieldMetadataEntity) => {
+          if (fieldMetadata.type === FieldMetadataType.RELATION) {
+            acc[fieldMetadata.objectMetadataId] = fieldMetadata;
+          }
+
+          return acc;
+        },
+        {},
+      );
+
+    await this.relationMetadataRepository.save([
+      {
+        workspaceId: workspaceId,
+        relationType: RelationMetadataType.ONE_TO_MANY,
+        fromObjectMetadataId: createdObjectMetadata.id,
+        toObjectMetadataId: attachmentObjectMetadata.id,
+        fromFieldMetadataId:
+          attachmentRelationFieldMetadataMap[createdObjectMetadata.id].id,
+        toFieldMetadataId:
+          attachmentRelationFieldMetadataMap[attachmentObjectMetadata.id].id,
+      },
+    ]);
+
+    return { attachmentObjectMetadata };
+  }
+
   private async createFavoriteRelation(
     workspaceId: string,
     createdObjectMetadata: ObjectMetadataEntity,