junma-show/src/store/user.ts

import { defineStore } from "pinia";
import { ref } from "vue";
import {
  login as apiLogin,
  register as apiRegister,
  profile,
  newGuest,
} from "@/services/api";
import { useStorage } from "@vueuse/core";
import { VipLevel } from "@/types/vip";

// 主页浏览状态接口定义
interface HomePageState {
  selectedMenu: string;
  currentPage: number;
  selectedSort: string;
  scrollPosition: number;
  isSearchMode: boolean;
  currentSearchKeyword: string;
  videoList?: any[];
  totalPages?: number;
  totalCount?: number;
}

export const useUserStore = defineStore("user", () => {
  const token = useStorage("token", "");
  const userInfo = ref<any>({});
  const userManuallyLoggedOut = useStorage("userManuallyLoggedOut", false);

  // 添加主页浏览状态存储
  const homePageState = ref<HomePageState>({
    selectedMenu: "",
    currentPage: 1,
    selectedSort: "time",
    scrollPosition: 0,
    isSearchMode: false,
    currentSearchKeyword: "",
  });

  const setToken = (newToken: string) => {
    token.value = newToken;
  };

  const setUserInfo = (info: any) => {
    userInfo.value = info;
  };

  // 保存主页浏览状态
  const saveHomePageState = (state: Partial<HomePageState>) => {
    homePageState.value = { ...homePageState.value, ...state };
  };

  // 获取主页浏览状态
  const getHomePageState = (): HomePageState => {
    return homePageState.value;
  };

  // 获取用户VIP等级
  const getVipLevel = (): VipLevel => {
    return userInfo.value?.vipLevel || VipLevel.GUEST;
  };

  const login = async (username: string, password: string) => {
    const response = await apiLogin(username, password);
    setToken(response.token);
    setUserInfo(response.user);
    userManuallyLoggedOut.value = false;
    return response;
  };

  const register = async (
    name: string,
    password: string,
    email?: string,
    phone?: string,
    code?: string
  ) => {
    const response = await apiRegister(name, password, email, phone, code);
    setToken(response.token);
    setUserInfo(response.user);
    userManuallyLoggedOut.value = false;
    return response;
  };

  const sync = async () => {
    const response = await profile();
    setUserInfo(response);
  };

  const logout = () => {
    token.value = "";
    userInfo.value = {};
    userManuallyLoggedOut.value = true;
  };

  const createGuest = async (code?: string, ref?: string) => {
    try {
      const response = await newGuest(code, ref);
      setToken(response.token);
      setUserInfo(response.user);
      userManuallyLoggedOut.value = false;
      return response;
    } catch (error) {
      console.error("创建游客账号失败", error);
      return null;
    }
  };

  return {
    token,
    userInfo,
    userManuallyLoggedOut,
    setUserInfo,
    getVipLevel,
    login,
    register,
    logout,
    sync,
    createGuest,
    homePageState,
    saveHomePageState,
    getHomePageState,
  };
});