snapbox 0.6.19

Snapshot testing toolbox
Documentation
/// Describes the structure of [`Data`][crate::Data]
#[derive(Clone, Debug, PartialEq, Eq, Copy, Hash, Default)]
#[non_exhaustive]
pub enum DataFormat {
    /// Processing of the [`Data`][crate::Data] failed
    Error,
    /// Non-textual, opaque data
    Binary,
    #[default]
    Text,
    #[cfg(feature = "json")]
    Json,
    /// Streamed JSON output according to <https://jsonlines.org/>
    #[cfg(feature = "json")]
    JsonLines,
    /// [ANSI escape codes](https://en.wikipedia.org/wiki/ANSI_escape_code#DOS_and_Windows)
    /// rendered as [svg](https://docs.rs/anstyle-svg)
    #[cfg(feature = "term-svg")]
    TermSvg,
}

impl DataFormat {
    /// Assumed file extension for the format
    pub fn ext(self) -> &'static str {
        match self {
            Self::Error => "txt",
            Self::Binary => "bin",
            Self::Text => "txt",
            #[cfg(feature = "json")]
            Self::Json => "json",
            #[cfg(feature = "json")]
            Self::JsonLines => "jsonl",
            #[cfg(feature = "term-svg")]
            Self::TermSvg => "term.svg",
        }
    }
}

impl From<&std::path::Path> for DataFormat {
    fn from(path: &std::path::Path) -> Self {
        let file_name = path
            .file_name()
            .and_then(|e| e.to_str())
            .unwrap_or_default();
        let (file_stem, mut ext) = file_name.split_once('.').unwrap_or((file_name, ""));
        if file_stem.is_empty() {
            (_, ext) = file_stem.split_once('.').unwrap_or((file_name, ""));
        }
        match ext {
            #[cfg(feature = "json")]
            "json" => DataFormat::Json,
            #[cfg(feature = "json")]
            "jsonl" => DataFormat::JsonLines,
            #[cfg(feature = "term-svg")]
            "term.svg" => Self::TermSvg,
            _ => DataFormat::Text,
        }
    }
}