vtcode-core 0.104.0

Core library for VT Code - a Rust-based terminal coding agent
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

use super::PatchError;
use super::lifecycle::OperationState;

pub(super) struct OperationJournal {
    applied: Vec<OperationState>,
}

impl OperationJournal {
    pub(super) fn new() -> Self {
        Self {
            applied: Vec::new(),
        }
    }

    pub(super) fn record(&mut self, state: OperationState) {
        self.applied.push(state);
    }

    pub(super) async fn rollback_all(&mut self) -> Result<(), PatchError> {
        let mut rollback_error = None;

        while let Some(state) = self.applied.pop() {
            if let Err(err) = state.rollback().await {
                rollback_error.get_or_insert(err);
            }
        }

        if let Some(err) = rollback_error {
            Err(err)
        } else {
            Ok(())
        }
    }

    pub(super) async fn commit_all(mut self) -> Result<(), PatchError> {
        while let Some(state) = self.applied.pop() {
            state.commit().await?;
        }
        Ok(())
    }
}