cartodb/lib/assets/javascripts/builder/helpers/analysis-node-utils.js
2020-06-15 10:58:47 +08:00

52 lines
1.2 KiB
JavaScript

// Get the source node from an analysis node
function getSourceNode (nodeModel) {
var source;
if (nodeModel.get('type') === 'source') {
source = nodeModel;
} else {
var primarySource = nodeModel.getPrimarySource && nodeModel.getPrimarySource();
if (primarySource && primarySource.get('type') === 'source') {
source = primarySource;
} else {
source = getSourceNode(primarySource);
}
}
return source;
}
// Evaluates predicate for the analysis chain from nodeModel
function someNode (nodeModel, predicate) {
if (predicate(nodeModel)) {
return nodeModel;
}
if (nodeModel.get('type') !== 'source') {
const source = nodeModel.getPrimarySource && nodeModel.getPrimarySource();
if (source) {
return someNode(source, predicate);
}
}
return false;
}
function hasAnalysisType (type) {
return function (node) {
return node.get('type') === type;
};
}
const hasTradeArea = hasAnalysisType('trade-area');
const hasSQLFunction = hasAnalysisType('deprecated-sql-function');
module.exports = {
nodeHasTradeArea: node => someNode(node, hasTradeArea),
nodeHasSQLFunction: node => someNode(node, hasSQLFunction),
// These are defined outside because they're recursive
getSourceNode,
someNode
};