@dhyasama/totem-models 9.10.0 → 9.11.0

package/lib/Fund.js

@@ -16,11 +16,6 @@ module.exports = function(mongoose, config) {
     closeDate: { type: Date },
     amount: { type: Number },
     crunchbase: { uuid: { type: String, default: '', unique: true } },
-    transactions: [{
-      type: { type: String, enum: ['fees', 'expenses', 'carry'] },
-      amount: { type: Number, default: 0 },
-      date: { type: Date, default: null },
-    }],
     customFields: { type: Schema.Types.Mixed },
     
     // the stakeholders names we match to on cap tables
@@ -143,18 +138,6 @@ module.exports = function(mongoose, config) {
 
   };
 
-  Fund.statics.removeTransactions = function removeTransactions(cb) {
-
-    const self = this;
-
-    let query = self.update(
-      { $set: { transactions: [] } },
-      { multi : true }
-    );
-    query.exec(cb);
-
-  };
-
   Fund.statics.modify = function(filter, update, cb) {
 
     // VERY IMPORTANT NOTE

package/lib/LimitedPartner.js

@@ -39,9 +39,10 @@ module.exports = function(mongoose, config) {
 
     transactions: [{
       fund: {type: Schema.ObjectId, ref: 'Fund'},
-      type: { type: String, enum: ['commitment', 'contribution', 'distribution'] },
+      type: { type: String, enum: ['commitment', 'contribution', 'distribution', 'fee', 'expense', 'carry', 'income', 'realized', 'unrealized'] },
       amount: { type: Number, default: 0 },
       date: { type: Date, default: null },
+      description: { type: String, default: '' },
     }],
 
     contact: {
@@ -134,6 +135,48 @@ module.exports = function(mongoose, config) {
     }, 0);
   });
 
+  LimitedPartner.virtual('totalFee').get(function () {
+    var self = this;
+    return _.reduce(self.transactions, function(memo, transaction) {
+      return memo + ((transaction.type === 'fee' && Math.abs(transaction.amount) > 0) ? transaction.amount : 0);
+    }, 0);
+  });
+
+  LimitedPartner.virtual('totalExpense').get(function () {
+    var self = this;
+    return _.reduce(self.transactions, function(memo, transaction) {
+      return memo + ((transaction.type === 'expense' && Math.abs(transaction.amount) > 0) ? transaction.amount : 0);
+    }, 0);
+  });
+
+  LimitedPartner.virtual('totalCarry').get(function () {
+    var self = this;
+    return _.reduce(self.transactions, function(memo, transaction) {
+      return memo + ((transaction.type === 'carry' && Math.abs(transaction.amount) > 0) ? transaction.amount : 0);
+    }, 0);
+  });
+
+  LimitedPartner.virtual('totalIncome').get(function () {
+    var self = this;
+    return _.reduce(self.transactions, function(memo, transaction) {
+      return memo + ((transaction.type === 'income' && Math.abs(transaction.amount) > 0) ? transaction.amount : 0);
+    }, 0);
+  });
+
+  LimitedPartner.virtual('totalRealized').get(function () {
+    var self = this;
+    return _.reduce(self.transactions, function(memo, transaction) {
+      return memo + ((transaction.type === 'realized' && Math.abs(transaction.amount) > 0) ? transaction.amount : 0);
+    }, 0);
+  });
+
+  LimitedPartner.virtual('totalUnrealized').get(function () {
+    var self = this;
+    return _.reduce(self.transactions, function(memo, transaction) {
+      return memo + ((transaction.type === 'unrealized' && Math.abs(transaction.amount) > 0) ? transaction.amount : 0);
+    }, 0);
+  });
+
   LimitedPartner.virtual('fundsInvested').get(function () {
     
     var self = this;
@@ -151,7 +194,13 @@ module.exports = function(mongoose, config) {
           abbreviation: transaction.fund.abbreviation,
           commitment: 0,
           contribution: 0,
-          distribution: 0
+          distribution: 0,
+          fee: 0,
+          expense: 0,
+          carry: 0,
+          income: 0,
+          realized: 0,
+          unrealized: 0
         });
         fundMatch = _.last(fundsInvested);
       }

package/package.json

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