@lskjs/db/Model.js

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = exports.Model = void 0;

var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));

var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));

var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));

var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));

var _bluebird = _interopRequireDefault(require("bluebird"));

var _forEach = _interopRequireDefault(require("lodash/forEach"));

var _get = _interopRequireDefault(require("lodash/get"));

var _set = _interopRequireDefault(require("lodash/set"));

var _mongoose = require("mongoose");

var Model = /*#__PURE__*/function () {
  function Model() {
    (0, _classCallCheck2["default"])(this, Model);
    this.__model = true;
  }

  (0, _createClass2["default"])(Model, [{
    key: "setState",
    value: function setState() {
      var _this = this;

      var state = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
      (0, _forEach["default"])(state, function (value, key) {
        (0, _set["default"])(_this, key, value); // @ts-ignore

        _this.markModified(key);
      });
    }
  }, {
    key: "getState",
    value: function getState(key, def) {
      if (!key) return this;
      return (0, _get["default"])(this, key, def);
    } // eslint-disable-next-line @typescript-eslint/no-unused-vars

  }], [{
    key: "prepareOne",
    value: function () {
      var _prepareOne = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee(obj, params) {
        return _regenerator["default"].wrap(function _callee$(_context) {
          while (1) {
            switch (_context.prev = _context.next) {
              case 0:
                return _context.abrupt("return", obj);

              case 1:
              case "end":
                return _context.stop();
            }
          }
        }, _callee);
      }));

      function prepareOne(_x, _x2) {
        return _prepareOne.apply(this, arguments);
      }

      return prepareOne;
    }()
  }, {
    key: "prepare",
    value: function () {
      var _prepare = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee2(obj) {
        var _this2 = this;

        var params,
            toObjectOne,
            res,
            _args2 = arguments;
        return _regenerator["default"].wrap(function _callee2$(_context2) {
          while (1) {
            switch (_context2.prev = _context2.next) {
              case 0:
                params = _args2.length > 1 && _args2[1] !== undefined ? _args2[1] : {};

                // // console.log('PREPARE params.toObject', params.toObject);
                // // console.log('params.toObject2@@@', params.toObject2);
                toObjectOne = function toObjectOne(o) {
                  if (!params.toObject2 || !Object.keys(params.toObject2).length) return o;
                  if (o && o.toObject) return o.toObject(params.toObject2);
                  return o;
                };

                if (!Array.isArray(obj)) {
                  _context2.next = 8;
                  break;
                }

                _context2.next = 5;
                return _bluebird["default"].map(obj, function (o) {
                  return _this2.prepareOne(o, params);
                });

              case 5:
                res = _context2.sent;
                _context2.next = 11;
                break;

              case 8:
                _context2.next = 10;
                return this.prepareOne(obj, params);

              case 10:
                res = _context2.sent;

              case 11:
                if (!Array.isArray(res)) {
                  _context2.next = 13;
                  break;
                }

                return _context2.abrupt("return", res.map(toObjectOne));

              case 13:
                return _context2.abrupt("return", toObjectOne(res));

              case 14:
              case "end":
                return _context2.stop();
            }
          }
        }, _callee2, this);
      }));

      function prepare(_x3) {
        return _prepare.apply(this, arguments);
      }

      return prepare;
    }()
  }]);
  return Model;
}();

exports.Model = Model;
Model.Schema = _mongoose.Schema;
Model.Types = _mongoose.Schema.Types;
Model.defaultOptions = {
  timestamps: true
};
Model.options = {};
var _default = Model;
exports["default"] = _default;
//# sourceMappingURL=Model.js.map