teamy-mft 0.7.1

TeamDman's Master File Table CLI and library for NTFS.
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

use crate::windows_utils::module::get_current_module;
use eyre::bail;
use tracing::debug;
use tracing::instrument;
use windows::Win32::Foundation::HINSTANCE;
use windows::Win32::UI::WindowsAndMessaging::HICON;
use windows::Win32::UI::WindowsAndMessaging::LoadIconW;
use windows::core::PCWSTR;
use windows::core::Param;

#[instrument]
pub fn get_icon_from_current_module(
    icon_name: impl Param<PCWSTR> + std::fmt::Debug,
) -> eyre::Result<HICON> {
    let handle = get_current_module()?;
    debug!(?handle, "Trying to load embedded icon from current module");
    let icon = unsafe { LoadIconW(Some(HINSTANCE(handle.0)), icon_name) };
    match icon {
        Ok(icon) => {
            debug!("Successfully loaded embedded icon from current module");
            Ok(icon)
        }
        Err(e) => bail!("Failed to load embedded icon from current module: {}", e),
    }
}