Run only most recent day by default

Author: swlody

Cargo.lock

@@ -15,6 +15,7 @@ bench = ["criterion"]
 
 [dependencies]
 attohttpc = { version = "0.16.0", default_features = false, features = ["tls"] }
+chrono = "0.4"
 clap = { version = "3.0.0-beta.2", default_features = false, features = ["std"] }
 colored = "2.0.0"
 dirs = "3.0.1"

Cargo.toml

@@ -3,6 +3,7 @@ pub mod parse;
 pub mod utils;
 
 // Reexport some crates for the generated main
+pub use chrono;
 pub use clap;
 pub use colored;
 
@@ -45,6 +46,7 @@ pub fn args(year: u16) -> App<'static> {
                 .long("bench")
                 .about("Run criterion benchmarks"),
         )
+        .arg(Arg::new("all").long("all").about("Run all days"))
 }
 
 #[macro_export]
@@ -54,16 +56,22 @@ macro_rules! base_main {
         use std::io::Read;
         use std::time::Instant;
 
+        use $crate::chrono::{prelude::*, FixedOffset, TimeZone};
+
         use $crate::{bench_day, extract_day, parse, run_day};
 
         const YEAR: u16 = $year;
 
         fn main() {
             let mut opt = $crate::args(YEAR).get_matches();
 
+
             if opt.is_present("bench") {
                 bench();
             } else {
+                let est = FixedOffset::west(5 * 3600).from_utc_datetime(&Utc::now().naive_utc());
+                let today = est.day().to_string();
+
                 let days: Vec<_> = {
                     if let Some(opt_days) = opt.values_of("days") {
                         let opt_days: Vec<_> = opt_days.collect();
@@ -84,10 +92,14 @@ macro_rules! base_main {
                             .filter(|day| days.contains(&format!("day{}", day).as_str()))
                             .collect()
                     } else {
-                        parse!(extract_day {}; $( $tail )*)
-                            .iter()
-                            .map(|s| &s[3..])
-                            .collect()
+                        if !opt.is_present("all") && est.year() == YEAR as i32 && est.month() == 12 && est.day() <= 25 {
+                            vec![&today]
+                        } else {
+                            parse!(extract_day {}; $( $tail )*)
+                                .iter()
+                                .map(|s| &s[3..])
+                                .collect()
+                        }
                     }
                 };