rbam 0.1.1

A blazing-fast Rust Bash Alias Manager (create, list, and remove shell aliases).
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

use std::collections::HashMap;
use std::fs;

pub fn read_aliases() -> HashMap<String, String> {
    let mut aliases = HashMap::new();
    let mut path = dirs::home_dir().expect("No home dir");
    path.push(".bash_aliases");

    let contents = match fs::read_to_string(&path) {
        Ok(c) => c,
        Err(_) => {
            eprintln!("Cannot read .bash_aliases");
            return aliases;
        }
    };

    for line in contents.lines() {
        let trimmed = line.trim();
        if trimmed.starts_with("alias ")
            && let Some((name, cmd)) = parse_alias_line(trimmed)
        {
            aliases.insert(name, cmd);
        }
    }

    aliases
}

pub fn parse_alias_line(line: &str) -> Option<(String, String)> {
    let without_alias = line.strip_prefix("alias ")?;
    let parts: Vec<&str> = without_alias.splitn(2, '=').collect();

    if parts.len() != 2 {
        return None;
    }

    let name = parts[0].to_string();
    let cmd = parts[1].trim_matches('\'').trim_matches('"').to_string();

    Some((name, cmd))
}

pub fn get_aliases_path() -> std::path::PathBuf {
    let mut path = dirs::home_dir().expect("No home dir");
    path.push(".bash_aliases");
    path
}

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

    #[test]
    fn test_parse_valid_alias_single_quotes() {
        let line = "alias cdn='ssh user@host'";
        let result = parse_alias_line(line);
        assert_eq!(
            result,
            Some(("cdn".to_string(), "ssh user@host".to_string()))
        );
    }

    #[test]
    fn test_parse_valid_alias_double_quotes() {
        let line = "alias prod=\"ssh deploy@prod.com\"";
        let result = parse_alias_line(line);
        assert_eq!(
            result,
            Some(("prod".to_string(), "ssh deploy@prod.com".to_string()))
        );
    }

    #[test]
    fn test_parse_alias_with_port() {
        let line = "alias staging='ssh -p 2222 user@host'";
        let result = parse_alias_line(line);
        assert_eq!(
            result,
            Some(("staging".to_string(), "ssh -p 2222 user@host".to_string()))
        );
    }

    #[test]
    fn test_parse_alias_with_equals_in_command() {
        let line = "alias test='echo KEY=VALUE'";
        let result = parse_alias_line(line);
        assert_eq!(
            result,
            Some(("test".to_string(), "echo KEY=VALUE".to_string()))
        );
    }

    #[test]
    fn test_parse_invalid_no_alias_prefix() {
        let line = "cdn='ssh user@host'";
        let result = parse_alias_line(line);
        assert_eq!(result, None);
    }

    #[test]
    fn test_parse_invalid_no_equals() {
        let line = "alias cdn";
        let result = parse_alias_line(line);
        assert_eq!(result, None);
    }

    #[test]
    fn test_parse_invalid_empty_name() {
        let line = "alias ='ssh user@host'";
        let result = parse_alias_line(line);
        assert!(result.is_some()); // Will parse but name is empty
        if let Some((name, _)) = result {
            assert_eq!(name, "");
        }
    }

    #[test]
    fn test_parse_alias_with_spaces() {
        let line = "alias my-server = 'ssh root@server.com'";
        let result = parse_alias_line(line);
        assert!(result.is_some());
    }
}