linagora-rse/fixtures/populate/index.js

Linagora RSE

'use strict';

var q = require('q');
var extend = require('extend');
var mongoose = require('mongoose');
require('../../backend/core/db/mongo/models/domain');
require('../../backend/core/db/mongo/models/community');
require('../../backend/core/db/mongo/models/user');
var Domain = mongoose.model('Domain');
var Community = mongoose.model('Community');
var User = mongoose.model('User');
var helpers = require('../../backend/core/db/mongo/plugins/helpers');
helpers.applyPlugins();
helpers.patchFindOneAndUpdate();

var userDomainModule = require('../../backend/core/user/domain');
var populateObjects = require('./data/populate-objects');

var ADMIN_OBJECT = populateObjects.ADMIN;

var USER_OBJECT = populateObjects.USER;

var DOMAIN_OBJECT = populateObjects.DOMAIN;

var COMMUNITY_OBJECT = populateObjects.COMMUNITY;

function _populateAdmin(adminObject) {
  console.log('[INFO] POPULATE admin');

  return q.ninvoke(new User(adminObject || ADMIN_OBJECT), 'save');
}

function _populateDomain(domainObject, admin) {
  console.log('[INFO] POPULATE domain');

  var object = extend({}, domainObject || DOMAIN_OBJECT, { administrators: [{ user_id: admin[0] }] });

  return q.ninvoke(new Domain(object), 'save').then(domain => [admin[0], domain[0]]);
}

function _joinDomain(user, domain) {
  var deferred = q.defer();
  userDomainModule.joinDomain(user, domain, function(err) {
    if (err) { deferred.reject(err); }
    deferred.resolve([user, domain]);
  });
  return deferred.promise;
}

function _buildMember(id) {
  return {
    member: {
      objectType: 'user',
      id: id
    },
    status: 'joined'
  };
}

function _populateCommunity(admin, domain) {
  console.log('[INFO] POPULATE community');
  var object = extend({}, COMMUNITY_OBJECT);
  object.creator = admin._id;
  object.domain_ids = [domain._id];
  object.members = [_buildMember(admin._id)];
  var community = new Community(object);
  return q.ninvoke(community, 'save')
    .then(function() {
      return [community, domain];
    });
}

function _createUser(index, community, domain) {
  var userToSave = {
    firstname: USER_OBJECT.firstname + index,
    lastname: USER_OBJECT.lastname + index,
    password: USER_OBJECT.password,
    accounts: [{
      type: USER_OBJECT.accounts[0].type,
      emails: [USER_OBJECT.accounts[0].emails[0].replace(/(\w+)@/, '$1' + index + '@')]
    }]
  };

  var user = new User(userToSave);
  return q.ninvoke(user, 'save')
    .then(function(user) {
      return user[0];
    })
    .then(function(user) {
      return _joinDomain(user, domain);
    })
    .spread(function(user) {
      return q.ninvoke(
        Community, 'update',
        {
          _id: community._id,
          'members.user': {$ne: user._id}
        },
        {
          $push: { members: _buildMember(user._id) }
        });
    });
}

function _populateMembers(community, domain) {
  console.log('[INFO] POPULATE members');
  var createUsers = [];

  for (var i = 0; i < 20; i++) {
    var promise = _createUser(i, community, domain);

    createUsers.push(promise);
  }

  return q.allSettled(createUsers);
}

function populateDomainConfigurationAndTechnicalUsers(host, admin, domain) {
  console.log('[INFO] POPULATE Domain configuration and technical user');

  var technicalUsers = require('./data/technical-users');
  var configuration = require('./data/configuration');

  return q.all([technicalUsers([domain]), configuration([domain], host)])
    .then(function() {
      return q([admin, domain]);
    });
}

function populateAll(host) {
  console.log('[INFO] POPULATE the ESN');

  return _populateAdmin()
    .then(_populateDomain.bind(null, null))
    .spread(populateDomainConfigurationAndTechnicalUsers.bind(null, host))
    .spread(_joinDomain)
    .spread(_populateCommunity)
    .spread(_populateMembers);
}

function provisionDomainAndAdministrator(email, password) {
  const parts = email.split('@'),
        login = parts[0],
        domainName = parts[1],
        admin = {
          firstname: 'Admin',
          lastname: 'Admin',
          password: password || login,
          accounts: [{
            type: 'email',
            hosted: true,
            emails: [email]
          }]
        },
        domain = {
          name: domainName,
          company_name: domainName
        };

  return _populateAdmin(admin)
    .then(_populateDomain.bind(null, domain))
    .spread(_joinDomain);
}

module.exports = {
  populateAll,
  provisionDomainAndAdministrator,
  populateDomainConfigurationAndTechnicalUsers
};