finale-rest/lib/Controllers/create.js

Create REST resources and controllers with Sequelize and Express or Restify

'use strict';

var _ = require('lodash'),
    util = require('util'),
    Base = require('./base');

var Create = function(args) {
  Create.super_.call(this, args);
};

util.inherits(Create, Base);

Create.prototype.action = 'create';
Create.prototype.method = 'post';
Create.prototype.plurality = 'plural';

Create.prototype.write = function(req, res, context) {
  context.attributes = _.defaults(context.attributes, req.body);
  var self = this;

  //Supporting an add_to_children context variable that can allow
  //some values to be injected/added to all first level child object creations
  var add_to_children = context.add_to_children || {};

  // Check associated data
  if (this.include && this.include.length) {
    _.values(self.resource.associationsInfo).forEach(function(association) {
      if (context.attributes.hasOwnProperty(association.as)) {
        var attr = context.attributes[association.as];

        if(attr) {
          //add the add_to_children attributes to the attr 
          if(_.isArray(attr)){
            //if array, add the add_to_children to each object
            for(var x=0;x<attr.length;x++) {
              attr[x] = Object.assign(attr[x],add_to_children);
            }
            context.attributes[association.as] = attr;
          } else {
            attr = Object.assign(attr,add_to_children);
            context.attributes[association.as] = attr;
          }
        }

        if (_.isObject(attr) && attr.hasOwnProperty(association.primaryKey)) {
          context.attributes[association.identifier] = attr[association.primaryKey];
          delete context.attributes[association.as];
        }
      }
    });
  }

var creation_include = _.assign([],this.include);

if(context.shallow) {
    creation_include = [];
  }

  return this.model
    .create(context.attributes, {
      include: creation_include
    })
    .then(function(instance) {
      if (self.resource) {
        var endpoint = self.resource.endpoints.singular;
        var location = endpoint.replace(/:(\w+)/g, function(match, $1) {
          return instance[$1];
        });

        res.header('Location', location);
      }

      if (self.resource.reloadInstances === true) {
        var reloadOptions = {};
        if (Array.isArray(self.include) && self.include.length)
          reloadOptions.include = self.include;
        if (!!self.resource.excludeAttributes)
          reloadOptions.attributes = { exclude: self.resource.excludeAttributes };

        if(context.shallow)
          delete reloadOptions.include;  
        return instance.reload(reloadOptions);
      }

      return instance;
    }).then(function(instance) {
      if (!!self.resource.excludeAttributes) {
        self.resource.excludeAttributes.forEach(function(attr) {
          delete instance.dataValues[attr];
        });
      }

      res.status(201);
      context.instance = instance;
      return context.continue;
    });
};

module.exports = Create;