ToolJet/server/data-migrations/1640683693031-BackfillCalendarWeekDateFormat.ts

import { AppVersion } from '@entities/app_version.entity';
import { MigrationInterface, QueryRunner } from 'typeorm';

export class BackfillCalendarWeekDateFormat1640683693031 implements MigrationInterface {
  public async up(queryRunner: QueryRunner): Promise<void> {
    const entityManager = queryRunner.manager;
    const queryBuilder = queryRunner.connection.createQueryBuilder();
    const appVersionRepository = entityManager.getRepository(AppVersion);

    const appVersions = await appVersionRepository.find();

    for (const version of appVersions) {
      const definition = version['definition'];
      let definitionUpdated = false;

      if (definition) {
        const components = definition['components'];

        for (const componentId of Object.keys(components)) {
          const componentDefinition = components[componentId];
          if (componentDefinition.component.component !== 'Calendar') continue;

          componentDefinition.component.definition.styles.weekDateFormat =
            this.determineDateFormatForBackfill(componentDefinition);

          components[componentId] = componentDefinition;
          definitionUpdated = true;
        }

        if (!definitionUpdated) continue;

        definition['components'] = components;
        version.definition = definition;

        await queryBuilder.update(AppVersion).set({ definition }).where('id = :id', { id: version.id }).execute();
      }
    }
  }

  determineDateFormatForBackfill(componentDefinition) {
    const dateFormat = componentDefinition.component.definition.styles.displayDayNamesInWeekView?.value
      ? 'ddd'
      : 'DD MMM';

    return { value: dateFormat };
  }

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