Expose Project.primaryPartnership

Author: CarsonF

src/components/partnership/dto/partnership.dto.ts

@@ -8,6 +8,7 @@ import {
   Secured,
   SecuredBoolean,
   SecuredDateNullable,
+  SecuredProperty,
   SecuredProps,
   Sensitivity,
   SensitivityField,
@@ -92,3 +93,8 @@ declare module '~/core/resources/map' {
     Partnership: typeof e.default.Partnership;
   }
 }
+
+@ObjectType({
+  description: SecuredProperty.descriptionFor('a partnership'),
+})
+export class SecuredPartnership extends SecuredProperty(Partnership) {}

src/components/project/dto/project.dto.ts

@@ -124,6 +124,8 @@ class Project extends Interfaces {
   @Calculated()
   readonly status: ProjectStatus;
 
+  readonly primaryPartnership: Secured<LinkTo<'Partnership'> | null>;
+
   readonly primaryLocation: Secured<LinkTo<'Location'> | null>;
 
   readonly marketingLocation: Secured<LinkTo<'Location'> | null>;

src/components/project/project.edgedb.repository.ts

@@ -27,6 +27,11 @@ const hydrate = e.shape(e.Project, (project) => ({
   __typename: project.__type__.name.slice(9, null),
 
   rootDirectory: true,
+  primaryPartnership: e
+    .select(project.partnerships, (p) => ({
+      filter: e.op(p.primary, '=', true),
+    }))
+    .assert_single(),
   primaryLocation: true,
   marketingLocation: true,
   marketingRegionOverride: true,

src/components/project/project.repository.ts

@@ -27,6 +27,7 @@ import {
   requestingUser,
   SortCol,
   sortWith,
+  variable,
 } from '~/core/database/query';
 import { Privileges } from '../authorization';
 import { locationSorters } from '../location/location.repository';
@@ -86,6 +87,13 @@ export class ProjectRepository extends CommonRepository {
           relation('out', '', 'rootDirectory', ACTIVE),
           node('rootDirectory', 'Directory'),
         ])
+        .optionalMatch([
+          node('node'),
+          relation('out', '', 'partnership', ACTIVE),
+          node('primaryPartnership', 'Partnership'),
+          relation('out', '', 'primary', ACTIVE),
+          node('', 'Property', { value: variable('true') }),
+        ])
         .optionalMatch([
           node('node'),
           relation('out', '', 'primaryLocation', ACTIVE),
@@ -125,6 +133,7 @@ export class ProjectRepository extends CommonRepository {
             type: 'node.type',
             pinned: 'exists((:User { id: $requestingUser })-[:pinned]->(node))',
             rootDirectory: 'rootDirectory { .id }',
+            primaryPartnership: 'primaryPartnership { .id }',
             primaryLocation: 'primaryLocation { .id }',
             marketingLocation: 'marketingLocation { .id }',
             fieldRegion: 'fieldRegion { .id }',

src/components/project/project.resolver.ts

@@ -43,6 +43,7 @@ import { SecuredOrganization } from '../organization/dto';
 import { PartnershipLoader } from '../partnership';
 import {
   PartnershipListInput,
+  SecuredPartnership,
   SecuredPartnershipList,
 } from '../partnership/dto';
 import { ProjectChangeRequestLoader } from '../project-change-request';
@@ -281,6 +282,16 @@ export class ProjectResolver {
     };
   }
 
+  @ResolveField(() => SecuredPartnership)
+  async primaryPartnership(
+    @Parent() project: Project,
+    @Loader(PartnershipLoader) partnerships: LoaderOf<PartnershipLoader>,
+  ): Promise<SecuredPartnership> {
+    return await mapSecuredValue(project.primaryPartnership, ({ id }) =>
+      partnerships.load({ id, view: { active: true } }),
+    );
+  }
+
   @ResolveField(() => SecuredLocation)
   async primaryLocation(
     @Parent() project: Project,