src/main.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
39
40
41
42
43
44

use std::env;
use std::io::{self, BufRead, Write};

mod alias;

#[cfg(test)]
mod tests;

use alias::*;

fn print_usage(program: &str) {
    println!("Usage: {} FILE", program);
}

fn main() {
    let args: Vec<String> = env::args().collect();
    let program = args[0].clone();

    let file = if args.len() > 1 {
        &args[1]
    } else {
        print_usage(&program);
        return;
    };

    let stdin = io::stdin();

    for line in stdin.lock().lines() {
        let line = line.unwrap();

        // Write the message to STDOUT so that it can be properly read by Mutt
        println!("{}", line);

        if line.starts_with("From: ") {
            let mut alias = Alias::new(&line);
            match alias.write_to_file(&file) {
                Ok(_)  => continue,
                Err(e @ AliasSearchError::NotFound) | Err(e @ AliasSearchError::EmailExists) =>
                    io::stderr().write(e.to_string().as_bytes()).ok(),
                Err(e) => io::stderr().write(e.to_string().as_bytes()).ok(),
            };
        }
    }
}