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

use std::fmt;

use crate::{ast, utility::parsing};
use nom::{combinator::map, IResult};
use serde::{Deserialize, Serialize};

/// An integer literal.
#[derive(Debug, Eq, PartialEq, Clone, Hash, Serialize, Deserialize)]
pub struct IntegerLiteral(pub i64);

impl From<i64> for IntegerLiteral {
    fn from(i: i64) -> Self {
        IntegerLiteral(i)
    }
}

impl From<ast::expression::integer_literal::IntegerLiteral> for IntegerLiteral {
    fn from(i: ast::expression::IntegerLiteral) -> Self {
        Self(i.0)
    }
}

impl fmt::Display for IntegerLiteral {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        self.0.fmt(f)
    }
}

/// Parse ir code to get an [`IntegerLiteral`].
pub fn parse(code: &str) -> IResult<&str, IntegerLiteral> {
    map(parsing::integer, IntegerLiteral)(code)
}

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

    #[test]
    fn test_parse() {
        assert_eq!(parse("123"), Ok(("", IntegerLiteral(123))));
        assert_eq!(parse("0"), Ok(("", IntegerLiteral(0))));
        assert_eq!(parse("-123"), Ok(("", IntegerLiteral(-123))));
        assert_eq!(parse("123abc"), Ok(("abc", IntegerLiteral(123))));
        assert!(parse("abc").is_err());
    }
}