Fix events handling on win32

src/win32/ffi.rs

@@ -381,6 +381,10 @@ extern "system" {
     // http://msdn.microsoft.com/en-us/library/windows/desktop/aa366730(v=vs.85).aspx
     pub fn LocalFree(hMem: HLOCAL) -> HLOCAL;
 
+    // http://msdn.microsoft.com/en-us/library/windows/desktop/ms644943(v=vs.85).aspx
+    pub fn PeekMessageW(lpMsg: *mut MSG, hWnd: HWND, wMsgFilterMin: UINT, wMsgFilterMax: UINT,
+        wRemoveMsg: UINT) -> BOOL;
+
     // http://msdn.microsoft.com/en-us/library/windows/desktop/ms644945(v=vs.85).aspx
     pub fn PostQuitMessage(nExitCode: libc::c_int);
 

src/win32/mod.rs

@@ -198,13 +198,35 @@ impl Window {
 
     // TODO: return iterator
     pub fn poll_events(&self) -> Vec<Event> {
-        unimplemented!()
+        use std::mem;
+
+        loop {
+            let mut msg = unsafe { mem::uninitialized() };
+
+            if unsafe { ffi::PeekMessageW(&mut msg, ptr::mut_null(), 0, 0, 1) } == 0 {
+                break
+            }
+
+            unsafe { ffi::TranslateMessage(&msg) };
+            unsafe { ffi::DispatchMessageW(&msg) };
+        }
+
+        let mut events = Vec::new();
+        loop {
+            match self.events_receiver.try_recv() {
+                Ok(ev) => events.push(ev),
+                Err(_) => break
+            }
+        }
+        
+        events
     }
 
     // TODO: return iterator
     pub fn wait_events(&self) -> Vec<Event> {
         use std::mem;
 
+        loop {
             {
                 let mut msg = unsafe { mem::uninitialized() };
 
@@ -227,7 +249,11 @@ impl Window {
                     Err(_) => break
                 }
             }
-        events
+            
+            if events.len() >= 1 {
+                return events
+            }
+        }
     }
 
     pub fn make_current(&self) {
@@ -276,8 +302,8 @@ extern "stdcall" fn callback(window: ffi::HWND, msg: ffi::UINT,
     match msg {
         ffi::WM_DESTROY => {
             use Closed;
-            send_event(window, Closed);
             unsafe { ffi::PostQuitMessage(0); }
+            send_event(window, Closed);
             0
         },