use super::arguments::Arguments;
use crate::{
syntax::{
ast::{
node::{
field::{GetConstField, GetField},
Call, Node,
},
Punctuator,
},
lexer::TokenKind,
parser::{
expression::Expression, AllowAwait, AllowYield, Cursor, ParseError, ParseResult,
TokenParser,
},
},
BoaProfiler,
};
use std::io::Read;
#[derive(Debug)]
pub(super) struct CallExpression {
allow_yield: AllowYield,
allow_await: AllowAwait,
first_member_expr: Node,
}
impl CallExpression {
pub(super) fn new<Y, A>(allow_yield: Y, allow_await: A, first_member_expr: Node) -> Self
where
Y: Into<AllowYield>,
A: Into<AllowAwait>,
{
Self {
allow_yield: allow_yield.into(),
allow_await: allow_await.into(),
first_member_expr,
}
}
}
impl<R> TokenParser<R> for CallExpression
where
R: Read,
{
type Output = Node;
fn parse(self, cursor: &mut Cursor<R>) -> ParseResult {
let _timer = BoaProfiler::global().start_event("CallExpression", "Parsing");
let token = cursor.peek(0)?.ok_or(ParseError::AbruptEnd)?;
let mut lhs = if token.kind() == &TokenKind::Punctuator(Punctuator::OpenParen) {
let args = Arguments::new(self.allow_yield, self.allow_await).parse(cursor)?;
Node::from(Call::new(self.first_member_expr, args))
} else {
let next_token = cursor.next()?.expect("token vanished");
return Err(ParseError::expected(
vec![TokenKind::Punctuator(Punctuator::OpenParen)],
next_token,
"call expression",
));
};
while let Some(tok) = cursor.peek(0)? {
let token = tok.clone();
match token.kind() {
TokenKind::Punctuator(Punctuator::OpenParen) => {
let args = Arguments::new(self.allow_yield, self.allow_await).parse(cursor)?;
lhs = Node::from(Call::new(lhs, args));
}
TokenKind::Punctuator(Punctuator::Dot) => {
cursor.next()?.ok_or(ParseError::AbruptEnd)?;
match &cursor.next()?.ok_or(ParseError::AbruptEnd)?.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,
"call expression",
));
}
}
}
TokenKind::Punctuator(Punctuator::OpenBracket) => {
let _ = cursor.next()?.ok_or(ParseError::AbruptEnd)?; let idx =
Expression::new(true, self.allow_yield, self.allow_await).parse(cursor)?;
cursor.expect(Punctuator::CloseBracket, "call expression")?;
lhs = GetField::new(lhs, idx).into();
}
_ => break,
}
}
Ok(lhs)
}
}