@dhyasama/totem-models 11.35.0 → 11.37.0

package/lib/LimitedPartner.js

@@ -446,6 +446,14 @@ module.exports = function(mongoose, config) {
 
       });
 
+      if (options.fundids) {
+        lp.transactions = _.filter(lp.transactions, function(transaction) {
+          return _.find(options.fundids, function(f) {
+            return transaction.fund._id.toString() === f.toString();
+          });
+        });
+      }
+
       return cb(null, lp);
 
     });
@@ -661,7 +669,23 @@ module.exports = function(mongoose, config) {
     query.populate('transactions.fund');
     query.populate('people');
     query.sort({ 'name': -1 });
-    query.exec(cb);
+    query.exec(function(err, result) {
+
+      if (err) return cb(err);
+
+      if (options.fundids) {
+        _.each(result, function(lp) {
+          lp.transactions = _.filter(lp.transactions, function(transaction) {
+            return _.find(options.fundids, function(f) {
+              return transaction.fund._id.toString() === f.toString();
+            });
+          });
+        });
+      }
+
+      return cb(null, result);
+
+    });
 
   };
 

package/lib/Round.js

@@ -665,7 +665,7 @@ module.exports = function(mongoose, config) {
 
     let query = self.find({ 'vehicles.fund': { $in: fundIds } });
 
-    query.select('organization');
+    query.select('organization vehicles');
     query.populate('organization', 'name website status ipo operating people');
 
     query.exec(function(err, rounds) {
@@ -685,13 +685,18 @@ module.exports = function(mongoose, config) {
       const grouped = _.groupBy(rounds, function(r) { return r.organization._id; });
 
       _.each(grouped, function(rounds) {
+          
+        var fundList = _.uniq(_.flatten(_.map(rounds, function(round) {
+          return round.vehicles ? _.map(round.vehicles, function(vehicle) { return vehicle.fund.toString(); }) : [];
+        })));
 
         portfolio.push({
           _id: rounds[0].organization._id,
           name: rounds[0].organization.name,
           website: rounds[0].organization.website,
           status: rounds[0].organization.status,
-          people: rounds[0].organization.people
+          people: rounds[0].organization.people,
+          funds: fundList
         });
 
       });

package/package.json

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