[EdgeDB] Engagement repo w/ polymorphic hydration

Author: CarsonF

src/components/engagement/dto/engagement.dto.ts

@@ -222,6 +222,11 @@ export class InternshipEngagement extends Engagement {
 export const engagementRange = (engagement: UnsecuredDto<Engagement>) =>
   DateInterval.tryFrom(engagement.startDate, engagement.endDate);
 
+export const EngagementConcretes = {
+  LanguageEngagement,
+  InternshipEngagement,
+};
+
 declare module '~/core/resources/map' {
   interface ResourceMap {
     Engagement: typeof Engagement;

src/components/engagement/engagement.edgedb.repository.ts

@@ -0,0 +1,130 @@
+import { Injectable, Type } from '@nestjs/common';
+import { ModuleRef } from '@nestjs/core';
+import { LazyGetter } from 'lazy-get-decorator';
+import { PublicOf } from '~/common';
+import { grabInstances } from '~/common/instance-maps';
+import { e, RepoFor } from '~/core/edgedb';
+import {
+  EngagementConcretes as ConcreteTypes,
+  CreateInternshipEngagement,
+  CreateLanguageEngagement,
+  IEngagement,
+  UpdateInternshipEngagement,
+  UpdateLanguageEngagement,
+} from './dto';
+import { EngagementRepository } from './engagement.repository';
+
+const baseHydrate = e.shape(e.Engagement, (engagement) => ({
+  ...engagement['*'],
+  __typename: engagement.__type__.name,
+  project: {
+    id: true,
+    status: true,
+    type: true,
+  },
+  parent: e.tuple({
+    identity: engagement.project.id,
+    labels: e.array_agg(e.set(engagement.project.__type__.name.slice(9, null))),
+    properties: e.tuple({
+      id: engagement.project.id,
+      createdAt: engagement.project.createdAt,
+    }),
+  }),
+  ceremony: true,
+  completeDate: engagement.completedDate, // TODO fix in schema
+}));
+
+const languageExtraHydrate = {
+  language: true,
+  firstScripture: true,
+  lukePartnership: true,
+  openToInvestorVisit: true,
+  sentPrintingDate: true,
+  paratextRegistryId: true,
+  pnp: true,
+  historicGoal: true,
+} as const;
+
+const internshipExtraHydrate = {
+  countryOfOrigin: true,
+  intern: true,
+  mentor: true,
+  position: true,
+  methodologies: true,
+  growthPlan: true,
+} as const;
+
+const languageHydrate = e.shape(e.LanguageEngagement, (le) => ({
+  ...baseHydrate(le),
+  __typename: le.__type__.name,
+  ...languageExtraHydrate,
+}));
+
+const internshipHydrate = e.shape(e.InternshipEngagement, (ie) => ({
+  ...baseHydrate(ie),
+  __typename: ie.__type__.name,
+  ...internshipExtraHydrate,
+}));
+
+const hydrate = e.shape(e.Engagement, (engagement) => ({
+  ...baseHydrate(engagement),
+  ...e.is(e.LanguageEngagement, languageExtraHydrate),
+  ...e.is(e.InternshipEngagement, internshipExtraHydrate),
+}));
+
+export const ConcreteRepos = {
+  LanguageEngagement: class LanguageEngagementRepository extends RepoFor(
+    ConcreteTypes.LanguageEngagement,
+    {
+      hydrate: languageHydrate,
+    },
+  ) {},
+
+  InternshipEngagement: class InternshipEngagementRepository extends RepoFor(
+    ConcreteTypes.InternshipEngagement,
+    {
+      hydrate: internshipHydrate,
+    },
+  ) {},
+} satisfies Record<keyof typeof ConcreteTypes, Type>;
+
+@Injectable()
+export class EngagementEdgeDBRepository
+  extends RepoFor(IEngagement, {
+    hydrate,
+    omit: ['create', 'update'],
+  })
+  implements PublicOf<EngagementRepository>
+{
+  constructor(private readonly moduleRef: ModuleRef) {
+    super();
+  }
+
+  @LazyGetter() protected get concretes() {
+    return grabInstances(this.moduleRef, ConcreteRepos);
+  }
+
+  async createLanguageEngagement(input: CreateLanguageEngagement) {
+    return await this.concretes.LanguageEngagement.create(input);
+  }
+
+  async createInternshipEngagement(input: CreateInternshipEngagement) {
+    return await this.concretes.InternshipEngagement.create(input);
+  }
+
+  get getActualLanguageChanges() {
+    return this.concretes.LanguageEngagement.getActualChanges;
+  }
+
+  get getActualInternshipChanges() {
+    return this.concretes.InternshipEngagement.getActualChanges;
+  }
+
+  async updateLanguage(input: UpdateLanguageEngagement) {
+    return await this.concretes.LanguageEngagement.update(input);
+  }
+
+  async updateInternship(input: UpdateInternshipEngagement) {
+    return await this.concretes.InternshipEngagement.update(input);
+  }
+}

src/components/engagement/engagement.module.ts

@@ -1,4 +1,5 @@
 import { forwardRef, Module } from '@nestjs/common';
+import { splitDb } from '~/core';
 import { AuthorizationModule } from '../authorization/authorization.module';
 import { CeremonyModule } from '../ceremony/ceremony.module';
 import { FileModule } from '../file/file.module';
@@ -7,6 +8,7 @@ import { LocationModule } from '../location/location.module';
 import { ProductModule } from '../product/product.module';
 import { ProjectModule } from '../project/project.module';
 import { EngagementStatusResolver } from './engagement-status.resolver';
+import { EngagementEdgeDBRepository } from './engagement.edgedb.repository';
 import { EngagementLoader } from './engagement.loader';
 import { EngagementRepository } from './engagement.repository';
 import { EngagementResolver } from './engagement.resolver';
@@ -38,7 +40,7 @@ import { EngagementProductConnectionResolver } from './product-connection.resolv
     EngagementProductConnectionResolver,
     EngagementRules,
     EngagementService,
-    EngagementRepository,
+    splitDb(EngagementRepository, EngagementEdgeDBRepository),
     EngagementLoader,
     ...Object.values(handlers),
     FixNullMethodologiesMigration,