grid_editor.rs 27 KB

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