// A helper class to guard sending control messages behind a Mutex.

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

use tokio::sync::Mutex;
use webtransport_quinn::{RecvStream, SendStream};

use crate::{message::Message, Error};

#[derive(Debug, Clone)]
pub(crate) struct Control {
	send: Arc<Mutex<SendStream>>,
	recv: Arc<Mutex<RecvStream>>,
}

impl Control {
	pub fn new(send: SendStream, recv: RecvStream) -> Self {
		Self {
			send: Arc::new(Mutex::new(send)),
			recv: Arc::new(Mutex::new(recv)),
		}
	}

	pub async fn send<T: Into<Message> + fmt::Debug>(&self, msg: T) -> Result<(), Error> {
		let mut stream = self.send.lock().await;
		log::info!("sending message: {:?}", msg);
		msg.into().encode(&mut *stream).await.map_err(|_e| Error::Write)
	}

	// It's likely a mistake to call this from two different tasks, but it's easier to just support it.
	pub async fn recv(&self) -> Result<Message, Error> {
		let mut stream = self.recv.lock().await;
		Message::decode(&mut *stream).await.map_err(|_e| Error::Read)
	}
}