cloud_admin_app/sheep/store/user.js

import {
	defineStore
} from 'pinia';
import $share from '@/sheep/platform/share';
import {
	clone,
	cloneDeep
} from 'lodash-es';
import cart from './cart';
import app from './app';
import {
	showAuthModal
} from '@/sheep/hooks/useModal';
import UserApi from '@/sheep/api/member/user';
import PayWalletApi from '@/sheep/api/pay/wallet';
import OrderApi from '@/sheep/api/trade/order';
import CouponApi from '@/sheep/api/promotion/coupon';

// 默认用户信息
const defaultUserInfo = {
	avatar: '', // 头像
	nickname: '', // 昵称
	gender: 0, // 性别
	mobile: '', // 手机号
	point: 0, // 积分
};

// 默认钱包信息
const defaultUserWallet = {
	balance: 0, // 余额
};

// 默认订单、优惠券等其他资产信息
const defaultNumData = {
	unusedCouponCount: 0,
	orderCount: {
		allCount: 0,
		unpaidCount: 0,
		undeliveredCount: 0,
		deliveredCount: 0,
		uncommentedCount: 0,
		afterSaleCount: 0,
	},
};

const user = defineStore({
	id: 'user',
	state: () => ({
		userInfo: clone(defaultUserInfo), // 用户信息
		userWallet: clone(defaultUserWallet), // 用户钱包信息
		userWalletPersonnelCar: clone(defaultUserWallet), // 车队个人用户钱包信息
		userWalletTeam: clone(defaultUserWallet), // 车队账户钱包信息
		isLogin: !!uni.getStorageSync('token'), // 登录状态
		numData: cloneDeep(defaultNumData), // 用户其他数据
		lastUpdateTime: 0, // 上次更新时间
	}),

	actions: {
		// 获取用户信息
		async getInfo() {
			const {
				code,
				data
			} = await UserApi.getUserInfo();
			if (code !== 0) {
				return;
			}
			this.userInfo = data;
			return Promise.resolve(data);
		},

		// 获得用户钱包
		async getWallet() {
			const {
				code,
				data
			} = await PayWalletApi.getPayWallet();
			if (code !== 0) {
				return;
			}
			this.userWallet = data;
		},
		// 获得个人车队用户钱包
		async getWallet_person_car() {
			const {
				code,
				data
			} = await PayWalletApi.getPayWalletPersonelCar();
			if (code !== 0) {
				return;
			}
			this.userWalletPersonnelCar = data;
		},
		
		// 获得车队钱包
		async getWalletTeam() {
			const {
				code,
				data
			} = await PayWalletApi.getPayWalletPersonelTeam();
			if (code !== 0) {
				return;
			}
			this.userWalletTeam = data;
		},

		// 获取订单、优惠券等其他资产信息
		getNumData() {
			OrderApi.getOrderCount().then((res) => {
				if (res.code === 0) {
					this.numData.orderCount = res.data;
				}
			});
			CouponApi.getUnusedCouponCount().then((res) => {
				if (res.code === 0) {
					this.numData.unusedCouponCount = res.data;
				}
			});
		},

		// 设置 token
		setToken(token = '', refreshToken = '') {
			if (token === '') {
				this.isLogin = false;
				uni.removeStorageSync('token');
				uni.removeStorageSync('refresh-token');
			} else {
				this.isLogin = true;
				uni.setStorageSync('token', token);
				uni.setStorageSync('refresh-token', refreshToken);
				this.loginAfter();
			}
			return this.isLogin;
		},

		// 更新用户相关信息 (手动限流，5 秒之内不刷新)
		async updateUserData() {
			if (!this.isLogin) {
				this.resetUserData();
				return;
			}
			// 防抖，5 秒之内不刷新
			const nowTime = new Date().getTime();
			if (this.lastUpdateTime + 5000 > nowTime) {
				return;
			}
			this.lastUpdateTime = nowTime;

			// 获取最新信息
			await this.getInfo();
			this.getWallet();
			this.getWallet_person_car();
			this.getNumData();
			return this.userInfo;
		},

		// 重置用户默认数据
		resetUserData() {
			// 清空 token
			this.setToken();
			// 清空用户相关的缓存
			this.userInfo = clone(defaultUserInfo);
			this.userWallet = clone(defaultUserWallet);
			this.numData = cloneDeep(defaultNumData);
			// 清空购物车的缓存
			cart().emptyList();
		},

		// 登录后，加载各种信息
		// TODO 芋艿：整理下；
		async loginAfter() {
			await this.updateUserData();

			// 加载购物车
			cart().getList();
			// 登录后设置全局分享参数
			$share.getShareInfo();

			// 提醒绑定手机号
			if (app().platform.bind_mobile && !this.userInfo.mobile) {
				showAuthModal('changeMobile');
			}

			// 绑定推广员
			$share.bindBrokerageUser();
		},

		// 登出系统
		async logout() {
			this.resetUserData();
			return !this.isLogin;
		},
	},
	persist: {
		enabled: true,
		strategies: [{
			key: 'user-store',
		}, ],
	},
});

export default user;