1 files changed, 88 insertions, 1 deletions

diff --git a/src/dates.rs b/src/dates.rs
index 09aac82..069af3f 100644
--- a/src/dates.rs
+++ b/src/dates.rs
@@ -1,5 +1,7 @@
+use std::fs;
 use std::path::Path;
-// use std::str::FromStr;
+use std::collections::HashSet;
+use crate::tasks_from_path;
 // use chrono::{Datelike, NaiveDate, NaiveTime, Weekday, Duration, Timelike};
 use chrono::{Datelike, NaiveDate, Duration};
 
@@ -47,6 +49,79 @@ pub fn relevant_files<'a>(start: &'a NaiveDate, end: &'a NaiveDate) -> Vec<WTDPa
     return paths;
 }
 
+/* This is a pretty rough way of doing it, but it should work. Basically:
+ * - Pull the tasks from wtd.md, record a set of the weeks involved.
+ * - Also pull all tasks from any (future) [week].md files, _removing_ those weeks from the set.
+ * - For every remaining week mentioned in wtd.md but not a [week].md, add a copy of weekly.md for
+ *   that week.
+ * - Finally, if there were no future [week].mds, add a copy of weekly.md for a date far in the
+ *   future (i.e., not to conflict with any other dates).
+ */
+pub fn relevant_files_starting<'a>(start: &'a NaiveDate) -> Vec<WTDPath> {
+    let start_week = week_start_of(start);
+    let mut paths: Vec<WTDPath> = vec![];
+    let mut weeks = HashSet::new();
+    let mut oldest_date: NaiveDate = *start;
+    if let Some(path) = path_if_exists("wtd.md") {
+        paths.push(WTDPath {
+            path: path,
+            week_start: None,
+            is_weekly_file: false,
+        });
+        for task in tasks_from_path(paths.last().unwrap()) {
+            if task.date < *start {
+                continue;
+            }
+            if task.date > oldest_date {
+                oldest_date = task.date;
+            }
+            let week = week_start_of(&task.date);
+            weeks.insert(week);
+        }
+    }
+
+    // https://stackoverflow.com/questions/26076005
+    for path in fs::read_dir("./").unwrap() {
+        let mut file_name = path.unwrap().path().to_str().unwrap().to_string();
+        file_name.remove(0);
+        file_name.remove(0); // removes the ./
+        if let Some(week) = maybe_valid_week_file(&file_name) {
+            if week < start_week {
+                continue;
+            }
+            if week + Duration::days(6) > oldest_date {
+                oldest_date = week + Duration::days(6);
+            }
+            paths.push(WTDPath {
+                path: path_if_exists(&file_name).unwrap(),
+                week_start: Some(week),
+                is_weekly_file: false,
+            });
+            weeks.remove(&week);
+        }
+    }
+
+    let weekly = path_if_exists("weekly.md");
+    if weekly.is_some() {
+        for week in &weeks {
+            paths.push(WTDPath {
+                path: weekly.as_ref().unwrap().to_string(),
+                week_start: Some(*week),
+                is_weekly_file: true,
+            });
+        }
+        if weeks.len() == 0 {
+            paths.push(WTDPath {
+                path: weekly.as_ref().unwrap().to_string(),
+                week_start: Some(week_start_of(&(oldest_date + Duration::days(7)))),
+                is_weekly_file: true,
+            });
+        }
+    }
+
+    return paths;
+}
+
 fn week_file(week_start: NaiveDate) -> Option<String> {
     let filename = week_start.format("%b_%d_%Y.md").to_string().to_lowercase();
     return path_if_exists(&filename);
@@ -62,6 +137,18 @@ pub fn parse_week_str(week_str: &str) -> Option<NaiveDate> {
     }
 }
 
+pub fn maybe_valid_week_file(week_file_name: &str) -> Option<NaiveDate> {
+    match week_file_name.rsplit_once(".") {
+        Some((name, "md")) => {
+            if let Some(date) = parse_week_str(name) {
+                return if week_start_of(&date) == date { Some(date) } else { None };
+            }
+            return None;
+        },
+        _ => return None,
+    }
+}
+
 fn path_if_exists(name: &str) -> Option<String> {
     return if Path::new(name).exists() { Some(name.to_string()) } else { None };
 }