2015-07-11 12:53:23 -07:00

111 lines
1.9 KiB
JavaScript

import * as t from "../../types";
/**
* [Please add a description.]
*/
export function _params(node, print) {
print.plain(node.typeParameters);
this.push("(");
print.list(node.params, {
iterator: (node) =>{
if (node.optional) this.push("?");
print.plain(node.typeAnnotation);
}
});
this.push(")");
if (node.returnType) {
print.plain(node.returnType);
}
}
/**
* [Please add a description.]
*/
export function _method(node, print) {
var value = node.value;
var kind = node.kind;
var key = node.key;
if (kind === "method" || kind === "init") {
if (value.generator) {
this.push("*");
}
}
if (kind === "get" || kind === "set") {
this.push(kind + " ");
}
if (value.async) this.push("async ");
if (node.computed) {
this.push("[");
print.plain(key);
this.push("]");
} else {
print.plain(key);
}
this._params(value, print);
this.space();
print.plain(value.body);
}
/**
* [Please add a description.]
*/
export function FunctionExpression(node, print) {
if (node.async) this.push("async ");
this.push("function");
if (node.generator) this.push("*");
if (node.id) {
this.push(" ");
print.plain(node.id);
} else {
this.space();
}
this._params(node, print);
this.space();
print.plain(node.body);
}
/**
* [Please add a description.]
*/
export { FunctionExpression as FunctionDeclaration };
/**
* [Please add a description.]
*/
export function ArrowFunctionExpression(node, print) {
if (node.async) this.push("async ");
if (node.params.length === 1 && t.isIdentifier(node.params[0])) {
print.plain(node.params[0]);
} else {
this._params(node, print);
}
this.push(" => ");
const bodyNeedsParens = t.isObjectExpression(node.body);
if (bodyNeedsParens) {
this.push("(");
}
print.plain(node.body);
if (bodyNeedsParens) {
this.push(")");
}
}