Auto-focus AccuDraw distance field

common/changes/@itwin/appui-react/fix-accudraw-focus_2024-03-04-13-06.json

@@ -0,0 +1,10 @@
+{
+  "changes": [
+    {
+      "packageName": "@itwin/appui-react",
+      "comment": "Fix polar mode AccuDraw input focus by correctly focusing the distance field.",
+      "type": "none"
+    }
+  ],
+  "packageName": "@itwin/appui-react"
+}

docs/changehistory/NextVersion.md

@@ -11,7 +11,9 @@ Table of contents:
 ### Changes
 
 - Removed arrow from status bar popup. [#750](https://github.com/iTwin/appui/pull/750)
+- Render AccuDraw Distance field above Angle field when in Polar mode.
 
 ### Fixes
 
-- Fix the issue when right-click + left-click starts a widget drag interaction. #730
+- Fix the issue when right-click + left-click starts a widget drag interaction. [#730](https://github.com/iTwin/appui/pull/730)
+- Fix polar mode AccuDraw input focus by correctly focusing the distance field.

ui/appui-react/src/appui-react/accudraw/AccuDrawFieldContainer.tsx

@@ -118,28 +118,59 @@ export function AccuDrawFieldContainer(props: AccuDrawFieldContainerProps) {
     React.CSSProperties | undefined
   >(undefined);
 
-  const getInputRef = (field: ItemField): React.RefObject<HTMLInputElement> => {
-    let inputRef: React.RefObject<HTMLInputElement>;
+  const getFieldInput = (field: ItemField) => {
     switch (field) {
       case ItemField.X_Item:
-        inputRef = xInputRef;
-        break;
+        return getCurrent(xInputRef);
       case ItemField.Y_Item:
-        inputRef = yInputRef;
-        break;
+        return getCurrent(yInputRef);
       case ItemField.Z_Item:
-        inputRef = zInputRef;
-        break;
+        return getCurrent(zInputRef);
       case ItemField.ANGLE_Item:
-        inputRef = angleInputRef;
-        break;
+        return getCurrent(angleInputRef);
       case ItemField.DIST_Item:
-        inputRef = distanceInputRef;
-        break;
+        return getCurrent(distanceInputRef);
     }
-    return inputRef;
   };
 
+  const getInputToFocus = React.useCallback((field: ItemField | undefined) => {
+    const fieldInput = field === undefined ? undefined : getFieldInput(field);
+    if (fieldInput) {
+      if (document.activeElement === fieldInput) return undefined;
+      return fieldInput;
+    }
+
+    const focusedInput = [
+      xInputRef,
+      yInputRef,
+      zInputRef,
+      angleInputRef,
+      distanceInputRef,
+    ].find((inputRef) => document.activeElement === inputRef.current);
+
+    // One of AccuDraw input fields is focused already.
+    if (focusedInput) return undefined;
+
+    // Focus one of fallback fields.
+    let fallbackInput = getFieldInput(ItemField.DIST_Item);
+    if (fallbackInput) return fallbackInput;
+
+    fallbackInput = getFieldInput(ItemField.X_Item);
+    return fallbackInput;
+  }, []);
+
+  // Moves focus to `field` input if not focused OR fallbacks to default field if none of accudraw fields are focused.
+  const setFocusToField = React.useCallback(
+    (field: ItemField | undefined) => {
+      const input = getInputToFocus(field);
+      if (!input) return;
+
+      input.focus();
+      input.select();
+    },
+    [getInputToFocus]
+  );
+
   React.useEffect(() => {
     return FrameworkAccuDraw.onAccuDrawSetFieldLockEvent.addListener((args) => {
       switch (args.field) {
@@ -162,16 +193,6 @@ export function AccuDrawFieldContainer(props: AccuDrawFieldContainerProps) {
     });
   }, []);
 
-  const setFocusToField = React.useCallback((field: ItemField) => {
-    const inputRef = getInputRef(field);
-
-    // istanbul ignore else
-    if (inputRef.current && document.activeElement !== inputRef.current) {
-      inputRef.current.focus();
-      inputRef.current.select();
-    }
-  }, []);
-
   React.useEffect(() => {
     return FrameworkAccuDraw.onAccuDrawSetFieldFocusEvent.addListener(
       (args) => {
@@ -183,7 +204,7 @@ export function AccuDrawFieldContainer(props: AccuDrawFieldContainerProps) {
 
   React.useEffect(() => {
     return FrameworkAccuDraw.onAccuDrawGrabInputFocusEvent.addListener(() => {
-      if (focusField.current) setFocusToField(focusField.current);
+      setFocusToField(focusField.current);
     });
   }, [setFocusToField]);
 
@@ -426,22 +447,6 @@ export function AccuDrawFieldContainer(props: AccuDrawFieldContainerProps) {
       )}
       {mode === CompassMode.Polar && (
         <>
-          <AccuDrawInputField
-            ref={angleInputRef}
-            isLocked={angleLock}
-            className="uifw-accudraw-angle-value"
-            style={angleStyle}
-            field={ItemField.ANGLE_Item}
-            id={`uifw-accudraw-angle-${containerIndex}`}
-            data-testid="uifw-accudraw-angle"
-            label={angleLabel}
-            iconSpec={angleIcon}
-            valueChangedDelay={delay}
-            onValueChanged={(stringValue) =>
-              handleValueChanged(ItemField.ANGLE_Item, stringValue)
-            }
-            onEscPressed={handleEscPressed}
-          />
           <AccuDrawInputField
             ref={distanceInputRef}
             isLocked={distanceLock}
@@ -458,8 +463,29 @@ export function AccuDrawFieldContainer(props: AccuDrawFieldContainerProps) {
             }
             onEscPressed={handleEscPressed}
           />
+          <AccuDrawInputField
+            ref={angleInputRef}
+            isLocked={angleLock}
+            className="uifw-accudraw-angle-value"
+            style={angleStyle}
+            field={ItemField.ANGLE_Item}
+            id={`uifw-accudraw-angle-${containerIndex}`}
+            data-testid="uifw-accudraw-angle"
+            label={angleLabel}
+            iconSpec={angleIcon}
+            valueChangedDelay={delay}
+            onValueChanged={(stringValue) =>
+              handleValueChanged(ItemField.ANGLE_Item, stringValue)
+            }
+            onEscPressed={handleEscPressed}
+          />
         </>
       )}
     </div>
   );
 }
+
+function getCurrent<T>(ref: React.RefObject<T>) {
+  if (ref.current === null) return undefined;
+  return ref.current;
+}