flextrek 0.3.2

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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154

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 mod listen_explorer_drag;

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 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 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
}

pub fn listen<F, Fut>(hotkey_str: String, callback: F) -> HotkeyHandle
where
    F: Fn() -> 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;
                    }
                    callback().await;
                    true
                }
            })
            .await;
        });
    });

    handle
}