Fix radio folder behavior when nested

seasketch · Nov 14, 2024 · d9485d3 · d9485d3
1 parent 472cd4f
commit d9485d3
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/packages/client/src/components/TreeView.tsx b/packages/client/src/components/TreeView.tsx
@@ -356,7 +356,7 @@ export default function TreeView({
         }
         onChecked(ids, isChecked);
         if (isChecked && item.parentId) {
-          const parent = data.find((i) => i.node.id === item.parentId);
+          const parent = findTreeNodeRecursive(data, item.parentId);
           if (parent?.radioFolder) {
             // Find and hide any visible siblings and their children
             const siblings = parent.children.filter(
@@ -684,3 +684,16 @@ export function overlayLayerFragmentsToTreeItems(
   }
   return items;
 }
+
+function findTreeNodeRecursive(data: TreeNode[], id: string): TreeNode | null {
+  for (const node of data) {
+    if (node.node.id === id) {
+      return node;
+    }
+    const found = findTreeNodeRecursive(node.children, id);
+    if (found) {
+      return found;
+    }
+  }
+  return null;
+}