feat(resolvers): Add callback beforeRecordMutate to resolveParams for createOne, updateById, updat

For `wrapResolve` you may provide `resolveParams.beforeRecordMutate` callback. Which will be called in

`createOne`, `updateById`, `updateOne` resolvers. To this callback will be provided Document, which

you may change before save. Or you may return Promise.reject('Msg') in this callback, if you want

prohibit operation.

Author: nodkz

src/definition.js

@@ -102,6 +102,8 @@ export type ResolverMWResolveFn = _ResolverMWResolveFn;
 export type ExtendedResolveParams = ResolveParams & {
   query: MongooseQuery,
   projection: ProjectionType,
+  beforeQuery(query: mixed): Promise<*>,
+  beforeRecordMutate(record: mixed): Promise<*>,
 };
 
 

src/resolvers/__tests__/createOne-test.js

@@ -95,6 +95,18 @@ describe('createOne() ->', () => {
       });
       expect(result).property('record').instanceof(UserModel);
     });
+
+    it('should call `beforeRecordMutate` method with created `record` as arg', async () => {
+      const result = await createOne(UserModel, UserTypeComposer).resolve({
+        args: { record: { name: 'NewUser' } },
+        beforeRecordMutate: (record) => {
+          record.name = 'OverridedName';
+          return record;
+        }
+      });
+      expect(result).property('record').instanceof(UserModel);
+      expect(result).have.deep.property('record.name', 'OverridedName');
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/__tests__/updateById-test.js

@@ -156,6 +156,19 @@ describe('updateById() ->', () => {
       });
       expect(result.record).instanceof(UserModel);
     });
+
+    it('should call `beforeRecordMutate` method with founded `record` as arg', async () => {
+      let beforeMutationId;
+      const result = await updateById(UserModel, UserTypeComposer).resolve({
+        args: { record: { _id: user1.id } },
+        beforeRecordMutate: (record) => {
+          beforeMutationId = record.id;
+          return record;
+        }
+      });
+      expect(result.record).instanceof(UserModel);
+      expect(beforeMutationId).to.equal(user1.id);
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/__tests__/updateOne-test.js

@@ -180,6 +180,19 @@ describe('updateOne() ->', () => {
       });
       expect(result.record).instanceof(UserModel);
     });
+
+    it('should call `beforeRecordMutate` method with founded `record` as arg', async () => {
+      let beforeMutationId;
+      const result = await updateOne(UserModel, UserTypeComposer).resolve({
+        args: { filter: { _id: user1.id } },
+        beforeRecordMutate: (record) => {
+          beforeMutationId = record.id;
+          return record;
+        }
+      });
+      expect(result.record).instanceof(UserModel);
+      expect(beforeMutationId).to.equal(user1.id);
+    });
   });
 
   describe('Resolver.getOutputType()', () => {

src/resolvers/createOne.js

@@ -71,7 +71,15 @@ export default function createOne(
         );
       }
 
-      return model.create(recordData)
+      // $FlowFixMe
+      return Promise.resolve(new model(recordData))
+        .then(doc => {
+          if (resolveParams.beforeRecordMutate) {
+            return resolveParams.beforeRecordMutate(doc);
+          }
+          return doc;
+        })
+        .then(doc => doc.save())
         .then(record => {
           if (record) {
             return {

src/resolvers/updateById.js

@@ -92,6 +92,12 @@ export default function updateById(
         (resolveParams.projection && resolveParams.projection.record) || {};
 
       return findByIdResolver.resolve(resolveParams)
+        .then(doc => {
+          if (resolveParams.beforeRecordMutate) {
+            return resolveParams.beforeRecordMutate(doc);
+          }
+          return doc;
+        })
         // save changes to DB
         .then(doc => {
           if (!doc) {

src/resolvers/updateOne.js

@@ -98,6 +98,12 @@ export default function updateOne(
         (resolveParams.projection && resolveParams.projection.record) || {};
 
       return findOneResolver.resolve(resolveParams)
+        .then(doc => {
+          if (resolveParams.beforeRecordMutate) {
+            return resolveParams.beforeRecordMutate(doc);
+          }
+          return doc;
+        })
         // save changes to DB
         .then(doc => {
           if (recordData) {