91 lines
1.9 KiB
JavaScript
91 lines
1.9 KiB
JavaScript
module.exports = Node;
|
|
|
|
var whitespace = require("./whitespace");
|
|
var parens = require("./parentheses");
|
|
var t = require("../../types");
|
|
var _ = require("lodash");
|
|
|
|
var find = function (obj, node, parent) {
|
|
var result;
|
|
|
|
_.each(obj, function (fn, type) {
|
|
if (t["is" + type](node)) {
|
|
result = fn(node, parent);
|
|
if (result != null) return false;
|
|
}
|
|
});
|
|
|
|
return result;
|
|
};
|
|
|
|
function Node(node, parent) {
|
|
this.parent = parent;
|
|
this.node = node;
|
|
}
|
|
|
|
Node.prototype.isUserWhitespacable = function () {
|
|
//var parent = this.parent;
|
|
var node = this.node;
|
|
|
|
if (t.isUserWhitespacable(node)) {
|
|
return true;
|
|
}
|
|
|
|
//if (t.isArrayExpression(parent)) {
|
|
// return true;
|
|
//}
|
|
|
|
return false;
|
|
};
|
|
|
|
Node.prototype.needsWhitespace = function (type) {
|
|
var parent = this.parent;
|
|
var node = this.node;
|
|
if (!node) return 0;
|
|
|
|
if (t.isExpressionStatement(node)) {
|
|
node = node.expression;
|
|
}
|
|
|
|
var lines = find(whitespace[type].nodes, node, parent);
|
|
if (lines) return lines;
|
|
|
|
_.each(find(whitespace[type].list, node, parent), function (expr) {
|
|
lines = Node.needsWhitespace(expr, node, type);
|
|
if (lines) return false;
|
|
});
|
|
return lines || 0;
|
|
};
|
|
|
|
Node.prototype.needsWhitespaceBefore = function () {
|
|
return this.needsWhitespace("before");
|
|
};
|
|
|
|
Node.prototype.needsWhitespaceAfter = function () {
|
|
return this.needsWhitespace("after");
|
|
};
|
|
|
|
Node.prototype.needsParens = function () {
|
|
var parent = this.parent;
|
|
var node = this.node;
|
|
|
|
if (!parent) return false;
|
|
|
|
if (t.isNewExpression(parent) && parent.callee === node) {
|
|
return t.isCallExpression(node) || _.some(node, function (val) {
|
|
return t.isCallExpression(val);
|
|
});
|
|
}
|
|
|
|
return find(parens, node, parent);
|
|
};
|
|
|
|
_.each(Node.prototype, function (fn, key) {
|
|
Node[key] = function (node, parent) {
|
|
var n = new Node(node, parent);
|
|
|
|
var args = _.toArray(arguments).slice(2);
|
|
return n[key].apply(n, args);
|
|
};
|
|
});
|