mango 0.4.0

The Mango programming language (UNDER CONSTRUCTION)
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

use ::std::str::FromStr;

use crate::common::error::{ErrMsg, MsgResult};
use crate::token::Token;
use crate::util::codeparts::Symbol;
use crate::util::encdec::ToText;

/// Equals symbol, which is used for associating a value with an identifier.
/// Also in-place operations like *=, += etc.
#[derive(Debug, PartialEq, Eq, Hash, Clone)]
pub struct AssociationToken {
    //TODO @mark: note that some symbols aren't allowed
    symbol: Option<Symbol>,
}

impl FromStr for AssociationToken {
    type Err = ErrMsg;

    fn from_str(association_txt: &str) -> MsgResult<AssociationToken> {
        if association_txt == "=" {
            return Ok(AssociationToken::from_unprefixed());
        }
        assert!(association_txt.ends_with("="));
        let symbol = Symbol::new(&association_txt[0..association_txt.len() - 1])?;
        AssociationToken::from_symbol(symbol)
    }
}

impl AssociationToken {
    pub fn from_unprefixed() -> Self {
        AssociationToken { symbol: Option::None }
    }

    pub fn from_symbol(symbol: Symbol) -> MsgResult<Self> {
        let is_valid = match symbol {
            Symbol::Plus => true,
            Symbol::Dash => true,
            Symbol::Asterisk => true,
            Symbol::Slash => true,
            Symbol::Exclamation => unimplemented!(),
            Symbol::Question => unimplemented!(),
            // Binary boolean operators are not allowed before '='
            Symbol::LT => false,
            Symbol::GT => false,
            Symbol::EQ => false,
            Symbol::LE => false,
            Symbol::GE => false,
        };
        if !is_valid {
            return Err(ErrMsg::new(format!("Symbol cannot be used as association (before an '='): '{}'", symbol)));
        }
        Ok(AssociationToken {
            symbol: Option::Some(symbol),
        })
    }
}

impl ToText for AssociationToken {
    fn to_text(&self) -> String {
        // LATER: this seems to compile, but IDEA flags it...
        match self.symbol {
            Option::None => "=".to_owned(),
            Option::Some(ref sym) => format!("{}=", sym),
        }
    }
}

impl Token for AssociationToken {}

#[cfg(test)]
mod from_str {
    use super::*;
    use crate::token::Tokens;
    use crate::common::tests::assert_panic_silent;

    #[test]
    fn empty() {
        assert_panic_silent(|| AssociationToken::from_str(""));
    }

    #[test]
    fn mismatch() {
        let err = AssociationToken::from_str("abc=").unwrap_err();
        assert!(err.as_str().to_lowercase().contains("unknown symbol"));
        assert!(err.as_str().to_lowercase().contains("abc"));
    }

    #[test]
    fn valid() {
        assert_eq!(AssociationToken::from_str("+=").unwrap(), AssociationToken::from_symbol(Symbol::Plus).unwrap());
        assert_eq!(AssociationToken::from_str("-=").unwrap(), AssociationToken::from_symbol(Symbol::Dash).unwrap());
        assert_eq!(AssociationToken::from_str("*=").unwrap(), AssociationToken::from_symbol(Symbol::Asterisk).unwrap());
        assert_eq!(AssociationToken::from_str("/=").unwrap(), AssociationToken::from_symbol(Symbol::Slash).unwrap());
        //assert_eq!(AssociationToken::from_str("?=").unwrap(), AssociationToken::from_symbol(Symbol::Exclamation));
        //assert_eq!(AssociationToken::from_str("!=").unwrap(), AssociationToken::from_symbol(Symbol::Question));
    }

    #[test]
    fn invalid() {
        assert!(AssociationToken::from_str("===").is_err());
        assert!(AssociationToken::from_str("<=").is_err());
        assert!(AssociationToken::from_str(">=").is_err());
        assert!(AssociationToken::from_str("<==").is_err());
        assert!(AssociationToken::from_str(">==").is_err());
    }
}