rbot 0.9.10 → 0.9.12

data/data/rbot/plugins/wow.rb

@@ -1,11 +1,13 @@
+#-- vim:sw=2:et
+#++
 #
-# World of Warcraft Realm Status plugin for rbot
-# by MrChucho (mrchucho@mrchucho.net)
-# Copyright (C) 2006 Ralph M. Churchill
+# :title: World of Warcraft Realm Status plugin for rbot
 #
-# Requires: insatiable appetite for World of Warcraft
+# Author:: MrChucho (mrchucho@mrchucho.net)
+# Copyright:: (C) 2006 Ralph M. Churchill
 #
-require 'open-uri'
+# Requires:: insatiable appetite for World of Warcraft
+
 require 'rexml/document'
 
 class Realm
@@ -16,25 +18,6 @@ class Realm
         self.type = pretty_type(type)
         self.pop = pretty_pop(pop)
     end
-    def Realm.get_realm_status(realm_name)
-        begin
-            open("http://www.worldofwarcraft.com/realmstatus/status.xml") do |xmldoc|
-                realm_list = (REXML::Document.new xmldoc).root
-                realm_data = realm_list.elements["r[@n=\"#{realm_name}\"]"]
-                if realm_data and realm_data.attributes.any? then
-                    realm = Realm.new(
-                        realm_data.attributes['n'],
-                        realm_data.attributes['s'].to_i,
-                        realm_data.attributes['t'].to_i,
-                        realm_data.attributes['l'].to_i)
-                else
-                    "Realm, #{realm_name}, not found."
-                end
-            end
-        rescue => err
-            "Error retrieving realm status: #{err}"
-        end
-    end
     def to_s
         "#{name} (#{type}) Status: #{status} Population: #{pop}"
     end
@@ -102,22 +85,43 @@ class RealmPlugin < Plugin
     def usage(m,params={})
         m.reply USAGE
     end
+    def get_realm_status(realm_name)
+        begin
+          xmldoc = @bot.httputil.get("http://www.worldofwarcraft.com/realmstatus/status.xml", :cache => false)
+          raise "unable to retrieve realm status" unless xmldoc
+          realm_list = (REXML::Document.new xmldoc).root
+          realm_data = realm_list.get_elements("//r[@n=\"#{realm_name}\"]").first
+          if realm_data and realm_data.attributes.any? then
+            realm = Realm.new(
+              realm_data.attributes['n'],
+              realm_data.attributes['s'].to_i,
+              realm_data.attributes['t'].to_i,
+              realm_data.attributes['l'].to_i)
+            realm.to_s
+          else
+            "realm #{realm_name} not found."
+          end
+        rescue => err
+          "error retrieving realm status: #{err}"
+        end
+    end
     def realm(m,params)
-        if params[:realm_name] and params[:realm_name].any?
-            realm_name = params[:realm_name].collect{|tok|
-                tok.capitalize
-            }.join(' ')
-            @registry[m.sourcenick] = realm_name
-            m.reply Realm.get_realm_status(realm_name)
+      if params[:realm_name] and params[:realm_name].any?
+        realm_name = params[:realm_name].collect{|tok|
+          tok.capitalize
+        }.join(' ')
+        @registry[m.sourcenick] = realm_name
+        m.reply get_realm_status(realm_name)
+      else
+        if @registry.has_key?(m.sourcenick)
+          realm_name = @registry[m.sourcenick]
+          m.reply get_realm_status(realm_name)
         else
-            if @registry.has_key?(m.sourcenick)
-                realm_name = @registry[m.sourcenick]
-                m.reply Realm.get_realm_status(realm_name)
-            else
-                m.reply "I don't know which realm you want.\n#{USAGE}"
-            end
+          m.reply "I don't know which realm you want.\n#{USAGE}"
         end
+      end
     end
 end
 plugin = RealmPlugin.new
-plugin.map 'realm *realm_name', :defaults => {:realm_name => false}
+plugin.map 'realm *realm_name',
+  :defaults => {:realm_name => false}, :thread => true

data/data/rbot/plugins/wserver.rb

@@ -1,7 +1,3 @@
-require 'net/http'
-require 'uri'
-Net::HTTP.version_1_2
-
 class WserverPlugin < Plugin
   def help(plugin, topic="")
     "wserver <uri> => try and determine what webserver <uri> is using"
@@ -29,30 +25,26 @@ class WserverPlugin < Plugin
         return
       end
         
-      http = @bot.httputil.get_proxy(uri)
-      http.open_timeout = 5
       
