openlogi-core 0.8.14

//! Binding-map construction: overlay the stored per-device (and per-app)
//! bindings on top of the built-in defaults.
//!
//! Keyed by `config_key` (`Option<&str>`) rather than any UI device record so
//! both the agent and the GUI can build the effective map from a [`Config`].

use std::collections::BTreeMap;

use crate::binding::{
    Action, Binding, ButtonId, GestureDirection, default_binding, default_binding_for,
};
use crate::config::Config;

/// Effective per-button single-action map for the device `config_key`, with
/// `app_bundle`'s per-app overlay applied. Unset buttons fall back to
/// [`default_binding`].
///
/// This projection is for one-shot consumers such as thumb-wheel rotation and
/// the GUI. Lifecycle-aware button paths use [`button_bindings_for`] so a long
/// press keeps both actions. `Binding::Gesture` is projected to its
/// `click_action()`; per-direction swipes are dispatched via the separate
/// [`hidpp_gesture_maps_for`] / [`oshook_gestures_for`] maps.
#[must_use]
pub fn bindings_for(
    config: &Config,
    config_key: Option<&str>,
    app_bundle: Option<&str>,
) -> BTreeMap<ButtonId, Action> {
    button_bindings_for(config, config_key, app_bundle)
        .into_iter()
        .map(|(button, binding)| (button, binding.click_action()))
        .collect()
}

/// Effective binding shapes for lifecycle-aware button consumers.
///
/// Unlike [`bindings_for`], this preserves threshold-based long presses. A
/// sparse gesture map receives only its missing `Click` default so it keeps its
/// gesture identity while retaining the same plain-click fallback.
#[must_use]
pub fn button_bindings_for(
    config: &Config,
    config_key: Option<&str>,
    app_bundle: Option<&str>,
) -> BTreeMap<ButtonId, Binding> {
    let stored = config_key
        .map(|key| config.effective_bindings(key, app_bundle))
        .unwrap_or_default();
    let mut bindings: BTreeMap<ButtonId, Binding> = ButtonId::ALL
        .iter()
        .copied()
        .map(|button| (button, Binding::Single(default_binding(button))))
        .collect();
    for (button, mut binding) in stored {
        if let Binding::Gesture(map) = &mut binding {
            map.entry(GestureDirection::Click)
                .or_insert_with(|| default_binding(button));
        }
        bindings.insert(button, binding);
    }
    bindings
}

/// Per-direction maps for every HID++ gesture source (DPI/ModeShift, the
/// dedicated gesture button, and the MX Master 4 haptic panel) in gesture mode
/// on `config_key`, with `app_bundle`'s single-action overrides applied,
/// keyed by the button its captured swipes dispatch as. Each map is seeded
/// via [`Binding::fill_gesture_defaults`] — the one canonical seeding rule —
/// so the watcher always dispatches the full five-direction set the GUI
/// shows. Empty when no HID++ source gestures (or `config_key` is `None`).
#[must_use]
pub fn hidpp_gesture_maps_for(
    config: &Config,
    config_key: Option<&str>,
    app_bundle: Option<&str>,
) -> BTreeMap<ButtonId, BTreeMap<GestureDirection, Action>> {
    let Some(key) = config_key else {
        return BTreeMap::new();
    };
    let stored = config.effective_bindings(key, app_bundle);
    ButtonId::ALL
        .iter()
        .copied()
        .filter(|button| button.is_hidpp_gesture_source())
        .filter_map(|button| {
            // The stored shape (or the button's canonical default) IS gesture
            // mode — a Single-shaped source simply drops out.
            let mut binding = stored
                .get(&button)
                .cloned()
                .unwrap_or_else(|| default_binding_for(button));
            binding.fill_gesture_defaults();
            match binding {
                Binding::Gesture(map) => Some((button, map)),
                Binding::Single(_) | Binding::LongPress(_) => None,
            }
        })
        .collect()
}

