28 lines
1.0 KiB
Rust
28 lines
1.0 KiB
Rust
|
|
use crate::parser::{NyashParser, ParseError};
|
||
|
|
use crate::parser::common::ParserUtils;
|
||
|
|
use crate::tokenizer::TokenType;
|
||
|
|
use crate::ast::{ASTNode, BinaryOperator, Span};
|
||
|
|
|
||
|
|
impl NyashParser {
|
||
|
|
pub(crate) fn expr_parse_shift(&mut self) -> Result<ASTNode, ParseError> {
|
||
|
|
let mut expr = self.expr_parse_factor()?;
|
||
|
|
loop {
|
||
|
|
if self.match_token(&TokenType::ShiftLeft) {
|
||
|
|
self.advance();
|
||
|
|
let rhs = self.expr_parse_factor()?;
|
||
|
|
expr = ASTNode::BinaryOp { operator: BinaryOperator::Shl, left: Box::new(expr), right: Box::new(rhs), span: Span::unknown() };
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
if self.match_token(&TokenType::ShiftRight) {
|
||
|
|
self.advance();
|
||
|
|
let rhs = self.expr_parse_factor()?;
|
||
|
|
expr = ASTNode::BinaryOp { operator: BinaryOperator::Shr, left: Box::new(expr), right: Box::new(rhs), span: Span::unknown() };
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
Ok(expr)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|