From b08179f174934f8f1ee40ae6d77bb572396d7930 Mon Sep 17 00:00:00 2001
From: German Attanasio Ruiz <germanattanasio@gmail.com>
Date: Thu, 1 Oct 2015 15:12:35 -0400
Subject: [PATCH] Added retrieve and rank

---
 .gitignore                       |   6 +-
 services/retrieve_and_rank/v1.js | 121 +++++++++++++++++++++++++++++++
 2 files changed, 122 insertions(+), 5 deletions(-)

diff --git a/.gitignore b/.gitignore
index 15045d67e7..9166216883 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,8 +2,4 @@ node_modules/*
 npm-debug.log
 coverage
 .DS_Store
-test/resources/auth.js
-test/resources/mobydick.txt
-test/test.all-services.js
-test/resources/tts-output.ogg
-test/resources/face.jpg
+test/integration/*
diff --git a/services/retrieve_and_rank/v1.js b/services/retrieve_and_rank/v1.js
index f3737395cb..b9e4519166 100644
--- a/services/retrieve_and_rank/v1.js
+++ b/services/retrieve_and_rank/v1.js
@@ -21,6 +21,9 @@ var url            = require('url');
 var extend         = require('extend');
 var requestFactory = require('../../lib/requestwrapper');
 var solr           = require('solr-client');
+var pick           = require('object.pick');
+var omit           = require('object.omit');
+var isStream       = require('isstream');
 
 function RetrieveAndRank(options) {
   var serviceDefaults = {
@@ -32,6 +35,124 @@ function RetrieveAndRank(options) {
   this._options = extend(serviceDefaults, options);
 }
 
+/**
+ * Creates a ranker
+ */
+RetrieveAndRank.prototype.createRanker = function(params, callback) {
+  params = params || {};
+
+  if (!params || !params.training_data) {
+    callback(new Error('Missing required parameters: training_data'));
+    return;
+  }
+  if ((typeof params.training_data !== 'string') || (!isStream(params.training_data))) {
+    callback(new Error('training_data needs to be a String or Stream'));
+    return;
+  }
+
+  var parameters = {
+    options: {
+      url: '/v1/rankers',
+      method: 'POST',
+      json: true,
+      formData: {
+        training_data: params.training_data,
+        training_metadata: JSON.stringify(omit(params, ['training_data']))
+      }
+    },
+    defaultOptions: this._options
+  };
+  return requestFactory(parameters, callback);
+};
+
+/**
+ * Returns the ranked candidates
+ */
+RetrieveAndRank.prototype.rank = function(params, callback) {
+  params = params || {};
+
+  if (!params || !params.answer_data) {
+    callback(new Error('Missing required parameters: answer_data'));
+    return;
+  }
+  if ((typeof params.answer_data !== 'string') || (!isStream(params.answer_data))) {
+    callback(new Error('answer_data needs to be a String or Stream'));
+    return;
+  }
+
+  var parameters = {
+    options: {
+      url: '/v1/rankers/{ranker_id}/rank',
+      method: 'POST',
+      json: true,
+      formData: {
+        answer_data: params.answer_data,
+        answer_metadata: JSON.stringify(omit(params, ['answer_data']))
+      },
+      path: pick(params, ['ranker_id'])
+    },
+    requiredParams: ['ranker_id'],
+    defaultOptions: this._options
+  };
+  return requestFactory(parameters, callback);
+};
+
+/**
+ * Returns the training status of the ranker
+ */
+RetrieveAndRank.prototype.rankerStatus = function(params, callback) {
+  params = params || {};
+
+  var parameters = {
+    options: {
+      url: '/v1/rankers/{ranker_id}',
+      method: 'GET',
+      json: true,
+      path: params
+    },
+    requiredParams: ['ranker_id'],
+    defaultOptions: this._options
+  };
+
+  return requestFactory(parameters, callback);
+};
+
+/**
+ * Retrieves the list of rankers for the user
+ */
+RetrieveAndRank.prototype.listRankers = function(params, callback) {
+  var parameters = {
+    options: {
+      url: '/v1/rankers',
+      method: 'GET',
+      json: true
+    },
+    defaultOptions: this._options
+  };
+
+  return requestFactory(parameters, callback);
+};
+
+/**
+ * Deletes a ranker
+ */
+RetrieveAndRank.prototype.deleteRanker = function(params, callback) {
+  params = params || {};
+
+  var parameters = {
+    options: {
+      url: '/v1/rakers/{ranker_id}',
+      method: 'DELETE',
+      path: params,
+      json: true
+    },
+    requiredParams: ['ranker_id'],
+    defaultOptions: this._options
+  };
+
+  return requestFactory(parameters, callback);
+};
+
 // Solr cluster lifecycle operations
 
 /**