whisp 0.5.0

Unobtrusive global speech-to-text
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

//! Tray icon management.

use std::path::Path;
use std::sync::LazyLock;

use tray_icon::Icon;

use crate::MicState;
use crate::color::{self, Color};

pub const ICON_PATH: &str = concat!(env!("CARGO_MANIFEST_DIR"), "/assets/icon.png");

fn load_color(color: Color) -> Icon {
    const CHOOSE_FN: fn(Color) -> (u8, u8, u8) = |color| color.accessible_dark;
    load_icon(ICON_PATH, Some(CHOOSE_FN(color)))
}

static IDLE: LazyLock<Icon> = LazyLock::new(|| load_color(color::WHITE));
static WAITING: LazyLock<Icon> = LazyLock::new(|| load_color(color::YELLOW));
static ACTIVE: LazyLock<Icon> = LazyLock::new(|| load_color(color::GREEN));
static WORKING: LazyLock<Icon> = LazyLock::new(|| load_color(color::YELLOW));

/// Extension trait to get icons for MicState.
pub trait MicStateIcon {
    /// Get the tray icon for this state.
    fn icon(&self) -> Icon;
}

impl MicStateIcon for MicState {
    fn icon(&self) -> Icon {
        match self {
            MicState::Activating => WAITING.clone(),
            MicState::Active => ACTIVE.clone(),
            MicState::Idle => IDLE.clone(),
            MicState::Processing => WORKING.clone(),
        }
    }
}

fn load_icon(path: impl AsRef<Path>, recolor: Option<(u8, u8, u8)>) -> Icon {
    let (icon_rgba, icon_width, icon_height) = {
        let mut image = image::open(path)
            .expect("Failed to open icon path")
            .into_rgba8();

        if let Some((r, g, b)) = recolor {
            for pixel in image.pixels_mut() {
                pixel[0] = r;
                pixel[1] = g;
                pixel[2] = b;
            }
        }

        let (width, height) = image.dimensions();
        let rgba = image.into_raw();
        (rgba, width, height)
    };
    Icon::from_rgba(icon_rgba, icon_width, icon_height).expect("Failed to open icon")
}