text-document-common 1.4.0

Shared entities, database, events, and undo/redo infrastructure for text-document
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

// Generated by Qleany v1.4.8 from common_da_use_cases_remove.tera

use super::traits::WriteUoWFactory;
use crate::snapshot::EntityTreeSnapshot;
use crate::types::EntityId;
use crate::undo_redo::UndoRedoCommand;
use anyhow::{Ok, Result};
use std::any::Any;

// ---------------------------------------------------------------------------
// Non-undoable remove
// ---------------------------------------------------------------------------

pub struct RemoveUseCase<F: WriteUoWFactory> {
    uow_factory: F,
}

impl<F: WriteUoWFactory> RemoveUseCase<F> {
    pub fn new(uow_factory: F) -> Self {
        RemoveUseCase { uow_factory }
    }

    pub fn execute(&mut self, id: &EntityId) -> Result<()> {
        self.execute_multi(&[*id])
    }

    pub fn execute_multi(&mut self, ids: &[EntityId]) -> Result<()> {
        let mut uow = self.uow_factory.create();
        uow.begin_transaction()?;
        for id in ids {
            if uow.get(id)?.is_none() {
                return Err(anyhow::anyhow!("Entity with id {} does not exist", id));
            }
        }
        uow.remove_multi(ids)?;
        uow.commit()?;
        Ok(())
    }
}

// ---------------------------------------------------------------------------
// Undoable remove
// ---------------------------------------------------------------------------

pub struct UndoableRemoveUseCase<F: WriteUoWFactory> {
    uow_factory: F,
    entity_tree_snapshot: Option<EntityTreeSnapshot>,
    original_ids: Option<Vec<EntityId>>,
}

impl<F: WriteUoWFactory> UndoableRemoveUseCase<F> {
    pub fn new(uow_factory: F) -> Self {
        UndoableRemoveUseCase {
            uow_factory,
            entity_tree_snapshot: None,
            original_ids: None,
        }
    }

    pub fn execute(&mut self, id: &EntityId) -> Result<()> {
        self.execute_multi(&[*id])
    }

    pub fn execute_multi(&mut self, ids: &[EntityId]) -> Result<()> {
        let mut uow = self.uow_factory.create();
        uow.begin_transaction()?;
        for id in ids {
            if uow.get(id)?.is_none() {
                return Err(anyhow::anyhow!("Entity with id {} does not exist", id));
            }
        }
        self.entity_tree_snapshot = Some(uow.snapshot(ids)?);
        uow.remove_multi(ids)?;
        uow.commit()?;
        self.original_ids = Some(ids.to_vec());
        Ok(())
    }
}

impl<F: WriteUoWFactory + Send + 'static> UndoRedoCommand for UndoableRemoveUseCase<F>
where
    F::Entity: 'static,
{
    fn undo(&mut self) -> Result<()> {
        if let Some(ref snap) = self.entity_tree_snapshot {
            let mut uow = self.uow_factory.create();
            uow.begin_transaction()?;
            uow.restore(snap)?;
            uow.commit()?;
        }
        Ok(())
    }

    fn redo(&mut self) -> Result<()> {
        if let Some(ref ids) = self.original_ids {
            let mut uow = self.uow_factory.create();
            uow.begin_transaction()?;
            self.entity_tree_snapshot = Some(uow.snapshot(ids)?);
            uow.remove_multi(ids)?;
            uow.commit()?;
        }
        Ok(())
    }

    fn as_any(&self) -> &dyn Any {
        self
    }
}