From 5fb6ca140b65c3fea7ec2d4c5fee44a6160805d6 Mon Sep 17 00:00:00 2001
From: Shrey Gupta <connectwithshrey@gmail.com>
Date: Mon, 30 Sep 2024 16:29:54 +0530
Subject: [PATCH] fix(notion): Fix button not re-rendering on page change

---
 src/content/notion.js | 79 ++++++++++++++++++++++++-------------------
 1 file changed, 45 insertions(+), 34 deletions(-)

diff --git a/src/content/notion.js b/src/content/notion.js
index ca3baf458..2ee4625ce 100644
--- a/src/content/notion.js
+++ b/src/content/notion.js
@@ -19,6 +19,7 @@ togglbutton.render(
   '.notion-peek-renderer:not(.toggl)',
   { observe: true },
   function (elem) {
+    if (!elem) return;
     function getDescription () {
       const descriptionElem = elem.querySelector('.notion-peek-renderer .notion-scroller h1[contenteditable]');
       return descriptionElem ? descriptionElem.textContent.trim() : '';
@@ -36,54 +37,63 @@ togglbutton.render(
     if (root) {
       root.parentElement.prepend(wrapper);
     } else {
-      elem.querySelector('div:nth-child(2) > div:nth-child(1) > div:nth-child(1) > div:nth-child(3)').prepend(wrapper)
+      const selector = elem.querySelector('div:nth-child(2) > div:nth-child(1) > div:nth-child(1) > div:nth-child(3)')
+      if (!selector) return;
+      selector.prepend(wrapper)
     }
   }
 );
 
-togglbutton.render(
-  '.notion-topbar-action-buttons:not(.toggl)',
-  { observe: true },
-  function (elem) {
-    if (!elem) return;
+setTimeout(() => {
+  togglbutton.render(
+    '.notion-topbar-action-buttons',
+    { observe: true, debounceInterval: 1000 },
+    function (elem) {
+      if (!elem) return;
+      const elements = document.querySelectorAll('.notion-topbar-action-buttons .toggl-button-notion-wrapper')
+      if(elements.length > 0) {
+        elements.forEach(element => element.remove())
+      }
 
-    elem.style.position = 'relative';
+      elem.style.position = 'relative';
 
-    function getDescription () {
-      const controls = document.querySelector('.notion-page-controls');
-      const topBar = document.querySelector('.notion-topbar');
-      let title = '';
+      function getDescription () {
+        const controls = document.querySelector('.notion-page-controls');
+        const topBar = document.querySelector('.notion-topbar');
+        let title = '';
 
-      if (controls) {
-        if (controls.nextElementSibling) {
-          title = controls.nextElementSibling;
-        } else {
-          const parent = controls.parentElement;
+        if (controls) {
+          if (controls.nextElementSibling) {
+            title = controls.nextElementSibling;
+          } else {
+            const parent = controls.parentElement;
 
-          if (!parent) return '';
+            if (!parent) return '';
 
-          title = parent ? parent.nextElementSibling : '';
-        }
-      } else if (topBar) {
-        const breadcrumbs = topBar.querySelector('div > .notranslate')
-        if (breadcrumbs) {
-          title = breadcrumbs.childNodes[breadcrumbs.childNodes.length - 1].querySelector('.notranslate:last-child')
+            title = parent ? parent.nextElementSibling : '';
+          }
+        } else if (topBar) {
+          const breadcrumbs = topBar.querySelector('div > .notranslate')
+          if (breadcrumbs) {
+            title = breadcrumbs.childNodes[breadcrumbs.childNodes.length - 1].querySelector('.notranslate:last-child')
+          }
         }
-      }
 
-      return title ? title.textContent.trim() : '';
-    }
+        return title ? title.textContent.trim() : '';
+      }
 
-    const link = togglbutton.createTimerLink({
-      className: 'notion',
-      description: getDescription,
-    });
+      const link = togglbutton.createTimerLink({
+        className: 'notion',
+        description: getDescription,
+      });
 
-    const wrapper = createWrapper(link);
+      const wrapper = createWrapper(link);
 
-    elem.prepend(wrapper);
-  }
-);
+      elem.prepend(wrapper);
+    },
+    '.notion-topbar .shadow-cursor-breadcrumb *,title'
+  );
+}, 2000)
 
 /**
  * @name Notion Calendar
@@ -94,6 +104,7 @@ togglbutton.render(
   'div[data-context-panel-root]:not(.toggl)',
   { observe: true },
   function (elem) {
+    if (!elem) return;
     function getDescription () {
       const descriptionElem = elem.querySelector('div[contenteditable="true"]');
       return descriptionElem ? descriptionElem.textContent.trim() : '';