Related Entity Location Sync Strategy.

- User selected sync strategy - User selected sync strategy fixes opensrp/fhircore#3052
onaio · Mar 13, 2024 · 82f6498 · 82f6498
1 parent f99738e
commit 82f6498
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 16 deletions.
diff --git a/plugins/src/main/java/org/smartregister/fhir/gateway/plugins/Constants.java b/plugins/src/main/java/org/smartregister/fhir/gateway/plugins/Constants.java
@@ -20,19 +20,19 @@ public class Constants {
     public static final String PROXY_TO_ENV = "PROXY_TO";
     public static final String SYNC_STRATEGY = "syncStrategy";
     public static final String TAG_SEARCH_PARAM = "_tag";
-
     public static final String AUTHORIZATION = "Authorization";
-
     public static final String KEYCLOAK_UUID = "keycloak-uuid";
     public static final String IDENTIFIER = "identifier";
-
     public static final String PAGINATION_PAGE_SIZE = "_count";
-
     public static final String PAGINATION_PAGE_NUMBER = "_page";
-
     public static final int PAGINATION_DEFAULT_PAGE_SIZE = 20;
-
     public static final int PAGINATION_DEFAULT_PAGE_NUMBER = 1;
+    public static final String SYNC_LOCATIONS = "_syncLocations";
+    public static final String RELATED_ENTITY_LOCATION = "RelatedEntityLocation";
+    public static final String RELATED_ENTITY_TAG_URL_ENV = "RELATED_ENTITY_TAG_URL";
+    public static final String DEFAULT_RELATED_ENTITY_TAG_URL =
+            "https://smartregister.org/related-entity-location-tag-id";
+    public static final String ROLE_ALL_LOCATIONS = "ROLE_ALL_LOCATIONS";
 
     public interface Literals {
         String EQUALS = "=";

diff --git a/plugins/src/main/java/org/smartregister/fhir/gateway/plugins/PermissionAccessChecker.java b/plugins/src/main/java/org/smartregister/fhir/gateway/plugins/PermissionAccessChecker.java
@@ -1,11 +1,6 @@
 package org.smartregister.fhir.gateway.plugins;
 
-import java.util.ArrayList;
-import java.util.Base64;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.Optional;
+import java.util.*;
 import java.util.stream.Collectors;
 
 import javax.inject.Named;
@@ -364,6 +359,7 @@ private static Map<String, List<String>> collateSyncStrategyIds(
             List<String> careTeamIds;
             List<String> organizationIds;
             List<String> locationIds;
+            List<String> relatedEntityLocationIds;
             if (StringUtils.isNotBlank(syncStrategy)) {
                 if (Constants.CARE_TEAM.equalsIgnoreCase(syncStrategy)) {
                     careTeams =
@@ -413,6 +409,18 @@ private static Map<String, List<String>> collateSyncStrategyIds(
                                     : new ArrayList<>();
 
                     resultMap = Map.of(syncStrategy, locationIds);
+                } else if (Constants.RELATED_ENTITY_LOCATION.equalsIgnoreCase(syncStrategy)) {
+                    // assigned locations
+                    relatedEntityLocationIds =
+                            practitionerDetails != null
+                                            && practitionerDetails.getFhirPractitionerDetails()
+                                                    != null
+                                    ? PractitionerDetailsEndpointHelper.getAttributedLocations(
+                                            practitionerDetails
+                                                    .getFhirPractitionerDetails()
+                                                    .getLocationHierarchyList())
+                                    : new ArrayList<>();
+                    resultMap = Map.of(syncStrategy, relatedEntityLocationIds);
                 }
             } else
                 throw new IllegalStateException(

diff --git a/...c/main/java/org/smartregister/fhir/gateway/plugins/PractitionerDetailsEndpointHelper.java b/...c/main/java/org/smartregister/fhir/gateway/plugins/PractitionerDetailsEndpointHelper.java
@@ -43,7 +43,7 @@ public class PractitionerDetailsEndpointHelper {
     public static final Bundle EMPTY_BUNDLE = new Bundle();
     private final IGenericClient r4FHIRClient;
 
-    private final LocationHierarchyEndpointHelper locationHierarchyEndpointHelper;
+    private static LocationHierarchyEndpointHelper locationHierarchyEndpointHelper;
 
     public PractitionerDetailsEndpointHelper(IGenericClient fhirClient) {
         this.r4FHIRClient = fhirClient;
@@ -546,7 +546,7 @@ private List<OrganizationAffiliation> mapBundleToOrganizationAffiliation(
                 : Collections.emptyList();
     }
 
-    private List<LocationHierarchy> getLocationsHierarchy(List<String> locationsIdentifiers) {
+    static List<LocationHierarchy> getLocationsHierarchy(List<String> locationsIdentifiers) {
         List<LocationHierarchy> locationHierarchyList = new ArrayList<>();
         LocationHierarchy locationHierarchy;
         for (String locationsIdentifier : locationsIdentifiers) {

diff --git a/plugins/src/main/java/org/smartregister/fhir/gateway/plugins/SyncAccessDecision.java b/plugins/src/main/java/org/smartregister/fhir/gateway/plugins/SyncAccessDecision.java
@@ -51,7 +51,7 @@ public class SyncAccessDecision implements AccessDecision {
     private final String syncStrategy;
     private final String applicationId;
     private final boolean accessGranted;
-    private final Map<String, List<String>> syncStrategyIds;
+    private Map<String, List<String>> syncStrategyIds;
     private final List<String> roles;
     private IgnoredResourcesConfig config;
     private final String keycloakUUID;
@@ -101,8 +101,25 @@ public RequestMutation getRequestMutation(RequestDetailsReader requestDetailsRea
         if (isSyncUrl(requestDetailsReader)
                 && !shouldSkipDataFiltering(
                         requestDetailsReader)) { // Check if it is the Sync URL and Skip app-wide
+
             // accessible resource requests
-            if (syncStrategyIds.isEmpty()
+            if (Constants.RELATED_ENTITY_LOCATION.equalsIgnoreCase(syncStrategy)) {
+                Map<String, String[]> parameters =
+                        new HashMap<>(requestDetailsReader.getParameters());
+                String[] syncLocations = parameters.get(Constants.SYNC_LOCATIONS);
+                List<String> relatedEntityLocationIds;
+                if (roles.contains(Constants.ROLE_ALL_LOCATIONS) && syncLocations != null) {
+                    // selected locations
+
+                    List<String> locationUuids = Arrays.asList(syncLocations[0].split(","));
+                    relatedEntityLocationIds =
+                            PractitionerDetailsEndpointHelper.getAttributedLocations(
+                                    PractitionerDetailsEndpointHelper.getLocationsHierarchy(
+                                            locationUuids));
+                    this.syncStrategyIds = Map.of(syncStrategy, relatedEntityLocationIds);
+                }
+
+            } else if (syncStrategyIds.isEmpty()
                     || StringUtils.isBlank(syncStrategy)
                     || (syncStrategyIds.containsKey(syncStrategy)
                             && syncStrategyIds.get(syncStrategy).isEmpty())) {
@@ -372,6 +389,9 @@ private String getSyncTagUrl(String syncStrategy) {
         } else if (syncStrategy.equalsIgnoreCase(Constants.CARE_TEAM)) {
             return getEnvironmentVar(
                     Constants.CARE_TEAM_TAG_URL_ENV, Constants.DEFAULT_CARE_TEAM_TAG_URL);
+        } else if (syncStrategy.equalsIgnoreCase(Constants.RELATED_ENTITY_LOCATION)) {
+            return getEnvironmentVar(
+                    Constants.RELATED_ENTITY_TAG_URL_ENV, Constants.DEFAULT_RELATED_ENTITY_TAG_URL);
         } else {
             return null;
         }