@azteam/mysql-model/lib/DataRepository.js

N/A

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = void 0;
var _sequelize = require("sequelize");
var _error = require("@azteam/error");
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
function _defineProperty(e, r, t) { return (r = _toPropertyKey(r)) in e ? Object.defineProperty(e, r, { value: t, enumerable: !0, configurable: !0, writable: !0 }) : e[r] = t, e; }
function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/babel/babel/blob/main/packages/babel-helpers/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return r; }; var t, r = {}, e = Object.prototype, n = e.hasOwnProperty, o = "function" == typeof Symbol ? Symbol : {}, i = o.iterator || "@@iterator", a = o.asyncIterator || "@@asyncIterator", u = o.toStringTag || "@@toStringTag"; function c(t, r, e, n) { Object.defineProperty(t, r, { value: e, enumerable: !n, configurable: !n, writable: !n }); } try { c({}, ""); } catch (t) { c = function c(t, r, e) { return t[r] = e; }; } function h(r, e, n, o) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype); return c(a, "_invoke", function (r, e, n) { var o = 1; return function (i, a) { if (3 === o) throw Error("Generator is already running"); if (4 === o) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var u = n.delegate; if (u) { var c = d(u, n); if (c) { if (c === f) continue; return c; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (1 === o) throw o = 4, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = 3; var h = s(r, e, n); if ("normal" === h.type) { if (o = n.done ? 4 : 2, h.arg === f) continue; return { value: h.arg, done: n.done }; } "throw" === h.type && (o = 4, n.method = "throw", n.arg = h.arg); } }; }(r, n, new Context(o || [])), !0), a; } function s(t, r, e) { try { return { type: "normal", arg: t.call(r, e) }; } catch (t) { return { type: "throw", arg: t }; } } r.wrap = h; var f = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var l = {}; c(l, i, function () { return this; }); var p = Object.getPrototypeOf, y = p && p(p(x([]))); y && y !== e && n.call(y, i) && (l = y); var v = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(l); function g(t) { ["next", "throw", "return"].forEach(function (r) { c(t, r, function (t) { return this._invoke(r, t); }); }); } function AsyncIterator(t, r) { function e(o, i, a, u) { var c = s(t[o], t, i); if ("throw" !== c.type) { var h = c.arg, f = h.value; return f && "object" == _typeof(f) && n.call(f, "__await") ? r.resolve(f.__await).then(function (t) { e("next", t, a, u); }, function (t) { e("throw", t, a, u); }) : r.resolve(f).then(function (t) { h.value = t, a(h); }, function (t) { return e("throw", t, a, u); }); } u(c.arg); } var o; c(this, "_invoke", function (t, n) { function i() { return new r(function (r, o) { e(t, n, r, o); }); } return o = o ? o.then(i, i) : i(); }, !0); } function d(r, e) { var n = e.method, o = r.i[n]; if (o === t) return e.delegate = null, "throw" === n && r.i["return"] && (e.method = "return", e.arg = t, d(r, e), "throw" === e.method) || "return" !== n && (e.method = "throw", e.arg = new TypeError("The iterator does not provide a '" + n + "' method")), f; var i = s(o, r.i, e.arg); if ("throw" === i.type) return e.method = "throw", e.arg = i.arg, e.delegate = null, f; var a = i.arg; return a ? a.done ? (e[r.r] = a.value, e.next = r.n, "return" !== e.method && (e.method = "next", e.arg = t), e.delegate = null, f) : a : (e.method = "throw", e.arg = new TypeError("iterator result is not an object"), e.delegate = null, f); } function w(t) { this.tryEntries.push(t); } function m(r) { var e = r[4] || {}; e.type = "normal", e.arg = t, r[4] = e; } function Context(t) { this.tryEntries = [[-1]], t.forEach(w, this), this.reset(!0); } function x(r) { if (null != r) { var e = r[i]; if (e) return e.call(r); if ("function" == typeof r.next) return r; if (!isNaN(r.length)) { var o = -1, a = function e() { for (; ++o < r.length;) if (n.call(r, o)) return e.value = r[o], e.done = !1, e; return e.value = t, e.done = !0, e; }; return a.next = a; } } throw new TypeError(_typeof(r) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, c(v, "constructor", GeneratorFunctionPrototype), c(GeneratorFunctionPrototype, "constructor", GeneratorFunction), c(GeneratorFunctionPrototype, u, GeneratorFunction.displayName = "GeneratorFunction"), r.isGeneratorFunction = function (t) { var r = "function" == typeof t && t.constructor; return !!r && (r === GeneratorFunction || "GeneratorFunction" === (r.displayName || r.name)); }, r.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, c(t, u, "GeneratorFunction")), t.prototype = Object.create(v), t; }, r.awrap = function (t) { return { __await: t }; }, g(AsyncIterator.prototype), c(AsyncIterator.prototype, a, function () { return this; }), r.AsyncIterator = AsyncIterator, r.async = function (t, e, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(h(t, e, n, o), i); return r.isGeneratorFunction(e) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, g(v), c(v, u, "Generator"), c(v, i, function () { return this; }), c(v, "toString", function () { return "[object Generator]"; }), r.keys = function (t) { var r = Object(t), e = []; for (var n in r) e.unshift(n); return function t() { for (; e.length;) if ((n = e.pop()) in r) return t.value = n, t.done = !1, t; return t.done = !0, t; }; }, r.values = x, Context.prototype = { constructor: Context, reset: function reset(r) { if (this.prev = this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(m), !r) for (var e in this) "t" === e.charAt(0) && n.call(this, e) && !isNaN(+e.slice(1)) && (this[e] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0][4]; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(r) { if (this.done) throw r; var e = this; function n(t) { a.type = "throw", a.arg = r, e.next = t; } for (var o = e.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i[4], u = this.prev, c = i[1], h = i[2]; if (-1 === i[0]) return n("end"), !1; if (!c && !h) throw Error("try statement without catch or finally"); if (null != i[0] && i[0] <= u) { if (u < c) return this.method = "next", this.arg = t, n(c), !0; if (u < h) return n(h), !1; } } }, abrupt: function abrupt(t, r) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var n = this.tryEntries[e]; if (n[0] > -1 && n[0] <= this.prev && this.prev < n[2]) { var o = n; break; } } o && ("break" === t || "continue" === t) && o[0] <= r && r <= o[2] && (o = null); var i = o ? o[4] : {}; return i.type = t, i.arg = r, o ? (this.method = "next", this.next = o[2], f) : this.complete(i); }, complete: function complete(t, r) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && r && (this.next = r), f; }, finish: function finish(t) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var e = this.tryEntries[r]; if (e[2] === t) return this.complete(e[4], e[3]), m(e), f; } }, "catch": function _catch(t) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var e = this.tryEntries[r]; if (e[0] === t) { var n = e[4]; if ("throw" === n.type) { var o = n.arg; m(e); } return o; } } throw Error("illegal catch attempt"); }, delegateYield: function delegateYield(r, e, n) { return this.delegate = { i: x(r), r: e, n: n }, "next" === this.method && (this.arg = t), f; } }, r; }
function asyncGeneratorStep(n, t, e, r, o, a, c) { try { var i = n[a](c), u = i.value; } catch (n) { return void e(n); } i.done ? t(u) : Promise.resolve(u).then(r, o); }
function _asyncToGenerator(n) { return function () { var t = this, e = arguments; return new Promise(function (r, o) { var a = n.apply(t, e); function _next(n) { asyncGeneratorStep(a, r, o, _next, _throw, "next", n); } function _throw(n) { asyncGeneratorStep(a, r, o, _next, _throw, "throw", n); } _next(void 0); }); }; }
function _classCallCheck(a, n) { if (!(a instanceof n)) throw new TypeError("Cannot call a class as a function"); }
function _defineProperties(e, r) { for (var t = 0; t < r.length; t++) { var o = r[t]; o.enumerable = o.enumerable || !1, o.configurable = !0, "value" in o && (o.writable = !0), Object.defineProperty(e, _toPropertyKey(o.key), o); } }
function _createClass(e, r, t) { return r && _defineProperties(e.prototype, r), t && _defineProperties(e, t), Object.defineProperty(e, "prototype", { writable: !1 }), e; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : i + ""; }
function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
var DataRepository = /*#__PURE__*/function () {
  function DataRepository(model) {
    var fks = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
    var dummyData = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : [];
    _classCallCheck(this, DataRepository);
    this._model = model;
    this._fks = fks;
    this._dummyData(dummyData);
  }
  return _createClass(DataRepository, [{
    key: "getModel",
    value: function getModel() {
      if (this._model) {
        return this._model;
      }
      throw new _error.ErrorException(_error.NOT_INIT_METHOD);
    }
  }, {
    key: "_dummyData",
    value: function _dummyData(data) {
      var _this = this;
      data.map(/*#__PURE__*/function () {
        var _ref = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(item) {
          var obj;
          return _regeneratorRuntime().wrap(function _callee$(_context) {
            while (1) switch (_context.prev = _context.next) {
              case 0:
                _context.next = 2;
                return _this.findOne(item.query);
              case 2:
                obj = _context.sent;
                if (obj) {
                  _context.next = 6;
                  break;
                }
                _context.next = 6;
                return _this.create(item.data);
              case 6:
              case "end":
                return _context.stop();
            }
          }, _callee);
        }));
        return function (_x) {
          return _ref.apply(this, arguments);
        };
      }());
    }
  }, {
    key: "_querySoftDelete",
    value: function _querySoftDelete(query) {
      var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
      var Model = this.getModel();
      if (Model.isSoftDelete) {
        var querySoft = query;
        if (options.force) {
          delete querySoft.deleted_at;
        } else if (options.trash) {
          querySoft.deleted_at = _defineProperty({}, _sequelize.Op.ne, 0);
        } else {
          querySoft.deleted_at = 0;
        }
        return querySoft;
      }
      return query;
    }
  }, {
    key: "findOne",
    value: function findOne() {
      var query = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
      var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
      var Model = this.getModel();
      return Model.findOne({
        where: this._querySoftDelete(query, options)
      });
    }
  }, {
    key: "findOneById",
    value: function findOneById(id) {
      return this.findOne({
        id: id
      });
    }
  }, {
    key: "findOneOrCreate",
    value: function () {
      var _findOneOrCreate = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(query, data) {
        var guard,
          allow,
          model,
          _args2 = arguments;
        return _regeneratorRuntime().wrap(function _callee2$(_context2) {
          while (1) switch (_context2.prev = _context2.next) {
            case 0:
              guard = _args2.length > 2 && _args2[2] !== undefined ? _args2[2] : [];
              allow = _args2.length > 3 && _args2[3] !== undefined ? _args2[3] : [];
              _context2.next = 4;
              return this.findOne(query);
            case 4:
              model = _context2.sent;
              if (model) {
                _context2.next = 9;
                break;
              }
              _context2.next = 8;
              return this.create(data, guard, allow);
            case 8:
              model = _context2.sent;
            case 9:
              return _context2.abrupt("return", model);
            case 10:
            case "end":
              return _context2.stop();
          }
        }, _callee2, this);
      }));
      function findOneOrCreate(_x2, _x3) {
        return _findOneOrCreate.apply(this, arguments);
      }
      return findOneOrCreate;
    }()
  }, {
    key: "find",
    value: function find() {
      var query = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
      var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
      var Model = this.getModel();
      var querySoft = this._querySoftDelete(query, options);
      var newOpt = options;
      if (!newOpt.sort) {
        newOpt.sort = {
          created_at: 'desc'
        };
      }
      if (newOpt.forceDisableSort) {
        delete newOpt.sort;
      }
      if (newOpt.page) {
        return Model.paginate(querySoft, newOpt);
      }
      var queryBuilder = Model.find(querySoft).sort(newOpt.sort);
      var limit = newOpt.limit && newOpt.limit < 10000 ? newOpt.limit : 10000;
      return queryBuilder.limit(limit);
    }
  }, {
    key: "create",
    value: function create() {
      var data = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
      var guard = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
      var allow = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : [];
      var createdId = arguments.length > 3 && arguments[3] !== undefined ? arguments[3] : null;
      var Model = this.getModel();
      var model = new Model();
      if (createdId) {
        model.created_id = createdId;
        model.modified_id = createdId;
      }
      model.loadData(data, guard, allow);
      return Model.create(data);
    }
  }, {
    key: "createByUser",
    value: function createByUser(createdId) {
      var data = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
      var guard = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : [];
      var allow = arguments.length > 3 && arguments[3] !== undefined ? arguments[3] : [];
      return this.create(data, guard, allow, createdId);
    }
  }, {
    key: "modify",
    value: function () {
      var _modify = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(query, data) {
        var Model;
        return _regeneratorRuntime().wrap(function _callee3$(_context3) {
          while (1) switch (_context3.prev = _context3.next) {
            case 0:
              Model = this.getModel();
              return _context3.abrupt("return", Model.updateOne(this._querySoftDelete(query), data));
            case 2:
            case "end":
              return _context3.stop();
          }
        }, _callee3, this);
      }));
      function modify(_x4, _x5) {
        return _modify.apply(this, arguments);
      }
      return modify;
    }()
  }, {
    key: "modifyById",
    value: function () {
      var _modifyById = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(id, data) {
        var Model;
        return _regeneratorRuntime().wrap(function _callee4$(_context4) {
          while (1) switch (_context4.prev = _context4.next) {
            case 0:
              Model = this.getModel();
              return _context4.abrupt("return", Model.updateOne(this._querySoftDelete({
                id: id
              }), data));
            case 2:
            case "end":
              return _context4.stop();
          }
        }, _callee4, this);
      }));
      function modifyById(_x6, _x7) {
        return _modifyById.apply(this, arguments);
      }
      return modifyById;
    }()
  }, {
    key: "modifyAll",
    value: function () {
      var _modifyAll = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(query, data) {
        var Model;
        return _regeneratorRuntime().wrap(function _callee5$(_context5) {
          while (1) switch (_context5.prev = _context5.next) {
            case 0:
              Model = this.getModel();
              return _context5.abrupt("return", Model.updateMany(this._querySoftDelete(query), data));
            case 2:
            case "end":
              return _context5.stop();
          }
        }, _callee5, this);
      }));
      function modifyAll(_x8, _x9) {
        return _modifyAll.apply(this, arguments);
      }
      return modifyAll;
    }()
  }, {
    key: "deleteById",
    value: function () {
      var _deleteById = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(id) {
        return _regeneratorRuntime().wrap(function _callee6$(_context6) {
          while (1) switch (_context6.prev = _context6.next) {
            case 0:
              return _context6.abrupt("return", this.modifyById(id, {
                deleted_at: Math.floor(Date.now() / 1000),
                $unset: {
                  deleted_id: 1
                }
              }));
            case 1:
            case "end":
              return _context6.stop();
          }
        }, _callee6, this);
      }));
      function deleteById(_x0) {
        return _deleteById.apply(this, arguments);
      }
      return deleteById;
    }()
  }, {
    key: "deleteAll",
    value: function () {
      var _deleteAll = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(query) {
        return _regeneratorRuntime().wrap(function _callee7$(_context7) {
          while (1) switch (_context7.prev = _context7.next) {
            case 0:
              return _context7.abrupt("return", this.modifyAll(query, {
                deleted_at: Math.floor(Date.now() / 1000),
                $unset: {
                  deleted_id: 1
                }
              }));
            case 1:
            case "end":
              return _context7.stop();
          }
        }, _callee7, this);
      }));
      function deleteAll(_x1) {
        return _deleteAll.apply(this, arguments);
      }
      return deleteAll;
    }()
  }, {
    key: "restoreAll",
    value: function () {
      var _restoreAll = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(query) {
        var Model;
        return _regeneratorRuntime().wrap(function _callee8$(_context8) {
          while (1) switch (_context8.prev = _context8.next) {
            case 0:
              Model = this.getModel();
              return _context8.abrupt("return", Model.update({
                deleted_at: 0,
                restored_id: 0
              }, {
                where: query
              }));
            case 2:
            case "end":
              return _context8.stop();
          }
        }, _callee8, this);
      }));
      function restoreAll(_x10) {
        return _restoreAll.apply(this, arguments);
      }
      return restoreAll;
    }()
  }, {
    key: "destroy",
    value: function destroy(id) {
      var Model = this.getModel();
      return Model.destroy({
        id: id
      });
    }
  }, {
    key: "destroyAll",
    value: function () {
      var _destroyAll = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(query) {
        var Model;
        return _regeneratorRuntime().wrap(function _callee9$(_context9) {
          while (1) switch (_context9.prev = _context9.next) {
            case 0:
              Model = this.getModel();
              return _context9.abrupt("return", Model.destroy({
                where: query
              }));
            case 2:
            case "end":
              return _context9.stop();
          }
        }, _callee9, this);
      }));
      function destroyAll(_x11) {
        return _destroyAll.apply(this, arguments);
      }
      return destroyAll;
    }()
  }]);
}();
var _default = exports["default"] = DataRepository;