import { create } from 'zustand'
import Taro from '@tarojs/taro'
import { registerApi, wxLoginApi } from '@/services/auth'
import type { RegisterParams, WxLoginParams } from '@/services/auth'
import type { User } from '@/types/user'

/** 存储 key */
const STORAGE_KEYS = {
  TOKEN: 'auth_token',
  USER: 'auth_user',
} as const

/** 从本地存储恢复登录态 */
function loadFromStorage(): { user: User | null; token: string | null } {
  try {
    const storedToken = Taro.getStorageSync(STORAGE_KEYS.TOKEN)
    const storedUser = Taro.getStorageSync(STORAGE_KEYS.USER)
    if (storedToken && storedUser) {
      return { token: storedToken, user: JSON.parse(storedUser) }
    }
  } catch {
    // 存储数据损坏，清除并视为未登录
    try { Taro.removeStorageSync(STORAGE_KEYS.TOKEN) } catch { /* noop */ }
    try { Taro.removeStorageSync(STORAGE_KEYS.USER) } catch { /* noop */ }
  }
  return { user: null, token: null }
}

/** 登录 / 注册成功后持久化 token 与用户信息 */
function persistAuth(token: string, user: User): void {
  try {
    Taro.setStorageSync(STORAGE_KEYS.TOKEN, token)
    Taro.setStorageSync(STORAGE_KEYS.USER, JSON.stringify(user))
  } catch {
    // storage 写入失败不阻塞登录流程
  }
}

interface AuthState {
  user: User | null
  token: string | null
  loading: boolean
  login: (params: WxLoginParams) => Promise<void>
  /** 微信注册（手机号授权 + 门店编码绑定门店） */
  register: (params: RegisterParams) => Promise<void>
  logout: () => void
  /** 更新用户信息（用于编辑资料后同步 store） */
  setUser: (user: User) => void
}

const useAuthStore = create<AuthState>((set) => {
  // 初始化时从 storage 恢复
  const initial = loadFromStorage()

  return {
    user: initial.user,
    token: initial.token,
    loading: !!(initial.token && initial.user), // 已恢复则立即 ready

    /** 登录（仅已注册用户可登录，未注册由页面引导去注册） */
    login: async (params: WxLoginParams) => {
      const res = await wxLoginApi(params)
      const { token, user } = res.data
      set({ user, token })
      persistAuth(token, user)
    },

    /** 注册：POST /mini/auth/register */
    register: async (params: RegisterParams) => {
      const res = await registerApi(params)
      const { token, user } = res.data
      set({ user, token })
      persistAuth(token, user)
    },

    /** 退出登录 */
    logout: () => {
      set({ user: null, token: null })
      try {
        Taro.removeStorageSync(STORAGE_KEYS.TOKEN)
        Taro.removeStorageSync(STORAGE_KEYS.USER)
      } catch {
        // noop
      }
    },

    /** 更新用户信息（编辑资料后同步 store + storage） */
    setUser: (user: User) => {
      set({ user })
      try {
        Taro.setStorageSync(STORAGE_KEYS.USER, JSON.stringify(user))
      } catch {
        // storage 写入失败不阻塞
      }
    },
  }
})

export default useAuthStore