AppFlowy/frontend/rust-lib/flowy-test/src/lib.rs

pub mod event_builder;
pub mod helper;

use crate::helper::*;

use flowy_core::{FlowySDK, FlowySDKConfig};
use flowy_document::entities::DocumentVersionPB;
use flowy_net::get_client_server_configuration;
use flowy_user::entities::UserProfilePB;
use nanoid::nanoid;

pub mod prelude {
    pub use crate::{event_builder::*, helper::*, *};
    pub use lib_dispatch::prelude::*;
}

#[derive(Clone)]
pub struct FlowySDKTest {
    pub inner: FlowySDK,
}

impl std::ops::Deref for FlowySDKTest {
    type Target = FlowySDK;

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

impl std::default::Default for FlowySDKTest {
    fn default() -> Self {
        Self::new(DocumentVersionPB::V0)
    }
}

impl FlowySDKTest {
    pub fn new(document_version: DocumentVersionPB) -> Self {
        let server_config = get_client_server_configuration().unwrap();
        let config = FlowySDKConfig::new(&root_dir(), nanoid!(6), server_config)
            .with_document_version(document_version)
            .log_filter("info");
        let sdk = std::thread::spawn(|| FlowySDK::new(config)).join().unwrap();
        std::mem::forget(sdk.dispatcher());
        Self { inner: sdk }
    }

    pub async fn sign_up(&self) -> SignUpContext {
        let context = async_sign_up(self.inner.dispatcher()).await;
        context
    }

    pub async fn init_user(&self) -> UserProfilePB {
        let context = async_sign_up(self.inner.dispatcher()).await;
        init_user_setting(self.inner.dispatcher()).await;
        context.user_profile
    }

    pub fn document_version(&self) -> DocumentVersionPB {
        self.inner.config.document.version.clone()
    }
}