45 lines
1.2 KiB
TypeScript
45 lines
1.2 KiB
TypeScript
import { useState, useEffect } from 'react';
|
|
import { useRouter } from 'next/router';
|
|
|
|
type CompilationStatus = 'ready' | 'compiling' | 'error' | 'initial';
|
|
|
|
const useDevCompilationStatus = (): CompilationStatus => {
|
|
const router = useRouter();
|
|
const [status, setStatus] = useState<CompilationStatus>('initial');
|
|
|
|
useEffect(() => {
|
|
if (process.env.NODE_ENV !== 'development') {
|
|
setStatus('ready');
|
|
return;
|
|
}
|
|
|
|
const handleRouteChangeStart = () => {
|
|
setStatus('compiling');
|
|
};
|
|
|
|
const handleRouteChangeComplete = () => {
|
|
setTimeout(() => setStatus('ready'), 300);
|
|
};
|
|
|
|
const handleRouteChangeError = () => {
|
|
setTimeout(() => setStatus('error'), 300);
|
|
};
|
|
|
|
router.events.on('routeChangeStart', handleRouteChangeStart);
|
|
router.events.on('routeChangeComplete', handleRouteChangeComplete);
|
|
router.events.on('routeChangeError', handleRouteChangeError);
|
|
|
|
setStatus('ready');
|
|
|
|
return () => {
|
|
router.events.off('routeChangeStart', handleRouteChangeStart);
|
|
router.events.off('routeChangeComplete', handleRouteChangeComplete);
|
|
router.events.off('routeChangeError', handleRouteChangeError);
|
|
};
|
|
}, [router]);
|
|
|
|
return status;
|
|
};
|
|
|
|
export default useDevCompilationStatus;
|