Add style to translate options

Author: ikhimenko

src/main/java/com/translated/lara/translator/DocumentOptions.java

@@ -4,6 +4,7 @@
 
 public class DocumentOptions {
     private String[] adaptTo = null;
+    private TranslationStyle style = null;
 
     public String[] getAdaptTo() {
         return adaptTo;
@@ -18,4 +19,13 @@ public DocumentOptions setAdaptTo(String... adaptTo) {
         this.adaptTo = adaptTo;
         return this;
     }
+
+    public TranslationStyle getStyle() {
+        return style;
+    }
+
+    public DocumentOptions setStyle(TranslationStyle style) {
+        this.style = style;
+        return this;
+    }
 }

src/main/java/com/translated/lara/translator/DocumentTranslateOptions.java

@@ -10,6 +10,7 @@ public class DocumentTranslateOptions {
     private String[] glossaries = null;
     private String outputFormat = null;
     private Boolean noTrace = null;
+    private TranslationStyle style = null;
 
     public String[] getAdaptTo() {
         return adaptTo;
@@ -57,11 +58,21 @@ public DocumentTranslateOptions setNoTrace(Boolean noTrace) {
         return this;
     }
 
+    public TranslationStyle getStyle() {
+        return style;
+    }
+
+    public DocumentTranslateOptions setStyle(TranslationStyle style) {
+        this.style = style;
+        return this;
+    }
+
     public HttpParams<Object> toParams() {
         HttpParams<Object> params = new HttpParams<>();
         params.set("adapt_to", adaptTo);
         params.set("glossaries", glossaries);
         params.set("output_format", outputFormat);
+        params.set("style", TranslationStyle.toString(style));
 
         return params;
     }

src/main/java/com/translated/lara/translator/DocumentUploadOptions.java

@@ -9,6 +9,7 @@ public class DocumentUploadOptions {
     private String[] adaptTo = null;
     private String[] glossaries = null;
     private Boolean noTrace = null;
+    private TranslationStyle style = null;
 
     public String[] getAdaptTo() {
         return adaptTo;
@@ -47,10 +48,20 @@ public DocumentUploadOptions setNoTrace(Boolean noTrace) {
         return this;
     }
 
+    public TranslationStyle getStyle() {
+        return style;
+    }
+
+    public DocumentUploadOptions setStyle(TranslationStyle style) {
+        this.style = style;
+        return this;
+    }
+
     public HttpParams<Object> toParams() {
         HttpParams<Object> params = new HttpParams<>();
         params.set("adapt_to", adaptTo);
         params.set("glossaries", glossaries);
+        params.set("style", TranslationStyle.toString(style));
 
         return params;
     }

src/main/java/com/translated/lara/translator/Documents.java

@@ -119,6 +119,7 @@ public InputStream translate(File input, String source, String target, DocumentT
         if (options != null && options.getAdaptTo() != null) uploadOptions.setAdaptTo(options.getAdaptTo());
         if (options != null && options.getGlossaries() != null) uploadOptions.setGlossaries(options.getGlossaries());
         if (options != null && options.getNoTrace()) uploadOptions.setNoTrace(true);
+        if (options != null && options.getStyle() != null) uploadOptions.setStyle(options.getStyle());
 
         Document document = upload(input, source, target, uploadOptions);
         document = pollDocumentUntilCompleted(document);

src/main/java/com/translated/lara/translator/TranslateOptions.java

@@ -28,6 +28,7 @@ public enum UseCache {
     private Integer cacheTTL = null;
     private Boolean noTrace = null;
     private Boolean verbose = null;
+    private TranslationStyle style = null;
     private Map<String, String> headers = null;
 
     public String getSourceHint() {
@@ -162,6 +163,15 @@ public TranslateOptions setHeaders(Map<String, String> headers) {
         return this;
     }
 
+    public TranslationStyle getStyle() {
+        return style;
+    }
+
+    public TranslateOptions setStyle(TranslationStyle style) {
+        this.style = style;
+        return this;
+    }
+
     public HttpParams<Object> toParams() {
         HttpParams<Object> params = new HttpParams<>();
         params.set("source_hint", sourceHint);
@@ -175,6 +185,7 @@ public HttpParams<Object> toParams() {
         params.set("use_cache", toString(useCache));
         params.set("cache_ttl", cacheTTL);
         params.set("verbose", verbose);
+        params.set("style", TranslationStyle.toString(style));
 
         return params;
     }

src/main/java/com/translated/lara/translator/TranslationStyle.java

@@ -0,0 +1,27 @@
+package com.translated.lara.translator;
+
+/**
+ * Translation style constants
+ */
+public enum TranslationStyle {
+    
+    FAITHFUL("faithful"),
+    FLUID("fluid"),
+    CREATIVE("creative");
+    
+    private final String value;
+    
+    TranslationStyle(String value) {
+        this.value = value;
+    }
+    
+    @Override
+    public String toString() {
+        return value;
+    }
+
+    public static String toString(TranslationStyle style) {
+        return style != null ? style.toString() : null;
+    }
+
+}