node-telegram-api/build/index.js
2015-11-08 06:52:06 +03:30

373 lines
12 KiB
JavaScript

'use strict';
Object.defineProperty(exports, '__esModule', {
value: true
});
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; }; })();
var _get = function get(_x2, _x3, _x4) { var _again = true; _function: while (_again) { var object = _x2, property = _x3, receiver = _x4; desc = parent = getter = undefined; _again = false; if (object === null) object = Function.prototype; var desc = Object.getOwnPropertyDescriptor(object, property); if (desc === undefined) { var parent = Object.getPrototypeOf(object); if (parent === null) { return undefined; } else { _x2 = parent; _x3 = property; _x4 = receiver; _again = true; continue _function; } } else if ('value' in desc) { return desc.value; } else { var getter = desc.get; if (getter === undefined) { return undefined; } return getter.call(receiver); } } };
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { 'default': obj }; }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError('Cannot call a class as a function'); } }
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) subClass.__proto__ = superClass; }
var _functionsApi = require('./functions/api');
var _functionsApi2 = _interopRequireDefault(_functionsApi);
var _functionsWebhook = require('./functions/webhook');
var _functionsWebhook2 = _interopRequireDefault(_functionsWebhook);
var _functionsPoll = require('./functions/poll');
var _functionsPoll2 = _interopRequireDefault(_functionsPoll);
var _functionsArgumentParser = require('./functions/argument-parser');
var _functionsArgumentParser2 = _interopRequireDefault(_functionsArgumentParser);
var _events = require('events');
var _typesMessage = require('./types/Message');
var _typesMessage2 = _interopRequireDefault(_typesMessage);
var DEFAULTS = {
update: {
offset: 0,
timeout: 20,
limit: 100
}
};
var REQUIRED = 0;
var OPTIONAL = 1;
/**
* Bot class used to connect to a new bot
* Bots have an api property which gives access to all Telegram API methods,
* see API class
*/
var Bot = (function (_EventEmitter) {
/**
* Create and connect to a new bot
* @param {object} options Bot properties.
*/
function Bot() {
var options = arguments[0] === undefined ? { update: {} } : arguments[0];
_classCallCheck(this, Bot);
_get(Object.getPrototypeOf(Bot.prototype), 'constructor', this).call(this);
if (!options.token) {
throw new Error('Token cannot be empty');
}
this.token = options.token;
this.update = Object.assign(options.update || {}, DEFAULTS.update);
this.api = new _functionsApi2['default'](this.token);
this.msg = {};
// EventEmitter
this._events = {};
this._userEvents = [];
this.setMaxListeners(100);
}
_inherits(Bot, _EventEmitter);
_createClass(Bot, [{
key: 'start',
/**
* Gets information about the bot and then
* 1) starts polling updates from API
* 2) sets a webhook as defined by the first parameter and listens for updates
* Emits an `update` event after polling with the response from server
* Returns a promise which is resolved after the bot information is received
* and set to it's `info` property i.e. bot.info
*
* @param {object} hook An object containg options passed to webhook
* properties:
* - url: HTTPS url to listen on POST requests coming
* from the Telegram API
* - port: the port to listen to, defaults to 443
* - server: An object passed to https.createServer
*
* @return {promise} A promise which is resolved with the response of getMe
*/
value: function start(hook) {
var _this = this;
if (hook) {
return (0, _functionsWebhook2['default'])(hook, this);
}
return this.api.getMe().then(function (response) {
_this.info = response.result;
_this.on('update', _this._update);
if (hook) {
return (0, _functionsWebhook2['default'])(hook, _this);
} else {
return (0, _functionsPoll2['default'])(_this);
}
});
}
}, {
key: 'get',
/**
* Listens on specific message matching the pattern which can be an string
* or a regexp.
* @param {string/regex} pattern
* @param {function} listener function to call when a message matching the
* pattern is found, gets the Update
* In case of string, the message should start
* with the string i.e. /^yourString/
* @return {object} returns the bot object
*/
value: function get(pattern, listener) {
if (typeof pattern === 'string') {
pattern = new RegExp('^' + pattern);
}
this._userEvents.push({
pattern: pattern, listener: listener
});
return this;
}
}, {
key: 'command',
/**
* Listens on a command
* @param {string} command the command string, should not include slash (/)
* @param {function} listener function to call when the command is received,
* gets the update
* @return {object} returns the bot object
*/
value: function command(_command, listener) {
var regex = /[^\s]+/;
var cmd = _command.match(regex)[0].trim();
this._userEvents.push({
pattern: new RegExp('^/' + cmd),
parse: _functionsArgumentParser2['default'].bind(null, _command),
listener: listener
});
return this;
}
}, {
key: 'send',
/**
* Sends the message provided
* @param {object} message The message to send. Gets it's send method called
* @return {unknown} returns the result of calling message's send method
*/
value: function send(message) {
return message.send(this)['catch'](console.error);
}
}, {
key: 'stop',
/**
* Stops the bot, deattaching all listeners and polling
*/
value: function stop() {
this._stop = true;
if (this._webhookServer) {
this._webhookServer.close();
}
this.removeListener('update', this._update);
this._events = {};
}
}, {
key: '_update',
/**
* The internal update event listener, used to parse messages and fire
* command/get events - YOU SHOULD NOT USE THIS
*
* @param {object} update
*/
value: function _update(update) {
var _this2 = this;
if (!this.update.offset) {
var updateId = update[update.length - 1].update_id;
this.update.offset = updateId;
}
if (this.update) {
this.update.offset += 1;
}
update.forEach(function (res) {
var marked3$0 = [getAnswer].map(regeneratorRuntime.mark);
var text = res.message.text;
if (!text) return;
var selfUsername = '@' + _this2.info.username;
if (text.startsWith('/') && text.indexOf(selfUsername) > -1) {
// Commands are sent in /command@thisusername format in groups
var regex = new RegExp('(/.*)@' + _this2.info.username);
text = text.replace(regex, '$1');
res.message.text = text;
}
var ev = _this2._userEvents.find(function (_ref) {
var pattern = _ref.pattern;
return pattern.test(text);
});
if (!ev) {
_this2.emit('command-notfound', res.message);
return;
}
if (!ev.parse) {
ev.listener(res.message);
return;
}
var _ev$parse = ev.parse(res.message.text);
var params = _ev$parse.params;
var args = _ev$parse.args;
res.message.args = args;
var requiredParams = Object.keys(params).filter(function (param) {
return params[param] === REQUIRED && !args[param];
});
if (!requiredParams.length) {
ev.listener(res.message);
return;
}
var bot = _this2;
function getAnswer() {
var _iteratorNormalCompletion, _didIteratorError, _iteratorError, _loop, _iterator, _step;
return regeneratorRuntime.wrap(function getAnswer$(context$4$0) {
var _this3 = this;
while (1) switch (context$4$0.prev = context$4$0.next) {
case 0:
_iteratorNormalCompletion = true;
_didIteratorError = false;
_iteratorError = undefined;
context$4$0.prev = 3;
_loop = regeneratorRuntime.mark(function callee$4$0() {
var param, msg;
return regeneratorRuntime.wrap(function callee$4$0$(context$5$0) {
while (1) switch (context$5$0.prev = context$5$0.next) {
case 0:
param = _step.value;
msg = new _typesMessage2['default']().to(res.message.chat.id).text('Enter value for ' + param);
context$5$0.next = 4;
return bot.send(msg).then(function (answer) {
args[param] = answer.text;
});
case 4:
case 'end':
return context$5$0.stop();
}
}, callee$4$0, _this3);
});
_iterator = requiredParams[Symbol.iterator]();
case 6:
if (_iteratorNormalCompletion = (_step = _iterator.next()).done) {
context$4$0.next = 11;
break;
}
return context$4$0.delegateYield(_loop(), 't0', 8);
case 8:
_iteratorNormalCompletion = true;
context$4$0.next = 6;
break;
case 11:
context$4$0.next = 17;
break;
case 13:
context$4$0.prev = 13;
context$4$0.t1 = context$4$0['catch'](3);
_didIteratorError = true;
_iteratorError = context$4$0.t1;
case 17:
context$4$0.prev = 17;
context$4$0.prev = 18;
if (!_iteratorNormalCompletion && _iterator['return']) {
_iterator['return']();
}
case 20:
context$4$0.prev = 20;
if (!_didIteratorError) {
context$4$0.next = 23;
break;
}
throw _iteratorError;
case 23:
return context$4$0.finish(20);
case 24:
return context$4$0.finish(17);
case 25:
case 'end':
return context$4$0.stop();
}
}, marked3$0[0], this, [[3, 13, 17, 25], [18,, 20, 24]]);
}
var iterator = getAnswer();
(function loop() {
var next = iterator.next();
if (next.done) {
ev.listener(res.message);
return;
}
next.value.then(loop);
})();
});
}
}]);
return Bot;
})(_events.EventEmitter);
exports['default'] = Bot;
module.exports = exports['default'];