rustdesk/libs/enigo/src/macos/macos_impl.rs

use core_graphics;
// TODO(dustin): use only the things i need

use self::core_graphics::display::*;
use self::core_graphics::event::*;
use self::core_graphics::event_source::*;
use std::collections::HashMap as Map;
use std::ffi::c_void;
use std::ffi::CStr;
use std::os::raw::*;
use std::ptr::null_mut;

use crate::macos::keycodes::*;
use crate::{Key, KeyboardControllable, MouseButton, MouseControllable};
use objc::runtime::Class;

struct MyCGEvent;
type TISInputSourceRef = *mut c_void;
type CFDataRef = *const c_void;
type OptionBits = u32;
type OSStatus = i32;
type UniChar = u16;
type UniCharCount = usize;
type Boolean = c_uchar;
type CFStringEncoding = u32;

#[repr(C)]
#[derive(Debug, Copy, Clone)]
struct __CFString([u8; 0]);
type CFStringRef = *const __CFString;

#[allow(non_upper_case_globals)]
const kCFStringEncodingUTF8: u32 = 134_217_984;
#[allow(non_upper_case_globals)]
const kUCKeyActionDisplay: u16 = 3;
#[allow(non_upper_case_globals)]
const kUCKeyTranslateDeadKeysBit: OptionBits = 1 << 31;
const BUF_LEN: usize = 4;

const MOUSE_EVENT_BUTTON_NUMBER_BACK: i64 = 3;
const MOUSE_EVENT_BUTTON_NUMBER_FORWARD: i64 = 4;

/// The event source user data value of cgevent.
pub const ENIGO_INPUT_EXTRA_VALUE: i64 = 100;

#[allow(improper_ctypes)]
#[allow(non_snake_case)]
#[link(name = "ApplicationServices", kind = "framework")]
#[link(name = "Carbon", kind = "framework")]
extern "C" {
    fn CFDataGetBytePtr(theData: CFDataRef) -> *const u8;
    fn TISCopyCurrentKeyboardInputSource() -> TISInputSourceRef;
    fn TISCopyCurrentKeyboardLayoutInputSource() -> TISInputSourceRef;
    fn TISCopyCurrentASCIICapableKeyboardLayoutInputSource() -> TISInputSourceRef;
    static kTISPropertyUnicodeKeyLayoutData: *mut c_void;
    static kTISPropertyInputSourceID: *mut c_void;
    fn UCKeyTranslate(
        keyLayoutPtr: *const u8, //*const UCKeyboardLayout,
        virtualKeyCode: u16,
        keyAction: u16,
        modifierKeyState: u32,
        keyboardType: u32,
        keyTranslateOptions: OptionBits,
        deadKeyState: *mut u32,
        maxStringLength: UniCharCount,
        actualStringLength: *mut UniCharCount,
        unicodeString: *mut [UniChar; BUF_LEN],
    ) -> OSStatus;
    fn LMGetKbdType() -> u8;
    fn CFStringGetCString(
        theString: CFStringRef,
        buffer: *mut c_char,
        bufferSize: CFIndex,
        encoding: CFStringEncoding,
    ) -> Boolean;

    fn CGEventPost(tapLocation: CGEventTapLocation, event: *mut MyCGEvent);
    // Actually return CFDataRef which is const here, but for coding convenience, return *mut c_void
    fn TISGetInputSourceProperty(source: TISInputSourceRef, property: *const c_void)
        -> *mut c_void;
    // not present in servo/core-graphics
    fn CGEventCreateScrollWheelEvent(
        source: &CGEventSourceRef,
        units: ScrollUnit,
        wheelCount: u32,
        wheel1: i32,
        ...
    ) -> *mut MyCGEvent;
    fn CGEventSourceKeyState(stateID: i32, key: u16) -> bool;
}

#[repr(C)]
#[derive(Clone, Copy)]
struct NSPoint {
    x: f64,
    y: f64,
}

// not present in servo/core-graphics
#[allow(dead_code)]
#[derive(Debug)]
enum ScrollUnit {
    Pixel = 0,
    Line = 1,
}
// hack

