refactor(auth): 将用户信息构建逻辑提取到单独的可组合函数中

hookehuyr
Commit 28c27d1f39ee29254fa4410b1d079538a70e611a 28c27d1f 1 parent 8e77f3fb
Showing 2 changed files with 68 additions and 109 deletions
src/composables/useUserInfo.js
src/contexts/auth.js
--- a/src/composables/useUserInfo.js
View file @28c27d1
+++ b/src/composables/useUserInfo.js
View file @28c27d1
@@ -10,6 +10,62 @@ import { ref } from 'vue'
 import { getUserInfoAPI } from '@/api/users'
 /**
+ * 构建当前用户对象
+ * @param {Object} data - 包含用户信息的响应数据
+ * @returns {Object|null} 合并后的当前用户对象，或 null 如果无法构建
+ */
+
+export const build_current_user = (data) => {
+  if (!data || typeof data !== 'object') return null
+
+  const merged_user = {
+    ...(data.user && typeof data.user === 'object' ? data.user : {}),
+    ...(data.checkin && typeof data.checkin === 'object' ? data.checkin : {}),
+  }
+
+  const unread_candidates = [
+    data?.unread_msg_count,
+    data?.user?.unread_msg_count,
+    data?.user?.unread,
+    data?.unread_msg,
+    data?.unread,
+  ]
+
+  for (const value of unread_candidates) {
+    const num = Number(value)
+    if (Number.isFinite(num)) {
+      merged_user.unread_msg_count = num
+      break
+    }
+  }
+
+  const teacher_candidates = [
+    data?.is_teacher,
+    data?.user?.is_teacher,
+    data?.user?.teacher,
+  ]
+
+  for (const value of teacher_candidates) {
+    if (value === null || value === undefined) continue
+    if (value === true || value === 'true') {
+      merged_user.is_teacher = 1
+      break
+    }
+    if (value === false || value === 'false') {
+      merged_user.is_teacher = 0
+      break
+    }
+    const num = Number(value)
+    if (Number.isFinite(num)) {
+      merged_user.is_teacher = num ? 1 : 0
+      break
+    }
+  }
+
+  return merged_user
+}
+
+/**
  * @function useUserInfo
  * @description 提供用户信息获取功能
  * @returns {Object} 包含用户信息状态和方法的对象
@@ -21,47 +77,6 @@ export function useUserInfo() {
   const error = ref(null)
   /**
-   * 从数据中提取未读消息数量
-   * @param {Object} data - 包含用户信息的响应数据
-   * @returns {number|null} 未读消息数量，或 null 如果无法提取
-   */
-
-  const get_unread_msg_count = (data) => {
-    const candidates = [
-      data?.unread_msg_count,
-      data?.user?.unread_msg_count,
-      data?.user?.unread,
-      data?.unread_msg,
-      data?.unread,
-    ]
-
-    for (const value of candidates) {
-      const num = Number(value)
-      if (Number.isFinite(num)) return num
-    }
-
-    return null
-  }
-
-  const get_is_teacher = (data) => {
-    const candidates = [
-      data?.is_teacher,
-      data?.user?.is_teacher,
-      data?.user?.teacher,
-    ]
-
-    for (const value of candidates) {
-      if (value === null || value === undefined) continue
-      if (value === true || value === 'true') return 1
-      if (value === false || value === 'false') return 0
-      const num = Number(value)
-      if (Number.isFinite(num)) return num ? 1 : 0
-    }
-
-    return null
-  }
-
-  /**
    * 刷新用户信息
    * @returns {Promise<Object>} 用户信息对象
    * @throws {Error} 获取失败时抛出错误
@@ -75,24 +90,12 @@ export function useUserInfo() {
       if (code === 1) {
         userInfo.value = data
-        // 合并用户和打卡数据
+        const mergedUser = build_current_user(data)
-        const mergedUser = {
-          ...data.user,
-          ...data.checkin
-        }
-
-        const unread = get_unread_msg_count(data)
-        if (unread !== null) {
-          mergedUser.unread_msg_count = unread
-        }
-
-        const is_teacher = get_is_teacher(data)
-        if (is_teacher !== null) {
-          mergedUser.is_teacher = is_teacher
-        }
         // 更新本地存储
-        localStorage.setItem('currentUser', JSON.stringify(mergedUser))
+        if (mergedUser) {
+          localStorage.setItem('currentUser', JSON.stringify(mergedUser))
+        }
         loading.value = false
         return mergedUser
--- a/src/contexts/auth.js
View file @28c27d1
+++ b/src/contexts/auth.js
View file @28c27d1
@@ -12,6 +12,7 @@ import { logoutAPI, getUserInfoAPI } from '@/api/users'
 import { getAuthInfoAPI } from '@/api/auth'
 import { clearAuthHeaders, setAuthHeaders } from '@/utils/axios'
 import { applyUserInfoAuth, getUserInfoFromStorage, removeUserInfoFromStorage } from '@/utils/auth_user_info'
+import { build_current_user } from '@/composables/useUserInfo'
 // 创建认证上下文的Symbol key，用于provide/inject依赖注入
 // 使用Symbol确保key的唯一性，避免命名冲突
@@ -31,41 +32,6 @@ export function provideAuth() {
   // 加载状态标志，用于控制加载动画等UI展示
   const loading = ref(true)
-  const get_unread_msg_count = (data) => {
-    const candidates = [
-      data?.unread_msg_count,
-      data?.user?.unread_msg_count,
-      data?.user?.unread,
-      data?.unread_msg,
-      data?.unread,
-    ]
-
-    for (const value of candidates) {
-      const num = Number(value)
-      if (Number.isFinite(num)) return num
-    }
-
-    return null
-  }
-
-  const get_is_teacher = (data) => {
-    const candidates = [
-      data?.is_teacher,
-      data?.user?.is_teacher,
-      data?.user?.teacher,
-    ]
-
-    for (const value of candidates) {
-      if (value === null || value === undefined) continue
-      if (value === true || value === 'true') return 1
-      if (value === false || value === 'false') return 0
-      const num = Number(value)
-      if (Number.isFinite(num)) return num ? 1 : 0
-    }
-
-    return null
-  }
-
   /**
    * 检查登录token是否过期
    * @returns {boolean} true表示token有效，false表示token已过期
@@ -101,15 +67,10 @@ export function provideAuth() {
       // 从服务器获取用户信息并更新本地存储
       const { code, data } = await getUserInfoAPI();
       if (code === 1) {
-        const unread = get_unread_msg_count(data)
+        currentUser.value = build_current_user(data)
-        const is_teacher = get_is_teacher(data)
+        if (currentUser.value) {
-        currentUser.value = {
+          localStorage.setItem('currentUser', JSON.stringify(currentUser.value))
-          ...data.user,
-          ...data.checkin,
-          ...(unread !== null ? { unread_msg_count: unread } : {}),
-          ...(is_teacher !== null ? { is_teacher } : {}),
         }
-        localStorage.setItem('currentUser', JSON.stringify(currentUser.value))
         // 重新设置认证头
         const userInfo = getUserInfoFromStorage()
         if (userInfo && userInfo.user_id && userInfo.HTTP_USER_TOKEN) {
@@ -138,14 +99,7 @@ export function provideAuth() {
           try {
             const userRes = await getUserInfoAPI();
             if (userRes.code === 1) {
-              const unread = get_unread_msg_count(userRes.data)
+              currentUser.value = build_current_user(userRes.data)
-              const is_teacher = get_is_teacher(userRes.data)
-              currentUser.value = {
-                ...userRes.data.user,
-                ...userRes.data.checkin,
-                ...(unread !== null ? { unread_msg_count: unread } : {}),
-                ...(is_teacher !== null ? { is_teacher } : {}),
-              }
             } else {
               currentUser.value = { ...data.user, ...data.checkin }
             }
@@ -154,7 +108,9 @@ export function provideAuth() {
             currentUser.value = { ...data.user, ...data.checkin }
           }
-          localStorage.setItem('currentUser', JSON.stringify(currentUser.value))
+          if (currentUser.value) {
+            localStorage.setItem('currentUser', JSON.stringify(currentUser.value))
+          }
         }
       }
     }