summaryrefslogtreecommitdiffstats
path: root/src/poem/anthology/alias.rs
blob: 6a8e7394e708784a5249762399b24da2799deeda (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
use crate::poem::Verse;
use std::collections::HashMap;

pub fn incant(verse: &Verse, aliases: &mut HashMap<String, String>) -> i32 {
    match verse.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 => {
            for (key, val) in aliases {
                if key.contains(' ') && val.contains(' ') {
                    println!("'{}'='{}'", key, val);
                } else if key.contains(' ') {
                    println!("'{}'={}", key, val);
                } else if val.contains(' ') {
                    println!("{}='{}'", key, val);
                } else if val.is_empty() {
                    println!("{}=''", key);
                } else {
                    println!("{}={}", key, val);
                }
            }
        }
    }
    0
}

pub fn unincant(verse: &Verse, aliases: &mut HashMap<String, String>) -> i32 {
    match verse.clause() {
        Some(clause) => {
            for stanza in clause {
                aliases.remove(&stanza);
            }
        }
        None => {
            eprintln!("unalias: not enough arguments");
            return 1;
        }
    }
    0
}