import React, { FunctionComponent, useRef, useState } from 'react';
import { AuthContext, AuthData, getAuthDataFromLocalStorage, RefreshTokenReminder } from '../Context/AuthContext';

const Auth: FunctionComponent = ({ children }) => {
    const [authData, setAuthData] = useState<AuthData | undefined>(getAuthDataFromLocalStorage());

    const refreshTokenReminder = useRef<RefreshTokenReminder>({});

    const setAuth = (data?: AuthData): void => {
        if (data) {
            localStorage.setItem('device_code', data.deviceCode);
            localStorage.setItem('access_token', data.accessToken);
            localStorage.setItem('refresh_token', data.refreshToken || '');
        } else {
            localStorage.removeItem('device_code');
            localStorage.removeItem('access_token');
            localStorage.removeItem('refresh_token');
        }

        setAuthData(data);
    };

    return (
        <AuthContext.Provider value={{ auth: authData, setAuth, refreshTokenReminder }}>
            { children }
        </AuthContext.Provider>
    );
};

export default Auth;