From 8a54616572369f1821c55de65ea069917d8b272c Mon Sep 17 00:00:00 2001
From: Gregorein <hello@gregore.in>
Date: Wed, 23 Apr 2025 19:56:51 +0200
Subject: [PATCH] feat: add multi-tile window suggestions feature

---
 ...e.shell.extensions.tilingshell.gschema.xml |  5 ++
 .../tilingLayoutWithSuggestions.ts            | 49 ++++++++++++++++++-
 src/prefs.ts                                  | 11 ++++-
 src/settings/settings.ts                      | 10 ++++
 4 files changed, 72 insertions(+), 3 deletions(-)

diff --git a/resources/schemas/org.gnome.shell.extensions.tilingshell.gschema.xml b/resources/schemas/org.gnome.shell.extensions.tilingshell.gschema.xml
index 7b441e47..1f30add9 100644
--- a/resources/schemas/org.gnome.shell.extensions.tilingshell.gschema.xml
+++ b/resources/schemas/org.gnome.shell.extensions.tilingshell.gschema.xml
@@ -187,6 +187,11 @@
             <summary>Enable window suggestions for screen edge snapping</summary>
             <description>Suggests windows to occupy empty tiles when snapping to screen edges.</description>
         </key>
+        <key name="enable-multi-tile-window-suggestions" type="b">
+            <default>false</default>
+            <summary>Enable multi-tile window suggestions</summary>
+            <description>Displays window suggestions in all available tiles after a snap.</description>
+        </key>
 
         <!-- keybindings -->
 		<key type="as" name="move-window-right">
diff --git a/src/components/windowsSuggestions/tilingLayoutWithSuggestions.ts b/src/components/windowsSuggestions/tilingLayoutWithSuggestions.ts
index 5b92c59b..1ec4a338 100644
--- a/src/components/windowsSuggestions/tilingLayoutWithSuggestions.ts
+++ b/src/components/windowsSuggestions/tilingLayoutWithSuggestions.ts
@@ -13,6 +13,7 @@ import SignalHandling from '@utils/signalHandling';
 import SuggestionsTilePreview from '@components/windowsSuggestions/suggestionsTilePreview';
 import TilingShellWindowManager from '@components/windowManager/tilingShellWindowManager';
 import { unmaximizeWindow } from '@utils/gnomesupport';
+import Settings from '@settings/settings';
 
 const debug = logger('TilingLayoutWithSuggestions');
 
@@ -138,12 +139,33 @@ export default class TilingLayoutWithSuggestions extends LayoutWidget<Suggestion
             return;
         }
 
-        // find the leftmost preview
+        // If multi-tile mode is enabled, show windows in all previews
+        if (Settings.ENABLE_MULTI_TILE_WINDOW_SUGGESTIONS) {
+            // Show windows in all previews at once
+            this._previews.forEach((preview) => {
+                this._showWindowsInPreview(
+                    preview,
+                    nontiledWindows,
+                    monitorIndex,
+                );
+            });
+            return;
+        }
+
+        // Original single-tile mode logic - find the leftmost preview
         let preview = this._previews[0];
         this._previews.forEach((prev) => {
             if (prev.x < preview.x) preview = prev;
         });
 
