user.ts 637 Bytes
import { useState } from 'react';

// src/models/userModel.ts
export default function Page() {
  const [token, setToken] = useState('');
  const [userInfo, setUserInfo] = useState({});

  const setUserToken = (token: string) => {
    localStorage.setItem('token', token);
    setToken(token);
  };

  const setUserLocalInfo = (token: string, userInfo: any) => {
    localStorage.setItem('token', token);
    setToken(token);

    localStorage.setItem('userInfo', JSON.stringify(userInfo));
    setUserInfo(userInfo);
  };

  return {
    token,
    setToken,
    userInfo,
    setUserInfo,
    setUserToken,
    setUserLocalInfo,
  };
}