User Context (#51)
* feat(frontend): user Context / useUser hook Adds a UserContext to wrap the app and load/cache the user when the website renders. Also adds the useUser hook to pull in user data anywhere its needed on the site. This commit also adds redirection to the login page for users who are not signed in * fix(frontend): use process.env.PORT for user request on server side (defaults to 3000) * docs(frontend): added documentation/notes for how the user context/login workspull/52/head
sct
4 years ago
committed by
GitHub
parent
350af66129
commit
190a8831c7
@ -0,0 +1,28 @@
|
||||
import React, { useEffect } from 'react';
|
||||
import { User, useUser } from '../hooks/useUser';
|
||||
import { useRouter } from 'next/dist/client/router';
|
||||
|
||||
interface UserContextProps {
|
||||
initialUser: User;
|
||||
}
|
||||
|
||||
/**
|
||||
* This UserContext serves the purpose of just preparing the useUser hooks
|
||||
* cache on server side render. It also will handle redirecting the user to
|
||||
* the login page if their session ever becomes invalid.
|
||||
*/
|
||||
export const UserContext: React.FC<UserContextProps> = ({
|
||||
initialUser,
|
||||
children,
|
||||
}) => {
|
||||
const { user } = useUser({ initialData: initialUser });
|
||||
const router = useRouter();
|
||||
|
||||
useEffect(() => {
|
||||
if (!router.pathname.match(/(setup|login)/) && !user) {
|
||||
router.push('/login');
|
||||
}
|
||||
}, [router, user]);
|
||||
|
||||
return <>{children}</>;
|
||||
};
|
||||
@ -0,0 +1,31 @@
|
||||
import useSwr from 'swr';
|
||||
import { useRef } from 'react';
|
||||
export interface User {
|
||||
id: number;
|
||||
email: string;
|
||||
}
|
||||
|
||||
interface UserHookResponse {
|
||||
user?: User;
|
||||
loading: boolean;
|
||||
error: string;
|
||||
revalidate: () => Promise<boolean>;
|
||||
}
|
||||
|
||||
export const useUser = ({
|
||||
id,
|
||||
initialData,
|
||||
}: { id?: number; initialData?: User } = {}): UserHookResponse => {
|
||||
const initialRef = useRef(initialData);
|
||||
const { data, error, revalidate } = useSwr<User>(
|
||||
id ? `/api/v1/user/${id}` : `/api/v1/auth/me`,
|
||||
{ initialData: initialRef.current }
|
||||
);
|
||||
|
||||
return {
|
||||
user: data,
|
||||
loading: !data && !error,
|
||||
error,
|
||||
revalidate,
|
||||
};
|
||||
};
|
||||
@ -1,20 +1,77 @@
|
||||
import React from 'react';
|
||||
import '../styles/globals.css';
|
||||
import App from 'next/app';
|
||||
import App, { AppInitialProps } from 'next/app';
|
||||
import { SWRConfig } from 'swr';
|
||||
import Layout from '../components/Layout';
|
||||
import { UserContext } from '../context/UserContext';
|
||||
import axios from 'axios';
|
||||
import { User } from '../hooks/useUser';
|
||||
|
||||
// Custom types so we can correctly type our GetInitialProps function
|
||||
// with our combined user prop
|
||||
// This is specific to _app.tsx. Other pages will not need to do this!
|
||||
type NextAppComponentType = typeof App;
|
||||
type GetInitialPropsFn = NextAppComponentType['getInitialProps'];
|
||||
|
||||
interface AppProps {
|
||||
user: User;
|
||||
}
|
||||
|
||||
class CoreApp extends App<AppProps> {
|
||||
public static getInitialProps: GetInitialPropsFn = async (initialProps) => {
|
||||
// Run the default getInitialProps for the main nextjs initialProps
|
||||
const appInitialProps: AppInitialProps = await App.getInitialProps(
|
||||
initialProps
|
||||
);
|
||||
const { ctx, router } = initialProps;
|
||||
let user = undefined;
|
||||
try {
|
||||
// Attempt to get the user by running a request to the local api
|
||||
const response = await axios.get<User>(
|
||||
`http://localhost:${process.env.PORT || 3000}/api/v1/auth/me`,
|
||||
{ headers: ctx.req ? { cookie: ctx.req.headers.cookie } : undefined }
|
||||
);
|
||||
user = response.data;
|
||||
} catch (e) {
|
||||
// If there is no user, and ctx.res is set (to check if we are on the server side)
|
||||
// _AND_ we are not already on the login or setup route, redirect to /login with a 307
|
||||
// before anything actually renders
|
||||
if (ctx.res && !router.pathname.match(/(login|setup)/)) {
|
||||
ctx.res.writeHead(307, {
|
||||
Location: '/login',
|
||||
});
|
||||
ctx.res.end();
|
||||
}
|
||||
}
|
||||
|
||||
return { ...appInitialProps, user };
|
||||
};
|
||||
|
||||
class CoreApp extends App {
|
||||
public render(): JSX.Element {
|
||||
const { Component, pageProps, router } = this.props;
|
||||
const { Component, pageProps, router, user } = this.props;
|
||||
|
||||
let component: React.ReactNode;
|
||||
|
||||
if (router.asPath === '/login') {
|
||||
return <Component {...pageProps} />;
|
||||
}
|
||||
return (
|
||||
component = <Component {...pageProps} />;
|
||||
} else {
|
||||
component = (
|
||||
<Layout>
|
||||
<Component {...pageProps} />
|
||||
</Layout>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<SWRConfig
|
||||
value={{
|
||||
fetcher: (url) => axios.get(url).then((res) => res.data),
|
||||
}}
|
||||
>
|
||||
<UserContext initialUser={user}>{component}</UserContext>
|
||||
</SWRConfig>
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
export default CoreApp;
|
||||
|
||||
Loading…
Reference in new issue