twenty/packages/twenty-server/src/database/commands/upgrade-version-command/1-2/1-2-add-enqueued-status-to-workflow-run-v2.command.ts

import { InjectDataSource, InjectRepository } from '@nestjs/typeorm';

import { Command } from 'nest-commander';
import { DataSource, Repository } from 'typeorm';

import {
  ActiveOrSuspendedWorkspacesMigrationCommandRunner,
  type RunOnWorkspaceArgs,
} from 'src/database/commands/command-runners/active-or-suspended-workspaces-migration.command-runner';
import { WorkspaceEntity } from 'src/engine/core-modules/workspace/workspace.entity';
import { FieldMetadataEntity } from 'src/engine/metadata-modules/field-metadata/field-metadata.entity';
import { ObjectMetadataEntity } from 'src/engine/metadata-modules/object-metadata/object-metadata.entity';
import { TwentyORMGlobalManager } from 'src/engine/twenty-orm/twenty-orm-global.manager';
import { getWorkspaceSchemaName } from 'src/engine/workspace-datasource/utils/get-workspace-schema-name.util';
import { WORKFLOW_RUN_STANDARD_FIELD_IDS } from 'src/engine/workspace-manager/workspace-sync-metadata/constants/standard-field-ids';
import { STANDARD_OBJECT_IDS } from 'src/engine/workspace-manager/workspace-sync-metadata/constants/standard-object-ids';
import { WorkflowRunStatus } from 'src/modules/workflow/common/standard-objects/workflow-run.workspace-entity';

@Command({
  name: 'upgrade:1-2:add-enqueued-status-to-workflow-run-v2',
  description: 'Add enqueued status to workflow run',
})
export class AddEnqueuedStatusToWorkflowRunV2Command extends ActiveOrSuspendedWorkspacesMigrationCommandRunner {
  constructor(
    @InjectRepository(WorkspaceEntity)
    protected readonly workspaceRepository: Repository<WorkspaceEntity>,
    protected readonly twentyORMGlobalManager: TwentyORMGlobalManager,
    @InjectRepository(ObjectMetadataEntity)
    private readonly objectMetadataRepository: Repository<ObjectMetadataEntity>,
    @InjectRepository(FieldMetadataEntity)
    private readonly fieldMetadataRepository: Repository<FieldMetadataEntity>,
    @InjectDataSource()
    private readonly coreDataSource: DataSource,
  ) {
    super(workspaceRepository, twentyORMGlobalManager);
  }

  override async runOnWorkspace({
    workspaceId,
    options,
  }: RunOnWorkspaceArgs): Promise<void> {
    this.logger.log(
      `Adding enqueued status to workflow run for workspace ${workspaceId}`,
    );

    const workflowRunObjectMetadata =
      await this.objectMetadataRepository.findOne({
        where: {
          workspaceId,
          standardId: STANDARD_OBJECT_IDS.workflowRun,
        },
      });

    if (!workflowRunObjectMetadata) {
      this.logger.error(
        `Workflow run object metadata not found for workspace ${workspaceId}`,
      );

      return;
    }

    const workflowRunStatusFieldMetadata =
      await this.fieldMetadataRepository.findOne({
        where: {
          standardId: WORKFLOW_RUN_STANDARD_FIELD_IDS.status,
          objectMetadataId: workflowRunObjectMetadata.id,
        },
      });

    if (!workflowRunStatusFieldMetadata) {
      this.logger.error(
        `Workflow run status field metadata not found for workspace ${workspaceId}`,
      );

      return;
    }

    const workflowRunStatusFieldMetadataOptions =
      workflowRunStatusFieldMetadata.options;

    // check if enqueued status is already in the field metadata options
    if (
      workflowRunStatusFieldMetadataOptions?.some(
        (option) => option.value === WorkflowRunStatus.ENQUEUED,
      )
    ) {
      this.logger.log(
        `Workflow run status field metadata options already contain enqueued status for workspace ${workspaceId}`,
      );

      return;
    } else if (options.dryRun) {
      this.logger.log(
        `Would add enqueued status to workflow run status field metadata for workspace ${workspaceId}`,
      );
    } else {
      workflowRunStatusFieldMetadataOptions?.push({
        value: WorkflowRunStatus.ENQUEUED,
        label: 'Enqueued',
        position: 4,
        color: 'blue',
      });

      await this.fieldMetadataRepository.save(workflowRunStatusFieldMetadata);

      this.logger.log(
        `Enqueued status added to workflow run status field metadata for workspace ${workspaceId}`,
      );
    }

    const schemaName = getWorkspaceSchemaName(workspaceId);

    if (options.dryRun) {
      this.logger.log(
        `Would try to add enqueued status to workflow run status enum for workspace ${workspaceId}`,
      );
    } else {
      try {
        await this.coreDataSource.query(
          `ALTER TYPE ${schemaName}."workflowRun_status_enum" ADD VALUE 'ENQUEUED'`,
        );
        this.logger.log(
          `Enqueued status added to workflow run status enum for workspace ${workspaceId}`,
        );
      } catch (error) {
        this.logger.error(
          `Error adding enqueued status to workflow run status enum for workspace ${workspaceId}: ${error}`,
        );
      }
    }
  }
}