graphql-compose-mongoose/lib/resolvers/updateOne.js

Plugin for `graphql-compose` which derive a graphql types from a mongoose model.

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _asyncToGenerator2 = require('babel-runtime/helpers/asyncToGenerator');

var _asyncToGenerator3 = _interopRequireDefault(_asyncToGenerator2);

var _extends2 = require('babel-runtime/helpers/extends');

var _extends3 = _interopRequireDefault(_extends2);

exports.default = updateOne;

var _helpers = require('./helpers');

var _findOne = require('./findOne');

var _findOne2 = _interopRequireDefault(_findOne);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function updateOne(model, tc, opts) {
  if (!model || !model.modelName || !model.schema) {
    throw new Error('First arg for Resolver updateOne() should be instance of Mongoose Model.');
  }
  if (!tc || tc.constructor.name !== 'TypeComposer') {
    throw new Error('Second arg for Resolver updateOne() should be instance of TypeComposer.');
  }

  const findOneResolver = (0, _findOne2.default)(model, tc, opts);

  const outputTypeName = `UpdateOne${tc.getTypeName()}Payload`;
  const outputType = tc.constructor.schemaComposer.getOrCreateTC(outputTypeName, t => {
    t.addFields({
      recordId: {
        type: 'MongoID',
        description: 'Updated document ID'
      },
      record: {
        type: tc.getType(),
        description: 'Updated document'
      }
    });
  });

  const resolver = new tc.constructor.schemaComposer.Resolver({
    name: 'updateOne',
    kind: 'mutation',
    description: 'Update one document: ' + '1) Retrieve one document via findOne. ' + '2) Apply updates to mongoose document. ' + '3) Mongoose applies defaults, setters, hooks and validation. ' + '4) And save it.',
    type: outputType,
    args: (0, _extends3.default)({}, (0, _helpers.recordHelperArgs)(tc, (0, _extends3.default)({
      recordTypeName: `UpdateOne${tc.getTypeName()}Input`,
      removeFields: ['id', '_id'],
      isRequired: true
    }, opts && opts.record)), (0, _helpers.filterHelperArgs)(tc, model, (0, _extends3.default)({
      filterTypeName: `FilterUpdateOne${tc.getTypeName()}Input`,
      model
    }, opts && opts.filter)), (0, _helpers.sortHelperArgs)(tc, model, (0, _extends3.default)({
      sortTypeName: `SortUpdateOne${tc.getTypeName()}Input`
    }, opts && opts.sort)), (0, _helpers.skipHelperArgs)()),
    resolve: (() => {
      var _ref = (0, _asyncToGenerator3.default)(function* (resolveParams) {
        const recordData = resolveParams.args && resolveParams.args.record || null;
        const filterData = resolveParams.args && resolveParams.args.filter || {};

        if (!(typeof filterData === 'object') || Object.keys(filterData).length === 0) {
          return Promise.reject(new Error(`${tc.getTypeName()}.updateOne resolver requires at least one value in args.filter`));
        }

        // We should get all data for document, cause Mongoose model may have hooks/middlewares
        // which required some fields which not in graphql projection
        // So empty projection returns all fields.
        resolveParams.projection = {};

        let doc = yield findOneResolver.resolve(resolveParams);

        if (resolveParams.beforeRecordMutate) {
          doc = yield resolveParams.beforeRecordMutate(doc, resolveParams);
        }

        if (doc && recordData) {
          doc.set(recordData);
          yield doc.save();
        }

        if (doc) {
          return {
            record: doc,
            recordId: tc.getRecordIdFn()(doc)
          };
        }

        return null;
      });

      return function resolve(_x) {
        return _ref.apply(this, arguments);
      };
    })()
  });

  return resolver;
}
/* eslint-disable no-param-reassign */