grid_editor.rs 32 KB

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