Simplify support for using "auto" to detect source language

[user/henk/code/ruby/rbot.git] / data / rbot / plugins / translator.rb

diff --git a/data/rbot/plugins/translator.rb b/data/rbot/plugins/translator.rb
index ad9c431c91e5e430426f28c544d448e69005db69..d4f554440965d2708be76019bf81029f42264e07 100644 (file)
--- a/data/rbot/plugins/translator.rb
+++ b/data/rbot/plugins/translator.rb
@@ -39,7 +39,7 @@ class Translator
     @cache = cache
   end
 
- 
+
   # whether the translator supports this direction
   def support?(from, to)
     from != to && @directions[from].include?(to)
@@ -182,35 +182,29 @@ end
 class GoogleTranslator < Translator
   INFO = 'Google Translate <http://www.google.com/translate_t>'
 
+  LANGUAGES =
+    %w[af sq am ar hy az eu be bn bh bg my ca chr zh zh_CN zh_TW hr
+    cs da dv en eo et tl fi fr gl ka de el gn gu iw hi hu is id iu
+    ga it ja kn kk km ko lv lt mk ms ml mt mr mn ne no or ps fa pl
+    pt_PT pa ro ru sa sr sd si sk sl es sw sv tg ta tl te th bo tr
+    uk ur uz ug vi cy yi auto]
   def initialize(cache={})
-    require 'mechanize'
-    load_form!
-
-    # we can probably safely assume that google translate is able to translate from
-    # any language in the source lang drop down list to any language in the target one
-    # so we create the language pairs based on that assumption
-    sl = @source_list.options.map { |o| o.value.sub('-', '_') }
-    tl = @target_list.options.map { |o| o.value.sub('-', '_') }
-    super(Translator::Direction.all_from_to(tl, sl), cache)
-  end
-
-  def load_form!
-    agent = WWW::Mechanize.new
-    # without faking the user agent, Google Translate will serve non-UTF-8 text
-    agent.user_agent_alias = 'Linux Konqueror'
-    @form = agent.get('http://www.google.com/translate_t').
-            forms_with(:action => '/translate_t').first
-    @source_list = @form.fields_with(:name => 'sl').last
-    @target_list = @form.fields_with(:name => 'tl').last
+    require "uri"
+    require "json"
+    super(Translator::Direction.all_to_all(LANGUAGES), cache)
   end
 
   def do_translate(text, from, to)
-    load_form!
+    langpair = [from == 'auto' ? '' : from, to].map { |e| e.tr('_', '-') }.join("|")
+    raw_json = Irc::Utils.bot.httputil.get_response(URI.escape(
+               "http://ajax.googleapis.com/ajax/services/language/translate?v=1.0&q=#{text}&langpair=#{langpair}")).body
+    response = JSON.parse(raw_json)
 
-    @source_list.value = from.sub('_', '-')
-    @target_list.value = to.sub('_', '-')
-    @form.fields_with(:name => 'text').last.value = text
-    @form.submit.parser.search('div#result_box').inner_html
+    if response["responseStatus"] != 200
+      raise Translator::NoTranslationError, response["responseDetails"]
+    else
+      response["responseData"]["translatedText"]
+    end
   end
 end
 
@@ -266,6 +260,9 @@ class TranslatorPlugin < Plugin
   Config.register Config::IntegerValue.new('translator.timeout',
     :default => 30, :validate => Proc.new{|v| v > 0},
     :desc => _("Number of seconds to wait for the translation service before timeout"))
+  Config.register Config::StringValue.new('translator.destination',
+    :default => "en",
+    :desc => _("Default destination language to be used with translate command"))
 
   TRANSLATORS = {
     'nifty' => NiftyTranslator,
@@ -315,17 +312,23 @@ class TranslatorPlugin < Plugin
     end
   end
 
+  def languages
+    @languages ||= @translators.map { |t| t.last.directions.keys }.flatten.uniq
+  end
+
   def update_default
-    @default_translators = bot.config['translator.default_list'] & @translators.keys 
+    @default_translators = bot.config['translator.default_list'] & @translators.keys
   end
 
   def cmd_translator(m, params)
-    from, to = params[:from], params[:to]
-    translator = @default_translators.find {|t| @translators[t].support?(from, to)}
+    params[:to] = @bot.config['translator.destination'] if params[:to].nil?
+    params[:from] ||= 'auto'
+    translator = @default_translators.find {|t| @translators[t].support?(params[:from], params[:to])}
+
     if translator
       cmd_translate m, params.merge({:translator => translator, :show_provider => true})
     else
-      m.reply _('None of the default translators (translator.default_list) supports translating from %{source} to %{target}') % {:source => from, :target => to}
+      m.reply _('None of the default translators (translator.default_list) supports translating from %{source} to %{target}') % {:source => params[:from], :target => params[:to]}
     end
   end
 
@@ -341,7 +344,7 @@ class TranslatorPlugin < Plugin
         end
         m.reply(if params[:show_provider]
                   _('%{translation} (provided by %{translator})') %
-                    {:translation => translation, :translator => tname}
+                    {:translation => translation, :translator => tname.gsub("_", " ")}
                 else
                   translation
                 end)
@@ -362,5 +365,9 @@ class TranslatorPlugin < Plugin
 end
 
 plugin = TranslatorPlugin.new
-plugin.map 'translator :from :to *phrase',
-           :action => :cmd_translator, :thread => true
+req = Hash[*%w(from to).map { |e| [e.to_sym, /#{plugin.languages.join("|")}/] }.flatten]
+
+plugin.map 'translate [:from] [:to] *phrase',
+           :action => :cmd_translator, :thread => true, :requirements => req
+plugin.map 'translator [:from] [:to] *phrase',
+           :action => :cmd_translator, :thread => true, :requirements => req