winit/src/platform_impl/macos/app.rs

use std::collections::VecDeque;

use objc2::foundation::NSObject;
use objc2::{declare_class, msg_send, ClassType};

use super::appkit::{NSApplication, NSEvent, NSEventModifierFlags, NSEventType, NSResponder};
use super::{app_state::AppState, event::EventWrapper, DEVICE_ID};
use crate::event::{DeviceEvent, ElementState, Event};

declare_class!(
    #[derive(Debug, PartialEq, Eq, Hash)]
    pub(super) struct WinitApplication {}

    unsafe impl ClassType for WinitApplication {
        #[inherits(NSResponder, NSObject)]
        type Super = NSApplication;
    }

    unsafe impl WinitApplication {
        // Normally, holding Cmd + any key never sends us a `keyUp` event for that key.
        // Overriding `sendEvent:` like this fixes that. (https://stackoverflow.com/a/15294196)
        // Fun fact: Firefox still has this bug! (https://bugzilla.mozilla.org/show_bug.cgi?id=1299553)
        #[sel(sendEvent:)]
        fn send_event(&self, event: &NSEvent) {
            // For posterity, there are some undocumented event types
            // (https://github.com/servo/cocoa-rs/issues/155)
            // but that doesn't really matter here.
            let event_type = event.type_();
            let modifier_flags = event.modifierFlags();
            if event_type == NSEventType::NSKeyUp
                && modifier_flags.contains(NSEventModifierFlags::NSCommandKeyMask)
            {
                if let Some(key_window) = self.keyWindow() {
                    unsafe { key_window.sendEvent(event) };
                }
            } else {
                maybe_dispatch_device_event(event);
                unsafe { msg_send![super(self), sendEvent: event] }
            }
        }
    }
);

fn maybe_dispatch_device_event(event: &NSEvent) {
    let event_type = event.type_();
    match event_type {
        NSEventType::NSMouseMoved
        | NSEventType::NSLeftMouseDragged
        | NSEventType::NSOtherMouseDragged
        | NSEventType::NSRightMouseDragged => {
            let mut events = VecDeque::with_capacity(3);

            let delta_x = event.deltaX() as f64;
            let delta_y = event.deltaY() as f64;

            if delta_x != 0.0 {
                events.push_back(EventWrapper::StaticEvent(Event::DeviceEvent {
                    device_id: DEVICE_ID,
                    event: DeviceEvent::Motion {
                        axis: 0,
                        value: delta_x,
                    },
                }));
            }

            if delta_y != 0.0 {
                events.push_back(EventWrapper::StaticEvent(Event::DeviceEvent {
                    device_id: DEVICE_ID,
                    event: DeviceEvent::Motion {
                        axis: 1,
                        value: delta_y,
                    },
                }));
            }

            if delta_x != 0.0 || delta_y != 0.0 {
                events.push_back(EventWrapper::StaticEvent(Event::DeviceEvent {
                    device_id: DEVICE_ID,
                    event: DeviceEvent::MouseMotion {
                        delta: (delta_x, delta_y),
                    },
                }));
            }

            AppState::queue_events(events);
        }
        NSEventType::NSLeftMouseDown
        | NSEventType::NSRightMouseDown
        | NSEventType::NSOtherMouseDown => {
            let mut events = VecDeque::with_capacity(1);

            events.push_back(EventWrapper::StaticEvent(Event::DeviceEvent {
                device_id: DEVICE_ID,
                event: DeviceEvent::Button {
                    button: event.buttonNumber() as u32,
                    state: ElementState::Pressed,
                },
            }));

            AppState::queue_events(events);
        }
        NSEventType::NSLeftMouseUp | NSEventType::NSRightMouseUp | NSEventType::NSOtherMouseUp => {
            let mut events = VecDeque::with_capacity(1);

            events.push_back(EventWrapper::StaticEvent(Event::DeviceEvent {
                device_id: DEVICE_ID,
                event: DeviceEvent::Button {
                    button: event.buttonNumber() as u32,
                    state: ElementState::Released,
                },
            }));

            AppState::queue_events(events);
        }
        _ => (),
    }
}