aboutsummaryrefslogtreecommitdiffstats
path: root/src/utils/hooks/use-on-route-change/use-on-route-change.ts
diff options
context:
space:
mode:
Diffstat (limited to 'src/utils/hooks/use-on-route-change/use-on-route-change.ts')
-rw-r--r--src/utils/hooks/use-on-route-change/use-on-route-change.ts35
1 files changed, 35 insertions, 0 deletions
diff --git a/src/utils/hooks/use-on-route-change/use-on-route-change.ts b/src/utils/hooks/use-on-route-change/use-on-route-change.ts
new file mode 100644
index 0000000..85e7060
--- /dev/null
+++ b/src/utils/hooks/use-on-route-change/use-on-route-change.ts
@@ -0,0 +1,35 @@
+import { useRouter } from 'next/router';
+import { useEffect } from 'react';
+
+export type OnRouteChangeStep = 'start' | 'end';
+
+export type OnRouteChangeHandler = () => void;
+
+/**
+ * React hook to trigger a callback function on route change.
+ *
+ * @param {OnRouteChangeHandler} handler - The callback to trigger.
+ * @param {OnRouteChangeStep} [step] - The event step.
+ */
+export const useOnRouteChange = (
+ handler: OnRouteChangeHandler,
+ step: OnRouteChangeStep = 'start'
+) => {
+ const router = useRouter();
+
+ useEffect(() => {
+ if (step === 'end') {
+ router.events.on('routeChangeComplete', handler);
+
+ return () => {
+ router.events.off('routeChangeComplete', handler);
+ };
+ }
+
+ router.events.on('routeChangeStart', handler);
+
+ return () => {
+ router.events.off('routeChangeStart', handler);
+ };
+ }, [handler, router.events, step]);
+};