/// The main struct for handling the event emitting
pub struct Enigo {
    event_source: Option<CGEventSource>,
    double_click_interval: u32,
    last_click_time: Option<std::time::Instant>,
    multiple_click: i64,
    ignore_flags: bool,
    flags: CGEventFlags,
    char_to_vkey_map: Map<String, Map<char, CGKeyCode>>,
}

impl Enigo {
    /// Set if ignore flags when posting events.
    pub fn set_ignore_flags(&mut self, ignore: bool) {
        self.ignore_flags = ignore;
    }

    ///
    pub fn reset_flag(&mut self) {
        self.flags = CGEventFlags::CGEventFlagNull;
    }

    ///
    pub fn add_flag(&mut self, key: &Key) {
        let flag = match key {
            &Key::CapsLock => CGEventFlags::CGEventFlagAlphaShift,
            &Key::Shift => CGEventFlags::CGEventFlagShift,
            &Key::Control => CGEventFlags::CGEventFlagControl,
            &Key::Alt => CGEventFlags::CGEventFlagAlternate,
            &Key::Meta => CGEventFlags::CGEventFlagCommand,
            &Key::NumLock => CGEventFlags::CGEventFlagNumericPad,
            _ => CGEventFlags::CGEventFlagNull,
        };
        self.flags |= flag;
    }

    // Just check F11 for minimal changes.
    // Since enigo (legacy mode) is deprecated, it is currently in maintenance only.
    fn post(&self, event: CGEvent, keycode: Option<u16>) {
        if keycode == Some(kVK_F11) {
            // Some key events require the flags to work.
            // We can't simply set the flag to `CGEventFlags::CGEventFlagNull`.
            // eg. `F11` requires flags `CGEventFlags::CGEventFlagSecondaryFn | 0x20000000` to work.
            self.post_event(event, false);
        } else {
            // macOS system may use the previous event flag to generate the next event.
            // Only found this issue when locking the screen.
            // When we use enigo to lock the screen, the next mouse event will have the flag
            // `CGEventFlagControl | CGEventFlagCommand | 0x20000000`.
            // The key event will also have the flag `CGEventFlagControl | CGEventFlagCommand | 0x20000000`.
            // Therefore, we need to set the flag to `event.set_flags(self.flags)` to avoid this.
            self.post_event(event, true);
        }
    }

    fn post_event(&self, event: CGEvent, force_flags: bool) {
        if !self.ignore_flags && (force_flags || self.flags != CGEventFlags::CGEventFlagNull) {
            event.set_flags(self.flags);
        }
        event.set_integer_value_field(EventField::EVENT_SOURCE_USER_DATA, ENIGO_INPUT_EXTRA_VALUE);
        event.post(CGEventTapLocation::HID);
    }
}

impl Default for Enigo {
    fn default() -> Self {
        let mut double_click_interval = 500;
        if let Some(ns_event) = Class::get("NSEvent") {
            let tm: f64 = unsafe { msg_send![ns_event, doubleClickInterval] };
            if tm > 0. {
                double_click_interval = (tm * 1000.) as u32;
                log::info!("double click interval: {}ms", double_click_interval);
            }
        }
        Self {
            // TODO(dustin): return error rather than panic here
            event_source: if let Ok(src) =
                CGEventSource::new(CGEventSourceStateID::CombinedSessionState)
            {
                Some(src)
            } else {
                None
            },
            double_click_interval,
            multiple_click: 1,
            last_click_time: None,
            ignore_flags: false,
            flags: CGEventFlags::CGEventFlagNull,
            char_to_vkey_map: Default::default(),
        }
    }
}

impl MouseControllable for Enigo {
    fn as_any(&self) -> &dyn std::any::Any {
        self
    }

    fn as_mut_any(&mut self) -> &mut dyn std::any::Any {
        self
    }

    fn mouse_move_to(&mut self, x: i32, y: i32) {
        // For absolute movement, we don't set delta values
        // This maintains backward compatibility
        self.mouse_move_to_impl(x, y, None);
    }

