mirror of
https://github.com/rustdesk/rustdesk.git
synced 2024-12-15 03:59:04 +08:00
a2f1ab80da
Signed-off-by: 21pages <pages21@163.com>
79 lines
2.4 KiB
Rust
79 lines
2.4 KiB
Rust
use std::{
|
|
collections::HashMap,
|
|
sync::{Arc, Mutex},
|
|
};
|
|
use trayicon::{MenuBuilder, TrayIconBuilder};
|
|
use winit::{
|
|
event::Event,
|
|
event_loop::{ControlFlow, EventLoop},
|
|
};
|
|
|
|
#[derive(Clone, Eq, PartialEq, Debug)]
|
|
enum Events {
|
|
DoubleClickTrayIcon,
|
|
StopService,
|
|
StartService,
|
|
}
|
|
|
|
pub fn start_tray(options: Arc<Mutex<HashMap<String, String>>>) {
|
|
let event_loop = EventLoop::<Events>::with_user_event();
|
|
let proxy = event_loop.create_proxy();
|
|
let icon = include_bytes!("../res/tray-icon.ico");
|
|
let mut tray_icon = TrayIconBuilder::new()
|
|
.sender_winit(proxy)
|
|
.icon_from_buffer(icon)
|
|
.tooltip("RustDesk")
|
|
.on_double_click(Events::DoubleClickTrayIcon)
|
|
.build()
|
|
.unwrap();
|
|
let old_state = Arc::new(Mutex::new(0));
|
|
let _sender = crate::ui_interface::SENDER.lock().unwrap();
|
|
event_loop.run(move |event, _, control_flow| {
|
|
if options.lock().unwrap().get("ipc-closed").is_some() {
|
|
*control_flow = ControlFlow::Exit;
|
|
return;
|
|
} else {
|
|
*control_flow = ControlFlow::Wait;
|
|
}
|
|
let stopped = if let Some(v) = options.lock().unwrap().get("stop-service") {
|
|
!v.is_empty()
|
|
} else {
|
|
false
|
|
};
|
|
let stopped = if stopped { 2 } else { 1 };
|
|
let old = *old_state.lock().unwrap();
|
|
if stopped != old {
|
|
hbb_common::log::info!("State changed");
|
|
let mut m = MenuBuilder::new();
|
|
if stopped == 2 {
|
|
m = m.item(
|
|
&crate::client::translate("Start Service".to_owned()),
|
|
Events::StartService,
|
|
);
|
|
} else {
|
|
m = m.item(
|
|
&crate::client::translate("Stop service".to_owned()),
|
|
Events::StopService,
|
|
);
|
|
}
|
|
tray_icon.set_menu(&m).ok();
|
|
*old_state.lock().unwrap() = stopped;
|
|
}
|
|
|
|
match event {
|
|
Event::UserEvent(e) => match e {
|
|
Events::DoubleClickTrayIcon => {
|
|
crate::run_me(Vec::<&str>::new()).ok();
|
|
}
|
|
Events::StopService => {
|
|
crate::ipc::set_option("stop-service", "Y");
|
|
}
|
|
Events::StartService => {
|
|
crate::ipc::set_option("stop-service", "");
|
|
}
|
|
},
|
|
_ => (),
|
|
}
|
|
});
|
|
}
|