X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Ftranslator.rb;h=41f4cf8e8ea73670eaef192e3d95380032847d5e;hb=56e4713c5c0498838ed77a409e44fbc3251acde2;hp=c4eaac4ff80c7cdcd1c414183efe89467d287220;hpb=b218f003e2625c2b9ebc9de39886fb2f54d68aa0;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git

diff --git a/data/rbot/plugins/translator.rb b/data/rbot/plugins/translator.rb
index c4eaac4f..41f4cf8e 100644
--- a/data/rbot/plugins/translator.rb
+++ b/data/rbot/plugins/translator.rb
@@ -34,18 +34,10 @@ class Translator
 
   attr_reader :directions, :cache
 
-  def initialize(directions, cache={})
+  def initialize(directions, cache={}, bot)
     @directions = directions
     @cache = cache
-  end
-
-  # Many translators use Mechanize, which changed namespace around version 1.0
-  # To support both pre-1.0 and post-1.0 namespaces, we use these auxiliary
-  # method. The translator still needs to require 'mechanize' on initialization
-  # if it needs it.
-  def mechanize
-    return Mechanize if defined? Mechanize
-    return WWW::Mechanize
+    @bot = bot
   end
 
   # whether the translator supports this direction
@@ -111,166 +103,29 @@ class Translator
   end
 end
 
+class YandexTranslator < Translator
+  INFO = 'Yandex Translator <http://translate.yandex.com/>'
+  LANGUAGES = %w{ar az be bg ca cs da de el en es et fi fr he hr hu hy it ka lt lv mk nl no pl pt ro ru sk sl sq sr sv tr uk}
 
-class NiftyTranslator < Translator
-  INFO = '@nifty Translation <http://nifty.amikai.com/amitext/indexUTF8.jsp>'
-
-  def initialize(cache={})
-   require 'mechanize'
-   super(Translator::Direction.all_from_to(%w[ja en zh_CN ko], %w[ja]), cache)
-  end
-
-  def do_translate(text, from, to)
-    @form ||= mechanize.new.
-              get('http://nifty.amikai.com/amitext/indexUTF8.jsp').
-              forms_with(:name => 'translateForm').last
-    @radio = @form.radiobuttons_with(:name => 'langpair').first
-    @radio.value = "#{from},#{to}".upcase
-    @radio.check
-    @form.fields_with(:name => 'sourceText').last.value = text
-
-    @form.submit(@form.buttons_with(:name => 'translate').last).
-          forms_with(:name => 'translateForm').last.fields_with(:name => 'translatedText').last.value
-  end
-end
-
-
-class ExciteTranslator < Translator
-  INFO = 'Excite.jp Translation <http://www.excite.co.jp/world/>'
-
-  def initialize(cache={})
-    require 'mechanize'
-    require 'iconv'
-
-    super(Translator::Direction.all_from_to(%w[ja en zh_CN zh_TW ko], %w[ja]), cache)
-
-    @forms = Hash.new do |h, k|
-      case k
-      when 'en'
-        h[k] = open_form('english')
-      when 'zh_CN', 'zh_TW'
-        # this way we don't need to fetch the same page twice
-        h['zh_CN'] = h['zh_TW'] = open_form('chinese')
-      when 'ko'
-        h[k] = open_form('korean')
-      end
-    end
-  end
-
-  def open_form(name)
-    mechanize.new.get("http://www.excite.co.jp/world/#{name}").
-                   forms_with(:name => 'world').first
-  end
-
-  def do_translate(text, from, to)
-    non_ja_language = from != 'ja' ? from : to
-    form = @forms[non_ja_language]
-
-    if non_ja_language =~ /zh_(CN|TW)/
-      form_with_fields(:name => 'wb_lp').first.value = "#{from}#{to}".sub(/_(?:CN|TW)/, '').upcase
-      form_with_fields(:name => 'big5').first.value = ($1 == 'TW' ? 'yes' : 'no')
-    else
-      # the en<->ja page is in Shift_JIS while other pages are UTF-8
-      text = Iconv.iconv('Shift_JIS', 'UTF-8', text) if non_ja_language == 'en'
-      form.fields_with(:name => 'wb_lp').first.value = "#{from}#{to}".upcase
-    end
-    form.fields_with(:name => 'before').first.value = text
-    result = form.submit.forms_with(:name => 'world').first.fields_with(:name => 'after').first.value
-    # the en<->ja page is in Shift_JIS while other pages are UTF-8
-    if non_ja_language == 'en'
-      Iconv.iconv('UTF-8', 'Shift_JIS', result)
-    else
-      result
-    end
-
-  end
-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 "uri"
-    require "json"
-    super(Translator::Direction.all_to_all(LANGUAGES), cache)
-  end
-
-  def do_translate(text, from, to)
-    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)
-
-    if response["responseStatus"] != 200
-      raise Translator::NoTranslationError, response["responseDetails"]
-    else
-      translation = response["responseData"]["translatedText"]
-      return Utils.decode_html_entities(translation)
-    end
-  end
-end
-
-
-class BabelfishTranslator < Translator
-  INFO = 'AltaVista Babel Fish Translation <http://babelfish.altavista.com/babelfish/>'
-
-  def initialize(cache)
-    require 'mechanize'
-    (_, lang_list) = parse_page
-    language_pairs = lang_list.options.map {|o| o.value.split('_')}.
-                                           reject {|p| p.empty?}
-    super(Translator::Direction.pairs(language_pairs), cache)
-  end
-
-  def parse_page
-    form = mechanize.new.get('http://babelfish.altavista.com/babelfish/').
-           forms_with(:name => 'frmTrText').first
-    lang_list = form.fields_with(:name => 'lp').first
-    [form, lang_list]
-  end
-
-  def do_translate(text, from, to)
-    unless @form && @lang_list
-      @form, @lang_list = parse_page
-    end
-    
-    if @form.fields_with(:name => 'trtext').empty?
-      @form.add_field!('trtext', text)
-    else
-      @form.fields_with(:name => 'trtext').first.value = text
-    end
-    @lang_list.value = "#{from}_#{to}"
-    @form.submit.parser.search("div[@id='result']/div[@style]").inner_html
-  end
-end
-
-class WorldlingoTranslator < Translator
-  INFO = 'WorldLingo Free Online Translator <http://www.worldlingo.com/en/products_services/worldlingo_translator.html>'
-
-  LANGUAGES = %w[en fr de it pt es ru nl el sv ar ja ko zh_CN zh_TW]
-  def initialize(cache)
+  URL = 'https://translate.yandex.net/api/v1.5/tr.json/translate?key=%s&lang=%s-%s&text=%s'
+  KEY = 'trnsl.1.1.20140326T031210Z.1e298c8adb4058ed.d93278fea8d79e0a0ba76b6ab4bfbf6ac43ada72'
+  def initialize(cache, bot)
     require 'uri'
