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

use ::lazy_static::lazy_static;
use ::regex::Regex;

use crate::util::strslice::char_ops::CharOps;

#[derive(Debug)]
pub enum IntParseFailReason {
    Invalid,
    Overflow,
    Underflow,
}

lazy_static! {
    /// This matches integer literals, either just numbers in base 10, or base 2-36 with prefix.
    /// The syntax for -37 in base 16 is -16b25 and 2748 is 16bABC.
    /// Incorrect values like 4b7 or 0b0 are not handled at the lexing stage.
    pub static ref INT_RE: Regex = Regex::new(r"^(?:(?P<base>(?:\+|-?)[1-9][0-9]*)b(?P<reb_val>(?:_?[0-9a-zA-Z])+)|(?P<b10_val>(?:\+|-?)[0-9](?:_?[0-9])*))").unwrap();
}

/// Convert a String that matches [int_pattern] to an i64 integer. Overflow is possible.
pub fn parse_int(text: &str) -> Result<i64, IntParseFailReason> {
    //TODO @mark: make sure no leftover chars (no $ at the end)
    match INT_RE.captures(text) {
        None => Err(IntParseFailReason::Invalid),
        Some(captures) => {
            //            // Sign
            //            let sign_str = captures.name("sign").unwrap().as_str();
            //            let sign = if sign_str == "+" || sign_str == "" {
            //                1 // positive
            //            } else {
            //                -1 // negative
            //            };
            // Check if base10 or special
            match captures.name("b10_val") {
                None => {
                    // There is a base provided.
                    if let Some(base) = captures.name("base") {
                        if let Some(value) = captures.name("reb_val") {
                            // TODO: implement
                            panic!(format!(
                                "Do not yet know how to deal with {} in base {}",
                                value.as_str().without_char('_'),
                                base.as_str()
                            ))
                        } else {
                            panic!("Expected 'reb_val' match in regex")
                        }
                    } else {
                        panic!("Expected 'base' match in regex")
                    }
                }
                Some(value) => {
                    // This is a 'normal' (base10) value.
                    // TODO: check for over/underflow
                    if value.as_str().len() < text.len() {
                        // Part of `text` did not match the regex, so this input is invalid.
                        return Err(IntParseFailReason::Invalid)
                    }
                    Ok(value.as_str().without_char('_').parse::<i64>().unwrap())
                }
            }
        }
    }
}

// TODO: possibly add a i32 version?

#[cfg(test)]
mod tests {
    use super::parse_int;

    #[test]
    fn parse_b10_int() {
        assert_eq!(42, parse_int("42").unwrap());
        assert_eq!(42, parse_int("4_2").unwrap());
        assert_eq!(123_456_789, parse_int("+1_2_3_4_5_6_7_8_9").unwrap());
        assert_eq!(-123_456_789, parse_int("-123456789").unwrap());
        assert_eq!(0, parse_int("-0").unwrap());
        assert_eq!(-1, parse_int("-1").unwrap());
        // Weird bases with 0 prefix are not supported.
        assert_eq!(9, parse_int("09").unwrap());
    }

    #[test]
    fn invalid_b10_ints() {
        assert!(parse_int("0x9").is_err());
        assert!(parse_int("A").is_err());
        assert!(parse_int("_0").is_err());
        assert!(parse_int("0_").is_err());
        // TODO: over/underflow
    }

    #[test]
    fn parse_based_ints() {
        // TODO: not implemented yet
    }
}