summaryrefslogtreecommitdiff
path: root/src/syntax/scope.rs
blob: aac2b1b8077d13c85c5fed0f0a922c514ed1242f (plain) (blame)
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
//! Mapping of function names to function parsers.

use std::collections::HashMap;
use std::fmt::{self, Debug, Formatter};

use super::parsing::{CallParser, ParseCall};
use super::tree::DynamicNode;

/// A map from identifiers to function parsers.
pub struct Scope {
    parsers: HashMap<String, Box<CallParser>>,
    fallback: Box<CallParser>,
}

impl Scope {
    /// Create a new empty scope with a fallback parser that is invoked when no
    /// match is found.
    pub fn new<F>() -> Self
    where
        F: ParseCall<Meta = ()> + DynamicNode + 'static
    {
        Self {
            parsers: HashMap::new(),
            fallback: make_parser::<F>(()),
        }
    }

    /// Associate the given function name with a dynamic node type.
    pub fn add<F>(&mut self, name: &str)
    where
        F: ParseCall<Meta = ()> + DynamicNode + 'static
    {
        self.add_with_meta::<F>(name, ());
    }

    /// Add a dynamic node type with additional metadata that is passed to the
    /// parser.
    pub fn add_with_meta<F>(&mut self, name: &str, metadata: <F as ParseCall>::Meta)
    where
        F: ParseCall + DynamicNode + 'static
    {
        self.parsers.insert(name.to_string(), make_parser::<F>(metadata));
    }

    /// Return the parser with the given name if there is one.
    pub fn get_parser(&self, name: &str) -> Option<&CallParser> {
        self.parsers.get(name).map(AsRef::as_ref)
    }

    /// Return the fallback parser.
    pub fn get_fallback_parser(&self) -> &CallParser {
        &*self.fallback
    }
}

impl Debug for Scope {
    fn fmt(&self, f: &mut Formatter) -> fmt::Result {
        f.debug_set().entries(self.parsers.keys()).finish()
    }
}

fn make_parser<F>(metadata: <F as ParseCall>::Meta) -> Box<CallParser>
where
    F: ParseCall + DynamicNode + 'static,
{
    Box::new(move |f, s| {
        F::parse(f, s, metadata.clone())
            .map(|tree| Box::new(tree) as Box<dyn DynamicNode>)
    })
}