From 35331d0a41a53b8ff5c642b8274c7377ae6c6182 Mon Sep 17 00:00:00 2001
From: Richard <richardsoncusto@gmail.com>
Date: Tue, 26 Jul 2022 16:46:12 -0300
Subject: Allow closing the window from user code

---
 winit/src/multi_window.rs | 77 +++++++++++++++++++++++++++++++++++------------
 1 file changed, 57 insertions(+), 20 deletions(-)

(limited to 'winit/src/multi_window.rs')

diff --git a/winit/src/multi_window.rs b/winit/src/multi_window.rs
index dc00d737..3c720a69 100644
--- a/winit/src/multi_window.rs
+++ b/winit/src/multi_window.rs
@@ -39,6 +39,8 @@ pub enum Event<Message> {
     // (maybe we should also allow users to listen/react to those internal messages?)
     NewWindow(window::Id, settings::Window),
     /// TODO(derezzedex)
+    CloseWindow(window::Id),
+    /// TODO(derezzedex)
     WindowCreated(window::Id, winit::window::Window),
 }
 
@@ -549,6 +551,27 @@ async fn run_instance<A, E, C>(
                     let _ = window_ids.insert(window.id(), id);
                     let _ = windows.insert(id, window);
                 }
+                Event::CloseWindow(id) => {
+                    // TODO(derezzedex): log errors
+                    if let Some(window) = windows.get(&id) {
+                        if window_ids.remove(&window.id()).is_none() {
+                            println!("Failed to remove from `window_ids`!");
+                        }
+                    }
+                    if states.remove(&id).is_none() {
+                        println!("Failed to remove from `states`!")
+                    }
+                    if interfaces.remove(&id).is_none() {
+                        println!("Failed to remove from `interfaces`!");
+                    }
+                    if windows.remove(&id).is_none() {
+                        println!("Failed to remove from `windows`!")
+                    }
+
+                    if windows.is_empty() {
+                        break 'main;
+                    }
+                }
                 Event::NewWindow(_, _) => unreachable!(),
             },
             event::Event::RedrawRequested(id) => {
@@ -651,29 +674,43 @@ async fn run_instance<A, E, C>(
                 window_id,
             } => {
                 // dbg!(window_id);
-                let window = window_ids
-                    .get(&window_id)
-                    .and_then(|id| windows.get(id))
-                    .unwrap();
-                let window_state = window_ids
-                    .get(&window_id)
-                    .and_then(|id| states.get_mut(id))
-                    .unwrap();
-
-                if requests_exit(&window_event, window_state.state.modifiers())
-                    && exit_on_close_request
+                if let Some(window) =
+                    window_ids.get(&window_id).and_then(|id| windows.get(id))
                 {
-                    break;
-                }
+                    if let Some(window_state) = window_ids
+                        .get(&window_id)
+                        .and_then(|id| states.get_mut(id))
+                    {
+                        if requests_exit(
+                            &window_event,
+                            window_state.state.modifiers(),
+                        ) && exit_on_close_request
+                        {
+                            break;
+                        }
 
-                window_state.state.update(window, &window_event, &mut debug);
+                        window_state.state.update(
+                            window,
+                            &window_event,
+                            &mut debug,
+                        );
 
-                if let Some(event) = conversion::window_event(
-                    &window_event,
-                    window_state.state.scale_factor(),
-                    window_state.state.modifiers(),
-                ) {
-                    events.push((window_ids.get(&window_id).cloned(), event));
+                        if let Some(event) = conversion::window_event(
+                            &window_event,
+                            window_state.state.scale_factor(),
+                            window_state.state.modifiers(),
+                        ) {
+                            events.push((
+                                window_ids.get(&window_id).cloned(),
+                                event,
+                            ));
+                        }
+                    } else {
+                        // TODO(derezzedex): log error
+                    }
+                } else {
+                    // TODO(derezzedex): log error
+                    // println!("{:?}: {:?}", window_id, window_event);
                 }
             }
             _ => {}
-- 
cgit