Icon microsoft (#9907)

Implementing the Outlook icon for CreatedBy, only for emails. Not in this PR original scope : The similar feature for calendar created records. Since it was straightforward, I added it to the scope of this PR. Fix https://github.com/twentyhq/core-team-issues/issues/252
2025-01-30 17:09:42 +01:00
parent 9ec524213c
commit a5273732b3
42 changed files with 530 additions and 93 deletions
--- a/packages/twenty-server/src/database/commands/upgrade-version/0-41/0-41-add-context-to-actor-composite-type.ts
+++ b/packages/twenty-server/src/database/commands/upgrade-version/0-41/0-41-add-context-to-actor-composite-type.ts
@ -0,0 +1,191 @@
+import { InjectRepository } from '@nestjs/typeorm';
+
+import chalk from 'chalk';
+import { Command } from 'nest-commander';
+import { FieldMetadataType } from 'twenty-shared';
+import { In, Repository, TableColumn } from 'typeorm';
+
+import {
+  ActiveWorkspacesCommandOptions,
+  ActiveWorkspacesCommandRunner,
+} from 'src/database/commands/active-workspaces.command';
+import { CommandLogger } from 'src/database/commands/logger';
+import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity';
+import { FieldActorSource } from 'src/engine/metadata-modules/field-metadata/composite-types/actor.composite-type';
+import { FieldMetadataEntity } from 'src/engine/metadata-modules/field-metadata/field-metadata.entity';
+import { WorkspaceMetadataVersionService } from 'src/engine/metadata-modules/workspace-metadata-version/services/workspace-metadata-version.service';
+import { TwentyORMGlobalManager } from 'src/engine/twenty-orm/twenty-orm-global.manager';
+import { computeTableName } from 'src/engine/utils/compute-table-name.util';
+import { WorkspaceDataSourceService } from 'src/engine/workspace-datasource/workspace-datasource.service';
+
+@Command({
+  name: 'upgrade-0.41:add-context-to-actor-composite-type',
+  description: 'Add context to actor composite type.',
+})
+export class AddContextToActorCompositeTypeCommand extends ActiveWorkspacesCommandRunner {
+  protected readonly logger;
+
+  constructor(
+    @InjectRepository(Workspace, 'core')
+    protected readonly workspaceRepository: Repository<Workspace>,
+    private readonly twentyORMGlobalManager: TwentyORMGlobalManager,
+    @InjectRepository(FieldMetadataEntity, 'metadata')
+    private readonly fieldMetadataRepository: Repository<FieldMetadataEntity>,
+    private readonly workspaceDataSourceService: WorkspaceDataSourceService,
+    private readonly workspaceMetadataVersionService: WorkspaceMetadataVersionService,
+  ) {
+    super(workspaceRepository);
+    this.logger = new CommandLogger({
+      constructorName: this.constructor.name,
+      verbose: false,
+    });
+    this.logger.setVerbose(false);
+  }
+
+  async executeActiveWorkspacesCommand(
+    _passedParam: string[],
+    options: ActiveWorkspacesCommandOptions,
+    workspaceIds: string[],
+  ): Promise<void> {
+    this.logger.log(`Running add-context-to-actor-composite-type command`);
+
+    if (options?.dryRun) {
+      this.logger.log(chalk.yellow('Dry run mode: No changes will be applied'));
+    }
+
+    for (const workspaceId of workspaceIds) {
+      try {
+        await this.execute(workspaceId, options?.dryRun);
+        this.logger.verbose(`Added for workspace: ${workspaceId}`);
+      } catch (error) {
+        this.logger.error(`Error for workspace: ${workspaceId}`, error);
+      }
+    }
+  }
+
+  private async execute(workspaceId: string, dryRun = false): Promise<void> {
+    this.logger.verbose(`Adding for workspace: ${workspaceId}`);
+    const actorFields = await this.fieldMetadataRepository.find({
+      where: {
+        type: FieldMetadataType.ACTOR,
+        workspaceId,
+      },
+      relations: ['object'],
+    });
+
+    // Filter and update fields with EMAIL or CALENDAR source
+    for (const field of actorFields) {
+      if (!field || !field.object) {
+        this.logger.verbose(
+          'field.objectMetadata is null',
+          workspaceId,
+          field.id,
+        );
+        continue;
+      }
+
+      await this.addContextColumn(
+        field,
+        `${field.name}Context`,
+        workspaceId,
+        dryRun,
+      );
+
+      const fieldRepository =
+        await this.twentyORMGlobalManager.getRepositoryForWorkspace(
+          workspaceId,
+          field.object.nameSingular,
+        );
+
+      if (!dryRun) {
+        await this.workspaceMetadataVersionService.incrementMetadataVersion(
+          workspaceId,
+        );
+      }
+
+      if (!dryRun) {
+        const rowsToUpdate = await fieldRepository.update(
+          {
+            [field.name + 'Source']: In([
+              FieldActorSource.EMAIL,
+              FieldActorSource.CALENDAR,
+            ]),
+            [field.name + 'Context']: {},
+          },
+          {
+            [field.name + 'Context']: {
+              provider: 'google',
+            },
+          },
+        );
+
+        this.logger.verbose(
+          `updated ${rowsToUpdate ? rowsToUpdate.affected : 0} rows`,
+        );
+      }
+    }
+  }
+
+  private async addContextColumn(
+    field: FieldMetadataEntity,
+    newColumnName: string,
+    workspaceId: string,
+    dryRun = false,
+  ): Promise<void> {
+    const workspaceDataSource =
+      await this.workspaceDataSourceService.connectToWorkspaceDataSource(
+        workspaceId,
+      );
+
+    if (!workspaceDataSource) {
+      this.logger.verbose('No workspace data source found');
+
+      return;
+    }
+
+    const queryRunner = workspaceDataSource?.createQueryRunner();
+
+    await queryRunner.connect();
+    await queryRunner.startTransaction();
+
+    const schemaName =
+      this.workspaceDataSourceService.getSchemaName(workspaceId);
+
+    try {
+      const hasColumn = await queryRunner.hasColumn(
+        `${schemaName}.${computeTableName(
+          field.object.nameSingular,
+          field?.object?.isCustom,
+        )}`,
+        newColumnName,
+      );
+
+      if (hasColumn) {
+        return;
+      }
+
+      if (!dryRun) {
+        await queryRunner.addColumn(
+          `${schemaName}.${computeTableName(
+            field.object.nameSingular,
+            field?.object?.isCustom,
+          )}`,
+          new TableColumn({
+            name: newColumnName,
+            type: 'jsonb',
+            default: `'{}'::"jsonb"`,
+            isNullable: true,
+          }),
+        );
+
+        await queryRunner.commitTransaction();
+      }
+    } catch (error) {
+      await queryRunner.rollbackTransaction();
+
+      throw error;
+    } finally {
+      await queryRunner.release();
+    }
+  }
+}
--- a/packages/twenty-server/src/database/commands/upgrade-version/0-41/0-41-upgrade-version.command.ts
+++ b/packages/twenty-server/src/database/commands/upgrade-version/0-41/0-41-upgrade-version.command.ts
@ -5,6 +5,7 @@ import { Repository } from 'typeorm';

 import { ActiveWorkspacesCommandRunner } from 'src/database/commands/active-workspaces.command';
 import { BaseCommandOptions } from 'src/database/commands/base.command';
+import { AddContextToActorCompositeTypeCommand } from 'src/database/commands/upgrade-version/0-41/0-41-add-context-to-actor-composite-type';
 import { MigrateRelationsToFieldMetadataCommand } from 'src/database/commands/upgrade-version/0-41/0-41-migrate-relations-to-field-metadata.command';
 import { RemoveDuplicateMcmasCommand } from 'src/database/commands/upgrade-version/0-41/0-41-remove-duplicate-mcmas';
 import { SeedWorkflowViewsCommand } from 'src/database/commands/upgrade-version/0-41/0-41-seed-workflow-views.command';
@ -22,6 +23,7 @@ export class UpgradeTo0_41Command extends ActiveWorkspacesCommandRunner {
    private readonly seedWorkflowViewsCommand: SeedWorkflowViewsCommand,
    private readonly syncWorkspaceMetadataCommand: SyncWorkspaceMetadataCommand,
    private readonly migrateRelationsToFieldMetadata: MigrateRelationsToFieldMetadataCommand,
+    private readonly addContextToActorCompositeType: AddContextToActorCompositeTypeCommand,
    private readonly removeDuplicateMcmasCommand: RemoveDuplicateMcmasCommand,
  ) {
    super(workspaceRepository);
@ -40,6 +42,12 @@ export class UpgradeTo0_41Command extends ActiveWorkspacesCommandRunner {
      workspaceIds,
    );

+    await this.addContextToActorCompositeType.executeActiveWorkspacesCommand(
+      passedParam,
+      options,
+      workspaceIds,
+    );
+
    await this.syncWorkspaceMetadataCommand.executeActiveWorkspacesCommand(
      passedParam,
      {
--- a/packages/twenty-server/src/database/commands/upgrade-version/0-41/0-41-upgrade-version.module.ts
+++ b/packages/twenty-server/src/database/commands/upgrade-version/0-41/0-41-upgrade-version.module.ts
@ -1,6 +1,7 @@
 import { Module } from '@nestjs/common';
 import { TypeOrmModule } from '@nestjs/typeorm';

+import { AddContextToActorCompositeTypeCommand } from 'src/database/commands/upgrade-version/0-41/0-41-add-context-to-actor-composite-type';
 import { MigrateRelationsToFieldMetadataCommand } from 'src/database/commands/upgrade-version/0-41/0-41-migrate-relations-to-field-metadata.command';
 import { RemoveDuplicateMcmasCommand } from 'src/database/commands/upgrade-version/0-41/0-41-remove-duplicate-mcmas';
 import { SeedWorkflowViewsCommand } from 'src/database/commands/upgrade-version/0-41/0-41-seed-workflow-views.command';
@ -10,6 +11,8 @@ import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity';
 import { DataSourceModule } from 'src/engine/metadata-modules/data-source/data-source.module';
 import { FieldMetadataEntity } from 'src/engine/metadata-modules/field-metadata/field-metadata.entity';
 import { ObjectMetadataModule } from 'src/engine/metadata-modules/object-metadata/object-metadata.module';
+import { WorkspaceMetadataVersionModule } from 'src/engine/metadata-modules/workspace-metadata-version/workspace-metadata-version.module';
+import { WorkspaceDataSourceModule } from 'src/engine/workspace-datasource/workspace-datasource.module';
 import { WorkspaceHealthModule } from 'src/engine/workspace-manager/workspace-health/workspace-health.module';
 import { SyncWorkspaceLoggerService } from 'src/engine/workspace-manager/workspace-sync-metadata/commands/services/sync-workspace-logger.service';
 import { SyncWorkspaceMetadataCommand } from 'src/engine/workspace-manager/workspace-sync-metadata/commands/sync-workspace-metadata.command';
@ -26,6 +29,8 @@ import { WorkspaceSyncMetadataModule } from 'src/engine/workspace-manager/worksp
    WorkspaceSyncMetadataCommandsModule,
    WorkspaceSyncMetadataModule,
    WorkspaceHealthModule,
+    WorkspaceDataSourceModule,
+    WorkspaceMetadataVersionModule,
  ],
  providers: [
    SyncWorkspaceLoggerService,
@ -33,6 +38,7 @@ import { WorkspaceSyncMetadataModule } from 'src/engine/workspace-manager/worksp
    SeedWorkflowViewsCommand,
    UpgradeTo0_41Command,
    MigrateRelationsToFieldMetadataCommand,
+    AddContextToActorCompositeTypeCommand,
    RemoveDuplicateMcmasCommand,
  ],
 })