Add Project.partnership(partner: id) (#3271)

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

Author: atGit2021

src/components/partnership/index.ts

@@ -1,2 +1,3 @@
 export * from './partnership.service';
 export * from './partnership.loader';
+export * from './partnership-by-project-and-partner.loader';

src/components/partnership/partnership-by-project-and-partner.loader.ts

@@ -0,0 +1,34 @@
+import { ID } from '~/common';
+import {
+  LoaderFactory,
+  LoaderOptionsOf,
+  SessionAwareLoaderStrategy,
+} from '~/core';
+import { Partnership } from './dto';
+import { PartnershipService } from './partnership.service';
+
+export interface PartnershipByProjectAndPartnerInput {
+  project: ID<'Project'>;
+  partner: ID<'Partner'>;
+}
+
+@LoaderFactory()
+export class PartnershipByProjectAndPartnerLoader extends SessionAwareLoaderStrategy<
+  { id: PartnershipByProjectAndPartnerInput; partnership: Partnership },
+  PartnershipByProjectAndPartnerInput,
+  string
+> {
+  constructor(private readonly service: PartnershipService) {
+    super();
+  }
+
+  getOptions() {
+    return {
+      cacheKeyFn: (input) => `${input.project}:${input.partner}`,
+    } satisfies LoaderOptionsOf<PartnershipByProjectAndPartnerLoader>;
+  }
+
+  async loadMany(input: readonly PartnershipByProjectAndPartnerInput[]) {
+    return await this.service.readManyByProjectAndPartner(input, this.session);
+  }
+}

src/components/partnership/partnership.edgedb.repository.ts

@@ -2,6 +2,7 @@ import { Injectable } from '@nestjs/common';
 import { ID, PublicOf } from '~/common';
 import { e, RepoFor } from '~/core/edgedb';
 import { Partnership } from './dto';
+import type { PartnershipByProjectAndPartnerInput } from './partnership-by-project-and-partner.loader';
 import { PartnershipRepository } from './partnership.repository';
 
 @Injectable()
@@ -28,6 +29,26 @@ export class PartnershipEdgeDBRepository
   })
   implements PublicOf<PartnershipRepository>
 {
+  async readManyByProjectAndPartner(
+    input: readonly PartnershipByProjectAndPartnerInput[],
+  ) {
+    return await this.db.run(this.readManyByProjectAndPartnerQuery, { input });
+  }
+  private readonly readManyByProjectAndPartnerQuery = e.params(
+    { input: e.array(e.tuple({ project: e.uuid, partner: e.uuid })) },
+    ({ input }) =>
+      e.for(e.array_unpack(input), ({ project, partner }) =>
+        e.select(e.Partnership, (partnership) => ({
+          ...this.hydrate(partnership),
+          filter: e.op(
+            e.op(partnership.project, '=', e.cast(e.Project, project)),
+            'and',
+            e.op(partnership.partner, '=', e.cast(e.Partner, partner)),
+          ),
+        })),
+      ),
+  );
+
   async isFirstPartnership(projectId: ID) {
     const project = e.cast(e.Project, e.uuid(projectId));
     const query = e.op('not', e.op('exists', project.partnerships));

src/components/partnership/partnership.module.ts

@@ -6,6 +6,7 @@ import { FileModule } from '../file/file.module';
 import { PartnerModule } from '../partner/partner.module';
 import { ProjectModule } from '../project/project.module';
 import * as handlers from './handlers';
+import { PartnershipByProjectAndPartnerLoader } from './partnership-by-project-and-partner.loader';
 import { PartnershipEdgeDBRepository } from './partnership.edgedb.repository';
 import { PartnershipLoader } from './partnership.loader';
 import { PartnershipRepository } from './partnership.repository';
@@ -25,6 +26,7 @@ import { PartnershipService } from './partnership.service';
     PartnershipService,
     splitDb(PartnershipRepository, PartnershipEdgeDBRepository),
     PartnershipLoader,
+    PartnershipByProjectAndPartnerLoader,
     ...Object.values(handlers),
   ],
   exports: [PartnershipService],

src/components/partnership/partnership.repository.ts

@@ -31,6 +31,7 @@ import {
   paginate,
   requestingUser,
   sortWith,
+  variable,
   whereNotDeletedInChangeset,
 } from '~/core/database/query';
 import { FileService } from '../file';
@@ -44,6 +45,7 @@ import {
   PartnershipListInput,
   UpdatePartnership,
 } from './dto';
+import type { PartnershipByProjectAndPartnerInput } from './partnership-by-project-and-partner.loader';
 
 @Injectable()
 export class PartnershipRepository extends DtoRepository<
@@ -162,6 +164,25 @@ export class PartnershipRepository extends DtoRepository<
       .run();
   }
 
