kraven 0.3.0

Manage named environment variable profiles
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use anyhow::Result;

use crate::config::resolve_profile_path;
use crate::profile::Profile;

pub fn run(profile_name: &str, mask_values: bool) -> Result<()> {
    let profile_path = resolve_profile_path(profile_name)?;
    let profile = Profile::load(profile_name, &profile_path)?;

    if profile.vars.is_empty() {
        println!("Profile '{profile_name}' is empty.");
        return Ok(());
    }

    for (key, value) in &profile.vars {
        if mask_values {
            println!("{key}={}", mask_value(value));
        } else if value.contains(' ') || value.contains('"') || value.contains('\'') {
            let escaped = value.replace('"', "\\\"");
            println!("{key}=\"{escaped}\"");
        } else {
            println!("{key}={value}");
        }
    }

    Ok(())
}

fn mask_value(value: &str) -> String {
    let char_count = value.chars().count();

    if char_count <= 4 {
        return "*".repeat(char_count);
    }

    // Show first 2 and last 2 characters
    let start: String = value.chars().take(2).collect();
    let end: String = value.chars().skip(char_count - 2).collect();
    let middle = "*".repeat(char_count - 4);
    format!("{start}{middle}{end}")
}

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

    #[test]
    fn test_mask_empty() {
        assert_eq!(mask_value(""), "");
    }

    #[test]
    fn test_mask_short() {
        assert_eq!(mask_value("ab"), "**");
        assert_eq!(mask_value("abcd"), "****");
    }

    #[test]
    fn test_mask_long() {
        assert_eq!(mask_value("abcdef"), "ab**ef");
        assert_eq!(mask_value("secretkey123"), "se********23");
    }

    #[test]
    fn test_mask_unicode() {
        // This would panic with byte indexing
        assert_eq!(mask_value("cafén"), "ca*én");
        // "🔑secret🔒" is 8 chars: 🔑 s e c r e t 🔒
        // First 2: 🔑 s, Last 2: t 🔒, Middle 4: ****
        assert_eq!(mask_value("🔑secret🔒"), "🔑s****t🔒");
    }
}