added class and instance method support

doc/views/index/code/instances/models-3.ejs → doc/views/index/code/instances/update.ejs

 // way 1
 task.title = 'a very different title now'
-task.save(function(){})
+task.save().on('success', function() {})
 
 // way 2
 task.updateAttributes({
   title: 'a very different title now'
-}, function(){})
+}).on('success', function() {})

doc/views/index/instances.ejs

@@ -13,4 +13,4 @@ To get it stored in the database, use the save method and catch the events, ...,
 
 Now lets change some values and save changes to the database... There are two ways to do that:
 
-<pre><%- koala(".js", partial("code/instances/models-3.ejs")) %></pre>
+<pre><%- koala(".js", partial("code/instances/update.ejs")) %></pre>

lib/sequelize/model-definition.js

@@ -16,9 +16,15 @@ var ModelDefinition = module.exports = function(name, attributes, options) {
   this.associations = {}
 
   this.addDefaultAttributes()
+  this.addOptionalClassMethods()
 }
 Utils.addEventEmitter(ModelDefinition)
 
+ModelDefinition.prototype.addOptionalClassMethods = function() {
+  var self = this
+  Utils._.each(this.options.classMethods || {}, function(fct, name) { self[name] = fct })
+}
+
 ModelDefinition.prototype.addDefaultAttributes = function() {
   var defaultAttributes = {id: {type: DataTypes.INTEGER, allowNull: false, primaryKey: true, autoIncrement: true}}
     , self = this
@@ -117,6 +123,7 @@ ModelDefinition.prototype.build = function(values, options) {
       instance.addAttribute(name, value)
     }
   })
+  Utils._.each(this.options.instanceMethods || {}, function(fct, name) { instance[name] = fct })
 
   Utils._.each(this.associations, function(association, associationName) {
     association.injectGetter(instance)

test/Sequelize/define.js

@@ -62,5 +62,17 @@ module.exports = {
   'tablenames should be pluralized if they are not frozen': function() {
     var User = sequelize.define('User', {}, {freezeTableName: false})
     assert.eql(User.tableName, 'Users')
+  },
+  'it should add the passed class/instance methods': function() {
+    var User = sequelize.define('User', {}, {
+      classMethods: { doSmth: function(){ return 1 } },
+      instanceMethods: { makeItSo: function(){ return 2}}
+    })
+    assert.isDefined(User.doSmth)
+    assert.eql(User.doSmth(), 1)
+    assert.isUndefined(User.makeItSo)
+    
+    assert.isDefined(User.build().makeItSo)
+    assert.eql(User.build().makeItSo(), 2)
   }
 }
\ No newline at end of file