@laser-ui/components 1.6.1 → 2.0.0

package/tree-select/TreeSelect.js

@@ -1,24 +1,24 @@
 import { __rest } from "tslib";
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
-import { useEventCallback, useForkRef, useResize } from '@laser-ui/hooks';
-import { findNested } from '@laser-ui/utils';
+import { useEventCallback, useResize } from '@laser-ui/hooks';
+import { findNested, setRef } from '@laser-ui/utils';
 import CancelFilled from '@material-design-icons/svg/filled/cancel.svg?react';
 import CloseOutlined from '@material-design-icons/svg/outlined/close.svg?react';
 import KeyboardArrowDownOutlined from '@material-design-icons/svg/outlined/keyboard_arrow_down.svg?react';
 import SearchOutlined from '@material-design-icons/svg/outlined/search.svg?react';
 import { isNull, isNumber, isUndefined } from 'lodash';
-import { createElement, forwardRef, useId, useImperativeHandle, useMemo, useRef, useState } from 'react';
+import { useId, useImperativeHandle, useMemo, useRef, useState } from 'react';
 import { TreeSelectSearchPanel } from './internal/TreeSelectSearchPanel';
 import { CLASSES } from './vars';
 import { BaseInput } from '../base-input';
+import { CircularProgress } from '../circular-progress';
 import { Dropdown } from '../dropdown';
-import { useComponentProps, useContainerScrolling, useControlled, useDesign, useFocusVisible, useJSS, useLayout, useMaxIndex, useNamespace, useScopedProps, useStyled, useTranslation, } from '../hooks';
+import { useComponentProps, useContainerScrolling, useControlled, useDesign, useFocusVisible, useLayout, useMaxIndex, useNamespace, useScopedProps, useStyled, useTranslation, } from '../hooks';
 import { Icon } from '../icon';
-import { CircularProgress } from '../internal/circular-progress';
 import { Portal } from '../internal/portal';
-import { Transition } from '../internal/transition';
 import { ROOT_DATA } from '../root/vars';
 import { Tag } from '../tag';
+import { Transition } from '../transition';
 import { TreePanel } from '../tree/internal/TreePanel';
 import { MultipleTreeNode } from '../tree/node/multiple-node';
 import { SingleTreeNode } from '../tree/node/single-node';
@@ -26,18 +26,15 @@ import { getTreeNodeLabel } from '../tree/utils';
 import { TREE_NODE_KEY } from '../tree/vars';
 import { getVerticalSidePosition, isPrintableCharacter, mergeCS } from '../utils';
 import { TTANSITION_DURING_POPUP, WINDOW_SPACE } from '../vars';
