X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Ffish.rb;h=2394fa098f7186a64917924e7d7cb2bd2db80610;hb=16336b4a240a4265d1f2df1e30d7b68d3a924287;hp=57aaafc209955a2b24f684d88cc7285dbbadcf8d;hpb=21949774b91eaec6ecde4eaa8ad121e2c0a36b87;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git

diff --git a/data/rbot/plugins/fish.rb b/data/rbot/plugins/fish.rb
index 57aaafc2..2394fa09 100644
--- a/data/rbot/plugins/fish.rb
+++ b/data/rbot/plugins/fish.rb
@@ -1,26 +1,50 @@
-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}
+  BASEURL = 'http://babelfish.yahoo.com'
+
+  Config.register Config::EnumValue.new('translate.default_from',
+    :values => LANGS, :default => 'en',
+    :desc => "Default language to translate from")
+  Config.register Config::EnumValue.new('translate.default_to',
+    :values => LANGS, :default => 'en',
+    :desc => "Default language to translate to")
+
   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: en, fr, de, it, pt, es, nl"
+    case topic
+    when 'cache'
+      "translate cache [view|clear] => view or clear the translate cache contents"
+    else
+      from = @bot.config['translate.default_from']
+      to = @bot.config['translate.default_to']
+      "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"
+    end
   end
+
   def translate(m, params)
-    langs = ["en", "fr", "de", "it", "pt", "es", "nl"]
-    trans_from = params[:fromlang] ? params[:fromlang] : 'en'
-    trans_to = params[:tolang] ? params[:tolang] : 'en'
+    langs = LANGS
+    trans_from = params[:fromlang] ? params[:fromlang] : @bot.config['translate.default_from']
+    trans_to = params[:tolang] ? params[:tolang] : @bot.config['translate.default_to']
     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
+    data_text = CGI.escape trans_text
     trans_pair = "#{trans_from}_#{trans_to}"
+
+    if (trans_text =~ /^http:\/\//) && (URI.parse(trans_text) rescue nil)
+      m.reply 'webpage translation is not currently supported'
+      return
+      # TODO FIXME
+      # url = BASEURL+'/translate_url' +
+      #   "?lp=#{trans_pair}&trurl=#{data_text}"
+      #
+      # return Utils.get_first_pars([url], 1, :message => m)
+    end
+
     data = "lp=#{trans_pair}&doit=done&intl=1&tt=urltext&urltext=#{data_text}"
 
     # check cache for previous lookups
@@ -29,33 +53,69 @@ class BabelPlugin < Plugin
       return
     end
 
-    http = @bot.httputil.get_proxy(URI.parse("http://babelfish.altavista.com"))
-
-    http.start {|http|
-      resp = http.post(query, data, {"content-type",
-      "application/x-www-form-urlencoded"})
-  
-  if (resp.code == "200")
-    resp.body.each_line do |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
+    headers = {
+      "content-type" => "application/x-www-form-urlencoded; charset=utf-8"
+    }
+
+    query = "/translate_txt"
+
+    begin
+      body = @bot.httputil.get(BASEURL+query,
+                               :method => :post,
+                               :body => data,
+                               :headers => headers)
+    rescue Exception => e
+      m.reply "http error: #{e.message}"
+      return
+    end
+
+    case body
+    when nil
+      m.reply "couldn't talk to babelfish :("
+    when /^\s*<div id="result"><div style="[^"]*">(.*?)<\/div><\/div>\s*$/
+      answer = $1
+      # cache the answer
+      if(answer.length > 0)
+        @registry["#{trans_pair}/#{data_text}"] = answer
       end
+      m.reply answer
+      return
+    when /^\s+<option value="#{trans_pair}"\s+SELECTED>/
+      m.reply "couldn't parse babelfish response html :("
+    else
+      m.reply "babelfish doesn't support translation from #{trans_from} to #{trans_to}"
     end
-    m.reply "couldn't parse babelfish response html :("
-  else
-    m.reply "couldn't talk to babelfish :("
   end
-  }
+
+  def cache_mgmt(m, params)
+    cmd = params[:cmd].intern
+    case cmd
+    when :view
+      cache = []
+      @registry.each { |key, val|
+        cache << "%s => %s" % [key, val]
+      }
+      m.reply "translate cache: #{cache.inspect}"
+    when :clear
+      keys = []
+      @registry.each { |key, val|
+        keys << key
+      }
+      keys.each { |key|
+        @registry.delete(key)
+      }
+      cache_mgmt(m, :cmd => 'view')
+    end
   end
+
 end
+
 plugin = BabelPlugin.new
-plugin.map 'translate to :tolang *phrase'
-plugin.map 'translate from :fromlang *phrase'
-plugin.map 'translate :fromlang :tolang *phrase'
+
+plugin.default_auth('cache', false)
+
+plugin.map 'translate to :tolang *phrase', :thread => true
+plugin.map 'translate from :fromlang *phrase', :thread => true
+plugin.map 'translate cache :cmd', :action => :cache_mgmt, :auth_path => 'cache!', :requirements => { :cmd => /view|clear/ }
+plugin.map 'translate :fromlang :tolang *phrase', :thread => true