deployd/lib/db.js

var config = require('./config').load('db.json')
  , mongodb = require('mongodb')
  , APP_NAME = 'deployd' // TODO - make app name dynamic
  , db
;

function ready(fn) {
  return function() {
    var _self = this
      , args = arguments
    ;
    
    db
      ? fn.apply(this, arguments)
      : mongodb.connect(config.host + APP_NAME, function(err, _db) {
          if(err || !_db) throw 'Could not connect to' + config.host + APP_NAME + ' (' + 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();
    
    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}
      , 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});
    });
  })
  
}