flextrek 0.2.0

A super-easy, windows-only crate to get focused explorer location or selected files path using hotkey!
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use std::path::PathBuf;
use std::sync::{Arc, Mutex};

use windows::Win32::UI::{
    Input::KeyboardAndMouse::{RegisterHotKey, UnregisterHotKey, HOT_KEY_MODIFIERS},
    WindowsAndMessaging::{GetMessageW, MSG, WM_HOTKEY},
};

use crate::get_explorer_location::get_focused_explorer_path;
use crate::get_explorer_selected_file::get_explorer_selected_file;
use crate::hotkey::parse_hotkey;
use futures::Future;
use tokio::sync::oneshot;

pub mod get_explorer_location;
pub mod get_explorer_selected_file;
pub mod hotkey;

pub struct HotkeyHandle {
    cancel_sender: oneshot::Sender<()>,
}

impl HotkeyHandle {
    pub fn unregister(self) {
        let _ = self.cancel_sender.send(());
        unsafe {
            let _ = UnregisterHotKey(None, 1);
        }
    }
}

async fn listen_hotkey<F, Fut>(hotkey_str: String, mut callback: F)
where
    F: FnMut() -> Fut,
    Fut: Future<Output = bool>,
{
    let hotkey_str_clone = hotkey_str.clone();
    if let Some((modifier, key)) = parse_hotkey(hotkey_str_clone) {
        unsafe {
            let _ = RegisterHotKey(None, 1, HOT_KEY_MODIFIERS(modifier), key);
            let mut msg = MSG::default();
            while GetMessageW(&mut msg, None, 0, 0).as_bool() {
                if msg.message == WM_HOTKEY && !callback().await {
                    break;
                }
            }
        }
    } else {
        println!("Invalid hotkey: {}", hotkey_str);
    }
}

pub async fn listen_path<F, Fut>(hotkey_str: String, callback: F) -> HotkeyHandle
where
    F: Fn(PathBuf) -> Fut + Send + Clone + 'static,
    Fut: Future<Output = ()> + Send,
{
    let (tx, rx) = oneshot::channel();
    let rx = Arc::new(Mutex::new(rx));

    let handle = HotkeyHandle { cancel_sender: tx };

    std::thread::spawn(move || {
        let rt = tokio::runtime::Runtime::new().unwrap();
        rt.block_on(async {
            let callback = callback.clone();
            let rx = rx.clone();
            listen_hotkey(hotkey_str, move || {
                let callback = callback.clone();
                let rx = rx.clone();
                async move {
                    if rx.lock().unwrap().try_recv().is_ok() {
                        return false;
                    }
                    if let Ok(path) = get_focused_explorer_path() {
                        callback(path).await;
                    }
                    true
                }
            })
            .await;
        });
    });

    handle
}

pub async fn listen_selected_files<F, Fut>(hotkey_str: String, callback: F) -> HotkeyHandle
where
    F: Fn(Vec<String>) -> Fut + Send + Clone + 'static,
    Fut: Future<Output = ()> + Send,
{
    let (tx, rx) = oneshot::channel();
    let rx = Arc::new(Mutex::new(rx));

    let handle = HotkeyHandle { cancel_sender: tx };

    std::thread::spawn(move || {
        let rt = tokio::runtime::Runtime::new().unwrap();
        rt.block_on(async {
            let callback = callback.clone();
            let rx = rx.clone();
            listen_hotkey(hotkey_str, move || {
                let callback = callback.clone();
                let rx = rx.clone();
                async move {
                    if rx.lock().unwrap().try_recv().is_ok() {
                        return false;
                    }
                    let file_list = get_explorer_selected_file();
                    callback(file_list).await;
                    true
                }
            })
            .await;
        });
    });

    handle
}