Ability to unregister umami

kumasuke · Sep 14, 2020 · 2ea91aa · 2ea91aa
1 parent 8727dbf
commit 2ea91aa
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/lib/web.js b/lib/web.js
@@ -39,12 +39,14 @@ export const put = (url, params) => apiRequest('put', url, JSON.stringify(params
 
 export const hook = (_this, method, callback) => {
   const orig = _this[method];
-
-  return (...args) => {
+  _this[method] = (...args) => {
     callback.apply(null, args);
-
     return orig.apply(_this, args);
   };
+
+  return () => {
+    _this[method] = orig;
+  };
 };
 
 export const doNotTrack = () => {

diff --git a/tracker/index.js b/tracker/index.js
@@ -73,8 +73,8 @@ import { removeTrailingSlash } from '../lib/url';
     pageView();
   };
 
-  history.pushState = hook(history, 'pushState', handlePush);
-  history.replaceState = hook(history, 'replaceState', handlePush);
+  const pushStateUnhook = hook(history, 'pushState', handlePush);
+  const replaceStateUnhook = hook(history, 'replaceState', handlePush);
 
   /* Handle events */
 
@@ -106,4 +106,12 @@ import { removeTrailingSlash } from '../lib/url';
   if (!window.umami) {
     window.umami = event_value => collect('event', { event_type: 'custom', event_value });
   }
+
+  if (!window.umamiUnregister) {
+    window.umamiUnregister = () => {
+      pushStateUnhook();
+      replaceStateUnhook();
+      removeEvents();
+    };
+  }
 })(window);