    fn mouse_move_relative(&mut self, x: i32, y: i32) {
        let (display_width, display_height) = Self::main_display_size();
        let (current_x, y_inv) = Self::mouse_location_raw_coords();
        let current_y = (display_height as i32) - y_inv;
        // Use saturating arithmetic to prevent overflow/wraparound
        let mut new_x = current_x.saturating_add(x);
        let mut new_y = current_y.saturating_add(y);

        // Define screen center and edge margins for cursor reset
        let center_x = (display_width / 2) as i32;
        let center_y = (display_height / 2) as i32;
        // Margin calculation: 5% of the smaller screen dimension with a minimum of 50px.
        // This provides a comfortable buffer zone to detect when the cursor is approaching
        // screen edges, allowing us to reset it to center before it hits the boundary.
        // This ensures continuous relative mouse movement without getting stuck at edges.
        let margin = (display_width.min(display_height) / 20).max(50) as i32;

        // Check if cursor is approaching screen boundaries
        // Use saturating_sub to prevent negative thresholds on very small displays
        let right = (display_width as i32).saturating_sub(margin);
        let bottom = (display_height as i32).saturating_sub(margin);
        let near_edge = new_x < margin
            || new_x > right
            || new_y < margin
            || new_y > bottom;

        if near_edge {
            // Reset cursor to screen center to allow continuous movement
            // The delta values are still passed correctly for games/apps
            new_x = center_x;
            new_y = center_y;
        }

        // Clamp to screen bounds as a safety measure.
        // Use saturating_sub(1) to ensure coordinates don't exceed the last valid pixel.
        let max_x = (display_width as i32).saturating_sub(1).max(0);
        let max_y = (display_height as i32).saturating_sub(1).max(0);
        new_x = new_x.clamp(0, max_x);
        new_y = new_y.clamp(0, max_y);

        // Pass delta values for relative movement
        // This is critical for browser Pointer Lock API support
        // The delta fields (MOUSE_EVENT_DELTA_X/Y) are used by browsers
        // to calculate movementX/Y in Pointer Lock mode
        self.mouse_move_to_impl(new_x, new_y, Some((x, y)));
    }

    fn mouse_down(&mut self, button: MouseButton) -> crate::ResultType {
        let now = std::time::Instant::now();
        if let Some(t) = self.last_click_time {
            if t.elapsed().as_millis() as u32 <= self.double_click_interval {
                self.multiple_click += 1;
            } else {
                self.multiple_click = 1;
            }
        }
        self.last_click_time = Some(now);
        let (current_x, current_y) = Self::mouse_location();
        let (button, event_type, btn_value) = match button {
            MouseButton::Left => (CGMouseButton::Left, CGEventType::LeftMouseDown, None),
            MouseButton::Middle => (CGMouseButton::Center, CGEventType::OtherMouseDown, None),
            MouseButton::Right => (CGMouseButton::Right, CGEventType::RightMouseDown, None),
            MouseButton::Back => (
                CGMouseButton::Left,
                CGEventType::OtherMouseDown,
                Some(MOUSE_EVENT_BUTTON_NUMBER_BACK),
            ),
            MouseButton::Forward => (
                CGMouseButton::Left,
                CGEventType::OtherMouseDown,
                Some(MOUSE_EVENT_BUTTON_NUMBER_FORWARD),
            ),
            _ => {
                log::info!("Unsupported button {:?}", button);
                return Ok(());
            }
        };
        let dest = CGPoint::new(current_x as f64, current_y as f64);
        if let Some(src) = self.event_source.as_ref() {
            if let Ok(event) = CGEvent::new_mouse_event(src.clone(), event_type, dest, button) {
                if self.multiple_click > 1 {
                    event.set_integer_value_field(
                        EventField::MOUSE_EVENT_CLICK_STATE,
                        self.multiple_click,
                    );
                }
                if let Some(v) = btn_value {
                    event.set_integer_value_field(EventField::MOUSE_EVENT_BUTTON_NUMBER, v);
                }
                self.post(event, None);
            }
        }
        Ok(())
    }

