From 5fe5f27b2882d0e7628d4374ddbc888417220b60 Mon Sep 17 00:00:00 2001
From: tomsmeding <tom.smeding@gmail.com>
Date: Sun, 2 Dec 2018 11:09:42 +0100
Subject: Day 2

---
 2018/src/main.rs | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

(limited to '2018/src/main.rs')

diff --git a/2018/src/main.rs b/2018/src/main.rs
index 7e683ca..162a56e 100644
--- a/2018/src/main.rs
+++ b/2018/src/main.rs
@@ -7,10 +7,12 @@ use std::process::exit;
 use argparse::{ArgumentParser, StoreTrue, Store};
 
 mod day1;
+mod day2;
 
 fn day_switch<T: BufRead>(day: i32, reader: T) -> io::Result<()> {
     match day {
         1 => day1::main(reader),
+        2 => day2::main(reader),
         _ => Err(Error::new(ErrorKind::Other, "Invalid day"))
     }
 }
@@ -65,13 +67,18 @@ fn main() -> io::Result<()> {
     }
 
     error_handler(||
-        match day_string.parse::<i32>() {
-            Ok(day) => run_day(day, &options),
-            Err(_) => {
-                for day in 1..1 {
-                    run_day(day, &options)?;
+        if day_string.len() == 0 {
+            for day in 1..3 {
+                run_day(day, &options)?;
+            }
+            Ok(())
+        } else {
+            match day_string.parse::<i32>() {
+                Ok(day) => run_day(day, &options),
+                Err(_) => {
+                    println!("Invalid day argument");
+                    exit(1)
                 }
-                Ok(())
             }
         }
     )
-- 
cgit v1.2.3-70-g09d2