+  async readManyByProjectAndPartner(
+    input: readonly PartnershipByProjectAndPartnerInput[],
+    session: Session,
+  ) {
+    return await this.db
+      .query()
+      .unwind([...input], 'input')
+      .match([
+        node('project', 'Project', { id: variable('input.project.id') }),
+        relation('out', '', 'partnership', ACTIVE),
+        node('node'),
+        relation('out', '', 'partner', ACTIVE),
+        node('partner', 'Partner', { id: variable('input.partner.id') }),
+      ])
+      .apply(this.hydrate(session))
+      .map('dto')
+      .run();
+  }
+
   protected override hydrate(session: Session, view?: ObjectView) {
     return (query: Query) =>
       query

src/components/partnership/partnership.service.ts

@@ -33,6 +33,7 @@ import {
   PartnershipUpdatedEvent,
   PartnershipWillDeleteEvent,
 } from './events';
+import type { PartnershipByProjectAndPartnerInput } from './partnership-by-project-and-partner.loader';
 import { PartnershipRepository } from './partnership.repository';
 
 @Injectable()
@@ -122,6 +123,20 @@ export class PartnershipService {
     return partnerships.map((dto) => this.secure(dto, session));
   }
 
+  async readManyByProjectAndPartner(
+    input: readonly PartnershipByProjectAndPartnerInput[],
+    session: Session,
+  ) {
+    const partnerships = await this.repo.readManyByProjectAndPartner(
+      input,
+      session,
+    );
+    return partnerships.map((dto) => ({
+      id: { project: dto.project.id, partner: dto.partner.id },
+      partnership: this.secure(dto, session),
+    }));
+  }
+
   secure(dto: UnsecuredDto<Partnership>, session: Session) {
     return this.privileges.for(session, Partnership).secure(dto);
   }

src/components/project/project.resolver.ts

@@ -40,8 +40,12 @@ import {
 } from '../location/dto';
 import { OrganizationLoader } from '../organization';
 import { SecuredOrganization } from '../organization/dto';
-import { PartnershipLoader } from '../partnership';
 import {
+  PartnershipByProjectAndPartnerLoader,
+  PartnershipLoader,
+} from '../partnership';
+import {
+  Partnership,
   PartnershipListInput,
   SecuredPartnership,
   SecuredPartnershipList,
@@ -254,6 +258,20 @@ export class ProjectResolver {
     return list;
   }
 
+  @ResolveField(() => Partnership)
+  async partnership(
+    @Parent() project: Project,
+    @IdArg({ name: 'partner' }) partnerId: ID,
+    @Loader(PartnershipByProjectAndPartnerLoader)
+    loader: LoaderOf<PartnershipByProjectAndPartnerLoader>,
+  ): Promise<Partnership> {
+    const result = await loader.load({
+      project: project.id,
+      partner: partnerId,
+    });
+    return result.partnership;
+  }
+
   @ResolveField(() => SecuredDirectory, {
     description: 'The root filesystem directory of this project',
   })