use crate::{managers::InFlight, RawSubscription};
use linera_alloy_primitives::U256;
use std::fmt;
use tokio::sync::oneshot;

/// Instructions for the pubsub service.
pub(crate) enum PubSubInstruction {
    /// Send a request.
    Request(InFlight),
    /// Get the subscription ID for a local ID.
    GetSub(U256, oneshot::Sender<RawSubscription>),
    /// Unsubscribe from a subscription.
    Unsubscribe(U256),
}

impl fmt::Debug for PubSubInstruction {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::Request(arg0) => f.debug_tuple("Request").field(arg0).finish(),
            Self::GetSub(arg0, _) => f.debug_tuple("GetSub").field(arg0).finish(),
            Self::Unsubscribe(arg0) => f.debug_tuple("Unsubscribe").field(arg0).finish(),
        }
    }
}