breadbar/src/main.rs

macro_rules! asset {
    ($n:literal) => {
        include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/assets/", $n))
    };
}

mod bar;
mod notifications;
mod osd;
mod theme;

use gtk4::prelude::*;
use gtk4_layer_shell::{Edge, Layer, LayerShell};
use hyprland::data::Workspace;
use hyprland::shared::WorkspaceId;
use relm4::prelude::*;
use std::cell::Cell;
use std::rc::Rc;

pub struct App {
    // ── Workspaces ────────────────────────────────────────────────────────
    workspaces: Vec<Workspace>,
    active_ws: WorkspaceId,
    workspace_box: gtk4::Box,
    button_map: std::collections::HashMap<WorkspaceId, gtk4::Button>,

    // ── Clock ─────────────────────────────────────────────────────────────
    time_str: String,
    clock_lbl: gtk4::Label,

    // ── Stats bar ─────────────────────────────────────────────────────────
    cpu_lbl: gtk4::Label,
    mem_lbl: gtk4::Label,
    pwr_lbl: gtk4::Label,
    bat_lbl: gtk4::Label,
    bat_img: gtk4::Image,
    bat_textures: std::collections::HashMap<usize, gtk4::gdk::Texture>,
    ac_img: gtk4::Image,
    bt_img: gtk4::Image,
    bt_textures: std::collections::HashMap<usize, gtk4::gdk::Texture>,
    wifi_lbl: gtk4::Label,
    wifi_img: gtk4::Image,
    wifi_textures: std::collections::HashMap<usize, gtk4::gdk::Texture>,

    // ── WiFi popover ──────────────────────────────────────────────────────
    wifi_popover_box: gtk4::Box,
    crumbs_status: Option<bar::wifi::CrumbsStatus>,
    wifi_popover_data: Option<bar::wifi::WifiPopoverData>,
    wifi_profile: Option<String>,
    current_ssid: String,

    // ── Media ─────────────────────────────────────────────────────────────
    media_widget: gtk4::Box,
    media_track_lbl: gtk4::Label,
    media_play_btn: gtk4::Button,
    media_last: Option<bar::media::MediaState>,
    media_paused_at: Option<std::time::Instant>,

    // ── Control panel ─────────────────────────────────────────────────────
    control_popover: gtk4::Popover,
    panel_vol_slider: gtk4::Scale,
    panel_bright_slider: gtk4::Scale,
    panel_loading: Rc<Cell<bool>>,
    panel_sink_store: gtk4::StringList,
    panel_sink_dropdown: gtk4::DropDown,
    panel_sink_signal: Option<gtk4::glib::SignalHandlerId>,
    panel_sinks: Vec<bar::control::AudioSink>,
    panel_temp_lbl: gtk4::Label,
    panel_gpu_lbl: gtk4::Label,
    panel_net_lbl: gtk4::Label,

    // ── Tray ──────────────────────────────────────────────────────────────
    tray_box: gtk4::Box,
    tray_items: std::collections::HashMap<String, gtk4::Button>,
}

#[derive(Debug)]
pub enum AppInput {
    WorkspaceList(Vec<Workspace>),
    ActiveWorkspace(WorkspaceId),
    ClockTick,
    StatsUpdate(bar::stats::Stats),
    TrayUpdate(bar::tray::TrayUpdate),
    CrumbsStatus(bar::wifi::CrumbsStatus),
    WifiPopoverData(bar::wifi::WifiPopoverData),
    SetProfile(String),
    MediaUpdate(bar::media::MediaState),
    ControlPanelData(bar::control::ControlPanelData),
}

#[relm4::component(pub)]
impl SimpleComponent for App {
    type Init = ();
    type Input = AppInput;
    type Output = ();

    view! {
        gtk::ApplicationWindow {
            add_css_class: "breadbar",
            set_title: Some("breadbar"),
            set_default_height: 32,

            #[name = "center_box"]
            gtk::CenterBox {
                #[wrap(Some)]
                set_start_widget = &gtk::Box {
                    set_orientation: gtk::Orientation::Horizontal,
                    set_spacing: 0,

                    #[name = "workspace_box"]
                    gtk::Box {
                        set_orientation: gtk::Orientation::Horizontal,
                        set_spacing: 4,
                    }
                },
            }
        }
    }

