use std::{collections::HashMap, sync::Arc};
use bonsaidb_core::{api, networking::CURRENT_PROTOCOL_VERSION, schema::ApiName};
#[cfg(not(target_arch = "wasm32"))]
use fabruic::Certificate;
#[cfg(not(target_arch = "wasm32"))]
use tokio::runtime::Handle;
use url::Url;
use crate::{
client::{AnyApiCallback, ApiCallback},
Client, Error,
};
#[must_use]
pub struct Builder {
url: Url,
protocol_version: &'static str,
custom_apis: HashMap<ApiName, Option<Arc<dyn AnyApiCallback>>>,
#[cfg(not(target_arch = "wasm32"))]
certificate: Option<fabruic::Certificate>,
#[cfg(not(target_arch = "wasm32"))]
tokio: Option<Handle>,
}
impl Builder {
pub(crate) fn new(url: Url) -> Self {
Self {
url,
protocol_version: CURRENT_PROTOCOL_VERSION,
custom_apis: HashMap::new(),
#[cfg(not(target_arch = "wasm32"))]
certificate: None,
#[cfg(not(target_arch = "wasm32"))]
tokio: Handle::try_current().ok(),
}
}
#[cfg(not(target_arch = "wasm32"))]
#[allow(clippy::missing_const_for_fn)]
pub fn with_runtime(mut self, handle: Handle) -> Self {
self.tokio = Some(handle);
self
}
pub fn with_api<Api: api::Api>(mut self) -> Self {
self.custom_apis.insert(Api::name(), None);
self
}
pub fn with_api_callback<Api: api::Api>(mut self, callback: ApiCallback<Api>) -> Self {
self.custom_apis
.insert(Api::name(), Some(Arc::new(callback)));
self
}
#[cfg(not(target_arch = "wasm32"))]
#[allow(clippy::missing_const_for_fn)]
pub fn with_certificate(mut self, certificate: Certificate) -> Self {
self.certificate = Some(certificate);
self
}
#[cfg(feature = "test-util")]
#[allow(clippy::missing_const_for_fn)]
pub fn with_protocol_version(mut self, version: &'static str) -> Self {
self.protocol_version = version;
self
}
pub fn finish(self) -> Result<Client, Error> {
Client::new_from_parts(
self.url,
self.protocol_version,
self.custom_apis,
#[cfg(not(target_arch = "wasm32"))]
self.certificate,
#[cfg(not(target_arch = "wasm32"))]
self.tokio,
)
}
}