import { defineStore } from 'pinia';
import { resetRouter, router, routerArrays, storageLocal, store, type userType } from '../utils';
import { getUserinfo, logout, refreshTokenApi, sendLoginEmail, userLogin } from '@/api/v1/system/adminUser';
import { useMultiTagsStoreHook } from '../multiTags';
import { type DataInfo, removeToken, setToken, userKey } from '@/utils/auth';
import { message, storeMessage } from '@/utils/message';
import { createUserRole, getRoleListByUserId } from '@/api/v1/system/role';

export const useUserStore = defineStore({
  id: 'system-user',
  state: (): userType => ({
    // 头像
    avatar: storageLocal().getItem<DataInfo<number>>(userKey)?.avatar ?? '',
    // 用户名
    username: storageLocal().getItem<DataInfo<number>>(userKey)?.username ?? '',
    // 昵称
    nickname: storageLocal().getItem<DataInfo<number>>(userKey)?.nickname ?? '',
    // 页面级别权限
    roles: storageLocal().getItem<DataInfo<number>>(userKey)?.roles ?? [],
    // 按钮级别权限
    permissions: storageLocal().getItem<DataInfo<number>>(userKey)?.permissions ?? [],
    // 是否勾选了登录页的免登录
    isRemembered: true,
    // 登录页的免登录存储几天，默认7天
    readMeDay: 7,
  }),
  actions: {
    /** 登入 */
    async loginByUsername(data: any) {
      data = this.isRemembered ? { ...data, readMeDay: this.readMeDay } : data;
      const result = await userLogin(data);

      if (result.code === 200) {
        setToken(result.data);
        return true;
      }
      return false;
    },

    /** 发送登录邮箱验证码 */
    async sendLoginEmailCode(email: string) {
      const response = await sendLoginEmail({ email });
      if (response.code === 200) {
        message(response.message, { type: 'success' });
        return true;
      }
      message(response.message, { type: 'error' });

      return false;
    },

    /** 前端登出 */
    async logOut() {
      const result = await logout();
      if (result.code == 200) {
        this.username = '';
        this.roles = [];
        this.permissions = [];
        removeToken();
        useMultiTagsStoreHook().handleTags('equal', [...routerArrays]);
        resetRouter();
        message(result.message, { type: 'success' });
      }

      await router.push('/login');
      return true;
    },

    /** 刷新`token` */
    async handRefreshToken(data: any) {
      const result = await refreshTokenApi({ ...data, readMeDay: this.readMeDay });
      if (result.code === 200) {
        setToken(data.data);
        return true;
      }
      message(result.message, { type: 'error' });
      return false;
    },

    /** 获取用户信息 */
    async loadUserinfo() {
      const result = await getUserinfo();
      if (result.code === 200) {
        const data = result.data;
        setToken(data);
        return data;
      }
      return {};
    },

    /** 根据用户id获取角色列表 */
    async loadRoleListByUserId(data: any) {
      const result = await getRoleListByUserId(data);
      if (result.code !== 200) return;
      return result.data;
    },

    /** 为用户分配角色 */
    async addUserRole(data: any) {
      const result = await createUserRole(data);
      return storeMessage(result);
    },
  },
});

export function useUserStoreHook() {
  return useUserStore(store);
}