Add custom datasource for dsdm refill points (#262)

Author: akileng56

packages/esm-ugandaemr-app/src/custom-expressions/custom-apis.ts

@@ -41,3 +41,9 @@ export async function getPatientPrograms(patientUuid: string) {
 
   return filteredDSDModels;
 }
+
+export async function getCohortCategorization(uuid: string) {
+  let apiUrl = `${restBaseUrl}/cohortm/cohort?v=custom:(name,uuid)&cohortType=${uuid}`;
+
+  return await openmrsFetch(apiUrl);
+}

packages/esm-ugandaemr-app/src/custom-expressions/custom-expressions.ts

@@ -1,6 +1,8 @@
-import { getConceptDataType, getLatestObs, getPatientPrograms } from './custom-apis';
+import { getCohortCategorization, getConceptDataType, getLatestObs, getPatientPrograms } from './custom-apis';
 import dayjs from 'dayjs';
 import { configSchema } from '@ugandaemr/esm-care-panel-app/src/config-schema';
+import { DataSource } from '@openmrs/openmrs-form-engine-lib';
+import { OpenmrsResource } from '@openmrs/esm-framework';
 
 export async function latestObs(patientId: string, conceptUuid: string) {
   const response = await getLatestObs(patientId, conceptUuid);
@@ -44,3 +46,24 @@ export function CalcMonthsOnART(artStartDate: Date, followupDate: Date) {
   }
   return artStartDate && followupDate ? resultMonthsOnART : null;
 }
+
+export class DSDMCategorizationDatasource implements DataSource<OpenmrsResource> {
+  fetchData(searchTerm: string, config?: Record<string, any>): Promise<any[]> {
+    return getCohortCategorization(config?.cohortUuid).then((response) => {
+      let data = [];
+      response?.data?.results?.map((dataItem) => {
+        data.push({
+          display: dataItem?.name,
+          uuid: dataItem?.uuid,
+        });
+      });
+      return data?.map((item) => this.toUuidAndDisplay(item));
+    });
+  }
+  toUuidAndDisplay(data: OpenmrsResource): OpenmrsResource {
+    if (typeof data.uuid === 'undefined' || typeof data.display === 'undefined') {
+      throw new Error("'uuid' or 'display' not found in the OpenMRS object.");
+    }
+    return data;
+  }
+}

packages/esm-ugandaemr-app/src/index.ts

@@ -1,7 +1,7 @@
 import { defineConfigSchema, getAsyncLifecycle, getSyncLifecycle, provide } from '@openmrs/esm-framework';
 import { configSchema } from './config-schema';
 import { moduleName } from './constants';
-import { registerExpressionHelper } from '@openmrs/openmrs-form-engine-lib';
+import { registerCustomDataSource, registerExpressionHelper } from '@openmrs/openmrs-form-engine-lib';
 
 import formBuilderAppMenu from './menu-app-items/form-builder-app-item/form-builder-app-item.component';
 import systemInfoAppMenu from './menu-app-items/system-info-app-item/system-info-app-item.component';
@@ -18,7 +18,12 @@ import SubjectiveFindingsComponent from './pages/clinical-patient-summary/clinic
 import ObjectiveFindingsComponent from './pages/clinical-patient-summary/clinical-patient-summary-tabs/objective-findings.component';
 import TreatmentPlanComponent from './pages/clinical-patient-summary/clinical-patient-summary-tabs/treatment-plan.component';
 import AssessmentComponent from './pages/clinical-patient-summary/clinical-patient-summary-tabs/assessment.component';
-import { CalcMonthsOnART, latestObs, patientDSDM } from './custom-expressions/custom-expressions';
+import {
+  CalcMonthsOnART,
+  DSDMCategorizationDatasource,
+  latestObs,
+  patientDSDM,
+} from './custom-expressions/custom-expressions';
 
 export const importTranslation = require.context('../translations', false, /.json$/, 'lazy');
 
@@ -40,6 +45,15 @@ export function startupApp() {
   registerExpressionHelper('cusGetLatestObs', latestObs);
   registerExpressionHelper('getPatientDSMD', patientDSDM);
   registerExpressionHelper('CustomMonthsOnARTCalc', CalcMonthsOnART);
+
+  registerCustomDataSource({
+    name: 'dsdm_categorization_datasource',
+    load: () => {
+      return Promise.resolve({
+        default: new DSDMCategorizationDatasource(),
+      });
+    },
+  });
 }
 
 export const systemInfoPage = getAsyncLifecycle(() => import('./pages/system-info/system-info.component'), {