setup.js

/**
 * File setup.js
 * Author Young Jun Choi
 * Script for setting up and initializing a repository instance within MongoDB
 * Most codes are adapted from Data Mechanics at http://datamechanics.org by Andrei Lapets
 */

// Load the configuration file.
var config = JSON.parse(cat("config.json"));

// Create role capable of evaluating stored functions.
db = new Mongo().getDB("admin");
db.dropRole("evaluator");
db.createRole({
  role: "evaluator",
  privileges: [{ resource: { anyResource: true }, actions: ["anyAction"] }],
  roles: []
});

// Create administration account for repository.
db = new Mongo().getDB(config.repo.name);
db.dropUser(config.admin.name);
db.createUser({
  user: config.admin.name,
  pwd: config.admin.pwd,
  roles: [
    { role: "evaluator", db: "admin" },
    { role: "userAdmin", db: config.repo.name },
    { role: "readWrite", db: config.repo.name }
  ]
});

// Create repository users if they are not already present.
var userName = config.user.name;
var userPwd = config.user.pwd;
if (db.system.users.find({ user: userName }).count() > 0) {
  print(
    "Found '" + userName + "' user in admin database; not creating a new user."
  );
} else {
  db.dropRole(userName);
  db.createRole({
    role: userName,
    privileges: [],
    roles: [{ role: "read", db: config.repo.name }]
  });
  db.dropUser(userName);
  db.createUser({
    user: userName,
    pwd: userPwd,
    roles: [{ role: userName, db: config.repo.name }]
  });
}

// Save the custom server-side functions.
var currentUser =
  // Return the current user as a string.
  function() {
    return db.runCommand({ connectionStatus: 1 }).authInfo.authenticatedUsers[0]
      .user;
  };
db.system.js.save({ _id: "currentUser", value: currentUser });

var createCreate = function() {
  // Build the function that creates a new collection and
  // grants the user that created it write permissions.
  return eval(
    "(function(collName, user, pwd) {" +
      "  /* By default, use current user. */" +
      "  if (user == null)" +
      "    user = currentUser();" +
      "  if (pwd == null)" +
      "    pwd = " +
      config.user.pwd +
      ";" +
      "  /* Validate collection name as <user>.<collection>. */" +
      "  if (collName.split('.')[0] != user)" +
      "    collName = user + '.' + collName;" +
      "  var repo = new Mongo().getDB('" +
      config.repo.name +
      "');" +
      "  repo.auth('" +
      config.admin.name +
      "', '" +
      config.admin.pwd +
      "');" +
      "  repo.createCollection(collName);" +
      "  repo.createCollection(collName + '.metadata');" +
      "  repo.runCommand({grantPrivilegesToRole:user," +
      "    privileges: [" +
      "        { resource:{db:'" +
      config.repo.name +
      "', collection:collName }," +
      "          actions:['find','insert','remove','update','createIndex'] }," +
      "        { resource:{db:'" +
      config.repo.name +
      "', collection:collName + '.metadata' }," +
      "          actions:['find','insert','remove','update','createIndex'] }" +
      "      ]" +
      "  });" +
      "  repo.auth(user, pwd);" +
      "  return collName;" +
      "})"
  ); // eval()
};
db.system.js.save({ _id: "createCollection", value: createCreate() });

var createDrop = function() {
  // Build the function that drops a collection.
  return eval(
    "(function(collName, user, pwd) {" +
      "  /* By default, use current user. */" +
      "  if (user == null)" +
      "    user = currentUser();" +
      "  if (pwd == null)" +
      "    pwd = " +
      config.user.pwd +
      ";" +
      "  /* Validate collection name as <user>.<collection>. */" +
      "  if (collName.split('.')[0] != user)" +
      "    collName = user + '.' + collName;" +
      "  var repo = new Mongo().getDB('" +
      config.repo.name +
      "');" +
      "  repo.auth('" +
      config.admin.name +
      "', '" +
      config.admin.pwd +
      "');" +
      "  repo[collName].drop();" +
      "  repo[collName + '.metadata'].drop();" +
      "  repo.auth(user, pwd);" +
      "  return collName;" +
      "})"
  ); // eval()
};
db.system.js.save({ _id: "dropCollection", value: createDrop() });

print('Saved custom functions and scripts to "' + config.repo.name + '".');

/* eof */