grid_editor.rs 33 KB

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