@dhyasama/totem-models 9.5.0 → 9.6.0

package/lib/CapTable.js

@@ -107,6 +107,22 @@ module.exports = function(mongoose, config) {
   // Properties that are not persisted to the database
   ///////////////////////////////////////////////////////////////////////////////////////
 
+   CapTable.virtual('shares').get(function() {
+
+    var self = this;
+
+    var shares = 0;
+
+    _.each(self.stakeholders, function(stakeholder) {
+      _.each(stakeholder.rounds, function(round) {
+        shares += round.shares;
+      });
+    });
+
+    return shares;
+
+  });
+
   ///////////////////////////////////////////////////////////////////////////////////////
   // METHODS
   // Methods operate on a single document that has already been returned
@@ -118,20 +134,47 @@ module.exports = function(mongoose, config) {
   // Statics operate on the entire collection
   //////////////////////////////////////////////////////
 
-  CapTable.statics.getForCustomer = function getForCustomer(orgId, customerId, cb) {
+  CapTable.statics.getForCustomer = function getForCustomer(options, cb) {
 
     var self = this;
 
-    self
-      .findOne({customer: customerId, organization: orgId })
-      .populate('stakeholders.lp', 'name')
-      .populate('stakeholders.org', 'name logoUrl')
-      .populate('stakeholders.person', 'name avatarUrl title')
-      .exec(cb);
+    if (!cb) { throw new Error('cb is required'); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    if (!options.CUSTOMER_ID) { return cb(new Error('options.CUSTOMER_ID is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(options.CUSTOMER_ID)) { return cb(new Error('options.CUSTOMER_ID is not a valid ObjectId'), null); }
+
+    var query;
+
+    query = self.find({
+      customer: options.CUSTOMER_ID,
+    });
+
+    query.populate('organization', 'name logoUrl');
+
+    if(options.populateStakeholders) {
+      query.populate('stakeholders.lp', 'name')
+      query.populate('stakeholders.org', 'name logoUrl')
+      query.populate('stakeholders.person', 'name avatarUrl title')
+    }
+    
+    query.exec(cb);
+
+  };
+
+  CapTable.statics.deleteForCustomer = function deleteForCustomer(options, cb) {
+
+    var self = this;
+
+    if (!cb) { throw new Error('cb is required'); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    if (!options.CUSTOMER_ID) { return cb(new Error('options.CUSTOMER_ID is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(options.CUSTOMER_ID)) { return cb(new Error('options.CUSTOMER_ID is not a valid ObjectId'), null); }
+
+    self.remove({ customer: options.CUSTOMER_ID, 'entered.by': { $ne: 'carta-parser' } }, cb);
 
   };
 
-  CapTable.statics.getAllForCustomer = function getAllForCustomer(options, cb) {
+  CapTable.statics.getForOrg = function getForOrg(options, cb) {
 
     var self = this;
 
@@ -144,6 +187,7 @@ module.exports = function(mongoose, config) {
 
     query = self.find({
       customer: options.CUSTOMER_ID,
+      organization: options.orgId
     });
 
     query.populate('organization', 'name logoUrl');
@@ -153,44 +197,64 @@ module.exports = function(mongoose, config) {
       query.populate('stakeholders.org', 'name logoUrl')
       query.populate('stakeholders.person', 'name avatarUrl title')
     }
-    
-    query.exec(cb);
 
   };
 
-  CapTable.statics.deleteAllForCustomer = function deleteAllForCustomer(customerId, cb) {
+  CapTable.statics.getByLP = function getByLP(options, cb) {
+
     var self = this;
-    self.remove({ customer: customerId, 'entered.by': { $ne: 'carta-parser' } }, cb);
-  };
 
-  CapTable.statics.getByLP = function getByLP(lpId, cb) {
+    if (!cb) { throw new Error('cb is required'); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    if (!options.CUSTOMER_ID) { return cb(new Error('options.CUSTOMER_ID is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(options.CUSTOMER_ID)) { return cb(new Error('options.CUSTOMER_ID is not a valid ObjectId'), null); }
 
-    var self = this;
+    query = self.find({
+      customer: options.CUSTOMER_ID,
+      'stakeholders.lp': options.lpId
+    });
 
     self
-      .find({ 'stakeholders.lp': lpId })
       .populate('organization', 'name logoUrl')
       .exec(cb);
 
   };
 
-  CapTable.statics.getByOrg = function getByOrg(orgId, cb) {
+  CapTable.statics.getByOrg = function getByOrg(options, cb) {
 
     var self = this;
 
+    if (!cb) { throw new Error('cb is required'); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    if (!options.CUSTOMER_ID) { return cb(new Error('options.CUSTOMER_ID is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(options.CUSTOMER_ID)) { return cb(new Error('options.CUSTOMER_ID is not a valid ObjectId'), null); }
+
+    query = self.find({
+      customer: options.CUSTOMER_ID,
+      'stakeholders.org': options.orgId
+    });
+
     self
-      .find({ 'stakeholders.org': orgId })
       .populate('organization', 'name logoUrl')
       .exec(cb);
 
   };
 
-  CapTable.statics.getByPerson = function getByPerson(personId, cb) {
+  CapTable.statics.getByPerson = function getByPerson(options, cb) {
 
     var self = this;
 
+    if (!cb) { throw new Error('cb is required'); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    if (!options.CUSTOMER_ID) { return cb(new Error('options.CUSTOMER_ID is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(options.CUSTOMER_ID)) { return cb(new Error('options.CUSTOMER_ID is not a valid ObjectId'), null); }
+
+    query = self.find({
+      customer: options.CUSTOMER_ID,
+      'stakeholders.person': options.personId
+    });
+
     self
-      .find({ 'stakeholders.person': personId })
       .populate('organization', 'name logoUrl')
       .exec(cb);
 

package/package.json

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