deployd/lib/db.js

var config = require('./config').load()
  , mongodb = require('mongodb')
  , connectionString = config['db-host'] + config['db']
  , db
;

function ready(fn) {
  return function() {
    var _self = this
      , args = arguments
    ;
    
    db
      ? fn.apply(this, arguments)
      : mongodb.connect(connectionString, function(err, _db) {
          if(err || !_db) throw 'Could not connect to' + connectionString + ' (' + err + ')';
          db = _db;
          fn.apply(_self, args);
        })
    ;
  }
}

function collection(db, model, fn) {
  db.collection(model.collection, fn);
}

module.exports = {
  
  find: ready(function(model) {
    var query = model.toQuery()
      , promise
    ;
    
    if(!query) throw new Error('model does not return a query object from toQuery()');
    
    collection(db, model, function(err, collection) {
      
      
      collection.find(query).toArray(function(err, results) {
          if(err) throw err;
          model.refresh(results);
        })
      ;
    });
  }),
  
  upsert: ready(function(model) {
    var query = model.toQuery()
      , changes = model.toJSON()
      , options = {safe: true, upsert: true}
      , callback = function(err, result) {
          model.refresh(changes);
        }
    ;  
    
    collection(db, model, function(err, collection) {
      if(query && query._id) {
        collection.update(query, changes, options, callback);
      } else {
        collection.insert(changes, callback)
      }
    }); 
  }),
  
  remove: ready(function(model) {
    collection(db, model, function(err, collection) {
      collection.remove(model.toQuery()._id);
      model.refresh({removed: true});
    });
  }),
  
  index: ready(function(model, property, options, fn) {
    collection(db, model, function(err, collection) {
      var index = {};
      index[property] = 1;
      
      collection.ensureIndex(index, options, fn);
    });
  })
  
}