gitea
/
node-telegram-bot-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Separated WebHook

Browse Source
experimental
yago 11 years ago
parent 3573d0949d
commit 923e65c789
3 changed files with 86 additions and 75 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 85
      src/telegram.js
  2. 66
      src/telegramWebHook.js
  3. 10
      test/index.js

85
src/telegram.js
Unescape View File

@ -1,10 +1,9 @@
var EventEmitter = require('events').EventEmitter;
var TelegramBotWebHook = require('./telegramWebHook');
var debug = require('debug')('node-telegram-bot-api');
var EventEmitter = require('events').EventEmitter;
var Promise = require("bluebird");
var request = require("request");
var stream = require('stream');
var https = require('https');
var http = require('http');
var util = require('util');
var mime = require('mime');
var path = require('path');
@ -42,70 +41,28 @@ var TelegramBot = function (token, options) {
}
if (options.webHook) {
var port = options.webHook.port || 8443;
var key = options.webHook.key;
var cert = options.webHook.cert;
var host = options.webHook.host;
this._configureWebHook(port, host, key, cert);
var binded = this._processUpdate.bind(this);
this._WebHook = new TelegramBotWebHook(token, options.webHook, binded);
}
};
util.inherits(TelegramBot, EventEmitter);
TelegramBot.prototype._configureWebHook = function (port, host, key, cert) {
var binded = this._requestListener.bind(this);
if (key && cert) { // HTTPS Server
debug('HTTPS WebHook enabled');
var options = {
key: fs.readFileSync(key),
cert: fs.readFileSync(cert)
};
this._webServer = https.createServer(options, binded);
} else {
debug('HTTP WebHook enabled');
this._webServer = http.createServer(binded);
}
this._webServer.listen(port, host, function () {
debug("WebHook listening on port %s", port);
});
};
TelegramBot.prototype._requestListener = function (req, res) {
TelegramBot.prototype._polling = function (timeout) {
var self = this;
var regex = new RegExp(this.token);
debug('WebHook request URL:', req.url);
debug('WebHook request headers: %j', req.headers);
// If there isn't token on URL
if (!regex.test(req.url)) {
debug('WebHook request unauthorized');
res.statusCode = 401;
res.end();
} else if (req.method === 'POST') {
var fullBody = '';
req.on('data', function (chunk) {
fullBody += chunk.toString();
});
req.on('end', function () {
try {
debug('WebHook request fullBody', fullBody);
var data = JSON.parse(fullBody);
self._processUpdate(data);
} catch (error) {
debug(error);
}
res.end('OK');
});
} else { // Authorized but not a POST
debug('WebHook request isn\'t a POST');
res.statusCode = 418; // I'm a teabot!
res.end();
}
this.getUpdates(timeout).then(function (data) {
self._processUpdates(data);
self._polling(timeout);
}).catch(function (err) {
debug('polling error: %j', err);
// Wait for 2 seconds before retry
setTimeout(self._polling.bind(self), 2000, timeout);
});
};
TelegramBot.prototype._processUpdate = function (update) {
debug('Process Update', update);
debug('Process Update message', update.message);
if (update.message) {
this.emit('message', update.message);
}
@ -117,18 +74,6 @@ TelegramBot.prototype._processUpdates = function (updates) {
}
};
TelegramBot.prototype._polling = function (timeout) {
var self = this;
this.getUpdates(timeout).then(function (data) {
self._processUpdates(data);
self._polling(timeout);
}).catch(function (err) {
debug('polling error: %j', err);
// Wait for 2 seconds before retry
setTimeout(self._polling.bind(self), 2000, timeout);
});
};
TelegramBot.prototype._request = function (path, options) {
if (!this.token) {
throw new Error('Telegram Bot Token not provided!');

66
src/telegramWebHook.js
Unescape View File

@ -0,0 +1,66 @@
var debug = require('debug')('node-telegram-bot-api');
var https = require('https');
var http = require('http');
var util = require('util');
var fs = require('fs');
var TelegramBotWebHook = function (token, options, callback) {
this.token = token;
this.callback = callback;
if (typeof options === 'boolean') {
options = {};
}
options.port = options.port || 8443;
var binded = this._requestListener.bind(this);
if (options.key && options.cert) { // HTTPS Server
debug('HTTPS WebHook enabled');
var opts = {
key: fs.readFileSync(options.key),
cert: fs.readFileSync(options.cert)
};
this._webServer = https.createServer(opts, binded);
} else {
debug('HTTP WebHook enabled');
this._webServer = http.createServer(binded);
}
this._webServer.listen(options.port, options.host, function () {
debug("WebHook listening on port %s", options.port);
});
};
TelegramBotWebHook.prototype._requestListener = function (req, res) {
var self = this;
var regex = new RegExp(this.token);
debug('WebHook request URL:', req.url);
debug('WebHook request headers: %j', req.headers);
// If there isn't token on URL
if (!regex.test(req.url)) {
debug('WebHook request unauthorized');
res.statusCode = 401;
res.end();
} else if (req.method === 'POST') {
var fullBody = '';
req.on('data', function (chunk) {
fullBody += chunk.toString();
});
req.on('end', function () {
try {
debug('WebHook request fullBody', fullBody);
var data = JSON.parse(fullBody);
self.callback(data);
} catch (error) {
debug(error);
}
res.end('OK');
});
} else { // Authorized but not a POST
debug('WebHook request isn\'t a POST');
res.statusCode = 418; // I'm a teabot!
res.end();
}
};
module.exports = TelegramBotWebHook;

10
test/index.js
Unescape View File

@ -60,7 +60,7 @@ describe('Telegram', function () {
method: 'POST'
}, function (error, response, body) {
response.statusCode.should.not.be.equal(200);
bot._webServer.close();
bot._WebHook._webServer.close();
done();
});
});
@ -72,7 +72,7 @@ describe('Telegram', function () {
method: 'GET'
}, function (error, response, body) {
response.statusCode.should.not.be.equal(200);
bot._webServer.close();
bot._WebHook._webServer.close();
done();
});
});
@ -80,7 +80,7 @@ describe('Telegram', function () {
it('should emit a `message` on HTTP WebHook', function (done) {
var bot = new Telegram(TOKEN, {webHook: true});
bot.on('message', function (msg) {
bot._webServer.close();
bot._WebHook._webServer.close();
done();
});
var url = 'http://localhost:8443/bot'+TOKEN;
@ -91,7 +91,7 @@ describe('Telegram', function () {
headers: {
"content-type": "application/json",
},
body: JSON.stringify({update_id: 0, message: {text: 'test'}})
body: {update_id: 0, message: {text: 'test'}}
});
});
@ -105,7 +105,7 @@ describe('Telegram', function () {
};
var bot = new Telegram(TOKEN, opts);
bot.on('message', function (msg) {
bot._webServer.close();
bot._WebHook._webServer.close();
done();
});
var url = 'https://localhost:8443/bot'+TOKEN;

Write Preview
Loading…
Cancel
Save