libp2p-episub 0.1.13

use crate::{codec::EpisubCodec, error::EpisubHandlerError};
use asynchronous_codec::Framed;
use futures::{future, AsyncRead, AsyncWrite};
use libp2p_core::{InboundUpgrade, OutboundUpgrade, UpgradeInfo};
use std::{future::Future, iter, pin::Pin};
use unsigned_varint::codec;

#[derive(Debug, Clone)]
pub struct EpisubProtocol {
  max_transmit_size: usize,
}

impl EpisubProtocol {
  pub fn new(max_transmit_size: usize) -> Self {
    Self { max_transmit_size }
  }
}

impl UpgradeInfo for EpisubProtocol {
  type Info = &'static [u8];
  type InfoIter = iter::Once<Self::Info>;

  fn protocol_info(&self) -> Self::InfoIter {
    iter::once(b"/beta/episub/1.0.0")
  }
}

impl<TSocket> InboundUpgrade<TSocket> for EpisubProtocol
where
  TSocket: AsyncRead + AsyncWrite + Send + Unpin + 'static,
{
  type Output = Framed<TSocket, EpisubCodec>;
  type Error = EpisubHandlerError;

  #[allow(clippy::type_complexity)] // oh well
  type Future =
    Pin<Box<dyn Future<Output = Result<Self::Output, Self::Error>> + Send>>;

  fn upgrade_inbound(self, socket: TSocket, _: Self::Info) -> Self::Future {
    let mut length_codec = codec::UviBytes::default();
    length_codec.set_max_len(self.max_transmit_size);
    Box::pin(future::ok(Framed::new(
      socket,
      EpisubCodec::new(length_codec),
    )))
  }
}

impl<TSocket> OutboundUpgrade<TSocket> for EpisubProtocol
where
  TSocket: AsyncRead + AsyncWrite + Send + Unpin + 'static,
{
  type Output = Framed<TSocket, EpisubCodec>;
  type Error = EpisubHandlerError;
  
  #[allow(clippy::type_complexity)] // oh well
  type Future =
    Pin<Box<dyn Future<Output = Result<Self::Output, Self::Error>> + Send>>;

  fn upgrade_outbound(self, socket: TSocket, _: Self::Info) -> Self::Future {
    let mut length_codec = codec::UviBytes::default();
    length_codec.set_max_len(self.max_transmit_size);
    Box::pin(future::ok(Framed::new(
      socket,
      EpisubCodec::new(length_codec),
    )))
  }
}