fix(tree): programmatic selection

chakra-ui · Sep 13, 2024 · cafa231 · cafa231
1 parent 8d4d226
commit cafa231
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/.changeset/lazy-geckos-complain.md b/.changeset/lazy-geckos-complain.md
@@ -0,0 +1,5 @@
+---
+"@zag-js/tree-view": patch
+---
+
+Fix issue where programmatic selection does not account for singular selection
diff --git a/packages/machines/tree-view/src/tree-view.connect.ts b/packages/machines/tree-view/src/tree-view.connect.ts
@@ -65,8 +65,17 @@ export function connect<T extends PropTypes>(state: State, send: Send, normalize
         send({ type: "SELECTED.ALL" })
         return
       }
-      const nextValue = new Set(selectedValue)
-      value.forEach((id) => nextValue.add(id))
+      const nextValue = new Set()
+      if (state.context.selectionMode === "single") {
+        // For single selection, only add the last item
+        if (value.length > 0) {
+          nextValue.add(value[value.length - 1])
+        }
+      } else {
+        // For multiple selection, add all items
+        value.forEach((id) => nextValue.add(id))
+        selectedValue.forEach((id) => nextValue.add(id))
+      }
       send({ type: "SELECTED.SET", value: nextValue, src: "select" })
     },
     focusBranch(id) {