translate: allow setting options.format

stephenplusplus · stephenplusplus · commit aabd9b8820df · 2016-09-29T11:21:40.000-04:00
diff --git a/packages/translate/package.json b/packages/translate/package.json
@@ -54,6 +54,7 @@
     "arrify": "^1.0.0",
     "extend": "^3.0.0",
     "is": "^3.0.1",
+    "is-html": "^1.0.0",
     "propprop": "^0.3.0"
   },
   "devDependencies": {
diff --git a/packages/translate/src/index.js b/packages/translate/src/index.js
@@ -24,6 +24,7 @@ var arrify = require('arrify');
 var common = require('@google-cloud/common');
 var extend = require('extend');
 var is = require('is');
+var isHtml = require('is-html');
 var prop = require('propprop');
 
 var PKG = require('../package.json');
@@ -268,6 +269,7 @@ Translate.prototype.getLanguages = function(target, callback) {
  *     target ISO 639-1 language code to translate the source input to. (e.g.
  *     `en` for English). If an object, you may also specify the source
  *     language.
+ * @param {string} options.format - Set the text's format as `html` or `text`.
  * @param {string} options.from - The ISO 639-1 language code the source input
  *     is written in.
  * @param {string} options.to - The ISO 639-1 language code to translate the
@@ -323,8 +325,11 @@ Translate.prototype.getLanguages = function(target, callback) {
  * });
  */
 Translate.prototype.translate = function(input, options, callback) {
+  input = arrify(input);
+
   var query = {
-    q: arrify(input)
+    q: input,
+    format: options.format || (isHtml(input[0]) ? 'html' : 'text')
   };
 
   if (is.string(options)) {
diff --git a/packages/translate/test/index.js b/packages/translate/test/index.js
@@ -266,6 +266,7 @@ describe('Translate', function() {
 
   describe('translate', function() {
     var INPUT = 'Hello';
+    var INPUT_HTML = '<html><body>Hello</body></html>';
     var SOURCE_LANG_CODE = 'en';
     var TARGET_LANG_CODE = 'es';
 
@@ -306,12 +307,46 @@ describe('Translate', function() {
       });
     });
 
+    describe('options.format', function() {
+      it('should default to text', function(done) {
+        translate.request = function(reqOpts) {
+          assert.strictEqual(reqOpts.qs.format, 'text');
+          done();
+        };
+
+        translate.translate(INPUT, OPTIONS, assert.ifError);
+      });
+
+      it('should recognize HTML', function(done) {
+        translate.request = function(reqOpts) {
+          assert.strictEqual(reqOpts.qs.format, 'html');
+          done();
+        };
+
+        translate.translate(INPUT_HTML, OPTIONS, assert.ifError);
+      });
+
+      it('should allow overriding the format', function(done) {
+        var options = extend({}, OPTIONS, {
+          format: 'custom format'
+        });
+
+        translate.request = function(reqOpts) {
+          assert.strictEqual(reqOpts.qs.format, options.format);
+          done();
+        };
+
+        translate.translate(INPUT_HTML, options, assert.ifError);
+      });
+    });
+
     it('should make the correct API request', function(done) {
       translate.request = function(reqOpts) {
         assert.strictEqual(reqOpts.uri, '');
         assert.strictEqual(reqOpts.useQuerystring, true);
         assert.deepEqual(reqOpts.qs, {
           q: [INPUT],
+          format: 'text',
           source: SOURCE_LANG_CODE,
           target: TARGET_LANG_CODE
         });
