veilid/veilid-core/src/veilid_core.rs

use crate::attachment_manager::*;
use crate::dht::crypto::Crypto;
use crate::intf::*;
use crate::veilid_api::*;
use crate::veilid_config::*;
use crate::xx::*;

cfg_if! {
    if #[cfg(target_arch = "wasm32")] {
        pub type UpdateCallback = Arc<dyn Fn(VeilidUpdate) -> SystemPinBoxFuture<()>>;
    } else {
        pub type UpdateCallback = Arc<dyn Fn(VeilidUpdate) -> SystemPinBoxFuture<()> + Send + Sync>;
    }
}

pub struct VeilidCoreSetup {
    pub update_callback: UpdateCallback,
    pub config_callback: ConfigCallback,
}

struct VeilidCoreInner {
    config: Option<VeilidConfig>,
    protected_store: Option<ProtectedStore>,
    table_store: Option<TableStore>,
    crypto: Option<Crypto>,
    attachment_manager: Option<AttachmentManager>,
    api: VeilidAPIWeak,
}

#[derive(Clone)]
pub struct VeilidCore {
    inner: Arc<Mutex<VeilidCoreInner>>,
}

impl Default for VeilidCore {
    fn default() -> Self {
        Self::new()
    }
}

impl VeilidCore {
    fn new_inner() -> VeilidCoreInner {
        VeilidCoreInner {
            config: None,
            table_store: None,
            protected_store: None,
            crypto: None,
            attachment_manager: None,
            api: VeilidAPIWeak::default(),
        }
    }
    pub fn new() -> Self {
        Self {
            inner: Arc::new(Mutex::new(Self::new_inner())),
        }
    }

    pub(crate) fn config(&self) -> VeilidConfig {
        self.inner.lock().config.as_ref().unwrap().clone()
    }

    pub(crate) fn table_store(&self) -> TableStore {
        self.inner.lock().table_store.as_ref().unwrap().clone()
    }

    pub(crate) fn protected_store(&self) -> ProtectedStore {
        self.inner.lock().protected_store.as_ref().unwrap().clone()
    }

    pub(crate) fn crypto(&self) -> Crypto {
        self.inner.lock().crypto.as_ref().unwrap().clone()
    }

    pub(crate) fn attachment_manager(&self) -> AttachmentManager {
        self.inner
            .lock()
            .attachment_manager
            .as_ref()
            .unwrap()
            .clone()
    }

    // internal startup
    async fn internal_startup(
        &self,
        inner: &mut VeilidCoreInner,
        setup: VeilidCoreSetup,
    ) -> Result<VeilidAPI, String> {
        trace!("VeilidCore::internal_startup starting");

        cfg_if! {
            if #[cfg(target_os = "android")] {
                if utils::android::ANDROID_GLOBALS.lock().is_none() {
                    error!("Android globals are not set up");
                    return Err("Android globals are not set up".to_owned());
                }
            }
        }

        // Set up config
        trace!("VeilidCore::internal_startup init config");
        let mut config = VeilidConfig::new();
        config.init(setup.config_callback).await?;
        inner.config = Some(config.clone());

        // Set up protected store
        trace!("VeilidCore::internal_startup init protected store");
        let protected_store = ProtectedStore::new(config.clone());
        protected_store.init().await?;
        inner.protected_store = Some(protected_store.clone());

        // Init node id from config now that protected store is set up
        config.init_node_id(protected_store).await?;

        // Set up tablestore
        trace!("VeilidCore::internal_startup init table store");
        let table_store = TableStore::new(config.clone());
        table_store.init().await?;
        inner.table_store = Some(table_store.clone());

        // Set up crypto
        trace!("VeilidCore::internal_startup init crypto");
        let crypto = Crypto::new(config.clone(), table_store.clone());
        crypto.init().await?;
        inner.crypto = Some(crypto.clone());

        // Set up block store
        // trace!("VeilidCore::internal_startup init block store");
        // let block_store = BlockStore::new(config.clone());
        // block_store.init().await?;
        // inner.block_store = Some(block_store.clone();)

        // Set up attachment manager
        trace!("VeilidCore::internal_startup init attachment manager");
        let cb = setup.update_callback;
        let attachment_manager =
            AttachmentManager::new(config.clone(), table_store.clone(), crypto.clone());
        attachment_manager
            .init(Arc::new(
                move |_old_state: AttachmentState, new_state: AttachmentState| {
                    cb(VeilidUpdate::Attachment(new_state))
                },
            ))
            .await?;
        inner.attachment_manager = Some(attachment_manager.clone());

        // Set up the API
        trace!("VeilidCore::internal_startup init API");
        let this = self.clone();
        let veilid_api = VeilidAPI::new(this);
        inner.api = veilid_api.weak();

        trace!("VeilidCore::internal_startup complete");

        Ok(veilid_api)
    }

    // called once at the beginning to start the node
    pub async fn startup(&self, setup: VeilidCoreSetup) -> Result<VeilidAPI, String> {
        // See if we have an API started up already
        let mut inner = self.inner.lock();
        if inner.api.upgrade().is_some() {
            // If so, return an error because we shouldn't try to do this more than once
            return Err("Veilid API is started".to_owned());
        }

        // Ensure we never end up partially initialized
        match self.internal_startup(&mut *inner, setup).await {
            Ok(v) => Ok(v),
            Err(e) => {
                Self::internal_shutdown(&mut *inner).await;
                Err(e)
            }
        }
    }

    async fn internal_shutdown(inner: &mut VeilidCoreInner) {
        trace!("VeilidCore::internal_shutdown starting");

        // Detach the API object
        inner.api = VeilidAPIWeak::default();

        // Shut down up attachment manager
        if let Some(attachment_manager) = &inner.attachment_manager {
            attachment_manager.terminate().await;
            inner.attachment_manager = None;
        }

        // Shut down crypto
        if let Some(crypto) = &inner.crypto {
            crypto.terminate().await;
            inner.crypto = None;
        }

        // Shut down table store
        if let Some(table_store) = &inner.table_store {
            table_store.terminate().await;
            inner.table_store = None;
        }

        // Shut down protected store
        if let Some(protected_store) = &inner.protected_store {
            protected_store.terminate().await;
            inner.protected_store = None;
        }

        // Shut down config
        if let Some(config) = &inner.config {
            config.terminate().await;
            inner.config = None;
        }

        trace!("VeilidCore::shutdown complete");
    }

    // stop the node gracefully because the veilid api was dropped
    pub(crate) async fn shutdown(self) {
        let mut inner = self.inner.lock();
        Self::internal_shutdown(&mut *inner).await;
    }

    //
}