ax-task 0.6.7

use core::sync::atomic::{AtomicBool, Ordering};

use crate::WaitQueue;

/// IRQ-safe deferred notification primitive.
///
/// `IrqNotify` separates a hard-IRQ notification from the slow work that must
/// run in task context. IRQ handlers call [`notify_irq`](Self::notify_irq) to
/// publish a pending bit and wake a deferred worker. The worker then drains the
/// bit and performs deferred work, such as consuming acknowledged device
/// completions.
pub struct IrqNotify {
    pending: AtomicBool,
    wait: WaitQueue,
}

impl Default for IrqNotify {
    fn default() -> Self {
        Self::new()
    }
}

impl IrqNotify {
    /// Creates an empty notification object.
    pub const fn new() -> Self {
        Self {
            pending: AtomicBool::new(false),
            wait: WaitQueue::new(),
        }
    }

    /// Publishes a pending notification from IRQ context.
    ///
    /// This method is IRQ-safe: it does not allocate, does not call arbitrary
    /// wakers, and does not perform slow poll wakeups. Repeated notifications
    /// coalesce into one pending bit until a worker drains it.
    pub fn notify_irq(&self) {
        self.pending.store(true, Ordering::Release);
        self.wait.notify_one_from_irq();
    }

    /// Publishes a pending notification from task context.
    ///
    /// Prefer [`notify_irq`](Self::notify_irq) inside hard IRQ callbacks. This
    /// method exists for task/deferred code that wants the same coalescing
    /// behavior while still allowing the scheduler to observe a normal wake.
    pub fn notify(&self) {
        self.pending.store(true, Ordering::Release);
        self.wait.notify_one(true);
    }

    /// Returns whether a notification is currently pending.
    pub fn is_pending(&self) -> bool {
        self.pending.load(Ordering::Acquire)
    }

    /// Drains the pending bit.
    ///
    /// Returns `true` if at least one notification was pending.
    pub fn drain(&self) -> bool {
        self.pending.swap(false, Ordering::AcqRel)
    }

    /// Blocks until at least one pending notification is available, then drains it.
    #[track_caller]
    pub fn wait(&self) {
        self.wait.wait_until(|| self.drain());
    }

    /// Blocks until a pending notification is consumed or `duration` elapses.
    ///
    /// Returns `true` only when the deadline elapsed without consuming a
    /// notification.
    #[track_caller]
    pub fn wait_timeout(&self, duration: core::time::Duration) -> bool {
        self.wait.wait_timeout_until(duration, || self.drain())
    }
}

#[cfg(axtest)]
pub(crate) fn irq_notify_constructor_and_pending_hold_for_test() -> bool {
    // Test IrqNotify::new() creates a non-pending instance
    let notify = IrqNotify::new();
    assert!(!notify.is_pending());

    // Test Default trait
    let default_notify = IrqNotify::default();
    assert!(!default_notify.is_pending());

    true
}

#[cfg(axtest)]
pub(crate) fn irq_notify_drain_logic_hold_for_test() -> bool {
    // Test drain on a fresh IrqNotify returns false (nothing pending)
    let notify = IrqNotify::new();
    assert!(!notify.drain()); // Nothing to drain
    assert!(!notify.is_pending()); // Still not pending after drain

    true
}