db.js

var
    async = require('async')
  , mongodb = require('mongodb')
  ;

var utils = require('./utils');


var connect = function(config) {
  // set up database stuff
  var host = config.mongodb.server || 'localhost';
  var port = config.mongodb.port || mongodb.Connection.DEFAULT_PORT;
  var dbOptions = {
    auto_reconnect: config.mongodb.autoReconnect,
    poolSize: config.mongodb.poolSize
  };
  var db = new mongodb.Db('local', new mongodb.Server(host, port, dbOptions), {safe:true});


  var connections = {};
  var databases = [];
  var collections = {};
  //get admin instance
  var adminDb = db.admin();
  var mainConn; // main db connection


  // update the collections list
  var updateCollections = function(db, dbName, callback) {
    db.listCollections().toArray(function (err, result) {
      var names = [];

      for (var r in result) {
        names.push(result[r].name);
      }

      collections[dbName] = names.sort();

      if (callback) {
        callback(err);
      }
    });
  };

  // update database list
  var updateDatabases = function(admin) {
    admin.listDatabases(function(err, dbs) {
      if (err) {
        //TODO: handle error
        console.error(err);
      }

      for (var key in dbs.databases) {
        var dbName = dbs.databases[key]['name'];

        //'local' is special database, ignore it
        if (dbName == 'local') {
          continue;
        }

        if (config.mongodb.whitelist.length != 0) {
          if (!_.include(config.mongodb.whitelist, dbName)) {
            continue;
          }
        }
        if (config.mongodb.blacklist.length != 0) {
          if (_.include(config.mongodb.blacklist, dbName)) {
            continue;
          }
        }

        connections[dbName] = mainConn.db(dbName);
        databases.push(dbName);

        updateCollections(connections[dbName], dbName);
      }

      //Sort database names
      databases = databases.sort();
    });
  };


  // connect to mongodb database
  db.open(function(err, db) {
    if (err) {
      throw err;
    }

    console.log('Database connected!');

    mainConn = db;

    //Check if admin features are on
    if (config.mongodb.admin === true) {
      if (config.mongodb.adminUsername.length == 0) {
        console.log('Admin Database connected');
        updateDatabases(adminDb);
      } else {
        //auth details were supplied, authenticate admin account with them
        adminDb.authenticate(config.mongodb.adminUsername, config.mongodb.adminPassword, function(err, result) {
          if (err) {
            //TODO: handle error
            console.error(err);
          }

          console.log('Admin Database connected');
          updateDatabases(adminDb);
        });
      }
    } else {
      //Regular user authentication
      if (typeof config.mongodb.auth == "undefined" || config.mongodb.auth.length == 0) {
        throw new Error('Add auth details to config or turn on admin!');
      }

      async.forEachSeries(config.mongodb.auth, function(auth, callback) {
        console.log("Connecting to " + auth.database + "...");
        connections[auth.database] = mainConn.db(auth.database);
        databases.push(auth.database);

        if (typeof auth.username != "undefined" && auth.username.length != 0) {
          connections[auth.database].authenticate(auth.username, auth.password, function(err, success) {
            if (err) {
              //TODO: handle error
              console.error(err);
            }

            if (!success) {
              console.error('Could not authenticate to database "' + auth.database + '"');
            }

            updateCollections(connections[auth.database], auth.database);
            console.log('Connected!');
            callback();
          });
        } else {
          updateCollections(connections[auth.database], auth.database);
          console.log('Connected!');
          callback();
        }
      });
    }
  });
  
  return {
    updateCollections: updateCollections,
    connections: connections,
    databases: databases,
    collections: collections,
    adminDb: adminDb,
    mainConn: mainConn
  }
};

module.exports = connect;