    fn mouse_up(&mut self, button: MouseButton) {
        let (current_x, current_y) = Self::mouse_location();
        let (button, event_type, btn_value) = match button {
            MouseButton::Left => (CGMouseButton::Left, CGEventType::LeftMouseUp, None),
            MouseButton::Middle => (CGMouseButton::Center, CGEventType::OtherMouseUp, None),
            MouseButton::Right => (CGMouseButton::Right, CGEventType::RightMouseUp, None),
            MouseButton::Back => (
                CGMouseButton::Left,
                CGEventType::OtherMouseUp,
                Some(MOUSE_EVENT_BUTTON_NUMBER_BACK),
            ),
            MouseButton::Forward => (
                CGMouseButton::Left,
                CGEventType::OtherMouseUp,
                Some(MOUSE_EVENT_BUTTON_NUMBER_FORWARD),
            ),
            _ => {
                log::info!("Unsupported button {:?}", button);
                return;
            }
        };
        let dest = CGPoint::new(current_x as f64, current_y as f64);
        if let Some(src) = self.event_source.as_ref() {
            if let Ok(event) = CGEvent::new_mouse_event(src.clone(), event_type, dest, button) {
                if self.multiple_click > 1 {
                    event.set_integer_value_field(
                        EventField::MOUSE_EVENT_CLICK_STATE,
                        self.multiple_click,
                    );
                }
                if let Some(v) = btn_value {
                    event.set_integer_value_field(EventField::MOUSE_EVENT_BUTTON_NUMBER, v);
                }
                self.post(event, None);
            }
        }
    }

    fn mouse_click(&mut self, button: MouseButton) {
        self.mouse_down(button).ok();
        self.mouse_up(button);
    }

    fn mouse_scroll_x(&mut self, length: i32) {
        let mut scroll_direction = -1; // 1 left -1 right;
        let mut length = length;

        if length < 0 {
            length *= -1;
            scroll_direction *= -1;
        }

        if let Some(src) = self.event_source.as_ref() {
            for _ in 0..length {
                unsafe {
                    let mouse_ev = CGEventCreateScrollWheelEvent(
                        &src,
                        ScrollUnit::Line,
                        2, // CGWheelCount 1 = y 2 = xy 3 = xyz
                        0,
                        scroll_direction,
                    );

                    CGEventPost(CGEventTapLocation::HID, mouse_ev);
                    CFRelease(mouse_ev as *const std::ffi::c_void);
                }
            }
        }
    }

    fn mouse_scroll_y(&mut self, length: i32) {
        let mut scroll_direction = -1; // 1 left -1 right;
        let mut length = length;

        if length < 0 {
            length *= -1;
            scroll_direction *= -1;
        }

        if let Some(src) = self.event_source.as_ref() {
            for _ in 0..length {
                unsafe {
                    let mouse_ev = CGEventCreateScrollWheelEvent(
                        &src,
                        ScrollUnit::Line,
                        1, // CGWheelCount 1 = y 2 = xy 3 = xyz
                        scroll_direction,
                    );

                    CGEventPost(CGEventTapLocation::HID, mouse_ev);
                    CFRelease(mouse_ev as *const std::ffi::c_void);
                }
            }
        }
    }
}

// https://stackoverflow.
// com/questions/1918841/how-to-convert-ascii-character-to-cgkeycode

impl KeyboardControllable for Enigo {
    fn as_any(&self) -> &dyn std::any::Any {
        self
    }

    fn as_mut_any(&mut self) -> &mut dyn std::any::Any {
        self
    }

    fn key_sequence(&mut self, sequence: &str) {
        // NOTE(dustin): This is a fix for issue https://github.com/enigo-rs/enigo/issues/68
        // TODO(dustin): This could be improved by aggregating 20 bytes worth of graphemes at a time
        // but i am unsure what would happen for grapheme clusters greater than 20 bytes ...
        use unicode_segmentation::UnicodeSegmentation;
        let clusters = UnicodeSegmentation::graphemes(sequence, true).collect::<Vec<&str>>();
        for cluster in clusters {
            if let Some(src) = self.event_source.as_ref() {
                if let Ok(event) = CGEvent::new_keyboard_event(src.clone(), 0, true) {
                    event.set_string(cluster);
                    self.post(event, None);
                }
            }
        }
    }

