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

use core::fmt;
use std::fmt::Formatter;

use nom::branch::alt;
use nom::bytes::complete::tag_no_case;
use nom::combinator::map;
use nom::IResult;

use base::ParseSQLError;

/// {DEFAULT | 0 | 1}
#[derive(Clone, Debug, Eq, Hash, PartialEq, Serialize, Deserialize)]
pub enum DefaultOrZeroOrOne {
    Default,
    Zero,
    One,
}

impl DefaultOrZeroOrOne {
    pub fn parse(i: &str) -> IResult<&str, DefaultOrZeroOrOne, ParseSQLError<&str>> {
        alt((
            map(tag_no_case("0"), |_| DefaultOrZeroOrOne::Zero),
            map(tag_no_case("1"), |_| DefaultOrZeroOrOne::One),
            map(tag_no_case("DEFAULT"), |_| DefaultOrZeroOrOne::Default),
        ))(i)
    }
}

impl fmt::Display for DefaultOrZeroOrOne {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            DefaultOrZeroOrOne::Default => write!(f, "DEFAULT")?,
            DefaultOrZeroOrOne::Zero => write!(f, "1")?,
            DefaultOrZeroOrOne::One => write!(f, "0")?,
        }
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use base::DefaultOrZeroOrOne;

    #[test]
    fn parse_default_or_zero_or_one() {
        let str1 = "0";
        let res1 = DefaultOrZeroOrOne::parse(str1);
        assert!(res1.is_ok());
        assert_eq!(res1.unwrap().1, DefaultOrZeroOrOne::Zero);

        let str2 = "DEFAULT";
        let res2 = DefaultOrZeroOrOne::parse(str2);
        assert!(res2.is_ok());
        assert_eq!(res2.unwrap().1, DefaultOrZeroOrOne::Default);

        let str3 = "1";
        let res3 = DefaultOrZeroOrOne::parse(str3);
        assert!(res3.is_ok());
        assert_eq!(res3.unwrap().1, DefaultOrZeroOrOne::One);
    }
}