quantoxide 0.5.5

use std::{
    fmt,
    sync::{Arc, Mutex, MutexGuard},
};

use tokio::sync::broadcast;

use crate::sync::SyncStatusNotSynced;

use super::{
    core::Signal,
    process::error::{SignalProcessFatalError, SignalProcessRecoverableError},
};

/// Detailed status when signal evaluation is not actively running.
///
/// Represents various states during the signal process lifecycle before achieving active signal
/// evaluation.
#[derive(Debug, Clone)]
pub enum LiveSignalStatusNotRunning {
    /// Signal process has not been started yet.
    NotInitiated,
    /// Signal process is initializing.
    Starting,
    /// Signal process is waiting for sync to complete before evaluating signals.
    WaitingForSync(SyncStatusNotSynced),
    /// Signal process encountered a recoverable error.
    Failed(Arc<SignalProcessRecoverableError>),
    /// Signal process is restarting after an error.
    Restarting,
}

impl PartialEq for LiveSignalStatusNotRunning {
    fn eq(&self, other: &Self) -> bool {
        match (self, other) {
            (Self::NotInitiated, Self::NotInitiated)
            | (Self::Starting, Self::Starting)
            | (Self::Restarting, Self::Restarting) => true,
            (Self::WaitingForSync(a), Self::WaitingForSync(b)) => a == b,
            (Self::Failed(a), Self::Failed(b)) => Arc::ptr_eq(a, b),
            _ => false,
        }
    }
}

impl fmt::Display for LiveSignalStatusNotRunning {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::NotInitiated => write!(f, "Not initiated"),
            Self::Starting => write!(f, "Starting"),
            Self::WaitingForSync(status) => {
                write!(f, "Waiting for sync ({status})")
            }
            Self::Failed(error) => write!(f, "Failed: {error}"),
            Self::Restarting => write!(f, "Restarting"),
        }
    }
}

/// Overall status of the live signal evaluation process.
///
/// Represents the high-level state of the signal process, including active evaluation, completion,
/// and shutdown states.
#[derive(Debug, Clone)]
pub enum LiveSignalStatus {
    /// Signal evaluation is not actively running.
    NotRunning(LiveSignalStatusNotRunning),
    /// Signal evaluation is actively running and generating signals.
    Running,
    /// Shutdown has been requested and is in progress.
    ShutdownInitiated,
    /// Signal process has been gracefully shut down.
    Shutdown,
    /// Signal process terminated due to a fatal error.
    Terminated(Arc<SignalProcessFatalError>),
}

impl PartialEq for LiveSignalStatus {
    fn eq(&self, other: &Self) -> bool {
        match (self, other) {
            (Self::NotRunning(a), Self::NotRunning(b)) => a == b,
            (Self::Running, Self::Running)
            | (Self::ShutdownInitiated, Self::ShutdownInitiated)
            | (Self::Shutdown, Self::Shutdown) => true,
            (Self::Terminated(a), Self::Terminated(b)) => Arc::ptr_eq(a, b),
            _ => false,
        }
    }
}

impl LiveSignalStatus {
    /// Returns `true` if the signal process has stopped (either shut down or terminated).
    pub fn is_stopped(&self) -> bool {
        matches!(self, Self::Shutdown | Self::Terminated(_))
    }
}

impl fmt::Display for LiveSignalStatus {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::NotRunning(status) => write!(f, "Not running ({status})"),
            Self::Running => write!(f, "Running"),
            Self::ShutdownInitiated => write!(f, "Shutdown initiated"),
            Self::Shutdown => write!(f, "Shutdown"),
            Self::Terminated(error) => write!(f, "Terminated: {error}"),
        }
    }
}

impl From<LiveSignalStatusNotRunning> for LiveSignalStatus {
    fn from(value: LiveSignalStatusNotRunning) -> Self {
        Self::NotRunning(value)
    }
}

impl From<SignalProcessRecoverableError> for LiveSignalStatus {
    fn from(value: SignalProcessRecoverableError) -> Self {
        LiveSignalStatusNotRunning::Failed(Arc::new(value)).into()
    }
}

impl From<Arc<SignalProcessFatalError>> for LiveSignalStatus {
    fn from(value: Arc<SignalProcessFatalError>) -> Self {
        Self::Terminated(value)
    }
}

impl From<SignalProcessFatalError> for LiveSignalStatus {
    fn from(value: SignalProcessFatalError) -> Self {
        Arc::new(value).into()
    }
}

/// Update events emitted by the live signal evaluation process.
///
/// These updates are broadcast to subscribers and include status changes and newly generated
/// trading signals.
#[derive(Debug, Clone)]
pub enum LiveSignalUpdate<S: Signal> {
    /// Signal process status has changed.
    Status(LiveSignalStatus),
    /// A new trading signal has been generated.
    Signal(S),
}

impl<S: Signal> From<LiveSignalStatus> for LiveSignalUpdate<S> {
    fn from(value: LiveSignalStatus) -> Self {
        Self::Status(value)
    }
}

pub(crate) type LiveSignalTransmitter<S> = broadcast::Sender<LiveSignalUpdate<S>>;

/// Receiver for subscribing to [`LiveSignalUpdate`]s.
pub type LiveSignalReceiver<S> = broadcast::Receiver<LiveSignalUpdate<S>>;

/// Trait for reading signal evaluation status and subscribing to updates.
///
/// Provides a read-only interface to the signal process state without the ability to control or
/// modify it.
pub trait LiveSignalReader<S: Signal>: Send + Sync + 'static {
    /// Creates a new [`LiveSignalReceiver`] for subscribing to signal updates.
    fn update_receiver(&self) -> LiveSignalReceiver<S>;

    /// Returns the current [`LiveSignalStatus`] as a snapshot.
    fn status_snapshot(&self) -> LiveSignalStatus;
}

#[derive(Debug)]
pub(crate) struct LiveSignalStatusManager<S: Signal> {
    status: Mutex<LiveSignalStatus>,
    update_tx: LiveSignalTransmitter<S>,
}

impl<S: Signal> LiveSignalStatusManager<S> {
    pub fn new(update_tx: LiveSignalTransmitter<S>) -> Arc<Self> {
        let status = Mutex::new(LiveSignalStatusNotRunning::NotInitiated.into());

        Arc::new(Self { status, update_tx })
    }

    fn lock_status(&self) -> MutexGuard<'_, LiveSignalStatus> {
        self.status
            .lock()
            .expect("`LiveSignalStatusManager` mutex can't be poisoned")
    }
    pub fn update(&self, new_status: LiveSignalStatus) {
        let mut status_guard = self.lock_status();

        if *status_guard == new_status {
            return;
        }

        *status_guard = new_status.clone();
        drop(status_guard);

        // Ignore no-receivers errors
        let _ = self.update_tx.send(new_status.into());
    }
}

impl<S: Signal> LiveSignalReader<S> for LiveSignalStatusManager<S> {
    fn update_receiver(&self) -> LiveSignalReceiver<S> {
        self.update_tx.subscribe()
    }

    fn status_snapshot(&self) -> LiveSignalStatus {
        self.lock_status().clone()
    }
}