moq-async 0.1.3

Media over QUIC - Async helpers and utilities
Documentation
use std::fmt;
use std::ops::{Deref, DerefMut};

// It's a cosmetic wrapper around Arc<Mutex<T>> on native platforms.
// On WASM, it uses Rc<RefCell<T>> instead.
pub struct Lock<T> {
	#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
	inner: std::sync::Arc<std::sync::Mutex<T>>,

	#[cfg(all(target_arch = "wasm32", not(target_os = "wasi")))]
	inner: std::rc::Rc<std::cell::RefCell<T>>,
}

#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
impl<T> Lock<T> {
	pub fn new(value: T) -> Self {
		Self {
			inner: std::sync::Arc::new(std::sync::Mutex::new(value)),
		}
	}
}

impl<T> Lock<T> {
	pub fn lock(&self) -> LockGuard<T> {
		LockGuard::new(&self.inner)
	}

	pub fn downgrade(&self) -> LockWeak<T> {
		LockWeak::new(&self.inner)
	}
}

#[cfg(all(target_arch = "wasm32", not(target_os = "wasi")))]
impl<T> Lock<T> {
	pub fn new(value: T) -> Self {
		Self {
			inner: std::rc::Rc::new(std::cell::RefCell::new(value)),
		}
	}
}

impl<T: Default> Default for Lock<T> {
	fn default() -> Self {
		Self::new(T::default())
	}
}

impl<T> Clone for Lock<T> {
	fn clone(&self) -> Self {
		Self {
			inner: self.inner.clone(),
		}
	}
}

pub struct LockGuard<'a, T> {
	#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
	inner: std::sync::MutexGuard<'a, T>,

	#[cfg(all(target_arch = "wasm32", not(target_os = "wasi")))]
	inner: std::cell::RefMut<'a, T>,
}

#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
impl<'a, T> LockGuard<'a, T> {
	fn new(inner: &'a std::sync::Arc<std::sync::Mutex<T>>) -> Self {
		Self {
			inner: inner.lock().unwrap(),
		}
	}
}

#[cfg(all(target_arch = "wasm32", not(target_os = "wasi")))]
impl<'a, T> LockGuard<'a, T> {
	fn new(inner: &'a std::rc::Rc<std::cell::RefCell<T>>) -> Self {
		Self {
			inner: inner.borrow_mut(),
		}
	}
}

impl<T> Deref for LockGuard<'_, T> {
	type Target = T;

	fn deref(&self) -> &Self::Target {
		&self.inner
	}
}

impl<T> DerefMut for LockGuard<'_, T> {
	fn deref_mut(&mut self) -> &mut Self::Target {
		&mut self.inner
	}
}

impl<T: fmt::Debug> fmt::Debug for LockGuard<'_, T> {
	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
		self.inner.fmt(f)
	}
}

pub struct LockWeak<T> {
	#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
	inner: std::sync::Weak<std::sync::Mutex<T>>,

	#[cfg(all(target_arch = "wasm32", not(target_os = "wasi")))]
	inner: std::rc::Weak<std::cell::RefCell<T>>,
}

#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
impl<T> LockWeak<T> {
	fn new(inner: &std::sync::Arc<std::sync::Mutex<T>>) -> Self {
		Self {
			inner: std::sync::Arc::downgrade(inner),
		}
	}
}

#[cfg(all(target_arch = "wasm32", not(target_os = "wasi")))]
impl<T> LockWeak<T> {
	fn new(inner: &std::rc::Rc<std::cell::RefCell<T>>) -> Self {
		Self {
			inner: std::rc::Rc::downgrade(inner),
		}
	}
}

impl<T> LockWeak<T> {
	pub fn upgrade(&self) -> Option<Lock<T>> {
		Some(Lock {
			inner: self.inner.upgrade()?,
		})
	}
}

impl<T> Clone for LockWeak<T> {
	fn clone(&self) -> Self {
		Self {
			inner: self.inner.clone(),
		}
	}
}