grid_editor.rs 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871
  1. use crate::dart_notification::{send_dart_notification, GridNotification};
  2. use crate::entities::GridCellIdParams;
  3. use crate::entities::*;
  4. use crate::manager::{GridTaskSchedulerRwLock, GridUser};
  5. use crate::services::block_manager::GridBlockManager;
  6. use crate::services::cell::{apply_cell_data_changeset, decode_any_cell_data, CellBytes};
  7. use crate::services::field::{
  8. default_type_option_builder_from_type, type_option_builder_from_bytes, type_option_builder_from_json_str,
  9. FieldBuilder,
  10. };
  11. use crate::services::filter::GridFilterService;
  12. use crate::services::grid_view_manager::GridViewManager;
  13. use crate::services::persistence::block_index::BlockIndexCache;
  14. use crate::services::row::{make_grid_blocks, make_rows_from_row_revs, GridBlockSnapshot, RowRevisionBuilder};
  15. use bytes::Bytes;
  16. use flowy_error::{ErrorCode, FlowyError, FlowyResult};
  17. use flowy_grid_data_model::revision::*;
  18. use flowy_revision::{
  19. RevisionCloudService, RevisionCompress, RevisionManager, RevisionObjectDeserializer, RevisionObjectSerializer,
  20. };
  21. use flowy_sync::client_grid::{GridRevisionChangeset, GridRevisionPad, JsonDeserializer};
  22. use flowy_sync::entities::revision::Revision;
  23. use flowy_sync::errors::{CollaborateError, CollaborateResult};
  24. use flowy_sync::util::make_operations_from_revisions;
  25. use lib_infra::future::{wrap_future, FutureResult};
  26. use lib_ot::core::EmptyAttributes;
  27. use std::collections::HashMap;
  28. use std::sync::Arc;
  29. use tokio::sync::RwLock;
  30. pub struct GridRevisionEditor {
  31. pub grid_id: String,
  32. user: Arc<dyn GridUser>,
  33. grid_pad: Arc<RwLock<GridRevisionPad>>,
  34. view_manager: Arc<GridViewManager>,
  35. rev_manager: Arc<RevisionManager>,
  36. block_manager: Arc<GridBlockManager>,
  37. #[allow(dead_code)]
  38. pub(crate) filter_service: Arc<GridFilterService>,
  39. }
  40. impl Drop for GridRevisionEditor {
  41. fn drop(&mut self) {
  42. tracing::trace!("Drop GridRevisionEditor");
  43. }
  44. }
  45. impl GridRevisionEditor {
  46. pub async fn new(
  47. grid_id: &str,
  48. user: Arc<dyn GridUser>,
  49. mut rev_manager: RevisionManager,
  50. persistence: Arc<BlockIndexCache>,
  51. task_scheduler: GridTaskSchedulerRwLock,
  52. ) -> FlowyResult<Arc<Self>> {
  53. let token = user.token()?;
  54. let cloud = Arc::new(GridRevisionCloudService { token });
  55. let grid_pad = rev_manager.load::<GridRevisionSerde>(Some(cloud)).await?;
  56. let rev_manager = Arc::new(rev_manager);
  57. let grid_pad = Arc::new(RwLock::new(grid_pad));
  58. // Block manager
  59. let block_meta_revs = grid_pad.read().await.get_block_meta_revs();
  60. let block_manager = Arc::new(GridBlockManager::new(&user, block_meta_revs, persistence).await?);
  61. let filter_service =
  62. GridFilterService::new(grid_pad.clone(), block_manager.clone(), task_scheduler.clone()).await;
  63. // View manager
  64. let view_manager = Arc::new(
  65. GridViewManager::new(
  66. grid_id.to_owned(),
  67. user.clone(),
  68. Arc::new(grid_pad.clone()),
  69. Arc::new(block_manager.clone()),
  70. Arc::new(task_scheduler.clone()),
  71. )
  72. .await?,
  73. );
  74. let editor = Arc::new(Self {
  75. grid_id: grid_id.to_owned(),
  76. user,
  77. grid_pad,
  78. rev_manager,
  79. block_manager,
  80. view_manager,
  81. filter_service: Arc::new(filter_service),
  82. });
  83. Ok(editor)
  84. }
  85. /// Save the type-option data to disk and send a `GridNotification::DidUpdateField` notification
  86. /// to dart side.
  87. ///
  88. /// It will do nothing if the passed-in type_option_data is empty
  89. /// # Arguments
  90. ///
  91. /// * `grid_id`: the id of the grid
  92. /// * `field_id`: the id of the field
  93. /// * `type_option_data`: the updated type-option data. The `type-option` data might be empty
  94. /// if there is no type-option config for that field. For example, the `RichTextTypeOptionPB`.
  95. ///
  96. pub async fn update_field_type_option(
  97. &self,
  98. grid_id: &str,
  99. field_id: &str,
  100. type_option_data: Vec<u8>,
  101. ) -> FlowyResult<()> {
  102. let result = self.get_field_rev(field_id).await;
  103. if result.is_none() {
  104. tracing::warn!("Can't find the field with id: {}", field_id);
  105. return Ok(());
  106. }
  107. let field_rev = result.unwrap();
  108. let changeset = FieldChangesetParams {
  109. field_id: field_id.to_owned(),
  110. grid_id: grid_id.to_owned(),
  111. type_option_data: Some(type_option_data),
  112. ..Default::default()
  113. };
  114. let _ = self.update_field_rev(changeset, field_rev.ty.into()).await?;
  115. let _ = self.notify_did_update_grid_field(field_id).await?;
  116. Ok(())
  117. }
  118. pub async fn next_field_rev(&self, field_type: &FieldType) -> FlowyResult<FieldRevision> {
  119. let name = format!("Property {}", self.grid_pad.read().await.get_fields().len() + 1);
  120. let field_rev = FieldBuilder::from_field_type(field_type).name(&name).build();
  121. Ok(field_rev)
  122. }
  123. pub async fn create_new_field_rev(
  124. &self,
  125. field_type: &FieldType,
  126. type_option_data: Option<Vec<u8>>,
  127. ) -> FlowyResult<FieldRevision> {
  128. let mut field_rev = self.next_field_rev(field_type).await?;
  129. if let Some(type_option_data) = type_option_data {
  130. let type_option_builder = type_option_builder_from_bytes(type_option_data, field_type);
  131. field_rev.insert_type_option(type_option_builder.serializer());
  132. }
  133. let _ = self
  134. .modify(|grid| Ok(grid.create_field_rev(field_rev.clone(), None)?))
  135. .await?;
  136. let _ = self.notify_did_insert_grid_field(&field_rev.id).await?;
  137. Ok(field_rev)
  138. }
  139. pub async fn contain_field(&self, field_id: &str) -> bool {
  140. self.grid_pad.read().await.contain_field(field_id)
  141. }
  142. pub async fn update_field(&self, params: FieldChangesetParams) -> FlowyResult<()> {
  143. let field_id = params.field_id.clone();
  144. let field_type: Option<FieldType> = self
  145. .grid_pad
  146. .read()
  147. .await
  148. .get_field_rev(params.field_id.as_str())
  149. .map(|(_, field_rev)| field_rev.ty.into());
  150. match field_type {
  151. None => Err(ErrorCode::FieldDoesNotExist.into()),
  152. Some(field_type) => {
  153. let _ = self.update_field_rev(params, field_type).await?;
  154. let _ = self.notify_did_update_grid_field(&field_id).await?;
  155. Ok(())
  156. }
  157. }
  158. }
  159. pub async fn modify_field_rev<F>(&self, field_id: &str, f: F) -> FlowyResult<()>
  160. where
  161. F: for<'a> FnOnce(&'a mut FieldRevision) -> FlowyResult<Option<()>>,
  162. {
  163. let mut is_changed = false;
  164. let _ = self
  165. .modify(|grid| {
  166. let changeset = grid.modify_field(field_id, |field_rev| {
  167. f(field_rev).map_err(|e| CollaborateError::internal().context(e))
  168. })?;
  169. is_changed = changeset.is_some();
  170. Ok(changeset)
  171. })
  172. .await?;
  173. if is_changed {
  174. match self.view_manager.did_update_view_field_type_option(field_id).await {
  175. Ok(_) => {}
  176. Err(e) => tracing::error!("View manager update field failed: {:?}", e),
  177. }
  178. let _ = self.notify_did_update_grid_field(field_id).await?;
  179. }
  180. Ok(())
  181. }
  182. pub async fn delete_field(&self, field_id: &str) -> FlowyResult<()> {
  183. let _ = self.modify(|grid_pad| Ok(grid_pad.delete_field_rev(field_id)?)).await?;
  184. let field_order = FieldIdPB::from(field_id);
  185. let notified_changeset = FieldChangesetPB::delete(&self.grid_id, vec![field_order]);
  186. let _ = self.notify_did_update_grid(notified_changeset).await?;
  187. Ok(())
  188. }
  189. pub async fn group_by_field(&self, field_id: &str) -> FlowyResult<()> {
  190. let _ = self.view_manager.group_by_field(field_id).await?;
  191. Ok(())
  192. }
  193. /// Switch the field with id to a new field type.
  194. ///
  195. /// If the field type is not exist before, the default type-option data will be created.
  196. /// Each field type has its corresponding data, aka, the type-option data. Check out [this](https://appflowy.gitbook.io/docs/essential-documentation/contribute-to-appflowy/architecture/frontend/grid#fieldtype)
  197. /// for more information
  198. ///
  199. /// # Arguments
  200. ///
  201. /// * `field_id`: the id of the field
  202. /// * `new_field_type`: the new field type of the field
  203. ///
  204. pub async fn switch_to_field_type(&self, field_id: &str, new_field_type: &FieldType) -> FlowyResult<()> {
  205. //
  206. let make_default_type_option = || -> String {
  207. return default_type_option_builder_from_type(new_field_type)
  208. .serializer()
  209. .json_str();
  210. };
  211. let type_option_transform =
  212. |prev_field_type: FieldTypeRevision, prev_type_option: Option<String>, current_type_option: String| {
  213. let prev_field_type: FieldType = prev_field_type.into();
  214. let mut type_option_builder = type_option_builder_from_json_str(&current_type_option, new_field_type);
  215. if let Some(prev_type_option) = prev_type_option {
  216. type_option_builder.transform(&prev_field_type, prev_type_option)
  217. }
  218. type_option_builder.serializer().json_str()
  219. };
  220. let _ = self
  221. .modify(|grid| {
  222. Ok(grid.switch_to_field(
  223. field_id,
  224. new_field_type.clone(),
  225. make_default_type_option,
  226. type_option_transform,
  227. )?)
  228. })
  229. .await?;
  230. let _ = self.notify_did_update_grid_field(field_id).await?;
  231. Ok(())
  232. }
  233. pub async fn duplicate_field(&self, field_id: &str) -> FlowyResult<()> {
  234. let duplicated_field_id = gen_field_id();
  235. let _ = self
  236. .modify(|grid| Ok(grid.duplicate_field_rev(field_id, &duplicated_field_id)?))
  237. .await?;
  238. let _ = self.notify_did_insert_grid_field(&duplicated_field_id).await?;
  239. Ok(())
  240. }
  241. pub async fn get_field_rev(&self, field_id: &str) -> Option<Arc<FieldRevision>> {
  242. let field_rev = self.grid_pad.read().await.get_field_rev(field_id)?.1.clone();
  243. Some(field_rev)
  244. }
  245. pub async fn get_field_revs(&self, field_ids: Option<Vec<String>>) -> FlowyResult<Vec<Arc<FieldRevision>>> {
  246. if field_ids.is_none() {
  247. let field_revs = self.grid_pad.read().await.get_field_revs(None)?;
  248. return Ok(field_revs);
  249. }
  250. let field_ids = field_ids.unwrap_or_default();
  251. let expected_len = field_ids.len();
  252. let field_revs = self.grid_pad.read().await.get_field_revs(Some(field_ids))?;
  253. if expected_len != 0 && field_revs.len() != expected_len {
  254. tracing::error!(
  255. "This is a bug. The len of the field_revs should equal to {}",
  256. expected_len
  257. );
  258. debug_assert!(field_revs.len() == expected_len);
  259. }
  260. Ok(field_revs)
  261. }
  262. /// Apply the changeset to field. Including the `name`,`field_type`,`width`,`visibility`,and `type_option_data`.
  263. /// Do nothing if the passed-in params doesn't carry any changes.
  264. ///
  265. /// # Arguments
  266. ///
  267. /// * `params`: contains the changesets that is going to applied to the field.
  268. /// Ignore the change if one of the properties is None.
  269. ///
  270. /// * `field_type`: is used by `TypeOptionJsonDeserializer` to deserialize the type_option_data
  271. ///
  272. #[tracing::instrument(level = "debug", skip_all, err)]
  273. async fn update_field_rev(&self, params: FieldChangesetParams, field_type: FieldType) -> FlowyResult<()> {
  274. let mut is_type_option_changed = false;
  275. let _ = self
  276. .modify(|grid| {
  277. let changeset = grid.modify_field(&params.field_id, |field| {
  278. if let Some(name) = params.name {
  279. field.name = name;
  280. }
  281. if let Some(desc) = params.desc {
  282. field.desc = desc;
  283. }
  284. if let Some(field_type) = params.field_type {
  285. field.ty = field_type;
  286. }
  287. if let Some(frozen) = params.frozen {
  288. field.frozen = frozen;
  289. }
  290. if let Some(visibility) = params.visibility {
  291. field.visibility = visibility;
  292. }
  293. if let Some(width) = params.width {
  294. field.width = width;
  295. }
  296. if let Some(type_option_data) = params.type_option_data {
  297. let deserializer = TypeOptionJsonDeserializer(field_type);
  298. is_type_option_changed = true;
  299. match deserializer.deserialize(type_option_data) {
  300. Ok(json_str) => {
  301. let field_type = field.ty;
  302. field.insert_type_option_str(&field_type, json_str);
  303. }
  304. Err(err) => {
  305. tracing::error!("Deserialize data to type option json failed: {}", err);
  306. }
  307. }
  308. }
  309. Ok(Some(()))
  310. })?;
  311. Ok(changeset)
  312. })
  313. .await?;
  314. let _ = self.view_manager.did_update_view_field(&params.field_id).await?;
  315. if is_type_option_changed {
  316. let _ = self
  317. .view_manager
  318. .did_update_view_field_type_option(&params.field_id)
  319. .await?;
  320. }
  321. Ok(())
  322. }
  323. pub async fn create_block(&self, block_meta_rev: GridBlockMetaRevision) -> FlowyResult<()> {
  324. let _ = self
  325. .modify(|grid_pad| Ok(grid_pad.create_block_meta_rev(block_meta_rev)?))
  326. .await?;
  327. Ok(())
  328. }
  329. pub async fn update_block(&self, changeset: GridBlockMetaRevisionChangeset) -> FlowyResult<()> {
  330. let _ = self
  331. .modify(|grid_pad| Ok(grid_pad.update_block_rev(changeset)?))
  332. .await?;
  333. Ok(())
  334. }
  335. pub async fn create_row(&self, params: CreateRowParams) -> FlowyResult<RowPB> {
  336. let mut row_rev = self.create_row_rev().await?;
  337. self.view_manager.will_create_row(&mut row_rev, &params).await;
  338. let row_pb = self.create_row_pb(row_rev, params.start_row_id.clone()).await?;
  339. self.view_manager.did_create_row(&row_pb, &params).await;
  340. Ok(row_pb)
  341. }
  342. #[tracing::instrument(level = "trace", skip_all, err)]
  343. pub async fn move_group(&self, params: MoveGroupParams) -> FlowyResult<()> {
  344. let _ = self.view_manager.move_group(params).await?;
  345. Ok(())
  346. }
  347. pub async fn insert_rows(&self, row_revs: Vec<RowRevision>) -> FlowyResult<Vec<RowPB>> {
  348. let block_id = self.block_id().await?;
  349. let mut rows_by_block_id: HashMap<String, Vec<RowRevision>> = HashMap::new();
  350. let mut row_orders = vec![];
  351. for row_rev in row_revs {
  352. row_orders.push(RowPB::from(&row_rev));
  353. rows_by_block_id
  354. .entry(block_id.clone())
  355. .or_insert_with(Vec::new)
  356. .push(row_rev);
  357. }
  358. let changesets = self.block_manager.insert_row(rows_by_block_id).await?;
  359. for changeset in changesets {
  360. let _ = self.update_block(changeset).await?;
  361. }
  362. Ok(row_orders)
  363. }
  364. pub async fn update_row(&self, changeset: RowChangeset) -> FlowyResult<()> {
  365. let row_id = changeset.row_id.clone();
  366. let _ = self.block_manager.update_row(changeset).await?;
  367. self.view_manager.did_update_cell(&row_id).await;
  368. Ok(())
  369. }
  370. pub async fn get_rows(&self, block_id: &str) -> FlowyResult<RepeatedRowPB> {
  371. let block_ids = vec![block_id.to_owned()];
  372. let mut grid_block_snapshot = self.grid_block_snapshots(Some(block_ids)).await?;
  373. // For the moment, we only support one block.
  374. // We can save the rows into multiple blocks and load them asynchronously in the future.
  375. debug_assert_eq!(grid_block_snapshot.len(), 1);
  376. if grid_block_snapshot.len() == 1 {
  377. let snapshot = grid_block_snapshot.pop().unwrap();
  378. let rows = make_rows_from_row_revs(&snapshot.row_revs);
  379. Ok(rows.into())
  380. } else {
  381. Ok(vec![].into())
  382. }
  383. }
  384. pub async fn get_row_rev(&self, row_id: &str) -> FlowyResult<Option<Arc<RowRevision>>> {
  385. match self.block_manager.get_row_rev(row_id).await? {
  386. None => Ok(None),
  387. Some(row_rev) => Ok(Some(row_rev)),
  388. }
  389. }
  390. pub async fn delete_row(&self, row_id: &str) -> FlowyResult<()> {
  391. let row_rev = self.block_manager.delete_row(row_id).await?;
  392. tracing::trace!("Did delete row:{:?}", row_rev);
  393. if let Some(row_rev) = row_rev {
  394. self.view_manager.did_delete_row(row_rev).await;
  395. }
  396. Ok(())
  397. }
  398. pub async fn duplicate_row(&self, _row_id: &str) -> FlowyResult<()> {
  399. Ok(())
  400. }
  401. pub async fn get_cell(&self, params: &GridCellIdParams) -> Option<GridCellPB> {
  402. let (field_type, cell_bytes) = self.decode_any_cell_data(params).await?;
  403. Some(GridCellPB::new(&params.field_id, field_type, cell_bytes.to_vec()))
  404. }
  405. pub async fn get_cell_bytes(&self, params: &GridCellIdParams) -> Option<CellBytes> {
  406. let (_, cell_data) = self.decode_any_cell_data(params).await?;
  407. Some(cell_data)
  408. }
  409. async fn decode_any_cell_data(&self, params: &GridCellIdParams) -> Option<(FieldType, CellBytes)> {
  410. let field_rev = self.get_field_rev(&params.field_id).await?;
  411. let row_rev = self.block_manager.get_row_rev(&params.row_id).await.ok()??;
  412. let cell_rev = row_rev.cells.get(&params.field_id)?.clone();
  413. Some(decode_any_cell_data(cell_rev.data, &field_rev))
  414. }
  415. pub async fn get_cell_rev(&self, row_id: &str, field_id: &str) -> FlowyResult<Option<CellRevision>> {
  416. let row_rev = self.block_manager.get_row_rev(row_id).await?;
  417. match row_rev {
  418. None => Ok(None),
  419. Some(row_rev) => {
  420. let cell_rev = row_rev.cells.get(field_id).cloned();
  421. Ok(cell_rev)
  422. }
  423. }
  424. }
  425. #[tracing::instrument(level = "trace", skip_all, err)]
  426. pub async fn update_cell(&self, cell_changeset: CellChangesetPB) -> FlowyResult<()> {
  427. let CellChangesetPB {
  428. grid_id,
  429. row_id,
  430. field_id,
  431. mut content,
  432. } = cell_changeset;
  433. match self.grid_pad.read().await.get_field_rev(&field_id) {
  434. None => {
  435. let msg = format!("Field:{} not found", &field_id);
  436. Err(FlowyError::internal().context(msg))
  437. }
  438. Some((_, field_rev)) => {
  439. tracing::trace!("field changeset: id:{} / value:{:?}", &field_id, content);
  440. let cell_rev = self.get_cell_rev(&row_id, &field_id).await?;
  441. // Update the changeset.data property with the return value.
  442. content = apply_cell_data_changeset(content, cell_rev, field_rev)?;
  443. let cell_changeset = CellChangesetPB {
  444. grid_id,
  445. row_id: row_id.clone(),
  446. field_id: field_id.clone(),
  447. content,
  448. };
  449. let _ = self.block_manager.update_cell(cell_changeset).await?;
  450. self.view_manager.did_update_cell(&row_id).await;
  451. Ok(())
  452. }
  453. }
  454. }
  455. pub async fn get_blocks(&self, block_ids: Option<Vec<String>>) -> FlowyResult<RepeatedBlockPB> {
  456. let block_snapshots = self.grid_block_snapshots(block_ids.clone()).await?;
  457. make_grid_blocks(block_ids, block_snapshots)
  458. }
  459. pub async fn get_block_meta_revs(&self) -> FlowyResult<Vec<Arc<GridBlockMetaRevision>>> {
  460. let block_meta_revs = self.grid_pad.read().await.get_block_meta_revs();
  461. Ok(block_meta_revs)
  462. }
  463. pub async fn delete_rows(&self, row_orders: Vec<RowPB>) -> FlowyResult<()> {
  464. let changesets = self.block_manager.delete_rows(row_orders).await?;
  465. for changeset in changesets {
  466. let _ = self.update_block(changeset).await?;
  467. }
  468. Ok(())
  469. }
  470. pub async fn get_grid_data(&self) -> FlowyResult<GridPB> {
  471. let pad_read_guard = self.grid_pad.read().await;
  472. let field_orders = pad_read_guard
  473. .get_field_revs(None)?
  474. .iter()
  475. .map(FieldIdPB::from)
  476. .collect();
  477. let mut block_orders = vec![];
  478. for block_rev in pad_read_guard.get_block_meta_revs() {
  479. let row_orders = self.block_manager.get_row_orders(&block_rev.block_id).await?;
  480. let block_order = BlockPB {
  481. id: block_rev.block_id.clone(),
  482. rows: row_orders,
  483. };
  484. block_orders.push(block_order);
  485. }
  486. Ok(GridPB {
  487. id: self.grid_id.clone(),
  488. fields: field_orders,
  489. blocks: block_orders,
  490. })
  491. }
  492. pub async fn get_grid_setting(&self) -> FlowyResult<GridSettingPB> {
  493. self.view_manager.get_setting().await
  494. }
  495. pub async fn get_grid_filter(&self) -> FlowyResult<Vec<GridFilterConfigurationPB>> {
  496. self.view_manager.get_filters().await
  497. }
  498. pub async fn insert_group(&self, params: InsertGroupParams) -> FlowyResult<()> {
  499. self.view_manager.insert_or_update_group(params).await
  500. }
  501. pub async fn delete_group(&self, params: DeleteGroupParams) -> FlowyResult<()> {
  502. self.view_manager.delete_group(params).await
  503. }
  504. pub async fn create_filter(&self, params: InsertFilterParams) -> FlowyResult<()> {
  505. let _ = self.view_manager.insert_or_update_filter(params).await?;
  506. Ok(())
  507. }
  508. pub async fn delete_filter(&self, params: DeleteFilterParams) -> FlowyResult<()> {
  509. let _ = self.view_manager.delete_filter(params).await?;
  510. Ok(())
  511. }
  512. pub async fn grid_block_snapshots(&self, block_ids: Option<Vec<String>>) -> FlowyResult<Vec<GridBlockSnapshot>> {
  513. let block_ids = match block_ids {
  514. None => self
  515. .grid_pad
  516. .read()
  517. .await
  518. .get_block_meta_revs()
  519. .iter()
  520. .map(|block_rev| block_rev.block_id.clone())
  521. .collect::<Vec<String>>(),
  522. Some(block_ids) => block_ids,
  523. };
  524. let snapshots = self.block_manager.get_block_snapshots(Some(block_ids)).await?;
  525. Ok(snapshots)
  526. }
  527. pub async fn move_row(&self, params: MoveRowParams) -> FlowyResult<()> {
  528. let MoveRowParams {
  529. view_id: _,
  530. from_row_id,
  531. to_row_id,
  532. } = params;
  533. match self.block_manager.get_row_rev(&from_row_id).await? {
  534. None => tracing::warn!("Move row failed, can not find the row:{}", from_row_id),
  535. Some(row_rev) => {
  536. match (
  537. self.block_manager.index_of_row(&from_row_id).await,
  538. self.block_manager.index_of_row(&to_row_id).await,
  539. ) {
  540. (Some(from_index), Some(to_index)) => {
  541. tracing::trace!("Move row from {} to {}", from_index, to_index);
  542. let _ = self
  543. .block_manager
  544. .move_row(row_rev.clone(), from_index, to_index)
  545. .await?;
  546. }
  547. (_, None) => tracing::warn!("Can not find the from row id: {}", from_row_id),
  548. (None, _) => tracing::warn!("Can not find the to row id: {}", to_row_id),
  549. }
  550. }
  551. }
  552. Ok(())
  553. }
  554. pub async fn move_group_row(&self, params: MoveGroupRowParams) -> FlowyResult<()> {
  555. let MoveGroupRowParams {
  556. view_id,
  557. from_row_id,
  558. to_group_id,
  559. to_row_id,
  560. } = params;
  561. match self.block_manager.get_row_rev(&from_row_id).await? {
  562. None => tracing::warn!("Move row failed, can not find the row:{}", from_row_id),
  563. Some(row_rev) => {
  564. let block_manager = self.block_manager.clone();
  565. self.view_manager
  566. .move_group_row(row_rev, to_group_id, to_row_id.clone(), |row_changeset| {
  567. wrap_future(async move {
  568. tracing::trace!("Row data changed: {:?}", row_changeset);
  569. let cell_changesets = row_changeset
  570. .cell_by_field_id
  571. .into_iter()
  572. .map(|(field_id, cell_rev)| CellChangesetPB {
  573. grid_id: view_id.clone(),
  574. row_id: row_changeset.row_id.clone(),
  575. field_id,
  576. content: cell_rev.data,
  577. })
  578. .collect::<Vec<CellChangesetPB>>();
  579. for cell_changeset in cell_changesets {
  580. match block_manager.update_cell(cell_changeset).await {
  581. Ok(_) => {}
  582. Err(e) => tracing::error!("Apply cell changeset error:{:?}", e),
  583. }
  584. }
  585. })
  586. })
  587. .await?;
  588. }
  589. }
  590. Ok(())
  591. }
  592. pub async fn move_field(&self, params: MoveFieldParams) -> FlowyResult<()> {
  593. let MoveFieldParams {
  594. grid_id: _,
  595. field_id,
  596. from_index,
  597. to_index,
  598. } = params;
  599. let _ = self
  600. .modify(|grid_pad| Ok(grid_pad.move_field(&field_id, from_index as usize, to_index as usize)?))
  601. .await?;
  602. if let Some((index, field_rev)) = self.grid_pad.read().await.get_field_rev(&field_id) {
  603. let delete_field_order = FieldIdPB::from(field_id);
  604. let insert_field = IndexFieldPB::from_field_rev(field_rev, index);
  605. let notified_changeset = FieldChangesetPB {
  606. grid_id: self.grid_id.clone(),
  607. inserted_fields: vec![insert_field],
  608. deleted_fields: vec![delete_field_order],
  609. updated_fields: vec![],
  610. };
  611. let _ = self.notify_did_update_grid(notified_changeset).await?;
  612. }
  613. Ok(())
  614. }
  615. pub async fn duplicate_grid(&self) -> FlowyResult<BuildGridContext> {
  616. let grid_pad = self.grid_pad.read().await;
  617. let grid_view_revision_data = self.view_manager.duplicate_grid_view().await?;
  618. let original_blocks = grid_pad.get_block_meta_revs();
  619. let (duplicated_fields, duplicated_blocks) = grid_pad.duplicate_grid_block_meta().await;
  620. let mut blocks_meta_data = vec![];
  621. if original_blocks.len() == duplicated_blocks.len() {
  622. for (index, original_block_meta) in original_blocks.iter().enumerate() {
  623. let grid_block_meta_editor = self
  624. .block_manager
  625. .get_block_editor(&original_block_meta.block_id)
  626. .await?;
  627. let duplicated_block_id = &duplicated_blocks[index].block_id;
  628. tracing::trace!("Duplicate block:{} meta data", duplicated_block_id);
  629. let duplicated_block_meta_data = grid_block_meta_editor.duplicate_block(duplicated_block_id).await;
  630. blocks_meta_data.push(duplicated_block_meta_data);
  631. }
  632. } else {
  633. debug_assert_eq!(original_blocks.len(), duplicated_blocks.len());
  634. }
  635. drop(grid_pad);
  636. Ok(BuildGridContext {
  637. field_revs: duplicated_fields.into_iter().map(Arc::new).collect(),
  638. block_metas: duplicated_blocks,
  639. blocks: blocks_meta_data,
  640. grid_view_revision_data,
  641. })
  642. }
  643. #[tracing::instrument(level = "trace", skip_all, err)]
  644. pub async fn load_groups(&self) -> FlowyResult<RepeatedGridGroupPB> {
  645. self.view_manager.load_groups().await
  646. }
  647. async fn create_row_rev(&self) -> FlowyResult<RowRevision> {
  648. let field_revs = self.grid_pad.read().await.get_field_revs(None)?;
  649. let block_id = self.block_id().await?;
  650. // insert empty row below the row whose id is upper_row_id
  651. let row_rev = RowRevisionBuilder::new(&block_id, &field_revs).build();
  652. Ok(row_rev)
  653. }
  654. async fn create_row_pb(&self, row_rev: RowRevision, start_row_id: Option<String>) -> FlowyResult<RowPB> {
  655. let row_pb = RowPB::from(&row_rev);
  656. let block_id = row_rev.block_id.clone();
  657. // insert the row
  658. let row_count = self.block_manager.create_row(row_rev, start_row_id).await?;
  659. // update block row count
  660. let changeset = GridBlockMetaRevisionChangeset::from_row_count(block_id, row_count);
  661. let _ = self.update_block(changeset).await?;
  662. Ok(row_pb)
  663. }
  664. async fn modify<F>(&self, f: F) -> FlowyResult<()>
  665. where
  666. F: for<'a> FnOnce(&'a mut GridRevisionPad) -> FlowyResult<Option<GridRevisionChangeset>>,
  667. {
  668. let mut write_guard = self.grid_pad.write().await;
  669. if let Some(changeset) = f(&mut *write_guard)? {
  670. let _ = self.apply_change(changeset).await?;
  671. }
  672. Ok(())
  673. }
  674. async fn apply_change(&self, change: GridRevisionChangeset) -> FlowyResult<()> {
  675. let GridRevisionChangeset { operations: delta, md5 } = change;
  676. let user_id = self.user.user_id()?;
  677. let (base_rev_id, rev_id) = self.rev_manager.next_rev_id_pair();
  678. let delta_data = delta.json_bytes();
  679. let revision = Revision::new(
  680. &self.rev_manager.object_id,
  681. base_rev_id,
  682. rev_id,
  683. delta_data,
  684. &user_id,
  685. md5,
  686. );
  687. let _ = self.rev_manager.add_local_revision(&revision).await?;
  688. Ok(())
  689. }
  690. async fn block_id(&self) -> FlowyResult<String> {
  691. match self.grid_pad.read().await.get_block_meta_revs().last() {
  692. None => Err(FlowyError::internal().context("There is no grid block in this grid")),
  693. Some(grid_block) => Ok(grid_block.block_id.clone()),
  694. }
  695. }
  696. #[tracing::instrument(level = "trace", skip_all, err)]
  697. async fn notify_did_insert_grid_field(&self, field_id: &str) -> FlowyResult<()> {
  698. if let Some((index, field_rev)) = self.grid_pad.read().await.get_field_rev(field_id) {
  699. let index_field = IndexFieldPB::from_field_rev(field_rev, index);
  700. let notified_changeset = FieldChangesetPB::insert(&self.grid_id, vec![index_field]);
  701. let _ = self.notify_did_update_grid(notified_changeset).await?;
  702. }
  703. Ok(())
  704. }
  705. #[tracing::instrument(level = "trace", skip_all, err)]
  706. async fn notify_did_update_grid_field(&self, field_id: &str) -> FlowyResult<()> {
  707. if let Some((_, field_rev)) = self
  708. .grid_pad
  709. .read()
  710. .await
  711. .get_field_rev(field_id)
  712. .map(|(index, field)| (index, field.clone()))
  713. {
  714. let updated_field = FieldPB::from(field_rev);
  715. let notified_changeset = FieldChangesetPB::update(&self.grid_id, vec![updated_field.clone()]);
  716. let _ = self.notify_did_update_grid(notified_changeset).await?;
  717. send_dart_notification(field_id, GridNotification::DidUpdateField)
  718. .payload(updated_field)
  719. .send();
  720. }
  721. Ok(())
  722. }
  723. async fn notify_did_update_grid(&self, changeset: FieldChangesetPB) -> FlowyResult<()> {
  724. send_dart_notification(&self.grid_id, GridNotification::DidUpdateGridField)
  725. .payload(changeset)
  726. .send();
  727. Ok(())
  728. }
  729. }
  730. #[cfg(feature = "flowy_unit_test")]
  731. impl GridRevisionEditor {
  732. pub fn rev_manager(&self) -> Arc<RevisionManager> {
  733. self.rev_manager.clone()
  734. }
  735. }
  736. pub struct GridRevisionSerde();
  737. impl RevisionObjectDeserializer for GridRevisionSerde {
  738. type Output = GridRevisionPad;
  739. fn deserialize_revisions(_object_id: &str, revisions: Vec<Revision>) -> FlowyResult<Self::Output> {
  740. let pad = GridRevisionPad::from_revisions(revisions)?;
  741. Ok(pad)
  742. }
  743. }
  744. impl RevisionObjectSerializer for GridRevisionSerde {
  745. fn combine_revisions(revisions: Vec<Revision>) -> FlowyResult<Bytes> {
  746. let operations = make_operations_from_revisions::<EmptyAttributes>(revisions)?;
  747. Ok(operations.json_bytes())
  748. }
  749. }
  750. struct GridRevisionCloudService {
  751. #[allow(dead_code)]
  752. token: String,
  753. }
  754. impl RevisionCloudService for GridRevisionCloudService {
  755. #[tracing::instrument(level = "trace", skip(self))]
  756. fn fetch_object(&self, _user_id: &str, _object_id: &str) -> FutureResult<Vec<Revision>, FlowyError> {
  757. FutureResult::new(async move { Ok(vec![]) })
  758. }
  759. }
  760. pub struct GridRevisionCompress();
  761. impl RevisionCompress for GridRevisionCompress {
  762. fn combine_revisions(&self, revisions: Vec<Revision>) -> FlowyResult<Bytes> {
  763. GridRevisionSerde::combine_revisions(revisions)
  764. }
  765. }
  766. struct TypeOptionJsonDeserializer(FieldType);
  767. impl JsonDeserializer for TypeOptionJsonDeserializer {
  768. fn deserialize(&self, type_option_data: Vec<u8>) -> CollaborateResult<String> {
  769. // The type_option_data sent from Dart is serialized by protobuf.
  770. let builder = type_option_builder_from_bytes(type_option_data, &self.0);
  771. let json = builder.serializer().json_str();
  772. tracing::trace!("Deserialize type-option data to: {}", json);
  773. Ok(json)
  774. }
  775. }