remove loop from core client.js

mods/core/client.js

@@ -25,7 +25,11 @@ module.exports = (store, __exports) => {
 
   const attempt_interval = setInterval(enhance, 500);
   async function enhance() {
-    if (!document.querySelector('.notion-frame')) return;
+    if (
+      !document.querySelector('.notion-frame') ||
+      !document.querySelector('.notion-sidebar')
+    )
+      return;
     clearInterval(attempt_interval);
 
     // scrollbars
@@ -72,60 +76,65 @@ module.exports = (store, __exports) => {
       .querySelector('.notion-topbar-share-menu')
       .parentElement.classList.add('notion-topbar-actions');
 
-    let sidebar_width;
+    const getStyle = (prop) =>
-    function communicationLoop() {
+      getComputedStyle(
-      const getStyle = (prop) =>
+        document.querySelector('.notion-app-inner')
-          getComputedStyle(
+      ).getPropertyValue(prop);
-            document.querySelector('.notion-app-inner')
-          ).getPropertyValue(prop),
-        mode = JSON.parse(localStorage.theme).mode;
 
-      // ctrl+f theming
+    // ctrl+f theming
-      notionIpc.sendNotionToIndex('search:set-theme', {
+    document.defaultView.addEventListener('keydown', (event) => {
-        'mode': mode,
+      if (event.key === 'f' && (event.ctrlKey || event.metaKey)) {
-        'colors': {
+        notionIpc.sendNotionToIndex('search:set-theme', {
-          'white': getStyle(`--theme--option_active-color`),
+          'mode': document.querySelector('.notion-dark-theme')
-          'blue': getStyle(`--theme--option_active-background`),
+            ? 'dark'
-        },
+            : 'light',
-        'borderRadius': 3,
+          'colors': {
-        'textColor': getStyle(`--theme--text`),
+            'white': getStyle(`--theme--option_active-color`),
-        'popoverBackgroundColor': getStyle(`--theme--card`),
+            'blue': getStyle(`--theme--option_active-background`),
-        'popoverBoxShadow': `0 0 0 1px ${getStyle(
+          },
-          `--theme--overlay`
+          'borderRadius': 3,
-        )}, 0 3px 6px ${getStyle(`--theme--overlay`)}`,
+          'textColor': getStyle(`--theme--text`),
-        'inputBoxShadow': `box-shadow: ${getStyle(
+          'popoverBackgroundColor': getStyle(`--theme--card`),
-          `--theme--primary`
+          'popoverBoxShadow': `0 0 0 1px ${getStyle(
-        )} 0px 0px 0px 1px inset, ${getStyle(
+            `--theme--overlay`
-          `--theme--primary_hover`
+          )}, 0 3px 6px ${getStyle(`--theme--overlay`)}`,
-        )} 0px 0px 0px 2px !important`,
+          'inputBoxShadow': `box-shadow: ${getStyle(
-        'inputBackgroundColor': getStyle(`--theme--main`),
+            `--theme--primary`
-        'dividerColor': getStyle(`--theme--table-border`),
+          )} 0px 0px 0px 1px inset, ${getStyle(
-        'shadowOpacity': 0.2,
+            `--theme--primary_hover`
-      });
+          )} 0px 0px 0px 2px !important`,
+          'inputBackgroundColor': getStyle(`--theme--main`),
+          'dividerColor': getStyle(`--theme--table-border`),
+          'shadowOpacity': 0.2,
+        });
+      }
+    });
 
-      // enhancer menu
+    // enhancer menu
-      electron.ipcRenderer.send('enhancer:set-theme', {
+    function setMenuTheme() {
-        mode,
+      electron.ipcRenderer.send('enhancer:set-menu-theme', {
+        mode: document.querySelector('.notion-dark-theme') ? 'dark' : 'light',
         rules: require('./css/variables.json').map((rule) => [
           rule,
           getStyle(rule),
         ]),
       });
-
-      // draggable area resizing
-      const sidebar = document.querySelector('.notion-sidebar');
-      if (store().frameless && sidebar) {
-        let new_sidebar_width =
-          sidebar.style.height === 'auto' ? '0px' : sidebar.style.width;
-        if (sidebar_width !== new_sidebar_width) {
-          sidebar_width = new_sidebar_width;
-          electron.ipcRenderer.sendToHost(
-            'enhancer:sidebar-width',
-            sidebar_width
-          );
-        }
-      }
     }
-    setInterval(communicationLoop, 500);
+    setMenuTheme();
+    electron.ipcRenderer.on('enhancer:get-menu-theme', setMenuTheme);
+
+    const observer = new MutationObserver(setSidebarWidth);
+    observer.observe(document.querySelector('.notion-sidebar'), {
+      attributes: true,
+    });
+    function setSidebarWidth(list, observer) {
+      if (!store().frameless) return;
+      electron.ipcRenderer.sendToHost(
+        'enhancer:sidebar-width',
+        list[0].target.style.height === 'auto'
+          ? '0px'
+          : list[0].target.style.width
+      );
+    }
   }
 };

mods/core/menu.js

@@ -41,7 +41,8 @@ window['__start'] = async () => {
     }
   });
 
-  electron.ipcRenderer.on('enhancer:set-theme', (event, theme) => {
+  electron.ipcRenderer.send('enhancer:get-menu-theme');
+  electron.ipcRenderer.on('enhancer:set-menu-theme', (event, theme) => {
     document.body.className = `notion-${theme.mode}-theme`;
     for (const style of theme.rules)
       document.body.style.setProperty(style[0], style[1]);

mods/core/render.js

@@ -16,8 +16,6 @@ module.exports = (store, __exports) => {
       ),
       default_styles = dragarea.getAttribute('style');
 
-    // document.body.innerText = document.body.innerHTML;
-
     document
       .getElementById('notion')
       .addEventListener('ipc-message', (event) => {

mods/core/tray.js

@@ -29,9 +29,14 @@ module.exports = (store, __exports) => {
           })
     );
 
-    electron.ipcMain.on('enhancer:set-theme', (event, arg) => {
+    electron.ipcMain.on('enhancer:set-menu-theme', (event, arg) => {
       if (!enhancer_menu) return;
-      enhancer_menu.webContents.send('enhancer:set-theme', arg);
+      enhancer_menu.webContents.send('enhancer:set-menu-theme', arg);
+    });
+    electron.ipcMain.on('enhancer:get-menu-theme', (event, arg) => {
+      electron.webContents
+        .getAllWebContents()
+        .forEach((webContents) => webContents.send('enhancer:get-menu-theme'));
     });
     electron.ipcMain.on('enhancer:open-extension-menu', openExtensionMenu);