Add support for incrementing year and month

1 files changed, 247 insertions, 85 deletions

diff --git a/helix-core/src/date.rs b/helix-core/src/date.rs
index 1332670d..e189fe03 100644
--- a/helix-core/src/date.rs
+++ b/helix-core/src/date.rs
@@ -1,70 +1,124 @@
-use chrono::{Duration, NaiveDate};
+use gregorian::{Date, DateResultExt};
+use regex::Regex;
 
 use std::borrow::Cow;
 
 use ropey::RopeSlice;
 
-use crate::{
-    textobject::{textobject_word, TextObject},
-    Range, Tendril,
-};
+use crate::{Range, Tendril};
+
+#[derive(Copy, Clone, Debug, PartialEq, Eq)]
+struct Format {
+    regex: &'static str,
+    separator: char,
+}
 
 // Only support formats that aren't region specific.
-static FORMATS: &[&str] = &["%Y-%m-%d", "%Y/%m/%d"];
+static FORMATS: &[Format] = &[
+    Format {
+        regex: r"(\d{4})-(\d{2})-(\d{2})",
+        separator: '-',
+    },
+    Format {
+        regex: r"(\d{4})/(\d{2})/(\d{2})",
+        separator: '/',
+    },
+];
 
-// We don't want to parse ambiguous dates like 10/11/12 or 7/8/10.
-// They must be YYYY-mm-dd or YYYY/mm/dd.
-// So 2021-01-05 works, but 2021-1-5 doesn't.
 const DATE_LENGTH: usize = 10;
 
