Next.js block navigation on unsaved changes

useNavigationLock.ts

import { useEffect, useState } from 'react';
import { useRouter } from 'next/router';

/** Ask for confirmation before changing page or leaving site.
 *
 * @see https://git.io/JOskG
 */
const useNavigationLock = (
  isEnabled = true,
  warningText = 'You have unsaved changes – are you sure you wish to leave this page?',
) => {
  const router = useRouter();

  useEffect(() => {
    const handleWindowClose = (e: BeforeUnloadEvent) => {
      if (!isEnabled) return;
      e.preventDefault();
      return (e.returnValue = warningText);
    };

    const handleBrowseAway = () => {
      if (!isEnabled) return;
      if (window.confirm(warningText)) return;
      router.events.emit('routeChangeError');
      throw 'routeChange aborted.';
    };

    window.addEventListener('beforeunload', handleWindowClose);

    router.events.on('routeChangeStart', handleBrowseAway);

    return () => {
      window.removeEventListener('beforeunload', handleWindowClose);
      router.events.off('routeChangeStart', handleBrowseAway);
    };
  }, [isEnabled]);
};

export default useNavigationLock;

I found an issue. When I click the cancel button it clear the states.