soil-network 0.2.0

// This file is part of Soil.

// Copyright (C) Soil contributors.
// Copyright (C) Parity Technologies (UK) Ltd.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

use super::{config::Config, error::Error, request::Request};
use futures::{
	channel::{mpsc, oneshot},
	SinkExt,
};
use std::future::Future;
use subsoil::core::Bytes;

/// The other end of an [`Api`]. This should be passed to [`run`](super::run::run).
pub struct ApiBackend {
	pub(super) request_receiver: mpsc::Receiver<Request>,
}

/// Interface to the mixnet service.
#[derive(Clone)]
pub struct Api {
	request_sender: mpsc::Sender<Request>,
}

impl Api {
	/// Create a new `Api`. The [`ApiBackend`] should be passed to [`run`](super::run::run).
	pub fn new(config: &Config) -> (Self, ApiBackend) {
		let (request_sender, request_receiver) = mpsc::channel(config.substrate.request_buffer);
		(Self { request_sender }, ApiBackend { request_receiver })
	}

	/// Submit an extrinsic via the mixnet.
	///
	/// Returns a [`Future`] which returns another `Future`.
	///
	/// The first `Future` resolves as soon as there is space in the mixnet service queue. The
	/// second `Future` resolves once a reply is received over the mixnet (or sooner if there is an
	/// error).
	///
	/// The first `Future` references `self`, but the second does not. This makes it possible to
	/// submit concurrent mixnet requests using a single `Api` instance.
	pub async fn submit_extrinsic(
		&mut self,
		extrinsic: Bytes,
	) -> impl Future<Output = Result<(), Error>> {
		let (reply_sender, reply_receiver) = oneshot::channel();
		let res = self
			.request_sender
			.feed(Request::SubmitExtrinsic { extrinsic, reply_sender })
			.await;
		async move {
			res.map_err(|_| Error::ServiceUnavailable)?;
			reply_receiver.await.map_err(|_| Error::ServiceUnavailable)?
		}
	}
}