@rivet-health/design-system 33.1.2 → 34.0.0

package/fesm2015/rivet-health-design-system.mjs

@@ -2001,6 +2001,7 @@ var RivSelect;
                 error: null,
             },
             sourceOptionGroups: [],
+            originalOptionGroups: [],
             selection: {
                 selected: new OptionSet(),
             },
@@ -2014,7 +2015,7 @@ var RivSelect;
         const actions = new Subject();
         const internalActions = new Subject();
         const coreStateWithSelectionSnapshot = merge(actions, internalActions).pipe(scan(([state, snapshot], action) => {
-            const newState = sideEffects(reduce(state, action), action, source, internalActions);
+            const newState = sideEffects(reduce(state, action, options), action, source, internalActions);
             const newSelectionSnapshot = (action.type === 'openChange' && action.open) ||
                 (action.type === 'loadComplete' &&
                     (newState.display.open || (options === null || options === void 0 ? void 0 : options.inline)))
@@ -2028,7 +2029,7 @@ var RivSelect;
         const state = connectable(coreStateWithSelectionSnapshot.pipe(map(([s, selectionAtOpen]) => {
             var _a, _b, _c, _d, _e, _f;
             const allOptionGroups = s.sourceOptionGroups.map(optionGroup => {
-                return Object.assign(Object.assign({}, optionGroup), { options: optionGroup.options.map(option => augmentOptionToFullOption(s, option)) });
+                return Object.assign(Object.assign({}, optionGroup), { options: optionGroup.options.map(option => augmentOptionToFullOption(s, option, options)) });
             });
             const topLevelOptions = getTopLevelOptions(allOptionGroups);
             const sortedOptionGroups = (options === null || options === void 0 ? void 0 : options.allowMultiSelect)
@@ -2227,7 +2228,10 @@ var RivSelect;
                         : null)));
                 });
             }
-            return [{ options: addHighlightsToOption(ordered) }];
+            return {
+                data: [{ options: addHighlightsToOption(ordered) }],
+                originalData: data,
+            };
         });
         return createManager(source, options);
     }
@@ -2358,7 +2362,16 @@ var RivSelect;
         if ((options === null || options === void 0 ? void 0 : options.selectedOptionLimit) === undefined) {
             return optionGroups;
         }
-        const selectedOrIndeterminateTopOptions = topLevelOptions.filter(option => option.selected != false);
+        // When useTopLevelOptionsForDisplay is true and search is active, count
+        // selected top-level options from the original (unfiltered) data to ensure
+        // already-selected options that don't match the search are still counted.
+        const useOriginalForCounting = (options === null || options === void 0 ? void 0 : options.useTopLevelOptionsForDisplay) &&
+            state.query.search &&
+            state.originalOptionGroups.length > 0;
+        const topOptionsForCounting = useOriginalForCounting
+            ? getTopLevelOptions(state.originalOptionGroups).map(option => augmentOptionToFullOption(state, option, options))
+            : topLevelOptions;
+        const selectedOrIndeterminateTopOptions = topOptionsForCounting.filter(option => option.selected != false);
         const numSelectedOptions = (options === null || options === void 0 ? void 0 : options.useTopLevelOptionsForDisplay)
             ? selectedOrIndeterminateTopOptions.length
             : state.selection.selected.size;
@@ -2431,13 +2444,43 @@ var RivSelect;
             return false;
         return 'indeterminate';
     }
