hfb_sys/frontend/src/stores/session.ts

import { defineStore } from 'pinia'

import { fetchMe, loginWithSms, updateMe, type AuthUser } from '@/features/auth/api/auth'
import {
  clearAuthStorage,
  getAccessToken,
  getRefreshToken,
  setAuthTokens,
} from '@/shared/utils/authStorage'
import { loginWithPassword, registerWithPassword } from '@/features/auth/api/auth'

export const useSessionStore = defineStore('session', {
  state: () => ({
    token: getAccessToken('user'),
    refreshToken: getRefreshToken('user'),
    userId: Number(localStorage.getItem('user_id') || 0),
    phone: localStorage.getItem('phone') || '',
    nickname: localStorage.getItem('nickname') || '',
    avatarUrl: localStorage.getItem('avatar_url') || '',
    realnameStatus: localStorage.getItem('realname_status') || 'unknown',
    _loadingMe: false, // 防止重复调用
  }),
  getters: {
    isLoggedIn: state => Boolean(state.token),
    displayName: state =>
      state.nickname || (state.phone ? `用户${state.phone.slice(-6)}` : `用户${state.userId || 0}`),
    avatarText: state => (state.nickname || state.phone || 'U').slice(0, 1),
  },
  actions: {
    async login(phone: string, code: string) {
      const result = await loginWithSms(phone, code)
      this.applySession(result.user, result.tokens.access_token, result.tokens.refresh_token)
      return result
    },
    async loginByPassword(phone: string, password: string) {
      const result = await loginWithPassword(phone, password)
      this.applySession(result.user, result.tokens.access_token, result.tokens.refresh_token)
      return result
    },
    async register(phone: string, code: string, password: string) {
      const result = await registerWithPassword(phone, code, password)
      this.applySession(result.user, result.tokens.access_token, result.tokens.refresh_token)
      return result
    },
    async loadMe() {
      if (this._loadingMe) {
        // 如果正在加载，等待完成
        await new Promise(resolve => {
          const checkInterval = setInterval(() => {
            if (!this._loadingMe) {
              clearInterval(checkInterval)
              resolve(true)
            }
          }, 50)
        })
        return
      }

      this._loadingMe = true
      try {
        const user = await fetchMe()
        this.applyUser(user)
        return user
      } finally {
        this._loadingMe = false
      }
    },
    async updateProfile(payload: { nickname: string; avatar_url: string }) {
      const user = await updateMe(payload)
      this.applyUser(user)
      return user
    },
    logout() {
      this.token = ''
      this.refreshToken = ''
      this.userId = 0
      this.phone = ''
      this.nickname = ''
      this.avatarUrl = ''
      this.realnameStatus = 'unknown'
      clearAuthStorage('user')
    },
    syncFromStorage() {
      this.token = getAccessToken('user')
      this.refreshToken = getRefreshToken('user')
      this.userId = Number(localStorage.getItem('user_id') || 0)
      this.phone = localStorage.getItem('phone') || ''
      this.nickname = localStorage.getItem('nickname') || ''
      this.avatarUrl = localStorage.getItem('avatar_url') || ''
      this.realnameStatus = localStorage.getItem('realname_status') || 'unknown'
    },
    applySession(user: AuthUser, accessToken: string, refreshToken: string) {
      this.token = accessToken
      this.refreshToken = refreshToken
      this.userId = user.id
      setAuthTokens('user', {
        access_token: accessToken,
        refresh_token: refreshToken,
      })
      localStorage.setItem('user_id', String(user.id))
      this.applyUser(user)
    },
    applyUser(user: AuthUser) {
      this.userId = user.id
      localStorage.setItem('user_id', String(user.id))
      this.phone = user.phone
      localStorage.setItem('phone', user.phone)
      this.nickname = user.nickname
      localStorage.setItem('nickname', user.nickname)
      this.avatarUrl = user.avatar_url
      localStorage.setItem('avatar_url', user.avatar_url)
      this.realnameStatus = user.realname_status
      localStorage.setItem('realname_status', user.realname_status)
    },
  },
})