156 lines
3.3 KiB
JavaScript
156 lines
3.3 KiB
JavaScript
|
"use strict";
|
||
|
|
||
|
Object.defineProperty(exports, "__esModule", {
|
||
|
value: true
|
||
|
});
|
||
|
exports.ArrowFunctionExpression = ArrowFunctionExpression;
|
||
|
exports.FunctionDeclaration = exports.FunctionExpression = FunctionExpression;
|
||
|
exports._functionHead = _functionHead;
|
||
|
exports._methodHead = _methodHead;
|
||
|
exports._param = _param;
|
||
|
exports._parameters = _parameters;
|
||
|
exports._params = _params;
|
||
|
exports._predicate = _predicate;
|
||
|
|
||
|
var _t = require("@babel/types");
|
||
|
|
||
|
const {
|
||
|
isIdentifier
|
||
|
} = _t;
|
||
|
|
||
|
function _params(node) {
|
||
|
this.print(node.typeParameters, node);
|
||
|
this.tokenChar(40);
|
||
|
|
||
|
this._parameters(node.params, node);
|
||
|
|
||
|
this.tokenChar(41);
|
||
|
this.print(node.returnType, node, node.type === "ArrowFunctionExpression");
|
||
|
}
|
||
|
|
||
|
function _parameters(parameters, parent) {
|
||
|
for (let i = 0; i < parameters.length; i++) {
|
||
|
this._param(parameters[i], parent);
|
||
|
|
||
|
if (i < parameters.length - 1) {
|
||
|
this.tokenChar(44);
|
||
|
this.space();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function _param(parameter, parent) {
|
||
|
this.printJoin(parameter.decorators, parameter);
|
||
|
this.print(parameter, parent);
|
||
|
|
||
|
if (parameter.optional) {
|
||
|
this.tokenChar(63);
|
||
|
}
|
||
|
|
||
|
this.print(parameter.typeAnnotation, parameter);
|
||
|
}
|
||
|
|
||
|
function _methodHead(node) {
|
||
|
const kind = node.kind;
|
||
|
const key = node.key;
|
||
|
|
||
|
if (kind === "get" || kind === "set") {
|
||
|
this.word(kind);
|
||
|
this.space();
|
||
|
}
|
||
|
|
||
|
if (node.async) {
|
||
|
this._catchUp("start", key.loc);
|
||
|
|
||
|
this.word("async");
|
||
|
this.space();
|
||
|
}
|
||
|
|
||
|
if (kind === "method" || kind === "init") {
|
||
|
if (node.generator) {
|
||
|
this.tokenChar(42);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (node.computed) {
|
||
|
this.tokenChar(91);
|
||
|
this.print(key, node);
|
||
|
this.tokenChar(93);
|
||
|
} else {
|
||
|
this.print(key, node);
|
||
|
}
|
||
|
|
||
|
if (node.optional) {
|
||
|
this.tokenChar(63);
|
||
|
}
|
||
|
|
||
|
this._params(node);
|
||
|
}
|
||
|
|
||
|
function _predicate(node) {
|
||
|
if (node.predicate) {
|
||
|
if (!node.returnType) {
|
||
|
this.tokenChar(58);
|
||
|
}
|
||
|
|
||
|
this.space();
|
||
|
this.print(node.predicate, node);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function _functionHead(node) {
|
||
|
if (node.async) {
|
||
|
this.word("async");
|
||
|
this.space();
|
||
|
}
|
||
|
|
||
|
this.word("function");
|
||
|
if (node.generator) this.tokenChar(42);
|
||
|
this.printInnerComments(node);
|
||
|
this.space();
|
||
|
|
||
|
if (node.id) {
|
||
|
this.print(node.id, node);
|
||
|
}
|
||
|
|
||
|
this._params(node);
|
||
|
|
||
|
if (node.type !== "TSDeclareFunction") {
|
||
|
this._predicate(node);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function FunctionExpression(node) {
|
||
|
this._functionHead(node);
|
||
|
|
||
|
this.space();
|
||
|
this.print(node.body, node);
|
||
|
}
|
||
|
|
||
|
function ArrowFunctionExpression(node) {
|
||
|
if (node.async) {
|
||
|
this.word("async");
|
||
|
this.space();
|
||
|
}
|
||
|
|
||
|
const firstParam = node.params[0];
|
||
|
|
||
|
if (!this.format.retainLines && !this.format.auxiliaryCommentBefore && !this.format.auxiliaryCommentAfter && node.params.length === 1 && isIdentifier(firstParam) && !hasTypesOrComments(node, firstParam)) {
|
||
|
this.print(firstParam, node);
|
||
|
} else {
|
||
|
this._params(node);
|
||
|
}
|
||
|
|
||
|
this._predicate(node);
|
||
|
|
||
|
this.space();
|
||
|
this.token("=>");
|
||
|
this.space();
|
||
|
this.print(node.body, node);
|
||
|
}
|
||
|
|
||
|
function hasTypesOrComments(node, param) {
|
||
|
var _param$leadingComment, _param$trailingCommen;
|
||
|
|
||
|
return !!(node.typeParameters || node.returnType || node.predicate || param.typeAnnotation || param.optional || (_param$leadingComment = param.leadingComments) != null && _param$leadingComment.length || (_param$trailingCommen = param.trailingComments) != null && _param$trailingCommen.length);
|
||
|
}
|