@dhyasama/totem-models 1.21.1 → 1.22.1

package/lib/List.js

@@ -122,6 +122,48 @@ module.exports = function(mongoose, config) {
       .exec(cb);
   };
 
+  // Get all lists that contain an lp
+  List.statics.getByLP = function(lpId, cb) {
+
+    var self = this;
+
+    self
+      .find({ 'customer': config.CUSTOMER_ID, limitedPartners: lpId })
+      .populate('limitedPartners', 'name logoUrl')
+      .populate('organizations', 'name logoUrl')
+      .populate('people', 'name title avatarUrl')
+      .exec(cb);
+
+  };
+
+  // Get all lists that contain an org
+  List.statics.getByOrg = function(orgId, cb) {
+
+    var self = this;
+
+    self
+      .find({ 'customer': config.CUSTOMER_ID, organizations: orgId })
+      .populate('limitedPartners', 'name logoUrl')
+      .populate('organizations', 'name logoUrl')
+      .populate('people', 'name title avatarUrl')
+      .exec(cb);
+
+  };
+
+  // Get all lists that contain a person
+  List.statics.getByPerson = function(personId, cb) {
+
+    var self = this;
+
+    self
+      .find({ 'customer': config.CUSTOMER_ID, people: personId })
+      .populate('limitedPartners', 'name logoUrl')
+      .populate('organizations', 'name logoUrl')
+      .populate('people', 'name title avatarUrl')
+      .exec(cb);
+
+  };
+
   List.statics.getLists = function(cb) {
 
     var self = this;
@@ -156,6 +198,10 @@ module.exports = function(mongoose, config) {
 
   };
 
+  List.statics.upsert = function (list, cb) {
+    list.save(cb);
+  };
+
   List.pre('save', function(next) {
 
     var self = this;

package/lib/Round.js

@@ -513,7 +513,8 @@ module.exports = function(mongoose, config) {
     var self = this;
 
     var query = self.find({ 'vehicles.fund': fundId });
-    query.select('vehicles');
+    query.select('organization roundName vehicles');
+    query.populate('organization', 'name logoUrl');
 
     if (config.CUSTOMER_ID == 'GLOBAL_PROCESS') {
       query.populate('vehicles.investments');
@@ -530,10 +531,21 @@ module.exports = function(mongoose, config) {
       if (err) return cb(err, null);
       if (!rounds) return cb(null, []);
 
-      var vehicles = _.flatten(_.pluck(rounds, 'vehicles'));
-      var investments = _.flatten(_.pluck(vehicles, 'investments'));
+      var fundInvestments = [];
+
+      _.each(rounds, function(round) {
+        _.each(round.vehicles, function(vehicle) {
+          if(vehicle.fund == fundId) {
+            _.each(vehicle.investments, function(investment) {
+              investment.organization = round.organization;
+              investment.roundName = round.roundName;
+              fundInvestments.push(investment);
+            });
+          }
+        });
+      });
 
-      return cb(null, investments);
+      return cb(null, fundInvestments);
 
     });
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dhyasama/totem-models",
-  "version": "1.21.1",
+  "version": "1.22.1",
   "author": "Jason Reynolds",
   "license": "UNLICENSED",
   "description": "Models for Totem platform",