@dhyasama/totem-models 9.5.0 → 9.8.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,33 +134,74 @@ module.exports = function(mongoose, config) {
   // Statics operate on the entire collection
   //////////////////////////////////////////////////////
 
-  CapTable.statics.getForCustomer = function getForCustomer(orgId, customerId, cb) {
+  CapTable.statics.getForCustomer = function getForCustomer(customerid, 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 (!customerid) { return cb(new Error('customerid is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(customerid)) { return cb(new Error('customerid is not a valid ObjectId'), null); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    
+    var query;
+
+    query = self.find({
+      customer: customerid,
+    });
+
+    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.getAllForCustomer = function getAllForCustomer(options, cb) {
+  CapTable.statics.deleteForCustomer = function deleteForCustomer(customerid, options, cb) {
 
     var self = this;
 
     if (!cb) { throw new Error('cb is required'); }
+    if (!customerid) { return cb(new Error('customerid is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(customerid)) { return cb(new Error('customerid is not a valid ObjectId'), null); }
     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: customerid, 'entered.by': { $ne: 'carta-parser' } }, cb);
+
+  };
 
+  CapTable.statics.getForOrg = function getForOrg(orgid, options, cb) {
+
+    var self = this;
+
+    if (!cb) { throw new Error('cb is required'); }
+    if (!orgid) { return cb(new Error('orgid is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(orgid)) { return cb(new Error('orgid is not a valid ObjectId'), null); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    
     var query;
 
-    query = self.find({
-      customer: options.CUSTOMER_ID,
-    });
+    if (options.isWorkerProcess) {
+      query = self.find({
+        organization: orgid
+      });
+    }
+
+    else {
+    
+      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,
+        organization: orgid
+      });
+
+    }
 
     query.populate('organization', 'name logoUrl');
 
@@ -153,46 +210,108 @@ 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(lpid, 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 (!lpid) { return cb(new Error('lpid is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(lpid)) { return cb(new Error('lpid is not a valid ObjectId'), null); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    
+    if (options.isWorkerProcess) {
+      query = self.find({
+        'stakeholders.lp': lpid
+      });
+    }
+
+    else {
 
-    var self = this;
+      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.lp': lpid
+      });
 
-    self
-      .find({ 'stakeholders.lp': lpId })
-      .populate('organization', 'name logoUrl')
-      .exec(cb);
+    }
+
+    query.populate('organization', 'name logoUrl')
+    query.exec(cb);
 
   };
 
-  CapTable.statics.getByOrg = function getByOrg(orgId, cb) {
+  CapTable.statics.getByOrg = function getByOrg(orgid, options, cb) {
 
     var self = this;
 
-    self
-      .find({ 'stakeholders.org': orgId })
-      .populate('organization', 'name logoUrl')
-      .exec(cb);
+    if (!cb) { throw new Error('cb is required'); }
+    if (!orgid) { return cb(new Error('orgid is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(orgid)) { return cb(new Error('orgid is not a valid ObjectId'), null); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    
+    var query;
+
+    if (options.isWorkerProcess) {
+      query = self.find({
+        'stakeholders.org': orgid
+      });
+    }
+
+    else {
+
+      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': orgid
+      });
+
+    }
+
+    query.populate('organization', 'name logoUrl')
+    query.exec(cb);
 
   };
 
-  CapTable.statics.getByPerson = function getByPerson(personId, cb) {
+  CapTable.statics.getByPerson = function getByPerson(personid, options, cb) {
 
     var self = this;
 
-    self
-      .find({ 'stakeholders.person': personId })
-      .populate('organization', 'name logoUrl')
-      .exec(cb);
+    if (!cb) { throw new Error('cb is required'); }
+    if (!personid) { return cb(new Error('personid is required'), null); }
+    if (!mongoose.Types.ObjectId.isValid(personid)) { return cb(new Error('personid is not a valid ObjectId'), null); }
+    if (!options) { return cb(new Error('options is required'), null); }
+    
+    var query;
+
+    if (options.isWorkerProcess) {
+      query = self.find({
+        'stakeholders.person': personid
+      });
+    }
+
+    else {
+
+      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': personid
+      });
+
+    }
+
+    query.populate('organization', 'name logoUrl')
+    query.exec(cb);
 
   };
 
@@ -203,7 +322,7 @@ module.exports = function(mongoose, config) {
     if (!capTable) { return cb(new Error('capTable is required'), null); }
     if (!username) { return cb(new Error('username is required'), null); }
 
-    capTable.constructor.getForCustomer(capTable.organization, capTable.customer, function(err, result) {
+    capTable.constructor.getForOrg(capTable.organization, { CUSTOMER_ID: capTable.customer }, function(err, result) {
 
       if (err) return cb(err, null);
       if (result) return cb(new Error('A cap table already exists for this customer/org combo'), null);

package/package.json

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