From 04308fbaea36a36e6e88e34ac9d76acf3c7f78ab Mon Sep 17 00:00:00 2001
From: "mr. m" <mr.m@tuta.com>
Date: Fri, 21 Mar 2025 00:48:43 +0100
Subject: [PATCH] fix(workspaces): ensure onWindowResize is triggered after
 workspace removal and optimize button alignment logic

---
 .../base/zen-components/ZenWorkspaces.mjs     | 21 +++++++++++--------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/src/browser/base/zen-components/ZenWorkspaces.mjs b/src/browser/base/zen-components/ZenWorkspaces.mjs
index 866831a0..2ca0adb5 100644
--- a/src/browser/base/zen-components/ZenWorkspaces.mjs
+++ b/src/browser/base/zen-components/ZenWorkspaces.mjs
@@ -873,6 +873,7 @@ var ZenWorkspaces = new (class extends ZenMultiWindowFeature {
     await ZenWorkspacesStorage.removeWorkspace(windowID);
     await this._propagateWorkspaceData();
     await this._updateWorkspacesChangeContextMenu();
+    this.onWindowResize();
   }
 
   isWorkspaceActive(workspace) {
@@ -2379,14 +2380,16 @@ var ZenWorkspaces = new (class extends ZenMultiWindowFeature {
     if (!parent) {
       return;
     }
-    // Once we are overflowing, we align the buttons to always stay inside the container,
-    // meaning we need to remove the overflow attribute to reset the width
-    parent.removeAttribute('overflow');
-    const overflow = parent.scrollWidth > parent.clientWidth;
-    parent.toggleAttribute('overflow', overflow);
-    // The maximum width a button has when it overflows based on the number of buttons
-    const numButtons = parent.children.length + 1; // +1 to exclude the active button
-    const maxWidth = 100 / numButtons;
-    parent.style.setProperty('--zen-overflowed-workspace-button-width', `${maxWidth}%`);
+    window.requestAnimationFrame(() => {
+      // Once we are overflowing, we align the buttons to always stay inside the container,
+      // meaning we need to remove the overflow attribute to reset the width
+      parent.removeAttribute('overflow');
+      const overflow = parent.scrollWidth > parent.clientWidth;
+      parent.toggleAttribute('overflow', overflow);
+      // The maximum width a button has when it overflows based on the number of buttons
+      const numButtons = parent.children.length + 1; // +1 to exclude the active button
+      const maxWidth = 100 / numButtons;
+      parent.style.setProperty('--zen-overflowed-workspace-button-width', `${maxWidth}%`);
+    });
   }
 })();