style: model list buttons

src/hooks/use-expanded-row-keys.ts

@@ -3,13 +3,16 @@ import { useCallback, useState } from 'react';
 export default function useExpandedRowKeys(defaultKeys: React.Key[] = []) {
   const [expandedRowKeys, setExpandedRowKeys] =
     useState<React.Key[]>(defaultKeys);
+  const [currentExpand, setCurrentExpand] = useState<React.Key | null>(null);
 
   const handleExpandChange = useCallback(
     (expanded: boolean, record: any, rowKey: any) => {
       if (expanded) {
         setExpandedRowKeys((keys) => [...keys, rowKey]);
+        setCurrentExpand(rowKey);
       } else {
         setExpandedRowKeys((keys) => keys.filter((key) => key !== rowKey));
+        setCurrentExpand(null);
       }
     },
     []
@@ -19,6 +22,7 @@ export default function useExpandedRowKeys(defaultKeys: React.Key[] = []) {
     (expanded: boolean, keys: React.Key[] = []) => {
       if (!expanded) {
         setExpandedRowKeys([]);
+        setCurrentExpand(null);
       }
       if (expanded) {
         setExpandedRowKeys((prevKeys) => [...new Set([...prevKeys, ...keys])]);
@@ -41,10 +45,13 @@ export default function useExpandedRowKeys(defaultKeys: React.Key[] = []) {
 
   const clearExpandedRowKeys = () => {
     setExpandedRowKeys([]);
+    setCurrentExpand(null);
   };
 
   return {
     expandedRowKeys,
+    currentExpand,
+    setCurrentExpand,
     clearExpandedRowKeys,
     updateExpandedRowKeys,
     removeExpandedRowKey,