-    super(Translator::Direction.all_to_all(LANGUAGES), cache)
+    require 'json'
+    super(Translator::Direction.all_to_all(LANGUAGES), cache, bot)
   end
 
   def translate(text, from, to)
-    response = Irc::Utils.bot.httputil.get_response(URI.escape(
-               "http://www.worldlingo.com/SEfpX0LV2xIxsIIELJ,2E5nOlz5RArCY,/texttranslate?wl_srcenc=utf-8&wl_trgenc=utf-8&wl_text=#{text}&wl_srclang=#{from.upcase}&wl_trglang=#{to.upcase}"))
-    # WorldLingo seems to respond an XML when error occurs
-    case response['Content-Type']
-    when %r'text/plain'
-      response.body
-    else
+    res = @bot.httputil.get_response(URL % [KEY, from, to, URI.escape(text)])
+    res = JSON.parse(res.body)
+
+    if res['code'] != 200
       raise Translator::NoTranslationError
+    else
+      res['text'].join(' ')
     end
   end
+
 end
 
 class TranslatorPlugin < Plugin
@@ -282,11 +137,7 @@ class TranslatorPlugin < Plugin
     :desc => _("Default destination language to be used with translate command"))
 
   TRANSLATORS = {
-    'nifty' => NiftyTranslator,
-    'excite' => ExciteTranslator,
-    'google_translate' => GoogleTranslator,
-    'babelfish' => BabelfishTranslator,
-    'worldlingo' => WorldlingoTranslator,
+    'yandex' => YandexTranslator,
   }
 
   def initialize
@@ -295,7 +146,7 @@ class TranslatorPlugin < Plugin
     @translators = {}
     TRANSLATORS.each_pair do |name, c|
       watch_for_fail(name) do
-        @translators[name] = c.new(@registry.sub_registry(name))
+        @translators[name] = c.new(@registry.sub_registry(name), @bot)
         map "#{name} :from :to *phrase",
           :action => :cmd_translate, :thread => true
       end
@@ -313,6 +164,8 @@ class TranslatorPlugin < Plugin
     begin
       yield
     rescue Exception
+      debug 'Translator error: '+$!.to_s
+      debug $@.join("\n")
       @failed_translators << { :name => name, :reason => $!.to_s }
 
       warning _("Translator %{name} cannot be used: %{reason}") %
@@ -382,22 +235,9 @@ class TranslatorPlugin < Plugin
     translator = @default_translators.find {|t| @translators[t].support?(params[:from], params[:to])}
 
     if translator
-      cmd_translate m, params.merge({:translator => translator, :show_provider => true})
+      cmd_translate m, params.merge({:translator => translator, :show_provider => false})
     else
-      # When translate command is used without source language, "auto" as source
-      # language is assumed. It means that google translator is used and we let google
-      # figure out what the source language is.
-      #
-      # Problem is that the google translator will fail if the system that the bot is
-      # running on does not have the json gem installed.
-      if params[:from] == 'auto'
-        m.reply _("Unable to auto-detect source language due to broken google translator, see %{reverse}%{prefix}help translate failed%{reverse} for details") % {
-          :reverse => Reverse,
-          :prefix => @bot.config['core.address_prefix'].first
-        }
-      else
-        m.reply _('None of the default translators (translator.default_list) supports translating from %{source} to %{target}') % {:source => params[:from], :target => params[:to]}
-      end
+      m.reply _('None of the default translators (translator.default_list) supports translating from %{source} to %{target}') % {:source => params[:from], :target => params[:to]}
     end
   end
 
@@ -453,10 +293,10 @@ end
 plugin = TranslatorPlugin.new
 req = Hash[*%w(from to).map { |e| [e.to_sym, /#{plugin.languages.join("|")}/] }.flatten]
 
-plugin.map 'translate [:from] [:to] *url',
-           :action => :cmd_translate_url, :requirements => req.merge(:url => %r{^https?://.*})
-plugin.map 'translator [:from] [:to] *url',
-           :action => :cmd_translate_url, :requirements => req.merge(:url => %r{^https?://.*})
+plugin.map 'translate [:from] [:to] :url',
+           :action => :cmd_translate_url, :requirements => req.merge(:url => %r{^https?://[^\s]*})
+plugin.map 'translator [:from] [:to] :url',
+           :action => :cmd_translate_url, :requirements => req.merge(:url => %r{^https?://[^\s]*})
 plugin.map 'translate [:from] [:to] *phrase',
            :action => :cmd_translator, :thread => true, :requirements => req
 plugin.map 'translator [:from] [:to] *phrase',