open sublist when adding item on tree

KaplanOpenSource · Nov 19, 2024 · f1e7c61 · f1e7c61
1 parent 1eece60
commit f1e7c61
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/client/src/App/TreeSublist.jsx b/client/src/App/TreeSublist.jsx
@@ -3,6 +3,8 @@ import AddIcon from '@mui/icons-material/Add';
 import { camelCaseToWords, createNewName } from "../Utils/utils";
 import { TreeItem } from "@mui/x-tree-view/TreeItem";
 import { assignUuids } from "../Context/TrackUuidUtils";
+import { ExperimentTreeNodesExpandedContext } from "../Experiment/ExperimentTreeNodesExpandedProvider";
+import { useContext } from "react";
 
 export const TreeSublist = ({
     nameTemplate,
@@ -15,6 +17,10 @@ export const TreeSublist = ({
     children,
     noAddButton,
 }) => {
+    const {
+        addExpandedNode,
+    } = useContext(ExperimentTreeNodesExpandedContext);
+
     const items = data[fieldName] || [];
     const key = parentKey + '_' + fieldName;
 
@@ -54,6 +60,7 @@ export const TreeSublist = ({
                                     const noNameData = newDataCreator ? newDataCreator() : {};
                                     const theData = assignUuids({ name, ...noNameData });
                                     setData({ ...data, [fieldName]: [...items, theData] });
+                                    addExpandedNode(key);
                                 }}
                             >
                                 <AddIcon />