ironbar/src/bar.rs

use crate::config::{BarPosition, ModuleConfig};
use crate::modules::{Module, ModuleInfo, ModuleLocation};
use crate::Config;
use color_eyre::Result;
use gtk::gdk::Monitor;
use gtk::prelude::*;
use gtk::{Application, ApplicationWindow, Orientation};
use tracing::{debug, info};

pub fn create_bar(
    app: &Application,
    monitor: &Monitor,
    monitor_name: &str,
    config: Config,
) -> Result<()> {
    let win = ApplicationWindow::builder().application(app).build();

    setup_layer_shell(&win, monitor, &config.position);

    let content = gtk::Box::builder()
        .orientation(Orientation::Horizontal)
        .spacing(0)
        .hexpand(false)
        .height_request(config.height)
        .name("bar")
        .build();

    let left = gtk::Box::builder().spacing(0).name("left").build();
    let center = gtk::Box::builder().spacing(0).name("center").build();
    let right = gtk::Box::builder().spacing(0).name("right").build();

    content.style_context().add_class("container");
    left.style_context().add_class("container");
    center.style_context().add_class("container");
    right.style_context().add_class("container");

    content.add(&left);
    content.set_center_widget(Some(&center));
    content.pack_end(&right, false, false, 0);

    load_modules(&left, &center, &right, app, config, monitor, monitor_name)?;
    win.add(&content);

    win.connect_destroy_event(|_, _| {
        info!("Shutting down");
        gtk::main_quit();
        Inhibit(false)
    });

    debug!("Showing bar");
    win.show_all();

    Ok(())
}

fn load_modules(
    left: &gtk::Box,
    center: &gtk::Box,
    right: &gtk::Box,
    app: &Application,
    config: Config,
    monitor: &Monitor,
    output_name: &str,
) -> Result<()> {
    if let Some(modules) = config.left {
        let info = ModuleInfo {
            app,
            location: ModuleLocation::Left,
            bar_position: &config.position,
            monitor,
            output_name,
        };

        add_modules(left, modules, &info)?;
    }

    if let Some(modules) = config.center {
        let info = ModuleInfo {
            app,
            location: ModuleLocation::Center,
            bar_position: &config.position,
            monitor,
            output_name,
        };

        add_modules(center, modules, &info)?;
    }

    if let Some(modules) = config.right {
        let info = ModuleInfo {
            app,
            location: ModuleLocation::Right,
            bar_position: &config.position,
            monitor,
            output_name,
        };

        add_modules(right, modules, &info)?;
    }

    Ok(())
}

fn add_modules(content: &gtk::Box, modules: Vec<ModuleConfig>, info: &ModuleInfo) -> Result<()> {
    macro_rules! add_module {
        ($module:expr, $name:literal) => {{
            let widget = $module.into_widget(&info)?;
            widget.set_widget_name($name);
            content.add(&widget);
            debug!("Added module of type {}", $name);
        }};
    }

    for config in modules {
        match config {
            ModuleConfig::Clock(module) => add_module!(module, "clock"),
            ModuleConfig::Mpd(module) => add_module!(module, "mpd"),
            ModuleConfig::Tray(module) => add_module!(module, "tray"),
            ModuleConfig::Workspaces(module) => add_module!(module, "workspaces"),
            ModuleConfig::SysInfo(module) => add_module!(module, "sysinfo"),
            ModuleConfig::Launcher(module) => add_module!(module, "launcher"),
            ModuleConfig::Script(module) => add_module!(module, "script"),
            ModuleConfig::Focused(module) => add_module!(module, "focused"),
        }
    }

    Ok(())
}

fn setup_layer_shell(win: &ApplicationWindow, monitor: &Monitor, position: &BarPosition) {
    gtk_layer_shell::init_for_window(win);
    gtk_layer_shell::set_monitor(win, monitor);
    gtk_layer_shell::set_layer(win, gtk_layer_shell::Layer::Top);
    gtk_layer_shell::auto_exclusive_zone_enable(win);

    gtk_layer_shell::set_margin(win, gtk_layer_shell::Edge::Top, 0);
    gtk_layer_shell::set_margin(win, gtk_layer_shell::Edge::Bottom, 0);
    gtk_layer_shell::set_margin(win, gtk_layer_shell::Edge::Left, 0);
    gtk_layer_shell::set_margin(win, gtk_layer_shell::Edge::Right, 0);

    gtk_layer_shell::set_anchor(
        win,
        gtk_layer_shell::Edge::Top,
        position == &BarPosition::Top,
    );
    gtk_layer_shell::set_anchor(
        win,
        gtk_layer_shell::Edge::Bottom,
        position == &BarPosition::Bottom,
    );
    gtk_layer_shell::set_anchor(win, gtk_layer_shell::Edge::Left, true);
    gtk_layer_shell::set_anchor(win, gtk_layer_shell::Edge::Right, true);
}