turso/sql_generation/model/query/transaction.rs

use std::fmt::Display;

use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum Begin {
    Deferred,
    Immediate,
    Concurrent,
}

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

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

impl Display for Begin {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let keyword = match self {
            Begin::Deferred => "",
            Begin::Immediate => "IMMEDIATE",
            Begin::Concurrent => "CONCURRENT",
        };
        write!(f, "BEGIN {keyword}")
    }
}

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")
    }
}