Merge pull request #6009 from masslight/bholyshevskyi/otr-1766-if-service-category-is-changed-from-urgent-care-to-workers

otr-1766

Author: VladMstv

packages/zambdas/src/ehr/visit-details/update-visit-details/index.ts

@@ -1,14 +1,15 @@
 import Oystehr, { BatchInputJSONPatchRequest, BatchInputPatchRequest, User } from '@oystehr/sdk';
 import { APIGatewayProxyResult } from 'aws-lambda';
 import { Operation } from 'fast-json-patch';
-import { Appointment, Coding, Encounter, Extension, Patient } from 'fhir/r4b';
+import { Account, Appointment, Coding, Encounter, Extension, Patient } from 'fhir/r4b';
 import { DateTime } from 'luxon';
 import {
   BOOKING_CONFIG,
   BookingDetails,
   cleanUpStaffHistoryTag,
   FHIR_EXTENSION,
   FHIR_RESOURCE_NOT_FOUND,
+  FHIR_RESOURCE_NOT_FOUND_CUSTOM,
   getCriticalUpdateTagOp,
   getReasonForVisitAndAdditionalDetailsFromAppointment,
   getSecret,
@@ -18,11 +19,13 @@ import {
   isValidUUID,
   MISSING_REQUEST_BODY,
   MISSING_REQUIRED_PARAMETERS,
+  OCCUPATIONAL_MEDICINE_ACCOUNT_TYPE,
   REASON_ADDITIONAL_MAX_CHAR,
   Secrets,
   SecretsKeys,
   userMe,
   VALUE_SETS,
+  WORKERS_COMP_ACCOUNT_TYPE,
 } from 'utils';
 import {
   checkOrCreateM2MClientToken,
@@ -31,6 +34,7 @@ import {
   wrapHandler,
   ZambdaInput,
 } from '../../../shared';
+import { accountMatchesType } from '../../shared/harvest';
 
 const ZAMBDA_NAME = 'update-visit-details';
 
@@ -66,10 +70,13 @@ interface EffectInput extends Input {
   user: User;
   appointment: Appointment;
   encounter: Encounter;
+  workersCompAccount?: Account;
+  occupationalMedicineAccount?: Account;
 }
 
 const performEffect = async (input: EffectInput, oystehr: Oystehr): Promise<void> => {
-  const { patient, appointment, bookingDetails, user, encounter } = input;
+  const { patient, appointment, bookingDetails, user, encounter, workersCompAccount, occupationalMedicineAccount } =
+    input;
 
   const patchRequests: BatchInputJSONPatchRequest[] = [];
   if (bookingDetails.confirmedDob) {
@@ -307,6 +314,41 @@ const performEffect = async (input: EffectInput, oystehr: Oystehr): Promise<void
   }
 
   if (bookingDetails.serviceCategory) {
+    const newEncounterAccounts = (encounter.account ?? [])
+      .filter((reference) => reference.reference === 'Account/' + occupationalMedicineAccount?.id)
+      .filter((reference) => reference.reference === 'Account/' + workersCompAccount?.id);
+
+    if (bookingDetails.serviceCategory.code === 'workers-comp') {
+      if (!workersCompAccount) {
+        throw FHIR_RESOURCE_NOT_FOUND_CUSTOM('Workmans Comp Account missing');
+      }
+      newEncounterAccounts.push({
+        reference: 'Account/' + workersCompAccount.id,
+      });
+    }
+
+    if (bookingDetails.serviceCategory.code === 'occupational-medicine') {
+      if (!occupationalMedicineAccount) {
+        throw FHIR_RESOURCE_NOT_FOUND_CUSTOM('Occupational Medicine Account missing');
+      }
+      newEncounterAccounts.push({
+        reference: 'Account/' + occupationalMedicineAccount.id,
+      });
+    }
+
+    const encounterPatch: BatchInputPatchRequest<Encounter> = {
+      method: 'PATCH',
+      url: `/Encounter/${encounter.id}`,
+      operations: [
+        {
+          op: encounter?.account ? 'replace' : 'add',
+          path: '/account',
+          value: newEncounterAccounts,
+        },
+      ],
+    };
+    patchRequests.push(encounterPatch);
+
     const appointmentPatch: BatchInputJSONPatchRequest = {
       url: '/Appointment/' + appointment.id,
       method: 'PATCH',
@@ -365,6 +407,13 @@ const complexValidation = async (input: Input, oystehr: Oystehr): Promise<Effect
     throw FHIR_RESOURCE_NOT_FOUND('Encounter');
   }
 
+  const accounts = (
+    await oystehr.fhir.search<Account>({
+      resourceType: 'Account',
+      params: [{ name: 'patient', value: patientResource.id }],
+    })
+  ).unbundle();
+
   // const selfPay = getPaymentVariantFromEncounter(encounterResource) === PaymentVariant.selfPay;
 
   return {
@@ -373,6 +422,10 @@ const complexValidation = async (input: Input, oystehr: Oystehr): Promise<Effect
     appointment,
     encounter: encounterResource,
     user,
+    workersCompAccount: accounts.find((account) => accountMatchesType(account, WORKERS_COMP_ACCOUNT_TYPE)),
+    occupationalMedicineAccount: accounts.find((account) =>
+      accountMatchesType(account, OCCUPATIONAL_MEDICINE_ACCOUNT_TYPE)
+    ),
   };
 };