rbot 0.9.10 → 0.9.12

data/data/rbot/plugins/seen.rb

@@ -1,4 +1,11 @@
-Saw = Struct.new("Saw", :nick, :time, :type, :where, :message)
+#-- vim:sw=2:et
+#++
+#
+# :title: Seen Plugin
+#
+# Keep a database of who last said/did what
+
+define_structure :Saw, :nick, :time, :type, :where, :message
 
 class SeenPlugin < Plugin
   def help(plugin, topic="")
@@ -21,39 +28,36 @@ class SeenPlugin < Plugin
   end
 
   def listen(m)
-    return if m.sourcenick.nil?
+    return unless m.source
     # keep database up to date with who last said what
-    if m.kind_of?(PrivMessage)
+    now = Time.new
+    case m
+    when PrivMessage
       return if m.private?
-      if m.action?
-        @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "ACTION", 
-                                          m.target, m.message.dup)
-      else
-        @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "PUBLIC",
-                                          m.target, m.message.dup)
-      end
-    elsif m.kind_of?(QuitMessage)
+      type = m.action? ? 'ACTION' : 'PUBLIC'
+      @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, now, type,
+                                          m.target.to_s, m.message.dup)
+    when QuitMessage
       return if m.address?
-      @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "QUIT", 
+      @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, now, "QUIT", 
                                         nil, m.message.dup)
-    elsif m.kind_of?(NickMessage)
+    when NickMessage
       return if m.address?
-      @registry[m.message] = Saw.new(m.sourcenick.dup, Time.new, "NICK", 
-                                        nil, m.message.dup)
-      @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "NICK", 
-                                        nil, m.message.dup)
-    elsif m.kind_of?(PartMessage)
+      saw = Saw.new(m.oldnick, now, "NICK", nil, m.newnick)
+      @registry[m.oldnick] = saw
+      @registry[m.newnick] = saw
+    when PartMessage
       return if m.address?
       @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "PART", 
-                                        m.target, m.message.dup)
-    elsif m.kind_of?(JoinMessage)
+                                        m.target.to_s, m.message.dup)
+    when JoinMessage
       return if m.address?
       @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "JOIN", 
-                                        m.target, m.message.dup)
-    elsif m.kind_of?(TopicMessage)
+                                        m.target.to_s, m.message.dup)
+    when TopicMessage
       return if m.address?
       @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "TOPIC", 
-                                        m.target, m.message.dup)
+                                        m.target.to_s, m.message.dup)
     end
   end
   
@@ -62,26 +66,26 @@ class SeenPlugin < Plugin
     ago = Time.new - saw.time
     
     if (ago.to_i == 0)
-      ret += "just now, "
+      ret << "just now, "
     else
-      ret += Utils.secs_to_string(ago) + " ago, "
+      ret << Utils.secs_to_string(ago) + " ago, "
     end
 
-    case saw.type
-    when "PUBLIC"
-      ret += "saying #{saw.message}"
-    when "ACTION"
-      ret += "doing #{saw.nick} #{saw.message}"
-    when "NICK"
-      ret += "changing nick from #{saw.nick} to #{saw.message}"
-    when "PART"
-      ret += "leaving #{saw.where}"
-    when "JOIN"
-      ret += "joining #{saw.where}"
-    when "QUIT"
-      ret += "quitting IRC (#{saw.message})"
-    when "TOPIC"
-      ret += "changing the topic of #{saw.where} to #{saw.message}"
+    case saw.type.to_sym
+    when :PUBLIC
+      ret << "saying #{saw.message}"
+    when :ACTION
+      ret << "doing #{saw.nick} #{saw.message}"
+    when :NICK
+      ret << "changing nick from #{saw.nick} to #{saw.message}"
+    when :PART
+      ret << "leaving #{saw.where}"
+    when :JOIN
+      ret << "joining #{saw.where}"
+    when :QUIT
+      ret << "quitting IRC (#{saw.message})"
+    when :TOPIC
+      ret << "changing the topic of #{saw.where} to #{saw.message}"
     end
   end
   

data/data/rbot/plugins/shortenurls.rb

