rotas

import React from 'react';
import PropTypes from 'prop-types';
import { Route, Redirect } from 'react-router-dom';

import AuthLayout from '~/screens/_layouts/auth';
import DefaultLayout from '~/screens/_layouts/default';
import { store } from '~/store';

export default function RouteWrapper({
  component: Component,
  isPrivate = false,
  layout,
  ...rest
}) {
  const { signed, firstLogin, aceptedTerms } = store.getState().auth;

  if (!aceptedTerms && signed && rest.path !== '/termos-de-servico') {
    return <Redirect to="termos-de-servico" />;
  }

  if (aceptedTerms && rest.path === '/termos-de-servico') {
    return <Redirect to="/" />;
  }

  if (!firstLogin && rest.path === '/abertura') {
    return <Redirect to="/" />;
  }
  
    if (!aceptedTerms && signed && rest.path !== '/termos-de-servico') {
    return <Redirect to="termos-de-servico" />;
  }

  if (aceptedTerms && rest.path === '/termos-de-servico') {
    return <Redirect to="/" />;
  }

  if (!firstLogin && rest.path === '/novo-ensino-medio') {
    return <Redirect to="/" />;
  }


  if (!signed && isPrivate) {
    return <Redirect to="/" />;
  }

  if (signed && !isPrivate) {
    return <Redirect to="principal" />;
  }

  const Layout = layout === 'auth' ? AuthLayout : DefaultLayout;

  return (
    <Route
      {...rest}
      render={props => (
        <Layout {...props}>
          <Component {...props} />
        </Layout>
      )}
    />
  );
}

RouteWrapper.propTypes = {
  isPrivate: PropTypes.bool,
  component: PropTypes.oneOfType([PropTypes.element, PropTypes.func])
    .isRequired,
  layout: PropTypes.string,
};

RouteWrapper.defaultProps = {
  isPrivate: false,
  layout: '',
};