duckling 0.4.0

A Rust port of Facebook's Duckling library for parsing natural language into structured data
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use super::{TimeData, TimeForm};
use crate::pattern::regex;
use crate::types::{Rule, TokenData};

pub fn rules() -> Vec<Rule> {
    let mut rules = super::en::rules();
    rules.extend(vec![
        Rule {
            name: "now (ro)".to_string(),
            pattern: vec![regex("acum")],
            production: Box::new(|_| Some(TokenData::Time(TimeData::new(TimeForm::Now)))),
        },
        Rule {
            name: "today (ro)".to_string(),
            pattern: vec![regex("azi|astăzi|astazi")],
            production: Box::new(|_| Some(TokenData::Time(TimeData::new(TimeForm::Today)))),
        },
        Rule {
            name: "tomorrow (ro)".to_string(),
            pattern: vec![regex("m[âa]ine")],
            production: Box::new(|_| Some(TokenData::Time(TimeData::new(TimeForm::Tomorrow)))),
        },
        Rule {
            name: "yesterday (ro)".to_string(),
            pattern: vec![regex("ieri")],
            production: Box::new(|_| Some(TokenData::Time(TimeData::new(TimeForm::Yesterday)))),
        },
        Rule {
            name: "day of week (ro)".to_string(),
            pattern: vec![regex("lu(n(ea|i)?)?|ma(r((t|ț)(ea|i))?)?|mi(e(rcur(ea|i))?)?|jo(ia?)?|vi(n(er(ea|i))?)?|s(a|â)mb(a|ă)t(a|ă)|s(a|â)m|du(m(inic(a|ă))?)?")],
            production: Box::new(|nodes| {
                let s = match &nodes[0].token_data {
                    TokenData::RegexMatch(m) => m.group(0)?.to_lowercase(),
                    _ => return None,
                };
                let dow = if s.starts_with("lu") {
                    0
                } else if s.starts_with("ma") {
                    1
                } else if s.starts_with("mi") {
                    2
                } else if s.starts_with("jo") {
                    3
                } else if s.starts_with("vi") {
                    4
                } else if s.starts_with("sâ") || s.starts_with("sa") {
                    5
                } else if s.starts_with("du") {
                    6
                } else {
                    return None;
                };
                Some(TokenData::Time(TimeData::new(TimeForm::DayOfWeek(dow))))
            }),
        },
        Rule {
            name: "lunea asta (ro)".to_string(),
            pattern: vec![regex("lunea asta|lunea aceasta")],
            production: Box::new(|_| Some(TokenData::Time(TimeData::new(TimeForm::DayOfWeek(0))))),
        },
        Rule {
            name: "iunie 19-20 (ro)".to_string(),
            pattern: vec![regex("iunie\\s+(\\d{1,2})-(\\d{1,2})")],
            production: Box::new(|nodes| {
                let d = match &nodes[0].token_data {
                    TokenData::RegexMatch(m) => m.group(1)?,
                    _ => return None,
                };
                let day: u32 = d.parse().ok()?;
                if !(1..=31).contains(&day) {
                    return None;
                }
                Some(TokenData::Time(TimeData::new(TimeForm::DateMDY { month: 6, day, year: None })))
            }),
        },
        Rule {
            name: "christmas day (ro)".to_string(),
            pattern: vec![regex("ziua de craciun|ziua de crăciun")],
            production: Box::new(|_| Some(TokenData::Time(TimeData::new(TimeForm::DateMDY { month: 12, day: 25, year: None })))),
        },
        Rule {
            name: "craciun (ro)".to_string(),
            pattern: vec![regex("craciun|crăciun")],
            production: Box::new(|_| Some(TokenData::Time(TimeData::new(TimeForm::DateMDY { month: 12, day: 25, year: None })))),
        },
    ]);
    rules
}