/// Per-direction maps for every OS-hook button with a stored gesture binding on
/// `config_key`, with `app_bundle`'s per-app overlay applied, for the OS hook to
/// resolve a hold+swipe. New gesture mode is offered only for Back/Forward, but
/// v0.8.0 also persisted Middle Click gesture maps; accepting all OS-hook
/// buttons here preserves those maps until the user turns that mode off.
/// Gesture mode is per-button (see [`Config::is_gesture_mode`]), so any number
/// of entries may be live at once — concurrency between them is the hook's
/// first-hold-wins policy, not a config concern.
///
/// Unlike [`hidpp_gesture_maps_for`] (whose maps seed every direction at
/// projection time), this returns each button's raw stored map. In practice
/// those maps are
/// already fully populated — [`Config::set_gesture_mode`] seeds all five
/// directions via [`Binding::fill_gesture_defaults`] when a button is
/// promoted — so only a hand-edited sparse map leaves a direction unbound, in
/// which case that swipe simply does nothing. The dedicated gesture button is
/// intentionally excluded: it never reaches the OS hook (it's captured over
/// HID++), so it has no entry here.
///
/// A per-app override of a gesture button turns it into a [`Binding::Single`]
/// for that app, so it stops being a gesture button there and falls through to
/// the single-action path (which applies the override) — mirroring how a single
/// binding is overridden per app.
#[must_use]
pub fn oshook_gestures_for(
    config: &Config,
    config_key: Option<&str>,
    app_bundle: Option<&str>,
) -> BTreeMap<ButtonId, BTreeMap<GestureDirection, Action>> {
    let Some(key) = config_key else {
        return BTreeMap::new();
    };
    // Read the per-app *effective* map: a per-app override replaces a gesture
    // button with a `Single`, dropping it from the gesture set for that app.
    config
        .effective_bindings(key, app_bundle)
        .into_iter()
        .filter(|(id, _)| id.is_os_hook_button())
        .filter_map(|(id, binding)| match binding {
            Binding::Gesture(map) => Some((id, map)),
            Binding::Single(_) | Binding::LongPress(_) => None,
        })
        .collect()
}

#[cfg(test)]
mod tests {
    use crate::binding::{LongPressBinding, default_gesture_binding};

    use super::*;

    #[test]
    fn click_less_gesture_keeps_default_click_in_projection() {
        // A gesture binding with no explicit `Click` (a migrated sparse v1 map or
        // a hand-edited config) must not project to `Action::None` and silently
        // disable the button — the button's default click survives.
        let mut cfg = Config::default();
        let mut map = BTreeMap::new();
        map.insert(GestureDirection::Up, Action::Copy);
        cfg.set_binding("2b042", ButtonId::GestureButton, Binding::Gesture(map));

        let projected = bindings_for(&cfg, Some("2b042"), None);
        assert_eq!(
            projected.get(&ButtonId::GestureButton),
            Some(&default_binding(ButtonId::GestureButton)),
            "a Click-less gesture must keep the default click, not None"
        );
    }

    #[test]
    fn lifecycle_projection_preserves_long_press_while_action_projection_uses_short() {
        let mut cfg = Config::default();
        let binding = Binding::LongPress(LongPressBinding::new(Action::Copy, Action::Paste));
        cfg.set_binding("2b042", ButtonId::Back, binding.clone());

        let lifecycle = button_bindings_for(&cfg, Some("2b042"), None);
        assert_eq!(lifecycle.get(&ButtonId::Back), Some(&binding));
        let actions = bindings_for(&cfg, Some("2b042"), None);
        assert_eq!(actions.get(&ButtonId::Back), Some(&Action::Copy));
    }

    #[test]
    fn a_rotation_rebind_leaves_the_wheels_tap_inert() {
        // Rebinding rotation (or moving the sensitivity slider) diverts the
        // wheel over 0x2150, which is what starts delivering its capacitive
        // taps. Those taps must stay inert until the user binds the tap
        // itself — a seeded action here fires from incidental thumb contact.
        let mut cfg = Config::default();
        cfg.set_binding(
            "2b034",
            ButtonId::ThumbwheelScrollUp,
            Action::VolumeUp.into(),
        );

        let projected = bindings_for(&cfg, Some("2b034"), None);
        assert_eq!(projected.get(&ButtonId::Thumbwheel), Some(&Action::None));
    }

