@dhyasama/totem-models 10.8.0 → 10.10.1

package/lib/Organization.js

@@ -33,6 +33,12 @@ module.exports = function(mongoose, config) {
         linkedin: { type: String, trim: true, unique: true, partialFilterExpression : { type :"string" } }
       },
 
+      social_handle: {
+        twitter: { type: String, trim: true },
+        facebook: { type: String, trim: true },
+        linkedin: { type: String, trim: true }
+      },
+
       contact: {
 
         phone: [{
@@ -1963,7 +1969,7 @@ module.exports = function(mongoose, config) {
 
   Organization.pre('save', function(next) {
 
-    let prepForSave = function prepForSave(doc) {
+    const prepForSave = function prepForSave(doc) {
 
       var cleanContactInfo = function cleanContactInfo(obj) {
 
@@ -2088,7 +2094,46 @@ module.exports = function(mongoose, config) {
 
     };
 
+    const setSocialHandles = function setSocialHandles(doc) {
+
+      const getHandle = function getHandle(url) {
+
+        let handle = "";
+
+        if (!url) return handle;
+
+        try {
+
+          const segments = url.split("/");
+
+          // normal case, e.g., http://facebook.com/hello
+          if (segments[segments.length - 1]) handle = segments[segments.length - 1];
+
+          // trailing slash, e.g., http://facebook.com/hello/
+          else if (segments[segments.length - 2]) handle = segments[segments.length - 2];
+
+          else handle = "";
+
+        }
+        catch (err) {
+          console.log('Error for url', url);
+          console.error(err);
+        }
+
+        return handle;
+
+      };
+
+      if (!doc) return;
+      if (!doc.social) return;
+      doc.social_handle.facebook = getHandle(doc.social.facebook);
+      doc.social_handle.twitter = getHandle(doc.social.twitter);
+      doc.social_handle.linkedin = getHandle(doc.social.linkedin);
+
+    }
+
     prepForSave(this);
+    setSocialHandles(this);
 
     return next();
 

package/lib/Round.js

@@ -399,7 +399,7 @@ module.exports = function(mongoose, config) {
 
     let query = self.find({ 'vehicles.fund': fundId });
     query.select('organization roundName vehicles');
-    query.populate('organization', 'name description logoUrl website websiteAliases filters status');
+    query.populate('organization', 'name description logoUrl website websiteAliases filters status ipo closed acquired operating');
 
     if (options.isWorkerProcess) {
       query.populate('vehicles.investments');

package/migrations/social_handles.js

@@ -0,0 +1,119 @@
+"use strict";
+
+const mongoose = require('mongoose');
+const models = require('../index')(mongoose, { suppressDebugLog: true });
+const Organization = mongoose.model('Organization');
+const db_uri = "mongodb://jason:T576G150HPXLA5q4BJ2o2zj747B5030x@production-shard-00-00-gv0f3.mongodb.net:27017,production-shard-00-01-gv0f3.mongodb.net:27017,production-shard-00-02-gv0f3.mongodb.net:27017/production?replicaSet=Production-shard-0&ssl=true&authSource=admin";
+const _ = require('underscore');
+const async = require('async');
+
+let dirtyOrgs = [];
+let savedOrgCount = 0;
+
+const getHandle = function getHandle(url) {
+
+  let handle = "";
+
+  try {
+
+    const segments = url.split("/");
+
+    // normal case, e.g., http://facebook.com/hello
+    if (segments[segments.length - 1]) handle = segments[segments.length - 1];
+
+    // trailing slash, e.g., http://facebook.com/hello/
+    else if (segments[segments.length - 2]) handle = segments[segments.length - 2];
+
+    else handle = "";
+
+  }
+  catch (err) {
+    console.log('Error for url', url);
+    console.error(err);
+  }
+
+  //console.log(handle, url);
+
+  return handle;
+
+};
+
+const saveOrg = function(org, cb) {
+
+  Organization.upsert(org, 'social-handle-transformation', function(err) {
+    if (err) console.error(err);
+    else {
+      savedOrgCount += 1;
+      //console.log(org.name);
+    }
+    return cb();
+  });
+
+}
+
+mongoose.connect(db_uri, function(err) {
+
+  if (err) {
+    console.error(err);
+    process.exit();
+  }
+
+  const page = 9;
+  const limit = 10000;
+  const skip = (page - 1) * limit;
+
+  console.log('Getting page', page);
+
+  Organization
+    .find({
+      $and: [
+        { social: { $exists: true } },
+        {
+          $or: [
+            { "social.facebook": { $exists: true } },
+            { "social.twitter": { $exists: true } },
+            { "social.linkedin": { $exists: true } }
+          ]
+        }
+      ]
+    })
+    .skip(skip)
+    .limit(limit)
+    .sort({"name": "asc"})
+    .exec(function(err, orgs) {
+
+      if (err) {
+        console.error(err);
+        process.exit();
+      }
+
+      console.log('Found', orgs.length, 'orgs');
+
+      orgs.forEach(function(org) {
+
+        let dirty = false;
+
+          if (org.social.facebook || org.social.twitter || org.social.linkedin) {
+            dirty = true;
+          }
+
+          if (dirty) dirtyOrgs.push(org);
+
+      });
+
+      console.log(dirtyOrgs.length, 'orgs need to be saved');
+
+      async.each(dirtyOrgs, saveOrg, function(err) {
+
+        if (err) console.error(err);
+
+        console.log('Saved', savedOrgCount, 'orgs');
+        console.log('Done page', page);
+
+        process.exit();
+
+      });
+
+    });
+
+});

package/package.json

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

package/test/Organization.js

@@ -19,7 +19,7 @@ var someOtherLp;
 var account;
 var customer1, customer2;
 
-describe('Organization', function() {
+describe.only('Organization', function() {
 
   var org1, org2, org3, org4, org5;
   var mergeId = new mongoose.Types.ObjectId();
@@ -448,6 +448,7 @@ describe('Organization', function() {
         org1 = org;
         done();
       });
+
     }); // end adds an org
 
     it('finds a fuzzy match by name', function (done) {
@@ -1148,6 +1149,28 @@ describe('Organization', function() {
 
     });
 
+    it('sets social handles', function (done) {
+
+      const o = new Organization();
+      o.name = 'Social Butterfly';
+      o.slug = 'sb';
+      o.social = {
+        facebook: 'https://facebook.com/socialbutterfly',
+        twitter: 'https://twitter.com/socialbutterfly',
+        linkedin: 'http://linkedin.com/in/socialbutterfly',
+      }
+
+      Organization.upsert(o, 'test-user', function (err, org) {
+        should.not.exist(err);
+        should.exist(org);
+        org.social_handle.facebook.should.equal('socialbutterfly');
+        org.social_handle.twitter.should.equal('socialbutterfly');
+        org.social_handle.linkedin.should.equal('socialbutterfly');
+        done();
+      });
+
+    });
+
   });
 
   describe('Deals', function() {