Qopy/src-tauri/src/api/hotkeys.rs

use crate::utils::commands::center_window_on_current_monitor;
use crate::utils::keys::KeyCode;
use global_hotkey::{
    hotkey::{ Code, HotKey, Modifiers },
    GlobalHotKeyEvent,
    GlobalHotKeyManager,
    HotKeyState,
};
use lazy_static::lazy_static;
use std::str::FromStr;
use std::sync::Mutex;
use tauri::{ AppHandle, Listener, Manager };
use tauri_plugin_aptabase::EventTracker;

lazy_static! {
    static ref HOTKEY_MANAGER: Mutex<Option<GlobalHotKeyManager>> = Mutex::new(None);
    static ref REGISTERED_HOTKEY: Mutex<Option<HotKey>> = Mutex::new(None);
}

pub fn setup(app_handle: tauri::AppHandle) {
    let app_handle_clone = app_handle.clone();

    let manager = match GlobalHotKeyManager::new() {
        Ok(manager) => manager,
        Err(err) => {
            eprintln!("Failed to initialize hotkey manager: {:?}", err);
            return;
        }
    };

    {
        let mut manager_guard = HOTKEY_MANAGER.lock().unwrap();
        *manager_guard = Some(manager);
    }

    let rt = app_handle.state::<tokio::runtime::Runtime>();
    let initial_keybind = rt
        .block_on(crate::db::settings::get_keybind(app_handle_clone.clone()))
        .expect("Failed to get initial keybind");

    if let Err(e) = register_shortcut(&initial_keybind) {
        eprintln!("Error registering initial shortcut: {:?}", e);
    }

    app_handle.listen("update-shortcut", move |event| {
        let payload_str = event.payload();

        if let Some(old_hotkey) = REGISTERED_HOTKEY.lock().unwrap().take() {
            let manager_guard = HOTKEY_MANAGER.lock().unwrap();
            if let Some(manager) = manager_guard.as_ref() {
                let _ = manager.unregister(old_hotkey);
            }
        }

        let payload: Vec<String> = serde_json::from_str(payload_str).unwrap_or_default();

        if let Err(e) = register_shortcut(&payload) {
            eprintln!("Error re-registering shortcut: {:?}", e);
        }
    });

    app_handle.listen("save_keybind", move |event| {
        let payload_str = event.payload().to_string();

        if let Some(old_hotkey) = REGISTERED_HOTKEY.lock().unwrap().take() {
            let manager_guard = HOTKEY_MANAGER.lock().unwrap();
            if let Some(manager) = manager_guard.as_ref() {
                let _ = manager.unregister(old_hotkey);
            }
        }

        let payload: Vec<String> = serde_json::from_str(&payload_str).unwrap_or_default();
        if let Err(e) = register_shortcut(&payload) {
            eprintln!("Error registering saved shortcut: {:?}", e);
        }
    });

    let app_handle_for_hotkey = app_handle.clone();
    tauri::async_runtime::spawn(async move {
        loop {
            match GlobalHotKeyEvent::receiver().recv() {
                Ok(event) => {
                    if event.state == HotKeyState::Released {
                        continue;
                    }
                    handle_hotkey_event(&app_handle_for_hotkey);
                }
                Err(e) => {
                    eprintln!("Error receiving hotkey event: {:?}", e);
                }
            }
        }
    });
}

fn register_shortcut(shortcut: &[String]) -> Result<(), Box<dyn std::error::Error>> {
    let hotkey = parse_hotkey(shortcut)?;

    let manager_guard = HOTKEY_MANAGER.lock().unwrap();
    if let Some(manager) = manager_guard.as_ref() {
        manager.register(hotkey.clone())?;
        *REGISTERED_HOTKEY.lock().unwrap() = Some(hotkey);
        Ok(())
    } else {
        Err("Hotkey manager not initialized".into())
    }
}

fn parse_hotkey(shortcut: &[String]) -> Result<HotKey, Box<dyn std::error::Error>> {
    let mut modifiers = Modifiers::empty();
    let mut code = None;

    for part in shortcut {
        match part.as_str() {
            "ControlLeft" => {
                modifiers |= Modifiers::CONTROL;
            }
            "AltLeft" => {
                modifiers |= Modifiers::ALT;
            }
            "ShiftLeft" => {
                modifiers |= Modifiers::SHIFT;
            }
            "MetaLeft" => {
                modifiers |= Modifiers::META;
            }
            key => {
                code = Some(Code::from(KeyCode::from_str(key)?));
            }
        }
    }

    let key_code = code.ok_or_else(|| "No valid key code found".to_string())?;
    Ok(HotKey::new(Some(modifiers), key_code))
}

fn handle_hotkey_event(app_handle: &AppHandle) {
    let window = app_handle.get_webview_window("main").unwrap();
    if window.is_visible().unwrap() {
        window.hide().unwrap();
    } else {
        window.set_always_on_top(true).unwrap();
        window.show().unwrap();
        window.set_focus().unwrap();

        let window_clone = window.clone();
        std::thread::spawn(move || {
            std::thread::sleep(std::time::Duration::from_millis(100));
            window_clone.set_always_on_top(false).unwrap();
        });

        center_window_on_current_monitor(&window);
    }

    let _ = app_handle.track_event(
        "hotkey_triggered",
        Some(
            serde_json::json!({
            "action": if window.is_visible().unwrap() { "hide" } else { "show" }
        })
        )
    );
}