From 1534a47adc730def470515999ae45ca37845f508 Mon Sep 17 00:00:00 2001
From: Vladyslav Matsiiako <matsiiako@gmail.com>
Date: Sun, 15 Jan 2023 10:45:41 -0800
Subject: [PATCH] Fixed the redirectbug with adding a new workspace

---
 frontend/components/basic/Layout.tsx | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/frontend/components/basic/Layout.tsx b/frontend/components/basic/Layout.tsx
index de58a90fa..7246836bc 100644
--- a/frontend/components/basic/Layout.tsx
+++ b/frontend/components/basic/Layout.tsx
@@ -43,8 +43,7 @@ const crypto = require("crypto");
 
 export default function Layout({ children }: LayoutProps) {
   const router = useRouter();
-  const [workspaceList, setWorkspaceList] = useState([]);
-  const [workspaceMapping, setWorkspaceMapping] = useState([{ "1": "2" }]);
+  const [workspaceMapping, setWorkspaceMapping] = useState<Map<string, string>[]>([]);
   const [workspaceSelected, setWorkspaceSelected] = useState("∞");
   const [newWorkspaceName, setNewWorkspaceName] = useState("");
   const [isOpen, setIsOpen] = useState(false);
@@ -140,7 +139,11 @@ export default function Layout({ children }: LayoutProps) {
             }
           });
         }
-        router.push("/dashboard/" + newWorkspaceId);
+        setWorkspaceMapping((prevState) => ({
+          ...prevState,
+          [workspaceName]: newWorkspaceId
+        }))
+        setWorkspaceSelected(workspaceName);
         setIsOpen(false);
         setNewWorkspaceName("");
       } else {
@@ -227,9 +230,7 @@ export default function Layout({ children }: LayoutProps) {
         ) {
           router.push("/dashboard/" + userWorkspaces[0]._id);
         } else {
-          setWorkspaceList(
-            userWorkspaces.map((workspace: any) => workspace.name)
-          );
+          console.log(99, Object.keys(workspaceMapping))
           setWorkspaceMapping(
             Object.fromEntries(
               userWorkspaces.map((workspace: any) => [
@@ -297,11 +298,11 @@ export default function Layout({ children }: LayoutProps) {
                   <div className="text-gray-400 self-start ml-1 mb-1 text-xs font-semibold tracking-wide">
                     {t("nav:menu.project")}
                   </div>
-                  {workspaceList.length > 0 ? (
+                  {Object.keys(workspaceMapping).length > 0 ? (
                     <Listbox
                       selected={workspaceSelected}
                       onChange={setWorkspaceSelected as any}
-                      data={workspaceList}
+                      data={Object.keys(workspaceMapping)}
                       buttonAction={openModal}
                       text=""
                       // workspaceMapping={workspaceMapping as any}
@@ -317,7 +318,7 @@ export default function Layout({ children }: LayoutProps) {
                   )}
                 </div>
                 <ul>
-                  {workspaceList.length > 0 &&
+                  {Object.keys(workspaceMapping).length > 0 &&
                     menuItems.map(({ href, title, emoji }) => (
                       <li className="mt-0.5 mx-2" key={title}>
                         {router.asPath.split("/")[1] === href.split("/")[1] &&