oxc/crates/oxc_ast/src/ast_builder.rs

1667 lines
49 KiB
Rust

#![allow(
clippy::fn_params_excessive_bools,
clippy::must_use_candidate, // must_use_candidate is too annoying for this file
clippy::too_many_arguments,
clippy::unused_self,
)]
use num_bigint::BigInt;
use std::mem;
use oxc_allocator::{Allocator, Box, String, Vec};
use oxc_span::{Atom, GetSpan, SourceType, Span};
use oxc_syntax::{
operator::{
AssignmentOperator, BinaryOperator, LogicalOperator, UnaryOperator, UpdateOperator,
},
NumberBase,
};
#[allow(clippy::wildcard_imports)]
use crate::ast::*;
/// AST builder for creating AST nodes
pub struct AstBuilder<'a> {
pub allocator: &'a Allocator,
}
impl<'a> AstBuilder<'a> {
pub fn new(allocator: &'a Allocator) -> Self {
Self { allocator }
}
#[inline]
pub fn alloc<T>(&self, value: T) -> Box<'a, T> {
Box(self.allocator.alloc(value))
}
#[inline]
pub fn new_vec<T>(&self) -> Vec<'a, T> {
Vec::new_in(self.allocator)
}
#[inline]
pub fn new_vec_with_capacity<T>(&self, capacity: usize) -> Vec<'a, T> {
Vec::with_capacity_in(capacity, self.allocator)
}
#[inline]
pub fn new_vec_single<T>(&self, value: T) -> Vec<'a, T> {
let mut vec = self.new_vec_with_capacity(1);
vec.push(value);
vec
}
#[inline]
pub fn new_str(&self, value: &str) -> &'a str {
String::from_str_in(value, self.allocator).into_bump_str()
}
pub fn copy<T>(&self, src: &T) -> T {
// SAFETY:
// This should be safe as long as `src` is an reference from the allocator.
// But honestly, I'm not really sure if this is safe.
unsafe { std::mem::transmute_copy(src) }
}
/// Moves the expression out by replacing it with a null expression.
pub fn move_expression(&self, expr: &mut Expression<'a>) -> Expression<'a> {
let null_literal = NullLiteral::new(expr.span());
let null_expr = self.literal_null_expression(null_literal);
mem::replace(expr, null_expr)
}
pub fn move_statement(&self, stmt: &mut Statement<'a>) -> Statement<'a> {
let empty_stmt = self.empty_statement(stmt.span());
mem::replace(stmt, empty_stmt)
}
pub fn move_statement_vec(&self, stmts: &mut Vec<'a, Statement<'a>>) -> Vec<'a, Statement<'a>> {
mem::replace(stmts, self.new_vec())
}
pub fn move_assignment_target(
&self,
target: &mut AssignmentTarget<'a>,
) -> AssignmentTarget<'a> {
let ident = IdentifierReference::new(Span::default(), "".into());
let dummy = self.simple_assignment_target_identifier(ident);
let dummy = AssignmentTarget::SimpleAssignmentTarget(dummy);
mem::replace(target, dummy)
}
pub fn program(
&self,
span: Span,
source_type: SourceType,
directives: Vec<'a, Directive>,
hashbang: Option<Hashbang>,
body: Vec<'a, Statement<'a>>,
) -> Program<'a> {
Program { span, source_type, directives, hashbang, body }
}
/* ---------- Constructors ---------- */
/// `void 0`
pub fn void_0(&self) -> Expression<'a> {
let left = self.number_literal(Span::default(), 0.0, "0", NumberBase::Decimal);
let num = self.literal_number_expression(left);
self.unary_expression(Span::default(), UnaryOperator::Void, num)
}
/* ---------- Literals ---------- */
pub fn number_literal(
&self,
span: Span,
value: f64,
raw: &'a str,
base: NumberBase,
) -> NumberLiteral<'a> {
NumberLiteral { span, value, raw, base }
}
pub fn boolean_literal(&self, span: Span, value: bool) -> BooleanLiteral {
BooleanLiteral { span, value }
}
pub fn bigint_literal(&self, span: Span, value: BigInt) -> BigintLiteral {
BigintLiteral { span, value }
}
pub fn template_literal(
&self,
span: Span,
quasis: Vec<'a, TemplateElement>,
expressions: Vec<'a, Expression<'a>>,
) -> TemplateLiteral<'a> {
TemplateLiteral { span, quasis, expressions }
}
pub fn template_element(
&self,
span: Span,
tail: bool,
value: TemplateElementValue,
) -> TemplateElement {
TemplateElement { span, tail, value }
}
pub fn template_element_value(&self, raw: Atom, cooked: Option<Atom>) -> TemplateElementValue {
TemplateElementValue { raw, cooked }
}
pub fn reg_exp_literal(&self, span: Span, pattern: Atom, flags: RegExpFlags) -> RegExpLiteral {
RegExpLiteral { span, value: EmptyObject, regex: RegExp { pattern, flags } }
}
pub fn literal_string_expression(&self, literal: StringLiteral) -> Expression<'a> {
Expression::StringLiteral(self.alloc(literal))
}
pub fn literal_boolean_expression(&self, literal: BooleanLiteral) -> Expression<'a> {
Expression::BooleanLiteral(self.alloc(literal))
}
pub fn literal_null_expression(&self, literal: NullLiteral) -> Expression<'a> {
Expression::NullLiteral(self.alloc(literal))
}
pub fn literal_regexp_expression(&self, literal: RegExpLiteral) -> Expression<'a> {
Expression::RegExpLiteral(self.alloc(literal))
}
pub fn literal_number_expression(&self, literal: NumberLiteral<'a>) -> Expression<'a> {
Expression::NumberLiteral(self.alloc(literal))
}
pub fn literal_bigint_expression(&self, literal: BigintLiteral) -> Expression<'a> {
Expression::BigintLiteral(self.alloc(literal))
}
pub fn literal_template_expression(&self, literal: TemplateLiteral<'a>) -> Expression<'a> {
Expression::TemplateLiteral(self.alloc(literal))
}
pub fn identifier_reference_expression(&self, ident: IdentifierReference) -> Expression<'a> {
Expression::Identifier(self.alloc(ident))
}
/* ---------- Statements ---------- */
pub fn directive(&self, span: Span, expression: StringLiteral, directive: Atom) -> Directive {
Directive { span, expression, directive }
}
pub fn hashbang(&self, span: Span, value: Atom) -> Hashbang {
Hashbang { span, value }
}
pub fn block(&self, span: Span, body: Vec<'a, Statement<'a>>) -> Box<'a, BlockStatement<'a>> {
self.alloc(BlockStatement { span, body })
}
pub fn block_statement(&self, block: Box<'a, BlockStatement<'a>>) -> Statement<'a> {
Statement::BlockStatement(
self.alloc(BlockStatement { span: block.span, body: block.unbox().body }),
)
}
pub fn break_statement(&self, span: Span, label: Option<LabelIdentifier>) -> Statement<'a> {
Statement::BreakStatement(self.alloc(BreakStatement { span, label }))
}
pub fn continue_statement(&self, span: Span, label: Option<LabelIdentifier>) -> Statement<'a> {
Statement::ContinueStatement(self.alloc(ContinueStatement { span, label }))
}
pub fn debugger_statement(&self, span: Span) -> Statement<'a> {
Statement::DebuggerStatement(self.alloc(DebuggerStatement { span }))
}
pub fn using_statement(
&self,
span: Span,
declarations: Vec<'a, VariableDeclarator<'a>>,
is_await: bool,
) -> Statement<'a> {
Statement::Declaration(Declaration::UsingDeclaration(self.alloc(UsingDeclaration {
span,
is_await,
declarations,
})))
}
pub fn do_while_statement(
&self,
span: Span,
body: Statement<'a>,
test: Expression<'a>,
) -> Statement<'a> {
Statement::DoWhileStatement(self.alloc(DoWhileStatement { span, body, test }))
}
pub fn empty_statement(&self, span: Span) -> Statement<'a> {
Statement::EmptyStatement(self.alloc(EmptyStatement { span }))
}
pub fn expression_statement(&self, span: Span, expression: Expression<'a>) -> Statement<'a> {
Statement::ExpressionStatement(self.alloc(ExpressionStatement { span, expression }))
}
pub fn for_in_statement(
&self,
span: Span,
left: ForStatementLeft<'a>,
right: Expression<'a>,
body: Statement<'a>,
) -> Statement<'a> {
Statement::ForInStatement(self.alloc(ForInStatement { span, left, right, body }))
}
pub fn for_of_statement(
&self,
span: Span,
r#await: bool,
left: ForStatementLeft<'a>,
right: Expression<'a>,
body: Statement<'a>,
) -> Statement<'a> {
Statement::ForOfStatement(self.alloc(ForOfStatement { span, r#await, left, right, body }))
}
pub fn for_statement(
&self,
span: Span,
init: Option<ForStatementInit<'a>>,
test: Option<Expression<'a>>,
update: Option<Expression<'a>>,
body: Statement<'a>,
) -> Statement<'a> {
Statement::ForStatement(self.alloc(ForStatement { span, init, test, update, body }))
}
pub fn if_statement(
&self,
span: Span,
test: Expression<'a>,
consequent: Statement<'a>,
alternate: Option<Statement<'a>>,
) -> Statement<'a> {
Statement::IfStatement(self.alloc(IfStatement { span, test, consequent, alternate }))
}
pub fn labeled_statement(
&self,
span: Span,
label: LabelIdentifier,
body: Statement<'a>,
) -> Statement<'a> {
Statement::LabeledStatement(self.alloc(LabeledStatement { span, label, body }))
}
pub fn return_statement(&self, span: Span, argument: Option<Expression<'a>>) -> Statement<'a> {
Statement::ReturnStatement(self.alloc(ReturnStatement { span, argument }))
}
pub fn switch_statement(
&self,
span: Span,
discriminant: Expression<'a>,
cases: Vec<'a, SwitchCase<'a>>,
) -> Statement<'a> {
Statement::SwitchStatement(self.alloc(SwitchStatement { span, discriminant, cases }))
}
pub fn switch_case(
&self,
span: Span,
test: Option<Expression<'a>>,
consequent: Vec<'a, Statement<'a>>,
) -> SwitchCase<'a> {
SwitchCase { span, test, consequent }
}
pub fn throw_statement(&self, span: Span, argument: Expression<'a>) -> Statement<'a> {
Statement::ThrowStatement(self.alloc(ThrowStatement { span, argument }))
}
pub fn try_statement(
&self,
span: Span,
block: Box<'a, BlockStatement<'a>>,
handler: Option<Box<'a, CatchClause<'a>>>,
finalizer: Option<Box<'a, BlockStatement<'a>>>,
) -> Statement<'a> {
Statement::TryStatement(self.alloc(TryStatement { span, block, handler, finalizer }))
}
pub fn catch_clause(
&self,
span: Span,
param: Option<BindingPattern<'a>>,
body: Box<'a, BlockStatement<'a>>,
) -> Box<'a, CatchClause<'a>> {
self.alloc(CatchClause { span, param, body })
}
pub fn while_statement(
&self,
span: Span,
test: Expression<'a>,
body: Statement<'a>,
) -> Statement<'a> {
Statement::WhileStatement(self.alloc(WhileStatement { span, test, body }))
}
pub fn with_statement(
&self,
span: Span,
object: Expression<'a>,
body: Statement<'a>,
) -> Statement<'a> {
Statement::WithStatement(self.alloc(WithStatement { span, object, body }))
}
/* ---------- Expressions ---------- */
pub fn super_(&self, span: Span) -> Expression<'a> {
Expression::Super(self.alloc(Super { span }))
}
pub fn meta_property(
&self,
span: Span,
meta: IdentifierName,
property: IdentifierName,
) -> Expression<'a> {
Expression::MetaProperty(self.alloc(MetaProperty { span, meta, property }))
}
pub fn array_expression(
&self,
span: Span,
elements: Vec<'a, ArrayExpressionElement<'a>>,
trailing_comma: Option<Span>,
) -> Expression<'a> {
Expression::ArrayExpression(self.alloc(ArrayExpression { span, elements, trailing_comma }))
}
pub fn arrow_expression(
&self,
span: Span,
expression: bool,
generator: bool,
r#async: bool,
params: Box<'a, FormalParameters<'a>>,
body: Box<'a, FunctionBody<'a>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
return_type: Option<Box<'a, TSTypeAnnotation<'a>>>,
) -> Expression<'a> {
Expression::ArrowExpression(self.alloc(ArrowExpression {
span,
expression,
generator,
r#async,
params,
body,
type_parameters,
return_type,
}))
}
pub fn assignment_expression(
&self,
span: Span,
operator: AssignmentOperator,
left: AssignmentTarget<'a>,
right: Expression<'a>,
) -> Expression<'a> {
Expression::AssignmentExpression(self.alloc(AssignmentExpression {
span,
operator,
left,
right,
}))
}
pub fn simple_assignment_target_identifier(
&self,
ident: IdentifierReference,
) -> SimpleAssignmentTarget<'a> {
SimpleAssignmentTarget::AssignmentTargetIdentifier(self.alloc(ident))
}
pub fn simple_assignment_target_member_expression(
&self,
expr: MemberExpression<'a>,
) -> SimpleAssignmentTarget<'a> {
SimpleAssignmentTarget::MemberAssignmentTarget(self.alloc(expr))
}
pub fn await_expression(&self, span: Span, argument: Expression<'a>) -> Expression<'a> {
Expression::AwaitExpression(self.alloc(AwaitExpression { span, argument }))
}
pub fn binary_expression(
&self,
span: Span,
left: Expression<'a>,
operator: BinaryOperator,
right: Expression<'a>,
) -> Expression<'a> {
Expression::BinaryExpression(self.alloc(BinaryExpression { span, left, operator, right }))
}
pub fn call_expression(
&self,
span: Span,
callee: Expression<'a>,
arguments: Vec<'a, Argument<'a>>,
optional: bool, // for optional chaining
type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
) -> Expression<'a> {
Expression::CallExpression(self.alloc(CallExpression {
span,
callee,
arguments,
optional,
type_parameters,
}))
}
pub fn chain_expression(&self, span: Span, expression: ChainElement<'a>) -> Expression<'a> {
Expression::ChainExpression(self.alloc(ChainExpression { span, expression }))
}
pub fn class_expression(&self, class: Box<'a, Class<'a>>) -> Expression<'a> {
Expression::ClassExpression(class)
}
pub fn conditional_expression(
&self,
span: Span,
test: Expression<'a>,
consequent: Expression<'a>,
alternate: Expression<'a>,
) -> Expression<'a> {
Expression::ConditionalExpression(self.alloc(ConditionalExpression {
span,
test,
consequent,
alternate,
}))
}
pub fn function_expression(&self, function: Box<'a, Function<'a>>) -> Expression<'a> {
Expression::FunctionExpression(function)
}
pub fn import_expression(
&self,
span: Span,
source: Expression<'a>,
arguments: Vec<'a, Expression<'a>>,
) -> Expression<'a> {
Expression::ImportExpression(self.alloc(ImportExpression { span, source, arguments }))
}
pub fn logical_expression(
&self,
span: Span,
left: Expression<'a>,
operator: LogicalOperator,
right: Expression<'a>,
) -> Expression<'a> {
Expression::LogicalExpression(self.alloc(LogicalExpression { span, left, operator, right }))
}
pub fn member_expression(&self, expr: MemberExpression<'a>) -> Expression<'a> {
Expression::MemberExpression(self.alloc(expr))
}
pub fn computed_member(
&self,
span: Span,
object: Expression<'a>,
expression: Expression<'a>,
optional: bool, // for optional chaining
) -> MemberExpression<'a> {
MemberExpression::ComputedMemberExpression(ComputedMemberExpression {
span,
object,
expression,
optional,
})
}
pub fn computed_member_expression(
&self,
span: Span,
object: Expression<'a>,
expression: Expression<'a>,
optional: bool, // for optional chaining
) -> Expression<'a> {
self.member_expression(self.computed_member(span, object, expression, optional))
}
pub fn static_member(
&self,
span: Span,
object: Expression<'a>,
property: IdentifierName,
optional: bool, // for optional chaining
) -> MemberExpression<'a> {
MemberExpression::StaticMemberExpression(StaticMemberExpression {
span,
object,
property,
optional,
})
}
pub fn static_member_expression(
&self,
span: Span,
object: Expression<'a>,
property: IdentifierName,
optional: bool, // for optional chaining
) -> Expression<'a> {
self.member_expression(self.static_member(span, object, property, optional))
}
pub fn private_field_expression(
&self,
span: Span,
object: Expression<'a>,
field: PrivateIdentifier,
optional: bool,
) -> Expression<'a> {
self.member_expression(MemberExpression::PrivateFieldExpression(PrivateFieldExpression {
span,
object,
field,
optional,
}))
}
pub fn new_expression(
&self,
span: Span,
callee: Expression<'a>,
arguments: Vec<'a, Argument<'a>>,
type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
) -> Expression<'a> {
Expression::NewExpression(self.alloc(NewExpression {
span,
callee,
arguments,
type_parameters,
}))
}
pub fn object_expression(
&self,
span: Span,
properties: Vec<'a, ObjectPropertyKind<'a>>,
trailing_comma: Option<Span>,
) -> Expression<'a> {
Expression::ObjectExpression(self.alloc(ObjectExpression {
span,
properties,
trailing_comma,
}))
}
pub fn object_property(
&self,
span: Span,
kind: PropertyKind,
key: PropertyKey<'a>,
value: Expression<'a>,
init: Option<Expression<'a>>,
method: bool,
shorthand: bool,
computed: bool,
) -> Box<'a, ObjectProperty<'a>> {
self.alloc(ObjectProperty { span, kind, key, value, init, method, shorthand, computed })
}
pub fn parenthesized_expression(
&self,
span: Span,
expression: Expression<'a>,
) -> Expression<'a> {
Expression::ParenthesizedExpression(
self.alloc(ParenthesizedExpression { span, expression }),
)
}
pub fn sequence_expression(
&self,
span: Span,
expressions: Vec<'a, Expression<'a>>,
) -> Expression<'a> {
Expression::SequenceExpression(self.alloc(SequenceExpression { span, expressions }))
}
pub fn tagged_template_expression(
&self,
span: Span,
tag: Expression<'a>,
quasi: TemplateLiteral<'a>,
type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
) -> Expression<'a> {
Expression::TaggedTemplateExpression(self.alloc(TaggedTemplateExpression {
span,
tag,
quasi,
type_parameters,
}))
}
pub fn template_literal_expression(
&self,
template_literal: TemplateLiteral<'a>,
) -> Expression<'a> {
Expression::TemplateLiteral(self.alloc(template_literal))
}
pub fn this_expression(&self, span: Span) -> Expression<'a> {
Expression::ThisExpression(self.alloc(ThisExpression { span }))
}
pub fn unary_expression(
&self,
span: Span,
operator: UnaryOperator,
argument: Expression<'a>,
) -> Expression<'a> {
Expression::UnaryExpression(self.alloc(UnaryExpression { span, operator, argument }))
}
pub fn update_expression(
&self,
span: Span,
operator: UpdateOperator,
prefix: bool,
argument: SimpleAssignmentTarget<'a>,
) -> Expression<'a> {
Expression::UpdateExpression(self.alloc(UpdateExpression {
span,
operator,
prefix,
argument,
}))
}
pub fn yield_expression(
&self,
span: Span,
delegate: bool,
argument: Option<Expression<'a>>,
) -> Expression<'a> {
Expression::YieldExpression(self.alloc(YieldExpression { span, delegate, argument }))
}
/* ---------- Functions ---------- */
pub fn function_declaration(&self, func: Box<'a, Function<'a>>) -> Statement<'a> {
Statement::Declaration(Declaration::FunctionDeclaration(func))
}
pub fn formal_parameters(
&self,
span: Span,
kind: FormalParameterKind,
items: Vec<'a, FormalParameter<'a>>,
rest: Option<Box<'a, RestElement<'a>>>,
) -> Box<'a, FormalParameters<'a>> {
self.alloc(FormalParameters { span, kind, items, rest })
}
pub fn formal_parameter(
&self,
span: Span,
pattern: BindingPattern<'a>,
accessibility: Option<TSAccessibility>,
readonly: bool,
decorators: Vec<'a, Decorator<'a>>,
) -> FormalParameter<'a> {
FormalParameter { span, pattern, accessibility, readonly, decorators }
}
pub fn function(
&self,
r#type: FunctionType,
span: Span,
id: Option<BindingIdentifier>,
expression: bool,
generator: bool,
r#async: bool,
params: Box<'a, FormalParameters<'a>>,
body: Option<Box<'a, FunctionBody<'a>>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
return_type: Option<Box<'a, TSTypeAnnotation<'a>>>,
modifiers: Modifiers<'a>,
) -> Box<'a, Function<'a>> {
self.alloc(Function {
r#type,
span,
id,
expression,
generator,
r#async,
params,
body,
type_parameters,
return_type,
modifiers,
})
}
pub fn function_body(
&self,
span: Span,
directives: Vec<'a, Directive>,
statements: Vec<'a, Statement<'a>>,
) -> Box<'a, FunctionBody<'a>> {
self.alloc(FunctionBody { span, directives, statements })
}
/* ---------- Class ---------- */
pub fn class(
&self,
r#type: ClassType,
span: Span,
id: Option<BindingIdentifier>,
super_class: Option<Expression<'a>>,
body: Box<'a, ClassBody<'a>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
super_type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
implements: Option<Vec<'a, Box<'a, TSClassImplements<'a>>>>,
decorators: Vec<'a, Decorator<'a>>,
modifiers: Modifiers<'a>,
) -> Box<'a, Class<'a>> {
self.alloc(Class {
r#type,
span,
id,
super_class,
body,
type_parameters,
super_type_parameters,
implements,
decorators,
modifiers,
})
}
pub fn class_body(
&self,
span: Span,
body: Vec<'a, ClassElement<'a>>,
) -> Box<'a, ClassBody<'a>> {
self.alloc(ClassBody { span, body })
}
pub fn class_declaration(&self, class: Box<'a, Class<'a>>) -> Statement<'a> {
Statement::Declaration(Declaration::ClassDeclaration(class))
}
pub fn static_block(&self, span: Span, body: Vec<'a, Statement<'a>>) -> ClassElement<'a> {
ClassElement::StaticBlock(self.alloc(StaticBlock { span, body }))
}
pub fn class_property(
&self,
span: Span,
key: PropertyKey<'a>,
value: Option<Expression<'a>>,
computed: bool,
r#static: bool,
decorators: Vec<'a, Decorator<'a>>,
) -> ClassElement<'a> {
ClassElement::PropertyDefinition(self.alloc(PropertyDefinition {
span,
key,
value,
computed,
r#static,
declare: false,
r#override: false,
optional: false,
definite: false,
readonly: false,
type_annotation: None,
accessibility: None,
decorators,
}))
}
pub fn accessor_property(
&self,
span: Span,
key: PropertyKey<'a>,
value: Option<Expression<'a>>,
computed: bool,
r#static: bool,
) -> ClassElement<'a> {
ClassElement::AccessorProperty(self.alloc(AccessorProperty {
span,
key,
value,
computed,
r#static,
}))
}
/* ---------- Declarations ---------- */
pub fn variable_declaration(
&self,
span: Span,
kind: VariableDeclarationKind,
declarations: Vec<'a, VariableDeclarator<'a>>,
modifiers: Modifiers<'a>,
) -> Box<'a, VariableDeclaration<'a>> {
self.alloc(VariableDeclaration { span, kind, declarations, modifiers })
}
pub fn variable_declarator(
&self,
span: Span,
kind: VariableDeclarationKind,
id: BindingPattern<'a>,
init: Option<Expression<'a>>,
definite: bool,
) -> VariableDeclarator<'a> {
VariableDeclarator { span, kind, id, init, definite }
}
pub fn using_declaration(
&self,
span: Span,
declarations: Vec<'a, VariableDeclarator<'a>>,
is_await: bool,
) -> UsingDeclaration<'a> {
UsingDeclaration { span, is_await, declarations }
}
/* ---------- Patterns ---------- */
pub fn binding_pattern(
&self,
kind: BindingPatternKind<'a>,
type_annotation: Option<Box<'a, TSTypeAnnotation<'a>>>,
optional: bool,
) -> BindingPattern<'a> {
BindingPattern { kind, type_annotation, optional }
}
pub fn binding_pattern_identifier(
&self,
identifier: BindingIdentifier,
) -> BindingPatternKind<'a> {
BindingPatternKind::BindingIdentifier(self.alloc(identifier))
}
pub fn object_pattern(
&self,
span: Span,
properties: Vec<'a, BindingProperty<'a>>,
rest: Option<Box<'a, RestElement<'a>>>,
) -> BindingPatternKind<'a> {
BindingPatternKind::ObjectPattern(self.alloc(ObjectPattern { span, properties, rest }))
}
pub fn binding_property(
&self,
span: Span,
key: PropertyKey<'a>,
value: BindingPattern<'a>,
shorthand: bool,
computed: bool,
) -> BindingProperty<'a> {
BindingProperty { span, key, value, shorthand, computed }
}
pub fn spread_element(
&self,
span: Span,
argument: Expression<'a>,
) -> Box<'a, SpreadElement<'a>> {
self.alloc(SpreadElement { span, argument })
}
pub fn array_pattern(
&self,
span: Span,
elements: Vec<'a, Option<BindingPattern<'a>>>,
rest: Option<Box<'a, RestElement<'a>>>,
) -> BindingPatternKind<'a> {
BindingPatternKind::ArrayPattern(self.alloc(ArrayPattern { span, elements, rest }))
}
pub fn assignment_pattern(
&self,
span: Span,
left: BindingPattern<'a>,
right: Expression<'a>,
) -> BindingPattern<'a> {
let pattern = self.alloc(AssignmentPattern { span, left, right });
BindingPattern {
kind: BindingPatternKind::AssignmentPattern(pattern),
type_annotation: None,
optional: false,
}
}
pub fn rest_element(
&self,
span: Span,
argument: BindingPattern<'a>,
) -> Box<'a, RestElement<'a>> {
self.alloc(RestElement { span, argument })
}
pub fn property_key_identifier(&self, ident: IdentifierName) -> PropertyKey<'a> {
PropertyKey::Identifier(self.alloc(ident))
}
pub fn property_key_expression(&self, expr: Expression<'a>) -> PropertyKey<'a> {
PropertyKey::Expression(expr)
}
/* ---------- Modules ---------- */
pub fn module_declaration(&self, decl: ModuleDeclaration<'a>) -> Statement<'a> {
Statement::ModuleDeclaration(self.alloc(decl))
}
pub fn import_declaration(
&self,
span: Span,
specifiers: Option<Vec<'a, ImportDeclarationSpecifier>>,
source: StringLiteral,
assertions: Option<Vec<'a, ImportAttribute>>,
import_kind: ImportOrExportKind,
) -> Box<'a, ImportDeclaration<'a>> {
self.alloc(ImportDeclaration { span, specifiers, source, assertions, import_kind })
}
pub fn export_all_declaration(
&self,
span: Span,
exported: Option<ModuleExportName>,
source: StringLiteral,
assertions: Option<Vec<'a, ImportAttribute>>,
export_kind: ImportOrExportKind,
) -> Box<'a, ExportAllDeclaration<'a>> {
self.alloc(ExportAllDeclaration { span, exported, source, assertions, export_kind })
}
pub fn export_default_declaration(
&self,
span: Span,
declaration: ExportDefaultDeclarationKind<'a>,
exported: ModuleExportName,
) -> Box<'a, ExportDefaultDeclaration<'a>> {
self.alloc(ExportDefaultDeclaration { span, declaration, exported })
}
pub fn export_named_declaration(
&self,
span: Span,
declaration: Option<Declaration<'a>>,
specifiers: Vec<'a, ExportSpecifier>,
source: Option<StringLiteral>,
export_kind: ImportOrExportKind,
) -> Box<'a, ExportNamedDeclaration<'a>> {
self.alloc(ExportNamedDeclaration { span, declaration, specifiers, source, export_kind })
}
/* ---------- JSX ----------------- */
pub fn jsx_element(
&self,
span: Span,
opening_element: Box<'a, JSXOpeningElement<'a>>,
closing_element: Option<Box<'a, JSXClosingElement<'a>>>,
children: Vec<'a, JSXChild<'a>>,
) -> Box<'a, JSXElement<'a>> {
self.alloc(JSXElement { span, opening_element, closing_element, children })
}
pub fn jsx_opening_element(
&self,
span: Span,
self_closing: bool,
name: JSXElementName<'a>,
attributes: Vec<'a, JSXAttributeItem<'a>>,
type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
) -> Box<'a, JSXOpeningElement<'a>> {
self.alloc(JSXOpeningElement { span, self_closing, name, attributes, type_parameters })
}
pub fn jsx_closing_element(
&self,
span: Span,
name: JSXElementName<'a>,
) -> Box<'a, JSXClosingElement<'a>> {
self.alloc(JSXClosingElement { span, name })
}
pub fn jsx_fragment(
&self,
span: Span,
opening_fragment: JSXOpeningFragment,
closing_fragment: JSXClosingFragment,
children: Vec<'a, JSXChild<'a>>,
) -> Box<'a, JSXFragment<'a>> {
self.alloc(JSXFragment { span, opening_fragment, closing_fragment, children })
}
pub fn jsx_opening_fragment(&self, span: Span) -> JSXOpeningFragment {
JSXOpeningFragment { span }
}
pub fn jsx_closing_fragment(&self, span: Span) -> JSXClosingFragment {
JSXClosingFragment { span }
}
pub fn jsx_namespaced_name(
&self,
span: Span,
namespace: JSXIdentifier,
property: JSXIdentifier,
) -> Box<'a, JSXNamespacedName> {
self.alloc(JSXNamespacedName { span, namespace, property })
}
pub fn jsx_member_expression(
&self,
span: Span,
object: JSXMemberExpressionObject<'a>,
property: JSXIdentifier,
) -> Box<'a, JSXMemberExpression<'a>> {
self.alloc(JSXMemberExpression { span, object, property })
}
pub fn jsx_expression_container(
&self,
span: Span,
expression: JSXExpression<'a>,
) -> JSXExpressionContainer<'a> {
JSXExpressionContainer { span, expression }
}
pub fn jsx_spread_child(&self, span: Span, expression: Expression<'a>) -> JSXSpreadChild<'a> {
JSXSpreadChild { span, expression }
}
pub fn jsx_empty_expression(&self, span: Span) -> JSXEmptyExpression {
JSXEmptyExpression { span }
}
pub fn jsx_attribute(
&self,
span: Span,
name: JSXAttributeName<'a>,
value: Option<JSXAttributeValue<'a>>,
) -> Box<'a, JSXAttribute<'a>> {
self.alloc(JSXAttribute { span, name, value })
}
pub fn jsx_spread_attribute(
&self,
span: Span,
argument: Expression<'a>,
) -> Box<'a, JSXSpreadAttribute<'a>> {
self.alloc(JSXSpreadAttribute { span, argument })
}
pub fn jsx_identifier(&self, span: Span, name: Atom) -> JSXIdentifier {
JSXIdentifier { span, name }
}
pub fn jsx_text(&self, span: Span, value: Atom) -> JSXText {
JSXText { span, value }
}
/* ---------- TypeScript ---------- */
pub fn ts_module_declaration(
&self,
span: Span,
id: TSModuleDeclarationName,
body: TSModuleDeclarationBody<'a>,
modifiers: Modifiers<'a>,
) -> Box<'a, TSModuleDeclaration<'a>> {
self.alloc(TSModuleDeclaration { span, id, body, modifiers })
}
pub fn ts_type_annotation(
&self,
span: Span,
type_annotation: TSType<'a>,
) -> Box<'a, TSTypeAnnotation<'a>> {
self.alloc(TSTypeAnnotation { span, type_annotation })
}
pub fn ts_literal_type(&self, span: Span, literal: TSLiteral<'a>) -> TSType<'a> {
TSType::TSLiteralType(self.alloc(TSLiteralType { span, literal }))
}
pub fn ts_union_type(&self, span: Span, types: Vec<'a, TSType<'a>>) -> TSType<'a> {
TSType::TSUnionType(self.alloc(TSUnionType { span, types }))
}
pub fn ts_intersection_type(&self, span: Span, types: Vec<'a, TSType<'a>>) -> TSType<'a> {
TSType::TSIntersectionType(self.alloc(TSIntersectionType { span, types }))
}
pub fn ts_type_operator_type(
&self,
span: Span,
operator: TSTypeOperator,
type_annotation: TSType<'a>,
) -> TSType<'a> {
TSType::TSTypeOperatorType(self.alloc(TSTypeOperatorType {
span,
operator,
type_annotation,
}))
}
pub fn ts_array_type(&self, span: Span, element_type: TSType<'a>) -> TSType<'a> {
TSType::TSArrayType(self.alloc(TSArrayType { span, element_type }))
}
pub fn ts_indexed_access_type(
&self,
span: Span,
object_type: TSType<'a>,
index_type: TSType<'a>,
) -> TSType<'a> {
TSType::TSIndexedAccessType(self.alloc(TSIndexedAccessType {
span,
object_type,
index_type,
}))
}
pub fn ts_tuple_type(
&self,
span: Span,
element_types: Vec<'a, TSTupleElement<'a>>,
) -> TSType<'a> {
TSType::TSTupleType(self.alloc(TSTupleType { span, element_types }))
}
pub fn ts_type_reference(
&self,
span: Span,
type_name: TSTypeName<'a>,
type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
) -> TSType<'a> {
TSType::TSTypeReference(self.alloc(TSTypeReference { span, type_name, type_parameters }))
}
pub fn ts_type_literal(&self, span: Span, members: Vec<'a, TSSignature<'a>>) -> TSType<'a> {
TSType::TSTypeLiteral(self.alloc(TSTypeLiteral { span, members }))
}
pub fn ts_type_implement(
&self,
span: Span,
expression: TSTypeName<'a>,
type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
) -> Box<'a, TSClassImplements<'a>> {
self.alloc(TSClassImplements { span, expression, type_parameters })
}
pub fn ts_type_parameter(
&self,
span: Span,
name: BindingIdentifier,
constraint: Option<TSType<'a>>,
default: Option<TSType<'a>>,
r#in: bool,
out: bool,
r#const: bool,
) -> Box<'a, TSTypeParameter<'a>> {
self.alloc(TSTypeParameter { span, name, constraint, default, r#in, out, r#const })
}
pub fn ts_type_parameters(
&self,
span: Span,
params: Vec<'a, Box<'a, TSTypeParameter<'a>>>,
) -> Box<'a, TSTypeParameterDeclaration<'a>> {
self.alloc(TSTypeParameterDeclaration { span, params })
}
pub fn ts_interface_heritages(
&self,
extends: Vec<'a, (Expression<'a>, Option<Box<'a, TSTypeParameterInstantiation<'a>>>, Span)>,
) -> Vec<'a, Box<'a, TSInterfaceHeritage<'a>>> {
Vec::from_iter_in(
extends.into_iter().map(|(expression, type_parameters, span)| {
self.alloc(TSInterfaceHeritage { span, expression, type_parameters })
}),
self.allocator,
)
}
pub fn ts_interface_body(
&self,
span: Span,
body: Vec<'a, TSSignature<'a>>,
) -> Box<'a, TSInterfaceBody<'a>> {
self.alloc(TSInterfaceBody { span, body })
}
pub fn ts_index_signature(
&self,
span: Span,
parameters: Vec<'a, Box<'a, TSIndexSignatureName<'a>>>,
type_annotation: Box<'a, TSTypeAnnotation<'a>>,
) -> TSSignature<'a> {
TSSignature::TSIndexSignature(self.alloc(TSIndexSignature {
span,
parameters,
type_annotation,
}))
}
pub fn ts_property_signature(
&self,
span: Span,
computed: bool,
optional: bool,
readonly: bool,
key: PropertyKey<'a>,
type_annotation: Option<Box<'a, TSTypeAnnotation<'a>>>,
) -> TSSignature<'a> {
TSSignature::TSPropertySignature(self.alloc(TSPropertySignature {
span,
computed,
optional,
readonly,
key,
type_annotation,
}))
}
pub fn ts_call_signature_declaration(
&self,
span: Span,
params: Box<'a, FormalParameters<'a>>,
return_type: Option<Box<'a, TSTypeAnnotation<'a>>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
) -> TSSignature<'a> {
TSSignature::TSCallSignatureDeclaration(self.alloc(TSCallSignatureDeclaration {
span,
params,
return_type,
type_parameters,
}))
}
pub fn ts_construct_signature_declaration(
&self,
span: Span,
params: Box<'a, FormalParameters<'a>>,
return_type: Option<Box<'a, TSTypeAnnotation<'a>>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
) -> TSSignature<'a> {
TSSignature::TSConstructSignatureDeclaration(self.alloc(TSConstructSignatureDeclaration {
span,
params,
return_type,
type_parameters,
}))
}
pub fn ts_method_signature(
&self,
span: Span,
key: PropertyKey<'a>,
computed: bool,
optional: bool,
kind: TSMethodSignatureKind,
params: Box<'a, FormalParameters<'a>>,
return_type: Option<Box<'a, TSTypeAnnotation<'a>>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
) -> TSSignature<'a> {
TSSignature::TSMethodSignature(self.alloc(TSMethodSignature {
span,
key,
computed,
optional,
kind,
params,
return_type,
type_parameters,
}))
}
pub fn ts_module_block(
&self,
span: Span,
body: Vec<'a, Statement<'a>>,
) -> Box<'a, TSModuleBlock<'a>> {
self.alloc(TSModuleBlock { span, body })
}
pub fn ts_type_arguments(
&self,
span: Span,
params: Vec<'a, TSType<'a>>,
) -> Box<'a, TSTypeParameterInstantiation<'a>> {
self.alloc(TSTypeParameterInstantiation { span, params })
}
pub fn ts_as_expression(
&self,
span: Span,
expression: Expression<'a>,
type_annotation: TSType<'a>,
) -> Expression<'a> {
Expression::TSAsExpression(self.alloc(TSAsExpression { span, expression, type_annotation }))
}
pub fn ts_satisfies_expression(
&self,
span: Span,
expression: Expression<'a>,
type_annotation: TSType<'a>,
) -> Expression<'a> {
Expression::TSSatisfiesExpression(self.alloc(TSSatisfiesExpression {
span,
expression,
type_annotation,
}))
}
pub fn ts_non_null_expression(&self, span: Span, expression: Expression<'a>) -> Expression<'a> {
Expression::TSNonNullExpression(self.alloc(TSNonNullExpression { span, expression }))
}
pub fn ts_type_assertion(
&self,
span: Span,
type_annotation: TSType<'a>,
expression: Expression<'a>,
) -> Expression<'a> {
Expression::TSTypeAssertion(self.alloc(TSTypeAssertion {
span,
expression,
type_annotation,
}))
}
pub fn ts_import_equals_declaration(
&self,
span: Span,
id: BindingIdentifier,
module_reference: TSModuleReference<'a>,
is_export: bool,
import_kind: ImportOrExportKind,
) -> Declaration<'a> {
Declaration::TSImportEqualsDeclaration(self.alloc(TSImportEqualsDeclaration {
span,
id,
module_reference: self.alloc(module_reference),
is_export,
import_kind,
}))
}
pub fn ts_interface_declaration(
&self,
span: Span,
id: BindingIdentifier,
body: Box<'a, TSInterfaceBody<'a>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
extends: Option<Vec<'a, Box<'a, TSInterfaceHeritage<'a>>>>,
modifiers: Modifiers<'a>,
) -> Declaration<'a> {
Declaration::TSInterfaceDeclaration(self.alloc(TSInterfaceDeclaration {
span,
id,
body,
type_parameters,
extends,
modifiers,
}))
}
pub fn ts_type_alias_declaration(
&self,
span: Span,
id: BindingIdentifier,
type_annotation: TSType<'a>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
modifiers: Modifiers<'a>,
) -> Declaration<'a> {
Declaration::TSTypeAliasDeclaration(self.alloc(TSTypeAliasDeclaration {
span,
id,
type_annotation,
type_parameters,
modifiers,
}))
}
pub fn ts_enum_declaration(
&self,
span: Span,
id: BindingIdentifier,
members: Vec<'a, TSEnumMember<'a>>,
modifiers: Modifiers<'a>,
) -> Declaration<'a> {
Declaration::TSEnumDeclaration(self.alloc(TSEnumDeclaration {
span,
id,
body: TSEnumBody { span, members },
modifiers,
}))
}
pub fn decorator(&self, span: Span, expression: Expression<'a>) -> Decorator<'a> {
Decorator { span, expression }
}
pub fn ts_void_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSVoidKeyword(self.alloc(TSVoidKeyword { span }))
}
pub fn ts_this_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSThisKeyword(self.alloc(TSThisKeyword { span }))
}
pub fn ts_any_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSAnyKeyword(self.alloc(TSAnyKeyword { span }))
}
pub fn ts_unknown_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSUnknownKeyword(self.alloc(TSUnknownKeyword { span }))
}
pub fn ts_number_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSNumberKeyword(self.alloc(TSNumberKeyword { span }))
}
pub fn ts_boolean_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSBooleanKeyword(self.alloc(TSBooleanKeyword { span }))
}
pub fn ts_object_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSObjectKeyword(self.alloc(TSObjectKeyword { span }))
}
pub fn ts_string_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSStringKeyword(self.alloc(TSStringKeyword { span }))
}
pub fn ts_bigint_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSBigIntKeyword(self.alloc(TSBigIntKeyword { span }))
}
pub fn ts_symbol_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSSymbolKeyword(self.alloc(TSSymbolKeyword { span }))
}
pub fn ts_null_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSNullKeyword(self.alloc(TSNullKeyword { span }))
}
pub fn ts_undefined_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSUndefinedKeyword(self.alloc(TSUndefinedKeyword { span }))
}
pub fn ts_never_keyword(&self, span: Span) -> TSType<'a> {
TSType::TSNeverKeyword(self.alloc(TSNeverKeyword { span }))
}
pub fn ts_template_literal_type(
&self,
span: Span,
quasis: Vec<'a, TemplateElement>,
types: Vec<'a, TSType<'a>>,
) -> TSType<'a> {
TSType::TSTemplateLiteralType(self.alloc(TSTemplateLiteralType { span, quasis, types }))
}
pub fn ts_type_query_type(
&self,
span: Span,
expr_name: TSTypeName<'a>,
type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
) -> TSType<'a> {
TSType::TSTypeQuery(self.alloc(TSTypeQuery { span, expr_name, type_parameters }))
}
pub fn ts_conditional_type(
&self,
span: Span,
check_type: TSType<'a>,
extends_type: TSType<'a>,
true_type: TSType<'a>,
false_type: TSType<'a>,
) -> TSType<'a> {
TSType::TSConditionalType(self.alloc(TSConditionalType {
span,
check_type,
extends_type,
true_type,
false_type,
}))
}
pub fn ts_mapped_type(
&self,
span: Span,
type_parameter: Box<'a, TSTypeParameter<'a>>,
name_type: Option<TSType<'a>>,
type_annotation: Option<Box<'a, TSTypeAnnotation<'a>>>,
optional: TSMappedTypeModifierOperator,
readonly: TSMappedTypeModifierOperator,
) -> TSType<'a> {
TSType::TSMappedType(self.alloc(TSMappedType {
span,
type_parameter,
name_type,
type_annotation,
optional,
readonly,
}))
}
pub fn ts_import_type(
&self,
span: Span,
is_type_of: bool,
parameter: TSType<'a>,
qualifier: Option<TSTypeName<'a>>,
type_parameters: Option<Box<'a, TSTypeParameterInstantiation<'a>>>,
) -> TSType<'a> {
TSType::TSImportType(self.alloc(TSImportType {
span,
is_type_of,
parameter,
qualifier,
type_parameters,
}))
}
pub fn ts_constructor_type(
&self,
span: Span,
r#abstract: bool,
params: Box<'a, FormalParameters<'a>>,
return_type: Box<'a, TSTypeAnnotation<'a>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
) -> TSType<'a> {
TSType::TSConstructorType(self.alloc(TSConstructorType {
span,
r#abstract,
params,
return_type,
type_parameters,
}))
}
pub fn ts_function_type(
&self,
span: Span,
params: Box<'a, FormalParameters<'a>>,
return_type: Box<'a, TSTypeAnnotation<'a>>,
type_parameters: Option<Box<'a, TSTypeParameterDeclaration<'a>>>,
) -> TSType<'a> {
TSType::TSFunctionType(self.alloc(TSFunctionType {
span,
params,
return_type,
type_parameters,
}))
}
pub fn ts_infer_type(
&self,
span: Span,
type_parameter: Box<'a, TSTypeParameter<'a>>,
) -> TSType<'a> {
TSType::TSInferType(self.alloc(TSInferType { span, type_parameter }))
}
pub fn ts_type_predicate(
&self,
span: Span,
parameter_name: TSTypePredicateName,
asserts: bool,
type_annotation: Option<Box<'a, TSTypeAnnotation<'a>>>,
) -> TSType<'a> {
TSType::TSTypePredicate(self.alloc(TSTypePredicate {
span,
parameter_name,
asserts,
type_annotation,
}))
}
/* JSDoc */
pub fn js_doc_nullable_type(
&self,
span: Span,
type_annotation: TSType<'a>,
postfix: bool,
) -> TSType<'a> {
TSType::JSDocNullableType(self.alloc(JSDocNullableType { span, type_annotation, postfix }))
}
pub fn js_doc_unknown_type(&self, span: Span) -> TSType<'a> {
TSType::JSDocUnknownType(self.alloc(JSDocUnknownType { span }))
}
}