use super::arguments::Arguments;
use crate::{
syntax::{
ast::{
node::{
field::{GetConstField, GetField},
Call, New, Node,
},
Keyword, Punctuator,
},
lexer::TokenKind,
parser::{
expression::{primary::PrimaryExpression, Expression},
AllowAwait, AllowYield, Cursor, ParseError, ParseResult, TokenParser,
},
},
BoaProfiler,
};
use std::io::Read;
#[derive(Debug, Clone, Copy)]
pub(super) struct MemberExpression {
allow_yield: AllowYield,
allow_await: AllowAwait,
}
impl MemberExpression {
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 MemberExpression
where
R: Read,
{
type Output = Node;
fn parse(self, cursor: &mut Cursor<R>) -> ParseResult {
let _timer = BoaProfiler::global().start_event("MemberExpression", "Parsing");
let mut lhs = if cursor.peek(0)?.ok_or(ParseError::AbruptEnd)?.kind()
== &TokenKind::Keyword(Keyword::New)
{
let _ = cursor.next().expect("new keyword disappeared");
let lhs = self.parse(cursor)?;
let args = Arguments::new(self.allow_yield, self.allow_await).parse(cursor)?;
let call_node = Call::new(lhs, args);
Node::from(New::from(call_node))
} else {
PrimaryExpression::new(self.allow_yield, self.allow_await).parse(cursor)?
};
while let Some(tok) = cursor.peek(0)? {
match tok.kind() {
TokenKind::Punctuator(Punctuator::Dot) => {
cursor.next()?.expect("dot punctuator token disappeared");
let token = cursor.next()?.ok_or(ParseError::AbruptEnd)?;
match token.kind() {
TokenKind::Identifier(name) => {
lhs = GetConstField::new(lhs, name.clone()).into()
}
TokenKind::Keyword(kw) => {
lhs = GetConstField::new(lhs, kw.to_string()).into()
}
_ => {
return Err(ParseError::expected(
vec![TokenKind::identifier("identifier")],
token,
"member expression",
));
}
}
}
TokenKind::Punctuator(Punctuator::OpenBracket) => {
cursor
.next()?
.expect("open bracket punctuator token disappeared"); let idx =
Expression::new(true, self.allow_yield, self.allow_await).parse(cursor)?;
cursor.expect(Punctuator::CloseBracket, "member expression")?;
lhs = GetField::new(lhs, idx).into();
}
_ => break,
}
}
Ok(lhs)
}
}