From 728c9d7edbb114117ec787c1c1109a6cbe018033 Mon Sep 17 00:00:00 2001
From: "marcel.kocisek" <marcel.kocisek@lutraconsulting.co.uk>
Date: Mon, 28 Oct 2024 16:07:38 +0100
Subject: [PATCH] Switch workspace with project change

---
 web-app/packages/lib/src/modules/project/store.ts | 9 +++++++++
 web-app/packages/lib/src/modules/user/store.ts    | 2 +-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/web-app/packages/lib/src/modules/project/store.ts b/web-app/packages/lib/src/modules/project/store.ts
index fad35874..1c8c2a41 100644
--- a/web-app/packages/lib/src/modules/project/store.ts
+++ b/web-app/packages/lib/src/modules/project/store.ts
@@ -129,6 +129,7 @@ export const useProjectStore = defineStore('projectModule', {
     },
 
     setProject(payload: { project: ProjectDetail }) {
+      const userStore = useUserStore()
       let enhancedProject: EnhancedProjectDetail = null
       if (payload.project) {
         enhancedProject = {
@@ -147,6 +148,14 @@ export const useProjectStore = defineStore('projectModule', {
         }
       }
       this.project = enhancedProject
+
+      // Update workspace
+      if (enhancedProject) {
+        userStore.setWorkspace({
+          workspaceId: enhancedProject?.workspace_id,
+          skipStorage: false
+        })
+      }
     },
     setProjects(payload: ProjectsPayload) {
       this.projects = payload.projects
diff --git a/web-app/packages/lib/src/modules/user/store.ts b/web-app/packages/lib/src/modules/user/store.ts
index 7a2b9beb..83f2fb9b 100644
--- a/web-app/packages/lib/src/modules/user/store.ts
+++ b/web-app/packages/lib/src/modules/user/store.ts
@@ -135,7 +135,7 @@ export const useUserStore = defineStore('userModule', {
       this.workspaces = payload.workspaces
     },
 
-    setWorkspaceId(payload) {
+    setWorkspaceId(payload: { id: number }) {
       this.workspaceId = payload.id
     },