node-nlp
Version:
Library for NLU (Natural Language Understanding) done in Node.js
216 lines (215 loc) • 6.83 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
var Message_1 = require("../Message");
var utils = require("../utils");
var readline = require("readline");
var async = require("async");
var ConsoleConnector = (function () {
function ConsoleConnector() {
this.replyCnt = 0;
}
ConsoleConnector.prototype.listen = function () {
var _this = this;
this.rl = readline.createInterface({ input: process.stdin, output: process.stdout, terminal: false });
this.rl.on('line', function (line) {
_this.replyCnt = 0;
line = line || '';
if (line.toLowerCase() == 'quit') {
_this.rl.close();
process.exit();
}
else {
_this.processMessage(line);
}
});
return this;
};
ConsoleConnector.prototype.processMessage = function (line) {
if (this.onEventHandler) {
var msg = new Message_1.Message()
.address({
channelId: 'console',
user: { id: 'user', name: 'User1' },
bot: { id: 'bot', name: 'Bot' },
conversation: { id: 'Convo1' }
})
.timestamp()
.text(line);
this.onEventHandler([msg.toMessage()]);
}
return this;
};
ConsoleConnector.prototype.processEvent = function (event) {
if (this.onEventHandler) {
this.onEventHandler([event]);
}
return this;
};
ConsoleConnector.prototype.onEvent = function (handler) {
this.onEventHandler = handler;
};
ConsoleConnector.prototype.onInvoke = function (handler) {
this.onInvokeHandler = handler;
};
ConsoleConnector.prototype.send = function (messages, done) {
var _this = this;
var addresses = [];
async.forEachOfSeries(messages, function (msg, idx, cb) {
try {
if (msg.type == 'delay') {
setTimeout(cb, msg.value);
}
else if (msg.type == 'message') {
if (_this.replyCnt++ > 0) {
console.log();
}
if (msg.text) {
log(msg.text);
}
if (msg.attachments && msg.attachments.length > 0) {
for (var j = 0; j < msg.attachments.length; j++) {
if (j > 0) {
console.log();
}
renderAttachment(msg.attachments[j]);
}
}
var adr = utils.clone(msg.address);
adr.id = idx.toString();
addresses.push(adr);
cb(null);
}
else {
cb(null);
}
}
catch (e) {
cb(e);
}
}, function (err) { return done(err, !err ? addresses : null); });
};
ConsoleConnector.prototype.startConversation = function (address, cb) {
var adr = utils.clone(address);
adr.conversation = { id: 'Convo1' };
cb(null, adr);
};
return ConsoleConnector;
}());
exports.ConsoleConnector = ConsoleConnector;
function renderAttachment(a) {
switch (a.contentType) {
case 'application/vnd.microsoft.card.hero':
case 'application/vnd.microsoft.card.thumbnail':
var tc = a.content;
if (tc.title) {
if (tc.title.length <= 40) {
line('=', 60, tc.title);
}
else {
line('=', 60);
wrap(tc.title, 60, 3);
}
}
if (tc.subtitle) {
wrap(tc.subtitle, 60, 3);
}
if (tc.text) {
wrap(tc.text, 60, 3);
}
renderImages(tc.images);
renderButtons(tc.buttons);
break;
case 'application/vnd.microsoft.card.signin':
case 'application/vnd.microsoft.card.receipt':
default:
line('.', 60, a.contentType);
if (a.contentUrl) {
wrap(a.contentUrl, 60, 3);
}
else {
log(JSON.stringify(a.content));
}
break;
}
}
function renderImages(images) {
if (images && images.length) {
line('.', 60, 'images');
var bullet = images.length > 1 ? '* ' : '';
for (var i = 0; i < images.length; i++) {
var img = images[i];
if (img.alt) {
wrap(bullet + img.alt + ': ' + img.url, 60, 3);
}
else {
wrap(bullet + img.url, 60, 3);
}
}
}
}
function renderButtons(actions) {
if (actions && actions.length) {
line('.', 60, 'buttons');
var bullet = actions.length > 1 ? '* ' : '';
for (var i = 0; i < actions.length; i++) {
var a = actions[i];
if (a.title == a.value) {
wrap(bullet + a.title, 60, 3);
}
else {
wrap(bullet + a.title + ' [' + a.value + ']', 60, 3);
}
}
}
}
function line(char, length, title) {
if (title) {
var txt = repeat(char, 2);
txt += '[' + title + ']';
if (length > txt.length) {
txt += repeat(char, length - txt.length);
}
log(txt);
}
else {
log(repeat(char, length));
}
}
function wrap(text, length, indent) {
if (indent === void 0) { indent = 0; }
var buffer = '';
var pad = indent ? repeat(' ', indent) : '';
var tokens = text.split(' ');
length -= pad.length;
for (var i = 0; i < tokens.length; i++) {
var t = tokens[i];
if (buffer.length) {
if ((buffer.length + 1 + t.length) > length) {
log(pad + buffer);
buffer = t;
}
else {
buffer += ' ' + t;
}
}
else if (t.length < length) {
buffer = t;
}
else {
log(pad + t);
}
}
if (buffer.length) {
log(pad + buffer);
}
}
function repeat(char, length) {
var txt = '';
for (var i = 0; i < length; i++) {
txt += char;
}
return txt;
}
function log(text) {
console.log(text);
}