celestia-client 0.4.0

use std::pin::Pin;
use std::sync::Arc;

use async_stream::try_stream;
use celestia_rpc::FraudClient;
use futures_util::{Stream, StreamExt};

use crate::Result;
use crate::api::fraud::{Proof, ProofType};
use crate::client::ClientInner;

/// Fraud API for quering bridge nodes.
pub struct FraudApi {
    inner: Arc<ClientInner>,
}

impl FraudApi {
    pub(crate) fn new(inner: Arc<ClientInner>) -> FraudApi {
        FraudApi { inner }
    }

    /// Fetches fraud proofs from node by their type.
    pub async fn get(&self, proof_type: ProofType) -> Result<Vec<Proof>> {
        Ok(self.inner.rpc.fraud_get(proof_type).await?)
    }

    /// Subscribe to fraud proof by its type.
    pub fn subscribe(
        &self,
        proof_type: ProofType,
    ) -> Pin<Box<dyn Stream<Item = Result<Proof>> + Send + 'static>> {
        let inner = self.inner.clone();

        // we need to re-stream it to map error and satisfy 'static
        try_stream! {
            let mut subscription = inner.rpc.fraud_subscribe(proof_type);
            while let Some(item) = subscription.next().await {
                // TODO: Should we validate proof?
                yield item?;
            }
        }
        .boxed()
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    use crate::test_utils::ensure_serializable_deserializable;

    #[allow(dead_code)]
    #[allow(unused_variables)]
    #[allow(unreachable_code)]
    #[allow(clippy::diverging_sub_expression)]
    async fn enforce_serde_bounds() {
        // intentionally no-run, compile only test
        let api = FraudApi::new(unimplemented!());

        let proof_type = ensure_serializable_deserializable(unimplemented!());
        ensure_serializable_deserializable(api.get(proof_type).await.unwrap());

        let proof_type = ensure_serializable_deserializable(unimplemented!());
        ensure_serializable_deserializable(
            api.subscribe(proof_type).next().await.unwrap().unwrap(),
        );
    }
}