summaryrefslogtreecommitdiffstats
path: root/src/poem/anthology/export.rs
blob: dfbaf814d1d3df5f1f59b42e4b1d8e4ad392e8cc (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
use crate::poem::Verse;
use std::env;

/// export
///
/// The builtin `export` command. Used to set global environment variables for
/// the current instance of the shell.
///
/// # Aliases
/// * export
/// * set
///
/// # Shell Examples
/// ```sh
/// export FOO=BAR
/// ```
pub fn incant(verse: &Verse) -> i32 {
    match verse.clause() {
        Some(clause) => {
            for stanza in clause {
                let (key, val) = match stanza.split_once("=") {
                    Some((key, val)) => (key, val),
                    None => continue,
                };
                env::set_var(key, val);
            }
        }
        None => {
            for (key, val) in env::vars() {
                println!("{}={}", key, val);
            }
        }
    }

    0
}

/// unset
///
/// The builtin `unset` command. Used to remove global environment variables
/// from the current instance of the shell, since `export` may be called with
/// an empty string as the value.
///
/// # Shell Examples
/// ```sh
/// unset FOO
/// ```
pub fn unincant(verse: &Verse) -> i32 {
    match verse.clause() {
        Some(clause) => {
            for stanza in clause {
                env::remove_var(stanza);
            }
        }
        None => {
            eprintln!("unset: not enough arguments");
            return 1;
        }
    }
    0
}