add detection for toMongo method

.gitignore

@@ -1,3 +1,4 @@
 node_modules
 script/
 coverage.html
+.tern-port

lib/mongo.js

@@ -52,7 +52,13 @@ module.exports = function(url) {
 
     Model.save = function(cb) {
       var self = this;
-      return db.insert(this.toJSON(), function(err, docs) {
+      var doc;
+      if (this.toMongo && typeof this.toMongo === 'function') {
+        doc = this.toMongo();
+      } else {
+        doc = this.toJSON();
+      }
+      return db.insert(doc, function(err, docs) {
         var doc = docs ? docs[0] : null;
         if(err) {
           // Check for duplicate index

test/test.js

@@ -15,6 +15,8 @@ var User = modella('User')
   .attr('email', {unique: true})
   .attr('password');
 
+User.use(mongo);
+
 var AtomicUser = modella('AtomicUser')
   .attr('_id')
   .attr('name')
@@ -24,10 +26,42 @@ var AtomicUser = modella('AtomicUser')
   .attr('email', {unique: true})
   .attr('password');
 
-
-User.use(mongo);
 AtomicUser.use(mongo);
 
+var OverrideUser = modella('OverrideUser')
+  .attr('_id')
+  .attr('name')
+  .attr('password');
+
+OverrideUser.use(mongo);
+
+OverrideUser.prototype.toMongo = function() {
+  var dump = {};
+  var self = this;
+
+  Object.keys(this.attrs).forEach(function (key) {
+    var val = self.attrs[key];
+
+    dump[key] = !!val.toJSON ? val.toJSON() : modella.utils.clone(val);
+  });
+
+  return dump;
+};
+
+OverrideUser.prototype.toJSON = function() {
+  var dump = {};
+  var self = this;
+
+  Object.keys(this.attrs).forEach(function (key) {
+    if (key === 'password') return;
+    var val = self.attrs[key];
+
+    dump[key] = !!val.toJSON ? val.toJSON() : modella.utils.clone(val);
+  });
+
+  return dump;
+};
+
 /**
  * Initialize
  */
@@ -36,12 +70,15 @@ var user = new User();
 
 var col = db.collection("User");
 var atomiccol = db.collection("AtomicUser");
+var overridecol = db.collection("OverrideUser");
 
 
 describe("Modella-Mongo", function() {
   before(function(done) {
     col.remove({}, function() {
-      atomiccol.remove({}, done);
+      atomiccol.remove({}, function() {
+        overridecol.remove({}, done);
+      });
     });
   });
 
@@ -77,6 +114,21 @@ describe("Modella-Mongo", function() {
         });
       });
 
+      it("saves the record using toMongo if present", function(done) {
+        var user = new OverrideUser({name: 'Ryan', email: 'ryan@slingingcode.com', password: 'foobar123'});
+        user.save(function(err, u) {
+          expect(user.primary()).to.be.ok();
+          overridecol.findOne({}, function(err, u) {
+            expect(u).to.be.ok();
+            expect(u).to.have.property('name', 'Ryan');
+            expect(u).to.have.property('password', 'foobar123');
+            var uJSON = user.toJSON();
+            expect(uJSON).to.not.have.property('password');
+            done();
+          });
+        });
+      });
+
       it("triggers errors if there is an error", function(done) {
         var user = new User({name: 'Ryan', email: 'ryan@slingingcode.com'});
         user.save();