use crate::{
Error,
lexer::TokenKind,
parser::{
AllowAwait, AllowYield, OrAbrupt, ParseResult, TokenParser, cursor::Cursor,
expression::Expression,
},
source::ReadChar,
};
use boa_ast::{self as ast, PositionGroup, Punctuator, Span, Spanned, expression::TaggedTemplate};
use boa_interner::Interner;
#[derive(Debug, Clone)]
pub(super) struct TaggedTemplateLiteral {
allow_yield: AllowYield,
allow_await: AllowAwait,
start: PositionGroup,
tag: ast::Expression,
}
impl TaggedTemplateLiteral {
pub(super) fn new<Y, A>(
allow_yield: Y,
allow_await: A,
start: PositionGroup,
tag: ast::Expression,
) -> Self
where
Y: Into<AllowYield>,
A: Into<AllowAwait>,
{
Self {
allow_yield: allow_yield.into(),
allow_await: allow_await.into(),
start,
tag,
}
}
}
impl<R> TokenParser<R> for TaggedTemplateLiteral
where
R: ReadChar,
{
type Output = TaggedTemplate;
fn parse(self, cursor: &mut Cursor<R>, interner: &mut Interner) -> ParseResult<Self::Output> {
let mut raws = Vec::new();
let mut cookeds = Vec::new();
let mut exprs = Vec::new();
let mut token = cursor.next(interner).or_abrupt()?;
loop {
match token.kind() {
TokenKind::TemplateMiddle(template_string) => {
raws.push(template_string.raw());
cookeds.push(template_string.cooked());
exprs.push(
Expression::new(true, self.allow_yield, self.allow_await)
.parse(cursor, interner)?,
);
cursor.expect(
TokenKind::Punctuator(Punctuator::CloseBlock),
"template literal",
interner,
)?;
}
TokenKind::TemplateNoSubstitution(template_string) => {
raws.push(template_string.raw());
cookeds.push(template_string.cooked());
return Ok(TaggedTemplate::new(
self.tag,
raws.into_boxed_slice(),
cookeds.into_boxed_slice(),
exprs.into_boxed_slice(),
cursor.tagged_template_identifier(),
Span::new(self.start.position(), token.span().end()),
));
}
_ => {
return Err(Error::general(
"cannot parse tagged template literal",
self.start,
));
}
}
token = cursor.lex_template(self.start, interner)?;
}
}
}