    fn init(
        _: Self::Init,
        root: Self::Root,
        sender: ComponentSender<Self>,
    ) -> ComponentParts<Self> {
        root.init_layer_shell();
        root.set_layer(Layer::Top);
        root.set_anchor(Edge::Top, true);
        root.set_anchor(Edge::Left, true);
        root.set_anchor(Edge::Right, true);
        root.set_exclusive_zone(32);

        // ── SVG icon sets ────────────────────────────────────────────────
        use bar::stats::{
            AC_POWER, BAT_HIGH, BAT_LOW, BAT_MID, BT_CONNECTED, BT_OFF, BT_ON, WIFI_MEDIUM,
            WIFI_OFF, WIFI_STRONG, WIFI_WEAK,
        };
        let bat_textures: std::collections::HashMap<usize, gtk4::gdk::Texture> =
            [BAT_HIGH, BAT_MID, BAT_LOW]
                .into_iter()
                .map(|p| (p.as_ptr() as usize, svg_texture(p)))
                .collect();
        let bt_textures: std::collections::HashMap<usize, gtk4::gdk::Texture> =
            [BT_OFF, BT_ON, BT_CONNECTED]
                .into_iter()
                .map(|p| (p.as_ptr() as usize, svg_texture(p)))
                .collect();
        let wifi_textures: std::collections::HashMap<usize, gtk4::gdk::Texture> =
            [WIFI_STRONG, WIFI_MEDIUM, WIFI_WEAK, WIFI_OFF]
                .into_iter()
                .map(|p| (p.as_ptr() as usize, svg_texture(p)))
                .collect();

        // ── Stat labels ──────────────────────────────────────────────────
        let cpu_lbl = stat_label();
        let mem_lbl = stat_label();
        let pwr_lbl = stat_label();
        let bat_lbl = stat_label();

        let bat_img = gtk4::Image::from_paintable(Some(
            bat_textures.get(&(BAT_MID.as_ptr() as usize)).unwrap(),
        ));
        let ac_img = gtk4::Image::from_paintable(Some(&svg_texture(AC_POWER)));
        ac_img.set_visible(false);
        let bt_img = gtk4::Image::from_paintable(Some(
            bt_textures.get(&(BT_OFF.as_ptr() as usize)).unwrap(),
        ));

        // ── WiFi pair + popover ──────────────────────────────────────────
        let wifi_lbl = gtk4::Label::new(None);
        wifi_lbl.add_css_class("stat-label");
        wifi_lbl.add_css_class("wifi-label");
        wifi_lbl.set_ellipsize(gtk4::pango::EllipsizeMode::End);
        wifi_lbl.set_max_width_chars(28);
        wifi_lbl.set_xalign(0.0);
        let wifi_img =
            gtk4::Image::from_paintable(Some(&svg_texture(asset!("WiFi Connecting.svg"))));

        let wifi_pair = gtk4::Box::new(gtk4::Orientation::Horizontal, 0);
        wifi_pair.add_css_class("stat-pair");
        wifi_pair.add_css_class("wifi-pair");
        wifi_img.add_css_class("stat-icon");
        wifi_pair.append(&wifi_img);
        wifi_pair.append(&wifi_lbl);

        let wifi_popover_box = gtk4::Box::new(gtk4::Orientation::Vertical, 0);
        wifi_popover_box.add_css_class("wifi-popover-inner");
        wifi_popover_box.set_margin_top(4);
        wifi_popover_box.set_margin_bottom(4);
        wifi_popover_box.set_margin_start(4);
        wifi_popover_box.set_margin_end(4);
        let loading_lbl = gtk4::Label::new(Some("Scanning…"));
        loading_lbl.add_css_class("wifi-popover-loading");
        wifi_popover_box.append(&loading_lbl);

        let wifi_popover = gtk4::Popover::new();
        wifi_popover.add_css_class("wifi-popover");
        wifi_popover.set_child(Some(&wifi_popover_box));
        wifi_popover.set_parent(&wifi_pair);

        let wpop = wifi_popover.clone();
        let gesture = gtk4::GestureClick::new();
        gesture.connect_released(move |_, _, _, _| {
            if wpop.is_visible() { wpop.popdown(); } else { wpop.popup(); }
        });
        wifi_pair.add_controller(gesture);

        let sender_wp = sender.clone();
        wifi_popover.connect_show(move |_| {
            bar::wifi::spawn_popover_load(sender_wp.clone());
        });

        // ── Media widget (center) ────────────────────────────────────────
        let media_widget = gtk4::Box::new(gtk4::Orientation::Horizontal, 4);
        media_widget.add_css_class("media-widget");
        media_widget.set_visible(false);

        let media_indicator = gtk4::Label::new(Some("▶"));
        media_indicator.add_css_class("media-indicator");

        let media_track_lbl = gtk4::Label::new(None);
        media_track_lbl.add_css_class("media-track-lbl");
        media_track_lbl.set_ellipsize(gtk4::pango::EllipsizeMode::End);
        media_track_lbl.set_max_width_chars(42);
        media_track_lbl.set_xalign(0.0);

        media_widget.append(&media_indicator);
        media_widget.append(&media_track_lbl);

        // Media controls popover
        let media_controls_box = gtk4::Box::new(gtk4::Orientation::Horizontal, 4);
        media_controls_box.add_css_class("media-controls");
        media_controls_box.set_margin_top(4);
        media_controls_box.set_margin_bottom(4);
        media_controls_box.set_margin_start(4);
        media_controls_box.set_margin_end(4);

        let prev_btn = gtk4::Button::with_label("⏮");
        prev_btn.add_css_class("flat");
        prev_btn.add_css_class("media-btn");
        prev_btn.connect_clicked(|_| bar::media::spawn_cmd("previous"));

        let media_play_btn = gtk4::Button::with_label("⏸");
        media_play_btn.add_css_class("flat");
        media_play_btn.add_css_class("media-btn");
        media_play_btn.add_css_class("media-play-btn");
        media_play_btn.connect_clicked(|_| bar::media::spawn_cmd("play-pause"));

        let next_btn = gtk4::Button::with_label("⏭");
        next_btn.add_css_class("flat");
        next_btn.add_css_class("media-btn");
        next_btn.connect_clicked(|_| bar::media::spawn_cmd("next"));

        media_controls_box.append(&prev_btn);
        media_controls_box.append(&media_play_btn);
        media_controls_box.append(&next_btn);

        let media_popover = gtk4::Popover::new();
        media_popover.add_css_class("media-popover");
        media_popover.set_child(Some(&media_controls_box));
        media_popover.set_parent(&media_widget);

        let mpop = media_popover.clone();
        let mgesture = gtk4::GestureClick::new();
        mgesture.connect_released(move |_, _, _, _| {
            if mpop.is_visible() { mpop.popdown(); } else { mpop.popup(); }
        });
        media_widget.add_controller(mgesture);

        // Clock label
        let clock_lbl = gtk4::Label::new(Some(&bar::clock::current()));
        clock_lbl.add_css_class("clock-label");

        // Center area: [media_widget · clock]
        let center_area = gtk4::Box::new(gtk4::Orientation::Horizontal, 10);
        center_area.add_css_class("center-area");
        center_area.append(&media_widget);
        center_area.append(&clock_lbl);

        // ── Stats box (right side) ───────────────────────────────────────
        let stats_box = gtk4::Box::new(gtk4::Orientation::Horizontal, 0);
        stats_box.add_css_class("stats-box");
        stats_box.append(&stat_pair(asset!("CPU.svg"), &cpu_lbl));
        stats_box.append(&stat_pair(asset!("RAM Usage.svg"), &mem_lbl));
        stats_box.append(&stat_pair(asset!("Power Draw.svg"), &pwr_lbl));

        let bat_box = gtk4::Box::new(gtk4::Orientation::Horizontal, 0);
        bat_box.add_css_class("stat-pair");
        bat_img.add_css_class("stat-icon");
        bat_lbl.add_css_class("stat-label");
        ac_img.add_css_class("stat-icon");
        ac_img.set_margin_start(6);
        bat_box.append(&bat_img);
        bat_box.append(&bat_lbl);
        bat_box.append(&ac_img);
        stats_box.append(&bat_box);

        bt_img.add_css_class("bt-icon");
        bt_img.add_css_class("clickable");
        let bt_gesture = gtk4::GestureClick::new();
        bt_gesture.connect_released(|_, _, _, _| {
            relm4::spawn(async {
                let _ = tokio::process::Command::new("blueman-manager").spawn();
            });
        });
        bt_img.add_controller(bt_gesture);
        stats_box.append(&bt_img);
        stats_box.append(&wifi_pair);

        // ── Control panel popover ────────────────────────────────────────
        let panel_inner = gtk4::Box::new(gtk4::Orientation::Vertical, 0);
        panel_inner.add_css_class("control-panel-inner");

        // Volume row
        let vol_row = build_slider_row("🔊", 0.0, 1.5, 0.02);
        let panel_vol_slider = vol_row.1.clone();
        panel_inner.append(&vol_row.0);

        // Brightness row
        let bright_row = build_slider_row("☀", 0.0, 1.0, 0.02);
        let panel_bright_slider = bright_row.1.clone();
        panel_inner.append(&bright_row.0);

        panel_inner.append(&gtk4::Separator::new(gtk4::Orientation::Horizontal));

        // Stats section
        let stats_section = gtk4::Box::new(gtk4::Orientation::Vertical, 6);
        stats_section.add_css_class("control-panel-stats");

        let panel_temp_lbl = gtk4::Label::new(Some("CPU  —"));
        panel_temp_lbl.add_css_class("control-panel-stat");
        panel_temp_lbl.set_xalign(0.0);

        let panel_gpu_lbl = gtk4::Label::new(Some("GPU  —"));
        panel_gpu_lbl.add_css_class("control-panel-stat");
        panel_gpu_lbl.set_xalign(0.0);

        let panel_net_lbl = gtk4::Label::new(Some("↓ —  ↑ —"));
        panel_net_lbl.add_css_class("control-panel-stat");
        panel_net_lbl.set_xalign(0.0);

        stats_section.append(&panel_temp_lbl);
        stats_section.append(&panel_gpu_lbl);
        stats_section.append(&panel_net_lbl);
        panel_inner.append(&stats_section);

        panel_inner.append(&gtk4::Separator::new(gtk4::Orientation::Horizontal));

        // Audio output section
        let sink_section = gtk4::Box::new(gtk4::Orientation::Vertical, 4);
        sink_section.add_css_class("control-panel-section");
        let sink_header = gtk4::Label::new(Some("Audio Output"));
        sink_header.add_css_class("control-panel-section-header");
        sink_header.set_xalign(0.0);

        let panel_sink_store = gtk4::StringList::new(&[]);
        let panel_sink_dropdown = gtk4::DropDown::new(
            Some(panel_sink_store.clone().upcast::<gtk4::gio::ListModel>()),
            Option::<gtk4::Expression>::None,
        );
        panel_sink_dropdown.add_css_class("control-panel-sink-dropdown");
        panel_sink_dropdown.set_hexpand(true);

        sink_section.append(&sink_header);
        sink_section.append(&panel_sink_dropdown);
        panel_inner.append(&sink_section);

        panel_inner.append(&gtk4::Separator::new(gtk4::Orientation::Horizontal));

        // Tray section
        let tray_section = gtk4::Box::new(gtk4::Orientation::Vertical, 4);
        tray_section.add_css_class("control-panel-section");
        let tray_header = gtk4::Label::new(Some("Apps"));
        tray_header.add_css_class("control-panel-section-header");
        tray_header.set_xalign(0.0);
        let tray_box = gtk4::Box::new(gtk4::Orientation::Horizontal, 4);
        tray_box.add_css_class("tray-box");
        tray_section.append(&tray_header);
        tray_section.append(&tray_box);
        panel_inner.append(&tray_section);

        let control_popover = gtk4::Popover::new();
        control_popover.add_css_class("control-panel");
        control_popover.set_child(Some(&panel_inner));

        // Hamburger button
        let hamburger_btn = gtk4::Button::with_label("☰");
        hamburger_btn.add_css_class("flat");
        hamburger_btn.add_css_class("control-panel-btn");

        control_popover.set_parent(&hamburger_btn);

        let cpop = control_popover.clone();
        hamburger_btn.connect_clicked(move |_| {
            if cpop.is_visible() { cpop.popdown(); } else { cpop.popup(); }
        });

        let sender_cp = sender.clone();
        control_popover.connect_show(move |_| {
            bar::control::spawn_load(sender_cp.clone());
        });

        // Slider signals — use Rc<Cell<bool>> to suppress feedback during data load
        let panel_loading = Rc::new(Cell::new(false));

        let loading_v = panel_loading.clone();
        panel_vol_slider.connect_value_changed(move |s| {
            if loading_v.get() { return; }
            bar::control::spawn_set_volume(s.value());
        });

        let loading_b = panel_loading.clone();
        panel_bright_slider.connect_value_changed(move |s| {
            if loading_b.get() { return; }
            bar::control::spawn_set_brightness(s.value());
        });

        stats_box.append(&hamburger_btn);

        // ── Assemble ─────────────────────────────────────────────────────
        let widgets = view_output!();
        widgets.center_box.set_center_widget(Some(&center_area));
        widgets.center_box.set_end_widget(Some(&stats_box));

        let mut model = App {
            workspaces: vec![],
            active_ws: 1,
            workspace_box: gtk4::Box::new(gtk4::Orientation::Horizontal, 4),
            button_map: std::collections::HashMap::new(),
            time_str: bar::clock::current(),
            clock_lbl,
            cpu_lbl,
            mem_lbl,
            pwr_lbl,
            bat_lbl,
            bat_img,
            bat_textures,
            ac_img,
            bt_img,
            bt_textures,
            wifi_lbl,
            wifi_img,
            wifi_textures,
            wifi_popover_box,
            crumbs_status: None,
            wifi_popover_data: None,
            wifi_profile: None,
            current_ssid: "—".to_string(),
            media_widget,
            media_track_lbl,
            media_play_btn,
            media_last: None,
            media_paused_at: None,
            control_popover,
            panel_vol_slider,
            panel_bright_slider,
            panel_loading,
            panel_sink_store,
            panel_sink_dropdown,
            panel_sink_signal: None,
            panel_sinks: vec![],
            panel_temp_lbl,
            panel_gpu_lbl,
            panel_net_lbl,
            tray_box,
            tray_items: std::collections::HashMap::new(),
        };
        model.workspace_box = widgets.workspace_box.clone();

        theme::apply();
        bar::workspaces::spawn_watcher(sender.clone());
        bar::clock::spawn_ticker(sender.clone());
        bar::stats::spawn_poller(sender.clone());
        bar::tray::spawn_watcher(sender.clone());
        bar::wifi::spawn_status_poller(sender.clone());
        bar::media::spawn_poller(sender.clone());
        notifications::spawn();
        osd::spawn();

        ComponentParts { model, widgets }
    }

