use crate::rdev::{Event, EventType, GrabError};
use crate::windows::common::{convert, set_key_hook, set_mouse_hook, HookError, HOOK, KEYBOARD};
use std::ptr::null_mut;
use std::time::SystemTime;
use winapi::um::winuser::{CallNextHookEx, GetMessageA, HC_ACTION};
static mut GLOBAL_CALLBACK: Option<Box<dyn FnMut(Event) -> Option<Event>>> = None;
unsafe extern "system" fn raw_callback(code: i32, param: usize, lpdata: isize) -> isize {
if code == HC_ACTION {
let opt = convert(param, lpdata);
if let Some(event_type) = opt {
let name = match &event_type {
EventType::KeyPress(_key) => match (*KEYBOARD).lock() {
Ok(mut keyboard) => keyboard.get_name(lpdata),
Err(_) => None,
},
_ => None,
};
let event = Event {
event_type,
time: SystemTime::now(),
name,
};
if let Some(callback) = &mut GLOBAL_CALLBACK {
if callback(event).is_none() {
return 1;
}
}
}
}
CallNextHookEx(HOOK, code, param, lpdata)
}
impl From<HookError> for GrabError {
fn from(error: HookError) -> Self {
match error {
HookError::Mouse(code) => GrabError::MouseHookError(code),
HookError::Key(code) => GrabError::KeyHookError(code),
}
}
}
pub fn grab<T>(callback: T) -> Result<(), GrabError>
where
T: FnMut(Event) -> Option<Event> + 'static,
{
unsafe {
GLOBAL_CALLBACK = Some(Box::new(callback));
set_key_hook(raw_callback)?;
set_mouse_hook(raw_callback)?;
GetMessageA(null_mut(), null_mut(), 0, 0);
}
Ok(())
}