Add traits plugin (resolves #374)

Gulpfile.js

@@ -132,7 +132,8 @@ var modularized = {
     './src/lib/plugins/grabAll.js',
     './src/lib/plugins/methods.js',
     './src/lib/plugins/ready.js',
-    './src/lib/plugins/refresh.js'
+    './src/lib/plugins/refresh.js',
+    './src/lib/plugins/traits.js'
   ],
   modules: {
     S: [ // searching

build/rekord.js

@@ -18634,6 +18634,45 @@ Rekord.Timestamp = Timestamp;
 Rekord.formatDate = noop;
 Rekord.convertDate = convertDate;
 
+addPlugin(function(model, db, options)
+{
+  var traits = options.traits || Defaults.traits;
+
+  if ( !isEmpty( traits ) )
+  {
+    if ( isFunction( traits ) )
+    {
+      traits = traits( model, db, options );
+    }
+
+    if ( isArray( traits ) )
+    {
+      for (var i = 0; i < traits.length; i++)
+      {
+        var trait = traits[ i ];
+
+        if ( isFunction( trait ) )
+        {
+          trait = trait( model, db, options );
+        }
+
+        if ( isObject( trait ) )
+        {
+          Class.methods( model, trait );
+        }
+        else
+        {
+          throw 'traits are expected to be an object with methods or a function which returns an object of methods';
+        }
+      }
+    }
+    else
+    {
+      throw 'traits are expected to be an array or a function which returns an array';
+    }
+  }
+});
+
 addPlugin(function(model, db, options)
 {
 

src/lib/plugins/traits.js

@@ -0,0 +1,38 @@
+addPlugin(function(model, db, options)
+{
+  var traits = options.traits || Defaults.traits;
+
+  if ( !isEmpty( traits ) )
+  {
+    if ( isFunction( traits ) )
+    {
+      traits = traits( model, db, options );
+    }
+
+    if ( isArray( traits ) )
+    {
+      for (var i = 0; i < traits.length; i++)
+      {
+        var trait = traits[ i ];
+
+        if ( isFunction( trait ) )
+        {
+          trait = trait( model, db, options );
+        }
+
+        if ( isObject( trait ) )
+        {
+          Class.methods( model, trait );
+        }
+        else
+        {
+          throw 'traits are expected to be an object with methods or a function which returns an object of methods';
+        }
+      }
+    }
+    else
+    {
+      throw 'traits are expected to be an array or a function which returns an array';
+    }
+  }
+});

test/cases/rekord-options.js

@@ -2037,3 +2037,83 @@ test( 'saveAlways field', function(assert)
 
   deepEqual( Task.Database.rest.lastRecord, {name: 't1', done: true} );
 });
+
+test( 'traits simple', function(assert)
+{
+  var prefix = 'traits_simple_';
+
+  var TraitFoo = {
+    foo: function() { return 1; }
+  };
+
+  var TraitBar = {
+    bar: function() { return 2; }
+  };
+
+  var Task = Rekord({
+    name: prefix + 'task',
+    fields: ['name', 'done'],
+    traits: [TraitFoo, TraitBar]
+  });
+
+  var t0 = Task.create({name: 't0', done: false});
+
+  strictEqual( t0.foo(), 1 );
+  strictEqual( t0.bar(), 2 );
+});
+
+
+test( 'traits function', function(assert)
+{
+  var prefix = 'traits_function_';
+
+  var TraitFoo = {
+    foo: function() { return 1; }
+  };
+
+  var TraitBar = function(model, db, options) {
+    return {
+      bar: function() { return options.name; }
+    }
+  };
+
+  var Task = Rekord({
+    name: prefix + 'task',
+    fields: ['name', 'done'],
+    traits: [TraitFoo, TraitBar]
+  });
+
+  var t0 = Task.create({name: 't0', done: false});
+
+  strictEqual( t0.foo(), 1 );
+  strictEqual( t0.bar(), 'traits_function_task' );
+});
+
+
+test( 'traits generator', function(assert)
+{
+  var prefix = 'traits_generator_';
+
+  var TraitFoo = {
+    foo: function() { return 1; }
+  };
+
+  var TraitBar = {
+    bar: function() { return 2; }
+  };
+
+  var TraitGenerator = function() {
+    return [TraitFoo, TraitBar];
+  };
+
+  var Task = Rekord({
+    name: prefix + 'task',
+    fields: ['name', 'done'],
+    traits: TraitGenerator
+  });
+
+  var t0 = Task.create({name: 't0', done: false});
+
+  strictEqual( t0.foo(), 1 );
+  strictEqual( t0.bar(), 2 );
+});