feat: [DHIS2-17970] Auto-select orgUnit if there is only one available

src/core_modules/capture-core/components/WidgetRelatedStages/WidgetRelatedStages.component.js

@@ -1,6 +1,7 @@
 // @flow
 import React, { forwardRef, useCallback, useEffect, useImperativeHandle, useState } from 'react';
 import { useRelatedStages } from './useRelatedStages';
+import { useOrgUnitAutoSelect } from './hooks/useAutoSelctOrgUnitRelatedStage';
 import type { Props, RelatedStageDataValueStates } from './WidgetRelatedStages.types';
 import { RelatedStagesActions } from './RelatedStagesActions';
 import { relatedStageStatus } from './constants';
@@ -36,6 +37,7 @@ const WidgetRelatedStagesPlain = ({
         orgUnit: undefined,
         linkedEventId: undefined,
     });
+    const { isLoading: orgUnitLoading } = useOrgUnitAutoSelect(setRelatedStageDataValues);
 
     const addErrorMessage = (message: ErrorMessagesForRelatedStages) => {
         setErrorMessages((prevMessages: Object) => ({
@@ -81,7 +83,7 @@ const WidgetRelatedStagesPlain = ({
         }
     }, [formIsValid, relatedStageDataValues]);
 
-    if (!currentRelatedStagesStatus || !selectedRelationshipType || isLoadingEvents) {
+    if (!currentRelatedStagesStatus || !selectedRelationshipType || isLoadingEvents || orgUnitLoading) {
         return null;
     }
 
@@ -103,8 +105,8 @@ const WidgetRelatedStagesPlain = ({
     );
 };
 
-export const WidgetRelatedStages = forwardRef<Props, {|
+export const WidgetRelatedStages = forwardRef < Props, {|
     eventHasLinkableStageRelationship: Function,
-    formIsValidOnSave: Function,
-    getLinkedStageValues: Function
-|}>(WidgetRelatedStagesPlain);
+        formIsValidOnSave: Function,
+            getLinkedStageValues: Function
+                |}>(WidgetRelatedStagesPlain);

src/core_modules/capture-core/components/WidgetRelatedStages/hooks/useAutoSelctOrgUnitRelatedStage.js

@@ -0,0 +1,33 @@
+// @flow
+import { useEffect } from 'react';
+import { useApiMetadataQuery } from '../../../utils/reactQueryHelpers';
+
+export const useOrgUnitAutoSelect = (setRelatedStageDataValues: any) => {
+    const queryKey = ['organisationUnits'];
+    const queryFn = {
+        resource: 'organisationUnits',
+        params: {
+            fields: 'id,path,displayName',
+            withinUserSearchHierarchy: true,
+            page: 2,
+        },
+    };
+    const { data, isLoading } = useApiMetadataQuery(queryKey, queryFn);
+
+    useEffect(() => {
+        if (!isLoading && data) {
+            const orgUnits = data.organisationUnits;
+            if (!orgUnits || orgUnits.length !== 1) return;
+            const [orgUnit] = orgUnits;
+            const { displayName, ...rest } = orgUnit;
+            setRelatedStageDataValues(prev => ({
+                ...prev,
+                orgUnit: { ...rest, name: displayName },
+            }));
+        }
+    }, [data, isLoading, setRelatedStageDataValues]);
+
+    return {
+        isLoading,
+    };
+};