adding cookie in access token (#1301)

TransformerOptimus · Oct 3, 2023 · 9720c65 · 9720c65
1 parent c46765d
commit 9720c65
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/gui/package-lock.json b/gui/package-lock.json
diff --git a/gui/package.json b/gui/package.json
@@ -18,6 +18,7 @@
     "echarts-for-react": "^3.0.2",
     "eslint": "8.40.0",
     "eslint-config-next": "13.4.2",
+    "js-cookie": "^3.0.5",
     "jszip": "^3.10.1",
     "mitt": "^3.0.0",
     "mixpanel-browser": "^2.47.0",

diff --git a/gui/pages/Dashboard/TopBar.js b/gui/pages/Dashboard/TopBar.js
@@ -5,6 +5,8 @@ import {useRouter} from 'next/router';
 import {ToastContainer, toast} from 'react-toastify';
 import 'react-toastify/dist/ReactToastify.css';
 import {refreshUrl, openNewTab, getUserClick} from "@/utils/utils";
+import Cookies from 'js-cookie';
+
 
 export default function TopBar({selectedProject, userName, env}) {
   const [dropdown, setDropdown] = useState(false);
@@ -18,6 +20,7 @@ export default function TopBar({selectedProject, userName, env}) {
     }
     getUserClick('Logged Out',{})
     localStorage.removeItem('accessToken');
+    Cookies.remove('accessToken');
     refreshUrl();
     router.reload();
   };

diff --git a/gui/pages/_app.js b/gui/pages/_app.js
@@ -26,6 +26,7 @@ import {refreshUrl, loadingTextEffect, getUTMParametersFromURL, setLocalStorageV
 import MarketplacePublic from "./Content/Marketplace/MarketplacePublic"
 import {toast} from "react-toastify";
 import mixpanel from 'mixpanel-browser';
+import Cookies from 'js-cookie';
 
 
 export default function App() {
@@ -61,6 +62,7 @@ export default function App() {
       });
   }
 
+
   const installFromMarketplace = () => {
     const toolkitName = localStorage.getItem('toolkit_to_install') || null;
     const agentTemplateId = localStorage.getItem('agent_to_install') || null;
@@ -152,6 +154,7 @@ export default function App() {
               if(signupSource) {
                 handleSignUpSource(signupSource)
               }
+              Cookies.set('accessToken', access_token, { domain: '.superagi.com', path: '/' });
               fetchOrganisation(response.data.id);
             })
             .catch((error) => {