@lskjs/db/lib/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