@@ -0,0 +1,93 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: ShortURL plugin for rbot
+#
+# Author:: Giuseppe Bilotta <giuseppe.bilotta@gmail.com>
+# Copyright:: (C) 2007 Giuseppe Bilotta
+# License:: GPL v2
+#
+# Plugin to handle ShortURL, merges the funcionality of the old rubyurl and tinyurl plugins
+# Note that it's called ShortenURLs and not ShortURL, to prevent conflicts with
+# the actual ruby package used
+
+require "shorturl"
+require "uri"
+
+class ShortenURLs < Plugin
+  # starting from about shorturl 0.8.4, the WWW module is not defined 
+  include WWW rescue nil
+
+  Config.register Config::ArrayValue.new('shortenurls.services_blacklist',
+    :default => ['rubyurl', 'shorterlink'],
+    :requires_rescan => true,
+    :desc => "List of nonfunctioning shorturl services")
+
+  attr_accessor :services
+  def initialize
+    super
+    @blacklist = @bot.config['shortenurls.services_blacklist'].map { |s| s.intern }
+    # Instead of catering for all the services, we only pick the ones with 'link' or 'url' in the name
+    @services = ShortURL.valid_services.select { |service| service.to_s =~ /(?:link|url)/ } - @blacklist
+    if @services.include?(:rubyurl)
+      @services << :shorturl
+    end
+  end
+
+  # return a help string when the bot is asked for help on this plugin
+  def help(plugin, topic="")
+    return "shorten urls. syntax: <service> <your long url> => creates a shortened url using the required service (choose between #{@services.join(', ')}). Example: #{@bot.nick}, tinyurl http://some.long.url/wow-this-is/really-long.html"
+  end
+
+  # do the dirty job. This method can be called by other plugins, in which case you
+  # should set the :called param to true
+  def shorten(m, params)
+    url = params[:url]
+    if url == "help"
+      m.reply help(m.plugin) unless params[:called]
+      return nil
+    end
+    begin
+      to_uri = URI.parse(url)
+      # We don't accept 'generic' URLs because almost everything gets in there
+      raise URI::InvalidURIError if to_uri.class == URI::Generic
+    rescue URI::InvalidURIError
+      m.reply "#{url} doesn't look like an URL to me ..." unless params[:called]
+      return nil
+    end
+
+    service = (params[:service] || m.plugin).to_sym
+    service = :rubyurl if service == :shorturl
+
+    tried = []
+    short = nil
+
+    begin
+      tried << service
+      raise InvalidService, "#{service} blacklisted" if @blacklist.include?(service)
+      short = ShortURL.shorten(url, service)
+      raise InvalidService, "#{service} returned an empty string for #{url}" unless short and not short.empty?
+    rescue InvalidService
+      pool = services - tried
+      if pool.empty?
+        m.reply "#{service} failed, and I don't know what else to try next" unless params[:called]
+        return nil
+      else
+        service = pool.pick_one
+        m.reply "#{tried.last} failed, I'll try #{service} instead" unless params[:called]
+        retry
+      end
+    end
+
+    m.reply "#{url} shortened to #{short}" unless params[:called]
+    return short
+  end
+
+end
+
+# create an instance of the RubyURL class and register it as a plugin
+plugin = ShortenURLs.new
+
+plugin.services.each { |service|
+  plugin.map "#{service} :url", :action => 'shorten'
+}

data/data/rbot/plugins/slashdot.rb

@@ -1,22 +1,60 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: Slashdot plugin for rbot
+
 require 'rexml/document'
-require 'uri/common'
 
 class SlashdotPlugin < Plugin
   include REXML
   def help(plugin, topic="")
     "slashdot search <string> [<max>=4] => search slashdot for <string>, slashdot [<max>=4] => return up to <max> slashdot headlines (use negative max to return that many headlines, but all on one line.)"
   end
