tabel/lib/relations/MorphMany.js

A simple orm for PostgreSQL which works with simple javascript objects and arrays

'use strict';

var _createClass = function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }

function _possibleConstructorReturn(self, call) { if (!self) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return call && (typeof call === "object" || typeof call === "function") ? call : self; }

function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function, not " + typeof superClass); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, enumerable: false, writable: true, configurable: true } }); if (superClass) Object.setPrototypeOf ? Object.setPrototypeOf(subClass, superClass) : subClass.__proto__ = superClass; }

var _require = require('lodash'),
    assign = _require.assign,
    isArray = _require.isArray;

var Relation = require('./Relation');
var MorphTo = require('./MorphTo');

var MorphMany = function (_Relation) {
  _inherits(MorphMany, _Relation);

  function MorphMany(ownerTable, toTable, inverse) {
    _classCallCheck(this, MorphMany);

    if (!(inverse instanceof MorphTo)) {
      throw new Error('inverse should be a MorphTo relation');
    }

    var _this = _possibleConstructorReturn(this, (MorphMany.__proto__ || Object.getPrototypeOf(MorphMany)).call(this, ownerTable));

    assign(_this, { fromTable: ownerTable.fork(), toTable: toTable, inverse: inverse });
    return _this;
  }

  _createClass(MorphMany, [{
    key: 'initRelation',
    value: function initRelation(fromModels) {
      var _this2 = this;

      return fromModels.map(function (m) {
        return assign(m, _defineProperty({}, _this2.relationName, []));
      });
    }
  }, {
    key: 'getRelated',
    value: function getRelated() {
      for (var _len = arguments.length, args = Array(_len), _key = 0; _key < _len; _key++) {
        args[_key] = arguments[_key];
      }

      if (args.length === 0) {
        if (this.activeModel !== null) {
          return this.getRelated([this.activeModel]);
        } else {
          return Promise.resolve([]);
        }
      }

      var fromModels = args[0];
      var fromTable = this.fromTable,
          inverse = this.inverse;

      var toTable = this.constraints.apply(this.toTable.fork());

      var foreignKey = inverse.foreignKey,
          typeField = inverse.typeField;

      var typeValue = fromTable.tableName();

      return toTable.where(_defineProperty({}, typeField, typeValue)).whereIn(foreignKey, fromModels.map(function (m) {
        return m[fromTable.key()];
      })).all();
    }
  }, {
    key: 'matchModels',
    value: function matchModels() {
      var fromModels = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
      var relatedModels = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
      var relationName = this.relationName,
          fromTable = this.fromTable,
          inverse = this.inverse;
      var foreignKey = inverse.foreignKey;


      var keyDict = relatedModels.reduce(function (dict, m) {
        var key = m[foreignKey];

        if (!isArray(dict[key])) {
          return assign(dict, _defineProperty({}, key, [m]));
        } else {
          return assign(dict, _defineProperty({}, key, dict[key].concat(m)));
        }
      }, {});

      return fromModels.map(function (m) {
        return assign(m, _defineProperty({}, relationName, isArray(keyDict[m[fromTable.key()]]) ? keyDict[m[fromTable.key()]] : []));
      });
    }
  }, {
    key: 'insert',
    value: function insert() {
      var _this3 = this;

      for (var _len2 = arguments.length, args = Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
        args[_key2] = arguments[_key2];
      }

      if (args.length === 0) {
        throw new Error('bad method call');
      }

      if (args.length === 1) {
        return this.insert.apply(this, [this.activeModel].concat(args));
      }

      var fromModel = args[0],
          values = args[1];
      var fromTable = this.fromTable,
          toTable = this.toTable,
          inverse = this.inverse;
      var foreignKey = inverse.foreignKey,
          typeField = inverse.typeField;


      return toTable.insert(function () {
        if (isArray(values)) {
          return values.map(function (v) {
            var _assign5;

            return assign(v, (_assign5 = {}, _defineProperty(_assign5, foreignKey, fromModel[fromTable.key()]), _defineProperty(_assign5, typeField, fromTable.tableName()), _assign5));
          });
        } else {
          var _assign6;

          return assign(values, (_assign6 = {}, _defineProperty(_assign6, foreignKey, fromModel[_this3.key]), _defineProperty(_assign6, typeField, fromTable.tableName()), _assign6));
        }
      }());
    }
  }, {
    key: 'update',
    value: function update() {
      var _assign7;

      for (var _len3 = arguments.length, args = Array(_len3), _key3 = 0; _key3 < _len3; _key3++) {
        args[_key3] = arguments[_key3];
      }

      if (args.length === 0) {
        throw new Error('bad method call');
      }

      if (args.length === 1) {
        return this.update.apply(this, [this.activeModel].concat(args));
      }

      var fromModel = args[0],
          values = args[1];
      var fromTable = this.fromTable,
          toTable = this.toTable,
          inverse = this.inverse;
      var foreignKey = inverse.foreignKey,
          typeField = inverse.typeField;


      return this.constraints.apply(toTable.fork()).where(typeField, fromTable.tableName()).where(foreignKey, fromModel[fromTable.key()]).update(assign(values, (_assign7 = {}, _defineProperty(_assign7, foreignKey, fromModel[fromTable.key()]), _defineProperty(_assign7, typeField, fromTable.tableName()), _assign7)));
    }
  }, {
    key: 'del',
    value: function del() {
      for (var _len4 = arguments.length, args = Array(_len4), _key4 = 0; _key4 < _len4; _key4++) {
        args[_key4] = arguments[_key4];
      }

      if (args.length === 0) {
        return this.del(this.activeModel);
      }

      var fromModel = args[0];
      var fromTable = this.fromTable,
          toTable = this.toTable,
          inverse = this.inverse;
      var foreignKey = inverse.foreignKey,
          typeField = inverse.typeField;


      return this.constraints.apply(toTable.fork()).where(typeField, fromTable.tableName()).where(foreignKey, fromModel[fromTable.key()]).del();
    }
  }, {
    key: 'join',
    value: function join() {
      var joiner = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : function () {};
      var label = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : null;

      label = this.jointLabel(label, {});
      var fromTable = this.fromTable,
          toTable = this.toTable,
          inverse = this.inverse;
      var foreignKey = inverse.foreignKey,
          typeField = inverse.typeField;


      if (this.ownerTable.scopeTrack.hasJoint(label)) {
        return this.ownerTable;
      } else {
        return this.ownerTable.joint(function (q) {
          q.join(toTable.tableName(), function (j) {
            j.on(toTable.c(typeField), '=', fromTable.orm.raw('?', [fromTable.tableName()])).on(toTable.c(foreignKey), '=', fromTable.keyCol());

            joiner(j);
          });
        }, label);
      }
    }
  }, {
    key: 'leftJoin',
    value: function leftJoin() {
      var joiner = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : function () {};
      var label = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : null;

      label = this.jointLabel(label, { isLeftJoin: true });
      var fromTable = this.fromTable,
          toTable = this.toTable,
          inverse = this.inverse;
      var foreignKey = inverse.foreignKey,
          typeField = inverse.typeField;


      if (this.ownerTable.scopeTrackhasJoint(label)) {
        return this.ownerTable;
      } else {
        return this.ownerTable.joint(function (q) {
          q.leftJoin(toTable.tableName(), function (j) {
            j.on(toTable.c(typeField), '=', fromTable.orm.raw('?', [fromTable.tableName()])).on(toTable.c(foreignKey), '=', fromTable.keyCol());

            joiner(j);
          });
        }, label);
      }
    }
  }]);

  return MorphMany;
}(Relation);

module.exports = MorphMany;