@lskjs/bots-plugin-notify
Version:
LSK.js plugin for @lskjs/bots module for telegram, slack, discord, whatsapp, twitter, instagram and vk bots creation
386 lines (385 loc) • 29.1 kB
JavaScript
"use strict";
var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
var _typeof = require("@babel/runtime/helpers/typeof");
Object.defineProperty(exports, "__esModule", {
value: true
});
exports["default"] = exports.NotifyPlugin = void 0;
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 _inherits2 = _interopRequireDefault(require("@babel/runtime/helpers/inherits"));
var _possibleConstructorReturn2 = _interopRequireDefault(require("@babel/runtime/helpers/possibleConstructorReturn"));
var _getPrototypeOf2 = _interopRequireDefault(require("@babel/runtime/helpers/getPrototypeOf"));
var _botsPlugin = _interopRequireDefault(require("@lskjs/bots-plugin"));
var _err = _interopRequireDefault(require("@lskjs/err"));
var _axios = _interopRequireDefault(require("axios"));
var _bluebird = _interopRequireDefault(require("bluebird"));
var _forEach = _interopRequireDefault(require("lodash/forEach"));
var _get = _interopRequireDefault(require("lodash/get"));
var _isEmpty = _interopRequireDefault(require("lodash/isEmpty"));
var _nodeCron = _interopRequireDefault(require("node-cron"));
var _createPost = require("./createPost");
function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return exports; }; var exports = {}, Op = Object.prototype, hasOwn = Op.hasOwnProperty, defineProperty = Object.defineProperty || function (obj, key, desc) { obj[key] = desc.value; }, $Symbol = "function" == typeof Symbol ? Symbol : {}, iteratorSymbol = $Symbol.iterator || "@@iterator", asyncIteratorSymbol = $Symbol.asyncIterator || "@@asyncIterator", toStringTagSymbol = $Symbol.toStringTag || "@@toStringTag"; function define(obj, key, value) { return Object.defineProperty(obj, key, { value: value, enumerable: !0, configurable: !0, writable: !0 }), obj[key]; } try { define({}, ""); } catch (err) { define = function define(obj, key, value) { return obj[key] = value; }; } function wrap(innerFn, outerFn, self, tryLocsList) { var protoGenerator = outerFn && outerFn.prototype instanceof Generator ? outerFn : Generator, generator = Object.create(protoGenerator.prototype), context = new Context(tryLocsList || []); return defineProperty(generator, "_invoke", { value: makeInvokeMethod(innerFn, self, context) }), generator; } function tryCatch(fn, obj, arg) { try { return { type: "normal", arg: fn.call(obj, arg) }; } catch (err) { return { type: "throw", arg: err }; } } exports.wrap = wrap; var ContinueSentinel = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var IteratorPrototype = {}; define(IteratorPrototype, iteratorSymbol, function () { return this; }); var getProto = Object.getPrototypeOf, NativeIteratorPrototype = getProto && getProto(getProto(values([]))); NativeIteratorPrototype && NativeIteratorPrototype !== Op && hasOwn.call(NativeIteratorPrototype, iteratorSymbol) && (IteratorPrototype = NativeIteratorPrototype); var Gp = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(IteratorPrototype); function defineIteratorMethods(prototype) { ["next", "throw", "return"].forEach(function (method) { define(prototype, method, function (arg) { return this._invoke(method, arg); }); }); } function AsyncIterator(generator, PromiseImpl) { function invoke(method, arg, resolve, reject) { var record = tryCatch(generator[method], generator, arg); if ("throw" !== record.type) { var result = record.arg, value = result.value; return value && "object" == _typeof(value) && hasOwn.call(value, "__await") ? PromiseImpl.resolve(value.__await).then(function (value) { invoke("next", value, resolve, reject); }, function (err) { invoke("throw", err, resolve, reject); }) : PromiseImpl.resolve(value).then(function (unwrapped) { result.value = unwrapped, resolve(result); }, function (error) { return invoke("throw", error, resolve, reject); }); } reject(record.arg); } var previousPromise; defineProperty(this, "_invoke", { value: function value(method, arg) { function callInvokeWithMethodAndArg() { return new PromiseImpl(function (resolve, reject) { invoke(method, arg, resolve, reject); }); } return previousPromise = previousPromise ? previousPromise.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(innerFn, self, context) { var state = "suspendedStart"; return function (method, arg) { if ("executing" === state) throw new Error("Generator is already running"); if ("completed" === state) { if ("throw" === method) throw arg; return doneResult(); } for (context.method = method, context.arg = arg;;) { var delegate = context.delegate; if (delegate) { var delegateResult = maybeInvokeDelegate(delegate, context); if (delegateResult) { if (delegateResult === ContinueSentinel) continue; return delegateResult; } } if ("next" === context.method) context.sent = context._sent = context.arg;else if ("throw" === context.method) { if ("suspendedStart" === state) throw state = "completed", context.arg; context.dispatchException(context.arg); } else "return" === context.method && context.abrupt("return", context.arg); state = "executing"; var record = tryCatch(innerFn, self, context); if ("normal" === record.type) { if (state = context.done ? "completed" : "suspendedYield", record.arg === ContinueSentinel) continue; return { value: record.arg, done: context.done }; } "throw" === record.type && (state = "completed", context.method = "throw", context.arg = record.arg); } }; } function maybeInvokeDelegate(delegate, context) { var methodName = context.method, method = delegate.iterator[methodName]; if (undefined === method) return context.delegate = null, "throw" === methodName && delegate.iterator["return"] && (context.method = "return", context.arg = undefined, maybeInvokeDelegate(delegate, context), "throw" === context.method) || "return" !== methodName && (context.method = "throw", context.arg = new TypeError("The iterator does not provide a '" + methodName + "' method")), ContinueSentinel; var record = tryCatch(method, delegate.iterator, context.arg); if ("throw" === record.type) return context.method = "throw", context.arg = record.arg, context.delegate = null, ContinueSentinel; var info = record.arg; return info ? info.done ? (context[delegate.resultName] = info.value, context.next = delegate.nextLoc, "return" !== context.method && (context.method = "next", context.arg = undefined), context.delegate = null, ContinueSentinel) : info : (context.method = "throw", context.arg = new TypeError("iterator result is not an object"), context.delegate = null, ContinueSentinel); } function pushTryEntry(locs) { var entry = { tryLoc: locs[0] }; 1 in locs && (entry.catchLoc = locs[1]), 2 in locs && (entry.finallyLoc = locs[2], entry.afterLoc = locs[3]), this.tryEntries.push(entry); } function resetTryEntry(entry) { var record = entry.completion || {}; record.type = "normal", delete record.arg, entry.completion = record; } function Context(tryLocsList) { this.tryEntries = [{ tryLoc: "root" }], tryLocsList.forEach(pushTryEntry, this), this.reset(!0); } function values(iterable) { if (iterable) { var iteratorMethod = iterable[iteratorSymbol]; if (iteratorMethod) return iteratorMethod.call(iterable); if ("function" == typeof iterable.next) return iterable; if (!isNaN(iterable.length)) { var i = -1, next = function next() { for (; ++i < iterable.length;) { if (hasOwn.call(iterable, i)) return next.value = iterable[i], next.done = !1, next; } return next.value = undefined, next.done = !0, next; }; return next.next = next; } } return { next: doneResult }; } function doneResult() { return { value: undefined, done: !0 }; } return GeneratorFunction.prototype = GeneratorFunctionPrototype, defineProperty(Gp, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), defineProperty(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, toStringTagSymbol, "GeneratorFunction"), exports.isGeneratorFunction = function (genFun) { var ctor = "function" == typeof genFun && genFun.constructor; return !!ctor && (ctor === GeneratorFunction || "GeneratorFunction" === (ctor.displayName || ctor.name)); }, exports.mark = function (genFun) { return Object.setPrototypeOf ? Object.setPrototypeOf(genFun, GeneratorFunctionPrototype) : (genFun.__proto__ = GeneratorFunctionPrototype, define(genFun, toStringTagSymbol, "GeneratorFunction")), genFun.prototype = Object.create(Gp), genFun; }, exports.awrap = function (arg) { return { __await: arg }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, asyncIteratorSymbol, function () { return this; }), exports.AsyncIterator = AsyncIterator, exports.async = function (innerFn, outerFn, self, tryLocsList, PromiseImpl) { void 0 === PromiseImpl && (PromiseImpl = Promise); var iter = new AsyncIterator(wrap(innerFn, outerFn, self, tryLocsList), PromiseImpl); return exports.isGeneratorFunction(outerFn) ? iter : iter.next().then(function (result) { return result.done ? result.value : iter.next(); }); }, defineIteratorMethods(Gp), define(Gp, toStringTagSymbol, "Generator"), define(Gp, iteratorSymbol, function () { return this; }), define(Gp, "toString", function () { return "[object Generator]"; }), exports.keys = function (val) { var object = Object(val), keys = []; for (var key in object) { keys.push(key); } return keys.reverse(), function next() { for (; keys.length;) { var key = keys.pop(); if (key in object) return next.value = key, next.done = !1, next; } return next.done = !0, next; }; }, exports.values = values, Context.prototype = { constructor: Context, reset: function reset(skipTempReset) { if (this.prev = 0, this.next = 0, this.sent = this._sent = undefined, this.done = !1, this.delegate = null, this.method = "next", this.arg = undefined, this.tryEntries.forEach(resetTryEntry), !skipTempReset) for (var name in this) { "t" === name.charAt(0) && hasOwn.call(this, name) && !isNaN(+name.slice(1)) && (this[name] = undefined); } }, stop: function stop() { this.done = !0; var rootRecord = this.tryEntries[0].completion; if ("throw" === rootRecord.type) throw rootRecord.arg; return this.rval; }, dispatchException: function dispatchException(exception) { if (this.done) throw exception; var context = this; function handle(loc, caught) { return record.type = "throw", record.arg = exception, context.next = loc, caught && (context.method = "next", context.arg = undefined), !!caught; } for (var i = this.tryEntries.length - 1; i >= 0; --i) { var entry = this.tryEntries[i], record = entry.completion; if ("root" === entry.tryLoc) return handle("end"); if (entry.tryLoc <= this.prev) { var hasCatch = hasOwn.call(entry, "catchLoc"), hasFinally = hasOwn.call(entry, "finallyLoc"); if (hasCatch && hasFinally) { if (this.prev < entry.catchLoc) return handle(entry.catchLoc, !0); if (this.prev < entry.finallyLoc) return handle(entry.finallyLoc); } else if (hasCatch) { if (this.prev < entry.catchLoc) return handle(entry.catchLoc, !0); } else { if (!hasFinally) throw new Error("try statement without catch or finally"); if (this.prev < entry.finallyLoc) return handle(entry.finallyLoc); } } } }, abrupt: function abrupt(type, arg) { for (var i = this.tryEntries.length - 1; i >= 0; --i) { var entry = this.tryEntries[i]; if (entry.tryLoc <= this.prev && hasOwn.call(entry, "finallyLoc") && this.prev < entry.finallyLoc) { var finallyEntry = entry; break; } } finallyEntry && ("break" === type || "continue" === type) && finallyEntry.tryLoc <= arg && arg <= finallyEntry.finallyLoc && (finallyEntry = null); var record = finallyEntry ? finallyEntry.completion : {}; return record.type = type, record.arg = arg, finallyEntry ? (this.method = "next", this.next = finallyEntry.finallyLoc, ContinueSentinel) : this.complete(record); }, complete: function complete(record, afterLoc) { if ("throw" === record.type) throw record.arg; return "break" === record.type || "continue" === record.type ? this.next = record.arg : "return" === record.type ? (this.rval = this.arg = record.arg, this.method = "return", this.next = "end") : "normal" === record.type && afterLoc && (this.next = afterLoc), ContinueSentinel; }, finish: function finish(finallyLoc) { for (var i = this.tryEntries.length - 1; i >= 0; --i) { var entry = this.tryEntries[i]; if (entry.finallyLoc === finallyLoc) return this.complete(entry.completion, entry.afterLoc), resetTryEntry(entry), ContinueSentinel; } }, "catch": function _catch(tryLoc) { for (var i = this.tryEntries.length - 1; i >= 0; --i) { var entry = this.tryEntries[i]; if (entry.tryLoc === tryLoc) { var record = entry.completion; if ("throw" === record.type) { var thrown = record.arg; resetTryEntry(entry); } return thrown; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(iterable, resultName, nextLoc) { return this.delegate = { iterator: values(iterable), resultName: resultName, nextLoc: nextLoc }, "next" === this.method && (this.arg = undefined), ContinueSentinel; } }, exports; }
function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = (0, _getPrototypeOf2["default"])(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = (0, _getPrototypeOf2["default"])(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return (0, _possibleConstructorReturn2["default"])(this, result); }; }
function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
var NotifyPlugin = /*#__PURE__*/function (_BaseBotPlugin) {
(0, _inherits2["default"])(NotifyPlugin, _BaseBotPlugin);
var _super = _createSuper(NotifyPlugin);
function NotifyPlugin() {
var _this;
(0, _classCallCheck2["default"])(this, NotifyPlugin);
for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
args[_key] = arguments[_key];
}
_this = _super.call.apply(_super, [this].concat(args));
_this.debug = 1;
_this.providers = ['telegram', 'slack'];
_this.crons = [];
_this.alertmanager = _createPost.alertmanager;
_this.github = _createPost.github;
_this.gitlab = _createPost.gitlab;
_this.graylog = _createPost.graylog;
_this.monitoring = _createPost.monitoring;
_this.manual = _createPost.manual;
return _this;
}
(0, _createClass2["default"])(NotifyPlugin, [{
key: "sendNotification",
value: // getEconnabortedErrorMessage = getEconnabortedErrorMessage;
// getOtherErrorMessage = getOtherErrorMessage;
// getRedirectErrorMessage = getRedirectErrorMessage;
// getWarningMessage = getWarningMessage;
function () {
var _sendNotification = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref) {
var _this2 = this;
var bot, message, provider, projectName, project, isDefault, res, _res, msg, options, chats;
return _regeneratorRuntime().wrap(function _callee$(_context) {
while (1) {
switch (_context.prev = _context.next) {
case 0:
bot = _ref.bot, message = _ref.message;
if (message) {
_context.next = 3;
break;
}
throw new _err["default"]('!message');
case 3:
provider = bot.provider;
projectName = message.projectName;
project = this.config.projects[projectName];
isDefault = false;
if (!project) {
isDefault = true;
project = this.config.projects._default;
}
if (this.debug) this.log.trace('NotifyPlugin.sendNotification.message', message);
if (!(message.type === 'gitlab')) {
_context.next = 15;
break;
}
_context.next = 12;
return this.gitlab(message, project, bot);
case 12:
res = _context.sent;
_context.next = 47;
break;
case 15:
if (!(message.type === 'github')) {
_context.next = 21;
break;
}
_context.next = 18;
return this.github(message, project, bot);
case 18:
res = _context.sent;
_context.next = 47;
break;
case 21:
if (!(message.type === 'alertmanager')) {
_context.next = 27;
break;
}
_context.next = 24;
return this.alertmanager(message, bot);
case 24:
res = _context.sent;
_context.next = 47;
break;
case 27:
if (!(message.type === 'graylog')) {
_context.next = 33;
break;
}
_context.next = 30;
return this.graylog(message);
case 30:
res = _context.sent;
_context.next = 47;
break;
case 33:
if (!(message.type === 'monitoring')) {
_context.next = 39;
break;
}
_context.next = 36;
return this.monitoring(message);
case 36:
res = _context.sent;
_context.next = 47;
break;
case 39:
if (!(message.type === 'manual')) {
_context.next = 45;
break;
}
_context.next = 42;
return this.manual(message);
case 42:
res = _context.sent;
_context.next = 47;
break;
case 45:
res = {
msg: "!type\n\n".concat(JSON.stringify(message))
};
this.log.warn('!type', {
message: message
});
case 47:
if (res) {
_context.next = 49;
break;
}
return _context.abrupt("return", null);
case 49:
_res = res, msg = _res.msg, options = _res.options;
if (message.md || message.isMd) {
if (!options) options = {};
options.parse_mode = 'MarkdownV2';
}
if (isDefault && msg || message.showChannel) {
msg = "/notify/".concat(projectName, "\n\n").concat(msg);
}
chats = project[provider] || [];
if (msg) {
_context.next = 55;
break;
}
throw new _err["default"]('!NotifyPlugin.sendNotification.msg');
case 55:
return _context.abrupt("return", _bluebird["default"].map(chats.filter(Boolean), function (chat) {
//eslint-disable-line
return bot.sendMessage(chat, msg, options)["catch"](function (err) {
if (err && err.response && err.response.error_code === 400 && err.response.description && err.response.description.startsWith("Bad Request: can't parse entities")) {
msg += "\n\n------P.S.-------\n".concat(err.response.description);
_this2.log.warn('[sendMessage]', err);
return bot.sendMessage(chat, msg, {});
}
_this2.log.error('[sendMessage]', err);
throw err;
});
}));
case 56:
case "end":
return _context.stop();
}
}
}, _callee, this);
}));
function sendNotification(_x) {
return _sendNotification.apply(this, arguments);
}
return sendNotification;
}()
}, {
key: "checkResourses",
value: function checkResourses(bot) {
var _this3 = this;
var projects = this.config.projects;
var timeout = 30 * 1000;
var timeoutWarn = 7 * 1000;
(0, _forEach["default"])(projects, function (project, projectName) {
// eslint-disable-next-line no-param-reassign
project.name = projectName;
});
return _bluebird["default"].map(Object.values(projects), /*#__PURE__*/function () {
var _ref2 = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(project) {
var monitorings;
return _regeneratorRuntime().wrap(function _callee3$(_context3) {
while (1) {
switch (_context3.prev = _context3.next) {
case 0:
monitorings = project.monitoring;
if (monitorings && monitorings.length > 0) {
_context3.next = 3;
break;
}
return _context3.abrupt("return", null);
case 3:
return _context3.abrupt("return", _bluebird["default"].map(monitorings,
/*#__PURE__*/
// eslint-disable-next-line no-shadow
function () {
var _ref3 = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(monitoring) {
var message, url, res;
return _regeneratorRuntime().wrap(function _callee2$(_context2) {
while (1) {
switch (_context2.prev = _context2.next) {
case 0:
message = {
type: 'monitoring',
projectName: project.name,
project: project,
monitoring: monitoring,
startedAt: Date.now()
};
_context2.prev = 1;
url = monitoring.url;
if (url) {
_context2.next = 5;
break;
}
return _context2.abrupt("return", null);
case 5:
_context2.next = 7;
return (0, _axios["default"])({
url: url,
timeout: timeout
});
case 7:
res = _context2.sent;
message.time = Date.now() - message.startedAt;
message.res = res;
if (res.status >= 300) {
message.status = 'error';
} else if (message.time >= timeoutWarn) {
message.status = 'warning';
} else {
message.status = 'success';
}
_context2.next = 19;
break;
case 13:
_context2.prev = 13;
_context2.t0 = _context2["catch"](1);
message.status = 'error';
message.err = _context2.t0;
message.time = Date.now() - message.startedAt;
message.response = _context2.t0 ? _context2.t0.response : null;
case 19:
_this3.log.trace("[checkResourses] ".concat(message.status));
_context2.next = 22;
return _this3.sendNotification({
bot: bot,
message: message
})["catch"](function (err) {
_this3.log.error('[sendNotification]', err, message);
});
case 22:
return _context2.abrupt("return", message);
case 23:
case "end":
return _context2.stop();
}
}
}, _callee2, null, [[1, 13]]);
}));
return function (_x3) {
return _ref3.apply(this, arguments);
};
}(), {
concurrency: 1
}));
case 4:
case "end":
return _context3.stop();
}
}
}, _callee3);
}));
return function (_x2) {
return _ref2.apply(this, arguments);
};
}());
}
}, {
key: "runMonitoring",
value: function () {
var _runMonitoring = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4(bot) {
var _this4 = this;
var cronConfig;
return _regeneratorRuntime().wrap(function _callee4$(_context4) {
while (1) {
switch (_context4.prev = _context4.next) {
case 0:
cronConfig = (0, _get["default"])(this.config, 'cron', '* * * * *');
if (!Array.isArray(cronConfig)) cronConfig = [cronConfig];
_context4.next = 4;
return this.checkResourses(bot);
case 4:
this.crons = cronConfig.map(function (config) {
var _config$time = config.time,
time = _config$time === void 0 ? '* * * * *' : _config$time,
_config$timeZone = config.timeZone,
timeZone = _config$timeZone === void 0 ? 'Europe/Moscow' : _config$timeZone;
return {
time: time,
timeZone: timeZone,
cron: _nodeCron["default"].schedule(time, function () {
_this4.checkResourses(bot);
}, {
timeZone: timeZone
})
};
});
this.log.info("Cron: [".concat(this.crons.map(function (_ref4) {
var time = _ref4.time;
return time;
}).join(','), "]. Provider: ").concat(bot.provider));
case 6:
case "end":
return _context4.stop();
}
}
}, _callee4, this);
}));
function runMonitoring(_x4) {
return _runMonitoring.apply(this, arguments);
}
return runMonitoring;
}()
}, {
key: "runBot",
value: function () {
var _runBot = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5(bot) {
return _regeneratorRuntime().wrap(function _callee5$(_context5) {
while (1) {
switch (_context5.prev = _context5.next) {
case 0:
if (!(!this.config || (0, _isEmpty["default"])(this.config))) {
_context5.next = 2;
break;
}
return _context5.abrupt("return");
case 2:
this.runMonitoring(bot);
case 3:
case "end":
return _context5.stop();
}
}
}, _callee5, this);
}));
function runBot(_x5) {
return _runBot.apply(this, arguments);
}
return runBot;
}()
}]);
return NotifyPlugin;
}(_botsPlugin["default"]);
exports.NotifyPlugin = NotifyPlugin;
var _default = NotifyPlugin;
exports["default"] = _default;
//# sourceMappingURL=NotifyPlugin.js.map