projeto-hit/frontend/src/hooks/useAuth.js/index.js

import { useState, useEffect } from 'react'
import { useHistory } from 'react-router-dom'
//import openSocket from 'socket.io-client'
import { socket } from '../../services/socket'

import { toast } from 'react-toastify'

import { i18n } from '../../translate/i18n'
import api from '../../services/api'
import toastError from '../../errors/toastError'

const useAuth = () => {
  const history = useHistory()
  const [isAuth, setIsAuth] = useState(false)
  const [loading, setLoading] = useState(true)
  const [user, setUser] = useState({})

  const [setting, setSetting] = useState({})

  const getSettingValue = (key) => {
    return setting?.find((s) => s?.key === key)?.value
  }


  api.interceptors.request.use(
    (config) => {
      const token = localStorage.getItem('token')
      if (token) {
        config.headers['Authorization'] = `Bearer ${JSON.parse(token)}`
        setIsAuth(true)
      }
      return config
    },
    (error) => {
      Promise.reject(error)
    }
  )

  api.interceptors.response.use(
    (response) => {
      return response
    },
    async (error) => {
      const originalRequest = error.config
      if (error?.response?.status === 403 && !originalRequest._retry) {
        originalRequest._retry = true

        const { data } = await api.post('/auth/refresh_token')
        if (data) {
          localStorage.setItem('token', JSON.stringify(data.token))
          api.defaults.headers.Authorization = `Bearer ${data.token}`
        }
        return api(originalRequest)
      }
      if (error?.response?.status === 401) {
        localStorage.removeItem('token')
        api.defaults.headers.Authorization = undefined
        setIsAuth(false)
      }
      return Promise.reject(error)
    }
  )

  useEffect(() => {
    const token = localStorage.getItem('token')
      ; (async () => {
        if (token) {
          try {
            const { data } = await api.post('/auth/refresh_token')
            api.defaults.headers.Authorization = `Bearer ${data.token}`
            setIsAuth(true)
            setUser(data.user)
          } catch (err) {
            toastError(err)
          }
        }
        setLoading(false)
      })()
  }, [])

  useEffect(() => {
    const fetchSession = async () => {
      try {
        const { data } = await api.get('/settings')
        setSetting(data.settings)
      } catch (err) {
        toastError(err)
      }
    }
    fetchSession()
  }, [])

  useEffect(() => {
    //const socket = openSocket(process.env.REACT_APP_BACKEND_URL)

    socket.on('user', (data) => {
      if (data.action === 'update' && data.user.id === user.id) {
        setUser(data.user)
      }
    })

    const onSettingsAuth = (data) => {
      if (data.action === 'update') {
        setSetting((prevState) => {
          const aux = [...prevState]
          const settingIndex = aux.findIndex((s) => s.key === data.setting.key)
          aux[settingIndex].value = data.setting.value
          return aux
        })
      }
    }

    socket.on('settings', onSettingsAuth)

    return () => {
      socket.removeAllListeners('user');
      socket.off('settings', onSettingsAuth);
    }
  }, [user])

  const handleLogin = async (userData) => {
    setLoading(true)

    try {
      const { data } = await api.post('/auth/login', userData)
      localStorage.setItem('token', JSON.stringify(data.token))
      api.defaults.headers.Authorization = `Bearer ${data.token}`
      setUser(data.user)
      setIsAuth(true)
      toast.success(i18n.t('auth.toasts.success'))
      history.push('/tickets')
      setLoading(false)
    } catch (err) {
      toastError(err)
      setLoading(false)
    }
  }

  const handleLogout = async () => {
    setLoading(true)

    try {
      await api.delete(`/auth/logout/${user.id}`);

      setIsAuth(false)
      setUser({})
      localStorage.removeItem('token')
      api.defaults.headers.Authorization = undefined
      setLoading(false)
      history.push('/login')
    } catch (err) {
      toastError(err)
      setLoading(false)
    }
  }

  return {
    isAuth,
    user,
    loading,
    handleLogin,
    handleLogout,
    setting,
    setSetting,
    getSettingValue
  }
}

export default useAuth