+
+  # This method defines a filter for /. pages. It's needed because the generic
+  # summarization grabs a comment, not the actual article.
+  #
+  # This filter relies on Hpricot being available, since REXML isn't too
+  # happy with the /. pages
+  def slashdot_filter(s)
+    return nil unless defined? Hpricot
+    loc = Utils.check_location(s, /slashdot\.org/)
+    return nil unless loc
+    h = Hpricot(s[:text])
+    # If we have no title tag in a head tag, return as this is not
+    # a /. page (it's probably a Slashdot RSS
+    ht = h/"head/title"
+    return nil if ht.empty?
+    title = ht.first.to_html.ircify_html
+    arts = (h/"div.article")
+    return nil if arts.empty?
+    if arts.length > 1
+      tits = []
+      arts.each { |el|
+        artitle = (el/"div.generaltitle").first.to_html.ircify_html
+        tits << artitle
+      }
+      content = tits.join(" | ")
+    else
+      det = (arts.first/"div.details").first.to_html.ircify_html
+      body = (arts.first/"div.body").first.to_html.ircify_html
+      content = [det, body].join(' ')
+    end
+    return {:title => title, :content => content}
+  end
+
+  def initialize
+    super
+    if defined? Hpricot
+      @bot.register_filter(:slashdot, :htmlinfo) { |s| slashdot_filter(s) }
+    end
+  end
   
   def search_slashdot(m, params)
-   max = params[:limit].to_i
-   search = params[:search].to_s
+    max = params[:limit].to_i
+    search = params[:search].to_s
 
-    begin
-      xml = @bot.httputil.get(URI.parse("http://slashdot.org/search.pl?content_type=rss&query=#{URI.escape(search)}"))
-    rescue URI::InvalidURIError, URI::BadURIError => e
-      m.reply "illegal search string #{search}"
-      return
-    end
+    xml = @bot.httputil.get("http://slashdot.org/search.pl?content_type=rss&query=#{CGI.escape(search)}")
     unless xml
       m.reply "search for #{search} failed"
       return
