From 43a2c98b65b61aa233fad3fbc44f427bd731e420 Mon Sep 17 00:00:00 2001
From: Andrew Welch <andrew@nystudio107.com>
Date: Sat, 19 Jun 2021 13:30:57 -0400
Subject: [PATCH] Add before/after router hooks

Add before/after router hooks so that can be added via `enhanceApp()`, to allow for code to be executed when new pages are routed to

Signed-off-by: Andrew Welch <andrew@nystudio107.com>
---
 src/client/app/router.ts | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/client/app/router.ts b/src/client/app/router.ts
index b7261ab4..c7aa7935 100644
--- a/src/client/app/router.ts
+++ b/src/client/app/router.ts
@@ -9,9 +9,15 @@ export interface Route {
   component: Component | null
 }
 
+export interface RouterHooks {
+  before?: (route: Route) => void
+  after?: (route: Route) => void
+}
+
 export interface Router {
   route: Route
   go: (href?: string) => Promise<void>
+  hooks?: RouterHooks
 }
 
 export const RouterSymbol: InjectionKey<Router> = Symbol()
@@ -29,6 +35,11 @@ const getDefaultRoute = (): Route => ({
   data: null as any
 })
 
+const getDefaultHooks = (): RouterHooks => ({
+  before: undefined,
+  after: undefined
+})
+
 interface PageModule {
   __pageData: string
   default: Component
@@ -39,6 +50,7 @@ export function createRouter(
   fallbackComponent?: Component
 ): Router {
   const route = reactive(getDefaultRoute())
+  const hooks = reactive(getDefaultHooks())
 
   function go(href: string = inBrowser ? location.href : '/') {
     // ensure correct deep link so page refresh lands on correct files.
@@ -78,6 +90,9 @@ export function createRouter(
         route.path = pendingPath
         route.component = markRaw(comp)
         route.data = readonly(JSON.parse(__pageData)) as PageData
+        if (hooks.before) {
+          hooks.before(route)
+        }
 
         if (inBrowser) {
           nextTick(() => {
@@ -104,6 +119,9 @@ export function createRouter(
         route.component = fallbackComponent ? markRaw(fallbackComponent) : null
       }
     }
+    if (hooks.after) {
+      hooks.after(route)
+    }
   }
 
   if (inBrowser) {
@@ -154,7 +172,8 @@ export function createRouter(
 
   return {
     route,
-    go
+    go,
+    hooks
   }
 }