+        this._showWindowsInPreview(preview, nontiledWindows, monitorIndex);
+    }
+
+    private _showWindowsInPreview(
+        preview: SuggestionsTilePreview,
+        nontiledWindows: Meta.Window[],
+        monitorIndex: number,
+    ): void {
         const clones = nontiledWindows.map((nonTiledWin) => {
             const winClone = new SuggestedWindowPreview(nonTiledWin);
             const winActor =
@@ -244,7 +266,30 @@ export default class TilingLayoutWithSuggestions extends LayoutWidget<Suggestion
                 this._oldPreviews.push(...removed);
                 nontiledWindows.splice(nontiledWindows.indexOf(nonTiledWin), 1);
                 preview.close(true);
-                this._recursivelyShowPopup(nontiledWindows, monitorIndex);
+
+                // In multi-tile mode, only close if all previews are filled or no more windows
+                if (Settings.ENABLE_MULTI_TILE_WINDOW_SUGGESTIONS) {
+                    if (
+                        this._previews.length === 0 ||
+                        nontiledWindows.length === 0
+                    ) {
+                        this.close();
+                    } else {
+                        // Remove the window from all other previews since it's now assigned
+                        this._previews.forEach((prev) => {
+                            prev.removeAllWindows();
+                            this._showWindowsInPreview(
+                                prev,
+                                nontiledWindows,
+                                monitorIndex,
+                            );
+                        });
+                    }
+                } else {
+                    // Original behavior: recursively show popup on the next vacant tile
+                    this._recursivelyShowPopup(nontiledWindows, monitorIndex);
+                }
+
                 return Clutter.EVENT_STOP; // Blocca la propagazione
             });
             return winClone;
diff --git a/src/prefs.ts b/src/prefs.ts
index 98f16d54..c141ea63 100644
--- a/src/prefs.ts
+++ b/src/prefs.ts
@@ -146,7 +146,7 @@ export default class TilingShellExtensionPreferences extends ExtensionPreference
             this._buildSwitchRow(
                 Settings.KEY_ENABLE_SMART_WINDOW_BORDER_RADIUS,
                 _('Smart border radius'),
-                _('Dynamically adapt to the window’s actual border radius'),
+                _("Dynamically adapt to the window's actual border radius"),
             ),
         );
         windowBorderRow.add_row(
@@ -383,6 +383,15 @@ export default class TilingShellExtensionPreferences extends ExtensionPreference
         );
         windowsSuggestionsGroup.add(screenEdgesWindowSuggestionRow);
 
+        const multiTileWindowSuggestionRow = this._buildSwitchRow(
+            Settings.KEY_ENABLE_MULTI_TILE_WINDOW_SUGGESTIONS,
+            _('Enable multi-tile window suggestions'),
+            _(
+                'Displays window suggestions in all available tiles after a snap',
+            ),
+        );
+        windowsSuggestionsGroup.add(multiTileWindowSuggestionRow);
+
         prefsPage.add(windowsSuggestionsGroup);
 
         // Layouts section
diff --git a/src/settings/settings.ts b/src/settings/settings.ts
index 3a890449..308828d4 100644
--- a/src/settings/settings.ts
+++ b/src/settings/settings.ts
@@ -121,6 +121,8 @@ export default class Settings {
         'enable-snap-assistant-windows-suggestions';
     static KEY_ENABLE_SCREEN_EDGES_WINDOWS_SUGGESTIONS =
         'enable-screen-edges-windows-suggestions';
+    static KEY_ENABLE_MULTI_TILE_WINDOW_SUGGESTIONS =
+        'enable-multi-tile-window-suggestions';
 
     static SETTING_MOVE_WINDOW_RIGHT = 'move-window-right';
     static SETTING_MOVE_WINDOW_LEFT = 'move-window-left';
@@ -468,6 +470,14 @@ export default class Settings {
         set_boolean(Settings.KEY_ENABLE_SCREEN_EDGES_WINDOWS_SUGGESTIONS, val);
     }
 
+    static get ENABLE_MULTI_TILE_WINDOW_SUGGESTIONS(): boolean {
+        return get_boolean(Settings.KEY_ENABLE_MULTI_TILE_WINDOW_SUGGESTIONS);
+    }
+
+    static set ENABLE_MULTI_TILE_WINDOW_SUGGESTIONS(val: boolean) {
+        set_boolean(Settings.KEY_ENABLE_MULTI_TILE_WINDOW_SUGGESTIONS, val);
+    }
+
     static get_inner_gaps(scaleFactor: number = 1): {
         top: number;
         bottom: number;