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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
extern crate clap;
extern crate lettre;
use clap::{App, Arg};
use lettre::transport::smtp::authentication::Credentials;
use lettre::{Message, SmtpTransport, Transport};
use rand::prelude::*;
use std::collections::HashMap;
use std::env;
use std::fs;
fn my_gift_to(players: &mut Vec<String>) -> Option<&String> {
let mut rng = rand::thread_rng();
players.shuffle(&mut rng);
Some(&players[0])
}
fn main() {
let matches = App::new("rsanta")
.version("0.1.0")
.author("Santo Cariotti <santo@dcariotti.me>")
.about("Be your friends' Secret Santa")
.arg(
Arg::with_name("file")
.short("f")
.long("file")
.value_name("FILE")
.help("Input file for players")
.takes_value(true),
)
.get_matches();
let config_file = matches
.value_of("file")
.expect("You must specify a config file");
let contents = fs::read_to_string(config_file)
.expect("Error occurs reading this file")
.to_string();
let config_name;
let config_email;
let config_password;
let host;
match env::var("NAME") {
Ok(v) => config_name = v.to_string(),
Err(_) => {
dbg!("Use default name!");
config_name = "Santa Claus".to_string();
}
}
match env::var("EMAIL") {
Ok(v) => config_email = v.to_string(),
Err(e) => panic!("Must provide EMAIL: {}", e),
}
match env::var("PASSWORD") {
Ok(v) => config_password = v.to_string(),
Err(e) => panic!("Must provide PASSWORD: {}", e),
}
match env::var("HOST") {
Ok(v) => host = v,
Err(e) => panic!("Must provide HOST: {}", e),
}
let mut members_data: Vec<&str> = contents.split(">\n").collect();
// ignore the last element, cause it's empty
members_data.pop();
let mut rng = rand::thread_rng();
members_data.shuffle(&mut rng);
let mut members = HashMap::new();
for member in members_data {
let details: Vec<&str> = member.split("<").collect();
// <email> : <name>
members.insert(details[1], details[0]);
}
let mut done: Vec<String> = Vec::new();
let mut emails: Vec<String> = Vec::new();
for (email, _) in &members {
emails.push(email.to_string());
}
let creds = Credentials::new(config_email.to_string(), config_password);
let mailer = SmtpTransport::relay(&host)
.unwrap()
.credentials(creds)
.build();
for (email, name) in members.iter() {
loop {
match my_gift_to(&mut emails) {
Some(gift_to_email) => {
if gift_to_email.to_string() != email.to_string()
&& !done.iter().any(|v| v == gift_to_email)
{
let gift_to_name;
match members.get(&gift_to_email[..]) {
Some(v) => gift_to_name = v,
None => continue
}
let mail = Message::builder()
.from(
format!("{} <{}>", config_name, config_email)
.parse()
.unwrap(),
)
.to(format!("{} <{}>", name, email).parse().unwrap())
.subject("Secret Santa!")
.body(format!("You are the Secret Santa of:\n{}", gift_to_name))
.unwrap();
match mailer.send(&mail) {
Ok(_) => println!("Email sent successfully to {}!", email),
Err(e) => panic!("Could not send email: {:?}", e),
}
done.push(gift_to_email.to_string());
break;
}
}
None => break,
}
}
}
}
|
