Feature: add password and extraction parameters for document translation

Author: ikhimenko

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

@@ -0,0 +1,16 @@
+package com.translated.lara.translator;
+
+import com.translated.lara.net.HttpParams;
+
+/**
+ * Abstract base class for extraction parameters by file format.
+ */
+public abstract class DocumentExtractionParams {
+
+    /**
+     * Convert to HTTP parameters for API requests.
+     * @return HttpParams containing the extraction parameters
+     */
+    public abstract HttpParams<Object> toParams();
+}
+

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

@@ -11,6 +11,8 @@ public class DocumentTranslateOptions {
     private String outputFormat = null;
     private Boolean noTrace = null;
     private TranslationStyle style = null;
+    private String password = null;
+    private DocumentExtractionParams extractionParams = null;
 
     public String[] getAdaptTo() {
         return adaptTo;
@@ -67,12 +69,31 @@ public DocumentTranslateOptions setStyle(TranslationStyle style) {
         return this;
     }
 
+    public String getPassword() {
+        return password;
+    }
+
+    public DocumentTranslateOptions setPassword(String password) {
+        this.password = password;
+        return this;
+    }
+
+    public DocumentExtractionParams getExtractionParams() {
+        return extractionParams;
+    }
+
+    public DocumentTranslateOptions setExtractionParams(DocumentExtractionParams extractionParams) {
+        this.extractionParams = extractionParams;
+        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));
+        params.set("password", password);
 
         return params;
     }

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

@@ -10,6 +10,8 @@ public class DocumentUploadOptions {
     private String[] glossaries = null;
     private Boolean noTrace = null;
     private TranslationStyle style = null;
+    private String password = null;
+    private DocumentExtractionParams extractionParams = null;
 
     public String[] getAdaptTo() {
         return adaptTo;
@@ -57,11 +59,31 @@ public DocumentUploadOptions setStyle(TranslationStyle style) {
         return this;
     }
 
+    public String getPassword() {
+        return password;
+    }
+
+    public DocumentUploadOptions setPassword(String password) {
+        this.password = password;
+        return this;
+    }
+
+    public DocumentExtractionParams getExtractionParams() {
+        return extractionParams;
+    }
+
+    public DocumentUploadOptions setExtractionParams(DocumentExtractionParams extractionParams) {
+        this.extractionParams = extractionParams;
+        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));
+        params.set("password", password);
+        params.set("extraction_params", extractionParams != null ? extractionParams.toParams() : null);
 
         return params;
     }

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

@@ -9,7 +9,6 @@
 import com.translated.lara.net.S3Client;
 
 import java.io.File;
-import java.io.IOException;
 import java.io.InputStream;
 import java.util.*;
 
@@ -120,6 +119,8 @@ public InputStream translate(File input, String source, String target, DocumentT
         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());
+        if (options != null && options.getPassword() != null) uploadOptions.setPassword(options.getPassword());
+        if (options != null && options.getExtractionParams() != null) uploadOptions.setExtractionParams(options.getExtractionParams());
 
         Document document = upload(input, source, target, uploadOptions);
         document = pollDocumentUntilCompleted(document);

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

@@ -0,0 +1,42 @@
+package com.translated.lara.translator;
+
+import com.translated.lara.net.HttpParams;
+
+/**
+ * Extraction parameters for DOCX files.
+ */
+public class DocxExtractionParams extends DocumentExtractionParams {
+
+    private Boolean extractComments = null;
+    private Boolean acceptRevisions = null;
+
+    public Boolean getExtractComments() {
+        return extractComments;
+    }
+
+    public DocxExtractionParams setExtractComments(Boolean extractComments) {
+        this.extractComments = extractComments;
+        return this;
+    }
+
+    public Boolean getAcceptRevisions() {
+        return acceptRevisions;
+    }
+
+    public DocxExtractionParams setAcceptRevisions(Boolean acceptRevisions) {
+        this.acceptRevisions = acceptRevisions;
+        return this;
+    }
+
+    @Override
+    public HttpParams<Object> toParams() {
+        HttpParams<Object> params = new HttpParams<>();
+        if (extractComments != null) {
+            params.set("extract_comments", extractComments);
+        }
+        if (acceptRevisions != null) {
+            params.set("accept_revisions", acceptRevisions);
+        }
+        return params;
+    }
+}