From f40e625ee13af2e95ce12ef48515608067daf2db Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Thu, 19 Jun 2025 21:43:57 +0800
Subject: [PATCH] fix: workspace (#3320)

---
 apps/users/serializers/user.py                | 11 +++++++-
 ui/src/api/type/user.ts                       |  1 +
 .../workspace-dropdown/index.vue              | 27 +++++--------------
 ui/src/stores/modules/user.ts                 | 11 ++++++++
 4 files changed, 29 insertions(+), 21 deletions(-)

diff --git a/apps/users/serializers/user.py b/apps/users/serializers/user.py
index 5490a1fda..79d8c0139 100644
--- a/apps/users/serializers/user.py
+++ b/apps/users/serializers/user.py
@@ -11,7 +11,6 @@ import os
 import random
 import re
 from collections import defaultdict
-from itertools import product
 
 from django.core.cache import cache
 from django.core.mail.backends.smtp import EmailBackend
@@ -71,6 +70,14 @@ def is_workspace_manage(user_id: str, workspace_id: str):
     return QuerySet(User).filter(id=user_id, role=RoleConstants.ADMIN.value.__str__()).exists()
 
 
+def get_workspace_list_by_user(user_id):
+    get_workspace_list = DatabaseModelManage.get_model('get_workspace_list_by_user')
+    license_is_valid = DatabaseModelManage.get_model('license_is_valid') or (lambda: False)
+    if get_workspace_list is not None and license_is_valid():
+        return get_workspace_list(user_id)
+    return [{'id': 'default', 'name': 'default'}]
+
+
 class UserProfileSerializer(serializers.Serializer):
     @staticmethod
     def profile(user: User, auth: Auth):
@@ -80,6 +87,7 @@ class UserProfileSerializer(serializers.Serializer):
         @param auth: 认证对象
         @return:
         """
+        workspace_list = get_workspace_list_by_user(user.id)
         return {
             'id': user.id,
             'username': user.username,
@@ -89,6 +97,7 @@ class UserProfileSerializer(serializers.Serializer):
             'permissions': auth.permission_list,
             'is_edit_password': user.role == RoleConstants.ADMIN.name and user.password == 'd880e722c47a34d8e9fce789fc62389d',
             'language': user.language,
+            'workspace_list': workspace_list
         }
 
 
diff --git a/ui/src/api/type/user.ts b/ui/src/api/type/user.ts
index 0d608c826..cad3e9471 100644
--- a/ui/src/api/type/user.ts
+++ b/ui/src/api/type/user.ts
@@ -27,6 +27,7 @@ interface User {
   IS_XPACK?: boolean
   XPACK_LICENSE_IS_VALID?: boolean
   language?: string
+  workspace_list?: Array<any>
 }
 
 interface LoginRequest {
diff --git a/ui/src/layout/layout-header/workspace-dropdown/index.vue b/ui/src/layout/layout-header/workspace-dropdown/index.vue
index c928615f2..f0bc9de2e 100644
--- a/ui/src/layout/layout-header/workspace-dropdown/index.vue
+++ b/ui/src/layout/layout-header/workspace-dropdown/index.vue
@@ -12,7 +12,7 @@
     <template #dropdown>
       <el-dropdown-menu v-loading="loading">
         <el-dropdown-item
-          v-for="item in workspaceList"
+          v-for="item in user.workspace_list"
           :key="item.id"
           :class="item.id === currentWorkspace?.id ? 'active' : ''"
           @click="changeWorkspace(item)"
@@ -35,33 +35,20 @@
 </template>
 
 <script setup lang="ts">
-import { onBeforeMount, ref } from 'vue'
-import WorkspaceApi from '@/api/workspace/workspace'
+import { computed, ref } from 'vue'
 import type { WorkspaceItem } from '@/api/type/workspace'
 import useStore from '@/stores'
+
 const { user } = useStore()
 const loading = ref(false)
-const workspaceList = ref<WorkspaceItem[]>([])
-const currentWorkspace = ref()
 
-async function getWorkspaceList() {
-  try {
-    const res = await WorkspaceApi.getWorkspaceListByUser(loading)
-    workspaceList.value = res.data
-  } catch (e) {
-    console.error(e)
-  }
-}
-
-onBeforeMount(async () => {
-  await getWorkspaceList()
-  const id = localStorage.getItem('workspace_id') ?? 'default'
-  currentWorkspace.value = workspaceList.value.find((item) => item.id === id)
+const currentWorkspace = computed(() => {
+  return user.workspace_list.find((w) => w.id == user.workspace_id)
 })
 
 function changeWorkspace(item: WorkspaceItem) {
-  if (item.id === currentWorkspace.value.id) return
-  currentWorkspace.value = item
+  if (item.id === user.workspace_id) return
+
   user.setWorkspaceId(item.id || 'default')
   window.location.reload()
 }
diff --git a/ui/src/stores/modules/user.ts b/ui/src/stores/modules/user.ts
index 7d62e022a..1e7937fa0 100644
--- a/ui/src/stores/modules/user.ts
+++ b/ui/src/stores/modules/user.ts
@@ -22,6 +22,7 @@ export interface userStateTypes {
   edition: 'CE' | 'PE' | 'EE'
   themeInfo: any
   workspace_id: string
+  workspace_list: Array<any>
 }
 
 const useUserStore = defineStore('user', {
@@ -33,6 +34,7 @@ const useUserStore = defineStore('user', {
     edition: 'CE',
     themeInfo: null,
     workspace_id: '',
+    workspace_list: [],
   }),
   actions: {
     getLanguage() {
@@ -125,6 +127,15 @@ const useUserStore = defineStore('user', {
     async profile(loading?: Ref<boolean>) {
       return UserApi.getUserProfile(loading).then((ok) => {
         this.userInfo = ok.data
+        const workspace_list =
+          ok.data.workspace_list && ok.data.workspace_list.length > 0
+            ? ok.data.workspace_list
+            : [{ id: 'default', name: 'default' }]
+        const workspace_id = this.getWorkspaceId()
+        if (!workspace_id || !workspace_list.some((w) => w.id == workspace_id)) {
+          this.setWorkspaceId(workspace_list[0].id)
+        }
+        this.workspace_list = workspace_list
         useLocalStorage<string>(localeConfigKey, 'en-US').value =
           ok?.data?.language || this.getLanguage()
         const theme = useThemeStore()