    fn key_click(&mut self, key: Key) {
        let keycode = self.key_to_keycode(key);
        if keycode == u16::MAX {
            return;
        }

        if let Some(src) = self.event_source.as_ref() {
            if let Ok(event) = CGEvent::new_keyboard_event(src.clone(), keycode, true) {
                self.post(event, Some(keycode));
            }

            if let Ok(event) = CGEvent::new_keyboard_event(src.clone(), keycode, false) {
                self.post(event, Some(keycode));
            }
        }
    }

    fn key_down(&mut self, key: Key) -> crate::ResultType {
        let code = self.key_to_keycode(key);
        if code == u16::MAX {
            return Err("".into());
        }
        if let Some(src) = self.event_source.as_ref() {
            if let Ok(event) = CGEvent::new_keyboard_event(src.clone(), code, true) {
                self.post(event, Some(code));
            }
        }
        Ok(())
    }

    fn key_up(&mut self, key: Key) {
        let code = self.key_to_keycode(key);
        if let Some(src) = self.event_source.as_ref() {
            if let Ok(event) = CGEvent::new_keyboard_event(src.clone(), code, false) {
                self.post(event, Some(code));
            }
        }
    }

    fn get_key_state(&mut self, key: Key) -> bool {
        let keycode = self.key_to_keycode(key);
        unsafe { CGEventSourceKeyState(1, keycode) }
    }
}

impl Enigo {
    fn pressed_buttons() -> usize {
        if let Some(ns_event) = Class::get("NSEvent") {
            unsafe { msg_send![ns_event, pressedMouseButtons] }
        } else {
            0
        }
    }

    /// Internal implementation for mouse movement with optional delta values.
    ///
    /// The `delta` parameter is crucial for browser Pointer Lock API support.
    /// When a browser enters Pointer Lock mode, it reads mouse delta values
    /// (MOUSE_EVENT_DELTA_X/Y) directly from CGEvent to calculate movementX/Y.
    /// Without setting these fields, the browser sees zero movement.
    fn mouse_move_to_impl(&mut self, x: i32, y: i32, delta: Option<(i32, i32)>) {
        let pressed = Self::pressed_buttons();

        // Determine event type and corresponding mouse button based on pressed buttons.
        // The CGMouseButton must match the event type for drag events.
        let (event_type, button) = if pressed & 1 > 0 {
            (CGEventType::LeftMouseDragged, CGMouseButton::Left)
        } else if pressed & 2 > 0 {
            (CGEventType::RightMouseDragged, CGMouseButton::Right)
        } else if pressed & 4 > 0 {
            (CGEventType::OtherMouseDragged, CGMouseButton::Center)
        } else {
            (CGEventType::MouseMoved, CGMouseButton::Left) // Button doesn't matter for MouseMoved
        };

        let dest = CGPoint::new(x as f64, y as f64);
        if let Some(src) = self.event_source.as_ref() {
            if let Ok(event) =
                CGEvent::new_mouse_event(src.clone(), event_type, dest, button)
            {
                // Set delta fields for relative mouse movement
                // This is essential for Pointer Lock API in browsers
                if let Some((dx, dy)) = delta {
                    event.set_integer_value_field(EventField::MOUSE_EVENT_DELTA_X, dx as i64);
                    event.set_integer_value_field(EventField::MOUSE_EVENT_DELTA_Y, dy as i64);
                }
                self.post(event, None);
            }
        }
    }

    /// Fetches the `(width, height)` in pixels of the main display
    pub fn main_display_size() -> (usize, usize) {
        let display_id = unsafe { CGMainDisplayID() };
        let width = unsafe { CGDisplayPixelsWide(display_id) };
        let height = unsafe { CGDisplayPixelsHigh(display_id) };
        (width, height)
    }

    /// Returns the current mouse location in Cocoa coordinates which have Y
    /// inverted from the Carbon coordinates used in the rest of the API.
    /// This function exists so that mouse_move_relative only has to fetch
    /// the screen size once.
    fn mouse_location_raw_coords() -> (i32, i32) {
        if let Some(ns_event) = Class::get("NSEvent") {
            let pt: NSPoint = unsafe { msg_send![ns_event, mouseLocation] };
            (pt.x as i32, pt.y as i32)
        } else {
            (0, 0)
        }
    }