-function TreeSelectFC(props, ref) {
-    const _a = useComponentProps('TreeSelect', props), { styleOverrides, styleProvider, formControl, list, model, defaultModel, expands: expandsProp, defaultExpands, visible: visibleProp, defaultVisible, placeholder, multiple = false, searchable = false, searchValue: searchValueProp, defaultSearchValue, onlyLeafSelectable = true, showLine = false, clearable: clearableProp = false, loading = false, size: sizeProp, disabled: disabledProp = false, virtual = false, escClosable = true, customItem, customSelected, customSearch, inputRef: inputRefProp, inputRender, popupRender, onModelChange, onFirstExpand, onExpandsChange, onVisibleChange, onSearch, onClear, afterVisibleChange, onScrollBottom } = _a, restProps = __rest(_a, ["styleOverrides", "styleProvider", "formControl", "list", "model", "defaultModel", "expands", "defaultExpands", "visible", "defaultVisible", "placeholder", "multiple", "searchable", "searchValue", "defaultSearchValue", "onlyLeafSelectable", "showLine", "clearable", "loading", "size", "disabled", "virtual", "escClosable", "customItem", "customSelected", "customSearch", "inputRef", "inputRender", "popupRender", "onModelChange", "onFirstExpand", "onExpandsChange", "onVisibleChange", "onSearch", "onClear", "afterVisibleChange", "onScrollBottom"]);
+export function TreeSelect(props) {
+    const _a = useComponentProps('TreeSelect', props), { ref, styleOverrides, styleProvider, formControl, list, model, defaultModel, expands: expandsProp, defaultExpands, visible: visibleProp, defaultVisible, placeholder, multiple = false, searchable = false, searchValue: searchValueProp, defaultSearchValue, onlyLeafSelectable = true, showLine = false, clearable: clearableProp = false, loading = false, size: sizeProp, disabled: disabledProp = false, virtual = false, escClosable = true, customItem, customSelected, customSearch, inputProps, popupRender, onModelChange, onFirstExpand, onExpandsChange, onVisibleChange, onSearch, onClear, afterVisibleChange, onScrollBottom } = _a, restProps = __rest(_a, ["ref", "styleOverrides", "styleProvider", "formControl", "list", "model", "defaultModel", "expands", "defaultExpands", "visible", "defaultVisible", "placeholder", "multiple", "searchable", "searchValue", "defaultSearchValue", "onlyLeafSelectable", "showLine", "clearable", "loading", "size", "disabled", "virtual", "escClosable", "customItem", "customSelected", "customSearch", "inputProps", "popupRender", "onModelChange", "onFirstExpand", "onExpandsChange", "onVisibleChange", "onSearch", "onClear", "afterVisibleChange", "onScrollBottom"]);
     const namespace = useNamespace();
     const styled = useStyled(CLASSES, {
         'tree-select': styleProvider === null || styleProvider === void 0 ? void 0 : styleProvider['tree-select'],
         tree: styleProvider === null || styleProvider === void 0 ? void 0 : styleProvider.tree,
         'tree-select-popup': styleProvider === null || styleProvider === void 0 ? void 0 : styleProvider['tree-select-popup'],
     }, styleOverrides);
-    const sheet = useJSS();
-    const dataRef = useRef({
-        expands: new Set(),
-    });
+    const expandSaved = useRef(new Set());
     const { t } = useTranslation();
     const uniqueId = useId();
     const listId = `${namespace}-tree-select-list-${uniqueId}`;
@@ -46,7 +43,6 @@ function TreeSelectFC(props, ref) {
     const boxRef = useRef(null);
     const popupRef = useRef(null);
     const inputRef = useRef(null);
-    const combineInputRef = useForkRef(inputRef, inputRefProp);
     const focusRef = useRef(null);
     const [nodes, nodesMap] = useMemo(() => {
         const nodes = list.map((item) => multiple
@@ -149,7 +145,7 @@ function TreeSelectFC(props, ref) {
         return searchList;
     })();
     const isEmpty = hasSearch ? searchList.length === 0 : nodes.length === 0;
-    const [focusVisible, focusVisibleWrapper] = useFocusVisible((code) => code.startsWith('Arrow') || ['Home', 'End', 'Enter', 'Space'].includes(code));
+    const [focusVisible, focusVisibleProps] = useFocusVisible((code) => code.startsWith('Arrow') || ['Home', 'End', 'Enter', 'Space'].includes(code));
     const [_itemFocusedWithoutSearch, setItemFocusedWithoutSearch] = useState();
     const itemFocusedWithoutSearch = (() => {
         let itemFocused;
@@ -179,7 +175,6 @@ function TreeSelectFC(props, ref) {
     })();
     const maxZIndex = useMaxIndex(visible);
     const zIndex = `calc(var(--${namespace}-zindex-fixed) + ${maxZIndex})`;
-    const transformOrigin = useRef();
     const updatePosition = useEventCallback(() => {
         if (visible && boxRef.current && popupRef.current) {
             const boxWidth = boxRef.current.offsetWidth;
@@ -190,17 +185,11 @@ function TreeSelectFC(props, ref) {
                 placement: 'bottom-left',
                 inWindow: WINDOW_SPACE,
             });
-            transformOrigin.current = position.transformOrigin;
-            if (sheet.classes.position) {
-                popupRef.current.classList.toggle(sheet.classes.position, false);
-            }
-            sheet.replaceRule('position', {
-                top: position.top,
-                left: position.left,
-                minWidth: Math.min(boxWidth, maxWidth),
-                maxWidth,
-            });
-            popupRef.current.classList.toggle(sheet.classes.position, true);
+            popupRef.current.style.setProperty(`--popup-down-transform-origin`, position.transformOrigin);
+            popupRef.current.style.top = position.top + 'px';
+            popupRef.current.style.left = position.left + 'px';
+            popupRef.current.style.minWidth = Math.min(boxWidth, maxWidth) + 'px';
+            popupRef.current.style.maxWidth = maxWidth + 'px';
         }
     });
     useContainerScrolling(boxRef, updatePosition, !visible);
@@ -224,8 +213,8 @@ function TreeSelectFC(props, ref) {
                 });
             }
             else {
-                if (!dataRef.current.expands.has(node.id)) {
-                    dataRef.current.expands.add(node.id);
+                if (!expandSaved.current.has(node.id)) {
+                    expandSaved.current.add(node.id);
                     onFirstExpand === null || onFirstExpand === void 0 ? void 0 : onFirstExpand(node.id, node.origin);
                 }
                 changeExpands((draft) => {
@@ -234,141 +223,9 @@ function TreeSelectFC(props, ref) {
             }
         }
     };
-    const scrollCallback = useRef();
+    const scrollCallback = useRef(() => { });
     const inputable = searchable && visible;
     const clearable = clearableProp && hasSelected && !visible && !loading && !disabled;
-    const inputNode = focusVisibleWrapper(createElement(searchable ? BaseInput : 'div', Object.assign(Object.assign(Object.assign(Object.assign({}, mergeCS(styled('tree-select__search'), {
-        style: {
-            opacity: inputable ? undefined : 0,
-            zIndex: inputable ? undefined : -1,
-        },
-    })), formControl === null || formControl === void 0 ? void 0 : formControl.inputAria), { ref: combineInputRef, tabIndex: disabled ? -1 : 0, role: 'combobox', 'aria-haspopup': 'listbox', 'aria-expanded': visible, 'aria-controls': listId, onBlur: () => {
-            changeVisible(false);
-        }, onKeyDown: (e) => {
-            if (e.code === 'Escape') {
-                if (visible && escClosable) {
-                    e.stopPropagation();
-                    e.preventDefault();
-                    changeVisible(false);
-                }
-            }
-            else {
-                const itemFocused = (code) => {
-                    if (focusRef.current) {
-                        const item = focusRef.current(code);
-                        if (item) {
-                            hasSearch ? setItemFocusedWithSearch(item) : setItemFocusedWithoutSearch(item);
-                            if (virtual === false && !code.includes('level')) {
-                                scrollCallback.current = () => {
-                                    const el = document.getElementById(getItemId(hasSearch ? item.value : item.id));
-                                    if (el) {
-                                        focusRef.current(el);
-                                    }
-                                };
-                                if (visible) {
-                                    scrollCallback.current();
-                                    scrollCallback.current = undefined;
-                                }
-                            }
-                        }
-                    }
-                };
-                if (visible) {
-                    switch (e.code) {
-                        case 'ArrowUp': {
-                            e.preventDefault();
-                            itemFocused('prev');
-                            break;
-                        }
-                        case 'ArrowDown': {
-                            e.preventDefault();
-                            itemFocused('next');
-                            break;
-                        }
-                        case 'ArrowLeft': {
-                            if (!hasSearch) {
-                                e.preventDefault();
-                                itemFocused('prev-level');
-                            }
-                            break;
-                        }
-                        case 'ArrowRight': {
-                            if (!hasSearch) {
-                                e.preventDefault();
-                                itemFocused('next-level');
-                            }
-                            break;
-                        }
-                        case 'Home': {
-                            e.preventDefault();
-                            itemFocused('first');
-                            break;
-                        }
-                        case 'End': {
-                            e.preventDefault();
-                            itemFocused('last');
-                            break;
-                        }
-                        default: {
-                            if (e.code === 'Enter' || (e.code === 'Space' && !searchable)) {
-                                e.preventDefault();
-                                if (hasSearch) {
-                                    if (itemFocusedWithSearch) {
-                                        changeSelectedByClickWithSearch(itemFocusedWithSearch);
-                                    }
-                                }
-                                else {
-                                    if (itemFocusedWithoutSearch) {
-                                        changeSelectedByClickWithoutSearch(itemFocusedWithoutSearch);
-                                    }
-                                }
-                            }
-                            break;
-                        }
-                    }
-                }
-                else if (!(searchable && ['Home', 'End', 'Enter', 'Space'].includes(e.code))) {
-                    switch (e.code) {
-                        case 'End':
-                        case 'ArrowUp': {
-                            e.preventDefault();
-                            changeVisible(true);
-                            itemFocused('last');
-                            break;
-                        }
-                        case 'Home':
-                        case 'ArrowDown': {
-                            e.preventDefault();
-                            changeVisible(true);
-                            itemFocused('first');
-                            break;
-                        }
-                        case 'Enter':
-                        case 'Space': {
-                            e.preventDefault();
-                            changeVisible(true);
-                            break;
-                        }
-                        default: {
-                            if (isPrintableCharacter(e.key)) {
-                                changeVisible(true);
-                            }
-                            break;
-                        }
-                    }
-                }
-            }
-        } }), searchable
-        ? {
-            type: 'text',
-            value: searchValue,
-            autoComplete: 'off',
-            disabled,
-            onValueChange: (val) => {
-                changeSearchValue(val);
-            },
-        }
-        : {})));
     const [selectedNode, suffixNode, selectedLabel] = (() => {
         let selectedNode = null;
         let suffixNode = null;
@@ -392,7 +249,7 @@ function TreeSelectFC(props, ref) {
                         draft.splice(draft.findIndex((v) => v === id), 1);
                     });
                     return false;
-                }, children: _jsx(Tag, { tabIndex: -1, size: size, children: selected.size }) }));
+                }, children: (dropdownProps) => (_jsx(Tag, Object.assign({}, dropdownProps, { tabIndex: -1, size: size, children: selected.size }))) }));
             selectedNode = selectedNodes.map(({ value, label, node }) => (_jsxs(Tag, { size: size, children: [label, !((node === null || node === void 0 ? void 0 : node.disabled) || disabled) && (_jsx("div", Object.assign({}, styled('tree-select__close'), { role: "button", "aria-label": t('Close'), onClick: (e) => {
                             e.stopPropagation();
                             const checkeds = node.changeStatus('UNCHECKED', selected);
@@ -418,7 +275,12 @@ function TreeSelectFC(props, ref) {
             }), {
                 className: restProps.className,
                 style: restProps.style,
-            }), designProps, { ref: boxRef, onMouseDown: (e) => {
+            }), designProps, { ref: (instance) => {
+                    boxRef.current = instance;
+                    return () => {
+                        boxRef.current = null;
+                    };
+                }, onMouseDown: (e) => {
                     var _a;
                     (_a = restProps.onMouseDown) === null || _a === void 0 ? void 0 : _a.call(restProps, e);
                     preventBlur(e);
@@ -431,7 +293,158 @@ function TreeSelectFC(props, ref) {
                     (_a = restProps.onClick) === null || _a === void 0 ? void 0 : _a.call(restProps, e);
                     (_b = inputRef.current) === null || _b === void 0 ? void 0 : _b.focus({ preventScroll: true });
                     changeVisible((draft) => (searchable ? true : !draft));
-                }, children: [_jsxs("div", Object.assign({}, styled('tree-select__container'), { title: selectedLabel, children: [inputRender ? inputRender(inputNode) : inputNode, !inputable &&
+                }, children: [_jsxs("div", Object.assign({}, styled('tree-select__container'), { title: selectedLabel, children: [(() => {
+                                const nodeProps = Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({}, inputProps), mergeCS(styled('tree-select__search'), {
+                                    style: {
+                                        opacity: inputable ? undefined : 0,
+                                        zIndex: inputable ? undefined : -1,
+                                    },
+                                })), formControl === null || formControl === void 0 ? void 0 : formControl.inputAria), { ref: (instance) => {
+                                        inputRef.current = instance;
+                                        const ret = setRef(inputProps === null || inputProps === void 0 ? void 0 : inputProps.ref, instance);
+                                        return () => {
+                                            inputRef.current = null;
+                                            ret();
+                                        };
+                                    }, tabIndex: disabled ? -1 : 0, role: 'combobox', 'aria-haspopup': 'listbox', 'aria-expanded': visible, 'aria-controls': listId, onFocus: (e) => {
+                                        var _a;
+                                        (_a = inputProps === null || inputProps === void 0 ? void 0 : inputProps.onFocus) === null || _a === void 0 ? void 0 : _a.call(inputProps, e);
+                                        focusVisibleProps.onFocus(e);
+                                    }, onBlur: (e) => {
+                                        var _a;
+                                        (_a = inputProps === null || inputProps === void 0 ? void 0 : inputProps.onBlur) === null || _a === void 0 ? void 0 : _a.call(inputProps, e);
+                                        focusVisibleProps.onBlur(e);
+                                        changeVisible(false);
+                                    }, onKeyDown: (e) => {
+                                        var _a;
+                                        (_a = inputProps === null || inputProps === void 0 ? void 0 : inputProps.onKeyDown) === null || _a === void 0 ? void 0 : _a.call(inputProps, e);
+                                        focusVisibleProps.onKeyDown(e);
+                                        if (e.code === 'Escape') {
+                                            if (visible && escClosable) {
+                                                e.stopPropagation();
+                                                e.preventDefault();
+                                                changeVisible(false);
+                                            }
+                                        }
+                                        else {
+                                            const itemFocused = (code) => {
+                                                if (focusRef.current) {
+                                                    const item = focusRef.current(code);
+                                                    if (item) {
+                                                        hasSearch ? setItemFocusedWithSearch(item) : setItemFocusedWithoutSearch(item);
+                                                        if (virtual === false && !code.includes('level')) {
+                                                            scrollCallback.current = () => {
+                                                                scrollCallback.current = () => { };
+                                                                const el = document.getElementById(getItemId(hasSearch ? item.value : item.id));
+                                                                if (el) {
+                                                                    focusRef.current(el);
+                                                                }
+                                                            };
+                                                            if (visible) {
+                                                                scrollCallback.current();
+                                                            }
+                                                        }
+                                                    }
+                                                }
+                                            };
+                                            if (visible) {
+                                                switch (e.code) {
+                                                    case 'ArrowUp': {
+                                                        e.preventDefault();
+                                                        itemFocused('prev');
+                                                        break;
+                                                    }
+                                                    case 'ArrowDown': {
+                                                        e.preventDefault();
+                                                        itemFocused('next');
+                                                        break;
+                                                    }
+                                                    case 'ArrowLeft': {
+                                                        if (!hasSearch) {
+                                                            e.preventDefault();
+                                                            itemFocused('prev-level');
+                                                        }
+                                                        break;
+                                                    }
+                                                    case 'ArrowRight': {
+                                                        if (!hasSearch) {
+                                                            e.preventDefault();
+                                                            itemFocused('next-level');
+                                                        }
+                                                        break;
+                                                    }
+                                                    case 'Home': {
+                                                        e.preventDefault();
+                                                        itemFocused('first');
+                                                        break;
+                                                    }
+                                                    case 'End': {
+                                                        e.preventDefault();
+                                                        itemFocused('last');
+                                                        break;
+                                                    }
+                                                    default: {
+                                                        if (e.code === 'Enter' || (e.code === 'Space' && !searchable)) {
+                                                            e.preventDefault();
+                                                            if (hasSearch) {
+                                                                if (itemFocusedWithSearch) {
+                                                                    changeSelectedByClickWithSearch(itemFocusedWithSearch);
+                                                                }
+                                                            }
+                                                            else {
+                                                                if (itemFocusedWithoutSearch) {
+                                                                    changeSelectedByClickWithoutSearch(itemFocusedWithoutSearch);
+                                                                }
+                                                            }
+                                                        }
+                                                        break;
+                                                    }
+                                                }
+                                            }
+                                            else if (!(searchable && ['Home', 'End', 'Enter', 'Space'].includes(e.code))) {
+                                                switch (e.code) {
+                                                    case 'End':
+                                                    case 'ArrowUp': {
+                                                        e.preventDefault();
+                                                        changeVisible(true);
+                                                        itemFocused('last');
+                                                        break;
+                                                    }
+                                                    case 'Home':
+                                                    case 'ArrowDown': {
+                                                        e.preventDefault();
+                                                        changeVisible(true);
+                                                        itemFocused('first');
+                                                        break;
+                                                    }
+                                                    case 'Enter':
+                                                    case 'Space': {
+                                                        e.preventDefault();
+                                                        changeVisible(true);
+                                                        break;
+                                                    }
+                                                    default: {
+                                                        if (isPrintableCharacter(e.key)) {
+                                                            changeVisible(true);
+                                                        }
+                                                        break;
+                                                    }
+                                                }
+                                            }
+                                        }
+                                    } }), searchable
+                                    ? {
+                                        type: 'text',
+                                        value: searchValue,
+                                        autoComplete: 'off',
+                                        disabled,
+                                        onValueChange: (val) => {
+                                            changeSearchValue(val);
+                                        },
+                                    }
+                                    : {});
+                                return searchable ? _jsx(BaseInput, Object.assign({}, nodeProps)) : _jsx("div", Object.assign({}, nodeProps));
+                            })(), !inputable &&
                                 (hasSelected ? (_jsx("div", Object.assign({}, styled('tree-select__content'), { children: selectedNode }))) : placeholder ? (_jsx("div", Object.assign({}, styled('tree-select__placeholder-wrapper'), { children: _jsx("div", Object.assign({}, styled('tree-select__placeholder'), { children: placeholder })) }))) : null)] })), multiple && (_jsx("div", Object.assign({}, styled('tree-select__multiple-count'), { onClick: (e) => {
                             e.stopPropagation();
                         }, children: suffixNode }))), clearable && (_jsx("div", Object.assign({}, styled('tree-select__clear'), { role: "button", "aria-label": t('Clear'), onClick: (e) => {
@@ -453,56 +466,46 @@ function TreeSelectFC(props, ref) {
                         document.body.appendChild(el);
                     }
                     return el;
-                }, children: _jsx(Transition, { enter: visible, during: TTANSITION_DURING_POPUP, afterRender: () => {
-                        var _a;
+                }, children: _jsx(Transition, { enter: visible, name: `${namespace}-popup-down`, duration: TTANSITION_DURING_POPUP, onSkipEnter: () => {
+                        updatePosition();
+                        scrollCallback.current();
+                    }, onBeforeEnter: () => {
                         updatePosition();
-                        (_a = scrollCallback.current) === null || _a === void 0 ? void 0 : _a.call(scrollCallback);
-                        scrollCallback.current = undefined;
-                    }, afterEnter: () => {
+                        scrollCallback.current();
+                    }, onAfterEnter: () => {
                         afterVisibleChange === null || afterVisibleChange === void 0 ? void 0 : afterVisibleChange(true);
-                    }, afterLeave: () => {
+                    }, onAfterLeave: () => {
                         afterVisibleChange === null || afterVisibleChange === void 0 ? void 0 : afterVisibleChange(false);
-                    }, children: (state) => {
-                        let transitionStyle = {};
-                        switch (state) {
-                            case 'enter':
-                                transitionStyle = { transform: 'scaleY(0.7)', opacity: 0 };
-                                break;
-                            case 'entering':
-                                transitionStyle = {
-                                    transition: ['transform', 'opacity'].map((attr) => `${attr} ${TTANSITION_DURING_POPUP}ms ease-out`).join(', '),
-                                    transformOrigin: transformOrigin.current,
-                                };
-                                break;
-                            case 'leaving':
-                                transitionStyle = {
-                                    transform: 'scaleY(0.7)',
-                                    opacity: 0,
-                                    transition: ['transform', 'opacity'].map((attr) => `${attr} ${TTANSITION_DURING_POPUP}ms ease-in`).join(', '),
-                                    transformOrigin: transformOrigin.current,
-                                };
-                                break;
-                            case 'leaved':
-                                transitionStyle = { display: 'none' };
-                                break;
-                            default:
-                                break;
-                        }
-                        return (_jsx("div", Object.assign({}, mergeCS(styled('tree-select-popup'), {
-                            style: Object.assign({ zIndex }, transitionStyle),
-                        }), { ref: popupRef, onMouseDown: (e) => {
-                                preventBlur(e);
-                            }, onMouseUp: (e) => {
-                                preventBlur(e);
-                            }, children: (() => {
-                                const el = (_jsxs("div", Object.assign({}, styled('tree-select-popup__content'), { children: [loading && (_jsx("div", Object.assign({}, styled('tree-select-popup__loading', {
-                                            'tree-select-popup__loading--empty': isEmpty,
-                                        }), { children: _jsx(Icon, { children: _jsx(CircularProgress, {}) }) }))), loading && isEmpty ? null : hasSearch ? (_jsx(TreeSelectSearchPanel, { ref: focusRef, namespace: namespace, styled: styled, id: listId, list: searchList, customItem: customItem, itemId: getItemId, itemFocused: itemFocusedWithSearch, multiple: multiple, onlyLeafSelectable: onlyLeafSelectable, virtual: virtual, focusVisible: focusVisible, onClick: (item) => {
-                                                setItemFocusedWithSearch(item);
-                                                changeSelectedByClickWithSearch(item);
-                                            } })) : (_jsx(TreePanel, { style: { maxHeight: 264, padding: '4px 0' }, ref: focusRef, id: listId, namespace: namespace, styled: styled, list: nodes, itemId: getItemId, itemSelected: !multiple && hasSelected ? nodesMap.get(selected) : undefined, itemFocused: itemFocusedWithoutSearch, expands: expands, customItem: customItem, showLine: showLine, multiple: multiple, onlyLeafSelectable: onlyLeafSelectable, disabled: false, virtual: virtual === false ? undefined : { listSize: 264, listPadding: 4, itemSize: isNumber(virtual) ? virtual : 32 }, focusVisible: focusVisible, onNodeFocus: setItemFocusedWithoutSearch, onNodeExpand: handleExpand, onNodeClick: changeSelectedByClickWithoutSearch, onScrollBottom: onScrollBottom }))] })));
-                                return popupRender ? popupRender(el) : el;
-                            })() })));
-                    } }) })] }));
+                    }, children: (transitionRef, leaved) => (_jsx("div", Object.assign({}, mergeCS(styled('tree-select-popup'), {
+                        style: Object.assign({ zIndex }, (leaved ? { display: 'none' } : undefined)),
+                    }), { ref: (instance) => {
+                            popupRef.current = instance;
+                            transitionRef(instance);
+                            return () => {
+                                popupRef.current = null;
+                                transitionRef(null);
+                            };
+                        }, onMouseDown: (e) => {
+                            preventBlur(e);
+                        }, onMouseUp: (e) => {
+                            preventBlur(e);
+                        }, children: (() => {
+                            const el = (_jsxs("div", Object.assign({}, styled('tree-select-popup__content'), { children: [loading && (_jsx("div", Object.assign({}, styled('tree-select-popup__loading', {
+                                        'tree-select-popup__loading--empty': isEmpty,
+                                    }), { children: _jsx(Icon, { children: _jsx(CircularProgress, {}) }) }))), loading && isEmpty ? null : hasSearch ? (_jsx(TreeSelectSearchPanel, { ref: (instance) => {
+                                            focusRef.current = instance;
+                                            return () => {
+                                                focusRef.current = null;
+                                            };
+                                        }, namespace: namespace, styled: styled, id: listId, list: searchList, customItem: customItem, itemId: getItemId, itemFocused: itemFocusedWithSearch, multiple: multiple, onlyLeafSelectable: onlyLeafSelectable, virtual: virtual, focusVisible: focusVisible, onClick: (item) => {
+                                            setItemFocusedWithSearch(item);
+                                            changeSelectedByClickWithSearch(item);
+                                        } })) : (_jsx(TreePanel, { style: { maxHeight: 264, padding: '4px 0' }, ref: (instance) => {
+                                            focusRef.current = instance;
+                                            return () => {
+                                                focusRef.current = null;
+                                            };
+                                        }, id: listId, namespace: namespace, styled: styled, list: nodes, itemId: getItemId, itemSelected: !multiple && hasSelected ? nodesMap.get(selected) : undefined, itemFocused: itemFocusedWithoutSearch, expands: expands, customItem: customItem, showLine: showLine, multiple: multiple, onlyLeafSelectable: onlyLeafSelectable, disabled: false, virtual: virtual === false ? undefined : { listSize: 264, listPadding: 4, itemSize: isNumber(virtual) ? virtual : 32 }, focusVisible: focusVisible, onNodeFocus: setItemFocusedWithoutSearch, onNodeExpand: handleExpand, onNodeClick: changeSelectedByClickWithoutSearch, onScrollBottom: onScrollBottom }))] })));
+                            return popupRender ? popupRender(el) : el;
+                        })() }))) }) })] }));
 }
-export const TreeSelect = forwardRef(TreeSelectFC);

package/tree-select/internal/TreeSelectSearchPanel.d.ts

@@ -4,6 +4,7 @@ import type { Styled } from '../../hooks/useStyled';
 import type { TreeItem } from '../../tree/types';
 import type { CLASSES } from '../vars';
 interface TreeSelectSearchPanelProps<V extends React.Key, T extends TreeItem<V>> {
+    ref?: React.Ref<(code: any) => TreeSelectSearchPanelItem<V, T> | undefined>;
     namespace: string;
     styled: Styled<typeof CLASSES>;
     id: string;
@@ -17,6 +18,5 @@ interface TreeSelectSearchPanelProps<V extends React.Key, T extends TreeItem<V>>
     focusVisible: boolean;
     onClick: (item: TreeSelectSearchPanelItem<V, T>) => void;
 }
-declare function TreeSelectSearchPanelFC<V extends React.Key, T extends TreeItem<V>>(props: TreeSelectSearchPanelProps<V, T>, ref: React.ForwardedRef<any>): JSX.Element | null;
-export declare const TreeSelectSearchPanel: <V extends React.Key, T extends TreeItem<V>>(props: TreeSelectSearchPanelProps<V, T> & React.RefAttributes<any>) => ReturnType<typeof TreeSelectSearchPanelFC>;
+export declare function TreeSelectSearchPanel<V extends React.Key, T extends TreeItem<V>>(props: TreeSelectSearchPanelProps<V, T>): React.ReactElement | null;
 export {};

package/tree-select/internal/TreeSelectSearchPanel.js

@@ -3,14 +3,14 @@ import { jsx as _jsx } from "react/jsx-runtime";
 import { useEventCallback } from '@laser-ui/hooks';
 import { scrollIntoViewIfNeeded } from '@laser-ui/utils';
 import { isNumber, isUndefined } from 'lodash';
-import { forwardRef, useImperativeHandle, useMemo, useRef } from 'react';
+import { useImperativeHandle, useMemo, useRef } from 'react';
 import { Checkbox } from '../../checkbox';
 import { Empty } from '../../empty';
 import { getTreeNodeLabel } from '../../tree/utils';
 import { TREE_NODE_KEY } from '../../tree/vars';
 import { VirtualScroll } from '../../virtual-scroll';
-function TreeSelectSearchPanelFC(props, ref) {
-    const { namespace, styled, id, list, customItem, itemId, itemFocused, multiple, onlyLeafSelectable, virtual, focusVisible, onClick } = props;
+export function TreeSelectSearchPanel(props) {
+    const { ref, namespace, styled, id, list, customItem, itemId, itemFocused, multiple, onlyLeafSelectable, virtual, focusVisible, onClick, } = props;
     const listRef = useRef(null);
     const vsRef = useRef(null);
     const handleKeyDown = useEventCallback((code) => {
@@ -46,7 +46,12 @@ function TreeSelectSearchPanelFC(props, ref) {
         itemSize: isNumber(virtual) ? virtual : 32,
         itemFocusable: (item) => item[TREE_NODE_KEY].enabled,
     }), [list, virtual]);
-    return (_jsx(VirtualScroll, Object.assign({}, vsProps, { ref: vsRef, enable: virtual !== false, listSize: 264, listPadding: 4, itemRender: (item, index, props) => {
+    return (_jsx(VirtualScroll, Object.assign({}, vsProps, { ref: (instance) => {
+            vsRef.current = instance;
+            return () => {
+                vsRef.current = null;
+            };
+        }, enable: virtual !== false, listSize: 264, listPadding: 4, itemRender: (item, index, props) => {
             const node = item[TREE_NODE_KEY];
             let inSelected = node.checked;
             if (!onlyLeafSelectable) {
@@ -68,6 +73,10 @@ function TreeSelectSearchPanelFC(props, ref) {
                 focusVisible && (itemFocused === null || itemFocused === void 0 ? void 0 : itemFocused.value) === item.value && _jsx("div", { className: `${namespace}-focus-outline` }),
                 multiple && (_jsx("div", Object.assign({}, styled('tree__option-checkbox'), { children: _jsx(Checkbox, { model: node.checked, disabled: node.disabled }) }))),
                 _jsx("div", Object.assign({}, styled('tree__option-content'), { children: customItem ? customItem(node.origin) : getTreeNodeLabel(node) }))));
-        }, itemFocused: itemFocused === null || itemFocused === void 0 ? void 0 : itemFocused.value, placeholder: "li", children: (vsList, onScroll) => (_jsx("ul", Object.assign({}, styled('tree'), { ref: listRef, id: id, tabIndex: -1, role: "listbox", "aria-multiselectable": multiple, "aria-activedescendant": isUndefined(itemFocused) ? undefined : itemId(itemFocused.value), onScroll: onScroll, children: list.length === 0 ? _jsx(Empty, { style: { padding: '12px 28px' }, image: Empty.SIMPLE_IMG }) : vsList }))) })));
+        }, itemFocused: itemFocused === null || itemFocused === void 0 ? void 0 : itemFocused.value, placeholder: "li", children: (vsList, onScroll) => (_jsx("ul", Object.assign({}, styled('tree'), { ref: (instance) => {
+                listRef.current = instance;
+                return () => {
+                    listRef.current = null;
+                };
+            }, id: id, tabIndex: -1, role: "listbox", "aria-multiselectable": multiple, "aria-activedescendant": isUndefined(itemFocused) ? undefined : itemId(itemFocused.value), onScroll: onScroll, children: list.length === 0 ? _jsx(Empty, { style: { padding: '12px 28px' }, image: Empty.SIMPLE_IMG }) : vsList }))) })));
 }
-export const TreeSelectSearchPanel = forwardRef(TreeSelectSearchPanelFC);

package/tree-select/types.d.ts

@@ -2,12 +2,13 @@
 import type { CLASSES } from './vars';
 import type { FormControlProvider } from '../form/types';
 import type { TreeItem } from '../tree/types';
-import type { BaseProps, CloneHTMLElement, Size } from '../types';
+import type { BaseProps, Size } from '../types';
 export {};
 export interface TreeSelectRef {
     updatePosition: () => void;
 }
 export interface TreeSelectProps<V extends React.Key, T extends TreeItem<V>> extends BaseProps<'tree-select' | 'tree' | 'tree-select-popup', typeof CLASSES>, Omit<React.HTMLAttributes<HTMLDivElement>, 'children'> {
+    ref?: React.Ref<TreeSelectRef>;
     formControl?: FormControlProvider;
     list: T[];
     model?: V | null | V[];
@@ -35,8 +36,7 @@ export interface TreeSelectProps<V extends React.Key, T extends TreeItem<V>> ext
         filter?: (value: string, item: T) => boolean;
         sort?: (a: T, b: T) => number;
     };
-    inputRef?: React.ForwardedRef<HTMLInputElement>;
-    inputRender?: CloneHTMLElement;
+    inputProps?: React.ComponentPropsWithRef<'input'>;
     popupRender?: (el: React.ReactElement) => React.ReactNode;
     onModelChange?: (value: any, origin: any) => void;
     onFirstExpand?: (value: V, origin: T) => void;