ToolJet/server/data-migrations/1686829426671-BackFillCurrentEnvironmentId.ts

import { App } from 'src/entities/app.entity';
import { AppVersion } from 'src/entities/app_version.entity';
import { Organization } from 'src/entities/organization.entity';
import { MigrationProgress } from 'src/helpers/utils.helper';
import { EntityManager, MigrationInterface, QueryRunner } from 'typeorm';

export class BackFillCurrentEnvironmentId1686829426671 implements MigrationInterface {
  public async up(queryRunner: QueryRunner): Promise<void> {
    //back fill current_environment_id to production env id
    await this.backFillNewColumn(queryRunner.manager);
  }

  async backFillNewColumn(manager: EntityManager) {
    const organizations = await manager.find(Organization, {
      select: ['id', 'appEnvironments'],
      relations: ['appEnvironments'],
    });

    const migrationProgress = new MigrationProgress('BackFillCurrentEnvironmentId1686829426671', organizations.length);

    for (const organization of organizations) {
      const productionEnvironment = organization.appEnvironments.find((appEnvironment) => appEnvironment.isDefault);
      const developmentEnvironment = organization.appEnvironments.find(
        (appEnvironment) => appEnvironment.priority === 1
      );
      const apps = await manager.find(App, {
        select: ['id', 'appVersions', 'currentVersionId'],
        where: {
          organizationId: organization.id,
        },
        relations: ['appVersions'],
      });

      for (const { appVersions, currentVersionId } of apps) {
        for (const appVersion of appVersions) {
          console.log('Updating app version =>', appVersion.id);
          let envToUpdate: string;

          /* For CE always the this condition will only work */
          if ((currentVersionId && currentVersionId === appVersion.id) || organization.appEnvironments.length === 1) {
            envToUpdate = productionEnvironment.id;
          } else {
            envToUpdate = developmentEnvironment.id;
          }

          await manager.update(
            AppVersion,
            { id: appVersion.id },
            {
              currentEnvironmentId: envToUpdate,
            }
          );
        }
      }
      migrationProgress.show();
    }
  }

  public async down(queryRunner: QueryRunner): Promise<void> {}
}