AppFlowy/rust-lib/flowy-workspace/src/services/workspace_controller.rs

use crate::{entities::workspace::*, errors::*, module::WorkspaceUser, sql_tables::workspace::*};
use flowy_database::{prelude::*, schema::workspace_table};
use futures_core::future::BoxFuture;
use std::sync::Arc;

pub struct WorkspaceController {
    pub user: Arc<dyn WorkspaceUser>,
}

impl WorkspaceController {
    pub fn new(user: Arc<dyn WorkspaceUser>) -> Self { Self { user } }

    pub async fn save_workspace(
        &self,
        params: CreateWorkspaceParams,
    ) -> Result<WorkspaceDetail, WorkspaceError> {
        let workspace = Workspace::new(params);
        let conn = self.user.db_connection()?;
        let detail: WorkspaceDetail = workspace.clone().into();

        let _ = diesel::insert_into(workspace_table::table)
            .values(workspace)
            .execute(&*conn)?;

        self.user.set_current_workspace(&detail.id);

        Ok(detail)
    }

    pub fn update_workspace(&self, params: UpdateWorkspaceParams) -> Result<(), WorkspaceError> {
        let changeset = WorkspaceChangeset::new(params);
        let conn = self.user.db_connection()?;
        diesel_update_table!(workspace_table, changeset, conn);

        Ok(())
    }
}

pub async fn save_workspace(
    controller: Arc<WorkspaceController>,
    params: CreateWorkspaceParams,
) -> Result<WorkspaceDetail, WorkspaceError> {
    let workspace = Workspace::new(params);
    let detail: WorkspaceDetail = workspace.clone().into();

    let _ = diesel::insert_into(workspace_table::table)
        .values(workspace)
        .execute(&*(controller.user.db_connection()?))?;

    // set_current_workspace(controller.clone(), &detail.id).await;

    Ok(detail)
}