import { defineStore } from 'pinia'
import { ref, computed } from 'vue'
import { login as loginApi, logout as logoutApi } from '@/api/auth'
import type { LoginRequest } from '@/types'

export const useUserStore = defineStore('user', () => {
  const token = ref(localStorage.getItem('token') || '')
  const userInfo = ref<{ username: string; realName: string } | null>(null)
  const roles = ref<string[]>([])

  const hasRole = (role: string) => {
    return roles.value.includes(role)
  }

  const hasAnyRole = (checkRoles: string[]) => {
    return checkRoles.some(role => roles.value.includes(role))
  }

  const isAdmin = computed(() => {
    return roles.value.includes('SYS_ADMIN')
  })

  const login = async (data: LoginRequest) => {
    const res = await loginApi(data)
    const loginData = res.data.data
    if (!loginData?.token) {
      throw new Error('登录失败：未获取到token')
    }
    const newToken = loginData.token
    token.value = newToken
    userInfo.value = {
      username: loginData.username,
      realName: loginData.realName
    }
    roles.value = loginData.roles || []
    localStorage.setItem('token', newToken)
    localStorage.setItem('userInfo', JSON.stringify(userInfo.value))
    localStorage.setItem('roles', JSON.stringify(roles.value))
  }

  const logout = async () => {
    await logoutApi()
    token.value = ''
    userInfo.value = null
    roles.value = []
    localStorage.removeItem('token')
    localStorage.removeItem('userInfo')
    localStorage.removeItem('roles')
  }

  const isLoggedIn = () => {
    const storedToken = localStorage.getItem('token')
    if (!storedToken || storedToken === 'null' || storedToken === 'undefined') {
      return false
    }
    const parts = storedToken.split('.')
    if (parts.length !== 3) {
      return false
    }
    try {
      const payload = JSON.parse(atob(parts[1]))
      if (payload.exp && payload.exp * 1000 < Date.now()) {
        logout()
        return false
      }
    } catch {
      return false
    }
    return true
  }

  const initFromStorage = () => {
    const storedRoles = localStorage.getItem('roles')
    if (storedRoles) {
      try {
        roles.value = JSON.parse(storedRoles)
      } catch {
        roles.value = []
      }
    }
    const storedUserInfo = localStorage.getItem('userInfo')
    if (storedUserInfo) {
      try {
        userInfo.value = JSON.parse(storedUserInfo)
      } catch {
        userInfo.value = null
      }
    }
  }

  initFromStorage()

  return {
    token,
    userInfo,
    roles,
    hasRole,
    hasAnyRole,
    isAdmin,
    login,
    logout,
    isLoggedIn
  }
})