node-red-contrib-chatbot
Version:
REDBot a Chat bot for a full featured chat bot for Telegram, Facebook Messenger and Slack. Almost no coding skills required
88 lines (76 loc) • 2.99 kB
JavaScript
const _ = require('underscore');
const utils = require('../lib/helpers/utils');
const RegisterType = require('../lib/node-installer');
const GlobalContextHelper = require('../lib/helpers/global-context-helper');
const isValidRegex = str => {
let isValid = true;
try {
new RegExp(str);
} catch(e) {
isValid = false;
}
return isValid;
};
module.exports = function(RED) {
const registerType = RegisterType(RED);
const globalContextHelper = GlobalContextHelper(RED);
function ChatBotNLPEntity(config) {
RED.nodes.createNode(this, config);
var node = this;
globalContextHelper.init(this.context().global);
this.entities = config.entities;
this.name = config.name;
this.language = config.language;
this.entityType = config.entityType;
this.regex = config.regex;
this.on('input', function(msg, send, done) {
// send/done compatibility for node-red < 1.0
send = send || function() { node.send.apply(node, arguments) };
done = done || function(error) { node.error.call(node, error, msg) };
const entities = utils.extractValue('arrayOfEntities', 'entities', node, msg);
const name = utils.extractValue('string', 'name', node, msg, false);
const language = utils.extractValue('string', 'language', node, msg, false);
const entityType = utils.extractValue('string', 'entityType', node, msg, false);
msg.payload = _.isObject(msg.payload) ? msg.payload : {};
// make sure we replace the entities[] in the payload
if (!_.isObject(msg.payload.entities) || _.isArray(msg.payload.entities)) {
msg.payload.entities = {};
}
if (!_.isObject(msg.payload.entities[language])) {
msg.payload.entities[language] = {};
}
if (!_.isObject(msg.payload.entities[language][name])) {
msg.payload.entities[language][name] = {};
}
// collect the lexicon of the node and mix with the one of the incoming payload
if (entityType === 'regex') {
if (isValidRegex(node.regex)) {
msg.payload.entities[language][name] = {
type: 'regex',
regex: node.regex
};
} else {
done(`Invalid NLP.js regex for entity "${name}"`);
return;
}
} else if (_.isEmpty(entityType) || entityType === 'enum') {
if (_.isArray(entities) && !_.isEmpty(entities)) {
msg.payload.entities[language][name] = {
type: 'enum',
entities: [
...(_.isArray(msg.payload.entities[language][name].entities) && msg.payload.entities[language][name].type === 'enum' ?
msg.payload.entities[language][name].entities : []),
...entities
],
};
}
} else {
done(`Invalid NLP.js entity type: ${entityType} for entity "${name}"`);
return;
}
send(msg);
done();
});
}
registerType('chatbot-nlpjs-entity', ChatBotNLPEntity);
};