mirror of
https://github.com/AppFlowy-IO/AppFlowy.git
synced 2024-08-30 18:12:39 +00:00
2cd88594e8
* refactor: weak passed-in params in handler * refactor: rename struct * chore: update tables * chore: update schema * chore: add permission * chore: update tables * chore: support transaction mode * chore: workspace database id * chore: add user workspace * feat: return list of workspaces * chore: add user to workspace * feat: separate database row table * refactor: update schema * chore: partition table * chore: use transaction * refactor: dir * refactor: collab db ref * fix: collab db lock * chore: rename files * chore: add tables descriptions * chore: update readme * docs: update documentation * chore: rename crate * chore: update ref * chore: update tests * chore: update tests * refactor: crate deps * chore: update crate ref * chore: remove unused deps * chore: remove unused deps * chore: update collab crate refs * chore: replace client with transaction in pooler * refactor: return error type * refactor: use anyhow error in deps * feat: supabase postgrest user signin (wip) * fix: Cargo.toml source git deps, changed Error to anyhow::Error * fix: uuid serialization * chore: fix conflict * chore: extend the response * feat: add implementation place holders * feat: impl get_user_workspaces * feat: impl get_user_profile * test: create workspace * fix: postgrest: field names and alias * chore: implement folder restful api * chore: implement collab storate with restful api * feat: added placeholders for impl: update_user_profile, check_user * feat: impl: update_user_profile * feat: impl: check_user * fix: use UidResponse, add more debug info for serde serialization error * fix: get_user_profile: use Optional<UserProfileResponse> * chore: imple init sync * chore: support soft delete * feat: postgresql: add migration test * feat: postgresql migration test: added UID display and colored output * feat: postgresql migration test: workspace role * feat: postgresql migration test: create shared common utils * feat: postgresql migration test: fixed shebang * chore: add flush_collab_update pg function * chore: implement datbaase and document restful api * chore: migrate to use restful api * chore: update table schema * chore: fix tests * chore: remove unused code * chore: format code * chore: remove unused env * fix: tauri build * fix: tauri build --------- Co-authored-by: Fu Zi Xiang <speed2exe@live.com.sg>
892 lines
32 KiB
Rust
892 lines
32 KiB
Rust
use std::sync::{Arc, Weak};
|
|
|
|
use collab_database::database::gen_row_id;
|
|
use collab_database::rows::RowId;
|
|
|
|
use flowy_error::{FlowyError, FlowyResult};
|
|
use lib_dispatch::prelude::{data_result_ok, AFPluginData, AFPluginState, DataResult};
|
|
use lib_infra::util::timestamp;
|
|
|
|
use crate::entities::*;
|
|
use crate::manager::DatabaseManager;
|
|
use crate::services::cell::CellBuilder;
|
|
use crate::services::field::checklist_type_option::ChecklistCellChangeset;
|
|
use crate::services::field::{
|
|
type_option_data_from_pb_or_default, DateCellChangeset, SelectOptionCellChangeset,
|
|
};
|
|
use crate::services::group::{GroupChangeset, GroupSettingChangeset};
|
|
use crate::services::share::csv::CSVFormat;
|
|
|
|
fn upgrade_manager(
|
|
database_manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<Arc<DatabaseManager>> {
|
|
let manager = database_manager
|
|
.upgrade()
|
|
.ok_or(FlowyError::internal().context("The database manager is already dropped"))?;
|
|
Ok(manager)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn get_database_data_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<DatabasePB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id: DatabaseViewIdPB = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
|
|
let data = database_editor.get_database_data(view_id.as_ref()).await?;
|
|
data_result_ok(data)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn open_database_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id: DatabaseViewIdPB = data.into_inner();
|
|
let database_id = manager
|
|
.get_database_id_with_view_id(view_id.as_ref())
|
|
.await?;
|
|
let _ = manager.open_database(&database_id).await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn get_database_id_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<DatabaseIdPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id: DatabaseViewIdPB = data.into_inner();
|
|
let database_id = manager
|
|
.get_database_id_with_view_id(view_id.as_ref())
|
|
.await?;
|
|
data_result_ok(DatabaseIdPB { value: database_id })
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn get_database_setting_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<DatabaseViewSettingPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id: DatabaseViewIdPB = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
|
|
let data = database_editor
|
|
.get_database_view_setting(view_id.as_ref())
|
|
.await?;
|
|
data_result_ok(data)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn update_database_setting_handler(
|
|
data: AFPluginData<DatabaseSettingChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: DatabaseSettingChangesetParams = data.into_inner().try_into()?;
|
|
let editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
|
|
if let Some(update_filter) = params.insert_filter {
|
|
editor.create_or_update_filter(update_filter).await?;
|
|
}
|
|
|
|
if let Some(delete_filter) = params.delete_filter {
|
|
editor.delete_filter(delete_filter).await?;
|
|
}
|
|
|
|
if let Some(update_sort) = params.alert_sort {
|
|
let _ = editor.create_or_update_sort(update_sort).await?;
|
|
}
|
|
if let Some(delete_sort) = params.delete_sort {
|
|
editor.delete_sort(delete_sort).await?;
|
|
}
|
|
|
|
if let Some(layout_type) = params.layout_type {
|
|
editor
|
|
.update_view_layout(¶ms.view_id, layout_type)
|
|
.await?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn get_all_filters_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RepeatedFilterPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id: DatabaseViewIdPB = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
|
|
let filters = database_editor.get_all_filters(view_id.as_ref()).await;
|
|
data_result_ok(filters)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn get_all_sorts_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RepeatedSortPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id: DatabaseViewIdPB = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
|
|
let sorts = database_editor.get_all_sorts(view_id.as_ref()).await;
|
|
data_result_ok(sorts)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn delete_all_sorts_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id: DatabaseViewIdPB = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
|
|
database_editor.delete_all_sorts(view_id.as_ref()).await;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn get_fields_handler(
|
|
data: AFPluginData<GetFieldPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RepeatedFieldPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: GetFieldParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let fields = database_editor
|
|
.get_fields(¶ms.view_id, params.field_ids)
|
|
.into_iter()
|
|
.map(FieldPB::from)
|
|
.collect::<Vec<FieldPB>>()
|
|
.into();
|
|
data_result_ok(fields)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn get_primary_field_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<FieldPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id = data.into_inner().value;
|
|
let database_editor = manager.get_database_with_view_id(&view_id).await?;
|
|
let mut fields = database_editor
|
|
.get_fields(&view_id, None)
|
|
.into_iter()
|
|
.filter(|field| field.is_primary)
|
|
.map(FieldPB::from)
|
|
.collect::<Vec<FieldPB>>();
|
|
|
|
if fields.is_empty() {
|
|
// The primary field should not be empty. Because it is created when the database is created.
|
|
// If it is empty, it must be a bug.
|
|
Err(FlowyError::record_not_found())
|
|
} else {
|
|
if fields.len() > 1 {
|
|
// The primary field should not be more than one. If it is more than one,
|
|
// it must be a bug.
|
|
tracing::error!("The primary field is more than one");
|
|
}
|
|
data_result_ok(fields.remove(0))
|
|
}
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn update_field_handler(
|
|
data: AFPluginData<FieldChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: FieldChangesetParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor.update_field(params).await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn update_field_type_option_handler(
|
|
data: AFPluginData<TypeOptionChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: TypeOptionChangesetParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
if let Some(old_field) = database_editor.get_field(¶ms.field_id) {
|
|
let field_type = FieldType::from(old_field.field_type);
|
|
let type_option_data =
|
|
type_option_data_from_pb_or_default(params.type_option_data, &field_type);
|
|
database_editor
|
|
.update_field_type_option(
|
|
¶ms.view_id,
|
|
¶ms.field_id,
|
|
type_option_data,
|
|
old_field,
|
|
)
|
|
.await?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn delete_field_handler(
|
|
data: AFPluginData<DeleteFieldPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: FieldIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor.delete_field(¶ms.field_id).await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn switch_to_field_handler(
|
|
data: AFPluginData<UpdateFieldTypePayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: EditFieldParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let old_field = database_editor.get_field(¶ms.field_id);
|
|
database_editor
|
|
.switch_to_field_type(¶ms.field_id, ¶ms.field_type)
|
|
.await?;
|
|
|
|
if let Some(new_type_option) = database_editor
|
|
.get_field(¶ms.field_id)
|
|
.map(|field| field.get_any_type_option(field.field_type))
|
|
{
|
|
match (old_field, new_type_option) {
|
|
(Some(old_field), Some(new_type_option)) => {
|
|
database_editor
|
|
.update_field_type_option(
|
|
¶ms.view_id,
|
|
¶ms.field_id,
|
|
new_type_option,
|
|
old_field,
|
|
)
|
|
.await?;
|
|
},
|
|
_ => {
|
|
tracing::warn!("Old field and the new type option should not be empty");
|
|
},
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn duplicate_field_handler(
|
|
data: AFPluginData<DuplicateFieldPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: FieldIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.duplicate_field(¶ms.view_id, ¶ms.field_id)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
/// Return the FieldTypeOptionData if the Field exists otherwise return record not found error.
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn get_field_type_option_data_handler(
|
|
data: AFPluginData<TypeOptionPathPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<TypeOptionPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: TypeOptionPathParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
if let Some((field, data)) = database_editor
|
|
.get_field_type_option_data(¶ms.field_id)
|
|
.await
|
|
{
|
|
let data = TypeOptionPB {
|
|
view_id: params.view_id,
|
|
field: FieldPB::from(field),
|
|
type_option_data: data.to_vec(),
|
|
};
|
|
data_result_ok(data)
|
|
} else {
|
|
Err(FlowyError::record_not_found())
|
|
}
|
|
}
|
|
|
|
/// Create TypeOptionPB and save it. Return the FieldTypeOptionData.
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn create_field_type_option_data_handler(
|
|
data: AFPluginData<CreateFieldPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<TypeOptionPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CreateFieldParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let (field, data) = database_editor
|
|
.create_field_with_type_option(¶ms.view_id, ¶ms.field_type, params.type_option_data)
|
|
.await;
|
|
|
|
let data = TypeOptionPB {
|
|
view_id: params.view_id,
|
|
field: FieldPB::from(field),
|
|
type_option_data: data.to_vec(),
|
|
};
|
|
data_result_ok(data)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn move_field_handler(
|
|
data: AFPluginData<MoveFieldPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: MoveFieldParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.move_field(
|
|
¶ms.view_id,
|
|
¶ms.field_id,
|
|
params.from_index,
|
|
params.to_index,
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
// #[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn get_row_handler(
|
|
data: AFPluginData<RowIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<OptionalRowPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: RowIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let row = database_editor
|
|
.get_row(¶ms.view_id, ¶ms.row_id)
|
|
.map(RowPB::from);
|
|
data_result_ok(OptionalRowPB { row })
|
|
}
|
|
|
|
pub(crate) async fn get_row_meta_handler(
|
|
data: AFPluginData<RowIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RowMetaPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: RowIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
match database_editor.get_row_meta(¶ms.view_id, ¶ms.row_id) {
|
|
None => Err(FlowyError::record_not_found()),
|
|
Some(row) => data_result_ok(row),
|
|
}
|
|
}
|
|
|
|
pub(crate) async fn update_row_meta_handler(
|
|
data: AFPluginData<UpdateRowMetaChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<()> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: UpdateRowMetaParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let row_id = RowId::from(params.id.clone());
|
|
database_editor.update_row_meta(&row_id, params).await;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn delete_row_handler(
|
|
data: AFPluginData<RowIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: RowIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor.delete_row(¶ms.row_id).await;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn duplicate_row_handler(
|
|
data: AFPluginData<RowIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: RowIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.duplicate_row(¶ms.view_id, params.group_id, ¶ms.row_id)
|
|
.await;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn move_row_handler(
|
|
data: AFPluginData<MoveRowPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: MoveRowParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.move_row(¶ms.view_id, params.from_row_id, params.to_row_id)
|
|
.await;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn create_row_handler(
|
|
data: AFPluginData<CreateRowPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RowMetaPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CreateRowParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let fields = database_editor.get_fields(¶ms.view_id, None);
|
|
let cells =
|
|
CellBuilder::with_cells(params.cell_data_by_field_id.unwrap_or_default(), &fields).build();
|
|
let view_id = params.view_id;
|
|
let group_id = params.group_id;
|
|
let params = collab_database::rows::CreateRowParams {
|
|
id: gen_row_id(),
|
|
cells,
|
|
height: 60,
|
|
visibility: true,
|
|
prev_row_id: params.start_row_id,
|
|
timestamp: timestamp(),
|
|
};
|
|
match database_editor
|
|
.create_row(&view_id, group_id, params)
|
|
.await?
|
|
{
|
|
None => Err(FlowyError::internal().context("Create row fail")),
|
|
Some(row) => data_result_ok(RowMetaPB::from(row.meta)),
|
|
}
|
|
}
|
|
|
|
// #[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn get_cell_handler(
|
|
data: AFPluginData<CellIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<CellPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CellIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let cell = database_editor
|
|
.get_cell_pb(¶ms.field_id, ¶ms.row_id)
|
|
.await
|
|
.unwrap_or_else(|| CellPB::empty(¶ms.field_id, params.row_id.into_inner()));
|
|
data_result_ok(cell)
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip_all, err)]
|
|
pub(crate) async fn update_cell_handler(
|
|
data: AFPluginData<CellChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CellChangesetPB = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.update_cell_with_changeset(
|
|
¶ms.view_id,
|
|
RowId::from(params.row_id),
|
|
¶ms.field_id,
|
|
params.cell_changeset.clone(),
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn new_select_option_handler(
|
|
data: AFPluginData<CreateSelectOptionPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<SelectOptionPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CreateSelectOptionParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let result = database_editor
|
|
.create_select_option(¶ms.field_id, params.option_name)
|
|
.await;
|
|
match result {
|
|
None => {
|
|
Err(FlowyError::record_not_found().context("Create select option fail. Can't find the field"))
|
|
},
|
|
Some(pb) => data_result_ok(pb),
|
|
}
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn insert_or_update_select_option_handler(
|
|
data: AFPluginData<RepeatedSelectOptionPayload>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.insert_select_options(
|
|
¶ms.view_id,
|
|
¶ms.field_id,
|
|
RowId::from(params.row_id),
|
|
params.items,
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn delete_select_option_handler(
|
|
data: AFPluginData<RepeatedSelectOptionPayload>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.delete_select_options(
|
|
¶ms.view_id,
|
|
¶ms.field_id,
|
|
RowId::from(params.row_id),
|
|
params.items,
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(data, manager), err)]
|
|
pub(crate) async fn get_select_option_handler(
|
|
data: AFPluginData<CellIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<SelectOptionCellDataPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CellIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let options = database_editor
|
|
.get_select_options(params.row_id, ¶ms.field_id)
|
|
.await;
|
|
data_result_ok(options)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn update_select_option_cell_handler(
|
|
data: AFPluginData<SelectOptionCellChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: SelectOptionCellChangesetParams = data.into_inner().try_into()?;
|
|
let database_editor = manager
|
|
.get_database_with_view_id(¶ms.cell_identifier.view_id)
|
|
.await?;
|
|
let changeset = SelectOptionCellChangeset {
|
|
insert_option_ids: params.insert_option_ids,
|
|
delete_option_ids: params.delete_option_ids,
|
|
};
|
|
database_editor
|
|
.update_cell_with_changeset(
|
|
¶ms.cell_identifier.view_id,
|
|
params.cell_identifier.row_id,
|
|
¶ms.cell_identifier.field_id,
|
|
changeset,
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn get_checklist_cell_data_handler(
|
|
data: AFPluginData<CellIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<ChecklistCellDataPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CellIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let data = database_editor
|
|
.get_checklist_option(params.row_id, ¶ms.field_id)
|
|
.await;
|
|
data_result_ok(data)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn update_checklist_cell_handler(
|
|
data: AFPluginData<ChecklistCellDataChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: ChecklistCellDataChangesetParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let changeset = ChecklistCellChangeset {
|
|
insert_options: params.insert_options,
|
|
selected_option_ids: params.selected_option_ids,
|
|
delete_option_ids: params.delete_option_ids,
|
|
update_options: params.update_options,
|
|
};
|
|
database_editor
|
|
.set_checklist_options(¶ms.view_id, params.row_id, ¶ms.field_id, changeset)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn update_date_cell_handler(
|
|
data: AFPluginData<DateChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> Result<(), FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let data = data.into_inner();
|
|
let cell_id: CellIdParams = data.cell_id.try_into()?;
|
|
let cell_changeset = DateCellChangeset {
|
|
date: data.date,
|
|
time: data.time,
|
|
include_time: data.include_time,
|
|
};
|
|
let database_editor = manager.get_database_with_view_id(&cell_id.view_id).await?;
|
|
database_editor
|
|
.update_cell_with_changeset(
|
|
&cell_id.view_id,
|
|
cell_id.row_id,
|
|
&cell_id.field_id,
|
|
cell_changeset,
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn get_groups_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RepeatedGroupPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: DatabaseViewIdPB = data.into_inner();
|
|
let database_editor = manager.get_database_with_view_id(params.as_ref()).await?;
|
|
let groups = database_editor.load_groups(params.as_ref()).await?;
|
|
data_result_ok(groups)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn get_group_handler(
|
|
data: AFPluginData<DatabaseGroupIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<GroupPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: DatabaseGroupIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let group = database_editor
|
|
.get_group(¶ms.view_id, ¶ms.group_id)
|
|
.await?;
|
|
data_result_ok(group)
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn set_group_by_field_handler(
|
|
data: AFPluginData<GroupByFieldPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<()> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: GroupByFieldParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.set_group_by_field(¶ms.view_id, ¶ms.field_id)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all, err)]
|
|
pub(crate) async fn update_group_handler(
|
|
data: AFPluginData<UpdateGroupPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<()> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: UpdateGroupParams = data.into_inner().try_into()?;
|
|
let view_id = params.view_id.clone();
|
|
let database_editor = manager.get_database_with_view_id(&view_id).await?;
|
|
let group_setting_changeset = GroupSettingChangeset {
|
|
update_groups: vec![GroupChangeset::from(params)],
|
|
};
|
|
database_editor
|
|
.update_group_setting(&view_id, group_setting_changeset)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn move_group_handler(
|
|
data: AFPluginData<MoveGroupPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<()> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: MoveGroupParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.move_group(¶ms.view_id, ¶ms.from_group_id, ¶ms.to_group_id)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn move_group_row_handler(
|
|
data: AFPluginData<MoveGroupRowPayloadPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<()> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: MoveGroupRowParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
database_editor
|
|
.move_group_row(
|
|
¶ms.view_id,
|
|
¶ms.to_group_id,
|
|
params.from_row_id,
|
|
params.to_row_id,
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(manager), err)]
|
|
pub(crate) async fn get_databases_handler(
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RepeatedDatabaseDescriptionPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let data = manager.get_all_databases_description().await;
|
|
data_result_ok(data)
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn set_layout_setting_handler(
|
|
data: AFPluginData<LayoutSettingChangesetPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<()> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: LayoutSettingChangeset = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let layout_params = LayoutSettingParams {
|
|
layout_type: params.layout_type,
|
|
calendar: params.calendar,
|
|
};
|
|
database_editor
|
|
.set_layout_setting(¶ms.view_id, layout_params)
|
|
.await;
|
|
Ok(())
|
|
}
|
|
|
|
pub(crate) async fn get_layout_setting_handler(
|
|
data: AFPluginData<DatabaseLayoutMetaPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<DatabaseLayoutSettingPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: DatabaseLayoutMeta = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let layout_setting_pb = database_editor
|
|
.get_layout_setting(¶ms.view_id, params.layout)
|
|
.await
|
|
.map(DatabaseLayoutSettingPB::from)
|
|
.unwrap_or_default();
|
|
data_result_ok(layout_setting_pb)
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn get_calendar_events_handler(
|
|
data: AFPluginData<CalendarEventRequestPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RepeatedCalendarEventPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CalendarEventRequestParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let events = database_editor
|
|
.get_all_calendar_events(¶ms.view_id)
|
|
.await;
|
|
data_result_ok(RepeatedCalendarEventPB { items: events })
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn get_no_date_calendar_events_handler(
|
|
data: AFPluginData<CalendarEventRequestPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RepeatedNoDateCalendarEventPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: CalendarEventRequestParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let _events = database_editor
|
|
.get_all_no_date_calendar_events(¶ms.view_id)
|
|
.await;
|
|
todo!()
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn get_calendar_event_handler(
|
|
data: AFPluginData<RowIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<CalendarEventPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let params: RowIdParams = data.into_inner().try_into()?;
|
|
let database_editor = manager.get_database_with_view_id(¶ms.view_id).await?;
|
|
let event = database_editor
|
|
.get_calendar_event(¶ms.view_id, params.row_id)
|
|
.await;
|
|
match event {
|
|
None => Err(FlowyError::record_not_found()),
|
|
Some(event) => data_result_ok(event),
|
|
}
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip(data, manager), err)]
|
|
pub(crate) async fn move_calendar_event_handler(
|
|
data: AFPluginData<MoveCalendarEventPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<()> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let data = data.into_inner();
|
|
let cell_id: CellIdParams = data.cell_path.try_into()?;
|
|
let cell_changeset = DateCellChangeset {
|
|
date: Some(data.timestamp.to_string()),
|
|
..Default::default()
|
|
};
|
|
let database_editor = manager.get_database_with_view_id(&cell_id.view_id).await?;
|
|
database_editor
|
|
.update_cell_with_changeset(
|
|
&cell_id.view_id,
|
|
cell_id.row_id,
|
|
&cell_id.field_id,
|
|
cell_changeset,
|
|
)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip_all, err)]
|
|
pub(crate) async fn create_database_view(
|
|
_data: AFPluginData<CreateDatabaseViewPayloadPB>,
|
|
_manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> FlowyResult<()> {
|
|
// let data: CreateDatabaseViewParams = data.into_inner().try_into()?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip_all, err)]
|
|
pub(crate) async fn export_csv_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<DatabaseExportDataPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id = data.into_inner().value;
|
|
let database = manager.get_database_with_view_id(&view_id).await?;
|
|
let data = database.export_csv(CSVFormat::Original).await?;
|
|
data_result_ok(DatabaseExportDataPB {
|
|
export_type: DatabaseExportDataType::CSV,
|
|
data,
|
|
})
|
|
}
|
|
|
|
#[tracing::instrument(level = "debug", skip_all, err)]
|
|
pub(crate) async fn get_snapshots_handler(
|
|
data: AFPluginData<DatabaseViewIdPB>,
|
|
manager: AFPluginState<Weak<DatabaseManager>>,
|
|
) -> DataResult<RepeatedDatabaseSnapshotPB, FlowyError> {
|
|
let manager = upgrade_manager(manager)?;
|
|
let view_id = data.into_inner().value;
|
|
let snapshots = manager.get_database_snapshots(&view_id).await?;
|
|
data_result_ok(RepeatedDatabaseSnapshotPB { items: snapshots })
|
|
}
|