turso/simulator/model/query/transaction.rs

use std::fmt::Display;

use serde::{Deserialize, Serialize};

use crate::{model::table::SimValue, runner::env::SimulatorEnv};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub(crate) struct Begin {
    pub(crate) immediate: bool,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub(crate) struct Commit;

#[derive(Debug, Clone, Serialize, Deserialize)]
pub(crate) struct Rollback;

impl Begin {
    pub(crate) fn shadow(&self, env: &mut SimulatorEnv) -> Vec<Vec<SimValue>> {
        env.tables_snapshot = Some(env.tables.clone());
        vec![]
    }
}

impl Commit {
    pub(crate) fn shadow(&self, env: &mut SimulatorEnv) -> Vec<Vec<SimValue>> {
        env.tables_snapshot = None;
        vec![]
    }
}

impl Rollback {
    pub(crate) fn shadow(&self, env: &mut SimulatorEnv) -> Vec<Vec<SimValue>> {
        if let Some(tables) = env.tables_snapshot.take() {
            env.tables = tables;
        }
        vec![]
    }
}

impl Display for Begin {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "BEGIN {}", if self.immediate { "IMMEDIATE" } else { "" })
    }
}

impl Display for Commit {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "COMMIT")
    }
}

impl Display for Rollback {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "ROLLBACK")
    }
}