aboutsummaryrefslogtreecommitdiff
path: root/helix-core/src/increment/date.rs
diff options
context:
space:
mode:
authorJason Rodney Hansen2021-11-21 17:38:41 +0000
committerIvan Tham2021-12-05 08:22:58 +0000
commitc9641fccedc51737a74ed47009279fa688462ea9 (patch)
treea057ce6a6a476224238dbb0341cbb8371e88497d /helix-core/src/increment/date.rs
parent2a0c685a7857a94db2ac61b586c92425da273ea7 (diff)
Add `Increment` trait
Diffstat (limited to 'helix-core/src/increment/date.rs')
-rw-r--r--helix-core/src/increment/date.rs474
1 files changed, 474 insertions, 0 deletions
diff --git a/helix-core/src/increment/date.rs b/helix-core/src/increment/date.rs
new file mode 100644
index 00000000..05442990
--- /dev/null
+++ b/helix-core/src/increment/date.rs
@@ -0,0 +1,474 @@
+use regex::Regex;
+
+use std::borrow::Cow;
+use std::cmp;
+
+use ropey::RopeSlice;
+
+use crate::{Range, Tendril};
+
+use chrono::{Datelike, Duration, NaiveDate};
+
+use super::Increment;
+
+fn ndays_in_month(year: i32, month: u32) -> u32 {
+ // The first day of the next month...
+ let (y, m) = if month == 12 {
+ (year + 1, 1)
+ } else {
+ (year, month + 1)
+ };
+ let d = NaiveDate::from_ymd(y, m, 1);
+
+ // ...is preceded by the last day of the original month.
+ d.pred().day()
+}
+
+fn add_days(date: NaiveDate, amount: i64) -> Option<NaiveDate> {
+ date.checked_add_signed(Duration::days(amount))
+}
+
+fn add_months(date: NaiveDate, amount: i64) -> Option<NaiveDate> {
+ let month = date.month0() as i64 + amount;
+ let year = date.year() + i32::try_from(month / 12).ok()?;
+ let year = if month.is_negative() { year - 1 } else { year };
+
+ // Normalize month
+ let month = month % 12;
+ let month = if month.is_negative() {
+ month + 13
+ } else {
+ month + 1
+ } as u32;
+
+ let day = cmp::min(date.day(), ndays_in_month(year, month));
+
+ Some(NaiveDate::from_ymd(year, month, day))
+}
+
+fn add_years(date: NaiveDate, amount: i64) -> Option<NaiveDate> {
+ let year = i32::try_from(date.year() as i64 + amount).ok()?;
+ let ndays = ndays_in_month(year, date.month());
+
+ if date.day() > ndays {
+ let d = NaiveDate::from_ymd(year, date.month(), ndays);
+ Some(d.succ())
+ } else {
+ date.with_year(year)
+ }
+}
+
+#[derive(Copy, Clone, Debug, PartialEq, Eq)]
+struct Format {
+ regex: &'static str,
+ separator: char,
+}
+
+// Only support formats that aren't region specific.
+static FORMATS: &[Format] = &[
+ Format {
+ regex: r"(\d{4})-(\d{2})-(\d{2})",
+ separator: '-',
+ },
+ Format {
+ regex: r"(\d{4})/(\d{2})/(\d{2})",
+ separator: '/',
+ },
+];
+
+const DATE_LENGTH: usize = 10;
+
+#[derive(Copy, Clone, Debug, PartialEq, Eq)]
+enum DateField {
+ Year,
+ Month,
+ Day,
+}
+
+#[derive(Debug, PartialEq, Eq)]
+pub struct DateIncrementor {
+ date: NaiveDate,
+ range: Range,
+ field: DateField,
+ format: Format,
+}
+
+impl DateIncrementor {
+ pub fn from_range(text: RopeSlice, range: Range) -> Option<DateIncrementor> {
+ let range = if range.is_empty() {
+ if range.anchor < text.len_bytes() {
+ // Treat empty range as a cursor range.
+ range.put_cursor(text, range.anchor + 1, true)
+ } else {
+ // The range is empty and at the end of the text.
+ return None;
+ }
+ } else {
+ range
+ };
+
+ 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();
+
+ FORMATS.iter().find_map(|&format| {
+ let re = Regex::new(format.regex).ok()?;
+ let captures = re.captures(&text)?;
+
+ let date = captures.get(0)?;
+ let offset = range.from() - from_in_text;
+ let range = Range::new(date.start() + offset, date.end() + offset);
+
+ let (year, month, day) = (captures.get(1)?, captures.get(2)?, captures.get(3)?);
+ let (year_range, month_range, day_range) = (year.range(), month.range(), day.range());
+
+ let field = if year_range.contains(&from_in_text)
+ && year_range.contains(&(to_in_text - 1))
+ {
+ DateField::Year
+ } else if month_range.contains(&from_in_text) && month_range.contains(&(to_in_text - 1))
+ {
+ DateField::Month
+ } else if day_range.contains(&from_in_text) && day_range.contains(&(to_in_text - 1)) {
+ DateField::Day
+ } else {
+ return None;
+ };
+
+ let date = NaiveDate::from_ymd_opt(
+ year.as_str().parse::<i32>().ok()?,
+ month.as_str().parse::<u32>().ok()?,
+ day.as_str().parse::<u32>().ok()?,
+ )?;
+
+ Some(DateIncrementor {
+ date,
+ field,
+ range,
+ format,
+ })
+ })
+ }
+}
+
+impl Increment for DateIncrementor {
+ fn increment(&self, amount: i64) -> (Range, Tendril) {
+ let date = match self.field {
+ DateField::Year => add_years(self.date, amount),
+ DateField::Month => add_months(self.date, amount),
+ DateField::Day => add_days(self.date, amount),
+ }
+ .unwrap_or(self.date);
+
+ (
+ self.range,
+ format!(
+ "{:04}{}{:02}{}{:02}",
+ date.year(),
+ self.format.separator,
+ date.month(),
+ self.format.separator,
+ date.day()
+ )
+ .into(),
+ )
+ }
+}
+
+#[cfg(test)]
+mod test {
+ use super::*;
+ use crate::Rope;
+
+ #[test]
+ fn test_create_incrementor_for_year_with_dashes() {
+ let rope = Rope::from_str("2021-11-15");
+
+ for cursor in 0..=3 {
+ let range = Range::new(cursor, cursor + 1);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ 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 cursor in 5..=6 {
+ let range = Range::new(cursor, cursor + 1);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ 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 cursor in 8..=9 {
+ let range = Range::new(cursor, cursor + 1);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ range: Range::new(0, 10),
+ field: DateField::Day,
+ format: FORMATS[0],
+ })
+ );
+ }
+ }
+
+ #[test]
+ fn test_try_create_incrementor_on_dashes() {
+ let rope = Rope::from_str("2021-11-15");
+
+ for &cursor in &[4, 7] {
+ let range = Range::new(cursor, cursor + 1);
+ 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");
+
+ for cursor in 0..=3 {
+ let range = Range::new(cursor, cursor + 1);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ 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 cursor in 5..=6 {
+ let range = Range::new(cursor, cursor + 1);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ 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 cursor in 8..=9 {
+ let range = Range::new(cursor, cursor + 1);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ 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 &cursor in &[4, 7] {
+ let range = Range::new(cursor, cursor + 1);
+ 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::new(3, 4);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ range: Range::new(3, 13),
+ field: DateField::Year,
+ format: FORMATS[0],
+ })
+ );
+ }
+
+ #[test]
+ fn test_date_in_single_quotes() {
+ let rope = Rope::from_str("date = '2021-11-15'");
+ let range = Range::new(10, 11);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ range: Range::new(8, 18),
+ field: DateField::Year,
+ format: FORMATS[0],
+ })
+ );
+ }
+
+ #[test]
+ fn test_date_in_double_quotes() {
+ let rope = Rope::from_str("let date = \"2021-11-15\";");
+ let range = Range::new(12, 13);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ range: Range::new(12, 22),
+ field: DateField::Year,
+ format: FORMATS[0],
+ })
+ );
+ }
+
+ #[test]
+ fn test_date_cursor_one_right_of_date() {
+ let rope = Rope::from_str("2021-11-15 ");
+ let range = Range::new(10, 11);
+ assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None);
+ }
+
+ #[test]
+ fn test_date_cursor_one_left_of_number() {
+ let rope = Rope::from_str(" 2021-11-15");
+ let range = Range::new(0, 1);
+ assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None);
+ }
+
+ #[test]
+ fn test_date_empty_range_at_beginning() {
+ 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),
+ field: DateField::Year,
+ format: FORMATS[0],
+ })
+ );
+ }
+
+ #[test]
+ fn test_date_empty_range_at_in_middle() {
+ let rope = Rope::from_str("2021-11-15");
+ let range = Range::point(5);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range),
+ Some(DateIncrementor {
+ date: NaiveDate::from_ymd(2021, 11, 15),
+ range: Range::new(0, 10),
+ field: DateField::Month,
+ format: FORMATS[0],
+ })
+ );
+ }
+
+ #[test]
+ fn test_date_empty_range_at_end() {
+ let rope = Rope::from_str("2021-11-15");
+ let range = Range::point(10);
+ assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None);
+ }
+
+ #[test]
+ fn test_invalid_dates() {
+ let tests = [
+ "0000-00-00",
+ "1980-2-21",
+ "1980-12-1",
+ "12345",
+ "2020-02-30",
+ "1999-12-32",
+ "19-12-32",
+ "1-2-3",
+ "0000/00/00",
+ "1980/2/21",
+ "1980/12/1",
+ "12345",
+ "2020/02/30",
+ "1999/12/32",
+ "19/12/32",
+ "1/2/3",
+ ];
+
+ for invalid in tests {
+ let rope = Rope::from_str(invalid);
+ let range = Range::new(0, 1);
+
+ assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None);
+ }
+ }
+
+ #[test]
+ fn test_increment_dates() {
+ let tests = [
+ // (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"),
+ ("2021-01-01", 5, -1, "2020-12-01"),
+ ("2021-01-31", 5, -2, "2020-11-30"),
+ ("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, cursor, amount, expected) in tests {
+ let rope = Rope::from_str(original);
+ let range = Range::new(cursor, cursor + 1);
+ assert_eq!(
+ DateIncrementor::from_range(rope.slice(..), range)
+ .unwrap()
+ .increment(amount)
+ .1,
+ expected.into()
+ );
+ }
+ }
+}