    fn update(&mut self, msg: Self::Input, sender: ComponentSender<Self>) {
        match msg {
            AppInput::WorkspaceList(list) => {
                let mut sorted = list;
                sorted.sort_by_key(|w| w.id);
                self.workspaces = sorted;
                self.rebuild_buttons();
            }
            AppInput::ActiveWorkspace(id) => {
                if let Some(old) = self.button_map.get(&self.active_ws) {
                    old.remove_css_class("active");
                }
                self.active_ws = id;
                if let Some(btn) = self.button_map.get(&self.active_ws) {
                    btn.add_css_class("active");
                }
            }
            AppInput::ClockTick => {
                self.time_str = bar::clock::current();
                self.clock_lbl.set_label(&self.time_str);
            }
            AppInput::StatsUpdate(stats) => {
                self.cpu_lbl.set_label(&stats.cpu);
                self.mem_lbl.set_label(&stats.mem);
                self.pwr_lbl.set_label(&stats.power);
                self.bat_lbl.set_label(&stats.bat);
                if let Some(tex) = self.bat_textures.get(&(stats.bat_icon.as_ptr() as usize)) {
                    self.bat_img.set_paintable(Some(tex));
                }
                self.ac_img.set_visible(stats.ac_connected);
                if let Some(tex) = self.bt_textures.get(&(stats.bt_icon.as_ptr() as usize)) {
                    self.bt_img.set_paintable(Some(tex));
                }
                self.current_ssid = stats.wifi_ssid.clone();
                if stats.wifi_profile.is_some() {
                    self.wifi_profile = stats.wifi_profile;
                }
                self.apply_wifi_label();
                let internet_ok = self
                    .crumbs_status
                    .as_ref()
                    .map(|s| s.internet && !s.captive_portal)
                    .unwrap_or(true);
                let icon = if !internet_ok && stats.wifi_ssid != "—" {
                    bar::stats::WIFI_OFF
                } else {
                    stats.wifi_icon
                };
                if let Some(tex) = self.wifi_textures.get(&(icon.as_ptr() as usize)) {
                    self.wifi_img.set_paintable(Some(tex));
                }

                // Live-update control panel stats while open
                if self.control_popover.is_visible() {
                    match stats.cpu_temp {
                        Some(t) => self.panel_temp_lbl.set_label(&format!("CPU  {t:.0}°C")),
                        None => self.panel_temp_lbl.set_label("CPU  —"),
                    }
                    match stats.gpu_usage {
                        Some(g) => self.panel_gpu_lbl.set_label(&format!("GPU  {g}%")),
                        None => self.panel_gpu_lbl.set_label("GPU  —"),
                    }
                    self.panel_net_lbl.set_label(&format!(
                        "↓ {}  ↑ {}",
                        fmt_speed(stats.net_rx_kbs),
                        fmt_speed(stats.net_tx_kbs),
                    ));
                }
            }
            AppInput::TrayUpdate(bar::tray::TrayUpdate::Add { id, icon, title }) => {
                if self.tray_items.contains_key(&id) {
                    return;
                }
                let btn = gtk4::Button::new();
                btn.add_css_class("tray-btn");
                btn.set_child(Some(&bar::tray::make_tray_image(icon.as_ref())));
                if !title.is_empty() {
                    btn.set_tooltip_text(Some(&title));
                }
                let id_click = id.clone();
                btn.connect_clicked(move |_| bar::tray::spawn_activate(id_click.clone()));
                self.tray_box.append(&btn);
                self.tray_items.insert(id, btn);
            }
            AppInput::TrayUpdate(bar::tray::TrayUpdate::Remove { id }) => {
                if let Some(btn) = self.tray_items.remove(&id) {
                    self.tray_box.remove(&btn);
                }
            }
            AppInput::CrumbsStatus(status) => {
                self.crumbs_status = Some(status);
            }
            AppInput::WifiPopoverData(data) => {
                self.wifi_popover_data = Some(data);
                self.rebuild_wifi_popover(&sender);
            }
            AppInput::SetProfile(name) => {
                self.wifi_profile = Some(name);
                self.apply_wifi_label();
            }
            AppInput::MediaUpdate(state) => {
                if state.has_player {
                    let label = if state.artist.is_empty() {
                        state.title.clone()
                    } else {
                        format!("{} · {}", state.artist, state.title)
                    };
                    self.media_track_lbl.set_label(&label);
                    self.media_play_btn
                        .set_label(if state.playing { "⏸" } else { "▶" });

                    if state.playing {
                        self.media_paused_at = None;
                    } else if self.media_paused_at.is_none() {
                        self.media_paused_at = Some(std::time::Instant::now());
                    }

                    let within_linger = self
                        .media_paused_at
                        .map_or(true, |t| t.elapsed().as_secs() < 30 * 60);
                    self.media_last = Some(state);
                    self.media_widget.set_visible(within_linger);
                } else {
                    // Player gone — honour linger from last pause
                    if let Some(paused_at) = self.media_paused_at {
                        if paused_at.elapsed().as_secs() < 30 * 60 {
                            self.media_widget.set_visible(true);
                        } else {
                            self.media_widget.set_visible(false);
                            self.media_last = None;
                            self.media_paused_at = None;
                        }
                    } else {
                        self.media_widget.set_visible(false);
                        self.media_last = None;
                    }
                }
            }
            AppInput::ControlPanelData(data) => {
                // Suppress slider value-changed signals during programmatic update
                self.panel_loading.set(true);
                self.panel_vol_slider.set_value(data.volume);
                self.panel_bright_slider.set_value(data.brightness);
                self.panel_loading.set(false);

                // Rebuild sink dropdown — disconnect, repopulate, reconnect
                if let Some(id) = self.panel_sink_signal.take() {
                    self.panel_sink_dropdown.disconnect(id);
                }
                // Clear store
                let n = self.panel_sink_store.n_items();
                for i in (0..n).rev() {
                    self.panel_sink_store.remove(i);
                }
                for sink in &data.sinks {
                    self.panel_sink_store.append(&sink.description);
                }
                if let Some(idx) = data.sinks.iter().position(|s| s.is_default) {
                    self.panel_sink_dropdown.set_selected(idx as u32);
                }
                self.panel_sinks = data.sinks;

                let sinks = self.panel_sinks.clone();
                let id = self.panel_sink_dropdown.connect_selected_notify(move |dd| {
                    let idx = dd.selected() as usize;
                    if let Some(sink) = sinks.get(idx) {
                        bar::control::spawn_set_sink(sink.name.clone());
                    }
                });
                self.panel_sink_signal = Some(id);
            }
        }
    }
}

