X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Ftranslator.rb;h=6a84275475f46e4388f47720c836fe3898dca10b;hb=de9ddb8a7eb436c0cb8db81289373dc169f484ba;hp=76392ccc85c5e32ebaa2c606e0100e5ecf5e78ff;hpb=b640b14d732d457fec50b89738206a911ec9de7a;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git diff --git a/data/rbot/plugins/translator.rb b/data/rbot/plugins/translator.rb index 76392ccc..6a842754 100644 --- a/data/rbot/plugins/translator.rb +++ b/data/rbot/plugins/translator.rb @@ -19,10 +19,14 @@ require 'timeout' # methods in the Direction module are convenient for initializing this # attribute class Translator + INFO = 'Some translation service' class UnsupportedDirectionError < ArgumentError end + class NoTranslationError < RuntimeError + end + attr_reader :directions, :cache def initialize(directions, cache={}) @@ -36,18 +40,26 @@ class Translator from != to && @directions[from].include?(to) end - # this implements checking of languages and caching. subclasses should define the + # this implements argument checking and caching. subclasses should define the # do_translate method to implement actual translation def translate(text, from, to) raise UnsupportedDirectionError unless support?(from, to) - @cache[[text, from, to]] ||= do_translate(text, from, to) + raise ArgumentError, _("Cannot translate empty string") if text.empty? + request = [text, from, to] + unless @cache.has_key? request + translation = do_translate(text, from, to) + raise NoTranslationError if translation.empty? + @cache[request] = translation + else + @cache[request] + end end module Direction # given the set of supported languages, return a hash suitable for the directions # attribute which includes any language to any other language def self.all_to_all(languages) - directions = all_to_all(languages) + directions = all_to_none(languages) languages.each {|l| directions[l] = languages.to_set} directions end @@ -88,6 +100,8 @@ end class NiftyTranslator < Translator + INFO = '@nifty Translation ' + def initialize(cache={}) require 'mechanize' super(Translator::Direction.all_from_to(%w[ja en zh_CN ko], %w[ja]), cache) @@ -107,6 +121,7 @@ end class ExciteTranslator < Translator + INFO = 'Excite.jp Translation ' def initialize(cache={}) require 'mechanize' @@ -158,6 +173,8 @@ end class GoogleTranslator < Translator + INFO = 'Google Translate ' + def initialize(cache={}) require 'mechanize' load_form! @@ -188,9 +205,11 @@ end class BabelfishTranslator < Translator + INFO = 'AltaVista Babel Fish Translation ' + def initialize(cache) require 'mechanize' - + @form = WWW::Mechanize.new.get('http://babelfish.altavista.com/babelfish/'). forms.name('frmTrText').first @lang_list = @form.fields.name('lp') @@ -210,76 +229,120 @@ class BabelfishTranslator < Translator end end +class WorldlingoTranslator < Translator + INFO = 'WorldLingo Free Online Translator ' + + LANGUAGES = %w[en fr de it pt es ru nl el sv ar ja ko zh_CN zh_TW] + def initialize(cache) + require 'uri' + super(Translator::Direction.all_to_all(LANGUAGES), cache) + 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 + raise Translator::NoTranslationError + end + end +end + class TranslatorPlugin < Plugin - BotConfig.register BotConfigIntegerValue.new('translate.timeout', + 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")) + TRANSLATORS = { + 'nifty' => NiftyTranslator, + 'excite' => ExciteTranslator, + 'google_translate' => GoogleTranslator, + 'babelfish' => BabelfishTranslator, + 'worldlingo' => WorldlingoTranslator, + } + def initialize super - translator_classes = { - 'nifty' => NiftyTranslator, - 'excite' => ExciteTranslator, - 'google_translate' => GoogleTranslator, - 'babelfish' => BabelfishTranslator - } @translators = {} - - translator_classes.each_pair do |name, c| + TRANSLATORS.each_pair do |name, c| begin @translators[name] = c.new(@registry.sub_registry(name)) - map "#{name} :from :to *phrase", :action => :cmd_translate - rescue + map "#{name} :from :to *phrase", + :action => :cmd_translate, :thread => true + rescue Exception warning _("Translator %{name} cannot be used: %{reason}") % {:name => name, :reason => $!} end end + + Config.register Config::ArrayValue.new('translator.default_list', + :default => TRANSLATORS.keys, + :validate => Proc.new {|l| l.all? {|t| TRANSLATORS.has_key?(t)}}, + :desc => _("List of translators to try in order when translator name not specified"), + :on_change => Proc.new {|bot, v| update_default}) + update_default end def help(plugin, topic=nil) if @translators.has_key?(topic) - _('Supported directions of translation for %{translator}: %{directions}') % { - :translator => topic, - :directions => @translators[topic].directions.map do |source, targets| + translator = @translators[topic] + _('%{info}, supported directions of translation: %{directions}') % { + :info => translator.class::INFO, + :directions => translator.directions.map do |source, targets| _('%{source} -> %{targets}') % {:source => source, :targets => targets.to_a.join(', ')} end.join(' | ') } else - _('Command: , where is one of: %{translators}. Use help to look up supported from and to languages') % + _('Command: , where is one of: %{translators}. If "translator" is used in place of the translator name, the first translator in translator.default_list which supports the specified direction will be picked automatically. Use "help translator " to look up supported from and to languages') % {:translators => @translators.keys.join(', ')} end end + def update_default + @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)} + if translator + cmd_translate m, params.merge({:translator => translator}) + else + m.reply _('None of the default translators (translator.default_list) supports translating from %{source} to %{target}') % {:source => from, :target => to} + end + end + def cmd_translate(m, params) # get the first word of the command - tname = m.message[/\A(\w+)\s/, 1] + tname = params[:translator] || m.message[/\A(\w+)\s/, 1] translator = @translators[tname] from, to, phrase = params[:from], params[:to], params[:phrase].to_s if translator begin - if translator.support?(from, to) - translation = Timeout.timeout(@bot.config['translate.timeout']) do - translator.translate(phrase, from, to) - end - if translation.empty? - m.reply _('No translation returned') - else - m.reply translation - end - else - m.reply _("%{translator} doesn't support translating from %{source} to %{target}") % - {:translator => tname, :source => from, :target => to} + translation = Timeout.timeout(@bot.config['translator.timeout']) do + translator.translate(phrase, from, to) end + m.reply translation + rescue Translator::UnsupportedDirectionError + m.reply _("%{translator} doesn't support translating from %{source} to %{target}") % + {:translator => tname, :source => from, :target => to} + rescue Translator::NoTranslationError + m.reply _('%{translator} failed to provide a translation') % + {:translator => tname} rescue Timeout::Error m.reply _('The translator timed out') end else - m.reply _('No translator called %{name}') % {:name => translator} + m.reply _('No translator called %{name}') % {:name => tname} end end end plugin = TranslatorPlugin.new - +plugin.map 'translator :from :to *phrase', + :action => :cmd_translator, :thread => true