614b486780
* migrate to named babel types imports * perf: transform babel types import to destructuring * fix merge errors * apply plugin to itself
68 lines
1.9 KiB
TypeScript
68 lines
1.9 KiB
TypeScript
import {
|
|
callExpression,
|
|
identifier,
|
|
isIdentifier,
|
|
isSpreadElement,
|
|
memberExpression,
|
|
optionalCallExpression,
|
|
optionalMemberExpression,
|
|
} from "@babel/types";
|
|
import type {
|
|
CallExpression,
|
|
Expression,
|
|
OptionalCallExpression,
|
|
SpreadElement,
|
|
} from "@babel/types";
|
|
|
|
/**
|
|
* A helper function that generates a new call expression with given thisNode.
|
|
It will also optimize `(...arguments)` to `.apply(arguments)`
|
|
*
|
|
* @export
|
|
* @param {Expression} callee The callee of call expression
|
|
* @param {Expression} thisNode The desired this of call expression
|
|
* @param {Readonly<Array<Expression | SpreadElement>>} args The arguments of call expression
|
|
* @param {boolean} optional Whether the call expression is optional
|
|
* @returns {CallExpression | OptionalCallExpression} The generated new call expression
|
|
*/
|
|
export default function optimiseCallExpression(
|
|
callee: Expression,
|
|
thisNode: Expression,
|
|
args: Readonly<Array<Expression | SpreadElement>>,
|
|
optional: boolean,
|
|
): CallExpression | OptionalCallExpression {
|
|
if (
|
|
args.length === 1 &&
|
|
isSpreadElement(args[0]) &&
|
|
isIdentifier(args[0].argument, { name: "arguments" })
|
|
) {
|
|
// a.b?.(...arguments);
|
|
if (optional) {
|
|
return optionalCallExpression(
|
|
optionalMemberExpression(callee, identifier("apply"), false, true),
|
|
[thisNode, args[0].argument],
|
|
false,
|
|
);
|
|
}
|
|
// a.b(...arguments);
|
|
return callExpression(memberExpression(callee, identifier("apply")), [
|
|
thisNode,
|
|
args[0].argument,
|
|
]);
|
|
} else {
|
|
// a.b?.(arg1, arg2)
|
|
if (optional) {
|
|
return optionalCallExpression(
|
|
optionalMemberExpression(callee, identifier("call"), false, true),
|
|
[thisNode, ...args],
|
|
false,
|
|
);
|
|
}
|
|
// a.b(arg1, arg2)
|
|
return callExpression(memberExpression(callee, identifier("call")), [
|
|
thisNode,
|
|
...args,
|
|
]);
|
|
}
|
|
}
|