    /// The mouse coordinates in points, only works on the main display
    pub fn mouse_location() -> (i32, i32) {
        let (x, y_inv) = Self::mouse_location_raw_coords();
        let (_, display_height) = Self::main_display_size();
        (x, (display_height as i32) - y_inv)
    }

    fn key_to_keycode(&mut self, key: Key) -> CGKeyCode {
        #[allow(deprecated)]
        // I mean duh, we still need to support deprecated keys until they're removed
        match key {
            Key::Alt => kVK_Option,
            Key::Backspace => kVK_Delete,
            Key::CapsLock => kVK_CapsLock,
            Key::Control => kVK_Control,
            Key::Delete => kVK_ForwardDelete,
            Key::DownArrow => kVK_DownArrow,
            Key::End => kVK_End,
            Key::Escape => kVK_Escape,
            Key::F1 => kVK_F1,
            Key::F10 => kVK_F10,
            Key::F11 => kVK_F11,
            Key::F12 => kVK_F12,
            Key::F2 => kVK_F2,
            Key::F3 => kVK_F3,
            Key::F4 => kVK_F4,
            Key::F5 => kVK_F5,
            Key::F6 => kVK_F6,
            Key::F7 => kVK_F7,
            Key::F8 => kVK_F8,
            Key::F9 => kVK_F9,
            Key::Home => kVK_Home,
            Key::LeftArrow => kVK_LeftArrow,
            Key::Option => kVK_Option,
            Key::PageDown => kVK_PageDown,
            Key::PageUp => kVK_PageUp,
            Key::Return => kVK_Return,
            Key::RightArrow => kVK_RightArrow,
            Key::Shift => kVK_Shift,
            Key::Space => kVK_Space,
            Key::Tab => kVK_Tab,
            Key::UpArrow => kVK_UpArrow,
            Key::Numpad0 => kVK_ANSI_Keypad0,
            Key::Numpad1 => kVK_ANSI_Keypad1,
            Key::Numpad2 => kVK_ANSI_Keypad2,
            Key::Numpad3 => kVK_ANSI_Keypad3,
            Key::Numpad4 => kVK_ANSI_Keypad4,
            Key::Numpad5 => kVK_ANSI_Keypad5,
            Key::Numpad6 => kVK_ANSI_Keypad6,
            Key::Numpad7 => kVK_ANSI_Keypad7,
            Key::Numpad8 => kVK_ANSI_Keypad8,
            Key::Numpad9 => kVK_ANSI_Keypad9,
            Key::Mute => kVK_Mute,
            Key::VolumeDown => kVK_VolumeUp,
            Key::VolumeUp => kVK_VolumeDown,
            Key::Help => kVK_Help,
            Key::Snapshot => kVK_F13,
            Key::Clear => kVK_ANSI_KeypadClear,
            Key::Decimal => kVK_ANSI_KeypadDecimal,
            Key::Multiply => kVK_ANSI_KeypadMultiply,
            Key::Add => kVK_ANSI_KeypadPlus,
            Key::Divide => kVK_ANSI_KeypadDivide,
            Key::NumpadEnter => kVK_ANSI_KeypadEnter,
            Key::Subtract => kVK_ANSI_KeypadMinus,
            Key::Equals => kVK_ANSI_KeypadEquals,
            Key::NumLock => kVK_ANSI_KeypadClear,
            Key::RWin => kVK_RIGHT_COMMAND,
            Key::RightShift => kVK_RightShift,
            Key::RightControl => kVK_RightControl,
            Key::RightAlt => kVK_RightOption,

            Key::Raw(raw_keycode) => raw_keycode,
            Key::Layout(c) => self.map_key_board(c),

            Key::Super | Key::Command | Key::Windows | Key::Meta => kVK_Command,
            _ => u16::MAX,
        }
    }

