aboutsummaryrefslogtreecommitdiff
path: root/helix-core/src
diff options
context:
space:
mode:
authorJason Rodney Hansen2021-11-18 13:23:27 +0000
committerIvan Tham2021-12-05 08:22:58 +0000
commit95cfeed2fa60420f584ef1ab051275780f55d4e9 (patch)
tree697f51f2d9d3899719fdd07e52ceb090e09b9ec5 /helix-core/src
parentc1f6167e37909517676c30b0a80203739e8492a5 (diff)
Add support for incrementing year and month
Diffstat (limited to 'helix-core/src')
-rw-r--r--helix-core/src/date.rs332
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()