feat: tree component API for virtual tree

- refactored Tree key handler based on the pr comments
momentum-design · Aug 27, 2024 · e84826d · e84826d
1 parent f377b57
commit e84826d
Showing 1 changed file with 19 additions and 12 deletions.
diff --git a/src/components/Tree/Tree.tsx b/src/components/Tree/Tree.tsx
@@ -119,18 +119,25 @@ const Tree: FC<Props> = (props: Props) => {
   const { keyboardProps } = useKeyboard({
     onKeyDown: (evt) => {
       const key = evt.key as TreeNavKeyCodes;
-      if (TREE_NAVIGATION_KEYS.includes(key)) {
-        evt.preventDefault();
-        const nextActiveNode = getNextActiveNode(
-          tree,
-          activeNodeId,
-          key,
-          excludeTreeRoot,
-          toggleTreeNode
-        );
-        setActiveNodeId(nextActiveNode);
-      } else {
-        evt.continuePropagation();
+      switch (key) {
+        case 'ArrowUp':
+        case 'ArrowDown':
+        case 'ArrowRight':
+        case 'ArrowLeft': {
+          evt.preventDefault();
+          const nextActiveNode = getNextActiveNode(
+            tree,
+            activeNodeId,
+            key,
+            excludeTreeRoot,
+            toggleTreeNode
+          );
+          setActiveNodeId(nextActiveNode);
+          break;
+        }
+        default:
+          evt.continuePropagation();
+          break;
       }
     },
   });