refactor(type): graffiti level models

src/field.js

@@ -1,48 +1,41 @@
-import {reduce, isDate} from 'lodash';
+import {isDate} from 'lodash';
 
 import {
   GraphQLString,
   GraphQLFloat,
   GraphQLBoolean,
   GraphQLList
-} from 'graphql/type';
+}
+from 'graphql/type';
 
 import {getProjection} from './projection';
 
 /**
- * @method get
- * @param {Object} path
+ * @method getField
+ * @param {Object} field
  * @param {Object} types
  * @param {Object} models
  * @return {Object} field
  */
-function get (path, types, models) {
-
-  // Convert array of models to map
-  var modelMap = reduce(models, (map, model) => {
-    map[model.modelName] = model;
-
-    return map;
-  }, {});
-
-  if (path.path === '__v') {
+function getField(field, types, models) {
+  if (field.name === '__v') {
     return;
   }
 
   // String
-  if (['String', 'ObjectID'].indexOf(path.instance) > -1) {
+  if (['String', 'ObjectID'].indexOf(field.instance) > -1) {
     return {
       type: GraphQLString
     };
 
-  // Number
-  } else if (path.instance === 'Number') {
+    // Number
+  } else if (field.instance === 'Number') {
     return {
       type: GraphQLFloat
     };
 
-  // Date
-  } else if (path.instance === 'Date') {
+    // Date
+  } else if (field.instance === 'Date') {
     return {
       type: GraphQLString,
       resolve: (modelInstance, params, source, fieldASTs) => {
@@ -55,47 +48,47 @@ function get (path, types, models) {
     };
 
     // Boolean
-  } else if (path.instance === 'Boolean') {
+  } else if (field.instance === 'Boolean') {
     return {
       type: GraphQLBoolean
     };
 
     // Array
-  } else if (path.instance === 'Array') {
+  } else if (field.instance === 'Array') {
     var type;
 
     // Array of refs
-    if (path.caster.instance === 'ObjectID') {
-      type = path.caster.options.ref;
+    if (field.caster.instance === 'ObjectID') {
+      type = field.caster.ref;
 
       return {
         type: new GraphQLList(types[type]),
         resolve: (modelInstance, params, source, fieldASTs) => {
           var projections = getProjection(fieldASTs);
-          return modelMap[type].find({
+          return models[type].model.find({
             _id: {
               // to make it easily testable
-              $in: modelInstance[path.path].map((id) => id.toString())
+              $in: modelInstance[field.name].map((id) => id.toString())
             }
           }, projections);
         }
       };
 
-    // Array of primitives
+      // Array of primitives
     } else {
-      if (path.caster.instance === 'Number') {
+      if (field.caster.instance === 'Number') {
         return {
           type: new GraphQLList(GraphQLFloat)
         };
-      } else if (path.caster.instance === 'String') {
+      } else if (field.caster.instance === 'String') {
         return {
           type: new GraphQLList(GraphQLString)
         };
-      } else if (path.caster.instance === 'Boolean') {
+      } else if (field.caster.instance === 'Boolean') {
         return {
           type: new GraphQLList(GraphQLBoolean)
         };
-      } else if (path.caster.instance === 'Date') {
+      } else if (field.caster.instance === 'Date') {
         return {
           type: new GraphQLList(GraphQLString),
           resolve: (modelInstance, params, source, fieldASTs) => {
@@ -114,5 +107,5 @@ function get (path, types, models) {
 }
 
 export default {
-  get: get
+  getField
 };

src/field.spec.js

@@ -7,13 +7,13 @@ import {
 } from 'graphql/type';
 
 import User from '../fixture/user';
-import {get} from './field';
+import {getField} from './field';
 
 var projection =  require('./projection');
 
 describe('field', () => {
   it('should resolve String properly', () => {
-    var field = get({
+    var field = getField({
       instance: 'String'
     });
 
@@ -23,7 +23,7 @@ describe('field', () => {
   });
 
   it('should resolve Number properly', () => {
-    var field = get({
+    var field = getField({
       instance: 'Number'
     });
 
@@ -33,7 +33,7 @@ describe('field', () => {
   });
 
   it('should resolve Boolean properly', () => {
-    var field = get({
+    var field = getField({
       instance: 'Boolean'
     });
 
@@ -43,7 +43,7 @@ describe('field', () => {
   });
 
   it('should resolve Date properly', () => {
-    var field = get({
+    var field = getField({
       instance: 'Date'
     });
 
@@ -73,18 +73,21 @@ describe('field', () => {
     });
     var findStub = this.sandbox.stub(User, 'find').returnsWithResolve(users);
 
-    var field = get({
+    var field = getField({
       instance: 'Array',
       path: 'value',
+      name: 'value',
       caster: {
         instance: 'ObjectID',
-        options: {
-          ref: 'User'
-        }
+        ref: 'User'
       }
     }, {
       User: 'foo'
-    }, [User]);
+    }, {
+      User: {
+        model: User
+      }
+    });
 
     var result = yield field.resolve({
       value: [users[0]._id, users[1]._id]
@@ -110,7 +113,7 @@ describe('field', () => {
   });
 
   it('should resolve Array of String properly', () => {
-    var field = get({
+    var field = getField({
       instance: 'Array',
       caster: {
         instance: 'String'
@@ -123,7 +126,7 @@ describe('field', () => {
   });
 
   it('should resolve Array of Number properly', () => {
-    var field = get({
+    var field = getField({
       instance: 'Array',
       caster: {
         instance: 'Number'
@@ -136,7 +139,7 @@ describe('field', () => {
   });
 
   it('should resolve Array of Boolean properly', () => {
-    var field = get({
+    var field = getField({
       instance: 'Array',
       caster: {
         instance: 'Boolean'
@@ -149,7 +152,7 @@ describe('field', () => {
   });
 
   it('should resolve Array of Dates properly', () => {
-    var field = get({
+    var field = getField({
       instance: 'Array',
       caster: {
         instance: 'Date'

src/index.js

@@ -1,9 +1,17 @@
 import {graphql} from 'graphql';
+
+import {getModels} from './model';
 import {getSchema} from './schema';
 import {getTypes} from './type';
 
-export default {
-  graphql,
-  getSchema,
-  getTypes
+module.exports.getSchema = function (mongooseModels) {
+  var models = getModels(mongooseModels);
+  return getSchema(models);
+};
+
+module.exports.getTypes = function (mongooseModels) {
+  var models = getModels(mongooseModels);
+  return getTypes(models);
 };
+
+module.exports.graphql = graphql;

src/integration.spec.js

@@ -3,10 +3,11 @@ import ObjectID from 'bson-objectid';
 import {graphql} from 'graphql';
 
 import {getSchema} from './schema';
+import {getModels} from './model';
 import User from '../fixture/user';
 
 describe('schema integration test', () => {
-  let schema = getSchema([User]);
+  let schema = getSchema(getModels([User]));
 
   describe('singular resource', () => {
     it('should get data by id with selected fields', function* () {

src/model.js

@@ -0,0 +1,54 @@
+import {map} from 'lodash';
+
+/**
+ * Turn mongoose model to graffiti model
+ * @method getModel
+ * @param {Object} mongooseModel
+ * @return {Object} graffiti model
+ */
+export function getModel (mongooseModel) {
+  var fields = map(mongooseModel.schema.paths, (schemaPath, name) => {
+
+    var options = schemaPath.options || {};
+
+    var field = {
+      name: name,
+      path: schemaPath.path,
+      instance: schemaPath.instance,
+      indexed: options.index ? true : false
+    };
+
+    if (schemaPath.caster) {
+      field.caster = {
+        path: schemaPath.caster.path,
+        instance: schemaPath.caster.instance
+      };
+
+      if (schemaPath.caster.options.ref) {
+        field.caster.ref = schemaPath.caster.options.ref;
+      }
+    }
+
+    return field;
+  });
+
+  return {
+    name: mongooseModel.modelName,
+    fields: fields,
+    model: mongooseModel
+  };
+}
+
+/**
+ * @method getModels
+ * @param {Array} mongooseModels
+ * @return {Object} - graffiti models
+ */
+export function getModels (mongooseModels) {
+  return mongooseModels
+    .map(getModel)
+    .reduce((models, model) => {
+      models[model.name] = model;
+      return models;
+    }, {});
+}