@dhyasama/totem-merge 6.0.1

package/lib/organizations.js

@@ -0,0 +1,408 @@
+"use strict";
+
+let async = require('async');
+let _ = require('underscore');
+let helpers = require('./helpers');
+let mongoose = require('mongoose');
+let ObjectId = mongoose.Types.ObjectId;
+let logger;
+
+var merge = function merge(context, target, source, mergedBy, customerId, cb) {
+
+  logger = context.logger;
+
+  logger.writeToInfoLog('Merge organizations...');
+
+  // validate parameters
+  if (!context) return cb(new Error('context is required'), null);
+  if (!target) return cb(new Error('target is required'), null);
+  if (!source) return cb(new Error('source is required'), null);
+  if (!mergedBy) return cb(new Error('mergedBy is required'), null);
+  if (!cb) throw new Error('cb is required');
+
+  var saveOrganization = function(org, username, callback) {
+
+    logger.writeToInfoLog('Save target organization...');
+    logger.writeToInfoLog('  ID: ' + org._id.toString());
+    logger.writeToInfoLog('  LinkedIn: ' + org.social.linkedin);
+    logger.writeToInfoLog('  Twitter: ' + org.social.twitter);
+    logger.writeToInfoLog('  Domains: ' + org.websites);
+
+    context.Organization.upsert(org, username, callback);
+
+  };
+
+  if (context.flagOnly) {
+
+    var url = 'https://admin.ffvc.com/orgs/' + source.id;
+    var text = 'This org should be merged with ' + url;
+
+    context.Organization.addFlag(target._id, target._id, text, customerId, function(err, result) {
+      if (err) return cb(err, null);
+      return cb(null, result);
+    });
+
+  }
+  else {
+
+    // mergedBy can be either an account or a username
+    var username = mergedBy.username ? mergedBy.username : mergedBy;
+    logger.writeToInfoLog('Username: ' + username);
+
+    async.series([
+
+      verifyOkToMerge.bind(null, target, source),
+
+      // Create a snapshot of the target
+      helpers.createSnapshot.bind(null, context, 'Organization', target, username),
+
+      // Create a snapshot of the source
+      helpers.createSnapshot.bind(null, context, 'Organization', source, username),
+
+      // Merge the source into the target
+      mergeSourceIntoTarget.bind(null, target, source),
+
+      // Delete the source
+      // Do this prior to saving merged target in order to avoid unique key issues
+      helpers.removeEntity.bind(null, context, source),
+
+      // Save the updated target
+      saveOrganization.bind(null, target, username),
+
+      // Change source references to target
+      updateReferences.bind(null, context, target, source, mergedBy, customerId)
+
+    ], function(err, results) {
+
+      if (err) { return cb(err, null); }
+
+      // result from saveOrganization
+      var updatedTarget = results[5];
+
+      // result from updateOrganizations within updateReferences
+      var updatedReferencesOrgs = results[6][4];
+
+      // Check if target had reference(s) updated
+      var targetMatch = _.find(updatedReferencesOrgs, function(o) {
+        return o.id.toString() === target.id.toString();
+      });
+
+      // Return up to date target
+      if (targetMatch) { return cb(null, targetMatch); }
+      else { return cb(null, updatedTarget); }
+
+    });
+
+  }
+
+};
+
+var mergeSourceIntoTarget = function mergeEntities(target, source, cb) {
+
+  logger.writeToInfoLog('Merge source into target...');
+
+  var doNotMergeKeys = [
+    'merged',
+    'deleted',
+    'entered',
+    '__v'
+  ];
+
+  helpers.mergeObjects(target, source, doNotMergeKeys);
+
+  // keep a list of copied orgs
+  target.merge(source.id);
+
+  // add copied org name as an alias if it's different from source name
+  if (target.name.toLowerCase() != source.name.toLowerCase()) {
+    target.aliases = target.aliases.concat([source.name]);
+    target.aliases = _.uniq(target.aliases);
+  }
+
+  // Add source website as alias for target
+  target.websiteAliases = target.websiteAliases.concat(source.website);
+  target.websiteAliases = _.uniq(target.websiteAliases);
+
+  // mark as entered by system
+  target.entered = {
+    by: 'ff-merge',
+    on: new Date()
+  };
+
+  return cb(null, target);
+
+};
+
+var verifyOkToMerge = function verifyOkToMerge(target, source, cb) {
+
+  logger.writeToInfoLog('Verify ok to merge...');
+
+  // Cannot merge if target is a customer
+  if (target && target.customer && target.customer.totemUrl) { return cb(true, null); }
+
+  // Cannot merge if source is a customer
+  else if (source && source.customer && source.customer.totemUrl) { return cb(true, null); }
+
+  // Ok to merge
+  else { return cb(null, null); }
+
+};
+
+var updateReferences = function updateReferences(context, target, source, mergedBy, customerId, cb) {
+
+  logger.writeToInfoLog('Update references...');
+
+  // validate parameters
+  if (!context) return cb(new Error('context is required'), null);
+  if (!target) return cb(new Error('target is required'), null);
+  if (!source) return cb(new Error('source is required'), null);
+  if (!cb) throw new Error('cb is required');
+
+  var oldOrgId = ObjectId(source.id);
+  var newOrgId = ObjectId(target.id);
+
+  logger.writeToInfoLog('  Target: ' + newOrgId.toString());
+  logger.writeToInfoLog('  Source: ' + oldOrgId.toString());
+
+  // mergedBy can be either an account or a username
+  var username = mergedBy.username ? mergedBy.username : mergedBy;
+
+  // Do not update cap tables, financials or rounds as these are complicated
+  // and get cleaned and updated on each pull anyway
+
+  async.parallel([
+    updateDeals.bind(context, newOrgId, oldOrgId, mergedBy, customerId),
+    updateLists.bind(context, newOrgId, oldOrgId, username),
+    updateMessages.bind(context, newOrgId, oldOrgId, username),
+    updateNews.bind(context, newOrgId, oldOrgId, username),
+    updateOrganizations.bind(context, newOrgId, oldOrgId, username)
+  ], cb)
+
+};
+
+var updateDeals = function updateDeals(newOrgId, oldOrgId, user, customerId, cb) {
+
+  logger.writeToInfoLog('  Updating deals...');
+
+  var self = this;
+
+  // Get all deals for this org
+  self.Deal.collection.find({ 'organization': oldOrgId }).toArray(function(err, items) {
+
+    if (err) return cb(err, null);
+    else if (!items || items.length === 0) return cb(null, []);
+
+    var tasks = [];
+    var updated = [];
+
+    _.each(items, function(item) {
+      tasks.push({
+        id: item._id,
+        update: { $set: { organization: newOrgId } }
+      });
+    });
+
+    logger.writeToInfoLog('    Found ' + tasks.length + ' deal references to update');
+
+    var queue = async.queue(function(task, callback) {
+      self.Deal.modifyById(task.id, task.update, { CUSTOMER_ID: customerId }, function(err, result) {
+        if (err) { return callback(err, null); }
+        else if (result) { updated.push(result); }
+        return callback(null, result);
+      });
+    }, 8);
+
+    queue.drain = function() { return cb(null, updated); };
+
+    queue.push(tasks);
+
+  });
+
+};
+
+var updateLists = function updateLists(newOrgId, oldOrgId, user, cb) {
+
+  logger.writeToInfoLog('  Updating lists...');
+
+  var self = this;
+
+  self.List.collection.find({ 'organizations': oldOrgId }).toArray(function(err, items) {
+
+    if (err) return cb(err, null);
+    else if (!items || items.length === 0) return cb(null, []);
+
+    var tasks = [];
+    var updated = [];
+
+    _.each(items, function(item) {
+
+      var orgs = _.map(item.organizations, function(o) {
+        if (o.toString() === oldOrgId.toString()) { o = newOrgId; }
+        return o;
+      });
+
+      tasks.push({
+        id: item._id,
+        update: { $set: { organizations: orgs } }
+      });
+
+    });
+
+    logger.writeToInfoLog('    Found ' + tasks.length + ' list references to update');
+
+    var queue = async.queue(function(task, callback) {
+      self.List.modifyById(task.id, task.update, function(err, result) {
+        if (err) { return callback(err, null); }
+        else if (result) { updated.push(result); }
+        return callback(null, result);
+      });
+    }, 8);
+
+    queue.drain = function() { return cb(null, updated); };
+
+    queue.push(tasks);
+
+  });
+
+};
+
+var updateMessages = function updateMessages(newOrgId, oldOrgId, user, cb) {
+
+  logger.writeToInfoLog('  Updating messages...');
+
+  var self = this;
+
+  self.Message.collection.find({ 'organization': oldOrgId }).toArray(function(err, items) {
+
+    if (err) return cb(err, null);
+    else if (!items || items.length === 0) return cb(null, []);
+
+    var tasks = [];
+    var updated = [];
+
+    _.each(items, function(item) {
+      tasks.push({
+        id: item._id,
+        update: { $set: { organization: newOrgId } }
+      });
+    });
+
+    logger.writeToInfoLog('    Found ' + tasks.length + ' message references to update');
+
+    var queue = async.queue(function(task, callback) {
+      self.Message.modifyById(task.id, task.update, function(err, result) {
+        if (err) { return callback(err, null); }
+        else if (result) { updated.push(result); }
+        return callback(null, result);
+      });
+    }, 8);
+
+    queue.drain = function() { return cb(null, updated); };
+
+    queue.push(tasks);
+
+  });
+
+};
+
+var updateNews = function updateNews(newOrgId, oldOrgId, user, cb) {
+
+  logger.writeToInfoLog('  Updating news...');
+
+  var self = this;
+
+  self.News.collection.find({ 'org': oldOrgId }).toArray(function(err, items) {
+
+    if (err) return cb(err, null);
+    else if (!items || items.length === 0) return cb(null, []);
+
+    var tasks = [];
+    var updated = [];
+
+    _.each(items, function(item) {
+      tasks.push({
+        id: item._id,
+        update: { $set: { org: newOrgId } }
+      });
+    });
+
+    logger.writeToInfoLog('    Found ' + tasks.length + ' news references to update');
+
+    var queue = async.queue(function(task, callback) {
+      self.News.modifyById(task.id, task.update, function(err, result) {
+        if (err) { return callback(err, null); }
+        else if (result) { updated.push(result); }
+        return callback(null, result);
+      });
+    }, 8);
+
+    queue.drain = function() { return cb(null, updated); };
+
+    queue.push(tasks);
+
+  });
+
+};
+
+var updateOrganizations = function updateOrganizations(newOrgId, oldOrgId, user, cb) {
+
+  logger.writeToInfoLog('  Updating orgs...');
+
+  var self = this;
+
+  self.Organization.collection.find({ $or: [{ 'merged': oldOrgId }, { 'operating.acquired.public.by': oldOrgId }, { 'operating.acquired.private.by': oldOrgId }, { 'related': oldOrgId }] }).toArray(function(err, items) {
+
+    if (err) return cb(err, null);
+    else if (!items || items.length === 0) return cb(null, []);
+
+    var tasks = [];
+    var updated = [];
+
+    _.each(items, function(item) {
+
+      var setObj = {};
+
+      setObj.merged = _.map(item.merged, function(o) {
+        if (o.toString() === oldOrgId) { o = newOrgId; }
+        return o;
+      });
+
+      setObj.related = _.map(item.related, function(o) {
+        if (o.toString() === oldOrgId) { o = newOrgId; }
+        return o;
+      });
+
+      var publicBy = item["operating.acquired.public.by"];
+      var privateBy = item["operating.acquired.private.by"];
+
+      if (publicBy && publicBy.toString() === oldOrgId) { setObj["operating.acquired.public.by"] = newOrgId; }
+      if (privateBy && privateBy.toString() === oldOrgId) { setObj["operating.acquired.private.by"] = newOrgId; }
+
+      tasks.push({
+        id: item._id,
+        update: {
+          $set: setObj
+        }
+      });
+
+    });
+
+    logger.writeToInfoLog('    Found ' + tasks.length + ' org references to update');
+
+    var queue = async.queue(function(task, callback) {
+      self.Organization.modifyById(task.id, task.update, function(err, result) {
+        if (err) { return callback(err, null); }
+        else if (result) { updated.push(result); }
+        return callback(null, result);
+      });
+    }, 8);
+
+    queue.drain = function() { return cb(null, updated); };
+
+    queue.push(tasks);
+
+  });
+
+};
+
+module.exports.merge = merge;