    #[test]
    fn an_explicitly_bound_tap_survives_the_inert_default() {
        let mut cfg = Config::default();
        cfg.set_binding("2b034", ButtonId::Thumbwheel, Action::AppExpose.into());

        let projected = bindings_for(&cfg, Some("2b034"), None);
        assert_eq!(
            projected.get(&ButtonId::Thumbwheel),
            Some(&Action::AppExpose)
        );
    }

    #[test]
    fn explicit_gesture_click_overrides_default_in_projection() {
        // A gesture binding that DOES define `Click` projects that action.
        let mut cfg = Config::default();
        let mut map = BTreeMap::new();
        map.insert(GestureDirection::Click, Action::Paste);
        cfg.set_binding("2b042", ButtonId::GestureButton, Binding::Gesture(map));

        let projected = bindings_for(&cfg, Some("2b042"), None);
        assert_eq!(
            projected.get(&ButtonId::GestureButton),
            Some(&Action::Paste)
        );
    }

    #[test]
    fn oshook_gestures_collects_only_os_hook_gesture_buttons() {
        let mut cfg = Config::default();
        // A gesture-mode Back (an OS-hook button) — included, raw map preserved.
        cfg.set_binding(
            "2b042",
            ButtonId::Back,
            Binding::Gesture(BTreeMap::from([(GestureDirection::Up, Action::Copy)])),
        );
        // A single-mode Middle — excluded (not a gesture button).
        cfg.set_binding("2b042", ButtonId::MiddleClick, Action::MiddleClick.into());
        // The dedicated HID++ gesture button — excluded (it never reaches the
        // OS hook, so it must not appear in the hook's gesture map).
        cfg.set_binding(
            "2b042",
            ButtonId::GestureButton,
            Binding::Gesture(BTreeMap::from([(
                GestureDirection::Up,
                Action::MissionControl,
            )])),
        );

        let oshook = oshook_gestures_for(&cfg, Some("2b042"), None);
        assert_eq!(oshook.len(), 1, "only the gesture-mode Back belongs here");
        assert_eq!(
            oshook.get(&ButtonId::Back),
            Some(&BTreeMap::from([(GestureDirection::Up, Action::Copy)]))
        );
        assert!(!oshook.contains_key(&ButtonId::MiddleClick));
        assert!(!oshook.contains_key(&ButtonId::GestureButton));
    }

    #[test]
    fn oshook_gestures_preserves_a_stored_middle_click_gesture() {
        let mut cfg = Config::default();
        cfg.set_gesture_mode("2b042", ButtonId::Back, true);
        cfg.set_gesture_mode("2b042", ButtonId::Forward, true);
        let middle = BTreeMap::from([
            (GestureDirection::Click, Action::MiddleClick),
            (GestureDirection::Up, Action::Copy),
        ]);
        cfg.set_binding(
            "2b042",
            ButtonId::MiddleClick,
            Binding::Gesture(middle.clone()),
        );

        let oshook = oshook_gestures_for(&cfg, Some("2b042"), None);
        assert!(oshook.contains_key(&ButtonId::Back), "got: {oshook:?}");
        assert!(oshook.contains_key(&ButtonId::Forward), "got: {oshook:?}");
        assert_eq!(
            oshook.get(&ButtonId::MiddleClick),
            Some(&middle),
            "an existing Middle Click gesture from v0.8.0 must keep dispatching"
        );
    }

    #[test]
    fn hidpp_gesture_maps_includes_every_gesture_mode_source() {
        // All HID++ sources in gesture mode dispatch simultaneously, each
        // through its own seeded direction map.
        let mut cfg = Config::default();
        cfg.set_gesture_mode("2b042", ButtonId::DpiToggle, true);
        cfg.set_gesture_mode("2b042", ButtonId::HapticPanel, true);

        let maps = hidpp_gesture_maps_for(&cfg, Some("2b042"), None);
        // The dedicated button gestures by default...
        let dedicated = maps
            .get(&ButtonId::GestureButton)
            .expect("the dedicated button's default gesture mode must survive");
        assert_eq!(
            dedicated.get(&GestureDirection::Up),
            Some(&default_gesture_binding(GestureDirection::Up))
        );
        // ...and the panel's promotion adds a second, fully-seeded map.
        let panel = maps
            .get(&ButtonId::HapticPanel)
            .expect("a gesture-mode panel must dispatch");
        for dir in GestureDirection::ALL {
            assert!(panel.contains_key(&dir), "unseeded panel arm {dir:?}");
        }
        let dpi = maps
            .get(&ButtonId::DpiToggle)
            .expect("a gesture-mode DPI button must dispatch through HID++");
        for dir in GestureDirection::ALL {
            assert!(dpi.contains_key(&dir), "unseeded DPI arm {dir:?}");
        }
    }

