Add isRequired option to input helper.

Author: nodkz

src/definition.js

@@ -43,8 +43,10 @@ export type MongooseFieldT = {
 
 export type MongooseFieldMapT = { [fieldName: string]: MongooseFieldT };
 
-export type ResolverNames = 'byId' | 'byIds' | 'findOne' | 'findMany' |
-                            'updateOne' | 'updateMany' | 'removeOne' | 'removeMany';
+export type ResolverNames = 'findById' | 'findByIds' | 'findOne' | 'findMany' |
+                            'updateById' | 'updateOne' | 'updateMany' |
+                            'removeById' | 'removeOne' | 'removeMany' |
+                            'createOne' | 'count';
 
 export type MongooseQuery = {
   exec(): Promise,

src/resolvers/helpers/input.js

@@ -12,6 +12,7 @@ export type inputHelperArgsGenOpts = {
   inputTypeName: string,
   removeFields?: string | string[],
   requiredFields?: string | string[],
+  isRequired?: boolean,
 };
 
 export const inputHelperArgsGen = (
@@ -36,7 +37,9 @@ export const inputHelperArgsGen = (
   return {
     input: {
       name: 'input',
-      type: inputComposer.getType(),
+      type: opts.isRequired
+        ? new GraphQLNonNull(inputComposer.getType())
+        : inputComposer.getType(),
     },
   };
 };

src/resolvers/updateById.js

@@ -43,6 +43,7 @@ export default function updateById(
       ...inputHelperArgsGen(gqType, {
         inputTypeName: `UpdateById${gqType.name}Input`,
         requiredFields: ['_id'],
+        isRequired: true,
       }),
     },
     resolve: (resolveParams: ExtendedResolveParams) => {