data/rbot/plugins/fish.rb

   1 require 'net/http'
   2 require 'uri/common'
   3 Net::HTTP.version_1_2
   4
   5 class BabelPlugin < Plugin
   6   LANGS = %w{en fr de it pt es nl ru zh zt el ja ko}
   7
   8   BotConfig.register BotConfigEnumValue.new('translate.default_from',
   9     :values => LANGS, :default => 'en',
  10     :desc => "Default language to translate from")
  11   BotConfig.register BotConfigEnumValue.new('translate.default_to',
  12     :values => LANGS, :default => 'en',
  13     :desc => "Default language to translate to")
  14
  15   def help(plugin, topic="")
  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(', ')}"
  19   end
  20
  21   def translate(m, params)
  22     langs = LANGS
  23     trans_from = params[:fromlang] ? params[:fromlang] : @bot.config['translate.default_from']
  24     trans_to = params[:tolang] ? params[:tolang] : @bot.config['translate.default_to']
  25     trans_text = params[:phrase].to_s
  26
  27     lang_match = langs.join("|")
  28     unless(trans_from =~ /^(#{lang_match})$/ && trans_to =~ /^(#{lang_match})$/)
  29       m.reply "invalid language: valid languagess are: #{langs.join(' ')}"
  30       return
  31     end
  32
  33     data_text = URI.escape trans_text
  34     trans_pair = "#{trans_from}_#{trans_to}"
  35
  36     if (trans_text =~ /^http:\/\//) && (URI.parse(trans_text) rescue nil)
  37       url = 'http://babelfish.altavista.com/babelfish/trurl_pagecontent' +
  38         "?lp=#{trans_pair}&url=#{data_text}"
  39
  40       return Utils.get_first_pars([url], 1, :message => m)
  41     end
  42
  43     data = "lp=#{trans_pair}&doit=done&intl=1&tt=urltext&urltext=#{data_text}"
  44
  45     # check cache for previous lookups
  46     if @registry.has_key?("#{trans_pair}/#{data_text}")
  47       m.reply @registry["#{trans_pair}/#{data_text}"]
  48       return
  49     end
  50
  51     headers = {
  52       "content-type" => "application/x-www-form-urlencoded; charset=utf-8"
  53     }
  54
  55     query = "/babelfish/tr"
  56
  57     begin
  58       body = @bot.httputil.get('http://babelfish.altavista.com'+query,
  59                                :method => :post,
  60                                :body => data,
  61                                :headers => headers)
  62     rescue Exception => e
  63       m.reply "http error: #{e.message}"
  64       return
  65     end
  66
  67     case body
  68     when nil
  69       m.reply "couldn't talk to babelfish :("
  70     when /^\s+<td bgcolor=white class=s><div style=padding:10px;>(.*)<\/div><\/td>\s*<\/tr>/m
  71       answer = $1.gsub(/\s*[\r\n]+\s*/,' ')
  72       # cache the answer
  73       if(answer.length > 0)
  74         @registry["#{trans_pair}/#{data_text}"] = answer
  75       end
  76       m.reply answer
  77       return
  78     when /^\s+<option value="#{trans_pair}"\s+SELECTED>/
  79       m.reply "couldn't parse babelfish response html :("
  80     else
  81       m.reply "babelfish doesn't support translation from #{trans_from} to #{trans_to}"
  82     end
  83   end
  84 end
  85
  86 plugin = BabelPlugin.new
  87 plugin.map 'translate to :tolang *phrase'
  88 plugin.map 'translate from :fromlang *phrase'
  89 plugin.map 'translate :fromlang :tolang *phrase'
  90