data/rbot/plugins/fish.rb

   1 class BabelPlugin < Plugin
   2   LANGS = %w{en fr de it pt es nl ru zh zt el ja ko}
   3
   4   BotConfig.register BotConfigEnumValue.new('translate.default_from',
   5     :values => LANGS, :default => 'en',
   6     :desc => "Default language to translate from")
   7   BotConfig.register BotConfigEnumValue.new('translate.default_to',
   8     :values => LANGS, :default => 'en',
   9     :desc => "Default language to translate to")
  10
  11   def help(plugin, topic="")
  12     case topic
  13     when 'cache'
  14       "translate cache [view|clear] => view or clear the translate cache contents"
  15     else
  16       from = @bot.config['translate.default_from']
  17       to = @bot.config['translate.default_to']
  18       "translate to <lang> <string> => translate from #{from} to <lang>, translate from <lang> <string> => translate to #{to} from <lang>, translate <fromlang> <tolang> <string> => translate from <fromlang> to <tolang>. If <string> is an http url, translates the referenced webpage and returns the 1st content paragraph. Languages: #{LANGS.join(', ')}. Other topics: cache"
  19     end
  20   end
  21
  22   def translate(m, params)
  23     langs = LANGS
  24     trans_from = params[:fromlang] ? params[:fromlang] : @bot.config['translate.default_from']
  25     trans_to = params[:tolang] ? params[:tolang] : @bot.config['translate.default_to']
  26     trans_text = params[:phrase].to_s
  27
  28     lang_match = langs.join("|")
  29     unless(trans_from =~ /^(#{lang_match})$/ && trans_to =~ /^(#{lang_match})$/)
  30       m.reply "invalid language: valid languagess are: #{langs.join(' ')}"
  31       return
  32     end
  33
  34     data_text = CGI.escape trans_text
  35     trans_pair = "#{trans_from}_#{trans_to}"
  36
  37     if (trans_text =~ /^http:\/\//) && (URI.parse(trans_text) rescue nil)
  38       url = 'http://babelfish.altavista.com/babelfish/trurl_pagecontent' +
  39         "?lp=#{trans_pair}&url=#{data_text}"
  40
  41       return Utils.get_first_pars([url], 1, :message => m)
  42     end
  43
  44     data = "lp=#{trans_pair}&doit=done&intl=1&tt=urltext&urltext=#{data_text}"
  45
  46     # check cache for previous lookups
  47     if @registry.has_key?("#{trans_pair}/#{data_text}")
  48       m.reply @registry["#{trans_pair}/#{data_text}"]
  49       return
  50     end
  51
  52     headers = {
  53       "content-type" => "application/x-www-form-urlencoded; charset=utf-8"
  54     }
  55
  56     query = "/babelfish/tr"
  57
  58     begin
  59       body = @bot.httputil.get('http://babelfish.altavista.com'+query,
  60                                :method => :post,
  61                                :body => data,
  62                                :headers => headers)
  63     rescue Exception => e
  64       m.reply "http error: #{e.message}"
  65       return
  66     end
  67
  68     case body
  69     when nil
  70       m.reply "couldn't talk to babelfish :("
  71     when /^\s+<td bgcolor=white class=s><div style=padding:10px;>(.*)<\/div><\/td>\s*<\/tr>/m
  72       answer = $1.gsub(/\s*[\r\n]+\s*/,' ')
  73       # cache the answer
  74       if(answer.length > 0)
  75         @registry["#{trans_pair}/#{data_text}"] = answer
  76       end
  77       m.reply answer
  78       return
  79     when /^\s+<option value="#{trans_pair}"\s+SELECTED>/
  80       m.reply "couldn't parse babelfish response html :("
  81     else
  82       m.reply "babelfish doesn't support translation from #{trans_from} to #{trans_to}"
  83     end
  84   end
  85
  86   def cache_mgmt(m, params)
  87     cmd = params[:cmd].intern
  88     case cmd
  89     when :view
  90       cache = []
  91       @registry.each { |key, val|
  92         cache << "%s => %s" % [key, val]
  93       }
  94       m.reply "translate cache: #{cache.inspect}"
  95     when :clear
  96       keys = []
  97       @registry.each { |key, val|
  98         keys << key
  99       }
 100       keys.each { |key|
 101         @registry.delete(key)
 102       }
 103       cache_mgmt(m, :cmd => 'view')
 104     end
 105   end
 106
 107 end
 108
 109 plugin = BabelPlugin.new
 110
 111 plugin.default_auth('cache', false)
 112
 113 plugin.map 'translate to :tolang *phrase'
 114 plugin.map 'translate from :fromlang *phrase'
 115 plugin.map 'translate cache :cmd', :action => :cache_mgmt, :auth_path => 'cache!', :requirements => { :cmd => /view|clear/ }
 116 plugin.map 'translate :fromlang :tolang *phrase'
 117