From c5b9bd361227e3df2fe58b80c87044e8507a7942 Mon Sep 17 00:00:00 2001
From: mitchmindtree <mitchell.nordine@gmail.com>
Date: Fri, 9 Jun 2017 19:40:22 +1000
Subject: [PATCH] Update macos backend to addition of ControlFlow (untested)

---
 src/platform/macos/events_loop.rs | 36 ++++++++++++++++---------------
 1 file changed, 19 insertions(+), 17 deletions(-)

diff --git a/src/platform/macos/events_loop.rs b/src/platform/macos/events_loop.rs
index 12fe2c0e..bed3df56 100644
--- a/src/platform/macos/events_loop.rs
+++ b/src/platform/macos/events_loop.rs
@@ -1,4 +1,4 @@
-use EventsLoopClosed;
+use {ControlFlow, EventsLoopClosed};
 use cocoa::{self, appkit, foundation};
 use cocoa::appkit::{NSApplication, NSEvent, NSView, NSWindow};
 use events::{self, ElementState, Event, MouseButton, TouchPhase, WindowEvent, ModifiersState, KeyboardInput};
@@ -11,7 +11,6 @@ pub struct EventsLoop {
     pub windows: std::sync::Mutex<Vec<std::sync::Weak<Window>>>,
     pub pending_events: std::sync::Mutex<std::collections::VecDeque<Event>>,
     modifiers: std::sync::Mutex<Modifiers>,
-    interrupted: std::sync::atomic::AtomicBool,
 
     // The user event callback given via either of the `poll_events` or `run_forever` methods.
     //
@@ -102,12 +101,11 @@ impl EventsLoop {
             windows: std::sync::Mutex::new(Vec::new()),
             pending_events: std::sync::Mutex::new(std::collections::VecDeque::new()),
             modifiers: std::sync::Mutex::new(modifiers),
-            interrupted: std::sync::atomic::AtomicBool::new(false),
             user_callback: UserCallback { mutex: std::sync::Mutex::new(None) },
         }
     }
 
-    pub fn poll_events<F>(&self, mut callback: F)
+    pub fn poll_events<F>(&mut self, mut callback: F)
         where F: FnMut(Event),
     {
         unsafe {
@@ -148,23 +146,33 @@ impl EventsLoop {
         self.user_callback.drop();
     }
 
-    pub fn run_forever<F>(&self, mut callback: F)
-        where F: FnMut(Event)
+    pub fn run_forever<F>(&mut self, mut callback: F)
+        where F: FnMut(Event) -> ControlFlow
     {
-        self.interrupted.store(false, std::sync::atomic::Ordering::Relaxed);
-
         unsafe {
             if !msg_send![cocoa::base::class("NSThread"), isMainThread] {
                 panic!("Events can only be polled from the main thread on macOS");
             }
         }
 
+        // Track whether or not control flow has changed.
+        let mut control_flow = std::cell::Cell::new(ControlFlow::Continue);
+
+        let mut callback = |event| {
+            if let ControlFlow::Complete = callback(event) {
+                control_flow.set(ControlFlow::Complete);
+            }
+        };
+
         self.user_callback.store(&mut callback);
 
         loop {
             unsafe {
                 // First, yield all pending events.
                 self.call_user_callback_with_pending_events();
+                if let ControlFlow::Complete = control_flow.get() {
+                    break;
+                }
 
                 let pool = foundation::NSAutoreleasePool::new(cocoa::base::nil);
 
@@ -183,22 +191,16 @@ impl EventsLoop {
 
                 if let Some(event) = maybe_event {
                     self.user_callback.call_with_event(event);
+                    if let ControlFlow::Complete = control_flow.get() {
+                        break;
+                    }
                 }
             }
-
-            if self.interrupted.load(std::sync::atomic::Ordering::Relaxed) {
-                self.interrupted.store(false, std::sync::atomic::Ordering::Relaxed);
-                break;
-            }
         }
 
         self.user_callback.drop();
     }
 
-    pub fn interrupt(&self) {
-        self.interrupted.store(true, std::sync::atomic::Ordering::Relaxed);
-    }
-
     // Removes the window with the given `Id` from the `windows` list.
     //
     // This is called when a window is either `Closed` or `Drop`ped.