mod impls;
use std::fmt;
use std::sync::Arc;
use crate::network::{Betanet, Mainnet, Sandbox, Testnet};
use crate::{Network, Result};
pub struct Worker<T: ?Sized> {
pub(crate) workspace: Arc<T>,
}
impl<T> Worker<T>
where
T: Network,
{
pub(crate) fn new(network: T) -> Self {
Self {
workspace: Arc::new(network),
}
}
}
impl<T: Network + 'static> Worker<T> {
pub(crate) fn coerce(self) -> Worker<dyn Network> {
Worker {
workspace: self.workspace,
}
}
}
impl<T: fmt::Debug> fmt::Debug for Worker<T> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("Worker")
.field("workspace", &self.workspace)
.finish()
}
}
pub async fn sandbox() -> Result<Worker<Sandbox>> {
Ok(Worker::new(Sandbox::new().await?))
}
pub async fn testnet() -> Result<Worker<Testnet>> {
Ok(Worker::new(Testnet::new().await?))
}
pub async fn testnet_archival() -> Result<Worker<Testnet>> {
Ok(Worker::new(Testnet::archival().await?))
}
pub async fn mainnet() -> Result<Worker<Mainnet>> {
Ok(Worker::new(Mainnet::new().await?))
}
pub async fn mainnet_archival() -> Result<Worker<Mainnet>> {
Ok(Worker::new(Mainnet::archival().await?))
}
pub async fn betanet() -> Result<Worker<Betanet>> {
Ok(Worker::new(Betanet::new().await?))
}
pub async fn with_sandbox<F, T>(task: F) -> Result<T::Output>
where
F: Fn(Worker<Sandbox>) -> T,
T: core::future::Future,
{
Ok(task(sandbox().await?).await)
}
pub async fn with_testnet<F, T>(task: F) -> Result<T::Output>
where
F: Fn(Worker<Testnet>) -> T,
T: core::future::Future,
{
Ok(task(testnet().await?).await)
}
pub async fn with_testnet_archival<F, T>(task: F) -> Result<T::Output>
where
F: Fn(Worker<Testnet>) -> T,
T: core::future::Future,
{
Ok(task(testnet_archival().await?).await)
}
pub async fn with_mainnet<F, T>(task: F) -> Result<T::Output>
where
F: Fn(Worker<Mainnet>) -> T,
T: core::future::Future,
{
Ok(task(mainnet().await?).await)
}
pub async fn with_mainnet_archival<F, T>(task: F) -> Result<T::Output>
where
F: Fn(Worker<Mainnet>) -> T,
T: core::future::Future,
{
Ok(task(mainnet_archival().await?).await)
}
pub async fn with_betanet<F, T>(task: F) -> Result<T::Output>
where
F: Fn(Worker<Betanet>) -> T,
T: core::future::Future,
{
Ok(task(betanet().await?).await)
}