-
Notifications
You must be signed in to change notification settings - Fork 30
/
patchRouter.ts
37 lines (32 loc) · 1.1 KB
/
patchRouter.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import { SingletonRouter, Router } from 'next/router'
import ReactDOM from 'react-dom'
import { Store } from 'redux'
import { onLocationChanged } from './actions'
import locationFromUrl from './utils/locationFromUrl'
type RouterToPatch = SingletonRouter & { router: Router }
const patchRouter = (Router: RouterToPatch, store: Store): (() => void) => {
const unpatchedMethods = {
set: Reflect.get(Router.router, 'set'),
}
Reflect.set(Router.router, 'set', function (this: any, ...args: any[]) {
if (!unpatchedMethods.set) {
return Promise.resolve()
}
return new Promise((resolve, reject) => {
ReactDOM.unstable_batchedUpdates(() => {
try {
const result = Reflect.apply(unpatchedMethods.set, Router.router, args)
if (result instanceof Promise) result.then(resolve, reject)
else resolve(result)
} catch (err) {
reject(err)
}
store.dispatch(onLocationChanged(locationFromUrl(Router.asPath)))
})
})
})
return () => {
Reflect.set(Router.router, 'set', unpatchedMethods.set)
}
}
export default patchRouter