JCSS / jcss-miniprogram

import { defineStore } from 'pinia';
import cache from '@/common/utils/cache';
import globalConfig from '@/common/config/global';
// 新增：导入 refreshToken 接口
import { login, getUserInfo, logout, moduleList, getSimpleDictDataList, refreshToken } from '@/api/user';

// 新增：定义刷新Token的定时器标识（避免重复创建定时器）
let refreshTokenTimer = null;

export const useUserStore = defineStore('user', {
  // 修复1：删除重复的 state 定义，只保留根层级的 state
  state: () => ({
    token: cache.get(globalConfig.cache.tokenKey) || '', // 初始值从缓存取（兼容持久化）
    refreshToken: cache.get(globalConfig.cache.refreshTokenKey) || '', // 新增：Refresh Token 状态
    userInfo: cache.get(globalConfig.cache.userInfoKey) || {},
    userId: cache.get(globalConfig.cache.userIdKey) || '',
    moduleListInfo: cache.get(globalConfig.cache.moduleListKey) || '',
    expireTime: cache.get(globalConfig.cache.expireTimeKey) || 0,
    dictData: cache.get(globalConfig.cache.dictDataKey) || {},
    logoutLoading: false
  }),

  getters: {
    isLogin: (state) => {
      if (!state.token) return false;
      if (state.expireTime && state.expireTime < Date.now()) {
        return false;
      }
      return true;
    },
    permissions: (state) => state.userInfo.permissions || [],
    // 新增：判断是否需要刷新Token（过期前30秒触发，避免已过期才刷新）
    needRefreshToken: (state) => {
      if (!state.token || !state.refreshToken) return false;
      const remainingTime = state.expireTime - Date.now();
      // 剩余时间大于0秒 且 小于30秒（即将过期）
      return remainingTime > 0 && remainingTime < 30 * 1000;
    }
  },

  actions: {
    async login(params) {
      try {
        const res = await login(params);
        // 新增：从登录接口返回值中获取 refreshToken（若接口返回字段名不一致，可调整，如 res.refresh_token）
        const { accessToken, expiresTime, userId, refreshToken } = res;

        if (!accessToken) {
          throw new Error('登录失败，未获取到令牌');
        }

        // 更新 Pinia state
        this.token = accessToken;
        this.refreshToken = refreshToken || ''; // 新增：存储Refresh Token
        this.expireTime = expiresTime;
        this.userId = userId;
        this.userInfo = {};

        // 等待 Pinia 持久化同步
        await new Promise(resolve => setTimeout(resolve, 50));

        // 获取用户信息
        const userInfo = await this.getUserInfo();
        this.userInfo = userInfo;

        // 获取模块列表
        let moduleListInfo = null;
        try {
          moduleListInfo = await this.getModuleList();
          this.moduleListInfo = moduleListInfo;
        } catch (moduleErr) {
          console.warn('获取模块列表失败（不影响登录）：', moduleErr);
          uni.showToast({ title: '获取模块列表失败，可正常登录', icon: 'none' });
        }

        // 修复2：调用字典接口时，不直接抛错（避免阻断登录）
        try {
          const dictRes = await this.getAndSaveDictData();
          // 仅接口返回成功时才更新字典
          this.dictData = dictRes || {};
          console.log('字典数据获取成功：', this.dictData);
        } catch (dictErr) {
          console.warn('获取字典失败（不影响登录）：', dictErr);
          uni.showToast({ title: '获取字典失败，可正常使用', icon: 'none' });
        }

        // 新增：登录成功后，启动 Token 自动刷新定时器
        this.startRefreshTokenTimer();

        return { ...res, userInfo, moduleListInfo };
      } catch (err) {
        console.error('登录流程失败：', err);
        throw err;
      }
    },

    // 新增：调用刷新Token接口，返回刷新是否成功
    async refreshTokenApi() {
      // 前置校验：无Refresh Token直接返回失败
      if (!this.refreshToken) {
        console.warn('无刷新令牌，无法刷新Token');
        this.logout(); // 无刷新令牌，直接退出登录
        return false;
      }

      try {
        // 调用修正后的 refreshToken 接口，传入当前的 refreshToken
        const res = await refreshToken(this.refreshToken);

        // 校验接口返回结果（根据你的实际接口返回格式调整）
        if (!res || !res.accessToken) {
          throw new Error('刷新Token失败，未获取到新令牌');
        }

        // 更新 Token 相关状态
        this.token = res.accessToken; // 新的访问令牌
        this.refreshToken = res.refreshToken || this.refreshToken; // 若接口返回新的Refresh Token则更新
        this.expireTime = res.expiresTime; // 新的过期时间
        console.log('Token 刷新成功');
        return true;
      } catch (err) {
        console.error('刷新Token失败：', err);
        // 刷新失败，直接退出登录
        this.logout();
        return false;
      }
    },

    // 新增：启动 Token 自动刷新定时器
    startRefreshTokenTimer() {
      // 先清除已有定时器，避免重复创建
      if (refreshTokenTimer) {
        clearTimeout(refreshTokenTimer);
        refreshTokenTimer = null;
      }

      // 计算剩余时间（过期前30秒执行刷新）
      const remainingTime = this.expireTime - Date.now() - 30 * 1000;
      if (remainingTime <= 0) {
        // 剩余时间不足，立即执行刷新
        this.refreshTokenApi();
        return;
      }

      // 设置定时器，到期后执行刷新
      refreshTokenTimer = setTimeout(async () => {
        const refreshSuccess = await this.refreshTokenApi();
        // 刷新成功后，重新启动定时器（循环自动刷新）
        if (refreshSuccess) {
          this.startRefreshTokenTimer();
        }
      }, remainingTime);
      console.log(`Token 自动刷新定时器已启动，将在 ${Math.ceil(remainingTime / 1000)} 秒后执行刷新`);
    },

    // 新增：停止 Token 自动刷新定时器（避免内存泄漏）
    stopRefreshTokenTimer() {
      if (refreshTokenTimer) {
        clearTimeout(refreshTokenTimer);
        refreshTokenTimer = null;
      }
    },

    // 修复3：重构字典获取方法（加 Token 校验 + 强制携带 Token + 宽松错误处理）
    async getAndSaveDictData() {
      // 前置校验：无登录态直接返回，不请求接口
      if (!this.isLogin) {
        console.warn('未登录，跳过字典获取');
        return { code: -1, msg: '未登录' };
      }

      try {
        // 强制携带 Token（和 getModuleList 保持一致，避免拦截器同步延迟）
        const res = await getSimpleDictDataList(
          {}, // 接口入参（按需传，比如 dictType: ['level']）
          { header: { 'Authorization': `Bearer ${this.token}` } }
        );

        // 校验接口返回码（核心：避免非 0 码数据存入）
        if (res.code !== 0) {
          console.warn('字典接口返回失败：', res.msg);
          return res; // 返回错误信息，但不抛错
        }
        return res;
      } catch (err) {
        // 修复4：宽松错误处理，只打印日志，不抛错（避免阻断登录）
        console.error('字典接口请求异常：', err);
        return { code: -2, msg: '接口请求异常：' + (err.message || '网络错误') };
      }
    },

    async getModuleList() {
      try {
        if (!this.token) {
          throw new Error('未获取到登录令牌，无法获取模块列表');
        }

        const res = await moduleList({}, {
          header: { 'Authorization': `Bearer ${this.token}` }
        });
        return res;
      } catch (err) {
        console.error('获取用户菜单失败：', err);
        if (err?.data?.code === 401 || err?.message.includes('401')) {
          throw new Error('登录态已过期，请重新登录');
        } else {
          throw new Error('获取用户菜单失败，请重新登录');
        }
      }
    },

    async getUserInfo() {
      try {
        const res = await getUserInfo();
        return res;
      } catch (err) {
        console.error('获取用户信息失败：', err);
        throw new Error('获取用户信息失败，请重新登录');
      }
    },

    logout() {
      // 新增：退出登录时，停止刷新定时器
      this.stopRefreshTokenTimer();

      const pages = getCurrentPages();
      if (pages.length === 0) return;

      const currentPageRoute = pages[pages.length - 1].route;
      const loginPath = globalConfig.router.loginPath
        .replace(/^\//, '')
        .split('?')[0];

      if (currentPageRoute === loginPath) {
        this.token = '';
        this.refreshToken = ''; // 新增：清空Refresh Token
        this.userInfo = {};
        this.userId = '';
        this.moduleListInfo = '';
        this.expireTime = 0;
        this.dictData = {};
        return;
      }

      const logoutWithLock = async () => {
        if (this.logoutLoading) return;
        this.logoutLoading = true;

        try {
          await logout();
        } catch (err) {
          console.error('退出登录接口调用失败：', err);
        } finally {
          this.token = '';
          this.refreshToken = ''; // 新增：清空Refresh Token
          this.userInfo = {};
          this.userId = '';
          this.moduleListInfo = '';
          this.dictData = {};
          this.expireTime = 0;
          this.logoutLoading = false;

          uni.redirectTo({
            url: globalConfig.router.loginPath
          });
        }
      };

      logoutWithLock();
    },

    checkLogin() {
      if (!this.isLogin) {
        const pages = getCurrentPages();
        if (pages.length === 0) return false;

        const currentPageRoute = pages[pages.length - 1].route;
        const loginPath = globalConfig.router.loginPath
          .replace(/^\//, '')
          .split('?')[0];

        if (currentPageRoute !== loginPath) {
          uni.redirectTo({
            url: globalConfig.router.loginPath
          });
        }
        return false;
      }

      // 新增：已登录时，启动刷新定时器（防止页面刷新后定时器丢失）
      this.startRefreshTokenTimer();
      return true;
    }
  },

  persist: {
    enabled: true,
    key: 'user_store',
    storage: {
      getItem: (key) => uni.getStorageSync(key),
      setItem: (key, value) => uni.setStorageSync(key, value),
      removeItem: (key) => uni.removeStorageSync(key)
    },
    serializer: {
      serialize: (state) => {
        uni.setStorageSync(globalConfig.cache.tokenKey, state.token);
        uni.setStorageSync(globalConfig.cache.refreshTokenKey, state.refreshToken); // 新增：持久化Refresh Token
        uni.setStorageSync(globalConfig.cache.userIdKey, state.userId);
        uni.setStorageSync(globalConfig.cache.expireTimeKey, state.expireTime);
        uni.setStorageSync(globalConfig.cache.userInfoKey, state.userInfo);
        uni.setStorageSync(globalConfig.cache.moduleListKey, state.moduleListInfo);
        uni.setStorageSync(globalConfig.cache.dictDataKey, state.dictData);
        return state;
      },
      deserialize: (value) => {
        return {
          token: uni.getStorageSync(globalConfig.cache.tokenKey) || '',
          refreshToken: uni.getStorageSync(globalConfig.cache.refreshTokenKey) || '', // 新增：读取Refresh Token
          userId: uni.getStorageSync(globalConfig.cache.userIdKey) || '',
          expireTime: uni.getStorageSync(globalConfig.cache.expireTimeKey) || 0,
          userInfo: uni.getStorageSync(globalConfig.cache.userInfoKey) || {},
          moduleListInfo: uni.getStorageSync(globalConfig.cache.moduleListKey) || '',
          dictData: uni.getStorageSync(globalConfig.cache.dictDataKey) || {},
          logoutLoading: false
        };
      }
    },
    // paths: []
  }
});