From 4ee5c26e7df7faea0f7dcd01f6027eaa60be2ef3 Mon Sep 17 00:00:00 2001
From: Gregory Schier <gschier1990@gmail.com>
Date: Wed, 15 Jan 2025 06:02:27 -0800
Subject: [PATCH] Update window state logic

---
 src-tauri/src/lib.rs | 45 ++++++++++++++++++++++----------------------
 1 file changed, 23 insertions(+), 22 deletions(-)

diff --git a/src-tauri/src/lib.rs b/src-tauri/src/lib.rs
index 08ea4a3a..28b5de63 100644
--- a/src-tauri/src/lib.rs
+++ b/src-tauri/src/lib.rs
@@ -1951,22 +1951,17 @@ pub fn run() {
                 }
                 _ => {}
             };
+
+            // Save window state on exit
             match event {
-                RunEvent::WindowEvent {
-                    label,
-                    event: WindowEvent::CloseRequested { .. },
-                    ..
-                } => {
-                    // Save window state on close because, for some reason, it doesn't
-                    if !label.starts_with(OTHER_WINDOW_PREFIX) {
-                        let handle = app_handle.clone();
-                        if let Err(e) = handle.save_window_state(StateFlags::all()) {
-                            warn!("Failed to save window state {e:?}");
-                        } else {
-                            debug!("Saved window state");
-                        };
-                    }
-                }
+                RunEvent::ExitRequested{..} => {
+                    let handle = app_handle.clone();
+                    if let Err(e) = handle.save_window_state(StateFlags::all()) {
+                        warn!("Failed to save window state {e:?}");
+                    } else {
+                        debug!("Saved window state");
+                    };
+                },
                 _ => {}
             };
         });
@@ -2005,7 +2000,19 @@ fn create_main_window(handle: &AppHandle, url: &str) -> WebviewWindow {
             100.0 + random::<f64>() * 20.0,
         ),
     };
-    create_window(handle, config)
+
+    let window = create_window(handle, config);
+
+    // Restore window state if it's a main window
+    if !label.starts_with(OTHER_WINDOW_PREFIX) {
+        if let Err(e) = window.restore_state(StateFlags::all()) {
+            warn!("Failed to restore window state {e:?}");
+        } else {
+            debug!("Restored window state");
+        }
+    }
+
+    window
 }
 
 struct CreateWindowConfig<'s> {
@@ -2099,12 +2106,6 @@ fn create_window(handle: &AppHandle, config: CreateWindowConfig) -> WebviewWindo
         }
     });
 
-    if let Err(e) = win.restore_state(StateFlags::all()) {
-        warn!("Failed to restore window state {e:?}");
-    } else {
-        debug!("Restored window state");
-    }
-
     win
 }