diff --git a/packages/twenty-server/src/database/commands/0-20-update-message-channel-sync-status-enum.command.ts b/packages/twenty-server/src/database/commands/0-20-update-message-channel-sync-status-enum.command.ts new file mode 100644 index 000000000..63e2405bd --- /dev/null +++ b/packages/twenty-server/src/database/commands/0-20-update-message-channel-sync-status-enum.command.ts @@ -0,0 +1,240 @@ +import { InjectRepository } from '@nestjs/typeorm'; +import { Logger } from '@nestjs/common'; + +import { Command, CommandRunner, Option } from 'nest-commander'; +import { Repository } from 'typeorm'; +import chalk from 'chalk'; +import { v4 } from 'uuid'; + +import { DataSourceService } from 'src/engine/metadata-modules/data-source/data-source.service'; +import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity'; +import { TypeORMService } from 'src/database/typeorm/typeorm.service'; +import { FieldMetadataEntity } from 'src/engine/metadata-modules/field-metadata/field-metadata.entity'; +import { WorkspaceCacheVersionService } from 'src/engine/metadata-modules/workspace-cache-version/workspace-cache-version.service'; +import { ObjectMetadataEntity } from 'src/engine/metadata-modules/object-metadata/object-metadata.entity'; +import { MessageChannelSyncStatus } from 'src/modules/messaging/standard-objects/message-channel.workspace-entity'; + +interface UpdateMessageChannelSyncStatusEnumCommandOptions { + workspaceId?: string; +} + +@Command({ + name: 'migrate-0.20:update-message-channel-sync-status-enum', + description: 'Update messageChannel syncStatus', +}) +export class UpdateMessageChannelSyncStatusEnumCommand extends CommandRunner { + private readonly logger = new Logger( + UpdateMessageChannelSyncStatusEnumCommand.name, + ); + constructor( + @InjectRepository(Workspace, 'core') + private readonly workspaceRepository: Repository, + @InjectRepository(FieldMetadataEntity, 'metadata') + private readonly fieldMetadataRepository: Repository, + @InjectRepository(ObjectMetadataEntity, 'metadata') + private readonly objectMetadataRepository: Repository, + private readonly typeORMService: TypeORMService, + private readonly dataSourceService: DataSourceService, + private readonly workspaceCacheVersionService: WorkspaceCacheVersionService, + ) { + super(); + } + + @Option({ + flags: '-w, --workspace-id [workspace_id]', + description: 'workspace id. Command runs on all workspaces if not provided', + required: false, + }) + parseWorkspaceId(value: string): string { + return value; + } + + async run( + _passedParam: string[], + options: UpdateMessageChannelSyncStatusEnumCommandOptions, + ): Promise { + let workspaceIds: string[] = []; + + if (options.workspaceId) { + workspaceIds = [options.workspaceId]; + } else { + workspaceIds = (await this.workspaceRepository.find()).map( + (workspace) => workspace.id, + ); + } + + if (!workspaceIds.length) { + this.logger.log(chalk.yellow('No workspace found')); + + return; + } else { + this.logger.log( + chalk.green(`Running command on ${workspaceIds.length} workspaces`), + ); + } + + for (const workspaceId of workspaceIds) { + const dataSourceMetadatas = + await this.dataSourceService.getDataSourcesMetadataFromWorkspaceId( + workspaceId, + ); + + for (const dataSourceMetadata of dataSourceMetadatas) { + const workspaceDataSource = + await this.typeORMService.connectToDataSource(dataSourceMetadata); + + if (workspaceDataSource) { + const queryRunner = workspaceDataSource.createQueryRunner(); + + await queryRunner.connect(); + await queryRunner.startTransaction(); + + try { + await queryRunner.query( + `ALTER TYPE "${dataSourceMetadata.schema}"."messageChannel_syncStatus_enum" RENAME TO "messageChannel_syncStatus_enum_old"`, + ); + await queryRunner.query( + `CREATE TYPE "${dataSourceMetadata.schema}"."messageChannel_syncStatus_enum" AS ENUM ('PENDING', + 'SUCCEEDED', + 'FAILED', + 'ONGOING', + 'NOT_SYNCED', + 'COMPLETED', + 'FAILED_INSUFFICIENT_PERMISSIONS', + 'FAILED_UNKNOWN')`, + ); + + await queryRunner.query( + `ALTER TABLE "${dataSourceMetadata.schema}"."messageChannel" ALTER COLUMN "syncStatus" DROP DEFAULT`, + ); + await queryRunner.query( + `ALTER TABLE "${dataSourceMetadata.schema}"."messageChannel" ALTER COLUMN "syncStatus" TYPE text`, + ); + + await queryRunner.query( + `ALTER TABLE "${dataSourceMetadata.schema}"."messageChannel" ALTER COLUMN "syncStatus" TYPE "${dataSourceMetadata.schema}"."messageChannel_syncStatus_enum" USING "syncStatus"::text::"${dataSourceMetadata.schema}"."messageChannel_syncStatus_enum"`, + ); + + await queryRunner.query( + `DROP TYPE "${dataSourceMetadata.schema}"."messageChannel_syncStatus_enum_old"`, + ); + await queryRunner.commitTransaction(); + } catch (error) { + await queryRunner.rollbackTransaction(); + this.logger.log( + chalk.red(`Running command on workspace ${workspaceId} failed`), + ); + throw error; + } finally { + await queryRunner.release(); + } + } + } + + const messageChannelObjectMetadata = + await this.objectMetadataRepository.findOne({ + where: { nameSingular: 'messageChannel', workspaceId }, + }); + + if (!messageChannelObjectMetadata) { + this.logger.log( + chalk.yellow( + `Object metadata for messageChannel not found in workspace ${workspaceId}`, + ), + ); + + continue; + } + + const syncStatusFieldMetadata = + await this.fieldMetadataRepository.findOne({ + where: { + name: 'syncStatus', + workspaceId, + objectMetadataId: messageChannelObjectMetadata.id, + }, + }); + + if (!syncStatusFieldMetadata) { + this.logger.log( + chalk.yellow( + `Field metadata for syncStatus not found in workspace ${workspaceId}`, + ), + ); + + continue; + } + + const newOptions = [ + { + id: v4(), + value: MessageChannelSyncStatus.PENDING, + label: 'Pending', + position: 0, + color: 'blue', + }, + { + id: v4(), + value: MessageChannelSyncStatus.SUCCEEDED, + label: 'Succeeded', + position: 2, + color: 'green', + }, + { + id: v4(), + value: MessageChannelSyncStatus.FAILED, + label: 'Failed', + position: 3, + color: 'red', + }, + { + id: v4(), + value: MessageChannelSyncStatus.ONGOING, + label: 'Ongoing', + position: 1, + color: 'yellow', + }, + { + id: v4(), + value: MessageChannelSyncStatus.NOT_SYNCED, + label: 'Not Synced', + position: 4, + color: 'blue', + }, + { + id: v4(), + value: MessageChannelSyncStatus.COMPLETED, + label: 'Completed', + position: 5, + color: 'green', + }, + { + id: v4(), + value: MessageChannelSyncStatus.FAILED_INSUFFICIENT_PERMISSIONS, + label: 'Failed Insufficient Permissions', + position: 6, + color: 'red', + }, + { + id: v4(), + value: MessageChannelSyncStatus.FAILED_UNKNOWN, + label: 'Failed Unknown', + position: 7, + color: 'red', + }, + ]; + + await this.fieldMetadataRepository.update(syncStatusFieldMetadata.id, { + options: newOptions, + }); + + await this.workspaceCacheVersionService.incrementVersion(workspaceId); + + this.logger.log( + chalk.green(`Running command on workspace ${workspaceId} done`), + ); + } + + this.logger.log(chalk.green(`Command completed!`)); + } +} diff --git a/packages/twenty-server/src/database/commands/database-command.module.ts b/packages/twenty-server/src/database/commands/database-command.module.ts index 1e85f0309..40d5b1a1b 100644 --- a/packages/twenty-server/src/database/commands/database-command.module.ts +++ b/packages/twenty-server/src/database/commands/database-command.module.ts @@ -1,4 +1,5 @@ import { Module } from '@nestjs/common'; +import { TypeOrmModule } from '@nestjs/typeorm'; import { ConfirmationQuestion } from 'src/database/commands/questions/confirmation.question'; import { WorkspaceManagerModule } from 'src/engine/workspace-manager/workspace-manager.module'; @@ -14,17 +15,28 @@ import { StopDataSeedDemoWorkspaceCronCommand } from 'src/database/commands/data import { WorkspaceAddTotalCountCommand } from 'src/database/commands/workspace-add-total-count.command'; import { DataSeedDemoWorkspaceCommand } from 'src/database/commands/data-seed-demo-workspace/data-seed-demo-workspace-command'; import { DataSeedDemoWorkspaceModule } from 'src/database/commands/data-seed-demo-workspace/data-seed-demo-workspace.module'; +import { UpdateMessageChannelSyncStatusEnumCommand } from 'src/database/commands/0-20-update-message-channel-sync-status-enum.command'; +import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity'; +import { FieldMetadataEntity } from 'src/engine/metadata-modules/field-metadata/field-metadata.entity'; +import { WorkspaceCacheVersionModule } from 'src/engine/metadata-modules/workspace-cache-version/workspace-cache-version.module'; +import { ObjectMetadataEntity } from 'src/engine/metadata-modules/object-metadata/object-metadata.entity'; @Module({ imports: [ WorkspaceManagerModule, DataSourceModule, TypeORMModule, + TypeOrmModule.forFeature([Workspace], 'core'), + TypeOrmModule.forFeature( + [FieldMetadataEntity, ObjectMetadataEntity], + 'metadata', + ), WorkspaceModule, WorkspaceDataSourceModule, WorkspaceSyncMetadataModule, ObjectMetadataModule, DataSeedDemoWorkspaceModule, + WorkspaceCacheVersionModule, ], providers: [ DataSeedWorkspaceCommand, @@ -33,6 +45,7 @@ import { DataSeedDemoWorkspaceModule } from 'src/database/commands/data-seed-dem ConfirmationQuestion, StartDataSeedDemoWorkspaceCronCommand, StopDataSeedDemoWorkspaceCronCommand, + UpdateMessageChannelSyncStatusEnumCommand, ], }) export class DatabaseCommandModule {}