Handle Mongoose refs on schema (#57)

Author: Lucas Bento

packages/generator/src/graphqlrc.json

@@ -7,9 +7,10 @@
     "mutation": "mutation",
     "mutation_test": "mutation/__tests__",
     "type": "type",
-    "type_test": "type/__tests__"
+    "type_test": "type/__tests__",
+    "interface": "interface"
   },
   "files": {
     "schema": "schema"
   }
-}
+}

packages/generator/src/loader/index.js

@@ -31,7 +31,7 @@ class LoaderGenerator extends Generator {
 
   generateLoader() {
     const schema = this.options.model ?
-      getMongooseModelSchema(this.options.model, true)
+      getMongooseModelSchema({ model: this.options.model, withTimestamps: true })
       : null;
 
     const name = uppercaseFirstLetter(this.options.name);

packages/generator/src/mutation/index.js

@@ -72,7 +72,7 @@ class MutationGenerator extends Generator {
   generateMutation() {
     let schema = null;
     if (this.options.model) {
-      const modelSchema = getMongooseModelSchema(this.options.model);
+      const modelSchema = getMongooseModelSchema({ model: this.options.model });
       schema = this._parseSchema(modelSchema);
     }
 
@@ -105,6 +105,8 @@ class MutationGenerator extends Generator {
       directories,
     };
 
+    // TODO: generate type and loader that do not exist yet
+
     Object.keys(mutations).forEach((mutationType) => {
       const { template, fileName } = mutations[mutationType];
 

packages/generator/src/type/__tests__/__snapshots__/TypeGenerator.spec.js.snap

@@ -6,6 +6,8 @@ Object {
 } from \'graphql\';
 import { globalIdField } from \'graphql-relay\';
 
+import { NodeInterface } from \'../interface/NodeInterface\';
+
 export default new GraphQLObjectType({
   name: \'Example\',
   description: \'Represents Example\',
@@ -16,6 +18,7 @@ export default new GraphQLObjectType({
       description: \'My example field\',
       resolve: obj => obj.example,
     },
+    interfaces: () => [NodeInterface],
   }),
 });",
   "typeTest": "import { graphql } from \'graphql\';
@@ -62,10 +65,13 @@ Object {
 import {
   GraphQLObjectType,
   GraphQLString,
-  GraphQLID,
 } from \'graphql\';
 import { globalIdField } from \'graphql-relay\';
 
+import { NodeInterface } from \'../interface/NodeInterface\';
+import UserType from \'./UserType\';
+import UserLoader from \'../loader/UserLoader\';
+
 export default new GraphQLObjectType({
   name: \'Post\',
   description: \'Represents Post\',
@@ -77,9 +83,9 @@ export default new GraphQLObjectType({
       resolve: obj => obj.title,
     },
     user: {
-      type: GraphQLID,
+      type: UserType,
       description: \'User that created this post\',
-      resolve: obj => obj.user,
+      resolve: async (obj, args, { user }) => await UserLoader.load(user, obj.user),
     },
     slug: {
       type: GraphQLString,
@@ -96,6 +102,7 @@ export default new GraphQLObjectType({
       description: \'\',
       resolve: obj => obj.updatedAt.toISOString(),
     },
+    interfaces: () => [NodeInterface],
   }),
 });
 ",
@@ -147,6 +154,8 @@ import {
 } from \'graphql\';
 import { globalIdField } from \'graphql-relay\';
 
+import { NodeInterface } from \'../interface/NodeInterface\';
+
 export default new GraphQLObjectType({
   name: \'User\',
   description: \'Represents User\',
@@ -172,6 +181,7 @@ export default new GraphQLObjectType({
       description: \'\',
       resolve: obj => obj.active,
     },
+    interfaces: () => [NodeInterface],
   }),
 });
 ",

packages/generator/src/type/index.js

@@ -3,6 +3,7 @@ import {
   getMongooseModelSchema,
   getConfigDir,
   uppercaseFirstLetter,
+  getRelativeConfigDir,
 } from '../utils';
 
 class TypeGenerator extends Generator {
@@ -23,13 +24,15 @@ class TypeGenerator extends Generator {
   }
 
   generateType() {
-    let schema = this.options.model ?
-      getMongooseModelSchema(this.options.model, true)
+    const schema = this.options.model ?
+      getMongooseModelSchema({
+        model: this.options.model,
+        withTimestamps: true,
+        ref: true,
+      })
       : null;
 
-    if (schema) {
-      schema = this._parseSchemaResolvers(schema);
-    }
+    const directories = this._getConfigDirectories();
 
     const name = uppercaseFirstLetter(this.options.name);
     const typeFileName = `${name}Type`;
@@ -42,50 +45,34 @@ class TypeGenerator extends Generator {
     const templateVars = {
       name,
       schema,
+      directories,
     };
 
     this._generateTypeTest({
       name,
       schema,
     });
 
+    // TODO: generate types and loaders that do not exist yet
+
     this.fs.copyTpl(templatePath, destinationPath, templateVars);
   }
 
-  /**
-   * Parse schema resolvers checking if the fields need different resolvers.
-   * @param schema {Array} The parsed Mongoose schema
-   * @returns {Array} The parsed schema with resolvers
-   */
-  _parseSchemaResolvers(schema) {
-    const fields = schema.fields.map((field) => {
-      if (field.originalType === 'Date') {
-        return {
-          ...field,
-          resolve: `obj.${field.name}.toISOString()`,
-        };
-      }
-
-      return {
-        ...field,
-        resolve: `obj.${field.name}`,
-      };
-    });
-
-    return {
-      ...schema,
-      fields,
-    };
+  _getConfigDirectories() {
+    return getRelativeConfigDir('type', ['model', 'type', 'loader', 'connection', 'interface']);
   }
 
   _generateTypeTest({ name, schema }) {
     const templatePath = this.templatePath('test/Type.js.template');
 
     const destinationPath = this.destinationPath(`${this.destinationDir}/__tests__/${name}Type.spec.js`);
 
+    const directories = this._getConfigDirectories();
+
     const templateVars = {
       name,
       schema,
+      directories,
     };
 
     this.fs.copyTpl(templatePath, destinationPath, templateVars);

packages/generator/src/type/templates/Type.js.template

@@ -4,6 +4,8 @@ import {
 } from 'graphql';
 import { globalIdField } from 'graphql-relay';
 
+import { NodeInterface } from '<%= directories.interface %>/NodeInterface';
+
 export default new GraphQLObjectType({
   name: '<%= name %>',
   description: 'Represents <%= name %>',
@@ -14,5 +16,6 @@ export default new GraphQLObjectType({
       description: 'My example field',
       resolve: obj => obj.example,
     },
+    interfaces: () => [NodeInterface],
   }),
 });

packages/generator/src/type/templates/TypeWithSchema.js.template

@@ -8,6 +8,14 @@ import {
 } from 'graphql';
 import { globalIdField } from 'graphql-relay';
 
+import { NodeInterface } from '<%= directories.interface %>/NodeInterface';
+<%_ for (i in schema.typeDependencies) { -%>
+import <%= schema.typeDependencies[i] %> from '<%= directories.type %>/<%=schema.typeDependencies[i]%>';
+<%_ } -%>
+<%_ for (i in schema.loaderDependencies) { -%>
+import <%= schema.loaderDependencies[i] %> from '<%= directories.loader %>/<%=schema.loaderDependencies[i]%>';
+<%_ } -%>
+
 export default new GraphQLObjectType({
   name: '<%= name %>',
   description: 'Represents <%= name %>',
@@ -17,8 +25,9 @@ export default new GraphQLObjectType({
     <%- field.name %>: {
       type: <%= field.type %>,
       description: '<%= field.description %>',
-      resolve: obj => <%= field.resolve %>,
+      resolve: <%if(field.resolveArgs){%><%=field.resolveArgs%><%}else{ %>obj<%} %> => <%= field.resolve %>,
     },
     <%_ } -%>
+    interfaces: () => [NodeInterface],
   }),
 });