#[derive(PartialEq)]
pub(crate) enum Shell {
    Sh,
    Bash,
    Fish,
    Zsh,
    Xonsh,
    Nu,
    Powershell,

    Unknown,
}

impl std::fmt::Display for Shell {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let shell = match self {
            Self::Bash => "bash",
            Self::Fish => "fish",
            Self::Zsh => "zsh",
            Self::Nu => "nu",
            Self::Xonsh => "xonsh",
            Self::Sh => "sh",
            Self::Powershell => "powershell",

            Self::Unknown => "unknown",
        };

        write!(f, "{shell}")
    }
}

impl Shell {
    pub(crate) fn from_env() -> Self {
        std::env::var("ATUIN_SHELL").map_or(Self::Unknown, |shell| {
            Self::from_string(shell.trim().to_lowercase().as_str())
        })
    }

    pub(crate) fn from_string(name: &str) -> Self {
        match name {
            "bash" => Self::Bash,
            "fish" => Self::Fish,
            "zsh" => Self::Zsh,
            "xonsh" => Self::Xonsh,
            "nu" => Self::Nu,
            "sh" => Self::Sh,
            "powershell" => Self::Powershell,

            _ => Self::Unknown,
        }
    }
}