impl App {
    fn apply_wifi_label(&self) {
        let label = match &self.wifi_profile {
            Some(p) => format!("{p} · {}", self.current_ssid),
            None => self.current_ssid.clone(),
        };
        self.wifi_lbl.set_label(&label);
    }

    fn rebuild_buttons(&mut self) {
        while let Some(child) = self.workspace_box.first_child() {
            self.workspace_box.remove(&child);
        }
        self.button_map.clear();
        for ws in &self.workspaces {
            let btn = bar::workspaces::make_button(ws.id, &ws.name, self.active_ws);
            self.workspace_box.append(&btn);
            self.button_map.insert(ws.id, btn);
        }
    }

    fn rebuild_wifi_popover(&mut self, sender: &ComponentSender<Self>) {
        while let Some(child) = self.wifi_popover_box.first_child() {
            self.wifi_popover_box.remove(&child);
        }

        if let Some(st) = &self.crumbs_status {
            let header = gtk4::Box::new(gtk4::Orientation::Vertical, 2);
            header.add_css_class("wifi-popover-header");
            header.set_margin_bottom(6);

            let ssid_str = st.ssid.as_deref().filter(|s| !s.is_empty()).unwrap_or("—");
            let ssid_lbl = gtk4::Label::new(Some(ssid_str));
            ssid_lbl.add_css_class("wifi-popover-ssid");
            ssid_lbl.set_xalign(0.0);
            header.append(&ssid_lbl);

            if let Some(ip) = &st.ip {
                let ip_lbl = gtk4::Label::new(Some(ip.as_str()));
                ip_lbl.add_css_class("wifi-popover-ip");
                ip_lbl.set_xalign(0.0);
                header.append(&ip_lbl);
            }

            let mut parts = Vec::new();
            if st.captive_portal {
                parts.push("captive portal");
            } else if st.internet {
                parts.push("internet ✓");
            } else {
                parts.push("internet ✗");
            }
            if st.tailscale_required {
                parts.push(if st.tailscale_ok { "tailscale ✓" } else { "tailscale ✗" });
            }
            let status_lbl = gtk4::Label::new(Some(&parts.join("   ")));
            status_lbl.add_css_class("wifi-popover-status");
            status_lbl.set_xalign(0.0);
            header.append(&status_lbl);

            self.wifi_popover_box.append(&header);
            self.wifi_popover_box
                .append(&gtk4::Separator::new(gtk4::Orientation::Horizontal));
        }

        let Some(data) = &self.wifi_popover_data else {
            let lbl = gtk4::Label::new(Some("Scanning…"));
            lbl.add_css_class("wifi-popover-loading");
            self.wifi_popover_box.append(&lbl);
            return;
        };

        let ph = gtk4::Label::new(Some("Profiles"));
        ph.add_css_class("wifi-popover-section");
        ph.set_xalign(0.0);
        ph.set_margin_top(6);
        ph.set_margin_bottom(2);
        self.wifi_popover_box.append(&ph);

        for (name, active) in &data.profiles {
            let row = gtk4::Button::new();
            row.add_css_class("flat");
            row.add_css_class("wifi-popover-row");
            if *active {
                row.add_css_class("wifi-popover-row-active");
            }
            let lbl = gtk4::Label::new(Some(&format!(
                "{}{}",
                if *active { "● " } else { "  " },
                name
            )));
            lbl.set_xalign(0.0);
            row.set_child(Some(&lbl));

            let name_clone = name.clone();
            let sender_clone = sender.clone();
            row.connect_clicked(move |btn| {
                sender_clone.input(AppInput::SetProfile(name_clone.clone()));
                bar::wifi::spawn_profile_set(name_clone.clone());
                close_parent_popover(btn);
            });
            self.wifi_popover_box.append(&row);
        }

        if !data.scan.is_empty() {
            self.wifi_popover_box
                .append(&gtk4::Separator::new(gtk4::Orientation::Horizontal));
            let nh = gtk4::Label::new(Some("Nearby"));
            nh.add_css_class("wifi-popover-section");
            nh.set_xalign(0.0);
            nh.set_margin_top(6);
            nh.set_margin_bottom(2);
            self.wifi_popover_box.append(&nh);

            for entry in &data.scan {
                let row = gtk4::Button::new();
                row.add_css_class("flat");
                row.add_css_class("wifi-popover-row");
                if !entry.saved {
                    row.add_css_class("wifi-popover-row-unsaved");
                    row.set_sensitive(false);
                }
                let is_current = entry.ssid == self.current_ssid;
                if is_current {
                    row.add_css_class("wifi-popover-row-active");
                }

                let row_box = gtk4::Box::new(gtk4::Orientation::Horizontal, 4);
                let icon_svg = wifi_icon_for_signal(entry.signal);
                if let Some(tex) = self.wifi_textures.get(&(icon_svg.as_ptr() as usize)) {
                    let img = gtk4::Image::from_paintable(Some(tex));
                    img.add_css_class("stat-icon");
                    row_box.append(&img);
                }
                let lbl = gtk4::Label::new(Some(&format!(
                    "{}{}",
                    if is_current { "● " } else { "  " },
                    entry.ssid,
                )));
                lbl.set_xalign(0.0);
                row_box.append(&lbl);
                row.set_child(Some(&row_box));

                if entry.saved {
                    let ssid_clone = entry.ssid.clone();
                    row.connect_clicked(move |btn| {
                        bar::wifi::spawn_join(ssid_clone.clone());
                        close_parent_popover(btn);
                    });
                }
                self.wifi_popover_box.append(&row);
            }
        }

        self.wifi_popover_box.set_visible(true);
    }
}

