src/poem/anthology/export.rs


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

use crate::poem::Verse;
use std::env;

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
}

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
}