use libc; use crate::{Key, KeyboardControllable, MouseButton, MouseControllable}; use self::libc::{c_char, c_int, c_void, useconds_t}; use std::{borrow::Cow, ffi::CString, ptr}; const CURRENT_WINDOW: c_int = 0; const DEFAULT_DELAY: u64 = 12000; type Window = c_int; type Xdo = *const c_void; #[link(name = "xdo")] extern "C" { fn xdo_free(xdo: Xdo); fn xdo_new(display: *const c_char) -> Xdo; fn xdo_click_window(xdo: Xdo, window: Window, button: c_int) -> c_int; fn xdo_mouse_down(xdo: Xdo, window: Window, button: c_int) -> c_int; fn xdo_mouse_up(xdo: Xdo, window: Window, button: c_int) -> c_int; fn xdo_move_mouse(xdo: Xdo, x: c_int, y: c_int, screen: c_int) -> c_int; fn xdo_move_mouse_relative(xdo: Xdo, x: c_int, y: c_int) -> c_int; fn xdo_enter_text_window( xdo: Xdo, window: Window, string: *const c_char, delay: useconds_t, ) -> c_int; fn xdo_send_keysequence_window( xdo: Xdo, window: Window, string: *const c_char, delay: useconds_t, ) -> c_int; fn xdo_send_keysequence_window_down( xdo: Xdo, window: Window, string: *const c_char, delay: useconds_t, ) -> c_int; fn xdo_send_keysequence_window_up( xdo: Xdo, window: Window, string: *const c_char, delay: useconds_t, ) -> c_int; fn xdo_get_input_state(xdo: Xdo) -> u32; } fn mousebutton(button: MouseButton) -> c_int { match button { MouseButton::Left => 1, MouseButton::Middle => 2, MouseButton::Right => 3, MouseButton::ScrollUp => 4, MouseButton::ScrollDown => 5, MouseButton::ScrollLeft => 6, MouseButton::ScrollRight => 7, } } /// The main struct for handling the event emitting pub struct Enigo { xdo: Xdo, delay: u64, } // This is safe, we have a unique pointer. // TODO: use Unique once stable. unsafe impl Send for Enigo {} impl Default for Enigo { /// Create a new Enigo instance fn default() -> Self { Self { xdo: unsafe { xdo_new(ptr::null()) }, delay: DEFAULT_DELAY, } } } impl Enigo { /// Get the delay per keypress. /// Default value is 12000. /// This is Linux-specific. pub fn delay(&self) -> u64 { self.delay } /// Set the delay per keypress. /// This is Linux-specific. pub fn set_delay(&mut self, delay: u64) { self.delay = delay; } } impl Drop for Enigo { fn drop(&mut self) { if self.xdo.is_null() { return; } unsafe { xdo_free(self.xdo); } } } impl MouseControllable for Enigo { fn mouse_move_to(&mut self, x: i32, y: i32) { if self.xdo.is_null() { return; } unsafe { xdo_move_mouse(self.xdo, x as c_int, y as c_int, 0); } } fn mouse_move_relative(&mut self, x: i32, y: i32) { if self.xdo.is_null() { return; } unsafe { xdo_move_mouse_relative(self.xdo, x as c_int, y as c_int); } } fn mouse_down(&mut self, button: MouseButton) -> crate::ResultType { if self.xdo.is_null() { return Ok(()); } unsafe { xdo_mouse_down(self.xdo, CURRENT_WINDOW, mousebutton(button)); } Ok(()) } fn mouse_up(&mut self, button: MouseButton) { if self.xdo.is_null() { return; } unsafe { xdo_mouse_up(self.xdo, CURRENT_WINDOW, mousebutton(button)); } } fn mouse_click(&mut self, button: MouseButton) { if self.xdo.is_null() { return; } unsafe { xdo_click_window(self.xdo, CURRENT_WINDOW, mousebutton(button)); } } fn mouse_scroll_x(&mut self, length: i32) { let button; let mut length = length; if length < 0 { button = MouseButton::ScrollLeft; } else { button = MouseButton::ScrollRight; } if length < 0 { length = -length; } for _ in 0..length { self.mouse_click(button); } } fn mouse_scroll_y(&mut self, length: i32) { let button; let mut length = length; if length < 0 { button = MouseButton::ScrollUp; } else { button = MouseButton::ScrollDown; } if length < 0 { length = -length; } for _ in 0..length { self.mouse_click(button); } } } fn keysequence<'a>(key: Key) -> Cow<'a, str> { if let Key::Layout(c) = key { return Cow::Owned(format!("U{:X}", c as u32)); } if let Key::Raw(k) = key { return Cow::Owned(format!("{}", k as u16)); } #[allow(deprecated)] // I mean duh, we still need to support deprecated keys until they're removed // https://www.rubydoc.info/gems/xdo/XDo/Keyboard // https://gitlab.com/cunidev/gestures/-/wikis/xdotool-list-of-key-codes Cow::Borrowed(match key { Key::Alt => "Alt", Key::Backspace => "BackSpace", Key::CapsLock => "Caps_Lock", Key::Control => "Control", Key::Delete => "Delete", Key::DownArrow => "Down", Key::End => "End", Key::Escape => "Escape", Key::F1 => "F1", Key::F10 => "F10", Key::F11 => "F11", Key::F12 => "F12", Key::F2 => "F2", Key::F3 => "F3", Key::F4 => "F4", Key::F5 => "F5", Key::F6 => "F6", Key::F7 => "F7", Key::F8 => "F8", Key::F9 => "F9", Key::Home => "Home", //Key::Layout(_) => unreachable!(), Key::LeftArrow => "Left", Key::Option => "Option", Key::PageDown => "Page_Down", Key::PageUp => "Page_Up", //Key::Raw(_) => unreachable!(), Key::Return => "Return", Key::RightArrow => "Right", Key::Shift => "Shift", Key::Space => "space", Key::Tab => "Tab", Key::UpArrow => "Up", Key::Numpad0 => "U30", //"KP_0", Key::Numpad1 => "U31", //"KP_1", Key::Numpad2 => "U32", //"KP_2", Key::Numpad3 => "U33", //"KP_3", Key::Numpad4 => "U34", //"KP_4", Key::Numpad5 => "U35", //"KP_5", Key::Numpad6 => "U36", //"KP_6", Key::Numpad7 => "U37", //"KP_7", Key::Numpad8 => "U38", //"KP_8", Key::Numpad9 => "U39", //"KP_9", Key::Decimal => "U2E", //"KP_Decimal", Key::Cancel => "Cancel", Key::Clear => "Clear", Key::Menu => "Menu", Key::Pause => "Pause", Key::Kana => "Kana", Key::Hangul => "Hangul", Key::Junja => "", Key::Final => "", Key::Hanja => "Hanja", Key::Kanji => "Kanji", Key::Convert => "", Key::Select => "Select", Key::Print => "Print", Key::Execute => "Execute", Key::Snapshot => "3270_PrintScreen", Key::Insert => "Insert", Key::Help => "Help", Key::Sleep => "", Key::Separator => "KP_Separator", Key::VolumeUp => "", Key::VolumeDown => "", Key::Mute => "", Key::Scroll => "Scroll_Lock", Key::NumLock => "Num_Lock", Key::RWin => "Super_R", Key::Apps => "", Key::Multiply => "KP_Multiply", Key::Add => "KP_Add", Key::Subtract => "KP_Subtract", Key::Divide => "KP_Divide", Key::Equals => "KP_Equal", Key::NumpadEnter => "KP_Enter", Key::RightShift => "Shift_R", Key::RightControl => "Control_R", Key::RightAlt => "Alt_R", Key::Command | Key::Super | Key::Windows | Key::Meta => "Super", _ => "", }) } impl KeyboardControllable for Enigo { fn get_key_state(&mut self, key: Key) -> bool { if self.xdo.is_null() { return false; } let mod_shift = 1 << 0; let mod_lock = 1 << 1; let mod_control = 1 << 2; let mod_alt = 1 << 3; let mod_numlock = 1 << 4; let mod_meta = 1 << 6; let mask = unsafe { xdo_get_input_state(self.xdo) }; // println!("{:b}", mask); match key { Key::Shift => mask & mod_shift != 0, Key::CapsLock => mask & mod_lock != 0, Key::Control => mask & mod_control != 0, Key::Alt => mask & mod_alt != 0, Key::NumLock => mask & mod_numlock != 0, Key::Meta => mask & mod_meta != 0, _ => false, } } fn key_sequence(&mut self, sequence: &str) { if self.xdo.is_null() { return; } if let Ok(string) = CString::new(sequence) { unsafe { xdo_enter_text_window( self.xdo, CURRENT_WINDOW, string.as_ptr(), self.delay as useconds_t, ); } } } fn key_down(&mut self, key: Key) -> crate::ResultType { if self.xdo.is_null() { return Ok(()); } let string = CString::new(&*keysequence(key))?; unsafe { xdo_send_keysequence_window_down( self.xdo, CURRENT_WINDOW, string.as_ptr(), self.delay as useconds_t, ); } Ok(()) } fn key_up(&mut self, key: Key) { if self.xdo.is_null() { return; } if let Ok(string) = CString::new(&*keysequence(key)) { unsafe { xdo_send_keysequence_window_up( self.xdo, CURRENT_WINDOW, string.as_ptr(), self.delay as useconds_t, ); } } } fn key_click(&mut self, key: Key) { if self.xdo.is_null() { return; } if let Ok(string) = CString::new(&*keysequence(key)) { unsafe { xdo_send_keysequence_window( self.xdo, CURRENT_WINDOW, string.as_ptr(), self.delay as useconds_t, ); } } } }