cf-modkit-db 0.7.2

//! `SQLite` PRAGMA parameter handling with typed enums.

use std::collections::HashMap;

/// `SQLite` journal mode options.
#[derive(Clone, Debug, PartialEq, Eq)]
pub enum JournalMode {
    Delete,
    Wal,
    Memory,
    Truncate,
    Persist,
    Off,
}

impl JournalMode {
    /// Parse from string (case-insensitive).
    fn from_str(s: &str) -> Option<Self> {
        match s.to_uppercase().as_str() {
            "DELETE" => Some(JournalMode::Delete),
            "WAL" => Some(JournalMode::Wal),
            "MEMORY" => Some(JournalMode::Memory),
            "TRUNCATE" => Some(JournalMode::Truncate),
            "PERSIST" => Some(JournalMode::Persist),
            "OFF" => Some(JournalMode::Off),
            _ => None,
        }
    }
}

/// `SQLite` synchronous mode options.
#[derive(Clone, Debug, PartialEq, Eq)]
pub enum SyncMode {
    Off,
    Normal,
    Full,
    Extra,
}

impl SyncMode {
    /// Parse from string (case-insensitive).
    fn from_str(s: &str) -> Option<Self> {
        match s.to_uppercase().as_str() {
            "OFF" => Some(SyncMode::Off),
            "NORMAL" => Some(SyncMode::Normal),
            "FULL" => Some(SyncMode::Full),
            "EXTRA" => Some(SyncMode::Extra),
            _ => None,
        }
    }
}

/// Parsed `SQLite` PRAGMA parameters.
#[derive(Clone, Debug, Default, PartialEq, Eq)]
pub struct Pragmas {
    pub journal_mode: Option<JournalMode>,
    pub synchronous: Option<SyncMode>,
    pub busy_timeout_ms: Option<i64>,
    /// Compatibility: support legacy `wal=true|false|1|0`
    pub wal_toggle: Option<bool>,
}

impl Pragmas {
    /// Parse PRAGMA parameters from a key-value map.
    pub(crate) fn from_pairs(pairs: &HashMap<String, String>) -> Self {
        let mut pragmas = Pragmas::default();

        for (key, value) in pairs {
            match key.to_lowercase().as_str() {
                "journal_mode" => {
                    pragmas.journal_mode = Self::parse_journal_mode(value);
                }
                "synchronous" => {
                    pragmas.synchronous = Self::parse_synchronous(value);
                }
                "busy_timeout" => {
                    pragmas.busy_timeout_ms = Self::parse_busy_timeout(value);
                }
                "wal" => {
                    pragmas.wal_toggle = Self::parse_wal_toggle(value);
                }
                _ => {
                    tracing::debug!("Unknown SQLite PRAGMA parameter: {}", key);
                }
            }
        }

        pragmas
    }

    /// Parse `journal_mode` PRAGMA value.
    fn parse_journal_mode(value: &str) -> Option<JournalMode> {
        if let Some(mode) = JournalMode::from_str(value) {
            Some(mode)
        } else {
            tracing::warn!("Invalid 'journal_mode' PRAGMA value '{}', ignoring", value);
            None
        }
    }

    /// Parse synchronous PRAGMA value.
    fn parse_synchronous(value: &str) -> Option<SyncMode> {
        if let Some(mode) = SyncMode::from_str(value) {
            Some(mode)
        } else {
            tracing::warn!("Invalid 'synchronous' PRAGMA value '{}', ignoring", value);
            None
        }
    }

    /// Parse `busy_timeout` PRAGMA value.
    fn parse_busy_timeout(value: &str) -> Option<i64> {
        match value.parse::<i64>() {
            Ok(timeout) if timeout >= 0 => Some(timeout),
            _ => {
                tracing::warn!("Invalid 'busy_timeout' PRAGMA value '{}', ignoring", value);
                None
            }
        }
    }

    /// Parse wal PRAGMA value (legacy compatibility).
    fn parse_wal_toggle(value: &str) -> Option<bool> {
        match value.to_lowercase().as_str() {
            "true" | "1" => Some(true),
            "false" | "0" => Some(false),
            _ => {
                tracing::warn!("Invalid 'wal' PRAGMA value '{}', ignoring", value);
                None
            }
        }
    }
}

#[cfg(test)]
#[cfg_attr(coverage_nightly, coverage(off))]
mod tests {
    use super::*;

    #[test]
    fn test_journal_mode_parsing() {
        assert_eq!(JournalMode::from_str("DELETE"), Some(JournalMode::Delete));
        assert_eq!(JournalMode::from_str("wal"), Some(JournalMode::Wal));
        assert_eq!(JournalMode::from_str("MEMORY"), Some(JournalMode::Memory));
        assert_eq!(JournalMode::from_str("invalid"), None);
    }

