75 lines
2.0 KiB
TypeScript
75 lines
2.0 KiB
TypeScript
import { createPersistStore } from "@/store/createPersistStore";
|
|
|
|
export interface User {
|
|
id: number;
|
|
account: string;
|
|
username: string;
|
|
phone: string;
|
|
avatar: string;
|
|
isAdmin: number;
|
|
companyId: number;
|
|
typeId: number;
|
|
status: number;
|
|
s2_accountId: string;
|
|
createTime: string;
|
|
updateTime: string | null;
|
|
lastLoginIp: string;
|
|
lastLoginTime: number;
|
|
}
|
|
|
|
interface UserState {
|
|
user: User | null;
|
|
token: string | null;
|
|
isLoggedIn: boolean;
|
|
setUser: (user: User) => void;
|
|
setToken: (token: string) => void;
|
|
clearUser: () => void;
|
|
login: (token: string, userInfo: User) => void;
|
|
logout: () => void;
|
|
}
|
|
|
|
export const useUserStore = createPersistStore<UserState>(
|
|
set => ({
|
|
user: null,
|
|
token: null,
|
|
isLoggedIn: false,
|
|
setUser: user => set({ user, isLoggedIn: true }),
|
|
setToken: token => set({ token }),
|
|
clearUser: () => set({ user: null, token: null, isLoggedIn: false }),
|
|
login: (token, userInfo) => {
|
|
// 只将token存储到localStorage
|
|
localStorage.setItem("token", token);
|
|
|
|
// 用户信息存储在状态管理中
|
|
const user: User = {
|
|
id: userInfo.id,
|
|
account: userInfo.account,
|
|
username: userInfo.username,
|
|
phone: userInfo.phone,
|
|
avatar: userInfo.avatar,
|
|
isAdmin: userInfo.isAdmin,
|
|
companyId: userInfo.companyId,
|
|
typeId: userInfo.typeId,
|
|
status: userInfo.status,
|
|
s2_accountId: userInfo.s2_accountId,
|
|
createTime: userInfo.createTime,
|
|
updateTime: userInfo.updateTime,
|
|
lastLoginIp: userInfo.lastLoginIp,
|
|
lastLoginTime: userInfo.lastLoginTime,
|
|
};
|
|
set({ user, token, isLoggedIn: true });
|
|
},
|
|
logout: () => {
|
|
// 清除localStorage中的token
|
|
localStorage.removeItem("token");
|
|
set({ user: null, token: null, isLoggedIn: false });
|
|
},
|
|
}),
|
|
"user-store",
|
|
state => ({
|
|
user: state.user,
|
|
token: state.token,
|
|
isLoggedIn: state.isLoggedIn,
|
|
}),
|
|
);
|