✨ Fixed Loss of Agent Data upon Login (#991)

reworkd · Jul 8, 2023 · 0b46593 · 0b46593
1 parent 3b27ce0
commit 0b46593
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/next/src/pages/index.tsx b/next/src/pages/index.tsx
@@ -92,6 +92,7 @@ const Home: NextPage = () => {
 
   const handleNewAgent = (name: string, goal: string) => {
     if (session === null) {
+      storeAgentDataInLocalStorage(name, goal);
       setShowSignInDialog(true);
       return;
     }
@@ -121,6 +122,28 @@ const Home: NextPage = () => {
     newAgent?.run().then(console.log).catch(console.error);
   };
 
+  const storeAgentDataInLocalStorage = (name: string, goal: string) => {
+    const agentData = { name, goal };
+    localStorage.setItem("agentData", JSON.stringify(agentData));
+  };
+
+  const getAgentDataFromLocalStorage = () => {
+    const agentData = localStorage.getItem("agentData");
+    return agentData ? JSON.parse(agentData) : null;
+  };
+
+  useEffect(() => {
+    if (session !== null) {
+      const agentData = getAgentDataFromLocalStorage();
+
+      if (agentData) {
+        setNameInput(agentData.name);
+        setGoalInput(agentData.goal);
+        localStorage.removeItem("agentData");
+      }
+    }
+  }, [session]);
+
   const handleRestart = () => {
     resetAllMessageSlices();
     resetAllTaskSlices();