summaryrefslogtreecommitdiff
path: root/data/rbot/plugins/fish.rb
blob: 8c115f90a3660022b74c723a157bd0711dae651a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
require 'net/http'
require 'uri/common'
Net::HTTP.version_1_2

class BabelPlugin < Plugin
  LANGS = %w{en fr de it pt es nl ru zh zt el ja ko}
  def help(plugin, topic="")
    "translate to <lang> <string> => translate from english to <lang>, translate from <lang> <string> => translate to english from <lang>, translate <fromlang> <tolang> <string> => translate from <fromlang> to <tolang>. Languages: #{LANGS.join(', ')}"
  end
  def translate(m, params)
    langs = LANGS
    trans_from = params[:fromlang] ? params[:fromlang] : 'en'
    trans_to = params[:tolang] ? params[:tolang] : 'en'
    trans_text = params[:phrase].to_s
    
    query = "/babelfish/tr"
    lang_match = langs.join("|")
    unless(trans_from =~ /^(#{lang_match})$/ && trans_to =~ /^(#{lang_match})$/)
      m.reply "invalid language: valid languagess are: #{langs.join(' ')}"
      return
    end

    data_text = URI.escape trans_text
    trans_pair = "#{trans_from}_#{trans_to}"
    data = "lp=#{trans_pair}&doit=done&intl=1&tt=urltext&urltext=#{data_text}"

    # check cache for previous lookups
    if @registry.has_key?("#{trans_pair}/#{data_text}")
      m.reply @registry["#{trans_pair}/#{data_text}"]
      return
    end

    http = @bot.httputil.get_proxy(URI.parse("http://babelfish.altavista.com"))

    headers = {
      "content-type" => "application/x-www-form-urlencoded; charset=utf-8",
      'accept-charset' => 'utf-8'
    }

    http.start {|http|
      resp = http.post(query, data, headers)
  
  if (resp.code == "200")
    lines = Array.new
    resp.body.each_line do |l|
      lines.push l
    end

    l = lines.join(" ")
    debug "babelfish response: #{l}"

    if(l =~ /^\s+<td bgcolor=white class=s><div style=padding:10px;>(.*)<\/div>/)
      answer = $1
      # cache the answer
      if(answer.length > 0)
        @registry["#{trans_pair}/#{data_text}"] = answer
      end
      m.reply answer
      return
    end
    m.reply "couldn't parse babelfish response html :("
  else
    m.reply "couldn't talk to babelfish :("
  end
  }
  end
end
plugin = BabelPlugin.new
plugin.map 'translate to :tolang *phrase'
plugin.map 'translate from :fromlang *phrase'
plugin.map 'translate :fromlang :tolang *phrase'