136 lines
3.6 KiB
JavaScript
136 lines
3.6 KiB
JavaScript
import 'babel/polyfill';
|
|
import API from './api';
|
|
import {EventEmitter} from 'events';
|
|
|
|
const DEFAULTS = {
|
|
update: {
|
|
offset: 0,
|
|
timeout: 0.5,
|
|
limit: 100
|
|
}
|
|
};
|
|
|
|
/**
|
|
* 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
|
|
*/
|
|
export default class Bot extends EventEmitter {
|
|
/**
|
|
* Create and connect to a new bot
|
|
* @param {object} options Bot properties.
|
|
*/
|
|
constructor(options = {update: {}}) {
|
|
super();
|
|
|
|
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 API(this.token);
|
|
|
|
this.msg = {};
|
|
|
|
// EventEmitter
|
|
this._events = {};
|
|
this._userEvents = [];
|
|
}
|
|
|
|
/**
|
|
* Gets information about the bot and then starts polling updates from API
|
|
* 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
|
|
* @return {promise} A promise which is resolved with the response of getMe
|
|
*/
|
|
start() {
|
|
let poll = function() {
|
|
this.api.getUpdates(this.update).then(response => {
|
|
setTimeout(poll, this.update.timeout * 1000);
|
|
|
|
const result = response.result;
|
|
if (!result.length) {
|
|
return;
|
|
}
|
|
|
|
if (!this.update.offset) {
|
|
const updateId = result[result.length - 1].update_id;
|
|
this.update.offset = updateId;
|
|
}
|
|
if (this.update) {
|
|
this.update.offset += 1;
|
|
}
|
|
|
|
result.forEach(res => {
|
|
let text = res.message.text;
|
|
if (text.startsWith('/')) {
|
|
// Commands are sent in /command@botusername format in groups
|
|
const regex = new RegExp(`@${this.info.username}$`);
|
|
text = text.replace(regex, '');
|
|
}
|
|
|
|
let ev = this._userEvents.find(({pattern}) => pattern.test(text));
|
|
ev.listener(res.message);
|
|
});
|
|
|
|
this.emit('update', result);
|
|
});
|
|
}.bind(this);
|
|
|
|
return this.api.getMe().then(response => {
|
|
this.info = response.result;
|
|
poll();
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 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
|
|
*/
|
|
get(pattern, listener) {
|
|
if (typeof pattern === 'string') {
|
|
pattern = new RegExp(`^${pattern}`);
|
|
}
|
|
|
|
this._userEvents.push({
|
|
pattern, listener
|
|
});
|
|
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Listens on a command
|
|
* @param {string} cmd 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
|
|
*/
|
|
command(cmd, listener) {
|
|
this._userEvents.push({
|
|
pattern: new RegExp(`/${cmd}`),
|
|
listener
|
|
});
|
|
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* 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
|
|
*/
|
|
send(message) {
|
|
return message.send(this);
|
|
}
|
|
}
|