feat: associate player to one another

2 files changed, 74 insertions, 1 deletions

diff --git a/Cargo.toml b/Cargo.toml
index 17b5ef0..a847dd7 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -7,3 +7,5 @@ edition = "2018"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+clap = "2.33.3"
+rand = "0.7.3"
diff --git a/src/main.rs b/src/main.rs
index e7a11a9..f80069f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,74 @@
+extern crate clap;
+use clap::{App, Arg};
+use rand::prelude::*;
+use std::collections::HashMap;
+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() {
-    println!("Hello, world!");
+    let matches = App::new("rsanta")
+        .version("0.1.0")
+        .author("Santo Cariotti <santo@dcariotti.me>")
+        .about("Make secret santa with your friends")
+        .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 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());
+    }
+    println!("{:?}", emails);
+
+    for (email, name) in members.into_iter() {
+        loop {
+            match my_gift_to(&mut emails) {
+                Some(gift_to) => {
+                    if gift_to.to_string() != email.to_string()
+                        && !done.iter().any(|v| v == gift_to)
+                    {
+                        println!("{} {}", email, gift_to);
+                        done.push(gift_to.to_string());
+                        break;
+                    }
+                }
+                None => break,
+            }
+        }
+    }
 }