|
1 | 1 | use std::{env, fs, io::BufReader, path::PathBuf}; |
2 | 2 |
|
3 | 3 | fn main() { |
4 | | - println!("cargo:rerun-if-changed=../../assets/icon.ico"); |
5 | | - |
| 4 | + let manifest_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap()); |
| 5 | + let ico_path = manifest_dir.join("../../assets/icon.ico"); |
6 | 6 | let out = PathBuf::from(env::var("OUT_DIR").unwrap()); |
7 | | - let ico_path = "../../assets/icon.ico"; |
8 | | - |
9 | | - let file = BufReader::new(fs::File::open(ico_path).expect("assets/icon.ico not found")); |
10 | | - let dir = ico::IconDir::read(file).expect("parse icon.ico"); |
11 | 7 |
|
12 | | - let entry = dir |
13 | | - .entries() |
14 | | - .iter() |
15 | | - .max_by_key(|e| e.width()) |
16 | | - .expect("icon.ico is empty"); |
17 | | - |
18 | | - let image = entry.decode().expect("decode icon entry"); |
19 | | - let (w, h) = (image.width(), image.height()); |
20 | | - let rgba = image.rgba_data().to_vec(); |
| 8 | + println!("cargo:rerun-if-changed=../../assets/icon.ico"); |
21 | 9 |
|
22 | | - fs::write(out.join("icon.rgba"), &rgba).expect("write icon.rgba"); |
23 | | - fs::write( |
24 | | - out.join("icon_meta.rs"), |
25 | | - format!( |
26 | | - "const ICON_WIDTH: u32 = {};\nconst ICON_HEIGHT: u32 = {};\n", |
27 | | - w, h |
28 | | - ), |
29 | | - ) |
30 | | - .expect("write icon_meta.rs"); |
| 10 | + if ico_path.exists() { |
| 11 | + let file = BufReader::new(fs::File::open(&ico_path).expect("open icon.ico")); |
| 12 | + let dir = ico::IconDir::read(file).expect("parse icon.ico"); |
| 13 | + let entry = dir |
| 14 | + .entries() |
| 15 | + .iter() |
| 16 | + .max_by_key(|e| e.width()) |
| 17 | + .expect("icon.ico is empty"); |
| 18 | + let image = entry.decode().expect("decode icon entry"); |
| 19 | + let (w, h) = (image.width(), image.height()); |
| 20 | + let rgba = image.rgba_data().to_vec(); |
| 21 | + fs::write(out.join("icon.rgba"), &rgba).expect("write icon.rgba"); |
| 22 | + fs::write( |
| 23 | + out.join("icon_meta.rs"), |
| 24 | + format!("const ICON_WIDTH: u32 = {w};\nconst ICON_HEIGHT: u32 = {h};\n"), |
| 25 | + ) |
| 26 | + .expect("write icon_meta.rs"); |
| 27 | + } else { |
| 28 | + // No icon available (e.g. during `cargo publish` verification). |
| 29 | + // Write a 1×1 transparent placeholder so the crate still compiles. |
| 30 | + fs::write(out.join("icon.rgba"), [0u8, 0, 0, 0]).expect("write icon.rgba"); |
| 31 | + fs::write( |
| 32 | + out.join("icon_meta.rs"), |
| 33 | + "const ICON_WIDTH: u32 = 1;\nconst ICON_HEIGHT: u32 = 1;\n", |
| 34 | + ) |
| 35 | + .expect("write icon_meta.rs"); |
| 36 | + } |
31 | 37 | } |
0 commit comments