@dhyasama/totem-models 8.107.0 → 8.109.0

package/lib/CapTable.js

@@ -58,7 +58,7 @@ module.exports = function(mongoose, config) {
         required: false,
         validate: {
           validator: function(v) {
-            if (v && this.lp || this.org || this.person) return false;
+            if (v && (this.lp || this.org || this.person)) return false;
             else return true;
           },
           message: 'A stakeholder can only be linked to one of the following: fund, lp, org, person'
@@ -73,7 +73,7 @@ module.exports = function(mongoose, config) {
         required: false,
         validate: {
           validator: function(v) {
-            if (v || this.fund || this.org || this.person) return false;
+            if (v && (this.fund || this.org || this.person)) return false;
             else return true;
           },
           message: 'A stakeholder can only be linked to one of the following: fund, lp, org, person'
@@ -88,7 +88,7 @@ module.exports = function(mongoose, config) {
         required: false,
         validate: {
           validator: function(v) {
-            if (v || this.fund || this.lp || this.person) return false;
+            if (v && (this.fund || this.lp || this.person)) return false;
             else return true;
           },
           message: 'A stakeholder can only be linked to one of the following: fund, lp, org, person'
@@ -103,7 +103,7 @@ module.exports = function(mongoose, config) {
         required: false,
         validate: {
           validator: function(v) {
-            if (v && this.fund || this.lp || this.org) return false;
+            if (v && (this.fund || this.lp || this.org)) return false;
             else return true;
           },
           message: 'A stakeholder can only be linked to one of the following: fund, lp, org, person'
@@ -269,64 +269,104 @@ module.exports = function(mongoose, config) {
     self.findByIdAndRemove(id, cb);
   };
 
+  CapTable.statics.getForCustomer = function getForCustomer(orgId, customerId, cb) {
+
+    var self = this;
+
+    self
+      .findOne({customer: customerId, organization: orgId })
+      query.populate('stakeholders.fund', 'name shortName')
+      query.populate('stakeholders.lp', 'name')
+      query.populate('stakeholders.org', 'name logoUrl')
+      query.populate('stakeholders.person', 'name avatarUrl title')
+      .exec(cb);
+
+  };
+
+  CapTable.statics.getAllForCustomer = function getAllForCustomer(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); }
+
+    var query;
+
+    query = self.find({
+      customer: options.CUSTOMER_ID,
+    });
+
+    query.populate('organization', 'name logoUrl');
+
+    if(options.populateStakeholders) {
+      query.populate('stakeholders.fund', 'name shortName')
+      query.populate('stakeholders.lp', 'name')
+      query.populate('stakeholders.org', 'name logoUrl')
+      query.populate('stakeholders.person', 'name avatarUrl title')
+    }
+    
+    query.exec(cb);
+
+  };
+
   CapTable.statics.deleteAllForCustomer = function(customerId, cb) {
     var self = this;
     self.remove({ customer: customerId, 'entered.by': { $ne: 'carta-parser' } }, cb);
   };
 
-  CapTable.statics.getAllForCustomer = function getAllForCustomer(customerId, cb) {
+  CapTable.statics.getByFunds = function getByFunds(fundIds, cb) {
 
     var self = this;
 
     self
-      .find({customer: customerId })
+      .find({ 'stakeholders.fund': { $in: fundIds } })
       .populate('organization', 'name logoUrl')
       .exec(cb);
 
   };
 
-  CapTable.statics.getAllForCustomer2 = function getAllForCustomer2(customerId, cb) {
+  CapTable.statics.getByFund = function getByFund(fundId, cb) {
 
     var self = this;
 
     self
-      .find({customer: customerId })
-      .populate('stakeholders.person', 'name avatarUrl title')
-      .populate('stakeholders.fund', 'name shortName')
+      .find({ 'stakeholders.fund': fundId })
+      .populate('organization', 'name logoUrl')
       .exec(cb);
 
   };
 
-  CapTable.statics.getByFunds = function getByFunds(fundIds, cb) {
+  CapTable.statics.getByLP = function getByLP(lpId, cb) {
 
     var self = this;
 
     self
-      .find({ 'stakeholders.fund': { $in: fundIds } })
+      .find({ 'stakeholders.lp': lpId })
       .populate('organization', 'name logoUrl')
       .exec(cb);
 
   };
 
-  CapTable.statics.getByPerson = function getByFunds(personId, cb) {
+  CapTable.statics.getByOrg = function getByOrg(orgId, cb) {
 
     var self = this;
 
     self
-      .find({ 'stakeholders.person': personId })
+      .find({ 'stakeholders.org': orgId })
       .populate('organization', 'name logoUrl')
       .exec(cb);
 
   };
 
-  CapTable.statics.getForCustomer = function getForCustomer(orgId, customerId, cb) {
+  CapTable.statics.getByPerson = function getByPerson(personId, cb) {
 
     var self = this;
 
     self
-      .findOne({customer: customerId, organization: orgId })
-      .populate('stakeholders.person', 'name avatarUrl title')
-      .populate('stakeholders.fund', 'name shortName')
+      .find({ 'stakeholders.person': personId })
+      .populate('organization', 'name logoUrl')
       .exec(cb);
 
   };

package/package.json

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