summaryrefslogtreecommitdiffstats
path: root/src/poem/anthology/alias.rs
blob: 0746b596109472dc9cef9353182afdd00a722649 (plain)
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
use std::collections::BTreeMap;
use std::collections::HashMap;
use std::os::unix::process::ExitStatusExt;
use std::process::{ExitStatus, Output};

/// alias
///
/// The builtin `alias` command. Used to monikers for other verbs, or entire
/// verses.
///
/// # Shell Example
/// ```sh
/// alias vim=nvim
/// ```
pub fn incant(
    clause: &Option<Vec<String>>,
    uout: bool,
    aliases: &mut HashMap<String, String>,
) -> Output {
    let status = 0;
    let mut out: Vec<u8> = Vec::new();
    let err: Vec<u8> = Vec::new();
    match clause {
        Some(clause) => {
            for stanza in clause {
                let (key, val) = match stanza.split_once("=") {
                    Some((key, val)) => (key, val),
                    None => continue,
                };
                aliases.insert(String::from(key), String::from(val));
            }
        }
        None => {
            let mut lines = Vec::new();
            let sorted: BTreeMap<_, _> = aliases.iter().collect();
            for (key, val) in sorted {
                let line = if key.contains(' ') && val.contains(' ') {
                    format!("'{}'='{}'", key, val)
                } else if key.contains(' ') {
                    format!("'{}'={}", key, val)
                } else if val.contains(' ') {
                    format!("{}='{}'", key, val)
                } else if val.is_empty() {
                    format!("{}=''", key)
                } else {
                    format!("{}={}", key, val)
                };
                lines.push(line);
            }

            if uout {
                out.append(&mut format!("{}\n", lines.join("\n")).as_bytes().to_vec());
            } else {
                println!("{}", lines.join("\n"));
            }
        }
    }

    Output {
        status: ExitStatus::from_raw(status),
        stdout: out,
        stderr: err,
    }
}

/// unalias
///
/// The builtin `unalias` command. Used to remove shell monikers, since `alias`
/// may be called with an empty string as the value.
///
/// # Shell Example
/// ```sh
/// unalias vim
/// ```
pub fn unincant(
    clause: &Option<Vec<String>>,
    uerr: bool,
    aliases: &mut HashMap<String, String>,
) -> Output {
    let out: Vec<u8> = Vec::new();
    let mut err: Vec<u8> = Vec::new();

    let status = match clause {
        Some(clause) => {
            for stanza in clause {
                aliases.remove(stanza);
            }
            0
        }
        None => {
            if uerr {
                err.append(&mut "unalias: not enough arguments".as_bytes().to_vec());
            } else {
                eprintln!("unalias: not enough arguments");
            }
            1
        }
    };

    Output {
        status: ExitStatus::from_raw(status),
        stdout: out,
        stderr: err,
    }
}