wait-timeout 0.1.2

use std::io;
use std::os::windows::prelude::*;
use std::process::Child;
use std::time::Duration;

use kernel32::*;
use winapi::*;

#[derive(Eq, PartialEq, Copy, Clone, Debug)]
pub struct ExitStatus(DWORD);

pub fn wait_timeout(child: &mut Child, dur: Duration)
                       -> io::Result<Option<ExitStatus>> {
    let ms = dur.as_secs().checked_mul(1000).and_then(|amt| {
        amt.checked_add((dur.subsec_nanos() / 1_000_000) as u64)
    }).expect("failed to convert duration to milliseconds");
    let ms = if ms > (DWORD::max_value() as u64) {
        DWORD::max_value()
    } else {
        ms as DWORD
    };
    unsafe {
        match WaitForSingleObject(child.as_raw_handle(), ms) {
            WAIT_OBJECT_0 => {}
            WAIT_TIMEOUT => return Ok(None),
            _ => return Err(io::Error::last_os_error()),
        }
        let mut status = 0;
        if GetExitCodeProcess(child.as_raw_handle(), &mut status) == FALSE {
            Err(io::Error::last_os_error())
        } else {
            Ok(Some(ExitStatus(status)))
        }
    }
}

impl ExitStatus {
    pub fn success(&self) -> bool { self.code() == Some(0) }
    pub fn code(&self) -> Option<i32> { Some(self.0 as i32) }
    pub fn unix_signal(&self) -> Option<i32> { None }
}