add support for attributeFields.only and document .exclude, closes #1, #26

Author: mickhansen

README.md

@@ -140,7 +140,11 @@ var Model = sequelize.define('User', {
 
 import {attributeFields} from 'graphql-sequelize';
 
-attributeFields(Model);
+attributeFields(Model, {
+  // ... options
+  exclude: [], // array of model attributes to ignore - default: []
+  only: [], // only generate definitions for these model attributes - default: null
+});
 
 /*
 {

src/attributeFields.js

@@ -6,6 +6,7 @@ module.exports = function (Model, options) {
 
   var result = Object.keys(Model.rawAttributes).reduce(function (memo, key) {
     if (options.exclude && ~options.exclude.indexOf(key)) return memo;
+    if (options.only && !~options.only.indexOf(key)) return memo;
 
     var attribute = Model.rawAttributes[key]
       , type = attribute.type;

test/attributeFields.test.js

@@ -91,6 +91,14 @@ describe('attributeFields', function () {
     expect(Object.keys(fields)).to.deep.equal(['firstName', 'lastName']);
   });
 
+  it('should be possible to specify specific fields', function () {
+    var fields = attributeFields(Model, {
+      only: ['id', 'email', 'list']
+    });
+
+    expect(Object.keys(fields)).to.deep.equal(['id', 'email', 'list']);
+  });
+
   it('should automatically name enum types', function () {
     var fields = attributeFields(Model);