-      http.start {|http|
-        resp = http.head('/')
-        server = resp['Server']
-        if(server && server.length > 0)
-          m.reply "#{uri.host} is running #{server}"
-        else
-          m.reply "couldn't tell what #{uri.host} is running"
-        end
-        
-        if(resp.code == "302" || resp.code == "301") 
-          newloc = resp['location']
-          newuri = URI.parse(newloc)
-          # detect and ignore incorrect redirects (to relative paths etc)
-          if (newuri.host != nil)
-            if(uri.host != newuri.host)
-              m.reply "#{uri.host} redirects to #{newuri.scheme}://#{newuri.host}"
-              raise resp['location']
-            end
+      resp = @bot.httputil.head(uri)
+      server = resp['Server']
+      if(server && server.length > 0)
+        m.reply "#{uri.host} is running #{server}"
+      else
+        m.reply "couldn't tell what #{uri.host} is running"
+      end
+
+      if(resp.code == "302" || resp.code == "301") 
+        newloc = resp['location']
+        newuri = URI.parse(newloc)
+        # detect and ignore incorrect redirects (to relative paths etc)
+        if (newuri.host != nil)
+          if(uri.host != newuri.host)
+            m.reply "#{uri.host} redirects to #{newuri.scheme}://#{newuri.host}"
+            raise resp['location']
           end
         end
-      }
+      end
     rescue TimeoutError => err
       m.reply "timed out connecting to #{uri.host}:#{uri.port} :("
       return

data/data/rbot/plugins/youtube.rb