    #[inline]
    fn map_key_board(&mut self, ch: char) -> CGKeyCode {
        // no idea why below char not working with shift, https://github.com/rustdesk/rustdesk/issues/406#issuecomment-1145157327
        // seems related to numpad char
        if ch == '-' || ch == '=' || ch == '.' || ch == '/' || (ch >= '0' && ch <= '9') {
            return self.map_key_board_en(ch);
        }
        let mut code = u16::MAX;
        unsafe {
            let (keyboard, layout) = get_layout();
            if !keyboard.is_null() && !layout.is_null() {
                let name_ref = TISGetInputSourceProperty(keyboard, kTISPropertyInputSourceID);
                if !name_ref.is_null() {
                    let name = get_string(name_ref as _);
                    if let Some(name) = name {
                        if let Some(m) = self.char_to_vkey_map.get(&name) {
                            code = *m.get(&ch).unwrap_or(&u16::MAX);
                        } else {
                            let m = get_map(&name, layout);
                            code = *m.get(&ch).unwrap_or(&u16::MAX);
                            self.char_to_vkey_map.insert(name.clone(), m);
                        }
                    }
                }
            }
            if !keyboard.is_null() {
                CFRelease(keyboard);
            }
        }
        if code != u16::MAX {
            return code;
        }
        self.map_key_board_en(ch)
    }

    #[inline]
    fn map_key_board_en(&mut self, ch: char) -> CGKeyCode {
        match ch {
            'a' => kVK_ANSI_A,
            'b' => kVK_ANSI_B,
            'c' => kVK_ANSI_C,
            'd' => kVK_ANSI_D,
            'e' => kVK_ANSI_E,
            'f' => kVK_ANSI_F,
            'g' => kVK_ANSI_G,
            'h' => kVK_ANSI_H,
            'i' => kVK_ANSI_I,
            'j' => kVK_ANSI_J,
            'k' => kVK_ANSI_K,
            'l' => kVK_ANSI_L,
            'm' => kVK_ANSI_M,
            'n' => kVK_ANSI_N,
            'o' => kVK_ANSI_O,
            'p' => kVK_ANSI_P,
            'q' => kVK_ANSI_Q,
            'r' => kVK_ANSI_R,
            's' => kVK_ANSI_S,
            't' => kVK_ANSI_T,
            'u' => kVK_ANSI_U,
            'v' => kVK_ANSI_V,
            'w' => kVK_ANSI_W,
            'x' => kVK_ANSI_X,
            'y' => kVK_ANSI_Y,
            'z' => kVK_ANSI_Z,
            '0' => kVK_ANSI_0,
            '1' => kVK_ANSI_1,
            '2' => kVK_ANSI_2,
            '3' => kVK_ANSI_3,
            '4' => kVK_ANSI_4,
            '5' => kVK_ANSI_5,
            '6' => kVK_ANSI_6,
            '7' => kVK_ANSI_7,
            '8' => kVK_ANSI_8,
            '9' => kVK_ANSI_9,
            '-' => kVK_ANSI_Minus,
            '=' => kVK_ANSI_Equal,
            '[' => kVK_ANSI_LeftBracket,
            ']' => kVK_ANSI_RightBracket,
            '\\' => kVK_ANSI_Backslash,
            ';' => kVK_ANSI_Semicolon,
            '\'' => kVK_ANSI_Quote,
            ',' => kVK_ANSI_Comma,
            '.' => kVK_ANSI_Period,
            '/' => kVK_ANSI_Slash,
            '`' => kVK_ANSI_Grave,
            _ => u16::MAX,
        }
    }

