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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

use crate::dimensions::numeral::helpers::numeral_data;
use crate::pattern::{dim, predicate, regex};
use crate::types::{DimensionKind, Rule, TokenData};

use super::{TemperatureData, TemperatureUnit};

fn temperature_data(td: &TokenData) -> Option<&TemperatureData> {
    match td {
        TokenData::Temperature(d) => Some(d),
        _ => None,
    }
}

fn is_value_only(allow_degree: bool) -> crate::types::PatternItem {
    predicate(move |td| match td {
        TokenData::Temperature(d) => {
            d.value.is_some()
                && d.min_value.is_none()
                && d.max_value.is_none()
                && (d.unit.is_none() || (allow_degree && d.unit == Some(TemperatureUnit::Degree)))
        }
        _ => false,
    })
}

fn is_simple_temperature(td: &TokenData) -> bool {
    matches!(td, TokenData::Temperature(d) if d.value.is_some() && d.unit.is_some() && d.min_value.is_none() && d.max_value.is_none())
}

pub fn rules() -> Vec<Rule> {
    vec![
        Rule {
            name: "number as temp".to_string(),
            pattern: vec![dim(DimensionKind::Numeral)],
            production: Box::new(|nodes| {
                let n = numeral_data(&nodes[0].token_data)?;
                Some(TokenData::Temperature(TemperatureData::new(n.value)))
            }),
        },
        Rule {
            name: "<latent temp> градус".to_string(),
            pattern: vec![is_value_only(false), regex("градус|°|хэм")],
            production: Box::new(|nodes| {
                let td = temperature_data(&nodes[0].token_data)?;
                Some(TokenData::Temperature(
                    td.clone().with_unit(TemperatureUnit::Degree),
                ))
            }),
        },
        Rule {
            name: "<temp> Celsius".to_string(),
            pattern: vec![is_value_only(true), regex("c(el[cs]?(ius)?)?\\.?")],
            production: Box::new(|nodes| {
                let td = temperature_data(&nodes[0].token_data)?;
                Some(TokenData::Temperature(
                    td.clone().with_unit(TemperatureUnit::Celsius),
                ))
            }),
        },
        Rule {
            name: "<temp> °C".to_string(),
            pattern: vec![is_value_only(true), regex("c")],
            production: Box::new(|nodes| {
                let td = temperature_data(&nodes[0].token_data)?;
                Some(TokenData::Temperature(
                    td.clone().with_unit(TemperatureUnit::Celsius),
                ))
            }),
        },
        Rule {
            name: "<temp> Fahrenheit".to_string(),
            pattern: vec![
                is_value_only(true),
                regex("((f(ah?rh?eh?n(h?eit)?)?\\.?)|фарангейт)"),
            ],
            production: Box::new(|nodes| {
                let td = temperature_data(&nodes[0].token_data)?;
                Some(TokenData::Temperature(
                    td.clone().with_unit(TemperatureUnit::Fahrenheit),
                ))
            }),
        },
        Rule {
            name: "<temp> below zero".to_string(),
            pattern: vec![is_value_only(true), regex("тэгээс доош")],
            production: Box::new(|nodes| {
                let mut td = temperature_data(&nodes[0].token_data)?.clone();
                td.value = Some(-td.value?);
                if td.unit.is_none() {
                    td.unit = Some(TemperatureUnit::Degree);
                }
                Some(TokenData::Temperature(td))
            }),
        },
        Rule {
            name: "under/less/lower/no more than <temp>".to_string(),
            pattern: vec![
                regex("доогуур|(бага|ихгүй|их биш)"),
                predicate(is_simple_temperature),
            ],
            production: Box::new(|nodes| {
                let d = temperature_data(&nodes[1].token_data)?;
                Some(TokenData::Temperature(
                    TemperatureData::unit_only(d.unit?).with_max(d.value?),
                ))
            }),
        },
        Rule {
            name: "over/above/at least/more than <temp>".to_string(),
            pattern: vec![
                regex("дээгүүр|их|багадаа"),
                predicate(is_simple_temperature),
            ],
            production: Box::new(|nodes| {
                let d = temperature_data(&nodes[1].token_data)?;
                Some(TokenData::Temperature(
                    TemperatureData::unit_only(d.unit?).with_min(d.value?),
                ))
            }),
        },
    ]
}