codex/codex-rs/utils/cli/src/format_env_display.rs

use std::collections::HashMap;

pub fn format_env_display(env: Option<&HashMap<String, String>>, env_vars: &[String]) -> String {
    let mut parts: Vec<String> = Vec::new();

    if let Some(map) = env {
        let mut pairs: Vec<_> = map.iter().collect();
        pairs.sort_by(|(a, _), (b, _)| a.cmp(b));
        parts.extend(pairs.into_iter().map(|(key, _)| format!("{key}=*****")));
    }

    if !env_vars.is_empty() {
        parts.extend(env_vars.iter().map(|var| format!("{var}=*****")));
    }

    if parts.is_empty() {
        "-".to_string()
    } else {
        parts.join(", ")
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn returns_dash_when_empty() {
        assert_eq!(format_env_display(None, &[]), "-");

        let empty_map = HashMap::new();
        assert_eq!(format_env_display(Some(&empty_map), &[]), "-");
    }

    #[test]
    fn formats_sorted_env_pairs() {
        let mut env = HashMap::new();
        env.insert("B".to_string(), "two".to_string());
        env.insert("A".to_string(), "one".to_string());

        assert_eq!(format_env_display(Some(&env), &[]), "A=*****, B=*****");
    }

    #[test]
    fn formats_env_vars_with_dollar_prefix() {
        let vars = vec!["TOKEN".to_string(), "PATH".to_string()];

        assert_eq!(format_env_display(None, &vars), "TOKEN=*****, PATH=*****");
    }

    #[test]
    fn combines_env_pairs_and_vars() {
        let mut env = HashMap::new();
        env.insert("HOME".to_string(), "/tmp".to_string());
        let vars = vec!["TOKEN".to_string()];

        assert_eq!(
            format_env_display(Some(&env), &vars),
            "HOME=*****, TOKEN=*****"
        );
    }
}