fix primary organization to be overwritten and exposed to the API

Author: rdonigian

src/components/user/user.repository.ts

@@ -290,13 +290,13 @@ export class UserRepository extends DtoRepository(User) {
         if (primary) {
           q.subQuery((sub) =>
             sub
-              .with('user, org')
+              .with('user')
               .match([
                 node('user'),
                 relation('out', 'oldRel', 'primaryOrganization', {
                   active: true,
                 }),
-                node('org'),
+                node('anyOrg', 'Organization'),
               ])
               .setValues({ 'oldRel.active': false })
               .return('oldRel as oldPrimaryRel')
@@ -333,6 +333,20 @@ export class UserRepository extends DtoRepository(User) {
     }
   }
 
+  async getPrimaryOrganizationId(userId: ID): Promise<ID | null> {
+    const result = await this.db
+      .query()
+      .match([
+        node('user', 'User', { id: userId }),
+        relation('out', '', 'primaryOrganization', ACTIVE),
+        node('org', 'Organization'),
+      ])
+      .return<{ orgId: ID }>('org.id as orgId')
+      .first();
+
+    return result?.orgId ?? null;
+  }
+
   async removeOrganizationFromUser(request: RemoveOrganizationFromUser) {
     const result = await this.db
       .query()

src/components/user/user.resolver.ts

@@ -25,8 +25,8 @@ import {
   OrganizationListInput,
   SecuredOrganizationList,
 } from '../organization/dto';
-import { PartnerLoader } from '../partner';
-import { PartnerListInput, SecuredPartnerList } from '../partner/dto';
+import { PartnerLoader, PartnerService } from '../partner';
+import { Partner, PartnerListInput, SecuredPartnerList } from '../partner/dto';
 import { TimeZoneService } from '../timezone';
 import { SecuredTimeZone } from '../timezone/timezone.dto';
 import {
@@ -68,6 +68,7 @@ class ModifyLocationArgs {
 export class UserResolver {
   constructor(
     private readonly userService: UserService,
+    private readonly partnerService: PartnerService,
     private readonly timeZoneService: TimeZoneService,
     private readonly identity: Identity,
   ) {}
@@ -162,6 +163,14 @@ export class UserResolver {
     return list;
   }
 
+  @ResolveField(() => Partner, { nullable: true })
+  async primaryOrganization(@Parent() { id }: User): Promise<Partner | null> {
+    const primaryOrgId = await this.userService.getPrimaryOrganizationId(id);
+    return primaryOrgId
+      ? await this.partnerService.readOnePartnerByOrgId(primaryOrgId)
+      : null;
+  }
+
   @ResolveField(() => SecuredPartnerList)
   async partners(
     @Parent() { id }: User,

src/components/user/user.service.ts

@@ -333,6 +333,11 @@ export class UserService {
     await this.userRepo.assignOrganizationToUser(request);
   }
 
+  async getPrimaryOrganizationId(userId: ID) {
+    const org = await this.userRepo.getPrimaryOrganizationId(userId);
+    return org ?? null;
+  }
+
   async removeOrganizationFromUser(
     request: RemoveOrganizationFromUser,
   ): Promise<void> {