# License:: GPLv2
#
# This plugin allows using rbot to translate text on a few translation services
+#
+# TODO
+#
+# * Configuration for whether to show translation engine
+# * Optionally sync default translators with karma.rb ranking
require 'set'
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={})
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
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)
class ExciteTranslator < Translator
+ INFO = 'Excite.jp Translation <http://www.excite.co.jp/world/>'
def initialize(cache={})
require 'mechanize'
class GoogleTranslator < Translator
+ INFO = 'Google Translate <http://www.google.com/translate_t>'
+
def initialize(cache={})
require 'mechanize'
load_form!
class BabelfishTranslator < Translator
+ INFO = 'AltaVista Babel Fish Translation <http://babelfish.altavista.com/babelfish/>'
+
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')
super(Translator::Direction.pairs(language_pairs), cache)
end
- def translate(text, from, to)
- super
+ def do_translate(text, from, to)
if @form.fields.name('trtext').empty?
@form.add_field!('trtext', text)
else
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)
+ 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|
+ if @translators.has_key?(plugin)
+ translator = @translators[plugin]
+ _('%{translator} <from> <to> <phrase> => Look up phrase using %{info}, supported from -> to languages: %{directions}') % {
+ :translator => plugin,
+ :info => translator.class::INFO,
+ :directions => translator.directions.map do |source, targets|
_('%{source} -> %{targets}') %
{:source => source, :targets => targets.to_a.join(', ')}
end.join(' | ')
}
else
- _('Command: <translator> <from> <to> <phrase>, where <translator> is one of: %{translators}. Use help <translator> to look up supported from and to languages') %
+ _('Command: <translator> <from> <to> <phrase>, where <translator> 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, :show_provider => true})
+ 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
- translator = @translators[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}") %
- {:source => from, :target => to}
+ translation = Timeout.timeout(@bot.config['translator.timeout']) do
+ translator.translate(phrase, from, to)
end
+ m.reply(if params[:show_provider]
+ _('%{translation} (provided by %{translator})') %
+ {:translation => translation, :translator => tname}
+ else
+ translation
+ end)
+
+ 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
projects / user / henk / code / ruby / rbot.git / blobdiff