event_handler.rs 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644
  1. use crate::entities::*;
  2. use crate::manager::DatabaseManager;
  3. use crate::services::cell::{FromCellString, ToCellChangesetString, TypeCellData};
  4. use crate::services::field::{
  5. default_type_option_builder_from_type, select_type_option_from_field_rev,
  6. type_option_builder_from_json_str, DateCellChangeset, DateChangesetPB, SelectOptionCellChangeset,
  7. SelectOptionCellChangesetPB, SelectOptionCellChangesetParams, SelectOptionCellDataPB,
  8. SelectOptionChangeset, SelectOptionChangesetPB, SelectOptionIds, SelectOptionPB,
  9. };
  10. use crate::services::row::make_row_from_row_rev;
  11. use database_model::FieldRevision;
  12. use flowy_error::{ErrorCode, FlowyError, FlowyResult};
  13. use lib_dispatch::prelude::{data_result_ok, AFPluginData, AFPluginState, DataResult};
  14. use std::sync::Arc;
  15. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  16. pub(crate) async fn get_database_data_handler(
  17. data: AFPluginData<DatabaseViewIdPB>,
  18. manager: AFPluginState<Arc<DatabaseManager>>,
  19. ) -> DataResult<DatabasePB, FlowyError> {
  20. let view_id: DatabaseViewIdPB = data.into_inner();
  21. let editor = manager.open_database_view(view_id.as_ref()).await?;
  22. let database = editor.get_database(view_id.as_ref()).await?;
  23. data_result_ok(database)
  24. }
  25. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  26. pub(crate) async fn get_database_setting_handler(
  27. data: AFPluginData<DatabaseViewIdPB>,
  28. manager: AFPluginState<Arc<DatabaseManager>>,
  29. ) -> DataResult<DatabaseViewSettingPB, FlowyError> {
  30. let view_id: DatabaseViewIdPB = data.into_inner();
  31. let editor = manager.open_database_view(view_id.as_ref()).await?;
  32. let database_setting = editor.get_setting(view_id.as_ref()).await?;
  33. data_result_ok(database_setting)
  34. }
  35. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  36. pub(crate) async fn update_database_setting_handler(
  37. data: AFPluginData<DatabaseSettingChangesetPB>,
  38. manager: AFPluginState<Arc<DatabaseManager>>,
  39. ) -> Result<(), FlowyError> {
  40. let params: DatabaseSettingChangesetParams = data.into_inner().try_into()?;
  41. let editor = manager.get_database_editor(&params.view_id).await?;
  42. if let Some(insert_params) = params.insert_group {
  43. editor.insert_group(insert_params).await?;
  44. }
  45. if let Some(delete_params) = params.delete_group {
  46. editor.delete_group(delete_params).await?;
  47. }
  48. if let Some(alter_filter) = params.insert_filter {
  49. editor.create_or_update_filter(alter_filter).await?;
  50. }
  51. if let Some(delete_filter) = params.delete_filter {
  52. editor.delete_filter(delete_filter).await?;
  53. }
  54. if let Some(alter_sort) = params.alert_sort {
  55. let _ = editor.create_or_update_sort(alter_sort).await?;
  56. }
  57. if let Some(delete_sort) = params.delete_sort {
  58. editor.delete_sort(delete_sort).await?;
  59. }
  60. Ok(())
  61. }
  62. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  63. pub(crate) async fn get_all_filters_handler(
  64. data: AFPluginData<DatabaseViewIdPB>,
  65. manager: AFPluginState<Arc<DatabaseManager>>,
  66. ) -> DataResult<RepeatedFilterPB, FlowyError> {
  67. let view_id: DatabaseViewIdPB = data.into_inner();
  68. let editor = manager.open_database_view(view_id.as_ref()).await?;
  69. let filters = RepeatedFilterPB {
  70. items: editor.get_all_filters(view_id.as_ref()).await?,
  71. };
  72. data_result_ok(filters)
  73. }
  74. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  75. pub(crate) async fn get_all_sorts_handler(
  76. data: AFPluginData<DatabaseViewIdPB>,
  77. manager: AFPluginState<Arc<DatabaseManager>>,
  78. ) -> DataResult<RepeatedSortPB, FlowyError> {
  79. let view_id: DatabaseViewIdPB = data.into_inner();
  80. let editor = manager.open_database_view(view_id.as_ref()).await?;
  81. let sorts = RepeatedSortPB {
  82. items: editor.get_all_sorts(view_id.as_ref()).await?,
  83. };
  84. data_result_ok(sorts)
  85. }
  86. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  87. pub(crate) async fn delete_all_sorts_handler(
  88. data: AFPluginData<DatabaseViewIdPB>,
  89. manager: AFPluginState<Arc<DatabaseManager>>,
  90. ) -> Result<(), FlowyError> {
  91. let view_id: DatabaseViewIdPB = data.into_inner();
  92. let editor = manager.open_database_view(view_id.as_ref()).await?;
  93. editor.delete_all_sorts(view_id.as_ref()).await?;
  94. Ok(())
  95. }
  96. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  97. pub(crate) async fn get_fields_handler(
  98. data: AFPluginData<GetFieldPayloadPB>,
  99. manager: AFPluginState<Arc<DatabaseManager>>,
  100. ) -> DataResult<RepeatedFieldPB, FlowyError> {
  101. let params: GetFieldParams = data.into_inner().try_into()?;
  102. let editor = manager.get_database_editor(&params.view_id).await?;
  103. let field_revs = editor.get_field_revs(params.field_ids).await?;
  104. let repeated_field: RepeatedFieldPB = field_revs
  105. .into_iter()
  106. .map(FieldPB::from)
  107. .collect::<Vec<_>>()
  108. .into();
  109. data_result_ok(repeated_field)
  110. }
  111. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  112. pub(crate) async fn update_field_handler(
  113. data: AFPluginData<FieldChangesetPB>,
  114. manager: AFPluginState<Arc<DatabaseManager>>,
  115. ) -> Result<(), FlowyError> {
  116. let changeset: FieldChangesetParams = data.into_inner().try_into()?;
  117. let editor = manager.get_database_editor(&changeset.view_id).await?;
  118. editor.update_field(changeset).await?;
  119. Ok(())
  120. }
  121. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  122. pub(crate) async fn update_field_type_option_handler(
  123. data: AFPluginData<TypeOptionChangesetPB>,
  124. manager: AFPluginState<Arc<DatabaseManager>>,
  125. ) -> Result<(), FlowyError> {
  126. let params: TypeOptionChangesetParams = data.into_inner().try_into()?;
  127. let editor = manager.get_database_editor(&params.view_id).await?;
  128. let old_field_rev = editor.get_field_rev(&params.field_id).await;
  129. editor
  130. .update_field_type_option(
  131. &params.view_id,
  132. &params.field_id,
  133. params.type_option_data,
  134. old_field_rev,
  135. )
  136. .await?;
  137. Ok(())
  138. }
  139. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  140. pub(crate) async fn delete_field_handler(
  141. data: AFPluginData<DeleteFieldPayloadPB>,
  142. manager: AFPluginState<Arc<DatabaseManager>>,
  143. ) -> Result<(), FlowyError> {
  144. let params: FieldIdParams = data.into_inner().try_into()?;
  145. let editor = manager.get_database_editor(&params.view_id).await?;
  146. editor.delete_field(&params.field_id).await?;
  147. Ok(())
  148. }
  149. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  150. pub(crate) async fn switch_to_field_handler(
  151. data: AFPluginData<UpdateFieldTypePayloadPB>,
  152. manager: AFPluginState<Arc<DatabaseManager>>,
  153. ) -> Result<(), FlowyError> {
  154. let params: EditFieldParams = data.into_inner().try_into()?;
  155. let editor = manager.get_database_editor(&params.view_id).await?;
  156. let old_field_rev = editor.get_field_rev(&params.field_id).await;
  157. editor
  158. .switch_to_field_type(&params.field_id, &params.field_type)
  159. .await?;
  160. // Get the field_rev with field_id, if it doesn't exist, we create the default FieldRevision from the FieldType.
  161. let new_field_rev = editor
  162. .get_field_rev(&params.field_id)
  163. .await
  164. .unwrap_or(Arc::new(editor.next_field_rev(&params.field_type).await?));
  165. // Update the type-option data after the field type has been changed
  166. let type_option_data = get_type_option_data(&new_field_rev, &params.field_type).await?;
  167. editor
  168. .update_field_type_option(
  169. &params.view_id,
  170. &new_field_rev.id,
  171. type_option_data,
  172. old_field_rev,
  173. )
  174. .await?;
  175. Ok(())
  176. }
  177. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  178. pub(crate) async fn duplicate_field_handler(
  179. data: AFPluginData<DuplicateFieldPayloadPB>,
  180. manager: AFPluginState<Arc<DatabaseManager>>,
  181. ) -> Result<(), FlowyError> {
  182. let params: FieldIdParams = data.into_inner().try_into()?;
  183. let editor = manager.get_database_editor(&params.view_id).await?;
  184. editor.duplicate_field(&params.field_id).await?;
  185. Ok(())
  186. }
  187. /// Return the FieldTypeOptionData if the Field exists otherwise return record not found error.
  188. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  189. pub(crate) async fn get_field_type_option_data_handler(
  190. data: AFPluginData<TypeOptionPathPB>,
  191. manager: AFPluginState<Arc<DatabaseManager>>,
  192. ) -> DataResult<TypeOptionPB, FlowyError> {
  193. let params: TypeOptionPathParams = data.into_inner().try_into()?;
  194. let editor = manager.get_database_editor(&params.view_id).await?;
  195. match editor.get_field_rev(&params.field_id).await {
  196. None => Err(FlowyError::record_not_found()),
  197. Some(field_rev) => {
  198. let field_type = field_rev.ty.into();
  199. let type_option_data = get_type_option_data(&field_rev, &field_type).await?;
  200. let data = TypeOptionPB {
  201. view_id: params.view_id,
  202. field: field_rev.into(),
  203. type_option_data,
  204. };
  205. data_result_ok(data)
  206. },
  207. }
  208. }
  209. /// Create FieldMeta and save it. Return the FieldTypeOptionData.
  210. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  211. pub(crate) async fn create_field_type_option_data_handler(
  212. data: AFPluginData<CreateFieldPayloadPB>,
  213. manager: AFPluginState<Arc<DatabaseManager>>,
  214. ) -> DataResult<TypeOptionPB, FlowyError> {
  215. let params: CreateFieldParams = data.into_inner().try_into()?;
  216. let editor = manager.get_database_editor(&params.view_id).await?;
  217. let field_rev = editor
  218. .create_new_field_rev_with_type_option(&params.field_type, params.type_option_data)
  219. .await?;
  220. let field_type: FieldType = field_rev.ty.into();
  221. let type_option_data = get_type_option_data(&field_rev, &field_type).await?;
  222. data_result_ok(TypeOptionPB {
  223. view_id: params.view_id,
  224. field: field_rev.into(),
  225. type_option_data,
  226. })
  227. }
  228. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  229. pub(crate) async fn move_field_handler(
  230. data: AFPluginData<MoveFieldPayloadPB>,
  231. manager: AFPluginState<Arc<DatabaseManager>>,
  232. ) -> Result<(), FlowyError> {
  233. let params: MoveFieldParams = data.into_inner().try_into()?;
  234. let editor = manager.get_database_editor(&params.view_id).await?;
  235. editor.move_field(params).await?;
  236. Ok(())
  237. }
  238. /// The [FieldRevision] contains multiple data, each of them belongs to a specific FieldType.
  239. async fn get_type_option_data(
  240. field_rev: &FieldRevision,
  241. field_type: &FieldType,
  242. ) -> FlowyResult<Vec<u8>> {
  243. let s = field_rev
  244. .get_type_option_str(field_type)
  245. .map(|value| value.to_owned())
  246. .unwrap_or_else(|| {
  247. default_type_option_builder_from_type(field_type)
  248. .serializer()
  249. .json_str()
  250. });
  251. let field_type: FieldType = field_rev.ty.into();
  252. let builder = type_option_builder_from_json_str(&s, &field_type);
  253. let type_option_data = builder.serializer().protobuf_bytes().to_vec();
  254. Ok(type_option_data)
  255. }
  256. // #[tracing::instrument(level = "debug", skip(data, manager), err)]
  257. pub(crate) async fn get_row_handler(
  258. data: AFPluginData<RowIdPB>,
  259. manager: AFPluginState<Arc<DatabaseManager>>,
  260. ) -> DataResult<OptionalRowPB, FlowyError> {
  261. let params: RowIdParams = data.into_inner().try_into()?;
  262. let editor = manager.get_database_editor(&params.view_id).await?;
  263. let row = editor
  264. .get_row_rev(&params.row_id)
  265. .await?
  266. .map(make_row_from_row_rev);
  267. data_result_ok(OptionalRowPB { row })
  268. }
  269. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  270. pub(crate) async fn delete_row_handler(
  271. data: AFPluginData<RowIdPB>,
  272. manager: AFPluginState<Arc<DatabaseManager>>,
  273. ) -> Result<(), FlowyError> {
  274. let params: RowIdParams = data.into_inner().try_into()?;
  275. let editor = manager.get_database_editor(&params.view_id).await?;
  276. editor.delete_row(&params.row_id).await?;
  277. Ok(())
  278. }
  279. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  280. pub(crate) async fn duplicate_row_handler(
  281. data: AFPluginData<RowIdPB>,
  282. manager: AFPluginState<Arc<DatabaseManager>>,
  283. ) -> Result<(), FlowyError> {
  284. let params: RowIdParams = data.into_inner().try_into()?;
  285. let editor = manager.get_database_editor(&params.view_id).await?;
  286. editor.duplicate_row(&params.row_id).await?;
  287. Ok(())
  288. }
  289. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  290. pub(crate) async fn move_row_handler(
  291. data: AFPluginData<MoveRowPayloadPB>,
  292. manager: AFPluginState<Arc<DatabaseManager>>,
  293. ) -> Result<(), FlowyError> {
  294. let params: MoveRowParams = data.into_inner().try_into()?;
  295. let editor = manager.get_database_editor(&params.view_id).await?;
  296. editor.move_row(params).await?;
  297. Ok(())
  298. }
  299. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  300. pub(crate) async fn create_row_handler(
  301. data: AFPluginData<CreateRowPayloadPB>,
  302. manager: AFPluginState<Arc<DatabaseManager>>,
  303. ) -> DataResult<RowPB, FlowyError> {
  304. let params: CreateRowParams = data.into_inner().try_into()?;
  305. let editor = manager.get_database_editor(params.view_id.as_ref()).await?;
  306. let row = editor.create_row(params).await?;
  307. data_result_ok(row)
  308. }
  309. #[tracing::instrument(level = "trace", skip_all, err)]
  310. pub(crate) async fn get_cell_handler(
  311. data: AFPluginData<CellIdPB>,
  312. manager: AFPluginState<Arc<DatabaseManager>>,
  313. ) -> DataResult<CellPB, FlowyError> {
  314. let params: CellIdParams = data.into_inner().try_into()?;
  315. let editor = manager.get_database_editor(&params.view_id).await?;
  316. match editor.get_cell(&params).await {
  317. None => data_result_ok(CellPB::empty(&params.field_id, &params.row_id)),
  318. Some(cell) => data_result_ok(cell),
  319. }
  320. }
  321. #[tracing::instrument(level = "trace", skip_all, err)]
  322. pub(crate) async fn update_cell_handler(
  323. data: AFPluginData<CellChangesetPB>,
  324. manager: AFPluginState<Arc<DatabaseManager>>,
  325. ) -> Result<(), FlowyError> {
  326. let changeset: CellChangesetPB = data.into_inner();
  327. let editor = manager.get_database_editor(&changeset.view_id).await?;
  328. editor
  329. .update_cell_with_changeset(
  330. &changeset.row_id,
  331. &changeset.field_id,
  332. changeset.type_cell_data,
  333. )
  334. .await?;
  335. Ok(())
  336. }
  337. #[tracing::instrument(level = "trace", skip_all, err)]
  338. pub(crate) async fn new_select_option_handler(
  339. data: AFPluginData<CreateSelectOptionPayloadPB>,
  340. manager: AFPluginState<Arc<DatabaseManager>>,
  341. ) -> DataResult<SelectOptionPB, FlowyError> {
  342. let params: CreateSelectOptionParams = data.into_inner().try_into()?;
  343. let editor = manager.get_database_editor(&params.view_id).await?;
  344. match editor.get_field_rev(&params.field_id).await {
  345. None => Err(ErrorCode::InvalidData.into()),
  346. Some(field_rev) => {
  347. let type_option = select_type_option_from_field_rev(&field_rev)?;
  348. let select_option = type_option.create_option(&params.option_name);
  349. data_result_ok(select_option)
  350. },
  351. }
  352. }
  353. #[tracing::instrument(level = "trace", skip_all, err)]
  354. pub(crate) async fn update_select_option_handler(
  355. data: AFPluginData<SelectOptionChangesetPB>,
  356. manager: AFPluginState<Arc<DatabaseManager>>,
  357. ) -> Result<(), FlowyError> {
  358. let changeset: SelectOptionChangeset = data.into_inner().try_into()?;
  359. let editor = manager
  360. .get_database_editor(&changeset.cell_path.view_id)
  361. .await?;
  362. let field_id = changeset.cell_path.field_id.clone();
  363. let (tx, rx) = tokio::sync::oneshot::channel();
  364. editor
  365. .modify_field_rev(&changeset.cell_path.view_id, &field_id, |field_rev| {
  366. let mut type_option = select_type_option_from_field_rev(field_rev)?;
  367. let mut cell_changeset_str = None;
  368. let mut is_changed = None;
  369. for option in changeset.insert_options {
  370. cell_changeset_str = Some(
  371. SelectOptionCellChangeset::from_insert_option_id(&option.id).to_cell_changeset_str(),
  372. );
  373. type_option.insert_option(option);
  374. is_changed = Some(());
  375. }
  376. for option in changeset.update_options {
  377. type_option.insert_option(option);
  378. is_changed = Some(());
  379. }
  380. for option in changeset.delete_options {
  381. cell_changeset_str = Some(
  382. SelectOptionCellChangeset::from_delete_option_id(&option.id).to_cell_changeset_str(),
  383. );
  384. type_option.delete_option(option);
  385. is_changed = Some(());
  386. }
  387. if is_changed.is_some() {
  388. field_rev.insert_type_option(&*type_option);
  389. }
  390. let _ = tx.send(cell_changeset_str);
  391. Ok(is_changed)
  392. })
  393. .await?;
  394. if let Ok(Some(cell_changeset_str)) = rx.await {
  395. match editor
  396. .update_cell_with_changeset(
  397. &changeset.cell_path.row_id,
  398. &changeset.cell_path.field_id,
  399. cell_changeset_str,
  400. )
  401. .await
  402. {
  403. Ok(_) => {},
  404. Err(e) => tracing::error!("{}", e),
  405. }
  406. }
  407. Ok(())
  408. }
  409. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  410. pub(crate) async fn get_select_option_handler(
  411. data: AFPluginData<CellIdPB>,
  412. manager: AFPluginState<Arc<DatabaseManager>>,
  413. ) -> DataResult<SelectOptionCellDataPB, FlowyError> {
  414. let params: CellIdParams = data.into_inner().try_into()?;
  415. let editor = manager.get_database_editor(&params.view_id).await?;
  416. match editor.get_field_rev(&params.field_id).await {
  417. None => {
  418. tracing::error!(
  419. "Can't find the select option field with id: {}",
  420. params.field_id
  421. );
  422. data_result_ok(SelectOptionCellDataPB::default())
  423. },
  424. Some(field_rev) => {
  425. //
  426. let cell_rev = editor
  427. .get_cell_rev(&params.row_id, &params.field_id)
  428. .await?;
  429. let type_option = select_type_option_from_field_rev(&field_rev)?;
  430. let type_cell_data: TypeCellData = match cell_rev {
  431. None => TypeCellData {
  432. cell_str: "".to_string(),
  433. field_type: field_rev.ty.into(),
  434. },
  435. Some(cell_rev) => cell_rev.try_into()?,
  436. };
  437. let ids = SelectOptionIds::from_cell_str(&type_cell_data.cell_str)?;
  438. let selected_options = type_option.get_selected_options(ids);
  439. data_result_ok(selected_options)
  440. },
  441. }
  442. }
  443. #[tracing::instrument(level = "trace", skip_all, err)]
  444. pub(crate) async fn update_select_option_cell_handler(
  445. data: AFPluginData<SelectOptionCellChangesetPB>,
  446. manager: AFPluginState<Arc<DatabaseManager>>,
  447. ) -> Result<(), FlowyError> {
  448. let params: SelectOptionCellChangesetParams = data.into_inner().try_into()?;
  449. let editor = manager
  450. .get_database_editor(&params.cell_identifier.view_id)
  451. .await?;
  452. let changeset = SelectOptionCellChangeset {
  453. insert_option_ids: params.insert_option_ids,
  454. delete_option_ids: params.delete_option_ids,
  455. };
  456. editor
  457. .update_cell_with_changeset(
  458. &params.cell_identifier.row_id,
  459. &params.cell_identifier.field_id,
  460. changeset,
  461. )
  462. .await?;
  463. Ok(())
  464. }
  465. #[tracing::instrument(level = "trace", skip_all, err)]
  466. pub(crate) async fn update_date_cell_handler(
  467. data: AFPluginData<DateChangesetPB>,
  468. manager: AFPluginState<Arc<DatabaseManager>>,
  469. ) -> Result<(), FlowyError> {
  470. let data = data.into_inner();
  471. let cell_path: CellIdParams = data.cell_path.try_into()?;
  472. let cell_changeset = DateCellChangeset {
  473. date: data.date,
  474. time: data.time,
  475. include_time: data.include_time,
  476. is_utc: data.is_utc,
  477. };
  478. let editor = manager.get_database_editor(&cell_path.view_id).await?;
  479. editor
  480. .update_cell(cell_path.row_id, cell_path.field_id, cell_changeset)
  481. .await?;
  482. Ok(())
  483. }
  484. #[tracing::instrument(level = "trace", skip_all, err)]
  485. pub(crate) async fn get_groups_handler(
  486. data: AFPluginData<DatabaseViewIdPB>,
  487. manager: AFPluginState<Arc<DatabaseManager>>,
  488. ) -> DataResult<RepeatedGroupPB, FlowyError> {
  489. let params: DatabaseViewIdPB = data.into_inner();
  490. let editor = manager.get_database_editor(&params.value).await?;
  491. let groups = editor.load_groups(&params.value).await?;
  492. data_result_ok(groups)
  493. }
  494. #[tracing::instrument(level = "trace", skip_all, err)]
  495. pub(crate) async fn get_group_handler(
  496. data: AFPluginData<DatabaseGroupIdPB>,
  497. manager: AFPluginState<Arc<DatabaseManager>>,
  498. ) -> DataResult<GroupPB, FlowyError> {
  499. let params: DatabaseGroupIdParams = data.into_inner().try_into()?;
  500. let editor = manager.get_database_editor(&params.view_id).await?;
  501. let group = editor.get_group(&params.view_id, &params.group_id).await?;
  502. data_result_ok(group)
  503. }
  504. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  505. pub(crate) async fn move_group_handler(
  506. data: AFPluginData<MoveGroupPayloadPB>,
  507. manager: AFPluginState<Arc<DatabaseManager>>,
  508. ) -> FlowyResult<()> {
  509. let params: MoveGroupParams = data.into_inner().try_into()?;
  510. let editor = manager.get_database_editor(params.view_id.as_ref()).await?;
  511. editor.move_group(params).await?;
  512. Ok(())
  513. }
  514. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  515. pub(crate) async fn move_group_row_handler(
  516. data: AFPluginData<MoveGroupRowPayloadPB>,
  517. manager: AFPluginState<Arc<DatabaseManager>>,
  518. ) -> FlowyResult<()> {
  519. let params: MoveGroupRowParams = data.into_inner().try_into()?;
  520. let editor = manager.get_database_editor(params.view_id.as_ref()).await?;
  521. editor.move_group_row(params).await?;
  522. Ok(())
  523. }
  524. #[tracing::instrument(level = "debug", skip(manager), err)]
  525. pub(crate) async fn get_databases_handler(
  526. manager: AFPluginState<Arc<DatabaseManager>>,
  527. ) -> DataResult<RepeatedDatabaseDescriptionPB, FlowyError> {
  528. let items = manager
  529. .get_databases()
  530. .await?
  531. .into_iter()
  532. .map(|database_info| DatabaseDescriptionPB {
  533. name: database_info.name,
  534. database_id: database_info.database_id,
  535. })
  536. .collect::<Vec<DatabaseDescriptionPB>>();
  537. data_result_ok(RepeatedDatabaseDescriptionPB { items })
  538. }
  539. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  540. pub(crate) async fn set_layout_setting_handler(
  541. data: AFPluginData<UpdateLayoutSettingPB>,
  542. manager: AFPluginState<Arc<DatabaseManager>>,
  543. ) -> FlowyResult<()> {
  544. let params: UpdateLayoutSettingParams = data.into_inner().try_into()?;
  545. let database_editor = manager.get_database_editor(params.view_id.as_ref()).await?;
  546. database_editor
  547. .set_layout_setting(&params.view_id, params.layout_setting)
  548. .await?;
  549. Ok(())
  550. }
  551. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  552. pub(crate) async fn get_layout_setting_handler(
  553. data: AFPluginData<DatabaseLayoutIdPB>,
  554. manager: AFPluginState<Arc<DatabaseManager>>,
  555. ) -> DataResult<LayoutSettingPB, FlowyError> {
  556. let params = data.into_inner();
  557. let database_editor = manager.get_database_editor(&params.view_id).await?;
  558. let layout_setting = database_editor
  559. .get_layout_setting(&params.view_id, params.layout)
  560. .await?;
  561. data_result_ok(layout_setting.into())
  562. }
  563. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  564. pub(crate) async fn get_calendar_events_handler(
  565. data: AFPluginData<CalendarEventRequestPB>,
  566. manager: AFPluginState<Arc<DatabaseManager>>,
  567. ) -> DataResult<RepeatedCalendarEventPB, FlowyError> {
  568. let params: CalendarEventRequestParams = data.into_inner().try_into()?;
  569. let database_editor = manager.get_database_editor(&params.view_id).await?;
  570. let events = database_editor
  571. .get_all_calendar_events(&params.view_id)
  572. .await;
  573. data_result_ok(RepeatedCalendarEventPB { items: events })
  574. }
  575. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  576. pub(crate) async fn get_calendar_event_handler(
  577. data: AFPluginData<RowIdPB>,
  578. manager: AFPluginState<Arc<DatabaseManager>>,
  579. ) -> DataResult<CalendarEventPB, FlowyError> {
  580. let params: RowIdParams = data.into_inner().try_into()?;
  581. let database_editor = manager.get_database_editor(&params.view_id).await?;
  582. let event = database_editor
  583. .get_calendar_event(&params.view_id, &params.row_id)
  584. .await;
  585. match event {
  586. None => Err(FlowyError::record_not_found()),
  587. Some(event) => data_result_ok(event),
  588. }
  589. }