folder_pad.rs 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820
  1. use crate::util::cal_diff;
  2. use crate::{
  3. client_folder::builder::FolderPadBuilder,
  4. entities::{
  5. folder_info::FolderDelta,
  6. revision::{md5, Revision},
  7. },
  8. errors::{CollaborateError, CollaborateResult},
  9. };
  10. use flowy_folder_data_model::entities::{app::App, trash::Trash, view::View, workspace::Workspace};
  11. use lib_ot::core::*;
  12. use crate::errors::internal_error;
  13. use lib_infra::util::move_vec_element;
  14. use serde::{Deserialize, Serialize};
  15. use std::sync::Arc;
  16. #[derive(Debug, Deserialize, Serialize, Clone, Eq, PartialEq)]
  17. pub struct FolderPad {
  18. pub(crate) workspaces: Vec<Arc<Workspace>>,
  19. pub(crate) trash: Vec<Arc<Trash>>,
  20. #[serde(skip)]
  21. pub(crate) delta: FolderDelta,
  22. }
  23. impl FolderPad {
  24. pub fn new(workspaces: Vec<Workspace>, trash: Vec<Trash>) -> CollaborateResult<Self> {
  25. FolderPadBuilder::new()
  26. .with_workspace(workspaces)
  27. .with_trash(trash)
  28. .build()
  29. }
  30. pub fn from_revisions(revisions: Vec<Revision>) -> CollaborateResult<Self> {
  31. FolderPadBuilder::new().build_with_revisions(revisions)
  32. }
  33. pub fn from_delta(delta: FolderDelta) -> CollaborateResult<Self> {
  34. FolderPadBuilder::new().build_with_delta(delta)
  35. }
  36. pub fn delta(&self) -> &FolderDelta {
  37. &self.delta
  38. }
  39. pub fn reset_folder(&mut self, delta: FolderDelta) -> CollaborateResult<String> {
  40. let folder = FolderPad::from_delta(delta)?;
  41. self.workspaces = folder.workspaces;
  42. self.trash = folder.trash;
  43. self.delta = folder.delta;
  44. Ok(self.md5())
  45. }
  46. pub fn compose_remote_delta(&mut self, delta: FolderDelta) -> CollaborateResult<String> {
  47. let composed_delta = self.delta.compose(&delta)?;
  48. self.reset_folder(composed_delta)
  49. }
  50. pub fn is_empty(&self) -> bool {
  51. self.workspaces.is_empty() && self.trash.is_empty()
  52. }
  53. #[tracing::instrument(level = "trace", skip(self, workspace), fields(workspace_name=%workspace.name), err)]
  54. pub fn create_workspace(&mut self, workspace: Workspace) -> CollaborateResult<Option<FolderChange>> {
  55. let workspace = Arc::new(workspace);
  56. if self.workspaces.contains(&workspace) {
  57. tracing::warn!("[RootFolder]: Duplicate workspace");
  58. return Ok(None);
  59. }
  60. self.modify_workspaces(move |workspaces| {
  61. workspaces.push(workspace);
  62. Ok(Some(()))
  63. })
  64. }
  65. pub fn update_workspace(
  66. &mut self,
  67. workspace_id: &str,
  68. name: Option<String>,
  69. desc: Option<String>,
  70. ) -> CollaborateResult<Option<FolderChange>> {
  71. self.with_workspace(workspace_id, |workspace| {
  72. if let Some(name) = name {
  73. workspace.name = name;
  74. }
  75. if let Some(desc) = desc {
  76. workspace.desc = desc;
  77. }
  78. Ok(Some(()))
  79. })
  80. }
  81. pub fn read_workspaces(&self, workspace_id: Option<String>) -> CollaborateResult<Vec<Workspace>> {
  82. match workspace_id {
  83. None => {
  84. let workspaces = self
  85. .workspaces
  86. .iter()
  87. .map(|workspace| workspace.as_ref().clone())
  88. .collect::<Vec<Workspace>>();
  89. Ok(workspaces)
  90. }
  91. Some(workspace_id) => {
  92. if let Some(workspace) = self.workspaces.iter().find(|workspace| workspace.id == workspace_id) {
  93. Ok(vec![workspace.as_ref().clone()])
  94. } else {
  95. Err(CollaborateError::record_not_found()
  96. .context(format!("Can't find workspace with id {}", workspace_id)))
  97. }
  98. }
  99. }
  100. }
  101. #[tracing::instrument(level = "trace", skip(self), err)]
  102. pub fn delete_workspace(&mut self, workspace_id: &str) -> CollaborateResult<Option<FolderChange>> {
  103. self.modify_workspaces(|workspaces| {
  104. workspaces.retain(|w| w.id != workspace_id);
  105. Ok(Some(()))
  106. })
  107. }
  108. #[tracing::instrument(level = "trace", skip(self), fields(app_name=%app.name), err)]
  109. pub fn create_app(&mut self, app: App) -> CollaborateResult<Option<FolderChange>> {
  110. let workspace_id = app.workspace_id.clone();
  111. self.with_workspace(&workspace_id, move |workspace| {
  112. if workspace.apps.contains(&app) {
  113. tracing::warn!("[RootFolder]: Duplicate app");
  114. return Ok(None);
  115. }
  116. workspace.apps.push(app);
  117. Ok(Some(()))
  118. })
  119. }
  120. pub fn read_app(&self, app_id: &str) -> CollaborateResult<App> {
  121. for workspace in &self.workspaces {
  122. if let Some(app) = workspace.apps.iter().find(|app| app.id == app_id) {
  123. return Ok(app.clone());
  124. }
  125. }
  126. Err(CollaborateError::record_not_found().context(format!("Can't find app with id {}", app_id)))
  127. }
  128. pub fn update_app(
  129. &mut self,
  130. app_id: &str,
  131. name: Option<String>,
  132. desc: Option<String>,
  133. ) -> CollaborateResult<Option<FolderChange>> {
  134. self.with_app(app_id, move |app| {
  135. if let Some(name) = name {
  136. app.name = name;
  137. }
  138. if let Some(desc) = desc {
  139. app.desc = desc;
  140. }
  141. Ok(Some(()))
  142. })
  143. }
  144. #[tracing::instrument(level = "trace", skip(self), err)]
  145. pub fn delete_app(&mut self, app_id: &str) -> CollaborateResult<Option<FolderChange>> {
  146. let app = self.read_app(app_id)?;
  147. self.with_workspace(&app.workspace_id, |workspace| {
  148. workspace.apps.retain(|app| app.id != app_id);
  149. Ok(Some(()))
  150. })
  151. }
  152. #[tracing::instrument(level = "trace", skip(self), err)]
  153. pub fn move_app(&mut self, app_id: &str, from: usize, to: usize) -> CollaborateResult<Option<FolderChange>> {
  154. let app = self.read_app(app_id)?;
  155. self.with_workspace(&app.workspace_id, |workspace| {
  156. match move_vec_element(&mut workspace.apps, |app| app.id == app_id, from, to).map_err(internal_error)? {
  157. true => Ok(Some(())),
  158. false => Ok(None),
  159. }
  160. })
  161. }
  162. #[tracing::instrument(level = "trace", skip(self), fields(view_name=%view.name), err)]
  163. pub fn create_view(&mut self, view: View) -> CollaborateResult<Option<FolderChange>> {
  164. let app_id = view.belong_to_id.clone();
  165. self.with_app(&app_id, move |app| {
  166. if app.belongings.contains(&view) {
  167. tracing::warn!("[RootFolder]: Duplicate view");
  168. return Ok(None);
  169. }
  170. app.belongings.push(view);
  171. Ok(Some(()))
  172. })
  173. }
  174. pub fn read_view(&self, view_id: &str) -> CollaborateResult<View> {
  175. for workspace in &self.workspaces {
  176. for app in &(*workspace.apps) {
  177. if let Some(view) = app.belongings.iter().find(|b| b.id == view_id) {
  178. return Ok(view.clone());
  179. }
  180. }
  181. }
  182. Err(CollaborateError::record_not_found().context(format!("Can't find view with id {}", view_id)))
  183. }
  184. pub fn read_views(&self, belong_to_id: &str) -> CollaborateResult<Vec<View>> {
  185. for workspace in &self.workspaces {
  186. for app in &(*workspace.apps) {
  187. if app.id == belong_to_id {
  188. return Ok(app.belongings.clone().take_items());
  189. }
  190. }
  191. }
  192. Ok(vec![])
  193. }
  194. pub fn update_view(
  195. &mut self,
  196. view_id: &str,
  197. name: Option<String>,
  198. desc: Option<String>,
  199. modified_time: i64,
  200. ) -> CollaborateResult<Option<FolderChange>> {
  201. let view = self.read_view(view_id)?;
  202. self.with_view(&view.belong_to_id, view_id, |view| {
  203. if let Some(name) = name {
  204. view.name = name;
  205. }
  206. if let Some(desc) = desc {
  207. view.desc = desc;
  208. }
  209. view.modified_time = modified_time;
  210. Ok(Some(()))
  211. })
  212. }
  213. #[tracing::instrument(level = "trace", skip(self), err)]
  214. pub fn delete_view(&mut self, view_id: &str) -> CollaborateResult<Option<FolderChange>> {
  215. let view = self.read_view(view_id)?;
  216. self.with_app(&view.belong_to_id, |app| {
  217. app.belongings.retain(|view| view.id != view_id);
  218. Ok(Some(()))
  219. })
  220. }
  221. #[tracing::instrument(level = "trace", skip(self), err)]
  222. pub fn move_view(&mut self, view_id: &str, from: usize, to: usize) -> CollaborateResult<Option<FolderChange>> {
  223. let view = self.read_view(view_id)?;
  224. self.with_app(&view.belong_to_id, |app| {
  225. match move_vec_element(&mut app.belongings, |view| view.id == view_id, from, to).map_err(internal_error)? {
  226. true => Ok(Some(())),
  227. false => Ok(None),
  228. }
  229. })
  230. }
  231. pub fn create_trash(&mut self, trash: Vec<Trash>) -> CollaborateResult<Option<FolderChange>> {
  232. self.with_trash(|t| {
  233. let mut new_trash = trash.into_iter().map(Arc::new).collect::<Vec<Arc<Trash>>>();
  234. t.append(&mut new_trash);
  235. Ok(Some(()))
  236. })
  237. }
  238. pub fn read_trash(&self, trash_id: Option<String>) -> CollaborateResult<Vec<Trash>> {
  239. match trash_id {
  240. None => Ok(self.trash.iter().map(|t| t.as_ref().clone()).collect::<Vec<Trash>>()),
  241. Some(trash_id) => match self.trash.iter().find(|t| t.id == trash_id) {
  242. Some(trash) => Ok(vec![trash.as_ref().clone()]),
  243. None => Ok(vec![]),
  244. },
  245. }
  246. }
  247. pub fn delete_trash(&mut self, trash_ids: Option<Vec<String>>) -> CollaborateResult<Option<FolderChange>> {
  248. match trash_ids {
  249. None => self.with_trash(|trash| {
  250. trash.clear();
  251. Ok(Some(()))
  252. }),
  253. Some(trash_ids) => self.with_trash(|trash| {
  254. trash.retain(|t| !trash_ids.contains(&t.id));
  255. Ok(Some(()))
  256. }),
  257. }
  258. }
  259. pub fn md5(&self) -> String {
  260. md5(&self.delta.to_delta_bytes())
  261. }
  262. pub fn to_json(&self) -> CollaborateResult<String> {
  263. serde_json::to_string(self)
  264. .map_err(|e| CollaborateError::internal().context(format!("serial trash to json failed: {}", e)))
  265. }
  266. }
  267. impl FolderPad {
  268. fn modify_workspaces<F>(&mut self, f: F) -> CollaborateResult<Option<FolderChange>>
  269. where
  270. F: FnOnce(&mut Vec<Arc<Workspace>>) -> CollaborateResult<Option<()>>,
  271. {
  272. let cloned_self = self.clone();
  273. match f(&mut self.workspaces)? {
  274. None => Ok(None),
  275. Some(_) => {
  276. let old = cloned_self.to_json()?;
  277. let new = self.to_json()?;
  278. match cal_diff::<PlainTextAttributes>(old, new) {
  279. None => Ok(None),
  280. Some(delta) => {
  281. self.delta = self.delta.compose(&delta)?;
  282. Ok(Some(FolderChange { delta, md5: self.md5() }))
  283. }
  284. }
  285. }
  286. }
  287. }
  288. fn with_workspace<F>(&mut self, workspace_id: &str, f: F) -> CollaborateResult<Option<FolderChange>>
  289. where
  290. F: FnOnce(&mut Workspace) -> CollaborateResult<Option<()>>,
  291. {
  292. self.modify_workspaces(|workspaces| {
  293. if let Some(workspace) = workspaces.iter_mut().find(|workspace| workspace_id == workspace.id) {
  294. f(Arc::make_mut(workspace))
  295. } else {
  296. tracing::warn!("[FolderPad]: Can't find any workspace with id: {}", workspace_id);
  297. Ok(None)
  298. }
  299. })
  300. }
  301. fn with_trash<F>(&mut self, f: F) -> CollaborateResult<Option<FolderChange>>
  302. where
  303. F: FnOnce(&mut Vec<Arc<Trash>>) -> CollaborateResult<Option<()>>,
  304. {
  305. let cloned_self = self.clone();
  306. match f(&mut self.trash)? {
  307. None => Ok(None),
  308. Some(_) => {
  309. let old = cloned_self.to_json()?;
  310. let new = self.to_json()?;
  311. match cal_diff::<PlainTextAttributes>(old, new) {
  312. None => Ok(None),
  313. Some(delta) => {
  314. self.delta = self.delta.compose(&delta)?;
  315. Ok(Some(FolderChange { delta, md5: self.md5() }))
  316. }
  317. }
  318. }
  319. }
  320. }
  321. fn with_app<F>(&mut self, app_id: &str, f: F) -> CollaborateResult<Option<FolderChange>>
  322. where
  323. F: FnOnce(&mut App) -> CollaborateResult<Option<()>>,
  324. {
  325. let workspace_id = match self
  326. .workspaces
  327. .iter()
  328. .find(|workspace| workspace.apps.iter().any(|app| app.id == app_id))
  329. {
  330. None => {
  331. tracing::warn!("[FolderPad]: Can't find any app with id: {}", app_id);
  332. return Ok(None);
  333. }
  334. Some(workspace) => workspace.id.clone(),
  335. };
  336. self.with_workspace(&workspace_id, |workspace| {
  337. // It's ok to unwrap because we get the workspace from the app_id.
  338. f(workspace.apps.iter_mut().find(|app| app_id == app.id).unwrap())
  339. })
  340. }
  341. fn with_view<F>(&mut self, belong_to_id: &str, view_id: &str, f: F) -> CollaborateResult<Option<FolderChange>>
  342. where
  343. F: FnOnce(&mut View) -> CollaborateResult<Option<()>>,
  344. {
  345. self.with_app(belong_to_id, |app| {
  346. match app.belongings.iter_mut().find(|view| view_id == view.id) {
  347. None => {
  348. tracing::warn!("[FolderPad]: Can't find any view with id: {}", view_id);
  349. Ok(None)
  350. }
  351. Some(view) => f(view),
  352. }
  353. })
  354. }
  355. }
  356. pub fn default_folder_delta() -> FolderDelta {
  357. PlainTextDeltaBuilder::new()
  358. .insert(r#"{"workspaces":[],"trash":[]}"#)
  359. .build()
  360. }
  361. pub fn initial_folder_delta(folder_pad: &FolderPad) -> CollaborateResult<FolderDelta> {
  362. let json = folder_pad.to_json()?;
  363. let delta = PlainTextDeltaBuilder::new().insert(&json).build();
  364. Ok(delta)
  365. }
  366. impl std::default::Default for FolderPad {
  367. fn default() -> Self {
  368. FolderPad {
  369. workspaces: vec![],
  370. trash: vec![],
  371. delta: default_folder_delta(),
  372. }
  373. }
  374. }
  375. pub struct FolderChange {
  376. pub delta: FolderDelta,
  377. /// md5: the md5 of the FolderPad's delta after applying the change.
  378. pub md5: String,
  379. }
  380. #[cfg(test)]
  381. mod tests {
  382. #![allow(clippy::all)]
  383. use crate::{client_folder::folder_pad::FolderPad, entities::folder_info::FolderDelta};
  384. use chrono::Utc;
  385. use flowy_folder_data_model::entities::{app::App, trash::Trash, view::View, workspace::Workspace};
  386. use lib_ot::core::{OperationTransformable, PlainTextDelta, PlainTextDeltaBuilder};
  387. #[test]
  388. fn folder_add_workspace() {
  389. let (mut folder, initial_delta, _) = test_folder();
  390. let _time = Utc::now();
  391. let mut workspace_1 = Workspace::default();
  392. workspace_1.name = "My first workspace".to_owned();
  393. let delta_1 = folder.create_workspace(workspace_1).unwrap().unwrap().delta;
  394. let mut workspace_2 = Workspace::default();
  395. workspace_2.name = "My second workspace".to_owned();
  396. let delta_2 = folder.create_workspace(workspace_2).unwrap().unwrap().delta;
  397. let folder_from_delta = make_folder_from_delta(initial_delta, vec![delta_1, delta_2]);
  398. assert_eq!(folder, folder_from_delta);
  399. }
  400. #[test]
  401. fn folder_update_workspace() {
  402. let (mut folder, initial_delta, workspace) = test_folder();
  403. assert_folder_equal(
  404. &folder,
  405. &make_folder_from_delta(initial_delta.clone(), vec![]),
  406. r#"{"workspaces":[{"id":"1","name":"😁 my first workspace","desc":"","apps":[],"modified_time":0,"create_time":0}],"trash":[]}"#,
  407. );
  408. let delta = folder
  409. .update_workspace(&workspace.id, Some("☺️ rename workspace️".to_string()), None)
  410. .unwrap()
  411. .unwrap()
  412. .delta;
  413. let folder_from_delta = make_folder_from_delta(initial_delta, vec![delta]);
  414. assert_folder_equal(
  415. &folder,
  416. &folder_from_delta,
  417. r#"{"workspaces":[{"id":"1","name":"☺️ rename workspace️","desc":"","apps":[],"modified_time":0,"create_time":0}],"trash":[]}"#,
  418. );
  419. }
  420. #[test]
  421. fn folder_add_app() {
  422. let (folder, initial_delta, _app) = test_app_folder();
  423. let folder_from_delta = make_folder_from_delta(initial_delta, vec![]);
  424. assert_eq!(folder, folder_from_delta);
  425. assert_folder_equal(
  426. &folder,
  427. &folder_from_delta,
  428. r#"{
  429. "workspaces": [
  430. {
  431. "id": "1",
  432. "name": "😁 my first workspace",
  433. "desc": "",
  434. "apps": [
  435. {
  436. "id": "",
  437. "workspace_id": "1",
  438. "name": "😁 my first app",
  439. "desc": "",
  440. "belongings": [],
  441. "version": 0,
  442. "modified_time": 0,
  443. "create_time": 0
  444. }
  445. ],
  446. "modified_time": 0,
  447. "create_time": 0
  448. }
  449. ],
  450. "trash": []
  451. }"#,
  452. );
  453. }
  454. #[test]
  455. fn folder_update_app() {
  456. let (mut folder, initial_delta, app) = test_app_folder();
  457. let delta = folder
  458. .update_app(&app.id, Some("🤪 rename app".to_owned()), None)
  459. .unwrap()
  460. .unwrap()
  461. .delta;
  462. let new_folder = make_folder_from_delta(initial_delta, vec![delta]);
  463. assert_folder_equal(
  464. &folder,
  465. &new_folder,
  466. r#"{
  467. "workspaces": [
  468. {
  469. "id": "1",
  470. "name": "😁 my first workspace",
  471. "desc": "",
  472. "apps": [
  473. {
  474. "id": "",
  475. "workspace_id": "1",
  476. "name": "🤪 rename app",
  477. "desc": "",
  478. "belongings": [],
  479. "version": 0,
  480. "modified_time": 0,
  481. "create_time": 0
  482. }
  483. ],
  484. "modified_time": 0,
  485. "create_time": 0
  486. }
  487. ],
  488. "trash": []
  489. }"#,
  490. );
  491. }
  492. #[test]
  493. fn folder_delete_app() {
  494. let (mut folder, initial_delta, app) = test_app_folder();
  495. let delta = folder.delete_app(&app.id).unwrap().unwrap().delta;
  496. let new_folder = make_folder_from_delta(initial_delta, vec![delta]);
  497. assert_folder_equal(
  498. &folder,
  499. &new_folder,
  500. r#"{
  501. "workspaces": [
  502. {
  503. "id": "1",
  504. "name": "😁 my first workspace",
  505. "desc": "",
  506. "apps": [],
  507. "modified_time": 0,
  508. "create_time": 0
  509. }
  510. ],
  511. "trash": []
  512. }"#,
  513. );
  514. }
  515. #[test]
  516. fn folder_add_view() {
  517. let (folder, initial_delta, _view) = test_view_folder();
  518. assert_folder_equal(
  519. &folder,
  520. &make_folder_from_delta(initial_delta, vec![]),
  521. r#"
  522. {
  523. "workspaces": [
  524. {
  525. "id": "1",
  526. "name": "😁 my first workspace",
  527. "desc": "",
  528. "apps": [
  529. {
  530. "id": "",
  531. "workspace_id": "1",
  532. "name": "😁 my first app",
  533. "desc": "",
  534. "belongings": [
  535. {
  536. "id": "",
  537. "belong_to_id": "",
  538. "name": "🎃 my first view",
  539. "desc": "",
  540. "view_type": "Blank",
  541. "version": 0,
  542. "belongings": [],
  543. "modified_time": 0,
  544. "create_time": 0
  545. }
  546. ],
  547. "version": 0,
  548. "modified_time": 0,
  549. "create_time": 0
  550. }
  551. ],
  552. "modified_time": 0,
  553. "create_time": 0
  554. }
  555. ],
  556. "trash": []
  557. }"#,
  558. );
  559. }
  560. #[test]
  561. fn folder_update_view() {
  562. let (mut folder, initial_delta, view) = test_view_folder();
  563. let delta = folder
  564. .update_view(&view.id, Some("😦 rename view".to_owned()), None, 123)
  565. .unwrap()
  566. .unwrap()
  567. .delta;
  568. let new_folder = make_folder_from_delta(initial_delta, vec![delta]);
  569. assert_folder_equal(
  570. &folder,
  571. &new_folder,
  572. r#"{
  573. "workspaces": [
  574. {
  575. "id": "1",
  576. "name": "😁 my first workspace",
  577. "desc": "",
  578. "apps": [
  579. {
  580. "id": "",
  581. "workspace_id": "1",
  582. "name": "😁 my first app",
  583. "desc": "",
  584. "belongings": [
  585. {
  586. "id": "",
  587. "belong_to_id": "",
  588. "name": "😦 rename view",
  589. "desc": "",
  590. "view_type": "Blank",
  591. "version": 0,
  592. "belongings": [],
  593. "modified_time": 123,
  594. "create_time": 0
  595. }
  596. ],
  597. "version": 0,
  598. "modified_time": 0,
  599. "create_time": 0
  600. }
  601. ],
  602. "modified_time": 0,
  603. "create_time": 0
  604. }
  605. ],
  606. "trash": []
  607. }"#,
  608. );
  609. }
  610. #[test]
  611. fn folder_delete_view() {
  612. let (mut folder, initial_delta, view) = test_view_folder();
  613. let delta = folder.delete_view(&view.id).unwrap().unwrap().delta;
  614. let new_folder = make_folder_from_delta(initial_delta, vec![delta]);
  615. assert_folder_equal(
  616. &folder,
  617. &new_folder,
  618. r#"{
  619. "workspaces": [
  620. {
  621. "id": "1",
  622. "name": "😁 my first workspace",
  623. "desc": "",
  624. "apps": [
  625. {
  626. "id": "",
  627. "workspace_id": "1",
  628. "name": "😁 my first app",
  629. "desc": "",
  630. "belongings": [],
  631. "version": 0,
  632. "modified_time": 0,
  633. "create_time": 0
  634. }
  635. ],
  636. "modified_time": 0,
  637. "create_time": 0
  638. }
  639. ],
  640. "trash": []
  641. }"#,
  642. );
  643. }
  644. #[test]
  645. fn folder_add_trash() {
  646. let (folder, initial_delta, _trash) = test_trash();
  647. assert_folder_equal(
  648. &folder,
  649. &make_folder_from_delta(initial_delta, vec![]),
  650. r#"{
  651. "workspaces": [],
  652. "trash": [
  653. {
  654. "id": "1",
  655. "name": "🚽 my first trash",
  656. "modified_time": 0,
  657. "create_time": 0,
  658. "ty": "Unknown"
  659. }
  660. ]
  661. }
  662. "#,
  663. );
  664. }
  665. #[test]
  666. fn folder_delete_trash() {
  667. let (mut folder, initial_delta, trash) = test_trash();
  668. let delta = folder.delete_trash(Some(vec![trash.id])).unwrap().unwrap().delta;
  669. assert_folder_equal(
  670. &folder,
  671. &make_folder_from_delta(initial_delta, vec![delta]),
  672. r#"{
  673. "workspaces": [],
  674. "trash": []
  675. }
  676. "#,
  677. );
  678. }
  679. fn test_folder() -> (FolderPad, FolderDelta, Workspace) {
  680. let mut folder = FolderPad::default();
  681. let folder_json = serde_json::to_string(&folder).unwrap();
  682. let mut delta = PlainTextDeltaBuilder::new().insert(&folder_json).build();
  683. let mut workspace = Workspace::default();
  684. workspace.name = "😁 my first workspace".to_owned();
  685. workspace.id = "1".to_owned();
  686. delta = delta
  687. .compose(&folder.create_workspace(workspace.clone()).unwrap().unwrap().delta)
  688. .unwrap();
  689. (folder, delta, workspace)
  690. }
  691. fn test_app_folder() -> (FolderPad, FolderDelta, App) {
  692. let (mut folder, mut initial_delta, workspace) = test_folder();
  693. let mut app = App::default();
  694. app.workspace_id = workspace.id;
  695. app.name = "😁 my first app".to_owned();
  696. initial_delta = initial_delta
  697. .compose(&folder.create_app(app.clone()).unwrap().unwrap().delta)
  698. .unwrap();
  699. (folder, initial_delta, app)
  700. }
  701. fn test_view_folder() -> (FolderPad, FolderDelta, View) {
  702. let (mut folder, mut initial_delta, app) = test_app_folder();
  703. let mut view = View::default();
  704. view.belong_to_id = app.id.clone();
  705. view.name = "🎃 my first view".to_owned();
  706. initial_delta = initial_delta
  707. .compose(&folder.create_view(view.clone()).unwrap().unwrap().delta)
  708. .unwrap();
  709. (folder, initial_delta, view)
  710. }
  711. fn test_trash() -> (FolderPad, FolderDelta, Trash) {
  712. let mut folder = FolderPad::default();
  713. let folder_json = serde_json::to_string(&folder).unwrap();
  714. let mut delta = PlainTextDeltaBuilder::new().insert(&folder_json).build();
  715. let mut trash = Trash::default();
  716. trash.name = "🚽 my first trash".to_owned();
  717. trash.id = "1".to_owned();
  718. delta = delta
  719. .compose(&folder.create_trash(vec![trash.clone()]).unwrap().unwrap().delta)
  720. .unwrap();
  721. (folder, delta, trash)
  722. }
  723. fn make_folder_from_delta(mut initial_delta: FolderDelta, deltas: Vec<PlainTextDelta>) -> FolderPad {
  724. for delta in deltas {
  725. initial_delta = initial_delta.compose(&delta).unwrap();
  726. }
  727. FolderPad::from_delta(initial_delta).unwrap()
  728. }
  729. fn assert_folder_equal(old: &FolderPad, new: &FolderPad, expected: &str) {
  730. assert_eq!(old, new);
  731. let json1 = old.to_json().unwrap();
  732. let json2 = new.to_json().unwrap();
  733. let expect_folder: FolderPad = serde_json::from_str(expected).unwrap();
  734. assert_eq!(json1, expect_folder.to_json().unwrap());
  735. assert_eq!(json1, json2);
  736. }
  737. }