From 192a125d016983835ffb8216641c94d22308db79 Mon Sep 17 00:00:00 2001
From: rd <1344903914@qq.com>
Date: Fri, 19 Sep 2025 09:40:06 +0800
Subject: [PATCH] =?UTF-8?q?refactor(user):=20=E9=87=8D=E6=9E=84=E7=94=A8?=
 =?UTF-8?q?=E6=88=B7=E4=BF=A1=E6=81=AF=E8=8E=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/App.vue                                        | 12 +++++++++---
 src/api/all/login.ts                               |  2 +-
 src/stores/modules/user/index.ts                   |  4 ++--
 src/utils/user.ts                                  |  2 +-
 src/views/login/components/login-form/index.vue    |  5 +++--
 src/views/login/components/register-form/index.vue |  8 +++++---
 6 files changed, 21 insertions(+), 12 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index bc49fec..5f23ee8 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -8,18 +8,20 @@
 <script setup>
 import { useUserStore } from '@/stores';
 // import { useChatStore } from '@/stores/modules/chat';
-
-import { initApp } from '@/utils/user';
+import { useRoute } from 'vue-router';
+import { handleUserHome, initApp } from '@/utils/user';
 import { useSidebarStore } from '@/stores/modules/side-bar';
+import { useEnterpriseStore } from '@/stores/modules/enterprise';
 
 import { ConfigProvider } from 'ant-design-vue';
 import zhCN from 'ant-design-vue/es/locale/zh_CN';
 import 'dayjs/locale/zh-cn';
 
 const userStore = useUserStore();
-// const route = useRoute();
+const route = useRoute();
 const sidebarStore = useSidebarStore();
 // const chatStore = useChatStore();
+const enterpriseStore = useEnterpriseStore();
 
 const redTheme = {
   token: {
@@ -34,6 +36,10 @@ const init = async () => {
   // 已开通
   if (isLogin) {
     await initApp();
+
+    if (enterpriseStore.isOpenEnterprise && route.name === 'Trial') {
+      handleUserHome();
+    }
   } else {
     sidebarStore.stopUnreadInfoPolling();
   }
diff --git a/src/api/all/login.ts b/src/api/all/login.ts
index 09047ea..da47c22 100644
--- a/src/api/all/login.ts
+++ b/src/api/all/login.ts
@@ -28,7 +28,7 @@ export const fetchLogOut = (params = {}) => {
 };
 
 // 导出一个名为fetchProfileInfo的函数，用于获取用户信息
-export const fetchProfileInfo = (params = {}) => {
+export const fetchUserInfo = (params = {}) => {
   // 使用Http.put方法，向/v1/me接口发送put请求，并将params作为参数传递
   return Http.get('/v1/me', params);
 };
diff --git a/src/stores/modules/user/index.ts b/src/stores/modules/user/index.ts
index 4cd8142..9139731 100644
--- a/src/stores/modules/user/index.ts
+++ b/src/stores/modules/user/index.ts
@@ -1,5 +1,5 @@
 import { defineStore } from 'pinia';
-import { fetchProfileInfo } from '@/api/all/login';
+import { fetchUserInfo } from '@/api/all/login';
 // import { useSidebarStore } from '@/stores/modules/side-bar';
 import { glsWithCatch, rlsWithCatch, slsWithCatch } from '@/utils/stroage';
 import { useEnterpriseStore } from '@/stores/modules/enterprise';
@@ -72,7 +72,7 @@ export const useUserStore = defineStore('user', {
     async getUserInfo() {
       const enterpriseStore = useEnterpriseStore();
 
-      const { code, data } = await fetchProfileInfo();
+      const { code, data } = await fetchUserInfo();
       if (code === 200) {
         this.setUserInfo(data);
 
diff --git a/src/utils/user.ts b/src/utils/user.ts
index 8d3b779..5515ff0 100644
--- a/src/utils/user.ts
+++ b/src/utils/user.ts
@@ -42,9 +42,9 @@ export async function initApp() {
   const enterpriseStore = useEnterpriseStore();
 
   await userStore.getUserInfo(); // 初始化用户信息
+  await getUserEnterpriseInfo(); // 初始化企业信息
 
   if (enterpriseStore.isOpenEnterprise) {
-    await getUserEnterpriseInfo(); // 初始化企业信息
     chatStore.getAgentInfo(); // 初始化智能体信息
 
     sidebarStore.startUnreadInfoPolling(); // 初始化未读信息
diff --git a/src/views/login/components/login-form/index.vue b/src/views/login/components/login-form/index.vue
index 488bdf5..722c5bd 100644
--- a/src/views/login/components/login-form/index.vue
+++ b/src/views/login/components/login-form/index.vue
@@ -105,7 +105,7 @@
 import { Button, Checkbox, Form, FormItem, Input, message, Tabs, Typography } from 'ant-design-vue';
 import PuzzleVerification from '../PuzzleVerification.vue';
 import SelectAccountModal from '../select-account-modal/index.vue';
-import { fetchAuthorizationsCaptcha, fetchLoginCaptCha, fetchProfileInfo, postLoginPassword } from '@/api/all/login';
+import { fetchAuthorizationsCaptcha, fetchLoginCaptCha, fetchUserInfo, postLoginPassword } from '@/api/all/login';
 import { postClearRateLimiter } from '@/api/all/common';
 import { joinEnterpriseByInviteCode } from '@/api/all';
 import { computed, onUnmounted, reactive, ref } from 'vue';
@@ -268,7 +268,7 @@ const onTabChange = () => {
 
 // 获取用户信息
 const getProfileInfo = async () => {
-  const { code, data } = await fetchProfileInfo();
+  const { code, data } = await fetchUserInfo();
   if (code === 200) {
     const enterprises = data['enterprises'];
     mobileNumber.value = data['mobile'];
@@ -283,6 +283,7 @@ const getProfileInfo = async () => {
         selectAccountModalRef.value.open();
       }
     } else {
+      userStore.setUserInfo(data);
       router.push({ name: 'Trial' });
     }
   }
diff --git a/src/views/login/components/register-form/index.vue b/src/views/login/components/register-form/index.vue
index bed7c7a..cd7ba5e 100644
--- a/src/views/login/components/register-form/index.vue
+++ b/src/views/login/components/register-form/index.vue
@@ -92,7 +92,7 @@ import PuzzleVerification from '../PuzzleVerification.vue';
 import SelectAccountModal from '../select-account-modal/index.vue';
 import { postClearRateLimiter } from '@/api/all/common';
 import {
-  fetchProfileInfo,
+  fetchUserInfo,
   postForgetPassword,
   postForgetPasswordCaptcha,
   postRegister,
@@ -295,8 +295,9 @@ const handleVerificationSubmit = async () => {
 
 // 获取用户信息
 const getProfileInfo = async () => {
-  const { code, data } = await fetchProfileInfo();
+  const { code, data } = await fetchUserInfo();
   if (code === 200) {
+
     const enterprises = data['enterprises'];
     mobileNumber.value = data['mobile'];
     accounts.value = enterprises;
@@ -311,6 +312,7 @@ const getProfileInfo = async () => {
         selectAccountModalRef.value.open();
       }
     } else {
+      userStore.setUserInfo(data);
       router.push({ name: 'Trial' });
     }
   }
@@ -348,7 +350,7 @@ const handleSubmit = async () => {
         }, 1500);
         return;
       }
-      ;
+
 
       userStore.setToken(data.access_token);