// ── Helpers ───────────────────────────────────────────────────────────────────

fn build_slider_row(icon: &str, min: f64, max: f64, step: f64) -> (gtk4::Box, gtk4::Scale) {
    let row = gtk4::Box::new(gtk4::Orientation::Horizontal, 8);
    row.add_css_class("control-panel-row");
    row.set_margin_top(2);
    row.set_margin_bottom(2);

    let icon_lbl = gtk4::Label::new(Some(icon));
    icon_lbl.add_css_class("control-panel-row-icon");
    icon_lbl.set_width_chars(2);

    let slider = gtk4::Scale::with_range(gtk4::Orientation::Horizontal, min, max, step);
    slider.set_draw_value(false);
    slider.set_hexpand(true);
    slider.set_width_request(180);
    slider.add_css_class("control-panel-slider");

    row.append(&icon_lbl);
    row.append(&slider);
    (row, slider)
}

fn fmt_speed(kbs: f32) -> String {
    if kbs >= 1024.0 {
        format!("{:.1} MB/s", kbs / 1024.0)
    } else {
        format!("{:.0} KB/s", kbs)
    }
}

fn wifi_icon_for_signal(pct: u8) -> &'static str {
    use bar::stats::{WIFI_MEDIUM, WIFI_OFF, WIFI_STRONG, WIFI_WEAK};
    match pct {
        75..=100 => WIFI_STRONG,
        50..=74 => WIFI_MEDIUM,
        25..=49 => WIFI_WEAK,
        _ => WIFI_OFF,
    }
}

