diff --git a/src/routing/PrivateRoute.tsx b/src/routing/PrivateRoute.tsx index c21de17..99b11a7 100644 --- a/src/routing/PrivateRoute.tsx +++ b/src/routing/PrivateRoute.tsx @@ -4,12 +4,12 @@ import Layout from 'pages/Layout/Layout'; import { Redirect, Route } from 'react-router-dom'; import { __TOKEN__ } from '../utils/constants'; -import cookies from '../utils/cookies'; +import { getUserStorageItem } from '../utils/storage'; import { CustomRouteProps } from './types'; import urls from './urls'; const PrivateRoute: React.FC = ({ component: Component, ...rest }) => { - const token = cookies.get(__TOKEN__); + const token = getUserStorageItem(__TOKEN__); if (!token) { return } />; diff --git a/src/routing/PublicRoute.tsx b/src/routing/PublicRoute.tsx index 9d9d0e4..5bfe5c5 100644 --- a/src/routing/PublicRoute.tsx +++ b/src/routing/PublicRoute.tsx @@ -4,12 +4,12 @@ import { Redirect, Route } from 'react-router-dom'; import Layout from '../pages/Layout'; import { __TOKEN__ } from '../utils/constants'; -import cookies from '../utils/cookies'; +import { getUserStorageItem } from '../utils/storage'; import { CustomRouteProps } from './types'; import urls from './urls'; const PublicRoute: React.FC = ({ component: Component, ...rest }) => { - const token = cookies.get(__TOKEN__); + const token = getUserStorageItem(__TOKEN__); if (token && rest.path === urls.login()) { return } />;