grid_editor.rs 28 KB

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