event_handler.rs 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744
  1. use collab_database::database::gen_row_id;
  2. use std::sync::Arc;
  3. use collab_database::rows::RowId;
  4. use lib_infra::util::timestamp;
  5. use flowy_error::{FlowyError, FlowyResult};
  6. use lib_dispatch::prelude::{data_result_ok, AFPluginData, AFPluginState, DataResult};
  7. use crate::entities::*;
  8. use crate::manager::DatabaseManager2;
  9. use crate::services::cell::CellBuilder;
  10. use crate::services::field::checklist_type_option::ChecklistCellChangeset;
  11. use crate::services::field::{
  12. type_option_data_from_pb_or_default, DateCellChangeset, SelectOptionCellChangeset,
  13. };
  14. use crate::services::group::{GroupChangeset, GroupSettingChangeset};
  15. use crate::services::share::csv::CSVFormat;
  16. #[tracing::instrument(level = "trace", skip_all, err)]
  17. pub(crate) async fn get_database_data_handler(
  18. data: AFPluginData<DatabaseViewIdPB>,
  19. manager: AFPluginState<Arc<DatabaseManager2>>,
  20. ) -> DataResult<DatabasePB, FlowyError> {
  21. let view_id: DatabaseViewIdPB = data.into_inner();
  22. let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
  23. let data = database_editor.get_database_data(view_id.as_ref()).await?;
  24. data_result_ok(data)
  25. }
  26. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  27. pub(crate) async fn get_database_setting_handler(
  28. data: AFPluginData<DatabaseViewIdPB>,
  29. manager: AFPluginState<Arc<DatabaseManager2>>,
  30. ) -> DataResult<DatabaseViewSettingPB, FlowyError> {
  31. let view_id: DatabaseViewIdPB = data.into_inner();
  32. let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
  33. let data = database_editor
  34. .get_database_view_setting(view_id.as_ref())
  35. .await?;
  36. data_result_ok(data)
  37. }
  38. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  39. pub(crate) async fn update_database_setting_handler(
  40. data: AFPluginData<DatabaseSettingChangesetPB>,
  41. manager: AFPluginState<Arc<DatabaseManager2>>,
  42. ) -> Result<(), FlowyError> {
  43. let params: DatabaseSettingChangesetParams = data.into_inner().try_into()?;
  44. let editor = manager.get_database_with_view_id(&params.view_id).await?;
  45. if let Some(update_filter) = params.insert_filter {
  46. editor.create_or_update_filter(update_filter).await?;
  47. }
  48. if let Some(delete_filter) = params.delete_filter {
  49. editor.delete_filter(delete_filter).await?;
  50. }
  51. if let Some(update_sort) = params.alert_sort {
  52. let _ = editor.create_or_update_sort(update_sort).await?;
  53. }
  54. if let Some(delete_sort) = params.delete_sort {
  55. editor.delete_sort(delete_sort).await?;
  56. }
  57. if let Some(layout_type) = params.layout_type {
  58. editor
  59. .update_view_layout(&params.view_id, layout_type)
  60. .await?;
  61. }
  62. Ok(())
  63. }
  64. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  65. pub(crate) async fn get_all_filters_handler(
  66. data: AFPluginData<DatabaseViewIdPB>,
  67. manager: AFPluginState<Arc<DatabaseManager2>>,
  68. ) -> DataResult<RepeatedFilterPB, FlowyError> {
  69. let view_id: DatabaseViewIdPB = data.into_inner();
  70. let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
  71. let filters = database_editor.get_all_filters(view_id.as_ref()).await;
  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<DatabaseManager2>>,
  78. ) -> DataResult<RepeatedSortPB, FlowyError> {
  79. let view_id: DatabaseViewIdPB = data.into_inner();
  80. let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
  81. let sorts = database_editor.get_all_sorts(view_id.as_ref()).await;
  82. data_result_ok(sorts)
  83. }
  84. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  85. pub(crate) async fn delete_all_sorts_handler(
  86. data: AFPluginData<DatabaseViewIdPB>,
  87. manager: AFPluginState<Arc<DatabaseManager2>>,
  88. ) -> Result<(), FlowyError> {
  89. let view_id: DatabaseViewIdPB = data.into_inner();
  90. let database_editor = manager.get_database_with_view_id(view_id.as_ref()).await?;
  91. database_editor.delete_all_sorts(view_id.as_ref()).await;
  92. Ok(())
  93. }
  94. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  95. pub(crate) async fn get_fields_handler(
  96. data: AFPluginData<GetFieldPayloadPB>,
  97. manager: AFPluginState<Arc<DatabaseManager2>>,
  98. ) -> DataResult<RepeatedFieldPB, FlowyError> {
  99. let params: GetFieldParams = data.into_inner().try_into()?;
  100. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  101. let fields = database_editor
  102. .get_fields(&params.view_id, params.field_ids)
  103. .into_iter()
  104. .map(FieldPB::from)
  105. .collect::<Vec<FieldPB>>()
  106. .into();
  107. data_result_ok(fields)
  108. }
  109. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  110. pub(crate) async fn update_field_handler(
  111. data: AFPluginData<FieldChangesetPB>,
  112. manager: AFPluginState<Arc<DatabaseManager2>>,
  113. ) -> Result<(), FlowyError> {
  114. let params: FieldChangesetParams = data.into_inner().try_into()?;
  115. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  116. database_editor.update_field(params).await?;
  117. Ok(())
  118. }
  119. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  120. pub(crate) async fn update_field_type_option_handler(
  121. data: AFPluginData<TypeOptionChangesetPB>,
  122. manager: AFPluginState<Arc<DatabaseManager2>>,
  123. ) -> Result<(), FlowyError> {
  124. let params: TypeOptionChangesetParams = data.into_inner().try_into()?;
  125. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  126. if let Some(old_field) = database_editor.get_field(&params.field_id) {
  127. let field_type = FieldType::from(old_field.field_type);
  128. let type_option_data =
  129. type_option_data_from_pb_or_default(params.type_option_data, &field_type);
  130. database_editor
  131. .update_field_type_option(
  132. &params.view_id,
  133. &params.field_id,
  134. type_option_data,
  135. old_field,
  136. )
  137. .await?;
  138. }
  139. Ok(())
  140. }
  141. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  142. pub(crate) async fn delete_field_handler(
  143. data: AFPluginData<DeleteFieldPayloadPB>,
  144. manager: AFPluginState<Arc<DatabaseManager2>>,
  145. ) -> Result<(), FlowyError> {
  146. let params: FieldIdParams = data.into_inner().try_into()?;
  147. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  148. database_editor.delete_field(&params.field_id).await?;
  149. Ok(())
  150. }
  151. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  152. pub(crate) async fn switch_to_field_handler(
  153. data: AFPluginData<UpdateFieldTypePayloadPB>,
  154. manager: AFPluginState<Arc<DatabaseManager2>>,
  155. ) -> Result<(), FlowyError> {
  156. let params: EditFieldParams = data.into_inner().try_into()?;
  157. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  158. let old_field = database_editor.get_field(&params.field_id);
  159. database_editor
  160. .switch_to_field_type(&params.field_id, &params.field_type)
  161. .await?;
  162. if let Some(new_type_option) = database_editor
  163. .get_field(&params.field_id)
  164. .map(|field| field.get_any_type_option(field.field_type))
  165. {
  166. match (old_field, new_type_option) {
  167. (Some(old_field), Some(new_type_option)) => {
  168. database_editor
  169. .update_field_type_option(
  170. &params.view_id,
  171. &params.field_id,
  172. new_type_option,
  173. old_field,
  174. )
  175. .await?;
  176. },
  177. _ => {
  178. tracing::warn!("Old field and the new type option should not be empty");
  179. },
  180. }
  181. }
  182. Ok(())
  183. }
  184. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  185. pub(crate) async fn duplicate_field_handler(
  186. data: AFPluginData<DuplicateFieldPayloadPB>,
  187. manager: AFPluginState<Arc<DatabaseManager2>>,
  188. ) -> Result<(), FlowyError> {
  189. let params: FieldIdParams = data.into_inner().try_into()?;
  190. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  191. database_editor
  192. .duplicate_field(&params.view_id, &params.field_id)
  193. .await?;
  194. Ok(())
  195. }
  196. /// Return the FieldTypeOptionData if the Field exists otherwise return record not found error.
  197. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  198. pub(crate) async fn get_field_type_option_data_handler(
  199. data: AFPluginData<TypeOptionPathPB>,
  200. manager: AFPluginState<Arc<DatabaseManager2>>,
  201. ) -> DataResult<TypeOptionPB, FlowyError> {
  202. let params: TypeOptionPathParams = data.into_inner().try_into()?;
  203. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  204. if let Some((field, data)) = database_editor
  205. .get_field_type_option_data(&params.field_id)
  206. .await
  207. {
  208. let data = TypeOptionPB {
  209. view_id: params.view_id,
  210. field: FieldPB::from(field),
  211. type_option_data: data.to_vec(),
  212. };
  213. data_result_ok(data)
  214. } else {
  215. Err(FlowyError::record_not_found())
  216. }
  217. }
  218. /// Create FieldMeta and save it. Return the FieldTypeOptionData.
  219. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  220. pub(crate) async fn create_field_type_option_data_handler(
  221. data: AFPluginData<CreateFieldPayloadPB>,
  222. manager: AFPluginState<Arc<DatabaseManager2>>,
  223. ) -> DataResult<TypeOptionPB, FlowyError> {
  224. let params: CreateFieldParams = data.into_inner().try_into()?;
  225. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  226. let (field, data) = database_editor
  227. .create_field_with_type_option(&params.view_id, &params.field_type, params.type_option_data)
  228. .await;
  229. let data = TypeOptionPB {
  230. view_id: params.view_id,
  231. field: FieldPB::from(field),
  232. type_option_data: data.to_vec(),
  233. };
  234. data_result_ok(data)
  235. }
  236. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  237. pub(crate) async fn move_field_handler(
  238. data: AFPluginData<MoveFieldPayloadPB>,
  239. manager: AFPluginState<Arc<DatabaseManager2>>,
  240. ) -> Result<(), FlowyError> {
  241. let params: MoveFieldParams = data.into_inner().try_into()?;
  242. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  243. database_editor
  244. .move_field(
  245. &params.view_id,
  246. &params.field_id,
  247. params.from_index,
  248. params.to_index,
  249. )
  250. .await?;
  251. Ok(())
  252. }
  253. // #[tracing::instrument(level = "debug", skip(data, manager), err)]
  254. pub(crate) async fn get_row_handler(
  255. data: AFPluginData<RowIdPB>,
  256. manager: AFPluginState<Arc<DatabaseManager2>>,
  257. ) -> DataResult<OptionalRowPB, FlowyError> {
  258. let params: RowIdParams = data.into_inner().try_into()?;
  259. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  260. let row = database_editor.get_row(&params.row_id).map(RowPB::from);
  261. data_result_ok(OptionalRowPB { row })
  262. }
  263. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  264. pub(crate) async fn delete_row_handler(
  265. data: AFPluginData<RowIdPB>,
  266. manager: AFPluginState<Arc<DatabaseManager2>>,
  267. ) -> Result<(), FlowyError> {
  268. let params: RowIdParams = data.into_inner().try_into()?;
  269. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  270. database_editor.delete_row(&params.row_id).await;
  271. Ok(())
  272. }
  273. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  274. pub(crate) async fn duplicate_row_handler(
  275. data: AFPluginData<RowIdPB>,
  276. manager: AFPluginState<Arc<DatabaseManager2>>,
  277. ) -> Result<(), FlowyError> {
  278. let params: RowIdParams = data.into_inner().try_into()?;
  279. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  280. database_editor
  281. .duplicate_row(&params.view_id, params.group_id, &params.row_id)
  282. .await;
  283. Ok(())
  284. }
  285. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  286. pub(crate) async fn move_row_handler(
  287. data: AFPluginData<MoveRowPayloadPB>,
  288. manager: AFPluginState<Arc<DatabaseManager2>>,
  289. ) -> Result<(), FlowyError> {
  290. let params: MoveRowParams = data.into_inner().try_into()?;
  291. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  292. database_editor
  293. .move_row(&params.view_id, params.from_row_id, params.to_row_id)
  294. .await;
  295. Ok(())
  296. }
  297. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  298. pub(crate) async fn create_row_handler(
  299. data: AFPluginData<CreateRowPayloadPB>,
  300. manager: AFPluginState<Arc<DatabaseManager2>>,
  301. ) -> DataResult<RowPB, FlowyError> {
  302. let params: CreateRowParams = data.into_inner().try_into()?;
  303. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  304. let fields = database_editor.get_fields(&params.view_id, None);
  305. let cells =
  306. CellBuilder::with_cells(params.cell_data_by_field_id.unwrap_or_default(), &fields).build();
  307. let view_id = params.view_id;
  308. let group_id = params.group_id;
  309. let params = collab_database::rows::CreateRowParams {
  310. id: gen_row_id(),
  311. cells,
  312. height: 60,
  313. visibility: true,
  314. prev_row_id: params.start_row_id,
  315. timestamp: timestamp(),
  316. };
  317. match database_editor
  318. .create_row(&view_id, group_id, params)
  319. .await?
  320. {
  321. None => Err(FlowyError::internal().context("Create row fail")),
  322. Some(row) => data_result_ok(RowPB::from(row)),
  323. }
  324. }
  325. // #[tracing::instrument(level = "trace", skip_all, err)]
  326. pub(crate) async fn get_cell_handler(
  327. data: AFPluginData<CellIdPB>,
  328. manager: AFPluginState<Arc<DatabaseManager2>>,
  329. ) -> DataResult<CellPB, FlowyError> {
  330. let params: CellIdParams = data.into_inner().try_into()?;
  331. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  332. let cell = database_editor
  333. .get_cell(&params.field_id, params.row_id)
  334. .await;
  335. data_result_ok(cell)
  336. }
  337. #[tracing::instrument(level = "debug", skip_all, err)]
  338. pub(crate) async fn update_cell_handler(
  339. data: AFPluginData<CellChangesetPB>,
  340. manager: AFPluginState<Arc<DatabaseManager2>>,
  341. ) -> Result<(), FlowyError> {
  342. let params: CellChangesetPB = data.into_inner();
  343. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  344. database_editor
  345. .update_cell_with_changeset(
  346. &params.view_id,
  347. RowId::from(params.row_id),
  348. &params.field_id,
  349. params.cell_changeset.clone(),
  350. )
  351. .await?;
  352. Ok(())
  353. }
  354. #[tracing::instrument(level = "trace", skip_all, err)]
  355. pub(crate) async fn new_select_option_handler(
  356. data: AFPluginData<CreateSelectOptionPayloadPB>,
  357. manager: AFPluginState<Arc<DatabaseManager2>>,
  358. ) -> DataResult<SelectOptionPB, FlowyError> {
  359. let params: CreateSelectOptionParams = data.into_inner().try_into()?;
  360. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  361. let result = database_editor
  362. .create_select_option(&params.field_id, params.option_name)
  363. .await;
  364. match result {
  365. None => {
  366. Err(FlowyError::record_not_found().context("Create select option fail. Can't find the field"))
  367. },
  368. Some(pb) => data_result_ok(pb),
  369. }
  370. }
  371. #[tracing::instrument(level = "trace", skip_all, err)]
  372. pub(crate) async fn insert_or_update_select_option_handler(
  373. data: AFPluginData<RepeatedSelectOptionPayload>,
  374. manager: AFPluginState<Arc<DatabaseManager2>>,
  375. ) -> Result<(), FlowyError> {
  376. let params = data.into_inner();
  377. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  378. database_editor
  379. .insert_select_options(
  380. &params.view_id,
  381. &params.field_id,
  382. RowId::from(params.row_id),
  383. params.items,
  384. )
  385. .await?;
  386. Ok(())
  387. }
  388. #[tracing::instrument(level = "trace", skip_all, err)]
  389. pub(crate) async fn delete_select_option_handler(
  390. data: AFPluginData<RepeatedSelectOptionPayload>,
  391. manager: AFPluginState<Arc<DatabaseManager2>>,
  392. ) -> Result<(), FlowyError> {
  393. let params = data.into_inner();
  394. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  395. database_editor
  396. .delete_select_options(
  397. &params.view_id,
  398. &params.field_id,
  399. RowId::from(params.row_id),
  400. params.items,
  401. )
  402. .await?;
  403. Ok(())
  404. }
  405. #[tracing::instrument(level = "trace", skip(data, manager), err)]
  406. pub(crate) async fn get_select_option_handler(
  407. data: AFPluginData<CellIdPB>,
  408. manager: AFPluginState<Arc<DatabaseManager2>>,
  409. ) -> DataResult<SelectOptionCellDataPB, FlowyError> {
  410. let params: CellIdParams = data.into_inner().try_into()?;
  411. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  412. let options = database_editor
  413. .get_select_options(params.row_id, &params.field_id)
  414. .await;
  415. data_result_ok(options)
  416. }
  417. #[tracing::instrument(level = "trace", skip_all, err)]
  418. pub(crate) async fn update_select_option_cell_handler(
  419. data: AFPluginData<SelectOptionCellChangesetPB>,
  420. manager: AFPluginState<Arc<DatabaseManager2>>,
  421. ) -> Result<(), FlowyError> {
  422. let params: SelectOptionCellChangesetParams = data.into_inner().try_into()?;
  423. let database_editor = manager
  424. .get_database_with_view_id(&params.cell_identifier.view_id)
  425. .await?;
  426. let changeset = SelectOptionCellChangeset {
  427. insert_option_ids: params.insert_option_ids,
  428. delete_option_ids: params.delete_option_ids,
  429. };
  430. database_editor
  431. .update_cell_with_changeset(
  432. &params.cell_identifier.view_id,
  433. params.cell_identifier.row_id,
  434. &params.cell_identifier.field_id,
  435. changeset,
  436. )
  437. .await?;
  438. Ok(())
  439. }
  440. #[tracing::instrument(level = "trace", skip_all, err)]
  441. pub(crate) async fn get_checklist_cell_data_handler(
  442. data: AFPluginData<CellIdPB>,
  443. manager: AFPluginState<Arc<DatabaseManager2>>,
  444. ) -> DataResult<ChecklistCellDataPB, FlowyError> {
  445. let params: CellIdParams = data.into_inner().try_into()?;
  446. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  447. let data = database_editor
  448. .get_checklist_option(params.row_id, &params.field_id)
  449. .await;
  450. data_result_ok(data)
  451. }
  452. #[tracing::instrument(level = "trace", skip_all, err)]
  453. pub(crate) async fn update_checklist_cell_handler(
  454. data: AFPluginData<ChecklistCellDataChangesetPB>,
  455. manager: AFPluginState<Arc<DatabaseManager2>>,
  456. ) -> Result<(), FlowyError> {
  457. let params: ChecklistCellDataChangesetParams = data.into_inner().try_into()?;
  458. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  459. let changeset = ChecklistCellChangeset {
  460. insert_options: params.insert_options,
  461. selected_option_ids: params.selected_option_ids,
  462. delete_option_ids: params.delete_option_ids,
  463. update_options: params.update_options,
  464. };
  465. database_editor
  466. .set_checklist_options(&params.view_id, params.row_id, &params.field_id, changeset)
  467. .await?;
  468. Ok(())
  469. }
  470. #[tracing::instrument(level = "trace", skip_all, err)]
  471. pub(crate) async fn update_date_cell_handler(
  472. data: AFPluginData<DateChangesetPB>,
  473. manager: AFPluginState<Arc<DatabaseManager2>>,
  474. ) -> Result<(), FlowyError> {
  475. let data = data.into_inner();
  476. let cell_id: CellIdParams = data.cell_path.try_into()?;
  477. let cell_changeset = DateCellChangeset {
  478. date: data.date,
  479. time: data.time,
  480. include_time: data.include_time,
  481. };
  482. let database_editor = manager.get_database_with_view_id(&cell_id.view_id).await?;
  483. database_editor
  484. .update_cell_with_changeset(
  485. &cell_id.view_id,
  486. cell_id.row_id,
  487. &cell_id.field_id,
  488. cell_changeset,
  489. )
  490. .await?;
  491. Ok(())
  492. }
  493. #[tracing::instrument(level = "trace", skip_all, err)]
  494. pub(crate) async fn get_groups_handler(
  495. data: AFPluginData<DatabaseViewIdPB>,
  496. manager: AFPluginState<Arc<DatabaseManager2>>,
  497. ) -> DataResult<RepeatedGroupPB, FlowyError> {
  498. let params: DatabaseViewIdPB = data.into_inner();
  499. let database_editor = manager.get_database_with_view_id(params.as_ref()).await?;
  500. let groups = database_editor.load_groups(params.as_ref()).await?;
  501. data_result_ok(groups)
  502. }
  503. #[tracing::instrument(level = "trace", skip_all, err)]
  504. pub(crate) async fn get_group_handler(
  505. data: AFPluginData<DatabaseGroupIdPB>,
  506. manager: AFPluginState<Arc<DatabaseManager2>>,
  507. ) -> DataResult<GroupPB, FlowyError> {
  508. let params: DatabaseGroupIdParams = data.into_inner().try_into()?;
  509. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  510. let group = database_editor
  511. .get_group(&params.view_id, &params.group_id)
  512. .await?;
  513. data_result_ok(group)
  514. }
  515. #[tracing::instrument(level = "trace", skip_all, err)]
  516. pub(crate) async fn set_group_by_field_handler(
  517. data: AFPluginData<GroupByFieldPayloadPB>,
  518. manager: AFPluginState<Arc<DatabaseManager2>>,
  519. ) -> FlowyResult<()> {
  520. let params: GroupByFieldParams = data.into_inner().try_into()?;
  521. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  522. database_editor
  523. .set_group_by_field(&params.view_id, &params.field_id)
  524. .await?;
  525. Ok(())
  526. }
  527. #[tracing::instrument(level = "trace", skip_all, err)]
  528. pub(crate) async fn update_group_handler(
  529. data: AFPluginData<UpdateGroupPB>,
  530. manager: AFPluginState<Arc<DatabaseManager2>>,
  531. ) -> FlowyResult<()> {
  532. let params: UpdateGroupParams = data.into_inner().try_into()?;
  533. let view_id = params.view_id.clone();
  534. let database_editor = manager.get_database_with_view_id(&view_id).await?;
  535. let group_setting_changeset = GroupSettingChangeset {
  536. update_groups: vec![GroupChangeset::from(params)],
  537. };
  538. database_editor
  539. .update_group_setting(&view_id, group_setting_changeset)
  540. .await?;
  541. Ok(())
  542. }
  543. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  544. pub(crate) async fn move_group_handler(
  545. data: AFPluginData<MoveGroupPayloadPB>,
  546. manager: AFPluginState<Arc<DatabaseManager2>>,
  547. ) -> FlowyResult<()> {
  548. let params: MoveGroupParams = data.into_inner().try_into()?;
  549. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  550. database_editor
  551. .move_group(&params.view_id, &params.from_group_id, &params.to_group_id)
  552. .await?;
  553. Ok(())
  554. }
  555. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  556. pub(crate) async fn move_group_row_handler(
  557. data: AFPluginData<MoveGroupRowPayloadPB>,
  558. manager: AFPluginState<Arc<DatabaseManager2>>,
  559. ) -> FlowyResult<()> {
  560. let params: MoveGroupRowParams = data.into_inner().try_into()?;
  561. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  562. database_editor
  563. .move_group_row(
  564. &params.view_id,
  565. &params.to_group_id,
  566. params.from_row_id,
  567. params.to_row_id,
  568. )
  569. .await?;
  570. Ok(())
  571. }
  572. #[tracing::instrument(level = "debug", skip(manager), err)]
  573. pub(crate) async fn get_databases_handler(
  574. manager: AFPluginState<Arc<DatabaseManager2>>,
  575. ) -> DataResult<RepeatedDatabaseDescriptionPB, FlowyError> {
  576. let data = manager.get_all_databases_description().await;
  577. data_result_ok(data)
  578. }
  579. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  580. pub(crate) async fn set_layout_setting_handler(
  581. data: AFPluginData<LayoutSettingChangesetPB>,
  582. manager: AFPluginState<Arc<DatabaseManager2>>,
  583. ) -> FlowyResult<()> {
  584. let params: LayoutSettingChangeset = data.into_inner().try_into()?;
  585. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  586. let layout_params = LayoutSettingParams {
  587. layout_type: params.layout_type,
  588. calendar: params.calendar,
  589. };
  590. database_editor
  591. .set_layout_setting(&params.view_id, layout_params)
  592. .await;
  593. Ok(())
  594. }
  595. pub(crate) async fn get_layout_setting_handler(
  596. data: AFPluginData<DatabaseLayoutMetaPB>,
  597. manager: AFPluginState<Arc<DatabaseManager2>>,
  598. ) -> DataResult<DatabaseLayoutSettingPB, FlowyError> {
  599. let params: DatabaseLayoutMeta = data.into_inner().try_into()?;
  600. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  601. let layout_setting_pb = database_editor
  602. .get_layout_setting(&params.view_id, params.layout)
  603. .await
  604. .map(DatabaseLayoutSettingPB::from)
  605. .unwrap_or_default();
  606. data_result_ok(layout_setting_pb)
  607. }
  608. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  609. pub(crate) async fn get_calendar_events_handler(
  610. data: AFPluginData<CalendarEventRequestPB>,
  611. manager: AFPluginState<Arc<DatabaseManager2>>,
  612. ) -> DataResult<RepeatedCalendarEventPB, FlowyError> {
  613. let params: CalendarEventRequestParams = data.into_inner().try_into()?;
  614. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  615. let events = database_editor
  616. .get_all_calendar_events(&params.view_id)
  617. .await;
  618. data_result_ok(RepeatedCalendarEventPB { items: events })
  619. }
  620. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  621. pub(crate) async fn get_no_date_calendar_events_handler(
  622. data: AFPluginData<CalendarEventRequestPB>,
  623. manager: AFPluginState<Arc<DatabaseManager2>>,
  624. ) -> DataResult<RepeatedNoDateCalendarEventPB, FlowyError> {
  625. let params: CalendarEventRequestParams = data.into_inner().try_into()?;
  626. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  627. let _events = database_editor
  628. .get_all_no_date_calendar_events(&params.view_id)
  629. .await;
  630. todo!()
  631. }
  632. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  633. pub(crate) async fn get_calendar_event_handler(
  634. data: AFPluginData<RowIdPB>,
  635. manager: AFPluginState<Arc<DatabaseManager2>>,
  636. ) -> DataResult<CalendarEventPB, FlowyError> {
  637. let params: RowIdParams = data.into_inner().try_into()?;
  638. let database_editor = manager.get_database_with_view_id(&params.view_id).await?;
  639. let event = database_editor
  640. .get_calendar_event(&params.view_id, params.row_id)
  641. .await;
  642. match event {
  643. None => Err(FlowyError::record_not_found()),
  644. Some(event) => data_result_ok(event),
  645. }
  646. }
  647. #[tracing::instrument(level = "debug", skip(data, manager), err)]
  648. pub(crate) async fn move_calendar_event_handler(
  649. data: AFPluginData<MoveCalendarEventPB>,
  650. manager: AFPluginState<Arc<DatabaseManager2>>,
  651. ) -> FlowyResult<()> {
  652. let data = data.into_inner();
  653. let cell_id: CellIdParams = data.cell_path.try_into()?;
  654. let cell_changeset = DateCellChangeset {
  655. date: Some(data.timestamp.to_string()),
  656. ..Default::default()
  657. };
  658. let database_editor = manager.get_database_with_view_id(&cell_id.view_id).await?;
  659. database_editor
  660. .update_cell_with_changeset(
  661. &cell_id.view_id,
  662. cell_id.row_id,
  663. &cell_id.field_id,
  664. cell_changeset,
  665. )
  666. .await?;
  667. Ok(())
  668. }
  669. #[tracing::instrument(level = "debug", skip_all, err)]
  670. pub(crate) async fn create_database_view(
  671. _data: AFPluginData<CreateDatabaseViewPayloadPB>,
  672. _manager: AFPluginState<Arc<DatabaseManager2>>,
  673. ) -> FlowyResult<()> {
  674. // let data: CreateDatabaseViewParams = data.into_inner().try_into()?;
  675. Ok(())
  676. }
  677. #[tracing::instrument(level = "debug", skip_all, err)]
  678. pub(crate) async fn export_csv_handler(
  679. data: AFPluginData<DatabaseViewIdPB>,
  680. manager: AFPluginState<Arc<DatabaseManager2>>,
  681. ) -> DataResult<DatabaseExportDataPB, FlowyError> {
  682. let view_id = data.into_inner().value;
  683. let database = manager.get_database_with_view_id(&view_id).await?;
  684. let data = database.export_csv(CSVFormat::Original).await?;
  685. data_result_ok(DatabaseExportDataPB {
  686. export_type: DatabaseExportDataType::CSV,
  687. data,
  688. })
  689. }