San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 1 | import { createSlice } from '@reduxjs/toolkit'; |
San3yuan | a2ee30b | 2025-06-05 21:20:17 +0800 | [diff] [blame] | 2 | import { isTokenExpired } from '@/utils/jwt'; |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 3 | |
| 4 | interface UserState { |
San3yuan | 6f2ed69 | 2025-04-16 20:24:49 +0800 | [diff] [blame] | 5 | userId: string; |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 6 | userName: string; |
| 7 | role: string; |
| 8 | isLogin: boolean; |
| 9 | uploadTraffic: number; |
| 10 | downloadTraffic: number; |
| 11 | downloadPoints: number; |
| 12 | avatar: string; |
| 13 | } |
| 14 | |
| 15 | const initialState: UserState = { |
San3yuan | 6f2ed69 | 2025-04-16 20:24:49 +0800 | [diff] [blame] | 16 | userId: '', |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 17 | userName: '', |
| 18 | role: '', |
San3yuan | a2ee30b | 2025-06-05 21:20:17 +0800 | [diff] [blame] | 19 | isLogin: localStorage.getItem('token')&& isTokenExpired(localStorage.getItem('token') as string) ==='0' ? true : false, |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 20 | uploadTraffic: 0, |
| 21 | downloadTraffic: 0, |
| 22 | downloadPoints: 0, |
| 23 | avatar: '', |
| 24 | }; |
| 25 | |
| 26 | |
| 27 | export const userSlice = createSlice({ |
| 28 | name: 'user', |
| 29 | initialState, |
| 30 | reducers: { |
| 31 | login: (state, action) => { |
San3yuan | ff75c54 | 2025-06-06 20:30:52 +0800 | [diff] [blame] | 32 | console.log(action.payload) |
San3yuan | 6f2ed69 | 2025-04-16 20:24:49 +0800 | [diff] [blame] | 33 | localStorage.setItem('token', action.payload.token); |
| 34 | state.isLogin = true; |
| 35 | }, |
| 36 | getUserInfo: (state, action) => { |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 37 | state.userId = action.payload.userId; |
| 38 | state.userName = action.payload.userName; |
| 39 | state.role = action.payload.role; |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 40 | state.uploadTraffic = action.payload.uploadTraffic; |
| 41 | state.downloadTraffic = action.payload.downloadTraffic; |
| 42 | state.downloadPoints = action.payload.downloadPoints; |
| 43 | state.avatar = action.payload.avatar; |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 44 | }, |
| 45 | logout: (state) => { |
San3yuan | 6f2ed69 | 2025-04-16 20:24:49 +0800 | [diff] [blame] | 46 | state.userId = ''; |
San3yuan | 4d0e803 | 2025-04-04 17:21:40 +0800 | [diff] [blame] | 47 | state.userName = ''; |
| 48 | state.role = ''; |
| 49 | state.isLogin = false; |
| 50 | state.uploadTraffic = 0; |
| 51 | state.downloadTraffic = 0; |
| 52 | state.downloadPoints = 0; |
| 53 | state.avatar = ''; |
| 54 | }, |
| 55 | updateTraffic: (state, action) => { |
| 56 | state.uploadTraffic = action.payload.uploadTraffic; |
| 57 | state.downloadTraffic = action.payload.downloadTraffic; |
| 58 | state.downloadPoints = action.payload.downloadPoints; |
| 59 | }, |
| 60 | updateAvatar: (state, action) => { |
| 61 | state.avatar = action.payload.avatar; |
| 62 | } |
| 63 | }, |
| 64 | |
| 65 | }); |
| 66 | |
| 67 | export const { login, logout, updateTraffic } = userSlice.actions; |
| 68 | export default userSlice.reducer; |