From bbc8c7a1e0b24ba0328b768643bbbe27276b2372 Mon Sep 17 00:00:00 2001
From: dragonwocky <thedragonring.bod@gmail.com>
Date: Sun, 12 May 2024 11:25:02 +1000
Subject: [PATCH] fix: replace .notion-help-button with .notion-ai-button

---
 src/core/islands/FloatingButton.mjs | 12 ++++++------
 src/core/islands/Panel.mjs          |  9 ++++-----
 src/extensions/focus/client.css     |  5 +++--
 3 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/src/core/islands/FloatingButton.mjs b/src/core/islands/FloatingButton.mjs
index 7c78a88..63940df 100644
--- a/src/core/islands/FloatingButton.mjs
+++ b/src/core/islands/FloatingButton.mjs
@@ -6,15 +6,15 @@
 
 let __$wrapper;
 const setupWrapper = () => {
-    const notionHelp = ".notion-help-button",
+    const notionAi = ".notion-ai-button",
       { html, addMutationListener } = globalThis.__enhancerApi,
       { removeMutationListener } = globalThis.__enhancerApi;
     return (__$wrapper ??= new Promise((res) => {
       const addToDom = () => {
-        const $help = document.querySelector(notionHelp);
-        if (!$help) return;
+        const $notionAi = document.querySelector(notionAi);
+        if (!$notionAi) return;
         const gap = 12,
-          computedStyles = getComputedStyle($help),
+          computedStyles = getComputedStyle($notionAi),
           visible = computedStyles.getPropertyValue("display") !== "none",
           width = computedStyles.getPropertyValue("width"),
           right = computedStyles.getPropertyValue("right"),
@@ -25,10 +25,10 @@ const setupWrapper = () => {
             style="right:${offset}px"
           ></div>`;
         removeMutationListener(addToDom);
-        $help.after($wrapper);
+        $notionAi.after($wrapper);
         res($wrapper);
       };
-      addMutationListener(notionHelp, addToDom);
+      addMutationListener(notionAi, addToDom);
       addToDom();
     }));
   },
diff --git a/src/core/islands/Panel.mjs b/src/core/islands/Panel.mjs
index 1e0291f..7cf9998 100644
--- a/src/core/islands/Panel.mjs
+++ b/src/core/islands/Panel.mjs
@@ -206,9 +206,8 @@ function Panel({
     };
 
   const animationState = { ...closedWidth },
-    easing = "cubic-bezier(0.4, 0, 0.2, 1)",
     animate = ($target, keyframes) => {
-      const opts = { fill: "forwards", duration: transitionDuration, easing };
+      const opts = { fill: "forwards", duration: transitionDuration, easing: "ease" };
       $target.animate(keyframes, opts);
     },
     animatePanel = (to) => {
@@ -303,10 +302,10 @@ function Panel({
   // moves help button out of the way of open panel.
   // normally would place outside of an island, but in
   // this case is necessary for syncing up animations
-  const notionHelp = ".notion-help-button",
+  const notionAi = ".notion-ai-button",
     floatingButtons = ".notion-enhancer--floating-buttons",
     repositionCorner = async (offset) => {
-      const $help = document.querySelector(notionHelp),
+      const $help = document.querySelector(notionAi),
         $floating = document.querySelector(floatingButtons);
       offset ??= await getWidth();
       if (isNaN(offset)) offset = minWidth;
@@ -329,7 +328,7 @@ function Panel({
       }
       if ($help || $floating) removeMutationListener(repositionCorner);
     };
-  const corner = `${notionHelp}, ${floatingButtons}`;
+  const corner = `${notionAi}, ${floatingButtons}`;
   addMutationListener(corner, repositionCorner, { subtree: false });
 
   $panel.pin = () => {
diff --git a/src/extensions/focus/client.css b/src/extensions/focus/client.css
index 8f4444b..e88c8e0 100644
--- a/src/extensions/focus/client.css
+++ b/src/extensions/focus/client.css
@@ -5,8 +5,9 @@
  * (https://notion-enhancer.github.io/) under the MIT license
  */
 
-/* hide topbar */
-.notion-sidebar-container[aria-hidden] ~ div .notion-topbar {
+/* hide topbar and ai */
+.notion-sidebar-container[aria-hidden] ~ div .notion-topbar,
+.notion-sidebar-container[aria-hidden] ~ div .notion-ai-button {
   opacity: 0 !important;
   transition: opacity 200ms ease-in-out !important;
 }