ToolJet/server/data-migrations/1684157120658-AddUniqueConstraintToAppName.ts

import { App } from 'src/entities/app.entity';
import { Organization } from 'src/entities/organization.entity';
import { DataBaseConstraints } from 'src/helpers/db_constraints.constants';
import { MigrationInterface, QueryRunner, TableUnique, EntityManager } from 'typeorm';

export class AddUniqueConstraintToAppName1684145489093 implements MigrationInterface {
  public async up(queryRunner: QueryRunner): Promise<void> {
    const entityManager = queryRunner.manager;
    await this.migrateAppNames(entityManager);
    await queryRunner.createUniqueConstraint(
      'apps',
      new TableUnique({
        name: DataBaseConstraints.APP_NAME_UNIQUE,
        columnNames: ['name', 'organization_id'],
      })
    );
  }

  public async migrateAppNames(entityManager: EntityManager) {
    const workspaces = await entityManager.find(Organization);
    for (const workspace of workspaces) {
      const { id: organizationId } = workspace;
      const apps = await entityManager.query(
        'select sub_query.name from (select count(*) as name_count, name from apps where organization_id=$1 group by name) sub_query where name_count > 1    ',
        [organizationId]
      );
      for (const app of apps) {
        const { name } = app;
        const sameApps = await entityManager.find(App, { where: { name, organizationId } });
        for (const appToChange of sameApps.slice(1)) {
          await entityManager.update(App, { id: appToChange.id }, { name: `${appToChange.name} ${Date.now()}` });
        }
      }
    }
  }

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