Add support features for entity extraction. Closes #52. (#53)

src/lib/initDb.js

@@ -67,6 +67,9 @@ exports.init = function(nlcFile){
 												}
 											}
 										}
+										if (p.entityfunction) {
+											p.entityfunction = `${name}_${p.entityfunction}`;
+										}
 									}
 								}
 								dfnObj.parameters = cls.parameters;

src/lib/nlcconfig.js

@@ -8,6 +8,20 @@
 
 const nlcDb = require('./nlcDb');
 
+const NLCCONFIG_KEY = Symbol.for('hubot-ibmcloud-cognitive-lib-nlcconfig');
+
+var globalSymbols = Object.getOwnPropertySymbols(global);
+if (globalSymbols.indexOf(NLCCONFIG_KEY) < 0) {
+	global[NLCCONFIG_KEY] = {};
+}
+
+var singleton = {};
+Object.defineProperty(singleton, 'instance', {
+	get: function(){
+		return global[NLCCONFIG_KEY];
+	}
+});
+
 /**
  * Return array of class definitions for all classes (emit target, textfile, and parameters).
  *
@@ -55,3 +69,17 @@ exports.getAutoApprove = function(){
 exports.setAutoApprove = function(value){
 	return nlcDb.setAutoApprove(value);
 };
+
+/**
+ * Sets entity function definitions.
+ */
+exports.setGlobalEntityFunction = function(name, entityFunction){
+	singleton.instance[name] = entityFunction;
+};
+
+/**
+ * Returns entity function definitions.
+ */
+exports.getGlobalEntityFunction = function(name){
+	return singleton.instance[name];
+};

test/nlc.config.test.js

@@ -89,4 +89,20 @@ describe('Testing NLC Configuration', function() {
 			nlcconfig.setAutoApprove(true);
 		});
 	});
+
+	context('Verify entity function setter/getter', function() {
+
+		function testFunc(paramName, parameters) {
+			return ['value1', 'value2', 'value3'];
+		}
+
+		it('Verify entity function setter/getter correct value', function() {
+			nlcconfig.setGlobalEntityFunction('testfunc', testFunc);
+			let retFunc = nlcconfig.getGlobalEntityFunction('testfunc');
+			expect(typeof (retFunc)).to.eql('function');
+			let retFuncResult = testFunc(null, null);
+			expect(retFuncResult.length).to.eql(3);
+			expect(retFuncResult[0]).to.eql('value1');
+		});
+	});
 });