AppFlowy/frontend/rust-lib/flowy-folder/src/event_map.rs

use crate::{
    entities::{
        app::{AppIdPB, CreateAppParams, UpdateAppParams},
        trash::RepeatedTrashIdPB,
        view::{CreateViewParams, RepeatedViewIdPB, UpdateViewParams, ViewIdPB},
        workspace::{CreateWorkspaceParams, UpdateWorkspaceParams, WorkspaceIdPB},
    },
    errors::FlowyError,
    manager::FolderManager,
    services::{app::event_handler::*, trash::event_handler::*, view::event_handler::*, workspace::event_handler::*},
};
use flowy_derive::{Flowy_Event, ProtoBuf_Enum};
use flowy_sqlite::{ConnectionPool, DBConnection};
use folder_model::{AppRevision, TrashRevision, ViewRevision, WorkspaceRevision};
use lib_dispatch::prelude::*;
use lib_infra::future::FutureResult;
use std::sync::Arc;
use strum_macros::Display;

pub trait WorkspaceDeps: WorkspaceUser + WorkspaceDatabase {}

pub trait WorkspaceUser: Send + Sync {
    fn user_id(&self) -> Result<String, FlowyError>;
    fn token(&self) -> Result<String, FlowyError>;
}

pub trait WorkspaceDatabase: Send + Sync {
    fn db_pool(&self) -> Result<Arc<ConnectionPool>, FlowyError>;

    fn db_connection(&self) -> Result<DBConnection, FlowyError> {
        let pool = self.db_pool()?;
        let conn = pool.get().map_err(|e| FlowyError::internal().context(e))?;
        Ok(conn)
    }
}

pub fn init(folder: Arc<FolderManager>) -> AFPlugin {
    let mut plugin = AFPlugin::new()
        .name("Flowy-Workspace")
        .state(folder.workspace_controller.clone())
        .state(folder.app_controller.clone())
        .state(folder.view_controller.clone())
        .state(folder.trash_controller.clone())
        .state(folder.clone());

    // Workspace
    plugin = plugin
        .event(FolderEvent::CreateWorkspace, create_workspace_handler)
        .event(FolderEvent::ReadCurrentWorkspace, read_cur_workspace_handler)
        .event(FolderEvent::ReadWorkspaces, read_workspaces_handler)
        .event(FolderEvent::OpenWorkspace, open_workspace_handler)
        .event(FolderEvent::ReadWorkspaceApps, read_workspace_apps_handler);

    // App
    plugin = plugin
        .event(FolderEvent::CreateApp, create_app_handler)
        .event(FolderEvent::ReadApp, read_app_handler)
        .event(FolderEvent::UpdateApp, update_app_handler)
        .event(FolderEvent::DeleteApp, delete_app_handler);

    // View
    plugin = plugin
        .event(FolderEvent::CreateView, create_view_handler)
        .event(FolderEvent::ReadView, read_view_handler)
        .event(FolderEvent::UpdateView, update_view_handler)
        .event(FolderEvent::DeleteView, delete_view_handler)
        .event(FolderEvent::DuplicateView, duplicate_view_handler)
        .event(FolderEvent::SetLatestView, set_latest_view_handler)
        .event(FolderEvent::CloseView, close_view_handler)
        .event(FolderEvent::MoveItem, move_item_handler);

    // Trash
    plugin = plugin
        .event(FolderEvent::ReadTrash, read_trash_handler)
        .event(FolderEvent::PutbackTrash, putback_trash_handler)
        .event(FolderEvent::DeleteTrash, delete_trash_handler)
        .event(FolderEvent::RestoreAllTrash, restore_all_trash_handler)
        .event(FolderEvent::DeleteAllTrash, delete_all_trash_handler);

    plugin
}

#[derive(Clone, Copy, PartialEq, Eq, Debug, Display, Hash, ProtoBuf_Enum, Flowy_Event)]
#[event_err = "FlowyError"]
pub enum FolderEvent {
    /// Create a new workspace
    #[event(input = "CreateWorkspacePayloadPB", output = "WorkspacePB")]
    CreateWorkspace = 0,

    /// Read the current opening workspace
    #[event(output = "WorkspaceSettingPB")]
    ReadCurrentWorkspace = 1,

    /// Open the workspace and mark it as the current workspace
    #[event(input = "WorkspaceIdPB", output = "RepeatedWorkspacePB")]
    ReadWorkspaces = 2,

    /// Delete the workspace
    #[event(input = "WorkspaceIdPB")]
    DeleteWorkspace = 3,

