layout/floating: allow size altering if window was maximized

Author: thegabriele97

src/shell/element/surface.rs

@@ -103,6 +103,9 @@ impl PartialEq<X11Surface> for CosmicSurface {
     }
 }
 
+#[derive(Default)]
+struct WasMaximized(AtomicBool);
+
 #[derive(Default)]
 struct Minimized(AtomicBool);
 
@@ -399,6 +402,16 @@ impl CosmicSurface {
     }
 
     pub fn set_maximized(&self, maximized: bool) {
+        let was_maximized = self.is_maximized(false);
+
+        // update was_maximized flag
+        self.0
+            .user_data()
+            .get_or_insert_threadsafe(WasMaximized::default)
+            .0
+            .store(was_maximized && !maximized, Ordering::SeqCst);
+
+
         match self.0.underlying_surface() {
             WindowSurface::Wayland(toplevel) => toplevel.with_pending_state(|state| {
                 if maximized {
@@ -436,6 +449,14 @@ impl CosmicSurface {
         }
     }
 
+    pub fn was_maximized(&self) -> bool {
+        self.0
+            .user_data()
+            .get_or_insert_threadsafe(WasMaximized::default)
+            .0
+            .load(Ordering::SeqCst)
+    }
+
     pub fn is_sticky(&self) -> bool {
         self.0
             .user_data()

src/shell/layout/floating/mod.rs

@@ -686,11 +686,11 @@ impl FloatingLayout {
                 mapped_geometry.size = last_size;
             }
         } else if !window.is_maximized(true) {
-            if window.active_window().has_pending_changes() {
+            if window.active_window().has_pending_changes() || window.active_window().was_maximized() {
                 if let Some(pending_size) = window.pending_size() {
                     mapped_geometry.size = pending_size.as_local();
                 }
-            }
+            }  
             *window.last_geometry.lock().unwrap() = Some(mapped_geometry);
         }