plugin.js

/**
 * We.js plugin file, main project file
 */
var watson = require('watson-developer-cloud'),
  fs = require('fs'),
  path = require('path');

module.exports = function loadPlugin (projectPath, Plugin) {
  var plugin = new Plugin(__dirname);

  plugin.nlc = watson.natural_language_classifier({
    "url": "https://gateway.watsonplatform.net/natural-language-classifier/api",
    "password": "zyRRvZSIFf7T",
    "username": "445ffcd9-85be-43f6-a000-7248278c87f3",
    version: 'v1'
  });

  /**
   * Create one classifier abd save in configurations
   */
  plugin.createClassifier = function createClassifier (we, done) {

  	var params = {
  	  language: 'pt-br',
  	  name: 'roupas',
  	  training_data: fs.createReadStream(path.resolve('./data/roupas_train.cvs'))
  	};

  	plugin.nlc.create(params, function (err, response) {
  	  if (err) return done(err);

      we.log.info('Watson response:', response);
      var data = 'module.exports.watsonClassifier = ' + JSON.stringify(response);

      fs.writeFile( path.resolve('./config/watsonClassifier.js'), data , done)
    }); 
  }

  /**
   * Preload one classifier
   */
  plugin.preloadClassifier = function preloadClassifier (we, next) {
    we.db.models.classifier.findOne({
      where: { name: 'roupas' }
    })
    .then(function (r) {
      plugin.watsonClassifier = r;
      next();
    })
    .catch(next)
  }

  /**
   * Execute one question in watson natural language
   */
  plugin.question = function question (we, text, done) {
    plugin.nlc.classify({
      text: text,
      classifier_id: we.config.watsonClassifier.classifier_id
    }, done);
  }

  return plugin;
};