Only set the window icon if we manage to get a primary window (resolves

#80)
NiklasEi · Dec 3, 2023 · 5e4cb8a · 5e4cb8a
1 parent d786f97
commit 5e4cb8a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main.rs b/src/main.rs
@@ -37,7 +37,9 @@ fn set_window_icon(
     primary_window: Query<Entity, With<PrimaryWindow>>,
 ) {
     let primary_entity = primary_window.single();
-    let primary = windows.get_window(primary_entity).unwrap();
+    let Some(primary) = windows.get_window(primary_entity) else {
+        return;
+    };
     let icon_buf = Cursor::new(include_bytes!(
         "../build/macos/AppIcon.iconset/icon_256x256.png"
     ));