1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
use std::iter;
use swc_atoms::js_word;
use swc_common::{Span, Spanned, DUMMY_SP};
use swc_ecma_ast::*;
pub trait ExprFactory: Into<Expr> {
#[inline]
fn as_arg(self) -> ExprOrSpread {
ExprOrSpread {
expr: Box::new(self.into()),
spread: None,
}
}
#[inline]
fn into_stmt(self) -> Stmt {
Stmt::Expr(ExprStmt {
span: DUMMY_SP,
expr: Box::new(self.into()),
})
}
#[inline]
fn as_callee(self) -> ExprOrSuper {
ExprOrSuper::Expr(Box::new(self.into()))
}
#[inline]
fn as_obj(self) -> ExprOrSuper {
ExprOrSuper::Expr(Box::new(self.into()))
}
fn apply(self, span: Span, this: Box<Expr>, args: Vec<ExprOrSpread>) -> Expr {
let apply = self.make_member(Ident::new(js_word!("apply"), span));
Expr::Call(CallExpr {
span,
callee: apply.as_callee(),
args: iter::once(this.as_arg()).chain(args).collect(),
type_args: None,
})
}
#[inline]
fn wrap_with_paren(self) -> Expr {
let expr = Box::new(self.into());
let span = expr.span();
Expr::Paren(ParenExpr { expr, span })
}
#[inline]
fn make_eq<T>(self, right: T) -> Expr
where
T: Into<Expr>,
{
self.make_bin(op!("==="), right)
}
#[inline]
fn make_bin<T>(self, op: BinaryOp, right: T) -> Expr
where
T: Into<Expr>,
{
let right = Box::new(right.into());
Expr::Bin(BinExpr {
span: DUMMY_SP,
left: Box::new(self.into()),
op,
right,
})
}
#[inline]
fn make_member<T>(self, prop: T) -> Expr
where
T: Into<Expr>,
{
Expr::Member(MemberExpr {
obj: ExprOrSuper::Expr(Box::new(self.into())),
span: DUMMY_SP,
computed: false,
prop: Box::new(prop.into()),
})
}
#[inline]
fn computed_member<T>(self, prop: T) -> Expr
where
T: Into<Expr>,
{
Expr::Member(MemberExpr {
obj: ExprOrSuper::Expr(Box::new(self.into())),
span: DUMMY_SP,
computed: true,
prop: Box::new(prop.into()),
})
}
}
impl<T: Into<Expr>> ExprFactory for T {}