jany 0.3.0

jev x any command: turn loosely ordered words into a command line, defined per command by a schema
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

//! What is attached to one word. Roles are the schema's strings (jind/jurl's enums turned into data).

use serde::Serialize;
use std::collections::BTreeMap;

#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize)]
#[serde(rename_all = "lowercase")]
pub enum Source {
    Rule,
    Jev,
}

/// An amount. `+7d` is fully decided by rules; for `7`, jev answers the unit and the direction.
#[derive(Debug, Clone, PartialEq, Serialize)]
pub struct Amount {
    pub n: f64,
    pub unit: Option<String>,
    /// Some(true) = at least, Some(false) = at most, None = undecided.
    pub at_least: Option<bool>,
}

#[derive(Debug, Clone, Serialize)]
pub struct Token {
    pub text: String,
    pub role: Option<String>,
    pub confidence: f32,
    pub source: Source,
    /// The corrected value (`files` → `f`, `psot` → `POST`). text if none.
    pub fixed: Option<String>,
    pub amount: Option<Amount>,
    pub tags: Vec<String>,
    /// A short note shown by `--explain`.
    pub note: Option<String>,
    /// jev's probability per role (none for words decided by rules). Used as evidence by repair.
    pub probs: Option<BTreeMap<String, f32>>,
}

impl Token {
    pub fn new(text: impl Into<String>) -> Self {
        Token { text: text.into(), role: None, confidence: 0.0, source: Source::Rule, fixed: None, amount: None, tags: Vec::new(), note: None, probs: None }
    }

    pub fn resolved(&self) -> bool {
        self.role.is_some()
    }

    pub fn is(&self, role: &str) -> bool {
        self.role.as_deref() == Some(role)
    }

    pub fn set_rule(&mut self, role: &str) {
        self.role = Some(role.to_string());
        self.confidence = 1.0;
        self.source = Source::Rule;
    }

    pub fn value(&self) -> &str {
        self.fixed.as_deref().unwrap_or(&self.text)
    }

    pub fn has_tag(&self, t: &str) -> bool {
        self.tags.iter().any(|x| x == t)
    }

    pub fn add_tag(&mut self, t: &str) {
        if !self.has_tag(t) {
            self.tags.push(t.to_string());
        }
    }

    /// Prepends a note (`old → new (why); previous note`).
    pub fn prepend_note(&mut self, s: String) {
        self.note = Some(match self.note.take() {
            Some(n) if !n.is_empty() => format!("{s}; {n}"),
            _ => s,
        });
    }

    pub fn role_str(&self) -> &str {
        self.role.as_deref().unwrap_or("?")
    }
}