+#[derive(Copy, Clone, Debug, PartialEq, Eq)]
+enum DateField {
+    Year,
+    Month,
+    Day,
+}
+
 #[derive(Debug, PartialEq, Eq)]
 pub struct DateIncrementor {
-    pub date: NaiveDate,
+    pub date: Date,
     pub range: Range,
-    pub format: &'static str,
+
+    field: DateField,
+    format: Format,
 }
 
 impl DateIncrementor {
     pub fn from_range(text: RopeSlice, range: Range) -> Option<DateIncrementor> {
-        // Don't increment if the cursor is one right of the date text.
-        if text.char(range.from()).is_whitespace() {
-            return None;
-        }
+        let from = range.from().saturating_sub(DATE_LENGTH);
+        let to = (range.from() + DATE_LENGTH).min(text.len_chars());
+        let (from_in_text, to_in_text) = (range.from() - from, range.to() - from);
+        let text: Cow<str> = text.slice(from..to).into();
 
-        let range = textobject_word(text, range, TextObject::Inside, 1, true);
-        let text: Cow<str> = text.slice(range.from()..range.to()).into();
+        FORMATS.iter().find_map(|&format| {
+            let re = Regex::new(format.regex).ok()?;
+            let captures = re.captures(&text)?;
 
-        let first = text.chars().next()?;
-        let last = text.chars().next_back()?;
+            let date = captures.get(0)?;
+            let offset = range.from() - from_in_text;
+            let range = Range::new(date.start() + offset, date.end() + offset);
 
-        // Allow date strings in quotes.
-        let (range, text) = if first == last && (first == '"' || first == '\'') {
-            (
-                Range::new(range.from() + 1, range.to() - 1),
-                Cow::from(&text[1..text.len() - 1]),
-            )
-        } else {
-            (range, text)
-        };
+            let year = captures.get(1)?;
+            let month = captures.get(2)?;
+            let day = captures.get(3)?;
 
-        if text.len() != DATE_LENGTH {
-            return None;
-        }
+            let year_range = year.range();
+            let month_range = month.range();
+            let day_range = day.range();
 
-        FORMATS.iter().find_map(|format| {
-            NaiveDate::parse_from_str(&text, format)
-                .ok()
-                .map(|date| DateIncrementor {
-                    date,
-                    range,
-                    format,
-                })
+            let to_inclusive = if to_in_text > from_in_text {
+                to_in_text - 1
+            } else {
+                to_in_text
+            };
+            let field = if year_range.contains(&from_in_text) && year_range.contains(&to_inclusive)
+            {
+                DateField::Year
+            } else if month_range.contains(&from_in_text) && month_range.contains(&to_inclusive) {
+                DateField::Month
+            } else if day_range.contains(&from_in_text) && day_range.contains(&to_inclusive) {
+                DateField::Day
+            } else {
+                return None;
+            };
+
+            let year: i16 = year.as_str().parse().ok()?;
+            let month: u8 = month.as_str().parse().ok()?;
+            let day: u8 = day.as_str().parse().ok()?;
+
+            let date = Date::new(year, month, day).ok()?;
+
+            Some(DateIncrementor {
+                date,
+                field,
+                range,
+                format,
+            })
         })
     }
 
     pub fn incremented_text(&self, amount: i64) -> Tendril {
-        let incremented_date = self.date + Duration::days(amount);
-        incremented_date.format(self.format).to_string().into()
+        let date = match self.field {
+            DateField::Year => self
+                .date
+                .add_years(amount.try_into().unwrap_or(0))
+                .or_next_valid(),
+            DateField::Month => self
+                .date
+                .add_months(amount.try_into().unwrap_or(0))
+                .or_prev_valid(),
+            DateField::Day => self.date.add_days(amount.try_into().unwrap_or(0)),
+        };
+
+        format!(
+            "{:04}{}{:02}{}{:02}",
+            date.year(),
+            self.format.separator,
+            date.month().to_number(),
+            self.format.separator,
+            date.day()
+        )
+        .into()
     }
 }
 
@@ -74,43 +128,144 @@ mod test {
     use crate::Rope;
 
     #[test]
-    fn test_date_dashes() {
+    fn test_create_incrementor_for_year_with_dashes() {
         let rope = Rope::from_str("2021-11-15");
-        let range = Range::point(0);
-        assert_eq!(
-            DateIncrementor::from_range(rope.slice(..), range),
-            Some(DateIncrementor {
-                date: NaiveDate::from_ymd(2021, 11, 15),
-                range: Range::new(0, 10),
-                format: "%Y-%m-%d",
-            })
-        );
+
+        for head in 0..=3 {
+            let range = Range::point(head);
+            assert_eq!(
+                DateIncrementor::from_range(rope.slice(..), range),
+                Some(DateIncrementor {
+                    date: Date::new(2021, 11, 15).unwrap(),
+                    range: Range::new(0, 10),
+                    field: DateField::Year,
+                    format: FORMATS[0],
+                })
+            );
+        }
+    }
+
+    #[test]
+    fn test_create_incrementor_for_month_with_dashes() {
+        let rope = Rope::from_str("2021-11-15");
+
+        for head in 5..=6 {
+            let range = Range::point(head);
+            assert_eq!(
+                DateIncrementor::from_range(rope.slice(..), range),
+                Some(DateIncrementor {
+                    date: Date::new(2021, 11, 15).unwrap(),
+                    range: Range::new(0, 10),
+                    field: DateField::Month,
+                    format: FORMATS[0],
+                })
+            );
+        }
+    }
+
+    #[test]
+    fn test_create_incrementor_for_day_with_dashes() {
+        let rope = Rope::from_str("2021-11-15");
+
+        for head in 8..=9 {
+            let range = Range::point(head);
+            assert_eq!(
+                DateIncrementor::from_range(rope.slice(..), range),
+                Some(DateIncrementor {
+                    date: Date::new(2021, 11, 15).unwrap(),
+                    range: Range::new(0, 10),
+                    field: DateField::Day,
+                    format: FORMATS[0],
+                })
+            );
+        }
     }
 
     #[test]
-    fn test_date_slashes() {
+    fn test_try_create_incrementor_on_dashes() {
+        let rope = Rope::from_str("2021-11-15");
+
+        for head in &[4, 7] {
+            let range = Range::point(*head);
+            assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None,);
+        }
+    }
+
+    #[test]
+    fn test_create_incrementor_for_year_with_slashes() {
         let rope = Rope::from_str("2021/11/15");
-        let range = Range::point(0);
-        assert_eq!(
-            DateIncrementor::from_range(rope.slice(..), range),
-            Some(DateIncrementor {
-                date: NaiveDate::from_ymd(2021, 11, 15),
-                range: Range::new(0, 10),
-                format: "%Y/%m/%d",
-            })
-        );
+
+        for head in 0..=3 {
+            let range = Range::point(head);
+            assert_eq!(
+                DateIncrementor::from_range(rope.slice(..), range),
+                Some(DateIncrementor {
+                    date: Date::new(2021, 11, 15).unwrap(),
+                    range: Range::new(0, 10),
+                    field: DateField::Year,
+                    format: FORMATS[1],
+                })
+            );
+        }
+    }
+
+    #[test]
+    fn test_create_incrementor_for_month_with_slashes() {
+        let rope = Rope::from_str("2021/11/15");
+
+        for head in 5..=6 {
+            let range = Range::point(head);
+            assert_eq!(
+                DateIncrementor::from_range(rope.slice(..), range),
+                Some(DateIncrementor {
+                    date: Date::new(2021, 11, 15).unwrap(),
+                    range: Range::new(0, 10),
+                    field: DateField::Month,
+                    format: FORMATS[1],
+                })
+            );
+        }
+    }
+
+    #[test]
+    fn test_create_incrementor_for_day_with_slashes() {
+        let rope = Rope::from_str("2021/11/15");
+
+        for head in 8..=9 {
+            let range = Range::point(head);
+            assert_eq!(
+                DateIncrementor::from_range(rope.slice(..), range),
+                Some(DateIncrementor {
+                    date: Date::new(2021, 11, 15).unwrap(),
+                    range: Range::new(0, 10),
+                    field: DateField::Day,
+                    format: FORMATS[1],
+                })
+            );
+        }
+    }
+
+    #[test]
+    fn test_try_create_incrementor_on_slashes() {
+        let rope = Rope::from_str("2021/11/15");
+
+        for head in &[4, 7] {
+            let range = Range::point(*head);
+            assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None,);
+        }
     }
 
     #[test]
     fn test_date_surrounded_by_spaces() {
         let rope = Rope::from_str("   2021-11-15  ");
-        let range = Range::point(10);
+        let range = Range::point(3);
         assert_eq!(
             DateIncrementor::from_range(rope.slice(..), range),
             Some(DateIncrementor {
-                date: NaiveDate::from_ymd(2021, 11, 15),
+                date: Date::new(2021, 11, 15).unwrap(),
                 range: Range::new(3, 13),
-                format: "%Y-%m-%d",
+                field: DateField::Year,
+                format: FORMATS[0],
             })
         );
     }
@@ -122,23 +277,25 @@ mod test {
         assert_eq!(
             DateIncrementor::from_range(rope.slice(..), range),
             Some(DateIncrementor {
-                date: NaiveDate::from_ymd(2021, 11, 15),
+                date: Date::new(2021, 11, 15).unwrap(),
                 range: Range::new(8, 18),
-                format: "%Y-%m-%d",
+                field: DateField::Year,
+                format: FORMATS[0],
             })
         );
     }
 
     #[test]
     fn test_date_in_double_quotes() {
-        let rope = Rope::from_str("date = \"2021-11-15\"");
-        let range = Range::point(10);
+        let rope = Rope::from_str("let date = \"2021-11-15\";");
+        let range = Range::point(12);
         assert_eq!(
             DateIncrementor::from_range(rope.slice(..), range),
             Some(DateIncrementor {
-                date: NaiveDate::from_ymd(2021, 11, 15),
-                range: Range::new(8, 18),
-                format: "%Y-%m-%d",
+                date: Date::new(2021, 11, 15).unwrap(),
+                range: Range::new(12, 22),
+                field: DateField::Year,
+                format: FORMATS[0],
             })
         );
     }
@@ -189,23 +346,28 @@ mod test {
     #[test]
     fn test_increment_dates() {
         let tests = [
-            ("1980-12-21", 1, "1980-12-22"),
-            ("1980-12-21", -1, "1980-12-20"),
-            ("1980-12-21", 100, "1981-03-31"),
-            ("1980-12-21", -100, "1980-09-12"),
-            ("1980-12-21", 1000, "1983-09-17"),
-            ("1980-12-21", -1000, "1978-03-27"),
-            ("1980/12/21", 1, "1980/12/22"),
-            ("1980/12/21", -1, "1980/12/20"),
-            ("1980/12/21", 100, "1981/03/31"),
-            ("1980/12/21", -100, "1980/09/12"),
-            ("1980/12/21", 1000, "1983/09/17"),
-            ("1980/12/21", -1000, "1978/03/27"),
+            // (original, cursor, amount, expected)
+            ("2020-02-28", 0, 1, "2021-02-28"),
+            ("2020-02-29", 0, 1, "2021-03-01"),
+            ("2020-01-31", 5, 1, "2020-02-29"),
+            ("2020-01-20", 5, 1, "2020-02-20"),
+            ("2020-02-28", 8, 1, "2020-02-29"),
+            ("2021-02-28", 8, 1, "2021-03-01"),
+            ("2021-02-28", 0, -1, "2020-02-28"),
+            ("2021-03-01", 0, -1, "2020-03-01"),
+            ("2020-02-29", 5, -1, "2020-01-29"),
+            ("2020-02-20", 5, -1, "2020-01-20"),
+            ("2020-02-29", 8, -1, "2020-02-28"),
+            ("2021-03-01", 8, -1, "2021-02-28"),
+            ("1980/12/21", 8, 100, "1981/03/31"),
+            ("1980/12/21", 8, -100, "1980/09/12"),
+            ("1980/12/21", 8, 1000, "1983/09/17"),
+            ("1980/12/21", 8, -1000, "1978/03/27"),
         ];
 
-        for (original, amount, expected) in tests {
+        for (original, cursor, amount, expected) in tests {
             let rope = Rope::from_str(original);
-            let range = Range::point(0);
+            let range = Range::point(cursor);
             assert_eq!(
                 DateIncrementor::from_range(rope.slice(..), range)
                     .unwrap()