konevsa/rustdesk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

load tray icon from asset

Browse Source
This commit is contained in:
rustdesk 2024-03-12 22:13:28 +08:00
parent fc875f7ad6
commit be301b825f
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/tray.rs
View File

@ -29,9 +29,10 @@ pub fn make_tray() -> hbb_common::ResultType<()> {
{ {
icon = include_bytes!("../res/tray-icon.ico"); icon = include_bytes!("../res/tray-icon.ico");
} }
let (icon_rgba, icon_width, icon_height) = { let (icon_rgba, icon_width, icon_height) = {
let image = image::load_from_memory(icon) let image = load_icon_from_asset()
.context("Failed to open icon path")? .unwrap_or(image::load_from_memory(icon).context("Failed to open icon path")?)
.into_rgba8(); .into_rgba8();
let (width, height) = image.dimensions(); let (width, height) = image.dimensions();
let rgba = image.into_raw(); let rgba = image.into_raw();
@ -202,3 +203,18 @@ async fn start_query_session_count(sender: std::sync::mpsc::Sender<Data>) {
hbb_common::sleep(1.).await; hbb_common::sleep(1.).await;
} }
} }
fn load_icon_from_asset() -> Option<image::DynamicImage> {
#[cfg(windows)]
if let Ok(cmd) = std::env::current_exe() {
let path = r".\data\flutter_assets\assets\icon.ico";
if let Some(path) = cmd.parent().map(|x| x.join(path)) {
if path.exists() {
if let Ok(image) = image::open(path) {
return Some(image);
}
}
}
}
None
}