-    function augmentOptionToFullOption(state, option) {
-        var _a;
+    function augmentOptionToFullOption(state, option, managerOptions) {
+        var _a, _b;
         if ((_a = option.children) === null || _a === void 0 ? void 0 : _a.length) {
-            const augmentedChildren = option.children.map(child => augmentOptionToFullOption(state, child));
-            return Object.assign(Object.assign({}, option), { children: augmentedChildren, selected: getParentMultiSelectionState(augmentedChildren), expanded: state.query.search
+            const augmentedChildren = option.children.map(child => augmentOptionToFullOption(state, child, managerOptions));
+            const selectableWhenSearching = (_b = managerOptions === null || managerOptions === void 0 ? void 0 : managerOptions.allowSelectNonLeafDuringSearch) !== null && _b !== void 0 ? _b : false;
+            // When allowSelectNonLeafDuringSearch is enabled and search is active,
+            // calculate selection state from ALL original children, not just the
+            // filtered ones visible in search results.
+            const useOriginalForSelectionState = (managerOptions === null || managerOptions === void 0 ? void 0 : managerOptions.allowSelectNonLeafDuringSearch) &&
+                state.query.search &&
+                state.originalOptionGroups.length > 0;
+            let selected;
+            if (useOriginalForSelectionState) {
+                const originalOption = findOption(getTopLevelOptions(state.originalOptionGroups), option.id);
+                if (originalOption) {
+                    const allLeaves = flattenOptions([originalOption], true);
+                    const selectedCount = allLeaves.filter(leaf => state.selection.selected.has(leaf)).length;
+                    if (selectedCount === 0) {
+                        selected = false;
+                    }
+                    else if (selectedCount === allLeaves.length) {
+                        selected = true;
+                    }
+                    else {
+                        selected = 'indeterminate';
+                    }
+                }
+                else {
+                    selected = getParentMultiSelectionState(augmentedChildren);
+                }
+            }
+            else {
+                selected = getParentMultiSelectionState(augmentedChildren);
+            }
+            return Object.assign(Object.assign({}, option), { children: augmentedChildren, selected, expanded: state.query.search
                     ? true
-                    : state.display.expandedOptions.has(option.id), selectable: !state.query.search, expandable: !state.query.search });
+                    : state.display.expandedOptions.has(option.id), selectable: !state.query.search || selectableWhenSearching, expandable: !state.query.search });
         }
         return Object.assign(Object.assign({}, option), { children: undefined, selected: state.selection.selected.has(option), expanded: false, selectable: true, expandable: false });
     }
@@ -2501,14 +2544,24 @@ var RivSelect;
             }
         }));
     }
-    function toggleSelected(state, id) {
+    function toggleSelected(state, id, managerOptions) {
+        var _a;
         const topLevelOptions = getTopLevelOptions(state.sourceOptionGroups);
         const option = findOption(topLevelOptions, id);
         if (!option)
             return state;
         const newSelection = new OptionSet(state.selection.selected);
-        const newValue = !(augmentOptionToFullOption(state, option).selected === true);
-        const leaves = flattenOptions([option], true);
+        const newValue = !(augmentOptionToFullOption(state, option, managerOptions).selected === true);
+        // When allowSelectNonLeafDuringSearch is enabled and search is active,
+        // use the original (unfiltered) data to get all children, not just
+        // those visible in the filtered search results.
+        const useOriginalData = (managerOptions === null || managerOptions === void 0 ? void 0 : managerOptions.allowSelectNonLeafDuringSearch) &&
+            state.query.search &&
+            state.originalOptionGroups.length > 0;
+        const optionForLeaves = useOriginalData
+            ? (_a = findOption(getTopLevelOptions(state.originalOptionGroups), id)) !== null && _a !== void 0 ? _a : option
+            : option;
+        const leaves = flattenOptions([optionForLeaves], true);
         for (const leaf of leaves) {
             if (newValue) {
                 newSelection.add(leaf);
@@ -2559,8 +2612,12 @@ var RivSelect;
             action.type === 'orderChange' ||
             (action.type === 'openChange' && action.open)) {
             source(state.query, action)
-                .then(data => {
-                internalActions.next({ type: 'loadComplete', data });
+                .then(result => {
+                const data = Array.isArray(result) ? result : result.data;
+                const originalData = Array.isArray(result)
+                    ? undefined
+                    : result.originalData;
+                internalActions.next({ type: 'loadComplete', data, originalData });
             })
                 .catch(error => {
                 // eslint-disable-next-line no-console
@@ -2574,7 +2631,8 @@ var RivSelect;
         }
         return state;
     }
-    function reduce(state, action) {
+    function reduce(state, action, managerOptions) {
+        var _a;
         switch (action.type) {
             case 'load':
                 return state;
@@ -2582,7 +2640,7 @@ var RivSelect;
                 return Object.assign(Object.assign({}, state), { load: {
                         loading: false,
                         error: null,
-                    }, sourceOptionGroups: action.data });
+                    }, sourceOptionGroups: action.data, originalOptionGroups: (_a = action.originalData) !== null && _a !== void 0 ? _a : [] });
             case 'loadError':
                 return Object.assign(Object.assign({}, state), { load: {
                         loading: false,
@@ -2597,7 +2655,7 @@ var RivSelect;
             case 'selectionChange':
                 return Object.assign(Object.assign({}, state), { selection: action.payload });
             case 'toggleOptionSelected':
-                return toggleSelected(state, action.id);
+                return toggleSelected(state, action.id, managerOptions);
             case 'setSelectedOption':
                 return setSelectedOption(state, action.id);
             case 'visibleSelectedChange':