use std::path::Path;

#[derive(PartialEq, Eq, Clone, Copy)]
pub enum Extension {
    #[allow(clippy::upper_case_acronyms)]
    CSV,
    #[allow(clippy::upper_case_acronyms)]
    JSON,
}

impl Extension {
    pub fn from(path: &Path) -> Option<Self> {
        if let Some(extension) = path.extension() {
            let extension = extension.to_str().unwrap();

            if extension == Self::CSV.to_str() {
                Some(Self::CSV)
            } else if extension == Self::JSON.to_str() {
                Some(Self::JSON)
            } else {
                None
            }
        } else {
            None
        }
    }

    pub fn to_str(&self) -> &str {
        match self {
            Extension::CSV => "csv",
            Extension::JSON => "json",
        }
    }
}