87 lines
2.4 KiB
Rust
87 lines
2.4 KiB
Rust
use std::error::Error;
|
|
use std::fs;
|
|
use std::collections::HashMap;
|
|
use std::str;
|
|
use regex::Regex;
|
|
|
|
pub fn both(filename: &str) -> Result<(), Box<dyn Error>> {
|
|
let contents = fs::read_to_string(filename)?;
|
|
|
|
let mut valid_passports_simple = 0;
|
|
let mut valid_passports_extra = 0;
|
|
let mut passport: HashMap<&str, &str> = HashMap::new();
|
|
for line in contents.lines() {
|
|
if line.trim() == "" {
|
|
if passport.len() == 8 || (passport.len() == 7 && !passport.contains_key("cid")) {
|
|
valid_passports_simple += 1;
|
|
if validate_passport(passport) {
|
|
valid_passports_extra += 1;
|
|
}
|
|
}
|
|
passport = HashMap::new();
|
|
continue
|
|
}
|
|
|
|
let entries = line.split(" ");
|
|
for entry in entries {
|
|
let kv: Vec<&str> = entry.split(":").collect();
|
|
passport.insert(kv[0], kv[1]);
|
|
}
|
|
}
|
|
|
|
println!("Valid Passports: {}", valid_passports_simple);
|
|
println!("Extra Valid Passports: {}", valid_passports_extra);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn validate_passport(passport: HashMap<&str, &str>) -> bool {
|
|
let byr: u32 = passport.get("byr").unwrap().parse().unwrap();
|
|
if byr < 1920 || byr > 2002 {
|
|
return false
|
|
}
|
|
|
|
let iyr: u32 = passport.get("iyr").unwrap().parse().unwrap();
|
|
if iyr < 2010 || iyr > 2020 {
|
|
return false
|
|
}
|
|
|
|
let eyr: u32 = passport.get("eyr").unwrap().parse().unwrap();
|
|
if eyr < 2020 || eyr > 2030 {
|
|
return false
|
|
}
|
|
|
|
let hgt = passport.get("hgt").unwrap();
|
|
if hgt.ends_with("cm") {
|
|
let hval: u32 = hgt.split("cm").nth(0).unwrap().parse().unwrap();
|
|
if hval < 150 || hval > 193 {
|
|
return false
|
|
}
|
|
} else if hgt.ends_with("in") {
|
|
let hval: u32 = hgt.split("in").nth(0).unwrap().parse().unwrap();
|
|
if hval < 59 || hval > 76 {
|
|
return false
|
|
}
|
|
} else {
|
|
return false
|
|
}
|
|
|
|
let hcl = passport.get("hcl").unwrap();
|
|
let hcl_re = Regex::new(r"\#[0-9a-f]{6}").unwrap();
|
|
if !hcl_re.is_match(hcl) {
|
|
return false
|
|
}
|
|
|
|
let ecl = passport.get("ecl").unwrap().trim();
|
|
if ecl.len() != 3 || !"amb blu brn gry grn hzl oth".contains(ecl) {
|
|
return false
|
|
}
|
|
|
|
let pid = passport.get("pid").unwrap();
|
|
let pid_re = Regex::new(r"^[0-9]{9}$").unwrap();
|
|
if !pid_re.is_match(pid) {
|
|
return false
|
|
}
|
|
|
|
true
|
|
} |