CAMS-283 - Add syncOfficeStaff to Offices use case

backend/functions/jest.config.js

@@ -23,6 +23,7 @@ module.exports = {
     'lib/cosmos-humble-objects/',
     'lib/testing/mock-data/index.ts',
     'lib/testing/local-data/',
+    'lib/testing/isolated-integration/',
     'lib/testing/testing-utilities.ts',
     'jest.*config.js',
     'lib/adapters/gateways/okta/HumbleVerifier.ts',

backend/functions/lib/adapters/gateways/consolidations.cosmosdb.repository.ts

@@ -29,6 +29,14 @@ export default class ConsolidationOrdersCosmosDbRepository
     throw new Error('Method not implemented.');
   }
 
+  upsert(
+    _context: ApplicationContext,
+    _partitionKey: string,
+    _data: ConsolidationOrder,
+  ): Promise<ConsolidationOrder> {
+    throw new Error('Method not implemented.');
+  }
+
   put(context: ApplicationContext, data: ConsolidationOrder): Promise<ConsolidationOrder> {
     return this.repo.put(context, data);
     throw new Error('Method not implemented.');

backend/functions/lib/adapters/gateways/cosmos/cosmos.repository.ts

@@ -79,6 +79,19 @@ export class CosmosDbRepository<T> implements DocumentRepository<T> {
     return this.execute<T>(context, lambdaToExecute);
   }
 
+  public async upsert(context: ApplicationContext, partitionKey: string, data: T): Promise<T> {
+    const lambdaToExecute = async <T>(): Promise<T> => {
+      const { resource } = await this.cosmosDbClient
+        .database(this.cosmosConfig.databaseName)
+        .container(this.containerName)
+        .items.upsert(data, { partitionKey });
+
+      context.logger.debug(this.moduleName, `${typeof data} Inserted/Updated ${resource.id}`);
+      return resource;
+    };
+    return this.execute<T>(context, lambdaToExecute);
+  }
+
   public async put(context: ApplicationContext, data: T): Promise<T> {
     const lambdaToExecute = async <T>(): Promise<T> => {
       const { resource } = await this.cosmosDbClient

backend/functions/lib/adapters/gateways/offices.cosmosdb.repository.test.ts

@@ -5,6 +5,9 @@ import { createMockApplicationContext } from '../../testing/testing-utilities';
 import MockData from '../../../../../common/src/cams/test-utilities/mock-data';
 import { CosmosDbRepository } from './cosmos/cosmos.repository';
 import { CamsRole } from '../../../../../common/src/cams/roles';
+import { OfficeStaff } from '../../../../../common/src/cams/staff';
+import { SYSTEM_USER_REFERENCE } from '../../../../../common/src/cams/auditable';
+import { getCamsUserReference } from '../../../../../common/src/cams/session';
 
 describe('offices cosmosDB repository tests', () => {
   let context: ApplicationContext;
@@ -21,8 +24,20 @@ describe('offices cosmosDB repository tests', () => {
 
   test('should query data with office code, staff doc type, and trial attorney role', async () => {
     const officeCode = 'test-office';
-    const attorneys = MockData.buildArray(MockData.getAttorneyUser, 5);
-    jest.spyOn(MockHumbleQuery.prototype, 'fetchAll').mockResolvedValue({ resources: attorneys });
+    const attorneys = MockData.buildArray(MockData.getAttorneyUser, 5).map((user) =>
+      getCamsUserReference(user),
+    );
+    const staffDocs: OfficeStaff[] = attorneys.map((user) => {
+      return {
+        id: user.id,
+        documentType: 'OFFICE_STAFF',
+        officeCode,
+        ...user,
+        updatedOn: '2024-10-01T00:00:00.000Z',
+        updatedBy: SYSTEM_USER_REFERENCE,
+      };
+    });
+    jest.spyOn(MockHumbleQuery.prototype, 'fetchAll').mockResolvedValue({ resources: staffDocs });
     const querySpy = jest.spyOn(CosmosDbRepository.prototype, 'query');
 
     const actual = await repo.getOfficeAttorneys(context, officeCode);

backend/functions/lib/adapters/gateways/offices.cosmosdb.repository.ts

@@ -3,17 +3,20 @@ import { ApplicationContext } from '../types/basic';
 import { AttorneyUser, CamsUserReference } from '../../../../../common/src/cams/users';
 import { CosmosDbRepository } from './cosmos/cosmos.repository';
 import { UstpOfficeDetails } from '../../../../../common/src/cams/courts';
+import { OfficeStaff } from '../../../../../common/src/cams/staff';
 import { CamsRole } from '../../../../../common/src/cams/roles';
+import { createAuditRecord } from '../../../../../common/src/cams/auditable';
+import { getCamsUserReference } from '../../../../../common/src/cams/session';
 
 const MODULE_NAME: string = 'COSMOS_DB_REPOSITORY_OFFICES';
 const CONTAINER_NAME: string = 'offices';
 
 export class OfficesCosmosDbRepository implements OfficesRepository {
-  private officeStaffRepo: CosmosDbRepository<CamsUserReference>;
+  private officeStaffRepo: CosmosDbRepository<OfficeStaff>;
   private officesRepo: CosmosDbRepository<UstpOfficeDetails>;
 
   constructor(context: ApplicationContext) {
-    this.officeStaffRepo = new CosmosDbRepository<CamsUserReference>(
+    this.officeStaffRepo = new CosmosDbRepository<OfficeStaff>(
       context,
       CONTAINER_NAME,
       MODULE_NAME,
@@ -24,6 +27,19 @@ export class OfficesCosmosDbRepository implements OfficesRepository {
       MODULE_NAME,
     );
   }
+  async putOfficeStaff(
+    context: ApplicationContext,
+    officeCode: string,
+    user: CamsUserReference,
+  ): Promise<void> {
+    const staff = createAuditRecord<OfficeStaff>({
+      id: user.id,
+      documentType: 'OFFICE_STAFF',
+      officeCode,
+      ...user,
+    });
+    await this.officeStaffRepo.upsert(context, officeCode, staff);
+  }
 
   async getOfficeAttorneys(
     context: ApplicationContext,
@@ -43,6 +59,7 @@ export class OfficesCosmosDbRepository implements OfficesRepository {
         },
       ],
     };
-    return await this.officeStaffRepo.query(context, querySpec);
+    const docs = await this.officeStaffRepo.query(context, querySpec);
+    return docs.map((doc) => getCamsUserReference(doc));
   }
 }

backend/functions/lib/testing/isolated-integration/test-okta-group-api.ts

@@ -1,6 +1,10 @@
+import { InvocationContext } from '@azure/functions';
+import applicationContextCreator from '../../../azure/application-context-creator';
 import OktaUserGroupGateway from '../../adapters/gateways/okta/okta-user-group-gateway';
 import { UserGroupGatewayConfig } from '../../adapters/types/authorization';
 import { getUserGroupGatewayConfig } from '../../configs/user-groups-gateway-configuration';
+import { OfficesUseCase } from '../../use-cases/offices/offices';
+import { LoggerImpl } from '../../adapters/services/logger.service';
 
 async function testOktaGroupApi() {
   console.log('Isolated Integration Test: Okta Group Api', '\n');
@@ -16,6 +20,21 @@ async function testOktaGroupApi() {
       const users = await OktaUserGroupGateway.getUserGroupUsers(config, group);
       console.log(`${group.name} users`, users, '\n');
     }
+
+    const context = await applicationContextCreator.getApplicationContext({
+      invocationContext: new InvocationContext(),
+      logger: new LoggerImpl('test-invocation'),
+    });
+    const useCase = new OfficesUseCase();
+    const results = await useCase.syncOfficeStaff(context);
+
+    const attorneys = await useCase.getOfficeAttorneys(
+      context,
+      'USTP_CAMS_Region_2_Office_Manhattan',
+    );
+    console.log('attorneys', attorneys, '\n');
+
+    console.log('syncOfficeStaff', JSON.stringify(results, null, 2), '\n');
   } catch (error) {
     console.error(error, '\n');
   } finally {

backend/functions/lib/testing/local-data/local-cosmos-db-repository.ts

@@ -17,6 +17,10 @@ export class LocalCosmosDbRepository<T extends Item> implements DocumentReposito
     throw new Error('Method not implemented.');
   }
 
+  upsert(_context: ApplicationContext, _partitionKey: string, _data: T): Promise<T> {
+    throw new Error('Method not implemented.');
+  }
+
   async put(_context: ApplicationContext, data: T): Promise<T> {
     const doc: T = { ...data, id: crypto.randomUUID() };
     this.container.push(doc);

backend/functions/lib/use-cases/gateways.types.ts

@@ -14,7 +14,7 @@ import {
 import { CaseAssignmentHistory, CaseHistory } from '../../../../common/src/cams/history';
 import { CaseDocket } from '../../../../common/src/cams/cases';
 import { OrdersSearchPredicate } from '../../../../common/src/api/search';
-import { AttorneyUser } from '../../../../common/src/cams/users';
+import { AttorneyUser, CamsUserReference } from '../../../../common/src/cams/users';
 
 export interface RepositoryResource {
   id?: string;
@@ -23,6 +23,7 @@ export interface RepositoryResource {
 export interface DocumentRepository<T extends RepositoryResource> {
   get(context: ApplicationContext, id: string, partitionKey: string): Promise<T>;
   update(context: ApplicationContext, id: string, partitionKey: string, data: T);
+  upsert(context: ApplicationContext, partitionKey: string, data: T): Promise<T>;
   put(context: ApplicationContext, data: T): Promise<T>;
   putAll(context: ApplicationContext, list: T[]): Promise<T[]>;
   delete(context: ApplicationContext, id: string, partitionKey: string);
@@ -82,6 +83,11 @@ export interface CasesRepository {
 
 export interface OfficesRepository {
   getOfficeAttorneys(context: ApplicationContext, officeCode: string): Promise<AttorneyUser[]>;
+  putOfficeStaff(
+    context: ApplicationContext,
+    officeCode: string,
+    user: CamsUserReference,
+  ): Promise<void>;
 }
 
 // TODO: Move these models to a top level models file?

backend/functions/lib/use-cases/offices/offices.ts

@@ -1,6 +1,12 @@
-import { OfficeDetails } from '../../../../../common/src/cams/courts';
+import { OfficeDetails, UstpOfficeDetails } from '../../../../../common/src/cams/courts';
+import { CamsUserReference } from '../../../../../common/src/cams/users';
 import { ApplicationContext } from '../../adapters/types/basic';
-import { getOfficesGateway, getOfficesRepository } from '../../factory';
+import {
+  getOfficesGateway,
+  getUserGroupGateway,
+  getOfficesRepository,
+  getStorageGateway,
+} from '../../factory';
 import { AttorneyUser } from '../../../../../common/src/cams/users';
 
 export class OfficesUseCase {
@@ -16,4 +22,64 @@ export class OfficesUseCase {
     const repository = getOfficesRepository(context);
     return repository.getOfficeAttorneys(context, officeCode);
   }
+
+  public async syncOfficeStaff(context: ApplicationContext): Promise<object> {
+    const config = context.config.userGroupGatewayConfig;
+    const repository = getOfficesRepository(context);
+    const gateway = getUserGroupGateway(context);
+    const storage = getStorageGateway(context);
+
+    // Get IdP to CAMS mappings.
+    const offices = storage.getUstpOffices();
+    const groupToRoleMap = storage.getRoleMapping();
+    const groupToOfficeMap = [...offices.values()].reduce((acc, office) => {
+      acc.set(office.idpGroupId, office);
+      return acc;
+    }, new Map<string, UstpOfficeDetails>());
+
+    // Filter out any groups not relevant to CAMS.
+    const userGroups = await gateway.getUserGroups(config);
+    const officeGroups = userGroups.filter((group) => groupToOfficeMap.has(group.name));
+    const roleGroups = userGroups.filter((group) => groupToRoleMap.has(group.name));
+
+    // Map roles to users.
+    const userMap = new Map<string, CamsUserReference>();
+    for (const roleGroup of roleGroups) {
+      const users = await gateway.getUserGroupUsers(config, roleGroup);
+      const role = groupToRoleMap.get(roleGroup.name);
+      for (const user of users) {
+        if (userMap.has(user.id)) {
+          userMap.get(user.id).roles.push(role);
+        } else {
+          user.roles = [role];
+          userMap.set(user.id, user);
+        }
+      }
+    }
+
+    // Write users with roles to the repo for each office.
+    const officesWithUsers: UstpOfficeDetails[] = [];
+    for (const officeGroup of officeGroups) {
+      const office = { ...groupToOfficeMap.get(officeGroup.name), staff: [] };
+
+      const users = await gateway.getUserGroupUsers(config, officeGroup);
+      for (const user of users) {
+        if (!userMap.has(user.id)) {
+          userMap.set(user.id, user);
+        }
+        const userWithRoles = userMap.has(user.id) ? userMap.get(user.id) : user;
+        office.staff.push(userWithRoles);
+        await repository.putOfficeStaff(context, office.officeCode, userWithRoles);
+      }
+
+      officesWithUsers.push(office);
+    }
+
+    // TODO: What to do with users with roles WITHOUT offices?
+    return {
+      userGroups,
+      users: [...userMap.values()],
+      officesWithUsers,
+    };
+  }
 }

common/src/cams/staff.ts

@@ -0,0 +1,8 @@
+import { Auditable } from './auditable';
+import { CamsUserReference } from './users';
+
+export type OfficeStaff = CamsUserReference &
+  Auditable & {
+    documentType: 'OFFICE_STAFF';
+    officeCode: string;
+  };