AppFlowy/frontend/rust-lib/flowy-database2/src/services/field/field_operation.rs

use std::sync::Arc;

use flowy_error::{FlowyError, FlowyResult};

use crate::entities::FieldType;
use crate::services::database::DatabaseEditor;
use crate::services::field::{MultiSelectTypeOption, SingleSelectTypeOption, TypeOption};

pub async fn edit_field_type_option<T: TypeOption>(
  field_id: &str,
  editor: Arc<DatabaseEditor>,
  action: impl FnOnce(&mut T),
) -> FlowyResult<()> {
  let field = editor
    .get_field(field_id)
    .await
    .ok_or_else(FlowyError::field_record_not_found)?;
  let field_type = FieldType::from(field.field_type);
  let get_type_option = field.get_type_option::<T>(field_type);

  if let Some(mut type_option) = get_type_option {
    if let Some(old_field) = editor.get_field(field_id).await {
      action(&mut type_option);
      let type_option_data = type_option.into();
      editor
        .update_field_type_option(field_id, type_option_data, old_field)
        .await?;
    }
  }

  Ok(())
}

pub async fn edit_single_select_type_option(
  field_id: &str,
  editor: Arc<DatabaseEditor>,
  action: impl FnOnce(&mut SingleSelectTypeOption),
) -> FlowyResult<()> {
  edit_field_type_option(field_id, editor, action).await
}

pub async fn edit_multi_select_type_option(
  field_id: &str,
  editor: Arc<DatabaseEditor>,
  action: impl FnOnce(&mut MultiSelectTypeOption),
) -> FlowyResult<()> {
  edit_field_type_option(field_id, editor, action).await
}