    #[test]
    fn test_sync_mode_parsing() {
        assert_eq!(SyncMode::from_str("OFF"), Some(SyncMode::Off));
        assert_eq!(SyncMode::from_str("normal"), Some(SyncMode::Normal));
        assert_eq!(SyncMode::from_str("FULL"), Some(SyncMode::Full));
        assert_eq!(SyncMode::from_str("invalid"), None);
    }

    #[test]
    fn test_pragmas_from_pairs() {
        let mut pairs = HashMap::new();
        pairs.insert("journal_mode".to_owned(), "WAL".to_owned());
        pairs.insert("synchronous".to_owned(), "NORMAL".to_owned());
        pairs.insert("busy_timeout".to_owned(), "5000".to_owned());
        pairs.insert("wal".to_owned(), "true".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);

        assert_eq!(pragmas.journal_mode, Some(JournalMode::Wal));
        assert_eq!(pragmas.synchronous, Some(SyncMode::Normal));
        assert_eq!(pragmas.busy_timeout_ms, Some(5000));
        assert_eq!(pragmas.wal_toggle, Some(true));
    }

    #[test]
    fn test_pragmas_invalid_values() {
        let mut pairs = HashMap::new();
        pairs.insert("journal_mode".to_owned(), "INVALID".to_owned());
        pairs.insert("synchronous".to_owned(), "INVALID".to_owned());
        pairs.insert("busy_timeout".to_owned(), "-1".to_owned());
        pairs.insert("wal".to_owned(), "maybe".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);

        assert_eq!(pragmas.journal_mode, None);
        assert_eq!(pragmas.synchronous, None);
        assert_eq!(pragmas.busy_timeout_ms, None);
        assert_eq!(pragmas.wal_toggle, None);
    }

    #[test]
    fn test_pragmas_case_insensitive() {
        let mut pairs = HashMap::new();
        pairs.insert("JOURNAL_MODE".to_owned(), "delete".to_owned());
        pairs.insert("SYNCHRONOUS".to_owned(), "off".to_owned());
        pairs.insert("WAL".to_owned(), "FALSE".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);

        assert_eq!(pragmas.journal_mode, Some(JournalMode::Delete));
        assert_eq!(pragmas.synchronous, Some(SyncMode::Off));
        assert_eq!(pragmas.wal_toggle, Some(false));
    }

    #[test]
    fn test_pragmas_unknown_keys() {
        let mut pairs = HashMap::new();
        pairs.insert("unknown_param".to_owned(), "value".to_owned());
        pairs.insert("journal_mode".to_owned(), "WAL".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);

        assert_eq!(pragmas.journal_mode, Some(JournalMode::Wal));
        // Unknown params should be ignored without error
    }

    #[test]
    fn test_pragmas_wal_numeric_toggle() {
        let mut pairs = HashMap::new();
        pairs.insert("wal".to_owned(), "1".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);
        assert_eq!(pragmas.wal_toggle, Some(true));

        let mut pairs = HashMap::new();
        pairs.insert("wal".to_owned(), "0".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);
        assert_eq!(pragmas.wal_toggle, Some(false));
    }

    #[test]
    fn test_pragmas_busy_timeout_zero() {
        let mut pairs = HashMap::new();
        pairs.insert("busy_timeout".to_owned(), "0".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);
        assert_eq!(pragmas.busy_timeout_ms, Some(0));
    }

    #[test]
    fn test_pragmas_busy_timeout_invalid() {
        let mut pairs = HashMap::new();
        pairs.insert("busy_timeout".to_owned(), "not_a_number".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);
        assert_eq!(pragmas.busy_timeout_ms, None);
    }

    #[test]
    fn test_pragmas_partial_map() {
        let mut pairs = HashMap::new();
        pairs.insert("synchronous".to_owned(), "FULL".to_owned());

        let pragmas = Pragmas::from_pairs(&pairs);
        assert_eq!(pragmas.journal_mode, None);
        assert_eq!(pragmas.synchronous, Some(SyncMode::Full));
        assert_eq!(pragmas.busy_timeout_ms, None);
        assert_eq!(pragmas.wal_toggle, None);
    }

    #[test]
    fn test_pragmas_empty_map() {
        let pairs = HashMap::new();

        let pragmas = Pragmas::from_pairs(&pairs);
        assert_eq!(pragmas.journal_mode, None);
        assert_eq!(pragmas.synchronous, None);
        assert_eq!(pragmas.busy_timeout_ms, None);
        assert_eq!(pragmas.wal_toggle, None);
    }
}