@@ -39,7 +77,7 @@ class SlashdotPlugin < Plugin
     doc.elements.each("*/item") {|e|
       desc = e.elements["title"].text
       desc.gsub!(/(.{150}).*/, '\1..')
-      reply = sprintf("%s | %s", e.elements["link"].text, desc)
+      reply = sprintf("%s | %s", e.elements["link"].text, desc.ircify_html)
       m.reply reply
       done += 1
       break if done >= max
@@ -53,7 +91,7 @@ class SlashdotPlugin < Plugin
     debug params.inspect
     max = params[:limit].to_i
     debug "max is #{max}"
-    xml = @bot.httputil.get(URI.parse("http://slashdot.org/slashdot.xml"))
+    xml = @bot.httputil.get('http://slashdot.org/slashdot.xml')
     unless xml
       m.reply "slashdot news parse failed"
       return

data/data/rbot/plugins/spell.rb

@@ -1,40 +1,50 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: Spell plugin
+
 class SpellPlugin < Plugin
+  Config.register Config::StringValue.new('spell.path',
+     :default => 'ispell',
+     :desc => _('Path to the program to use to check spelling'))
+  Config.register Config::StringValue.new('spell.command_line',
+     :default => '%s -a -S',
+     :desc => _('Command line used to call the spell.path. Use %s as a placeholder for the executable name'))
+
   def help(plugin, topic="")
-    "spell <word> => check spelling of <word>, suggest alternatives"
+    _("spell <word> => check spelling of <word>, suggest alternatives")
   end
   def privmsg(m)
     unless(m.params && m.params =~ /^\S+$/)
-      m.reply "incorrect usage: " + help(m.plugin)
+      m.reply _("incorrect usage: ") + help(m.plugin)
       return
     end
-    p = IO.popen("ispell -a -S", "w+")
-    if(p)
-      p.puts m.params
-      p.close_write
-      p.each_line {|l|
-        if(l =~ /^\*/)
-          m.reply "#{m.params} may be spelled correctly"
-          p.close
-          return
-        elsif(l =~ /^\s*&.*: (.*)$/)
-          m.reply "#{m.params}: #$1"
-          p.close
-          return
-        elsif(l =~ /^\s*\+ (.*)$/)
-          m.reply "#{m.params} is presumably derived from " + $1.downcase
-          p.close
-          return
-        elsif(l =~ /^\s*#/)
-          m.reply "#{m.params}: no suggestions"
-          p.close
-          return
-        end
+
+    begin
+      IO.popen(@bot.config['spell.command_line'] % @bot.config['spell.path'], "w+") { |p|
+        p.puts m.params
+        p.close_write
+        p.each_line { |l|
+          case l
+          when /^\*/
+            m.reply(_("%{word} may be spelled correctly") % {:word => m.params})
+          when /^\s*&.*: (.*)$/
+            m.reply "#{m.params}: #$1"
+          when /^\s*\+ (.*)$/
+            m.reply((_("%{word} is presumably derived from ") % {:word => m.params}) + $1.downcase)
+          when /^\s*#/
+            m.reply(_("%{word}: no suggestions") % {:word => m.params})
+          end
+          return if m.replied?
+        }
       }
-      p.close
-    else
-      m.reply "couldn't exec ispell :("
+    rescue
+      m.reply(_("couldn't exec %{prog} :(") % {:prog => @bot.config['spell.path']})
       return
     end
+    m.reply(_("something odd happened while checking %{word} with %{prog}") % {
+      :word => m.params, :prog => @bot.config['spell.path']
+    })
   end
 end
 plugin = SpellPlugin.new

data/data/rbot/plugins/theyfightcrime.rb

@@ -16,21 +16,21 @@ HE = [
     'Green Beret','waffle chef','vampire hunter','messiah','astronaut','sorceror','card sharp','matador',
     'barbarian'],
     
-    ['with a robot buddy named Sparky.','whom everyone believes is mad.','gone bad.',
-    'with a mysterious suitcase handcuffed to his arm.','living undercover at Ringling Bros. Circus.',
-    'searching for his wife''s true killer.','who dotes on his loving old ma.','looking for ''the Big One.''',
-    'who knows the secret of the alien invasion.','on the edge.','on a mission from God.','with a secret.',
-    'in drag.','plagued by the memory of his family''s brutal murder.','looking for a cure to the poison coursing through his veins.',
-    'moving from town to town, helping folk in trouble.',
-    'who must take medication to keep him sane.','who hangs with the wrong crowd.',
-    'possessed of the uncanny powers of an insect.','with a winning smile and a way with the ladies.',
-    'fleeing from a secret government programme.','from the ''hood.','haunted by an iconic dead American confidante.',
-    'with a passion for fast cars.','trapped in a world he never made.','in a wheelchair.',
-    'on the hunt for the last specimen of a great and near-mythical creature.','on the run.',
-    'for the 21st century.','who hides his scarred face behind a mask.','on the wrong side of the law.',
-    'with no name.','from the Mississippi delta.','with acid for blood.','with nothing left to lose.',
-    'haunted by memories of ''Nam.','on a search for his missing sister.','on his last day in the job.',
-    'from a doomed world.','who believes he can never love again.']
+    ['with a robot buddy named Sparky','whom everyone believes is mad','gone bad',
+    'with a mysterious suitcase handcuffed to his arm','living undercover at Ringling Bros. Circus',
+    'searching for his wife''s true killer','who dotes on his loving old ma','looking for ''the Big One''',
+    'who knows the secret of the alien invasion','on the edge','on a mission from God','with a secret',
+    'in drag','plagued by the memory of his family''s brutal murder','looking for a cure to the poison coursing through his veins',
+    'moving from town to town, helping folk in trouble',
+    'who must take medication to keep him sane','who hangs with the wrong crowd',
+    'possessed of the uncanny powers of an insect','with a winning smile and a way with the ladies',
+    'fleeing from a secret government programme','from the ''hood','haunted by an iconic dead American confidante',
+    'with a passion for fast cars','trapped in a world he never made','in a wheelchair',
+    'on the hunt for the last specimen of a great and near-mythical creature','on the run',
+    'for the 21st century','who hides his scarred face behind a mask','on the wrong side of the law',
+    'with no name','from the Mississippi delta','with acid for blood','with nothing left to lose',
+    'haunted by memories of ''Nam','on a search for his missing sister','on his last day in the job',
+    'from a doomed world','who believes he can never love again']
 
 ]
 
@@ -55,21 +55,21 @@ SHE = [
     'safe cracker','traffic cop','research scientist','queen of the dead','Hell''s Angel','museum curator',
     'advertising executive','widow','mercenary','socialite'],
     
-    ['on her way to prison for a murder she didn''t commit.','trying to make a difference in a man''s world.',
-    'with the soul of a mighty warrior.','looking for love in all the wrong places.','with an MBA from Harvard.',
-    'who hides her beauty behind a pair of thick-framed spectacles.','with the power to see death.',
-    'descended from a line of powerful witches.','from a family of eight older brothers.','with a flame-thrower.',
-    'with her own daytime radio talk show.','living on borrowed time.','who can talk to animals.',
-    'prone to fits of savage, blood-crazed rage.','who don''t take no shit from nobody.','with a knack for trouble.',
-    'who believes she is the reincarnation of an ancient Egyptian queen.','fleeing from a Satanic cult.',
-    'on the trail of a serial killer.','with a birthmark shaped like Liberty''s torch.',
-    'in the witness protection scheme.','from out of town.','from aristocratic European stock.',
-    'living homeless in New York''s sewers.','with only herself to blame.','from beyond the grave',
-    'married to the Mob.','from the wrong side of the tracks.','from a secret island of warrior women.',
-    'from Mars.','with someone else''s memories.','from a different time and place.','operating on the wrong side of the law.',
-    'who inherited a spooky stately manor from her late maiden aunt.','who dreams of becoming Elvis.',
-    'with a song in her heart and a spring in her step.','in the wrong place at the wrong time.',
-    'with an incredible destiny.','with the power to bend men''s minds.','with an evil twin sister.']
+    ['on her way to prison for a murder she didn''t commit','trying to make a difference in a man''s world',
+    'with the soul of a mighty warrior','looking for love in all the wrong places','with an MBA from Harvard',
+    'who hides her beauty behind a pair of thick-framed spectacles','with the power to see death',
+    'descended from a line of powerful witches','from a family of eight older brothers','with a flame-thrower',
+    'with her own daytime radio talk show','living on borrowed time','who can talk to animals',
+    'prone to fits of savage, blood-crazed rage','who don''t take no shit from nobody','with a knack for trouble',
+    'who believes she is the reincarnation of an ancient Egyptian queen','fleeing from a Satanic cult',
+    'on the trail of a serial killer','with a birthmark shaped like Liberty''s torch',
+    'in the witness protection scheme','from out of town','from aristocratic European stock',
+    'living homeless in New York''s sewers','with only herself to blame','from beyond the grave',
+    'married to the Mob','from the wrong side of the tracks','from a secret island of warrior women',
+    'from Mars','with someone else''s memories','from a different time and place','operating on the wrong side of the law',
+    'who inherited a spooky stately manor from her late maiden aunt','who dreams of becoming Elvis',
+    'with a song in her heart and a spring in her step','in the wrong place at the wrong time',
+    'with an incredible destiny','with the power to bend men''s minds','with an evil twin sister']
 
 ]
 
@@ -98,24 +98,21 @@ class TheyFightCrime < Plugin
     end
     
     def get_random_things(thing_array)
-        thing_array.map { |things| things[rand(things.length)] }
+        thing_array.map { |things| things.pick_one }
     end
     
-    def do_movieplot(m, params)
+    def movieplot(m, params)
         he_things = get_random_things(HE)
         she_things = get_random_things(SHE)
-        m.reply "Summary: He's #{he_things.join " "} She's #{she_things.join " "} They fight crime."
+        m.reply "Summary: He's #{he_things.join " "}. She's #{she_things.join " "}. They fight crime."
     end
 
-    def do_movietitle(m, params)
+    def movietitle(m, params)
         title_things = get_random_things(TITLE)
         m.reply "Title: #{title_things.join " "}"
     end
 end
 
 plugin = TheyFightCrime.new
-plugin.map 'movieplot', :action => 'do_movieplot'
-plugin.map 'movietitle', :action => 'do_movietitle'
-
-
-
+plugin.map 'movieplot'
+plugin.map 'movietitle'