AFFiNE-Mirror/apps/desktop/src-tauri/src/commands/user.rs

use cloud_database::{CreateUser, User};
use ipc_types::{document::CreateDocumentParameter, user::GetUserParameters};

use crate::state::AppState;

use super::document::create_doc;

#[tauri::command]
/// create new user and a private workspace
pub async fn create_user<'s>(
  state: tauri::State<'s, AppState>,
  parameters: CreateUser,
) -> Result<User, String> {
  let new_user_result = &state
    .0
    .lock()
    .await
    .metadata_db
    .create_user(parameters.clone())
    .await;
  match new_user_result {
    Ok(new_user_option) => match new_user_option {
      Some((new_user, new_workspace)) => {
        // a new private workspace is created, we have to create a yDoc for it
        create_doc(
          state,
          CreateDocumentParameter {
            workspace_id: new_workspace.id.clone(),
            workspace_name: parameters.name.clone(),
          },
        )
        .await
        .ok();
        Ok(new_user.clone())
      }
      None => Err("User creation failed".to_string()),
    },
    Err(error_message) => Err(error_message.to_string()),
  }
}

#[tauri::command]
/// get the only one user in local sqlite
pub async fn get_user<'s>(
  state: tauri::State<'s, AppState>,
  parameters: GetUserParameters,
) -> Result<User, String> {
  let db = &state.0.lock().await.metadata_db;
  match db.get_user_by_email(&parameters.email).await.ok().unwrap() {
    Some(user) => Ok(user),
    None => Err("User not found".to_string()),
  }
}