    #[inline]
    fn mouse_scroll_impl(&mut self, length: i32, is_track_pad: bool, is_horizontal: bool) {
        let mut scroll_direction = -1; // 1 left -1 right;
        let mut length = length;

        if length < 0 {
            length *= -1;
            scroll_direction *= -1;
        }

        if let Some(src) = self.event_source.as_ref() {
            for _ in 0..length {
                unsafe {
                    let units = if is_track_pad {
                        ScrollUnit::Pixel
                    } else {
                        ScrollUnit::Line
                    };
                    let mouse_ev = if is_horizontal {
                        CGEventCreateScrollWheelEvent(
                            &src,
                            units,
                            2, // CGWheelCount 1 = y 2 = xy 3 = xyz
                            0,
                            scroll_direction,
                        )
                    } else {
                        CGEventCreateScrollWheelEvent(
                            &src,
                            units,
                            1, // CGWheelCount 1 = y 2 = xy 3 = xyz
                            scroll_direction,
                        )
                    };

                    CGEventPost(CGEventTapLocation::HID, mouse_ev);
                    CFRelease(mouse_ev as *const std::ffi::c_void);
                }
            }
        }
    }

    /// handle scroll vertically
    pub fn mouse_scroll_y(&mut self, length: i32, is_track_pad: bool) {
        self.mouse_scroll_impl(length, is_track_pad, false)
    }

    /// handle scroll horizontally
    pub fn mouse_scroll_x(&mut self, length: i32, is_track_pad: bool) {
        self.mouse_scroll_impl(length, is_track_pad, true)
    }
}

#[inline]
unsafe fn get_string(cf_string: CFStringRef) -> Option<String> {
    if !cf_string.is_null() {
        let mut buf: [i8; 255] = [0; 255];
        let success = CFStringGetCString(
            cf_string,
            buf.as_mut_ptr(),
            buf.len() as _,
            kCFStringEncodingUTF8,
        );
        if success != 0 {
            let name: &CStr = CStr::from_ptr(buf.as_ptr());
            if let Ok(name) = name.to_str() {
                return Some(name.to_string());
            }
        }
    }
    None
}

#[inline]
unsafe fn get_layout() -> (TISInputSourceRef, *const u8) {
    let mut keyboard = TISCopyCurrentKeyboardInputSource();
    let mut layout = null_mut();
    if !keyboard.is_null() {
        layout = TISGetInputSourceProperty(keyboard, kTISPropertyUnicodeKeyLayoutData);
    }
    if layout.is_null() {
        if !keyboard.is_null() {
            CFRelease(keyboard);
        }
        // https://github.com/microsoft/vscode/issues/23833
        keyboard = TISCopyCurrentKeyboardLayoutInputSource();
        if !keyboard.is_null() {
            layout = TISGetInputSourceProperty(keyboard, kTISPropertyUnicodeKeyLayoutData);
        }
    }
    if layout.is_null() {
        if !keyboard.is_null() {
            CFRelease(keyboard);
        }
        keyboard = TISCopyCurrentASCIICapableKeyboardLayoutInputSource();
        if !keyboard.is_null() {
            layout = TISGetInputSourceProperty(keyboard, kTISPropertyUnicodeKeyLayoutData);
        }
    }
    if layout.is_null() {
        if !keyboard.is_null() {
            CFRelease(keyboard);
        }
        return (null_mut(), null_mut());
    }
    let layout_ptr = CFDataGetBytePtr(layout as _);
    if layout_ptr.is_null() {
        if !keyboard.is_null() {
            CFRelease(keyboard);
        }
        return (null_mut(), null_mut());
    }
    (keyboard, layout_ptr)
}

#[inline]
fn get_map(name: &str, layout: *const u8) -> Map<char, CGKeyCode> {
    log::info!("Create keyboard map for {}", name);
    let mut keys_down: u32 = 0;
    let mut map = Map::new();
    for keycode in 0..128 {
        let mut buff = [0_u16; BUF_LEN];
        let kb_type = unsafe { LMGetKbdType() };
        let mut length: UniCharCount = 0;
        let _retval = unsafe {
            UCKeyTranslate(
                layout,
                keycode,
                kUCKeyActionDisplay as _,
                0,
                kb_type as _,
                kUCKeyTranslateDeadKeysBit as _,
                &mut keys_down,
                BUF_LEN,
                &mut length,
                &mut buff,
            )
        };
        if length > 0 {
            if let Ok(str) = String::from_utf16(&buff[..length]) {
                if let Some(chr) = str.chars().next() {
                    map.insert(chr, keycode as _);
                }
            }
        }
    }
    map
}
unsafe impl Send for Enigo {}