    /// Open the workspace and mark it as the current workspace
    #[event(input = "WorkspaceIdPB", output = "WorkspacePB")]
    OpenWorkspace = 4,

    /// Return a list of apps that belong to this workspace
    #[event(input = "WorkspaceIdPB", output = "RepeatedAppPB")]
    ReadWorkspaceApps = 5,

    /// Create a new app
    #[event(input = "CreateAppPayloadPB", output = "AppPB")]
    CreateApp = 101,

    /// Delete the app
    #[event(input = "AppIdPB")]
    DeleteApp = 102,

    /// Read the app
    #[event(input = "AppIdPB", output = "AppPB")]
    ReadApp = 103,

    /// Update the app's properties including the name,description, etc.
    #[event(input = "UpdateAppPayloadPB")]
    UpdateApp = 104,

    /// Create a new view in the corresponding app
    #[event(input = "CreateViewPayloadPB", output = "ViewPB")]
    CreateView = 201,

    /// Return the view info
    #[event(input = "ViewIdPB", output = "ViewPB")]
    ReadView = 202,

    /// Update the view's properties including the name,description, etc.
    #[event(input = "UpdateViewPayloadPB", output = "ViewPB")]
    UpdateView = 203,

    /// Move the view to the trash folder
    #[event(input = "RepeatedViewIdPB")]
    DeleteView = 204,

    /// Duplicate the view
    #[event(input = "ViewPB")]
    DuplicateView = 205,

    /// Close and release the resources that are used by this view.
    /// It should get called when the 'View' page get destroy
    #[event(input = "ViewIdPB")]
    CloseView = 206,

    #[event()]
    CopyLink = 220,

    /// Set the current visiting view
    #[event(input = "ViewIdPB")]
    SetLatestView = 221,

    /// Move the view or app to another place
    #[event(input = "MoveFolderItemPayloadPB")]
    MoveItem = 230,

    /// Read the trash that was deleted by the user
    #[event(output = "RepeatedTrashPB")]
    ReadTrash = 300,

    /// Put back the trash to the origin folder
    #[event(input = "TrashIdPB")]
    PutbackTrash = 301,

    /// Delete the trash from the disk
    #[event(input = "RepeatedTrashIdPB")]
    DeleteTrash = 302,

    /// Put back all the trash to its original folder
    #[event()]
    RestoreAllTrash = 303,

    /// Delete all the trash from the disk
    #[event()]
    DeleteAllTrash = 304,
}

pub trait FolderCouldServiceV1: Send + Sync {
    fn init(&self);

    // Workspace
    fn create_workspace(
        &self,
        token: &str,
        params: CreateWorkspaceParams,
    ) -> FutureResult<WorkspaceRevision, FlowyError>;

    fn read_workspace(&self, token: &str, params: WorkspaceIdPB) -> FutureResult<Vec<WorkspaceRevision>, FlowyError>;

    fn update_workspace(&self, token: &str, params: UpdateWorkspaceParams) -> FutureResult<(), FlowyError>;

    fn delete_workspace(&self, token: &str, params: WorkspaceIdPB) -> FutureResult<(), FlowyError>;

    // View
    fn create_view(&self, token: &str, params: CreateViewParams) -> FutureResult<ViewRevision, FlowyError>;

    fn read_view(&self, token: &str, params: ViewIdPB) -> FutureResult<Option<ViewRevision>, FlowyError>;

    fn delete_view(&self, token: &str, params: RepeatedViewIdPB) -> FutureResult<(), FlowyError>;

    fn update_view(&self, token: &str, params: UpdateViewParams) -> FutureResult<(), FlowyError>;

    // App
    fn create_app(&self, token: &str, params: CreateAppParams) -> FutureResult<AppRevision, FlowyError>;

    fn read_app(&self, token: &str, params: AppIdPB) -> FutureResult<Option<AppRevision>, FlowyError>;

    fn update_app(&self, token: &str, params: UpdateAppParams) -> FutureResult<(), FlowyError>;

    fn delete_app(&self, token: &str, params: AppIdPB) -> FutureResult<(), FlowyError>;

    // Trash
    fn create_trash(&self, token: &str, params: RepeatedTrashIdPB) -> FutureResult<(), FlowyError>;

    fn delete_trash(&self, token: &str, params: RepeatedTrashIdPB) -> FutureResult<(), FlowyError>;

    fn read_trash(&self, token: &str) -> FutureResult<Vec<TrashRevision>, FlowyError>;
}