Support for the RU date format (#46)

rturovtsev · web-flow · commit 987a8eed4b47 · 2024-11-03T06:43:12.000Z
diff --git a/rules/ru/date.go b/rules/ru/date.go
@@ -0,0 +1,66 @@
+package ru
+
+import (
+	"regexp"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/olebedev/when/rules"
+	"github.com/pkg/errors"
+)
+
+// https://go.dev/play/p/YsVdaraCwIP
+
+func Date(s rules.Strategy) rules.Rule {
+	return &rules.F{
+		RegExp: regexp.MustCompile(`(?i)(?:\b|^)(\d{1,2})\s*(` + MONTHS_PATTERN + `)(?:\s*(\d{4}))?(?:\s*в\s*(\d{1,2}):(\d{2}))?(?:\b|$)`),
+		Applier: func(m *rules.Match, c *rules.Context, o *rules.Options, ref time.Time) (bool, error) {
+			if (c.Day != nil || c.Month != nil || c.Year != nil) || s != rules.Override {
+				return false, nil
+			}
+
+			day, err := strconv.Atoi(m.Captures[0])
+			if err != nil {
+				return false, errors.Wrap(err, "date rule: day")
+			}
+
+			month, ok := MONTHS[strings.ToLower(m.Captures[1])]
+			if !ok {
+				return false, errors.New("date rule: invalid month")
+			}
+
+			year := time.Now().Year()
+			if m.Captures[2] != "" {
+				year, err = strconv.Atoi(m.Captures[2])
+				if err != nil {
+					return false, errors.Wrap(err, "date rule: year")
+				}
+			}
+
+			hour, minute := 0, 0
+			if m.Captures[3] != "" && m.Captures[4] != "" {
+				hour, err = strconv.Atoi(m.Captures[3])
+				if err != nil {
+					return false, errors.Wrap(err, "date rule: hour")
+				}
+				minute, err = strconv.Atoi(m.Captures[4])
+				if err != nil {
+					return false, errors.Wrap(err, "date rule: minute")
+				}
+			}
+
+			c.Day = &day
+			c.Month = pointerToInt(int(month))
+			c.Year = &year
+			c.Hour = &hour
+			c.Minute = &minute
+
+			return true, nil
+		},
+	}
+}
+
+func pointerToInt(v int) *int {
+	return &v
+}
diff --git a/rules/ru/date_test.go b/rules/ru/date_test.go
@@ -0,0 +1,41 @@
+package ru_test
+
+import (
+	"github.com/olebedev/when"
+	"github.com/olebedev/when/rules"
+	"github.com/olebedev/when/rules/ru"
+	"testing"
+	"time"
+)
+
+func TestDate(t *testing.T) {
+	w := when.New(nil)
+	w.Add(ru.Date(rules.Override))
+
+	fixt := []Fixture{
+		// Simple dates
+		{"встреча 15 января 2024", 15, "15 января 2024", time.Date(2024, 1, 15, 0, 0, 0, 0, time.UTC).Sub(null)},
+		{"5 марта 2025 запланирована встреча", 0, "5 марта 2025", time.Date(2025, 3, 5, 0, 0, 0, 0, time.UTC).Sub(null)},
+		{"31 декабря 2023", 0, "31 декабря 2023", time.Date(2023, 12, 31, 0, 0, 0, 0, time.UTC).Sub(null)},
+
+		// Dates with time
+		{"15 января 2024 в 9:30", 0, "15 января 2024 в 9:30", time.Date(2024, 1, 15, 9, 30, 0, 0, time.UTC).Sub(null)},
+		{"5 марта 2025 в 15:00 запланирована встреча", 0, "5 марта 2025 в 15:00", time.Date(2025, 3, 5, 15, 0, 0, 0, time.UTC).Sub(null)},
+		{"31 декабря 2023 в 23:59", 0, "31 декабря 2023 в 23:59", time.Date(2023, 12, 31, 23, 59, 0, 0, time.UTC).Sub(null)},
+	}
+
+	ApplyFixtures(t, "ru.Date", w, fixt)
+}
+
+func TestDateNil(t *testing.T) {
+	w := when.New(nil)
+	w.Add(ru.Date(rules.Override))
+
+	fixt := []Fixture{
+		{"это текст без даты", 0, "", 0},
+		{"15", 0, "", 0},
+		{"15 чего-то", 0, "", 0},
+	}
+
+	ApplyFixturesNil(t, "ru.Date nil", w, fixt)
+}
diff --git a/rules/ru/dot_date_time.go b/rules/ru/dot_date_time.go
@@ -0,0 +1,61 @@
+package ru
+
+import (
+	"regexp"
+	"strconv"
+	"time"
+
+	"github.com/olebedev/when/rules"
+	"github.com/pkg/errors"
+)
+
+// https://go.dev/play/p/vRzLhHHupUJ
+
+func DotDateTime(s rules.Strategy) rules.Rule {
+	return &rules.F{
+		RegExp: regexp.MustCompile(`(?i)(?:^|\b)(\d{2})\.(\d{2})\.(\d{4})(?:\s+(\d{2}):(\d{2}))?(?:\b|$)`),
+		Applier: func(m *rules.Match, c *rules.Context, o *rules.Options, ref time.Time) (bool, error) {
+			if (c.Day != nil || c.Month != nil || c.Year != nil || c.Hour != nil || c.Minute != nil) && s != rules.Override {
+				return false, nil
+			}
+
+			day, err := strconv.Atoi(m.Captures[0])
+			if err != nil {
+				return false, errors.Wrap(err, "dot date time rule: day")
+			}
+
+			month, err := strconv.Atoi(m.Captures[1])
+			if err != nil {
+				return false, errors.Wrap(err, "dot date time rule: month")
+			}
+
+			year, err := strconv.Atoi(m.Captures[2])
+			if err != nil {
+				return false, errors.Wrap(err, "dot date time rule: year")
+			}
+
+			hour, minute := 0, 0
+			if m.Captures[3] != "" && m.Captures[4] != "" {
+				hour, err = strconv.Atoi(m.Captures[3])
+				if err != nil {
+					return false, errors.Wrap(err, "dot date time rule: hour")
+				}
+				minute, err = strconv.Atoi(m.Captures[4])
+				if err != nil {
+					return false, errors.Wrap(err, "dot date time rule: minute")
+				}
+			}
+
+			if day > 0 && day <= 31 && month > 0 && month <= 12 {
+				c.Day = &day
+				c.Month = &month
+				c.Year = &year
+				c.Hour = &hour
+				c.Minute = &minute
+				return true, nil
+			}
+
+			return false, nil
+		},
+	}
+}
diff --git a/rules/ru/dot_date_time_test.go b/rules/ru/dot_date_time_test.go
@@ -0,0 +1,37 @@
+package ru_test
+
+import (
+	"github.com/olebedev/when"
+	"github.com/olebedev/when/rules"
+	"github.com/olebedev/when/rules/ru"
+	"testing"
+	"time"
+)
+
+func TestDotDateTime(t *testing.T) {
+	w := when.New(nil)
+	w.Add(ru.DotDateTime(rules.Override))
+
+	fixt := []Fixture{
+		// Basic date/time formats
+		{"встреча 15.01.2024 09:30", 15, "15.01.2024 09:30", time.Date(2024, 1, 15, 9, 30, 0, 0, time.UTC).Sub(null)},
+		{"05.03.2025 15:00 запланирована встреча", 0, "05.03.2025 15:00", time.Date(2025, 3, 5, 15, 0, 0, 0, time.UTC).Sub(null)},
+		{"31.12.2023 23:59", 0, "31.12.2023 23:59", time.Date(2023, 12, 31, 23, 59, 0, 0, time.UTC).Sub(null)},
+	}
+
+	ApplyFixtures(t, "ru.DateTime", w, fixt)
+}
+
+func TestDotDateTimeNil(t *testing.T) {
+	w := when.New(nil)
+	w.Add(ru.DotDateTime(rules.Override))
+
+	fixt := []Fixture{
+		{"это текст без даты и времени", 0, "", 0},
+		{"15.01", 0, "", 0},
+		{"32.01.2024 15:00", 0, "", 0}, // некорректный день
+		{"15.13.2024 15:00", 0, "", 0}, // некорректный месяц
+	}
+
+	ApplyFixturesNil(t, "ru.DateTime nil", w, fixt)
+}
diff --git a/rules/ru/hour_minute.go b/rules/ru/hour_minute.go
@@ -22,7 +22,7 @@ import (
 	{"11.1pm", 0, "11.1pm", 0},
 	{"11.10 pm", 0, "11.10 pm", 0},
 
-	https://play.golang.org/p/PmPBjHK4PA
+	https://go.dev/play/p/QiSvUkrni6N
 */
 
 // 1. - int
@@ -31,12 +31,12 @@ import (
 
 func HourMinute(s rules.Strategy) rules.Rule {
 	return &rules.F{
-		RegExp: regexp.MustCompile("(?i)(?:\\W|\\D|^)" +
+		RegExp: regexp.MustCompile("(?i)(?:\\A|\\s|\\D)" +
 			"((?:[0-1]{0,1}[0-9])|(?:2[0-3]))" +
 			"(?:\\:|：|\\-|\\.)" +
 			"((?:[0-5][0-9]))" +
 			"(?:\\s*(утра|вечера|дня))?" +
-			"(?:\\P{L}|$)"),
+			"(?:\\s|\\D|\\z)"),
 		Applier: func(m *rules.Match, c *rules.Context, o *rules.Options, ref time.Time) (bool, error) {
 			if (c.Hour != nil || c.Minute != nil) && s != rules.Override {
 				return false, nil
diff --git a/rules/ru/ru.go b/rules/ru/ru.go
@@ -1,6 +1,9 @@
 package ru
 
-import "github.com/olebedev/when/rules"
+import (
+	"github.com/olebedev/when/rules"
+	"time"
+)
 
 var All = []rules.Rule{
 	Weekday(rules.Override),
@@ -9,6 +12,8 @@ var All = []rules.Rule{
 	Hour(rules.Override),
 	HourMinute(rules.Override),
 	Deadline(rules.Override),
+	Date(rules.Override),
+	DotDateTime(rules.Override),
 }
 
 var WEEKDAY_OFFSET = map[string]int{
@@ -18,29 +23,29 @@ var WEEKDAY_OFFSET = map[string]int{
 	"понедельник":  1,
 	"понедельнику": 1,
 	"понедельника": 1,
-	"пн":       1,
-	"вторник":  2,
-	"вторника": 2,
-	"вторнику": 2,
-	"вт":       2,
-	"среда":    3,
-	"среду":    3,
-	"среде":    3,
-	"ср":       3,
-	"четверг":  4,
-	"четверга": 4,
-	"четвергу": 4,
-	"чт":       4,
-	"пятница":  5,
-	"пятнице":  5,
-	"пятницы":  5,
-	"пятницу":  5,
-	"пт":       5,
-	"суббота":  6,
-	"субботы":  6,
-	"субботе":  6,
-	"субботу":  6,
-	"сб":       6,
+	"пн":           1,
+	"вторник":      2,
+	"вторника":     2,
+	"вторнику":     2,
+	"вт":           2,
+	"среда":        3,
+	"среду":        3,
+	"среде":        3,
+	"ср":           3,
+	"четверг":      4,
+	"четверга":     4,
+	"четвергу":     4,
+	"чт":           4,
+	"пятница":      5,
+	"пятнице":      5,
+	"пятницы":      5,
+	"пятницу":      5,
+	"пт":           5,
+	"суббота":      6,
+	"субботы":      6,
+	"субботе":      6,
+	"субботу":      6,
+	"сб":           6,
 }
 
 var WEEKDAY_OFFSET_PATTERN = "(?:воскресенье|воскресенья|воск|понедельник|понедельнику|понедельника|пн|вторник|вторника|вторнику|вт|среда|среду|среде|ср|четверг|четверга|четвергу|чт|пятница|пятнице|пятницы|пятницу|пт|суббота|субботы|субботе|субботу|сб)"
@@ -65,3 +70,20 @@ var INTEGER_WORDS = map[string]int{
 }
 
 var INTEGER_WORDS_PATTERN = `(?:час|один|одну|одного|два|две|три|четыре|пять|шесть|семь|восемь|девять|десять|одиннадцать|двенадцать)`
+
+var MONTHS = map[string]time.Month{
+	"января":   time.January,
+	"февраля":  time.February,
+	"марта":    time.March,
+	"апреля":   time.April,
+	"мая":      time.May,
+	"июня":     time.June,
+	"июля":     time.July,
+	"августа":  time.August,
+	"сентября": time.September,
+	"октября":  time.October,
+	"ноября":   time.November,
+	"декабря":  time.December,
+}
+
+var MONTHS_PATTERN = `(?:января|февраля|марта|апреля|мая|июня|июля|августа|сентября|октября|ноября|декабря)`
diff --git a/rules/ru/ru_test.go b/rules/ru/ru_test.go
@@ -66,6 +66,18 @@ func TestAll(t *testing.T) {
 		{"написать письмо до утра субботы ", 30, "до утра субботы", ((3 * 24) + 8) * time.Hour},
 		{"написать письмо к субботе после обеда ", 30, "к субботе после обеда", ((3 * 24) + 15) * time.Hour},
 		{"В субботу вечером", 0, "В субботу вечером", ((3 * 24) + 18) * time.Hour},
+
+		{"встреча 15 января 2024", 15, "15 января 2024", time.Date(2024, 1, 15, 0, 0, 0, 0, time.UTC).Sub(null)},
+		{"5 марта 2025 запланирована встреча", 0, "5 марта 2025", time.Date(2025, 3, 5, 0, 0, 0, 0, time.UTC).Sub(null)},
+		{"31 декабря 2023", 0, "31 декабря 2023", time.Date(2023, 12, 31, 0, 0, 0, 0, time.UTC).Sub(null)},
+		{"15 января 2024 в 9:30", 0, "15 января 2024 в 9:30", time.Date(2024, 1, 15, 9, 30, 0, 0, time.UTC).Sub(null)},
+		{"5 марта 2025 в 15:00 запланирована встреча", 0, "5 марта 2025 в 15:00", time.Date(2025, 3, 5, 15, 0, 0, 0, time.UTC).Sub(null)},
+		{"31 декабря 2023 в 23:59", 0, "31 декабря 2023 в 23:59", time.Date(2023, 12, 31, 23, 59, 0, 0, time.UTC).Sub(null)},
+		{"31 декабря", 0, "31 декабря", time.Date(time.Now().Year(), 12, 31, 0, 0, 0, 0, time.UTC).Sub(null)},
+		{"встреча 15.01.2024 09:30", 15, "15.01.2024 09:30", time.Date(2024, 1, 15, 9, 30, 0, 0, time.UTC).Sub(null)},
+		{"05.03.2025 15:00 запланирована встреча", 0, "05.03.2025 15:00", time.Date(2025, 3, 5, 15, 0, 0, 0, time.UTC).Sub(null)},
+		{"31.12.2023 23:59", 0, "31.12.2023 23:59", time.Date(2023, 12, 31, 23, 59, 0, 0, time.UTC).Sub(null)},
+		{"31.12.2023", 0, "31.12.2023", time.Date(2023, 12, 31, 0, 0, 0, 0, time.UTC).Sub(null)},
 	}
 
 	ApplyFixtures(t, "ru.All...", w, fixt)
diff --git a/rules/zh/casual_date_test.go b/rules/zh/casual_date_test.go
@@ -10,8 +10,6 @@ import (
 )
 
 func TestCasualDate(t *testing.T) {
-	// current is Monday
-	now := time.Now()
 	fixt := []Fixture{
 		{"后天中午", 0, "后天", (2 * 24) * time.Hour},
 		{"大后天中午", 0, "大后天", (3 * 24) * time.Hour},

Original file line number	Diff line number	Diff line change
`@@ -10,8 +10,6 @@ import (`
`10`	`10`	`)`
`11`	`11`
`12`	`12`	`func TestCasualDate(t *testing.T) {`
`13`		`- // current is Monday`
`14`		`- now := time.Now()`
`15`	`13`	`fixt := []Fixture{`
`16`	`14`	`{"后天中午", 0, "后天", (2 * 24) * time.Hour},`
`17`	`15`	`{"大后天中午", 0, "大后天", (3 * 24) * time.Hour},`