feat: record tree expend keys

Author: moonrailgun

src/renderer/components/SideTree.tsx

@@ -59,12 +59,13 @@ export const SideTree: React.FC = React.memo(() => {
   const {
     treeData,
     selectedNode,
+    expandedKeys,
+    setExpandedKeys,
     moveTreeNode,
     setSelectedNode,
     addTreeNodeChildren,
     deleteTreeNode,
   } = useTreeStore();
-  const [expandedKeys, setExpandedKeys] = useState<string[]>([]);
 
   return (
     <div>

src/renderer/store/tree.ts

@@ -18,8 +18,10 @@ export type WebsiteTreeNode = {
 interface TreeStoreState {
   selectedNode: WebsiteTreeNode | null;
   treeData: WebsiteTreeNode[];
+  expandedKeys: string[];
   setSelectedNode: (selectedNode: WebsiteTreeNode | null) => void;
   setTreeData: (treeData: WebsiteTreeNode[]) => void;
+  setExpandedKeys: (expandedKeys: string[]) => void;
   addTreeNode: (treeNode: WebsiteTreeNode) => void;
   addTreeNodeChildren: (parentKey: string, treeNode: WebsiteTreeNode) => void;
   moveTreeNode: (
@@ -63,6 +65,7 @@ export const useTreeStore = create<TreeStoreState>()(
     immer((set) => ({
       treeData: defaultTreeData,
       selectedNode: null,
+      expandedKeys: [],
       setSelectedNode: (selectedNode: WebsiteTreeNode | null) => {
         set({
           selectedNode,
@@ -73,6 +76,11 @@ export const useTreeStore = create<TreeStoreState>()(
           treeData,
         });
       },
+      setExpandedKeys: (expandedKeys: string[]) => {
+        set({
+          expandedKeys,
+        });
+      },
       addTreeNode: (treeNode: WebsiteTreeNode) => {
         set((state) => {
           state.treeData.push(treeNode);
@@ -181,7 +189,10 @@ export const useTreeStore = create<TreeStoreState>()(
     })),
     {
       name: 'webbox-tree',
-      partialize: (state) => ({ treeData: state.treeData }),
+      partialize: (state) => ({
+        treeData: state.treeData,
+        expandedKeys: state.expandedKeys,
+      }),
     }
   )
 );