@@ -0,0 +1,237 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: YouTube plugin for rbot
+#
+# Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
+#
+# Copyright:: (C) 2008 Giuseppe Bilotta
+
+
+class YouTubePlugin < Plugin
+  YOUTUBE_SEARCH = "http://gdata.youtube.com/feeds/api/videos?vq=%{words}&orderby=relevance"
+  YOUTUBE_VIDEO = "http://gdata.youtube.com/feeds/api/videos/%{id}"
+
+  YOUTUBE_VIDEO_URLS = %r{youtube.com/(?:watch\?v=|v/)(.*?)(&.*)?$}
+
+  Config.register Config::IntegerValue.new('youtube.hits',
+    :default => 3,
+    :desc => "Number of hits to return from YouTube searches")
+  Config.register Config::IntegerValue.new('youtube.descs',
+    :default => 3,
+    :desc => "When set to n > 0, the bot will return the description of the first n videos found")
+  Config.register Config::BooleanValue.new('youtube.formats',
+    :default => true,
+    :desc => "Should the bot display alternative URLs (swf, rstp) for YouTube videos?")
+
+  def youtube_filter(s)
+    loc = Utils.check_location(s, /youtube\.com/)
+    return nil unless loc
+    if s[:text].include? '<div id="watch-vid-title">'
+      vid = @bot.filter(:"youtube.video", s)
+      return nil unless vid
+      content = _("Category: %{cat}. Rating: %{rating}. Author: %{author}. Duration: %{duration}. %{views} views, faved %{faves} times. %{desc}") % vid
+      return vid.merge(:content => content)
+    elsif s[:text].include? '<!-- start search results -->'
+      vids = @bot.filter(:"youtube.search", s)[:videos]
+      if !vids.empty?
+        return nil # TODO
+      end
+    end
+    # otherwise, just grab the proper div
+    if defined? Hpricot
+      content = (Hpricot(s[:text])/".watch-video-desc").to_html.ircify_html
+    end
+    # suboptimal, but still better than the default HTML info extractor
+    dm = /<div\s+class="watch-video-desc"[^>]*>/.match(s[:text])
+    content ||= dm ? dm.post_match.ircify_html : '(no description found)'
+    return {:title => s[:text].ircify_html_title, :content => content}
+  end
+
+  def youtube_apivideo_filter(s)
+    # This filter can be used either
+    e = s[:rexml] || REXML::Document.new(s[:text]).elements["entry"]
+    # TODO precomputing mg doesn't work on my REXML, despite what the doc
+    # says?
+    #   mg = e.elements["media:group"]
+    #   :title => mg["media:title"].text
+    # fails because "media:title" is not an Integer. Bah
+    vid = {
+      :formats => [],
+      :author => (e.elements["author/name"].text rescue nil),
+      :title =>  (e.elements["media:group/media:title"].text rescue nil),
+      :desc =>   (e.elements["media:group/media:description"].text rescue nil),
+      :cat => (e.elements["media:group/media:category"].text rescue nil),
+      :seconds => (e.elements["media:group/yt:duration/@seconds"].value.to_i rescue nil),
+      :url => (e.elements["media:group/media:player/@url"].value rescue nil),
+      :rating => (("%s/%s" % [e.elements["gd:rating/@average"].value, e.elements["gd:rating/@max"].value]) rescue nil),
+      :views => (e.elements["yt:statistics/@viewCount"].value rescue nil),
+      :faves => (e.elements["yt:statistics/@favoriteCount"].value rescue nil)
+    }
+    if vid[:desc]
+      vid[:desc].gsub!(/\s+/m, " ")
+    end
+    if secs = vid[:seconds]
+      vid[:duration] = Utils.secs_to_short(secs)
+    else
+      vid[:duration] = _("unknown duration")
+    end
+    e.elements.each("media:group/media:content") { |c|
+      if url = (c.elements["@url"].value rescue nil)
+        type = c.elements["@type"].value rescue nil
+        medium = c.elements["@medium"].value rescue nil
+        expression = c.elements["@expression"].value rescue nil
+        seconds = c.elements["@duration"].value.to_i rescue nil
+        fmt = case num_fmt = (c.elements["@yt:format"].value rescue nil)
+              when "1"
+                "h263+amr"
+              when "5"
+                "swf"
+              when "6"
+                "mp4+aac"
+              when nil
+                nil
+              else
+                num_fmt
+              end
+        vid[:formats] << {
+          :url => url, :type => type,
+          :medium => medium, :expression => expression,
+          :seconds => seconds,
+          :numeric_format => num_fmt,
+          :format => fmt
+        }.delete_if { |k, v| v.nil? }
+        if seconds
+          vid[:formats].last[:duration] = Utils.secs_to_short(seconds)
+        else
+          vid[:formats].last[:duration] = _("unknown duration")
+        end
+      end
+    }
+    debug vid
+    return vid
+  end
+
+  def youtube_apisearch_filter(s)
+    vids = []
+    title = nil
+    begin
+      doc = REXML::Document.new(s[:text])
+      title = doc.elements["feed/title"].text
+      doc.elements.each("*/entry") { |e|
+        vids << @bot.filter(:"youtube.apivideo", :rexml => e)
+      }
+      debug vids
+    rescue => e
+      debug e
+    end
+    return {:title => title, :vids => vids}
+  end
+
+  def youtube_search_filter(s)
+    # TODO
+    # hits = s[:hits] || @bot.config['youtube.hits']
+    # scrap the videos
+    return []
+  end
+
+  # Filter a YouTube video URL
+  def youtube_video_filter(s)
+    id = s[:youtube_video_id]
+    if not id
+      url = s.key?(:headers) ? s[:headers]['x-rbot-location'].first : s[:url]
+      debug url
+      id = YOUTUBE_VIDEO_URLS.match(url).captures.first rescue nil
+    end
+    return nil unless id
+
+    debug id
+
+    url = YOUTUBE_VIDEO % {:id => id}
+    resp, xml = @bot.httputil.get_response(url)
+    unless Net::HTTPSuccess === resp
+      debug("error looking for movie %{id} on youtube: %{e}" % {:id => id, :e => xml})
+      return nil
+    end
+    debug xml
+    begin
+      return @bot.filter(:"youtube.apivideo", DataStream.new(xml, s))
+    rescue => e
+      debug e
+      return nil
+    end
+  end
+
+  def initialize
+    super
+    @bot.register_filter(:youtube, :htmlinfo) { |s| youtube_filter(s) }
+    @bot.register_filter(:apisearch, :youtube) { |s| youtube_apisearch_filter(s) }
+    @bot.register_filter(:apivideo, :youtube) { |s| youtube_apivideo_filter(s) }
+    @bot.register_filter(:search, :youtube) { |s| youtube_search_filter(s) }
+    @bot.register_filter(:video, :youtube) { |s| youtube_video_filter(s) }
+  end
+
+  def info(m, params)
+    movie = params[:movie]
+    id = nil
+    if movie =~ /^[A-Za-z0-9]+$/
+      id = movie.dup
+    end
+
+    vid = @bot.filter(:"youtube.video", :url => movie, :youtube_video_id => id)
+    if vid
+      str = _("%{bold}%{title}%{bold} [%{cat}] %{rating} @ %{url} by %{author} (%{duration}). %{views} views, faved %{faves} times. %{desc}") %
+        {:bold => Bold}.merge(vid)
+      if @bot.config['youtube.formats'] and not vid[:formats].empty?
+        str << _("\n -- also available at: ")
+        str << vid[:formats].inject([]) { |list, fmt|
+          list << ("%{url} %{type} %{format} (%{duration} %{expression} %{medium})" % fmt)
+        }.join(', ')
+      end
+      m.reply str
+    else
+      m.reply(_("couldn't retrieve video info") % {:id => id})
+    end
+  end
+
+  def search(m, params)
+    what = params[:words].to_s
+    searchfor = CGI.escape what
+    url = YOUTUBE_SEARCH % {:words => searchfor}
+    resp, xml = @bot.httputil.get_response(url)
+    unless Net::HTTPSuccess === resp
+      m.reply(_("error looking for %{what} on youtube: %{e}") % {:what => what, :e => xml})
+      return
+    end
+    debug "filtering XML"
+    vids = @bot.filter(:"youtube.apisearch", DataStream.new(xml, params))[:vids][0, @bot.config['youtube.hits']]
+    debug vids
+    case vids.length
+    when 0
+      m.reply _("no videos found for %{what}") % {:what => what}
+      return
+    when 1
+      show = "%{title} (%{duration}) [%{desc}] @ %{url}" % vids.first
+      m.reply _("One video found for %{what}: %{show}") % {:what => what, :show => show}
+    else
+      idx = 0
+      shorts = vids.inject([]) { |list, el|
+        idx += 1
+        list << ("#{idx}. %{bold}%{title}%{bold} (%{duration}) @ %{url}" % {:bold => Bold}.merge(el))
+      }.join(" | ")
+      m.reply(_("Videos for %{what}: %{shorts}") % {:what =>what, :shorts => shorts},
+              :split_at => /\s+\|\s+/)
+      if (descs = @bot.config['youtube.descs']) > 0
+        vids[0, descs].each_with_index { |v, i|
+          m.reply("[#{i+1}] %{title} (%{duration}): %{desc}" % v, :overlong => :truncate)
+        }
+      end
+    end
+  end
+
+end
+
+plugin = YouTubePlugin.new
+
+plugin.map "youtube info :movie", :action => 'info', :threaded => true
+plugin.map "youtube [search] *words", :action => 'search', :threaded => true

