import { create } from 'zustand';
import type { User } from '../types';
import * as api from '../api';

interface AuthState {
  user: User | null;
  loading: boolean;
  login: (username: string, password: string, captchaToken?: string, rememberMe?: boolean) => Promise<void>;
  logout: () => Promise<void>;
  checkAuth: () => Promise<void>;
  changePassword: (oldPwd: string, newPwd: string) => Promise<void>;
  refreshUser: () => Promise<void>;
  setUserCredits: (credits: number) => void;
}

export const useAuthStore = create<AuthState>((set) => ({
  user: null,
  loading: true,

  login: async (username, password, captchaToken?, rememberMe?) => {
    const user = await api.login(username, password, captchaToken, rememberMe);
    set({ user });
  },

  logout: async () => {
    await api.logout();
    set({ user: null });
  },

   checkAuth: async () => {
    try {
      const token = localStorage.getItem('auth_token');
      if (!token) { set({ user: null, loading: false }); return; }
      const user = await api.getUser();
      set({ user, loading: false });
    } catch (error: any) {
      if (error?.message?.includes('401') || error?.message?.includes('Unauthorized')) {
        localStorage.removeItem('auth_token');
      }
      set({ user: null, loading: false });
    }
  },

  changePassword: async (oldPwd, newPwd) => {
    await api.changePassword(oldPwd, newPwd);
  },

  refreshUser: async () => {
    try {
      const user = await api.getUser();
      set({ user });
    } catch (error) {
      console.error('Failed to refresh user:', error);
    }
  },

  // 用通知接口返回的 credits.balance 直接覆盖当前 user.credits
  // 避免再请求一次 getUser 接口
  setUserCredits: (credits: number) => {
    set((state) => (state.user ? { user: { ...state.user, credits } } : {}));
  },
}));