grid_editor.rs 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681
  1. use crate::dart_notification::{send_dart_notification, GridNotification};
  2. use crate::entities::CellIdentifier;
  3. use crate::manager::GridUser;
  4. use crate::services::block_manager::GridBlockManager;
  5. use crate::services::field::{default_type_option_builder_from_type, type_option_builder_from_bytes, FieldBuilder};
  6. use crate::services::filter::GridFilterService;
  7. use crate::services::persistence::block_index::BlockIndexCache;
  8. use crate::services::row::*;
  9. use crate::services::tasks::GridTaskScheduler;
  10. use bytes::Bytes;
  11. use flowy_error::{ErrorCode, FlowyError, FlowyResult};
  12. use flowy_grid_data_model::entities::*;
  13. use flowy_grid_data_model::revision::*;
  14. use flowy_revision::{RevisionCloudService, RevisionCompactor, RevisionManager, RevisionObjectBuilder};
  15. use flowy_sync::client_grid::{GridChangeset, GridRevisionPad, JsonDeserializer};
  16. use flowy_sync::entities::revision::Revision;
  17. use flowy_sync::errors::CollaborateResult;
  18. use flowy_sync::util::make_delta_from_revisions;
  19. use lib_infra::future::FutureResult;
  20. use lib_ot::core::PlainTextAttributes;
  21. use std::collections::HashMap;
  22. use std::sync::Arc;
  23. use tokio::sync::RwLock;
  24. pub struct GridRevisionEditor {
  25. pub(crate) grid_id: String,
  26. user: Arc<dyn GridUser>,
  27. grid_pad: Arc<RwLock<GridRevisionPad>>,
  28. rev_manager: Arc<RevisionManager>,
  29. block_manager: Arc<GridBlockManager>,
  30. task_scheduler: Arc<RwLock<GridTaskScheduler>>,
  31. pub(crate) filter_service: Arc<GridFilterService>,
  32. }
  33. impl Drop for GridRevisionEditor {
  34. fn drop(&mut self) {
  35. tracing::trace!("Drop GridRevisionEditor");
  36. }
  37. }
  38. impl GridRevisionEditor {
  39. pub async fn new(
  40. grid_id: &str,
  41. user: Arc<dyn GridUser>,
  42. mut rev_manager: RevisionManager,
  43. persistence: Arc<BlockIndexCache>,
  44. task_scheduler: Arc<RwLock<GridTaskScheduler>>,
  45. ) -> FlowyResult<Arc<Self>> {
  46. let token = user.token()?;
  47. let cloud = Arc::new(GridRevisionCloudService { token });
  48. let grid_pad = rev_manager.load::<GridPadBuilder>(Some(cloud)).await?;
  49. let rev_manager = Arc::new(rev_manager);
  50. let grid_pad = Arc::new(RwLock::new(grid_pad));
  51. let block_meta_revs = grid_pad.read().await.get_block_meta_revs();
  52. let block_manager = Arc::new(GridBlockManager::new(grid_id, &user, block_meta_revs, persistence).await?);
  53. let filter_service = Arc::new(GridFilterService::new());
  54. let editor = Arc::new(Self {
  55. grid_id: grid_id.to_owned(),
  56. user,
  57. grid_pad,
  58. rev_manager,
  59. block_manager,
  60. filter_service,
  61. task_scheduler: task_scheduler.clone(),
  62. });
  63. task_scheduler.write().await.register_handler(editor.clone());
  64. Ok(editor)
  65. }
  66. pub async fn insert_field(&self, params: InsertFieldParams) -> FlowyResult<()> {
  67. let InsertFieldParams {
  68. field,
  69. type_option_data,
  70. start_field_id,
  71. grid_id,
  72. } = params;
  73. let field_id = field.id.clone();
  74. if self.contain_field(&field_id).await {
  75. let _ = self
  76. .modify(|grid| {
  77. let deserializer = TypeOptionJsonDeserializer(field.field_type.clone());
  78. let changeset = FieldChangesetParams {
  79. field_id: field.id,
  80. grid_id,
  81. name: Some(field.name),
  82. desc: Some(field.desc),
  83. field_type: Some(field.field_type),
  84. frozen: Some(field.frozen),
  85. visibility: Some(field.visibility),
  86. width: Some(field.width),
  87. type_option_data: Some(type_option_data),
  88. };
  89. Ok(grid.update_field_rev(changeset, deserializer)?)
  90. })
  91. .await?;
  92. let _ = self.notify_did_update_grid_field(&field_id).await?;
  93. } else {
  94. let _ = self
  95. .modify(|grid| {
  96. let builder = type_option_builder_from_bytes(type_option_data, &field.field_type);
  97. let field_rev = FieldBuilder::from_field(field, builder).build();
  98. Ok(grid.create_field_rev(field_rev, start_field_id)?)
  99. })
  100. .await?;
  101. let _ = self.notify_did_insert_grid_field(&field_id).await?;
  102. }
  103. Ok(())
  104. }
  105. pub async fn update_field_type_option(
  106. &self,
  107. grid_id: &str,
  108. field_id: &str,
  109. type_option_data: Vec<u8>,
  110. ) -> FlowyResult<()> {
  111. let result = self.get_field_rev(field_id).await;
  112. if result.is_none() {
  113. tracing::warn!("Can't find the field with id: {}", field_id);
  114. return Ok(());
  115. }
  116. let field_rev = result.unwrap();
  117. let _ = self
  118. .modify(|grid| {
  119. let deserializer = TypeOptionJsonDeserializer(field_rev.field_type.clone());
  120. let changeset = FieldChangesetParams {
  121. field_id: field_id.to_owned(),
  122. grid_id: grid_id.to_owned(),
  123. type_option_data: Some(type_option_data),
  124. ..Default::default()
  125. };
  126. Ok(grid.update_field_rev(changeset, deserializer)?)
  127. })
  128. .await?;
  129. let _ = self.notify_did_update_grid_field(field_id).await?;
  130. Ok(())
  131. }
  132. pub async fn next_field_rev(&self, field_type: &FieldType) -> FlowyResult<FieldRevision> {
  133. let name = format!("Property {}", self.grid_pad.read().await.fields().len() + 1);
  134. let field_rev = FieldBuilder::from_field_type(field_type).name(&name).build();
  135. Ok(field_rev)
  136. }
  137. pub async fn create_next_field_rev(&self, field_type: &FieldType) -> FlowyResult<FieldRevision> {
  138. let field_rev = self.next_field_rev(field_type).await?;
  139. let _ = self
  140. .modify(|grid| Ok(grid.create_field_rev(field_rev.clone(), None)?))
  141. .await?;
  142. let _ = self.notify_did_insert_grid_field(&field_rev.id).await?;
  143. Ok(field_rev)
  144. }
  145. pub async fn contain_field(&self, field_id: &str) -> bool {
  146. self.grid_pad.read().await.contain_field(field_id)
  147. }
  148. pub async fn update_field(&self, params: FieldChangesetParams) -> FlowyResult<()> {
  149. let field_id = params.field_id.clone();
  150. let json_deserializer = match self.grid_pad.read().await.get_field_rev(params.field_id.as_str()) {
  151. None => return Err(ErrorCode::FieldDoesNotExist.into()),
  152. Some((_, field_rev)) => TypeOptionJsonDeserializer(field_rev.field_type.clone()),
  153. };
  154. let _ = self
  155. .modify(|grid| Ok(grid.update_field_rev(params, json_deserializer)?))
  156. .await?;
  157. let _ = self.notify_did_update_grid_field(&field_id).await?;
  158. Ok(())
  159. }
  160. pub async fn replace_field(&self, field_rev: FieldRevision) -> FlowyResult<()> {
  161. let field_id = field_rev.id.clone();
  162. let _ = self
  163. .modify(|grid_pad| Ok(grid_pad.replace_field_rev(field_rev)?))
  164. .await?;
  165. let _ = self.notify_did_update_grid_field(&field_id).await?;
  166. Ok(())
  167. }
  168. pub async fn delete_field(&self, field_id: &str) -> FlowyResult<()> {
  169. let _ = self.modify(|grid_pad| Ok(grid_pad.delete_field_rev(field_id)?)).await?;
  170. let field_order = FieldOrder::from(field_id);
  171. let notified_changeset = GridFieldChangeset::delete(&self.grid_id, vec![field_order]);
  172. let _ = self.notify_did_update_grid(notified_changeset).await?;
  173. Ok(())
  174. }
  175. pub async fn switch_to_field_type(&self, field_id: &str, field_type: &FieldType) -> FlowyResult<()> {
  176. // let block_ids = self
  177. // .get_block_metas()
  178. // .await?
  179. // .into_iter()
  180. // .map(|block_meta| block_meta.block_id)
  181. // .collect();
  182. // let cell_revs = self
  183. // .block_meta_manager
  184. // .get_cell_revs(block_ids, field_id, None)
  185. // .await?;
  186. let type_option_json_builder = |field_type: &FieldType| -> String {
  187. return default_type_option_builder_from_type(field_type).entry().json_str();
  188. };
  189. let _ = self
  190. .modify(|grid| Ok(grid.switch_to_field(field_id, field_type.clone(), type_option_json_builder)?))
  191. .await?;
  192. let _ = self.notify_did_update_grid_field(field_id).await?;
  193. Ok(())
  194. }
  195. pub async fn duplicate_field(&self, field_id: &str) -> FlowyResult<()> {
  196. let duplicated_field_id = gen_field_id();
  197. let _ = self
  198. .modify(|grid| Ok(grid.duplicate_field_rev(field_id, &duplicated_field_id)?))
  199. .await?;
  200. let _ = self.notify_did_insert_grid_field(&duplicated_field_id).await?;
  201. Ok(())
  202. }
  203. pub async fn get_field_rev(&self, field_id: &str) -> Option<FieldRevision> {
  204. let field_rev = self.grid_pad.read().await.get_field_rev(field_id)?.1.clone();
  205. Some(field_rev)
  206. }
  207. pub async fn get_field_revs<T>(&self, field_ids: Option<Vec<T>>) -> FlowyResult<Vec<FieldRevision>>
  208. where
  209. T: Into<FieldOrder>,
  210. {
  211. if field_ids.is_none() {
  212. let field_revs = self.grid_pad.read().await.get_field_revs(None)?;
  213. return Ok(field_revs);
  214. }
  215. let to_field_orders = |item: Vec<T>| item.into_iter().map(|data| data.into()).collect();
  216. let field_orders = field_ids.map_or(vec![], to_field_orders);
  217. let expected_len = field_orders.len();
  218. let field_revs = self.grid_pad.read().await.get_field_revs(Some(field_orders))?;
  219. if expected_len != 0 && field_revs.len() != expected_len {
  220. tracing::error!(
  221. "This is a bug. The len of the field_revs should equal to {}",
  222. expected_len
  223. );
  224. debug_assert!(field_revs.len() == expected_len);
  225. }
  226. Ok(field_revs)
  227. }
  228. pub async fn create_block(&self, block_meta_rev: GridBlockMetaRevision) -> FlowyResult<()> {
  229. let _ = self
  230. .modify(|grid_pad| Ok(grid_pad.create_block_meta_rev(block_meta_rev)?))
  231. .await?;
  232. Ok(())
  233. }
  234. pub async fn update_block(&self, changeset: GridBlockMetaRevisionChangeset) -> FlowyResult<()> {
  235. let _ = self
  236. .modify(|grid_pad| Ok(grid_pad.update_block_rev(changeset)?))
  237. .await?;
  238. Ok(())
  239. }
  240. pub async fn create_row(&self, start_row_id: Option<String>) -> FlowyResult<RowOrder> {
  241. let field_revs = self.grid_pad.read().await.get_field_revs(None)?;
  242. let block_id = self.block_id().await?;
  243. // insert empty row below the row whose id is upper_row_id
  244. let row_rev_ctx = CreateRowRevisionBuilder::new(&field_revs).build();
  245. let row_rev = make_row_rev_from_context(&block_id, row_rev_ctx);
  246. let row_order = RowOrder::from(&row_rev);
  247. // insert the row
  248. let row_count = self.block_manager.create_row(&block_id, row_rev, start_row_id).await?;
  249. // update block row count
  250. let changeset = GridBlockMetaRevisionChangeset::from_row_count(&block_id, row_count);
  251. let _ = self.update_block(changeset).await?;
  252. Ok(row_order)
  253. }
  254. pub async fn insert_rows(&self, contexts: Vec<CreateRowRevisionPayload>) -> FlowyResult<Vec<RowOrder>> {
  255. let block_id = self.block_id().await?;
  256. let mut rows_by_block_id: HashMap<String, Vec<RowRevision>> = HashMap::new();
  257. let mut row_orders = vec![];
  258. for ctx in contexts {
  259. let row_rev = make_row_rev_from_context(&block_id, ctx);
  260. row_orders.push(RowOrder::from(&row_rev));
  261. rows_by_block_id
  262. .entry(block_id.clone())
  263. .or_insert_with(Vec::new)
  264. .push(row_rev);
  265. }
  266. let changesets = self.block_manager.insert_row(rows_by_block_id).await?;
  267. for changeset in changesets {
  268. let _ = self.update_block(changeset).await?;
  269. }
  270. Ok(row_orders)
  271. }
  272. pub async fn update_row(&self, changeset: RowMetaChangeset) -> FlowyResult<()> {
  273. let field_revs = self.get_field_revs::<FieldOrder>(None).await?;
  274. self.block_manager
  275. .update_row(changeset, |row_rev| make_row_from_row_rev(&field_revs, row_rev))
  276. .await
  277. }
  278. pub async fn get_rows(&self, block_id: &str) -> FlowyResult<RepeatedRow> {
  279. let block_ids = vec![block_id.to_owned()];
  280. let mut grid_block_snapshot = self.grid_block_snapshots(Some(block_ids)).await?;
  281. // For the moment, we only support one block.
  282. // We can save the rows into multiple blocks and load them asynchronously in the future.
  283. debug_assert_eq!(grid_block_snapshot.len(), 1);
  284. if grid_block_snapshot.len() == 1 {
  285. let snapshot = grid_block_snapshot.pop().unwrap();
  286. let field_revs = self.get_field_revs::<FieldOrder>(None).await?;
  287. let rows = make_rows_from_row_revs(&field_revs, &snapshot.row_revs);
  288. Ok(rows.into())
  289. } else {
  290. Ok(vec![].into())
  291. }
  292. }
  293. pub async fn get_row(&self, row_id: &str) -> FlowyResult<Option<Row>> {
  294. match self.block_manager.get_row_rev(row_id).await? {
  295. None => Ok(None),
  296. Some(row_rev) => {
  297. let field_revs = self.get_field_revs::<FieldOrder>(None).await?;
  298. let row_revs = vec![row_rev];
  299. let mut rows = make_rows_from_row_revs(&field_revs, &row_revs);
  300. debug_assert!(rows.len() == 1);
  301. Ok(rows.pop())
  302. }
  303. }
  304. }
  305. pub async fn delete_row(&self, row_id: &str) -> FlowyResult<()> {
  306. let _ = self.block_manager.delete_row(row_id).await?;
  307. Ok(())
  308. }
  309. pub async fn duplicate_row(&self, _row_id: &str) -> FlowyResult<()> {
  310. Ok(())
  311. }
  312. pub async fn get_cell(&self, params: &CellIdentifier) -> Option<Cell> {
  313. let field_rev = self.get_field_rev(&params.field_id).await?;
  314. let row_rev = self.block_manager.get_row_rev(&params.row_id).await.ok()??;
  315. let cell_rev = row_rev.cells.get(&params.field_id)?.clone();
  316. let data = decode_cell_data(cell_rev.data, &field_rev).data;
  317. Some(Cell::new(&params.field_id, data))
  318. }
  319. pub async fn get_cell_rev(&self, row_id: &str, field_id: &str) -> FlowyResult<Option<CellRevision>> {
  320. let row_rev = self.block_manager.get_row_rev(row_id).await?;
  321. match row_rev {
  322. None => Ok(None),
  323. Some(row_rev) => {
  324. let cell_rev = row_rev.cells.get(field_id).cloned();
  325. Ok(cell_rev)
  326. }
  327. }
  328. }
  329. #[tracing::instrument(level = "trace", skip_all, err)]
  330. pub async fn update_cell(&self, cell_changeset: CellChangeset) -> FlowyResult<()> {
  331. if cell_changeset.cell_content_changeset.as_ref().is_none() {
  332. return Ok(());
  333. }
  334. let CellChangeset {
  335. grid_id,
  336. row_id,
  337. field_id,
  338. mut cell_content_changeset,
  339. } = cell_changeset;
  340. match self.grid_pad.read().await.get_field_rev(&field_id) {
  341. None => {
  342. let msg = format!("Field not found with id: {}", &field_id);
  343. Err(FlowyError::internal().context(msg))
  344. }
  345. Some((_, field_rev)) => {
  346. tracing::trace!("field changeset: id:{} / value:{:?}", &field_id, cell_content_changeset);
  347. let cell_rev = self.get_cell_rev(&row_id, &field_id).await?;
  348. // Update the changeset.data property with the return value.
  349. cell_content_changeset = Some(apply_cell_data_changeset(
  350. cell_content_changeset.unwrap(),
  351. cell_rev,
  352. field_rev,
  353. )?);
  354. let field_revs = self.get_field_revs::<FieldOrder>(None).await?;
  355. let cell_changeset = CellChangeset {
  356. grid_id,
  357. row_id,
  358. field_id,
  359. cell_content_changeset,
  360. };
  361. let _ = self
  362. .block_manager
  363. .update_cell(cell_changeset, |row_rev| make_row_from_row_rev(&field_revs, row_rev))
  364. .await?;
  365. Ok(())
  366. }
  367. }
  368. }
  369. pub async fn get_blocks(&self, block_ids: Option<Vec<String>>) -> FlowyResult<RepeatedGridBlock> {
  370. let block_snapshots = self.grid_block_snapshots(block_ids.clone()).await?;
  371. make_grid_blocks(block_ids, block_snapshots)
  372. }
  373. pub async fn get_block_meta_revs(&self) -> FlowyResult<Vec<Arc<GridBlockMetaRevision>>> {
  374. let block_meta_revs = self.grid_pad.read().await.get_block_meta_revs();
  375. Ok(block_meta_revs)
  376. }
  377. pub async fn delete_rows(&self, row_orders: Vec<RowOrder>) -> FlowyResult<()> {
  378. let changesets = self.block_manager.delete_rows(row_orders).await?;
  379. for changeset in changesets {
  380. let _ = self.update_block(changeset).await?;
  381. }
  382. Ok(())
  383. }
  384. pub async fn get_grid_data(&self) -> FlowyResult<Grid> {
  385. let pad_read_guard = self.grid_pad.read().await;
  386. let field_orders = pad_read_guard.get_field_orders();
  387. let mut block_orders = vec![];
  388. for block_rev in pad_read_guard.get_block_meta_revs() {
  389. let row_orders = self.block_manager.get_row_orders(&block_rev.block_id).await?;
  390. let block_order = GridBlock {
  391. id: block_rev.block_id.clone(),
  392. row_orders,
  393. };
  394. block_orders.push(block_order);
  395. }
  396. Ok(Grid {
  397. id: self.grid_id.clone(),
  398. field_orders,
  399. blocks: block_orders,
  400. })
  401. }
  402. pub async fn get_grid_setting(&self) -> FlowyResult<GridSetting> {
  403. let read_guard = self.grid_pad.read().await;
  404. let grid_setting_rev = read_guard.get_grid_setting_rev();
  405. Ok(grid_setting_rev.into())
  406. }
  407. pub async fn get_grid_filter(&self, layout_type: &GridLayoutType) -> FlowyResult<Vec<GridFilter>> {
  408. let layout_type: GridLayoutRevision = layout_type.clone().into();
  409. let read_guard = self.grid_pad.read().await;
  410. match read_guard.get_grid_setting_rev().filter.get(&layout_type) {
  411. Some(filter_revs) => Ok(filter_revs.iter().map(GridFilter::from).collect::<Vec<GridFilter>>()),
  412. None => Ok(vec![]),
  413. }
  414. }
  415. pub async fn update_grid_setting(&self, params: GridSettingChangesetParams) -> FlowyResult<()> {
  416. let _ = self
  417. .modify(|grid_pad| Ok(grid_pad.update_grid_setting_rev(params)?))
  418. .await?;
  419. Ok(())
  420. }
  421. pub async fn grid_block_snapshots(&self, block_ids: Option<Vec<String>>) -> FlowyResult<Vec<GridBlockSnapshot>> {
  422. let block_ids = match block_ids {
  423. None => self
  424. .grid_pad
  425. .read()
  426. .await
  427. .get_block_meta_revs()
  428. .iter()
  429. .map(|block_rev| block_rev.block_id.clone())
  430. .collect::<Vec<String>>(),
  431. Some(block_ids) => block_ids,
  432. };
  433. let snapshots = self.block_manager.make_block_snapshots(block_ids).await?;
  434. Ok(snapshots)
  435. }
  436. pub async fn move_item(&self, params: MoveItemParams) -> FlowyResult<()> {
  437. match params.ty {
  438. MoveItemType::MoveField => {
  439. self.move_field(&params.item_id, params.from_index, params.to_index)
  440. .await
  441. }
  442. MoveItemType::MoveRow => self.move_row(&params.item_id, params.from_index, params.to_index).await,
  443. }
  444. }
  445. pub async fn move_field(&self, field_id: &str, from: i32, to: i32) -> FlowyResult<()> {
  446. let _ = self
  447. .modify(|grid_pad| Ok(grid_pad.move_field(field_id, from as usize, to as usize)?))
  448. .await?;
  449. if let Some((index, field_rev)) = self.grid_pad.read().await.get_field_rev(field_id) {
  450. let delete_field_order = FieldOrder::from(field_id);
  451. let insert_field = IndexField::from_field_rev(field_rev, index);
  452. let notified_changeset = GridFieldChangeset {
  453. grid_id: self.grid_id.clone(),
  454. inserted_fields: vec![insert_field],
  455. deleted_fields: vec![delete_field_order],
  456. updated_fields: vec![],
  457. };
  458. let _ = self.notify_did_update_grid(notified_changeset).await?;
  459. }
  460. Ok(())
  461. }
  462. pub async fn move_row(&self, row_id: &str, from: i32, to: i32) -> FlowyResult<()> {
  463. let _ = self.block_manager.move_row(row_id, from as usize, to as usize).await?;
  464. Ok(())
  465. }
  466. pub async fn delta_bytes(&self) -> Bytes {
  467. self.grid_pad.read().await.delta_bytes()
  468. }
  469. pub async fn duplicate_grid(&self) -> FlowyResult<BuildGridContext> {
  470. let grid_pad = self.grid_pad.read().await;
  471. let original_blocks = grid_pad.get_block_meta_revs();
  472. let (duplicated_fields, duplicated_blocks) = grid_pad.duplicate_grid_block_meta().await;
  473. let mut blocks_meta_data = vec![];
  474. if original_blocks.len() == duplicated_blocks.len() {
  475. for (index, original_block_meta) in original_blocks.iter().enumerate() {
  476. let grid_block_meta_editor = self.block_manager.get_editor(&original_block_meta.block_id).await?;
  477. let duplicated_block_id = &duplicated_blocks[index].block_id;
  478. tracing::trace!("Duplicate block:{} meta data", duplicated_block_id);
  479. let duplicated_block_meta_data = grid_block_meta_editor.duplicate_block(duplicated_block_id).await;
  480. blocks_meta_data.push(duplicated_block_meta_data);
  481. }
  482. } else {
  483. debug_assert_eq!(original_blocks.len(), duplicated_blocks.len());
  484. }
  485. drop(grid_pad);
  486. Ok(BuildGridContext {
  487. field_revs: duplicated_fields,
  488. blocks: duplicated_blocks,
  489. blocks_meta_data,
  490. })
  491. }
  492. async fn modify<F>(&self, f: F) -> FlowyResult<()>
  493. where
  494. F: for<'a> FnOnce(&'a mut GridRevisionPad) -> FlowyResult<Option<GridChangeset>>,
  495. {
  496. let mut write_guard = self.grid_pad.write().await;
  497. if let Some(changeset) = f(&mut *write_guard)? {
  498. let _ = self.apply_change(changeset).await?;
  499. }
  500. Ok(())
  501. }
  502. async fn apply_change(&self, change: GridChangeset) -> FlowyResult<()> {
  503. let GridChangeset { delta, md5 } = change;
  504. let user_id = self.user.user_id()?;
  505. let (base_rev_id, rev_id) = self.rev_manager.next_rev_id_pair();
  506. let delta_data = delta.to_delta_bytes();
  507. let revision = Revision::new(
  508. &self.rev_manager.object_id,
  509. base_rev_id,
  510. rev_id,
  511. delta_data,
  512. &user_id,
  513. md5,
  514. );
  515. let _ = self
  516. .rev_manager
  517. .add_local_revision(&revision, Box::new(GridRevisionCompactor()))
  518. .await?;
  519. Ok(())
  520. }
  521. async fn block_id(&self) -> FlowyResult<String> {
  522. match self.grid_pad.read().await.get_block_meta_revs().last() {
  523. None => Err(FlowyError::internal().context("There is no grid block in this grid")),
  524. Some(grid_block) => Ok(grid_block.block_id.clone()),
  525. }
  526. }
  527. #[tracing::instrument(level = "trace", skip_all, err)]
  528. async fn notify_did_insert_grid_field(&self, field_id: &str) -> FlowyResult<()> {
  529. if let Some((index, field_rev)) = self.grid_pad.read().await.get_field_rev(field_id) {
  530. let index_field = IndexField::from_field_rev(field_rev, index);
  531. let notified_changeset = GridFieldChangeset::insert(&self.grid_id, vec![index_field]);
  532. let _ = self.notify_did_update_grid(notified_changeset).await?;
  533. }
  534. Ok(())
  535. }
  536. #[tracing::instrument(level = "trace", skip_all, err)]
  537. async fn notify_did_update_grid_field(&self, field_id: &str) -> FlowyResult<()> {
  538. if let Some((_, field_rev)) = self
  539. .grid_pad
  540. .read()
  541. .await
  542. .get_field_rev(field_id)
  543. .map(|(index, field)| (index, field.clone()))
  544. {
  545. let updated_field = Field::from(field_rev);
  546. let notified_changeset = GridFieldChangeset::update(&self.grid_id, vec![updated_field.clone()]);
  547. let _ = self.notify_did_update_grid(notified_changeset).await?;
  548. send_dart_notification(field_id, GridNotification::DidUpdateField)
  549. .payload(updated_field)
  550. .send();
  551. }
  552. Ok(())
  553. }
  554. async fn notify_did_update_grid(&self, changeset: GridFieldChangeset) -> FlowyResult<()> {
  555. send_dart_notification(&self.grid_id, GridNotification::DidUpdateGridField)
  556. .payload(changeset)
  557. .send();
  558. Ok(())
  559. }
  560. }
  561. #[cfg(feature = "flowy_unit_test")]
  562. impl GridRevisionEditor {
  563. pub fn rev_manager(&self) -> Arc<RevisionManager> {
  564. self.rev_manager.clone()
  565. }
  566. }
  567. pub struct GridPadBuilder();
  568. impl RevisionObjectBuilder for GridPadBuilder {
  569. type Output = GridRevisionPad;
  570. fn build_object(object_id: &str, revisions: Vec<Revision>) -> FlowyResult<Self::Output> {
  571. let pad = GridRevisionPad::from_revisions(object_id, revisions)?;
  572. Ok(pad)
  573. }
  574. }
  575. struct GridRevisionCloudService {
  576. #[allow(dead_code)]
  577. token: String,
  578. }
  579. impl RevisionCloudService for GridRevisionCloudService {
  580. #[tracing::instrument(level = "trace", skip(self))]
  581. fn fetch_object(&self, _user_id: &str, _object_id: &str) -> FutureResult<Vec<Revision>, FlowyError> {
  582. FutureResult::new(async move { Ok(vec![]) })
  583. }
  584. }
  585. struct GridRevisionCompactor();
  586. impl RevisionCompactor for GridRevisionCompactor {
  587. fn bytes_from_revisions(&self, revisions: Vec<Revision>) -> FlowyResult<Bytes> {
  588. let delta = make_delta_from_revisions::<PlainTextAttributes>(revisions)?;
  589. Ok(delta.to_delta_bytes())
  590. }
  591. }
  592. struct TypeOptionJsonDeserializer(FieldType);
  593. impl JsonDeserializer for TypeOptionJsonDeserializer {
  594. fn deserialize(&self, type_option_data: Vec<u8>) -> CollaborateResult<String> {
  595. // The type_option_data sent from Dart is serialized by protobuf.
  596. let builder = type_option_builder_from_bytes(type_option_data, &self.0);
  597. let json = builder.entry().json_str();
  598. tracing::trace!("Deserialize type option data to: {}", json);
  599. Ok(json)
  600. }
  601. }