nwg-notifications 0.3.4

use crate::state::NotificationState;
use crate::ui::dnd_menu;
use crate::ui::panel::NotificationPanel;
use gtk4::glib;
use nwg_common::signals;
use std::cell::RefCell;
use std::rc::Rc;
use std::sync::mpsc;

/// Commands the notification daemon responds to via signals.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum NotificationCommand {
    TogglePanel,
    ToggleDnd,
    ShowDndMenu,
}

/// Starts signal listener for notification-specific signals.
///
/// Must be called BEFORE app.connect_activate (before GTK starts).
pub fn start_signal_listener() -> Rc<mpsc::Receiver<NotificationCommand>> {
    let (tx, rx) = mpsc::channel();

    let sig_panel = signals::sig_notification_toggle();
    let sig_dnd = signals::sig_notification_dnd();
    let sig_menu = signals::sig_notification_dnd_menu();

    signals::setup_sigterm_handler();

    // Block notification signals in the main thread BEFORE spawning.
    // Uses raw libc because nix's Signal enum doesn't support RT signals.
    unsafe {
        let mut set: libc::sigset_t = std::mem::zeroed();
        libc::sigemptyset(&mut set);
        libc::sigaddset(&mut set, sig_panel);
        libc::sigaddset(&mut set, sig_dnd);
        libc::sigaddset(&mut set, sig_menu);
        libc::pthread_sigmask(libc::SIG_BLOCK, &set, std::ptr::null_mut());
    }

    std::thread::spawn(move || {
        loop {
            let mut sig: i32 = 0;
            // SAFETY: sigwait blocks until a signal from the set is pending.
            unsafe {
                let mut set: libc::sigset_t = std::mem::zeroed();
                libc::sigemptyset(&mut set);
                libc::sigaddset(&mut set, sig_panel);
                libc::sigaddset(&mut set, sig_dnd);
                libc::sigaddset(&mut set, sig_menu);
                libc::sigwait(&set, &mut sig);
            }

            let cmd = if sig == sig_panel {
                Some(NotificationCommand::TogglePanel)
            } else if sig == sig_dnd {
                Some(NotificationCommand::ToggleDnd)
            } else if sig == sig_menu {
                Some(NotificationCommand::ShowDndMenu)
            } else {
                None
            };

            if let Some(cmd) = cmd
                && tx.send(cmd).is_err()
            {
                break;
            }
        }
    });

    Rc::new(rx)
}

/// Polls the signal receiver on the GTK main thread.
pub fn poll_signals(
    sig_rx: &Rc<mpsc::Receiver<NotificationCommand>>,
    panel: &Rc<RefCell<NotificationPanel>>,
    state: &Rc<RefCell<NotificationState>>,
    on_state_change: &Rc<dyn Fn()>,
    dnd_menu: &Rc<RefCell<dnd_menu::DndMenu>>,
) {
    let panel = Rc::clone(panel);
    let state = Rc::clone(state);
    let rx = Rc::clone(sig_rx);
    let on_change = Rc::clone(on_state_change);
    let menu = Rc::clone(dnd_menu);

    glib::timeout_add_local(std::time::Duration::from_millis(100), move || {
        while let Ok(cmd) = rx.try_recv() {
            match cmd {
                NotificationCommand::TogglePanel => {
                    panel.borrow().toggle();
                }
                NotificationCommand::ToggleDnd => {
                    let new_dnd = !state.borrow().dnd;
                    state.borrow_mut().dnd = new_dnd;
                    log::info!(
                        "DND {} via signal",
                        if new_dnd { "enabled" } else { "disabled" }
                    );
                    on_change();
                }
                NotificationCommand::ShowDndMenu => {
                    menu.borrow().toggle();
                }
            }
        }
        glib::ControlFlow::Continue
    });
}