fix(TreeSelect): fix render container method

gravity-ui · Feb 13, 2024 · f0c888c · f0c888c
1 parent 8eacc82
commit f0c888c
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/src/components/TreeSelect/TreeSelect.tsx b/src/components/TreeSelect/TreeSelect.tsx
@@ -56,7 +56,7 @@ export const TreeSelect = React.forwardRef(function TreeSelect<T>(
         onOpenChange,
         renderControl,
         renderItem,
-        renderContainer: RenderContainer = TreeListContainer,
+        renderContainer = TreeListContainer,
         onItemClick,
     } = props;
 
@@ -210,18 +210,18 @@ export const TreeSelect = React.forwardRef(function TreeSelect<T>(
         <SelectControl
             {...controlProps}
             selectedOptionsContent={React.Children.toArray(
-                value.map((id) => {
+                value.map((itemId) => {
                     if ('renderControlContent' in props) {
-                        return props.renderControlContent(listParsedState.itemsById[id]).title;
+                        return props.renderControlContent(listParsedState.itemsById[itemId]).title;
                     }
 
-                    const items = listParsedState.itemsById[id];
+                    const item = listParsedState.itemsById[itemId];
 
-                    if (isKnownStructureGuard(items)) {
-                        return items.title;
+                    if (isKnownStructureGuard(item)) {
+                        return item.title;
                     }
 
-                    return items as string;
+                    return item as string;
                 }),
             ).join(', ')}
             view="normal"
@@ -262,15 +262,15 @@ export const TreeSelect = React.forwardRef(function TreeSelect<T>(
                 id={`tree-select-popup-${treeSelectId}`}
             >
                 {slotBeforeListBody}
-                <RenderContainer
-                    size={size}
-                    containerRef={containerRef}
-                    id={`list-${treeSelectId}`}
-                    {...listParsedState}
-                    {...listState}
-                    renderItem={(id, index, renderContextProps) => {
+                {renderContainer({
+                    size,
+                    containerRef,
+                    id: `list-${treeSelectId}`,
+                    ...listParsedState,
+                    ...listState,
+                    renderItem: (itemId, index, renderContextProps) => {
                         const renderState = getItemRenderState({
-                            id,
+                            id: itemId,
                             size,
                             onItemClick: handleItemClick,
                             ...listParsedState,
@@ -291,7 +291,7 @@ export const TreeSelect = React.forwardRef(function TreeSelect<T>(
                             });
                         }
 
-                        const itemData = listParsedState.itemsById[id];
+                        const itemData = listParsedState.itemsById[itemId];
 
                         return (
                             <TreeSelectItem
@@ -305,8 +305,8 @@ export const TreeSelect = React.forwardRef(function TreeSelect<T>(
                                 {...renderContextProps}
                             />
                         );
-                    }}
-                />
+                    },
+                })}
                 {slotAfterListBody}
             </SelectPopup>
         </Flex>