data/data/rbot/templates/lart/larts-italian

@@ -0,0 +1,25 @@
+scambia lo shampoo di <who> con crema depilatoria
+installa Windows 3.1 sul computer di <who>
+costringe <who> a programmare in Perl per 3 settimane
+manda l'indirizzo email di <who> a 50 noti spammer
+restringe il terminale di <who> a 40x24
+ruba il the di <who>
+picchia <who> a sangue sulle gengive con una mazza ferrata arrugginata
+vende <who> su eBay
+ficca un peperoncino su per il culo di <who>
+fa <who> a fettine
+punzecchia la bambola voodoo di <who>
+riformatta il disco fisso di <who>
+brucia la collezione di fumetti di <who>
+brucia la biblioteca di <who>
+brucia la macchina di <who>
+mette zucchero nel serbatoio della moto di <who>
+legge a <who> un po' di poesia Vogon
+svuota il frigorifero di <who>
+picchia <who> in faccia con la 'Ricerca' di Proust 
+usa <who> come cavia in esperimenti di guerra batteriologica
+pubblica su internet la foto di <who> sedut@ al cesso
+scrive il numero di telefono di <who> su tutti i cessi pubblici
+scollega la connessione ad internet di <who>
+strappa le braccia di <who> e le usa per picchiarl@ a morte
+nasconde un nido di vipere nel cuscino di <who>

data/data/rbot/templates/lart/larts-japanese

@@ -0,0 +1,10 @@
+<who>のパソコンにWindowsをインスタールする
+<who>に３週間Perlを使わせる
+<who>を叩き付ける
+<who>の手を剥して、その手で<who>に打つ
+Ebayで<who>を売る
+<who>のコーヒーに灰汁を入れる
+<who>をはらきりさせる
+<who>を刺身に作る
+<who>に心臓麻痺させる
+<who>の名前をデスノートに書く

data/data/rbot/templates/lart/praises-italian

@@ -0,0 +1,4 @@
+abbraccia <who>
+adora <who>
+invita <who> ad una cena luculliana
+da' un'amichevole pacca sulle spalle a <who>

data/data/rbot/templates/lart/praises-japanese

@@ -0,0 +1,3 @@
+<who>とチュウする
+<who>にクキーあげる
+<who>ダイスキ！

data/data/rbot/templates/quiz/win_messages

@@ -0,0 +1,11 @@
+BINGO!! <who> got it right. The answer was: <answer>
+OMG!! PONIES!! <who> is the cutest. The answer was: <answer>
+HUZZAAAH! <who> did it again. The answer was: <answer>
+YEEEHA! Cowboy <who> scored again. The answer was: <answer>
+STRIKE! <who> pwned you all. The answer was: <answer>
+YAY :)) <who> is totally invited to my next sleepover. The answer was: <answer>
+And the crowd GOES WILD for <who>. The answer was: <answer>
+GOOOAAALLLL! That was one fine strike by <who>. The answer was: <answer>
+HOO-RAY, <who> deserves a medal! Only <who> could have known the answer: <answer>
+OKAY, <who> is officially a spermatologist! Answer was: <answer>
+WOOO, I bet that <who> knows where the word 'trivia' comes from too! Answer was: <answer>