Add support of importing user defined translations

README.md

@@ -22,8 +22,15 @@ gradlew.bat run
 ## Formats
 - Kindle (sqlite) Default file located /Volumes/Kindle/system/vocabulary/vocab.db
 - Text. All words should be in one column.
+- Text with translations. (CSV separation with ";" sign)
 
 ## Text example:
+### Text witout translations
 apple  
 orange  
 watermelon
+### Text with translations
+adversarial;состязательный  
+cinnamon;корица     
+draw up;составлять (документы)  
+vigorously;энергично, решительно    

src/main/java/com/lingualeo/Importer.java

@@ -7,6 +7,7 @@
 
 import javax.naming.AuthenticationException;
 import java.io.IOException;
+import java.util.HashMap;
 import java.util.Iterator;
 import java.util.List;
 import java.util.logging.Level;
@@ -15,6 +16,7 @@
 public class Importer {
     private final List<Word> words;
     private final ApiClient client;
+    private HashMap<String, String> translations;
     private static final Logger logger = Logger.getLogger(Importer.class.getName());
     private ProgressBar progressBar;
 
@@ -30,6 +32,10 @@ public Importer(List<Word> words, ApiClient client) {
         this.client = client;
     }
 
+    public void setTranslations(HashMap<String, String> translations) {
+        this.translations = translations;
+    }
+
     public void startImport() {
         if (words.isEmpty()) {
             logger.warning("You don't have words in the file");
@@ -50,8 +56,16 @@ public void startImport() {
 
         for (Word word : words) {
             try {
-                Iterator<TranslateDto> it = client.getTranslates(word.getName()).iterator();
-                processWord(word, it);
+                if (translations == null
+                        || translations.isEmpty()
+                        || !translations.containsKey(word.getName())) {
+
+                    Iterator<TranslateDto> it = client.getTranslates(word.getName()).iterator();
+                    processWord(word, it);
+                } else {
+                    client.addWord(word.getName(), translations.get(word.getName()), word.getContext());
+                    logger.finest("Word added: " + word.getName());
+                }
 
                 float percent = i / count;
                 updateProgress(percent);
@@ -61,6 +75,7 @@ public void startImport() {
                 logger.log(Level.SEVERE, e.getMessage(), e);
             }
         }
+
         updateProgress(1);
         logger.finest("End import to Lingualeo...");
     }

src/main/java/com/lingualeo/controller/FormController.java

@@ -10,6 +10,7 @@
 import javafx.stage.FileChooser;
 
 import java.io.File;
+import java.util.HashMap;
 import java.util.List;
 
 public class FormController {
@@ -18,14 +19,17 @@ public class FormController {
     public Button startButton;
     public TextArea textField;
     public ProgressBar progressBar;
+    public CheckBox translationsCheckBox;
     private List<Word> words;
+    private HashMap<String, String> translations;
 
     @FXML
     protected void handleSubmitButtonAction() {
         Thread t = new Thread(
                 () -> {
                     String password = passwordField.getText();
                     Importer importer = new Importer(words, new ApiClient(emailField.getText(), password), progressBar);
+                    importer.setTranslations(translations);
                     startButton.setDisable(true);
                     importer.startImport();
                 }
@@ -44,6 +48,7 @@ protected void handleFileChooserAction() {
         if (file != null) {
             BaseReader reader = ReaderFactory.create(file);
             words = reader.read();
+            translations = reader.getTranslations();
             this.startButton.setDisable(false);
         }
     }

src/main/java/com/lingualeo/reader/BaseReader.java

@@ -2,11 +2,13 @@
 
 import java.io.File;
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
 
 public abstract class BaseReader {
     File file;
     List<Word> words = new ArrayList<>();
+    HashMap<String, String> translations = new HashMap<>();
 
     BaseReader(File file) {
         this.file = file;
@@ -15,6 +17,7 @@ public abstract class BaseReader {
     List<Word> getWords() {
         return words;
     }
+    public HashMap<String, String> getTranslations() { return translations; }
 
     public abstract List<Word> read();
 }

src/main/java/com/lingualeo/reader/TextReader.java

@@ -22,8 +22,14 @@ public List<Word> read() {
             BufferedReader br = new BufferedReader(fileReader);
             String line;
             while ((line = br.readLine()) != null) {
-                line = line.trim();
-                this.words.add(new Word(line.toLowerCase()));
+                String[] split_str = line.split(";", 2);
+                if (split_str.length == 1) {
+                    line = line.trim();
+                    this.words.add(new Word(line.toLowerCase()));
+                } else {
+                    this.words.add(new Word(split_str[0].trim()));
+                    this.translations.put(split_str[0].trim(), split_str[1].trim());
+                }
             }
             fileReader.close();
         } catch (IOException e) {