fn close_parent_popover(widget: &gtk4::Button) {
    if let Some(w) = widget.ancestor(gtk4::Popover::static_type()) {
        if let Ok(p) = w.downcast::<gtk4::Popover>() {
            p.popdown();
        }
    }
}

fn stat_pair(icon_svg: &str, label: &gtk4::Label) -> gtk4::Box {
    let pair = gtk4::Box::new(gtk4::Orientation::Horizontal, 0);
    pair.add_css_class("stat-pair");
    let img = gtk4::Image::from_paintable(Some(&svg_texture(icon_svg)));
    img.add_css_class("stat-icon");
    pair.append(&img);
    pair.append(label);
    pair
}

fn svg_texture(svg_src: &str) -> gtk4::gdk::Texture {
    use resvg::{tiny_skia, usvg};
    let fg = theme::fg_color();
    let svg = svg_src
        .replace("currentColor", &fg)
        .replace(r#"width="24" height="24""#, r#"width="16" height="16""#);
    let tree = usvg::Tree::from_str(&svg, &usvg::Options::default()).expect("parse svg");
    let size = tree.size().to_int_size();
    let (w, h) = (size.width(), size.height());
    let mut pixmap = tiny_skia::Pixmap::new(w, h).expect("alloc pixmap");
    resvg::render(&tree, tiny_skia::Transform::identity(), &mut pixmap.as_mut());
    let bytes = gtk4::glib::Bytes::from_owned(pixmap.take());
    gtk4::gdk::MemoryTexture::new(
        w as i32,
        h as i32,
        gtk4::gdk::MemoryFormat::R8g8b8a8Premultiplied,
        &bytes,
        (w * 4) as usize,
    )
    .upcast()
}

fn stat_label() -> gtk4::Label {
    let lbl = gtk4::Label::new(None);
    lbl.add_css_class("stat-label");
    lbl.set_xalign(0.0);
    lbl
}

fn main() {
    relm4::spawn(async {
        use tokio::signal::unix::{signal, SignalKind};
        let mut stream = signal(SignalKind::hangup()).expect("SIGHUP handler");
        loop {
            stream.recv().await;
            gtk4::glib::MainContext::default().invoke(theme::apply);
        }
    });

    let app = RelmApp::new("sh.breadway.breadbar");
    app.run::<App>(());
}