import { Router } from 'next/router'
import { useEffect } from 'react'

export const useWarnUnsavedChanges = (hasUnsavedChanges: boolean) => {
  useEffect(() => {
    if (!hasUnsavedChanges) return

    const confirmationMessage = 'Changes you made may not be saved.'

    const warnUnsavedChanges = (e: BeforeUnloadEvent) => {
      const event = e || window.event
      event.returnValue = confirmationMessage
      return confirmationMessage
    }

    const beforeRouteHandler = (href: string) => {
      const pathname = href.split('?')[0]
      // Don't warn if the user is navigating to the same page.
      if (pathname === location.pathname) return

      if (!confirm(confirmationMessage)) {
        Router.events.emit('routeChangeError')
        throw 'Abort route change. Please ignore this error.'
      }
    }

    window.addEventListener('beforeunload', warnUnsavedChanges)
    Router.events.on('routeChangeStart', (href) => beforeRouteHandler(href))

    return () => {
      window.removeEventListener('beforeunload', warnUnsavedChanges)
      Router.events.off('routeChangeStart', beforeRouteHandler)
    }
  }, [hasUnsavedChanges])
}