use super::left_hand_side::LeftHandSideExpression;
use crate::{
profiler::BoaProfiler,
syntax::{
ast::{node, op::UnaryOp, Node, Punctuator},
lexer::TokenKind,
parser::{AllowAwait, AllowYield, Cursor, ParseError, ParseResult, TokenParser},
},
};
use std::io::Read;
#[derive(Debug, Clone, Copy)]
pub(super) struct UpdateExpression {
allow_yield: AllowYield,
allow_await: AllowAwait,
}
impl UpdateExpression {
pub(super) fn new<Y, A>(allow_yield: Y, allow_await: A) -> Self
where
Y: Into<AllowYield>,
A: Into<AllowAwait>,
{
Self {
allow_yield: allow_yield.into(),
allow_await: allow_await.into(),
}
}
}
impl<R> TokenParser<R> for UpdateExpression
where
R: Read,
{
type Output = Node;
fn parse(self, cursor: &mut Cursor<R>) -> ParseResult {
let _timer = BoaProfiler::global().start_event("UpdateExpression", "Parsing");
let tok = cursor.peek(0)?.ok_or(ParseError::AbruptEnd)?;
match tok.kind() {
TokenKind::Punctuator(Punctuator::Inc) => {
cursor.next()?.expect("Punctuator::Inc token disappeared");
return Ok(node::UnaryOp::new(
UnaryOp::IncrementPre,
LeftHandSideExpression::new(self.allow_yield, self.allow_await)
.parse(cursor)?,
)
.into());
}
TokenKind::Punctuator(Punctuator::Dec) => {
cursor.next()?.expect("Punctuator::Dec token disappeared");
return Ok(node::UnaryOp::new(
UnaryOp::DecrementPre,
LeftHandSideExpression::new(self.allow_yield, self.allow_await)
.parse(cursor)?,
)
.into());
}
_ => {}
}
let lhs = LeftHandSideExpression::new(self.allow_yield, self.allow_await).parse(cursor)?;
if let Some(tok) = cursor.peek(0)? {
match tok.kind() {
TokenKind::Punctuator(Punctuator::Inc) => {
cursor.next()?.expect("Punctuator::Inc token disappeared");
return Ok(node::UnaryOp::new(UnaryOp::IncrementPost, lhs).into());
}
TokenKind::Punctuator(Punctuator::Dec) => {
cursor.next()?.expect("Punctuator::Dec token disappeared");
return Ok(node::UnaryOp::new(UnaryOp::DecrementPost, lhs).into());
}
_ => {}
}
}
Ok(lhs)
}
}