    #[test]
    fn per_app_override_drops_the_owner_from_the_oshook_gesture_set() {
        // Back is the gesture owner globally...
        let mut cfg = Config::default();
        cfg.set_gesture_mode("2b042", ButtonId::Back, true);
        assert!(
            oshook_gestures_for(&cfg, Some("2b042"), None).contains_key(&ButtonId::Back),
            "Back gestures globally"
        );

        // ...but a per-app override makes it a single action in that app, so it
        // must drop out of the gesture set there (and fall through to the
        // single-action path, which applies the override).
        cfg.set_per_app_binding(
            "2b042",
            "com.apple.Safari",
            ButtonId::Back,
            Some(Action::NextTab),
        );
        assert!(
            oshook_gestures_for(&cfg, Some("2b042"), Some("com.apple.Safari")).is_empty(),
            "a per-app override of the owner removes it from the gesture set"
        );
        // Other apps are unaffected — Back still gestures.
        assert!(
            oshook_gestures_for(&cfg, Some("2b042"), Some("com.other.App"))
                .contains_key(&ButtonId::Back)
        );
    }

    #[test]
    fn per_app_overrides_replace_hidpp_gestures_and_clearing_restores_them() {
        for button in [
            ButtonId::DpiToggle,
            ButtonId::GestureButton,
            ButtonId::HapticPanel,
        ] {
            let mut cfg = Config::default();
            cfg.set_gesture_mode("2b042", button, true);
            let global = hidpp_gesture_maps_for(&cfg, Some("2b042"), None);
            assert!(global.contains_key(&button));

            for action in [Action::Paste, Action::None] {
                cfg.set_per_app_binding(
                    "2b042",
                    "com.example.Editor",
                    button,
                    Some(action.clone()),
                );
                let mut expected = global.clone();
                expected.remove(&button);
                assert_eq!(
                    hidpp_gesture_maps_for(&cfg, Some("2b042"), Some("com.example.Editor")),
                    expected
                );
                assert_eq!(
                    bindings_for(&cfg, Some("2b042"), Some("com.example.Editor")).get(&button),
                    Some(&action)
                );
                assert_eq!(hidpp_gesture_maps_for(&cfg, Some("2b042"), None), global);
                assert_eq!(
                    hidpp_gesture_maps_for(&cfg, Some("2b042"), Some("com.example.Other")),
                    global
                );
            }
            cfg.set_per_app_binding("2b042", "com.example.Editor", button, None);
            assert_eq!(
                hidpp_gesture_maps_for(&cfg, Some("2b042"), Some("com.example.Editor")),
                global
            );
        }
    }

    #[test]
    fn hidpp_maps_silent_for_a_demoted_dedicated_button() {
        // Default device: the dedicated HID++ gesture button gestures, with its
        // defaults seeded.
        let mut cfg = Config::default();
        let maps = hidpp_gesture_maps_for(&cfg, Some("2b042"), None);
        assert_eq!(
            maps.get(&ButtonId::GestureButton)
                .and_then(|m| m.get(&GestureDirection::Up)),
            Some(&default_gesture_binding(GestureDirection::Up)),
            "the dedicated button gestures by default, seeded"
        );

        // Demoting it silences the watcher for 0x00c3 — and promoting an
        // OS-hook button never resurrects it.
        cfg.set_gesture_mode("2b042", ButtonId::GestureButton, false);
        cfg.set_gesture_mode("2b042", ButtonId::Back, true);
        assert!(
            hidpp_gesture_maps_for(&cfg, Some("2b042"), None).is_empty(),
            "a demoted dedicated button must dispatch nothing over HID++"
        );
    }
}