summaryrefslogtreecommitdiffstats
path: root/third_party/rust/fluent-bundle/src/entry.rs
blob: 1ac8ecf01b782c96509b4bfdd79d34730d708dc6 (plain)
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
//! `Entry` is used to store Messages, Terms and Functions in `FluentBundle` instances.

use std::borrow::Borrow;

use fluent_syntax::ast;

use crate::args::FluentArgs;
use crate::bundle::FluentBundle;
use crate::resource::FluentResource;
use crate::types::FluentValue;

pub type FluentFunction =
    Box<dyn for<'a> Fn(&[FluentValue<'a>], &FluentArgs) -> FluentValue<'a> + Send + Sync>;

pub enum Entry {
    Message((usize, usize)),
    Term((usize, usize)),
    Function(FluentFunction),
}

pub trait GetEntry {
    fn get_entry_message(&self, id: &str) -> Option<&ast::Message<&str>>;
    fn get_entry_term(&self, id: &str) -> Option<&ast::Term<&str>>;
    fn get_entry_function(&self, id: &str) -> Option<&FluentFunction>;
}

impl<'bundle, R: Borrow<FluentResource>, M> GetEntry for FluentBundle<R, M> {
    fn get_entry_message(&self, id: &str) -> Option<&ast::Message<&str>> {
        self.entries.get(id).and_then(|ref entry| match entry {
            Entry::Message(pos) => {
                let res = self.resources.get(pos.0)?.borrow();
                if let ast::Entry::Message(ref msg) = res.get_entry(pos.1)? {
                    Some(msg)
                } else {
                    None
                }
            }
            _ => None,
        })
    }

    fn get_entry_term(&self, id: &str) -> Option<&ast::Term<&str>> {
        self.entries.get(id).and_then(|ref entry| match entry {
            Entry::Term(pos) => {
                let res = self.resources.get(pos.0)?.borrow();
                if let ast::Entry::Term(ref msg) = res.get_entry(pos.1)? {
                    Some(msg)
                } else {
                    None
                }
            }
            _ => None,
        })
    }

    fn get_entry_function(&self, id: &str) -> Option<&FluentFunction> {
        self.entries.get(id).and_then(|ref entry| match entry {
            Entry::Function(function) => Some(function),
            _ => None,
        })
    }
}