Merge pull request #14 from KicksiteDev/master

added optional format to translate method
aegorov · Mar 4, 2020 · 066fda8 · 066fda8
2 parents c30ab6f + 6bf52c8
commit 066fda8
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/lib/yandex/translator.rb b/lib/yandex/translator.rb
@@ -26,8 +26,8 @@ def get_langs
         translator.get_langs
       end
 
-      def translate(text, *lang)
-        translator.translate(text, *lang)
+      def translate(text, from: nil, to: nil)
+        translator.translate(text, from: from, to: to)
       end
 
       def detect(text)
@@ -51,13 +51,10 @@ def detect(text)
       lang == '' ? nil : lang
     end
 
-    def translate(text, *lang)
-      if lang.last.is_a?(Hash)
-        lang_options = lang.last
-        lang = [lang_options[:to], lang_options[:from]].compact
-      end
+    def translate(text, from: nil, to: nil, format: 'plain')
+      lang = (to.nil? && from.nil?) ? [] : [to, from].compact
 
-      options = { text: text, lang: lang.reverse.join('-') }
+      options = { text: text, lang: lang.reverse.join('-'), format: format }
 
       result = visit('/translate', options)['text']