AppFlowy/rust-lib/flowy-workspace/src/module.rs

use flowy_dispatch::prelude::*;

use crate::{
    errors::WorkspaceError,
    event::WorkspaceEvent,
    services::{AppController, WorkspaceController},
};
use flowy_database::DBConnection;

use crate::{
    handlers::*,
    services::{server::construct_workspace_server, ViewController},
};
use std::sync::Arc;

pub trait WorkspaceDeps: WorkspaceUser + WorkspaceDatabase {}

pub trait WorkspaceUser: Send + Sync {
    fn user_id(&self) -> Result<String, WorkspaceError>;
    fn token(&self) -> Result<String, WorkspaceError>;
    // fn set_cur_workspace_id(&self, id: &str) -> DispatchFuture<Result<(),
    // WorkspaceError>>; fn get_cur_workspace(&self) ->
    // DispatchFuture<Result<CurrentWorkspace, WorkspaceError>>;
}

pub trait WorkspaceDatabase: Send + Sync {
    fn db_connection(&self) -> Result<DBConnection, WorkspaceError>;
}

pub fn create(user: Arc<dyn WorkspaceUser>, database: Arc<dyn WorkspaceDatabase>) -> Module {
    let server = construct_workspace_server();
    let view_controller = Arc::new(ViewController::new(database.clone(), server.clone()));

    let app_controller = Arc::new(AppController::new(
        user.clone(),
        database.clone(),
        view_controller.clone(),
        server.clone(),
    ));

    let workspace_controller = Arc::new(WorkspaceController::new(
        user.clone(),
        database.clone(),
        app_controller.clone(),
        server.clone(),
    ));

    let mut module = Module::new()
        .name("Flowy-Workspace")
        .data(workspace_controller)
        .data(app_controller)
        .data(view_controller);

    module = module
        .event(WorkspaceEvent::CreateWorkspace, create_workspace)
        .event(WorkspaceEvent::ReadCurWorkspace, read_cur_workspace)
        .event(WorkspaceEvent::ReadWorkspaces, read_workspaces)
        .event(WorkspaceEvent::OpenWorkspace, open_workspace);

    module = module
        .event(WorkspaceEvent::CreateApp, create_app)
        .event(WorkspaceEvent::ReadApp, read_app)
        .event(WorkspaceEvent::UpdateApp, update_app)
        .event(WorkspaceEvent::DeleteApp, delete_app);

    module = module
        .event(WorkspaceEvent::CreateView, create_view)
        .event(WorkspaceEvent::ReadView, read_view)
        .event(WorkspaceEvent::UpdateView, update_view)
        .event(WorkspaceEvent::DeleteView, delete_view);

    module
}