Abilities to set focus by passing focusedId props

src/TreeView/constants.ts

@@ -36,6 +36,7 @@ export const treeTypes = {
   changeSelectMany: "SELECT_MANY",
   exclusiveChangeSelectMany: "EXCLUSIVE_CHANGE_SELECT_MANY",
   focus: "FOCUS",
+  clearFocus: "CLEAR_FOCUS",
   blur: "BLUR",
   disable: "DISABLE",
   enable: "ENABLE",

src/TreeView/index.tsx

@@ -33,6 +33,7 @@ import {
   noop,
   isBranchNotSelectedAndHasOnlySelectedChild,
   getOnSelectTreeAction,
+  getBranchNodesToExpand,
 } from "./utils";
 import { Node } from "./node";
 import {
@@ -62,6 +63,7 @@ interface IUseTreeProps {
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   onLoadData?: (props: ITreeViewOnLoadDataProps) => Promise<any>;
   togglableSelect?: boolean;
+  focusedId?: NodeId;
 }
 
 const useTree = ({
@@ -82,6 +84,7 @@ const useTree = ({
   propagateSelect,
   propagateSelectUpwards,
   treeRef,
+  focusedId,
 }: IUseTreeProps) => {
   const treeParentNode = getTreeParent(data);
   const [state, dispatch] = useReducer(treeReducer, {
@@ -419,18 +422,49 @@ const useTree = ({
       nodeRefs?.current != null &&
       leafRefs?.current != null
     ) {
-      const isTreeActive = (treeRef?.current == null) || 
-        (document.activeElement && treeRef.current.contains(document.activeElement));
+      const isTreeActive =
+        treeRef?.current == null ||
+        (document.activeElement &&
+          treeRef.current.contains(document.activeElement));
       if (isTreeActive) {
         // Only scroll and focus on the tree when it is the active element on the page.
         // This prevents controlled updates from scrolling to the tree and giving it focus.
         const tabbableNode = nodeRefs.current[tabbableId];
-        const leafNode = leafRefs.current[lastInteractedWith]; 
+        const leafNode = leafRefs.current[lastInteractedWith];
         scrollToRef(leafNode);
         focusRef(tabbableNode);
       }
     }
   }, [tabbableId, nodeRefs, leafRefs, lastInteractedWith]);
+  //Controlled focus
+  useEffect(() => {
+    if (!focusedId || disabledIds.has(focusedId)) {
+      dispatch({
+        type: treeTypes.clearFocus,
+        id: treeParentNode.children[0],
+      });
+    }
+
+    if (
+      focusedId &&
+      getTreeNode(data, focusedId) &&
+      !disabledIds.has(focusedId)
+    ) {
+      const nodesToExpand = getBranchNodesToExpand(data, focusedId);
+      if (nodesToExpand.length) {
+        dispatch({
+          type: treeTypes.expandMany,
+          ids: nodesToExpand,
+          lastInteractedWith: focusedId,
+        });
+      }
+      dispatch({
+        type: treeTypes.focus,
+        id: focusedId,
+        lastInteractedWith: focusedId,
+      });
+    }
+  }, [focusedId]);
 
   // The "as const" technique tells Typescript that this is a tuple not an array
   return [state, dispatch] as const;
@@ -518,6 +552,8 @@ export interface ITreeViewProps {
     treeState: ITreeViewState;
     dispatch: React.Dispatch<TreeViewAction>;
   }) => void;
+  /** Id of the node to focus */
+  focusedId?: NodeId;
 }
 
 const TreeView = React.forwardRef<HTMLUListElement, ITreeViewProps>(
@@ -543,6 +579,7 @@ const TreeView = React.forwardRef<HTMLUListElement, ITreeViewProps>(
       clickAction = clickActions.select,
       nodeAction = "select",
       expandedIds,
+      focusedId,
       onBlur,
       ...other
     },
@@ -572,7 +609,8 @@ const TreeView = React.forwardRef<HTMLUListElement, ITreeViewProps>(
       multiSelect,
       propagateSelect,
       propagateSelectUpwards,
-	  treeRef: innerRef,
+      treeRef: innerRef,
+      focusedId,
     });
     propagateSelect = propagateSelect && multiSelect;
 
@@ -1006,6 +1044,9 @@ TreeView.propTypes = {
 
   /** Function called to load data asynchronously on expand */
   onLoadData: PropTypes.func,
+
+  /** Id of the node to focus on */
+  focusedId: PropTypes.oneOfType([PropTypes.string, PropTypes.number]),
 };
 
 export default TreeView;

src/TreeView/reducer.ts

@@ -60,6 +60,7 @@ export type TreeViewAction =
       lastManuallyToggled?: NodeId | null;
     }
   | { type: "FOCUS"; id: NodeId; lastInteractedWith?: NodeId | null }
+  | { type: "CLEAR_FOCUS"; id: NodeId }
   | { type: "BLUR" }
   | { type: "DISABLE"; id: NodeId }
   | { type: "ENABLE"; id: NodeId }
@@ -355,6 +356,13 @@ export const treeReducer = (
         ...state,
         isFocused: false,
       };
+    case treeTypes.clearFocus:
+      return {
+        ...state,
+        isFocused: false,
+        lastInteractedWith: null,
+        tabbableId: action.id,
+      };
     case treeTypes.disable: {
       const disabledIds = new Set<NodeId>(state.disabledIds);
       disabledIds.add(action.id);

src/TreeView/utils.ts

@@ -51,6 +51,18 @@ export const isBranchNode = (data: INode[], i: NodeId) => {
   return !!node.children?.length;
 };
 
+export const getBranchNodesToExpand = (data: INode[], id: NodeId): NodeId[] => {
+  const parentId = getParent(data, id);
+  const isNodeExpandable =
+    parentId &&
+    (isBranchNode(data, parentId) || getTreeNode(data, parentId).isBranch);
+
+  if (!parentId || !isNodeExpandable) {
+    return [];
+  }
+  return [parentId, ...getBranchNodesToExpand(data, parentId)];
+};
+
 export const scrollToRef = (ref: INodeRef) => {
   if (ref != null && ref.scrollIntoView) {
     ref.scrollIntoView({ block: "nearest" });
@@ -290,7 +302,10 @@ export const getAccessibleRange = ({
 /**
  * This is to help consumers to understand that we do not currently support metadata that is a nested object. If this is needed, make an issue in Github
  */
-export type IFlatMetadata = Record<string, string | number | boolean | undefined | null>;
+export type IFlatMetadata = Record<
+  string,
+  string | number | boolean | undefined | null
+>;
 
 interface ITreeNode<M extends IFlatMetadata> {
   id?: NodeId;