Add initial aoc common code

1 files changed, 13 insertions, 0 deletions

diff --git a/2020/rust/aoc/src/lib.rs b/2020/rust/aoc/src/lib.rs
new file mode 100644
index 0000000..bd0c31e
--- /dev/null
+++ b/2020/rust/aoc/src/lib.rs
@@ -0,0 +1,13 @@
+use std::env;
+
+pub fn do_parts() -> (bool, bool) {
+    match env::var("AOC_PARTS") {
+        Ok(parts) => {
+            let do_part_1 = parts.contains('1');
+            let do_part_2 = parts.contains('2');
+
+            (do_part_1, do_part_2)
+        },
+        Err(_) => (true, true)
+    }
+}