2017-06-27 12:15:00 -05:00

21 lines
514 B
JavaScript

import * as t from "babel-types";
export default function(callee, thisNode, args) {
if (
args.length === 1 &&
t.isSpreadElement(args[0]) &&
t.isIdentifier(args[0].argument, { name: "arguments" })
) {
// eg. super(...arguments);
return t.callExpression(t.memberExpression(callee, t.identifier("apply")), [
thisNode,
args[0].argument,
]);
} else {
return t.callExpression(t.memberExpression(callee, t.identifier("call")), [
thisNode,
...args,
]);
}
}