fnm/tests/shellcode/eval_fnm_env.rs

use super::expression::Expression;
use super::shell::{Bash, Fish, PowerShell, WinCmd, Zsh};
use std::fmt::Write;

#[derive(Debug, Default)]
pub(crate) struct EvalFnmEnv {
    use_on_cd: bool,
    log_level: Option<&'static str>,
}

impl EvalFnmEnv {
    pub(crate) fn use_on_cd(self, use_on_cd: bool) -> Self {
        Self { use_on_cd, ..self }
    }

    pub(crate) fn log_level(self, log_level: Option<&'static str>) -> Self {
        Self { log_level, ..self }
    }
}

impl std::fmt::Display for EvalFnmEnv {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "fnm")?;
        if let Some(log_level) = &self.log_level {
            write!(f, " --log-level='{}'", log_level)?;
        }
        write!(f, " env")?;
        if self.use_on_cd {
            write!(f, " --use-on-cd")?;
        }
        Ok(())
    }
}

impl Expression<WinCmd> for EvalFnmEnv {
    fn write_shell(&self, writer: &mut impl Write) -> std::fmt::Result {
        write!(writer, r#"FOR /f "tokens=*" %i IN ('{}') DO CALL %i"#, self)
    }
}

impl Expression<PowerShell> for EvalFnmEnv {
    fn write_shell(&self, writer: &mut impl Write) -> std::fmt::Result {
        write!(writer, r#"{} | Out-String | Invoke-Expression"#, self)
    }
}

impl Expression<Zsh> for EvalFnmEnv {
    fn write_shell(&self, writer: &mut impl Write) -> std::fmt::Result {
        write!(writer, r#"eval "$({})""#, self)
    }
}

impl Expression<Bash> for EvalFnmEnv {
    fn write_shell(&self, writer: &mut impl Write) -> std::fmt::Result {
        write!(writer, r#"eval "$({})""#, self)
    }
}

impl Expression<Fish> for EvalFnmEnv {
    fn write_shell(&self, writer: &mut impl Write) -> std::fmt::Result {
        write!(writer, r#"{} | source"#, self)
    }
}