tkbar 0.2.0

A minimalist, hardened status bar for Wayland.
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150

#![forbid(unsafe_code)]
#[cfg(feature = "cli")]
pub use conf::CLI_CONFIG_PATH;
pub use {crate::ui::build_window, conf::Config};
use {gtk::CssProvider, gtk4 as gtk};

mod conf;
mod log;
mod ui;

/// Initializes GTK, aborting with a diagnostic if it fails.
///
/// `gtk::init` is the first point where a missing display server
/// (`WAYLAND_DISPLAY`/`DISPLAY`) surfaces. Ignoring its error would only defer
/// the failure to `load_css`, with a far less actionable message.
pub fn init_gtk() {
    if let Err(err) = gtk::init() {
        crate::log::error("gtk", &format!("cannot initialize GTK: {err}"));
        std::process::exit(1);
    }
}

pub fn load_css() {
    let Some(display) = gtk::gdk::Display::default() else {
        crate::log::error("gtk", "no default display");
        std::process::exit(1);
    };

    let provider = CssProvider::new();
    provider.load_from_string(THEME_CSS);
    gtk::style_context_add_provider_for_display(
        &display,
        &provider,
        gtk::STYLE_PROVIDER_PRIORITY_APPLICATION,
    );

    #[cfg(feature = "config")]
    load_custom_css(&display);
}

#[cfg(feature = "config")]
fn load_custom_css(display: &gtk::gdk::Display) {
    let Some(base) = directories::BaseDirs::new() else {
        return;
    };
    let path = base.config_dir().join("tkbar").join("style.css");
    let css = match std::fs::read_to_string(&path) {
        Ok(css) => css,
        Err(err) if err.kind() == std::io::ErrorKind::NotFound => return,
        Err(err) => {
            crate::log::warn("config", &format!("cannot read {}: {err}", path.display()));
            return;
        },
    };

    let provider = CssProvider::new();
    provider.load_from_string(&css);
    gtk::style_context_add_provider_for_display(
        display,
        &provider,
        gtk::STYLE_PROVIDER_PRIORITY_USER,
    );
}

/// Exactly one color feature must be enabled; it selects the compiled-in base
/// stylesheet. Using `#[cfg]` (rather than runtime `cfg!`) means only the
/// selected stylesheet is `include_str!`-ed into the binary — the other nine are
/// not linked at all. `color_feature_count` + the const assert below enforce the
/// "exactly one" rule with a clear message; without it, zero colors would just
/// surface as a "cannot find `THEME_CSS`" error and two as a duplicate
/// definition.
#[cfg(feature = "black")]
const THEME_CSS: &str = include_str!("ui/styles/black.css");
#[cfg(feature = "blue")]
const THEME_CSS: &str = include_str!("ui/styles/blue.css");
#[cfg(feature = "cyan")]
const THEME_CSS: &str = include_str!("ui/styles/cyan.css");
#[cfg(feature = "green")]
const THEME_CSS: &str = include_str!("ui/styles/green.css");
#[cfg(feature = "orange")]
const THEME_CSS: &str = include_str!("ui/styles/orange.css");
#[cfg(feature = "pink")]
const THEME_CSS: &str = include_str!("ui/styles/pink.css");
#[cfg(feature = "purple")]
const THEME_CSS: &str = include_str!("ui/styles/purple.css");
#[cfg(feature = "red")]
const THEME_CSS: &str = include_str!("ui/styles/red.css");
#[cfg(feature = "white")]
const THEME_CSS: &str = include_str!("ui/styles/white.css");
#[cfg(feature = "yellow")]
const THEME_CSS: &str = include_str!("ui/styles/yellow.css");

const fn color_feature_count() -> usize {
    let mut n = 0;
    if cfg!(feature = "black") {
        n += 1;
    }
    if cfg!(feature = "blue") {
        n += 1;
    }
    if cfg!(feature = "cyan") {
        n += 1;
    }
    if cfg!(feature = "green") {
        n += 1;
    }
    if cfg!(feature = "orange") {
        n += 1;
    }
    if cfg!(feature = "pink") {
        n += 1;
    }
    if cfg!(feature = "purple") {
        n += 1;
    }
    if cfg!(feature = "red") {
        n += 1;
    }
    if cfg!(feature = "white") {
        n += 1;
    }
    if cfg!(feature = "yellow") {
        n += 1;
    }
    n
}

const _: () = assert!(
    color_feature_count() == 1,
    "exactly one color feature must be enabled: black, blue, cyan, green, orange, pink, purple, red, white, yellow"
);

const fn workspace_backend_count() -> usize {
    let mut n = 0;
    if cfg!(feature = "niri") {
        n += 1;
    }
    if cfg!(feature = "hyprland") {
        n += 1;
    }
    if cfg!(feature = "sway") {
        n += 1;
    }
    n
}

const _: () = assert!(
    workspace_backend_count() <= 1,
    "zero or one of the workspace backends must be enabled: niri, hyprland, sway"
);