Initial commit

lib/assets/javascripts/builder/data/legends/legend-definitions-collection.js

@@ -0,0 +1,104 @@
+var Backbone = require('backbone');
+var _ = require('underscore');
+var LegendBaseDefModel = require('./legend-base-definition-model');
+var layerTypesAndKinds = require('builder/data/layer-types-and-kinds');
+
+var CategoryModel = require('./legend-category-definition-model');
+var BubbleModel = require('./legend-bubble-definition-model');
+var ChoroplethModel = require('./legend-choropleth-definition-model');
+var CustomChoroplethModel = require('./legend-custom-choropleth-definition-model');
+var CustomModel = require('./legend-custom-definition-model');
+var TorqueModel = require('./legend-torque-definition-model');
+
+var LEGEND_MODEL_TYPE = {
+  'bubble': BubbleModel,
+  'category': CategoryModel,
+  'choropleth': ChoroplethModel,
+  'custom_choropleth': CustomChoroplethModel,
+  'custom': CustomModel,
+  'html': CustomModel,
+  'torque': TorqueModel
+};
+
+module.exports = Backbone.Collection.extend({
+  model: LegendBaseDefModel,
+
+  initialize: function (models, options) {
+    if (!options.configModel) throw new Error('configModel is required');
+    if (!options.layerDefinitionsCollection) throw new Error('layerDefinitionsCollection is required');
+    if (!options.vizId) throw new Error('vizId is required');
+
+    this.configModel = options.configModel;
+    this.layerDefinitionsCollection = options.layerDefinitionsCollection;
+    this.vizId = options.vizId;
+  },
+
+  /**
+   * Intended to be called from entry point, to make sure initial layers are taken into account
+   */
+  resetByData: function (vizJSON) {
+    var models = [];
+    var layers = _.filter(vizJSON.layers, function (layer) {
+      return this._isDataLayer(layer.type);
+    }, this);
+
+    _.each(layers, function (layer) {
+      var layerDefModel = this._findLayerDefinitionModel(layer.id);
+      var legends = [];
+      if (layer.legends) {
+        legends = layer.legends.map(function (legend) {
+          var Klass = LEGEND_MODEL_TYPE[legend.type];
+
+          return new Klass(legend, {
+            layerDefinitionModel: layerDefModel,
+            configModel: this.configModel,
+            vizId: this.vizId,
+            parse: true
+          });
+        }, this);
+      }
+
+      models.push(legends);
+    }, this);
+
+    this.reset(_.flatten(models));
+  },
+
+  fetch: function () {
+    throw new Error('This collection should not make any fetch calls. It should be populated from the vizJSON.');
+  },
+
+  _isDataLayer: function (layerType) {
+    return layerTypesAndKinds.isCartoDBType(layerType) ||
+      layerTypesAndKinds.isTorqueType(layerType);
+  },
+
+  _findLayerDefinitionModel: function (id) {
+    return this.layerDefinitionsCollection.findWhere({id: id});
+  },
+
+  findByLayerDefModel: function (layerDefModel) {
+    var id = layerDefModel.id;
+    return this.filter(function (legendDefModel) {
+      var layerDefModel = legendDefModel.layerDefinitionModel;
+      return layerDefModel.id === id;
+    });
+  },
+
+  findByLayerDefModelAndType: function (layerDefModel, type) {
+    var id = layerDefModel.id;
+    return this.find(function (legendDefModel) {
+      var layerDefModel = legendDefModel.layerDefinitionModel;
+      return layerDefModel.id === id && legendDefModel.get('type') === type;
+    });
+  },
+
+  findByLayerDefModelAndTypes: function (layerDefModel, types) {
+    var id = layerDefModel.id;
+    return this.filter(function (legendDefModel) {
+      var layerDefModel = legendDefModel.layerDefinitionModel;
+      var type = legendDefModel.get('type');
+      return layerDefModel.id === id && types.indexOf(type) !== -1;
+    });
+  }
+});