use nom::branch::alt;
use nom::bytes::complete::tag;
use nom::combinator::map;
use crate::parsers::{ParseResult, Span};
use crate::types::{multi_op::names, MultiOp};
pub fn parse_multi_op<'a, T: Into<Span<'a>>>(input: T) -> ParseResult<'a, MultiOp> {
map(
alt((tag(names::MULTIPLICATION), tag(names::ADDITION))),
|x: Span| MultiOp::try_from(*x.fragment()).expect("unhandled variant"),
)(input.into())
}
impl crate::parsers::Parser for MultiOp {
type Item = MultiOp;
fn parse<'a, S: Into<Span<'a>>>(input: S) -> ParseResult<'a, Self::Item> {
parse_multi_op(input)
}
}
#[cfg(test)]
mod tests {
use crate::parsers::UnwrapValue;
use crate::{MultiOp, Parser};
#[test]
fn test_parse() {
assert!(MultiOp::parse("*").is_value(MultiOp::Multiplication));
assert!(MultiOp::parse("+").is_value(MultiOp::Addition));
}
}