grid_editor.rs 36 KB

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