0701 default marketing region migration (#3044)

Co-authored-by: Carson Full <[email protected]>

Author: willdch

src/components/location/location.module.ts

@@ -7,6 +7,7 @@ import { LocationLoader } from './location.loader';
 import { LocationRepository } from './location.repository';
 import { LocationResolver } from './location.resolver';
 import { LocationService } from './location.service';
+import { DefaultMarketingRegionMigration } from './migrations/default-marketing-region.migration';
 
 @Module({
   imports: [
@@ -20,6 +21,7 @@ import { LocationService } from './location.service';
     LocationService,
     LocationRepository,
     LocationLoader,
+    DefaultMarketingRegionMigration,
   ],
   exports: [LocationService],
 })

src/components/location/migrations/default-marketing-region.migration.ts

@@ -0,0 +1,91 @@
+import { mapEntries } from '@seedcompany/common';
+import { node, relation } from 'cypher-query-builder';
+import { ValueOf } from 'type-fest';
+import { ID, IdOf } from '~/common';
+import { BaseMigration, Migration } from '~/core/database';
+import { ACTIVE } from '~/core/database/query';
+import { Location } from '../dto';
+import { LocationService } from '../location.service';
+
+@Migration('2024-01-16T11:00:00')
+export class DefaultMarketingRegionMigration extends BaseMigration {
+  constructor(private readonly locationService: LocationService) {
+    super();
+  }
+  async up() {
+    const session = this.fakeAdminSession;
+
+    const fieldRegionNameToMarketingRegionName = {
+      'Africa - Anglophone East': 'Africa',
+      'Africa - Anglophone West': 'Africa',
+      'Africa - Congo Basin': 'Africa',
+      'Africa - Sahel': 'Africa',
+      'Africa - Southern': 'Africa',
+      Americas: 'Americas',
+      Eurasia: 'Europe and the Middle East',
+      Pacific: 'Pacific',
+      'Asia - Islands': 'Pacific',
+      'Asia - Mainland': 'Asia',
+      'Asia - South': 'Asia',
+    } as const;
+
+    const marketingRegionList = await this.db
+      .query()
+      .match([
+        node('location', 'Location'),
+        relation('out', '', 'type', ACTIVE),
+        node('', 'LocationType', { value: 'Region' }),
+      ])
+      .match([
+        node('location'),
+        relation('out', '', 'name', ACTIVE),
+        node('locName', 'LocationName'),
+      ])
+      .return<{
+        id: IdOf<Location>;
+        name: ValueOf<typeof fieldRegionNameToMarketingRegionName>;
+      }>(['location.id as id', 'locName.value as name'])
+      .run();
+    const marketingRegionNameToId = mapEntries(marketingRegionList, (loc) => [
+      loc.name,
+      loc.id,
+    ]).asRecord;
+
+    const countries = await this.db
+      .query()
+      .match([
+        node('location', 'Location'),
+        relation('out', '', 'type', ACTIVE),
+        node('', 'LocationType', { value: 'Country' }),
+      ])
+      .match([
+        node('location', 'Location'),
+        relation('out', '', 'defaultFieldRegion', ACTIVE),
+        node('fieldRegion', 'FieldRegion'),
+        relation('out', '', 'name', ACTIVE),
+        node('fieldRegionName', 'FieldRegionName'),
+      ])
+      .return<{
+        id: ID;
+        fieldRegionName: keyof typeof fieldRegionNameToMarketingRegionName;
+      }>(['location.id as id', 'fieldRegionName.value as fieldRegionName'])
+      .run();
+
+    for (const country of countries) {
+      const marketingRegionName =
+        fieldRegionNameToMarketingRegionName[country.fieldRegionName];
+      const marketingRegionId = marketingRegionNameToId[marketingRegionName];
+      if (marketingRegionId === undefined) {
+        continue;
+      }
+
+      await this.locationService.update(
+        {
+          id: country.id,
+          defaultMarketingRegionId: marketingRegionId,
+        },
+        session,
+      );
+    }
+  }
+}