rbot 0.9.14 → 0.9.15

data/data/rbot/plugins/seen.rb

@@ -8,10 +8,33 @@
 define_structure :Saw, :nick, :time, :type, :where, :message
 
 class SeenPlugin < Plugin
+
+  MSG_PUBLIC = N_("saying \"%{message}\" in %{where}")
+  MSG_ACTION = N_("doing *%{nick} %{message}* in %{where}")
+  MSG_NICK = N_("changing nick from %{nick} to %{message}")
+  MSG_PART = N_("leaving %{where} (%{message})")
+  MSG_PART_EMPTY = N_("leaving %{where}")
+  MSG_JOIN = N_("joining %{where}")
+  MSG_QUIT = N_("quitting IRC (%{message})")
+  MSG_TOPIC = N_("changing the topic of %{where} to \"%{message}\"")
+
+  CHANPRIV_CHAN      = N_("a private channel")
+  CHANPRIV_MSG_TOPIC  = N_("changing the topic of %{where}")
+
+  MSGPRIV_MSG_PUBLIC = N_("speaking in %{where}")
+  MSGPRIV_MSG_ACTION = N_("doing something in %{where}")
+
+  FORMAT_NORMAL = N_("%{nick} was last seen %{when}, %{doing}")
+  FORMAT_WITH_BEFORE = N_("%{nick} was last seen %{when}, %{doing} and %{time} before %{did_before}")
+
+  Config.register Config::IntegerValue.new('seen.max_results',
+    :default => 3, :validate => Proc.new{|v| v >= 0},
+    :desc => _("Maximum number of seen users to return in search (0 = no limit)."))
+
   def help(plugin, topic="")
-    "seen <nick> => have you seen, or when did you last see <nick>"
+    _("seen <nick> => have you seen, or when did you last see <nick>")
   end
-  
+
   def privmsg(m)
     unless(m.params =~ /^(\S)+$/)
       m.reply "incorrect usage: " + help(m.plugin)
@@ -23,7 +46,17 @@ class SeenPlugin < Plugin
     if @registry.has_key?(m.params)
       m.reply seen(@registry[m.params])
     else
-      m.reply "nope!"
+      rx = Regexp.new(m.params, true)
+      num_matched = 0
+      @registry.each {|nick, saw|
+        if nick.match(rx)
+          m.reply seen(saw)
+          num_matched += 1
+          break if num_matched == @bot.config['seen.max_results']
+        end
+      }
+
+      m.reply _("nope!") if num_matched.zero?
     end
   end
 
@@ -35,60 +68,158 @@ class SeenPlugin < Plugin
     when PrivMessage
       return if m.private?
       type = m.action? ? 'ACTION' : 'PUBLIC'
-      @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, now, type,
-                                          m.target.to_s, m.message.dup)
+      store m, 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, now, "QUIT", 
-                                        nil, m.message.dup)
+      store m, Saw.new(m.sourcenick.dup, now, "QUIT",
+                       nil, m.message.dup)
     when NickMessage
       return if m.address?
-      saw = Saw.new(m.oldnick, now, "NICK", nil, m.newnick)
-      @registry[m.oldnick] = saw
-      @registry[m.newnick] = saw
+      store m, Saw.new(m.oldnick, now, "NICK", nil, m.newnick)
     when PartMessage
       return if m.address?
-      @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "PART", 
-                                        m.target.to_s, m.message.dup)
+      store m, Saw.new(m.sourcenick.dup, Time.new, "PART",
+                       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.to_s, m.message.dup)
+      store m, Saw.new(m.sourcenick.dup, Time.new, "JOIN",
+                       m.target.to_s, m.message.dup)
     when TopicMessage
       return if m.address? or m.info_or_set == :info
-      @registry[m.sourcenick] = Saw.new(m.sourcenick.dup, Time.new, "TOPIC", 
-                                        m.target.to_s, m.message.dup)
+      store m, Saw.new(m.sourcenick.dup, Time.new, "TOPIC",
+                       m.target.to_s, m.message.dup)
     end
   end
-  
-  def seen(saw)
-    ret = "#{saw.nick} was last seen "
+
+  def seen(reg)
+    saw = case reg
+    when Struct::Saw
+      reg # for backwards compatibility
+    when Array
+      reg.last
+    end
+
+    if reg.kind_of? Array
+      before = reg.first
+    end
+
+    # TODO: a message should not be disclosed if:
+    # - it was said in a channel that was/is invite-only, private or secret
+    # - UNLESS the requester is also in the channel now, or the request is made
+    #   in the channel?
+    msg_privacy = false
+    # TODO: a channel or it's topic should not be disclosed if:
+    # - the channel was/is private or secret
+    # - UNLESS the requester is also in the channel now, or the request is made
+    #   in the channel?
+    chan_privacy = false
+
+    # What should be displayed for channel?
+    where = chan_privacy ? _(CHANPRIV_CHAN) : saw.where
+
+    formats = {
+      :normal      => _(FORMAT_NORMAL),
+      :with_before => _(FORMAT_WITH_BEFORE)
+    }
+
+    if before && [:PART, :QUIT].include?(saw.type.to_sym) &&
+       [:PUBLIC, :ACTION].include?(before.type.to_sym)
+      # TODO see chan_privacy
+      prev_chan_privacy = false
+      prev_where = prev_chan_privacy ? _(CHANPRIV_CHAN) : before.where
+      did_before = case before.type.to_sym
+      when :PUBLIC
+        _(msg_privacy ? MSGPRIV_MSG_PUBLIC : MSG_PUBLIC)
+      when :ACTION
+        _(msg_privacy ? MSGPRIV_MSG_ACTION : MSG_ACTION)
+      end % {
+        :nick => saw.nick,
+        :message => before.message,
+        :where => prev_where
+      }
+
+      format = :with_before
+
+      time_diff = saw.time - before.time
+      if time_diff < 300
+        time_before = _("a moment")
+      elsif time_diff < 3600
+        time_before = _("a while")
+      else
+        format = :normal
+      end
+    else
+      format = :normal
+    end
+
+    nick = saw.nick
     ago = Time.new - saw.time
-    
+
     if (ago.to_i == 0)
-      ret << "just now, "
+      time = _("just now")
     else
-      ret << Utils.secs_to_string(ago) + " ago, "
+      time = _("%{time} ago") % { :time => Utils.secs_to_string(ago) }
     end
 
-    case saw.type.to_sym
+    doing = case saw.type.to_sym
     when :PUBLIC
-      ret << "saying #{saw.message}"
+      _(msg_privacy ? MSGPRIV_MSG_PUBLIC : MSG_PUBLIC)
     when :ACTION
-      ret << "doing #{saw.nick} #{saw.message}"
+      _(msg_privacy ? MSGPRIV_MSG_ACTION : MSG_ACTION)
     when :NICK
-      ret << "changing nick from #{saw.nick} to #{saw.message}"
+      _(MSG_NICK)
     when :PART
-      ret << "leaving #{saw.where}"
+      if saw.message.empty?
+        _(MSG_PART_EMPTY)
+      else
+        _(MSG_PART)
+      end
     when :JOIN
-      ret << "joining #{saw.where}"
+      _(MSG_JOIN)
     when :QUIT
-      ret << "quitting IRC (#{saw.message})"
+      _(MSG_QUIT)
     when :TOPIC
-      ret << "changing the topic of #{saw.where} to #{saw.message}"
+      _(chan_privacy ? CHANPRIV_MSG_TOPIC : MSG_TOPIC)
+    end % { :message => saw.message, :where => where, :nick => saw.nick }
+
+    case format
+    when :normal
+      formats[:normal] % {
+        :nick  => saw.nick,
+        :when  => time,
+        :doing => doing,
+      }
+    when :with_before
+      formats[:with_before] % {
+        :nick  => saw.nick,
+        :when  => time,
+        :doing => doing,
+        :time  => time_before,
+        :did_before => did_before
+      }
     end
   end
-  
+
+  def store(m, saw)
+    # TODO: we need to store the channel state INVITE/SECRET/PRIVATE here, in
+    # some symbolic form, so that we know the prior state of the channel when
+    # it comes time to display.
+    reg = @registry[saw.nick]
+
+    if reg && reg.is_a?(Array)
+      reg.shift if reg.size > 1
+      reg.push(saw)
+    else
+      reg = [saw]
+    end
+
+    if m.is_a? NickMessage
+      @registry[m.newnick] = reg
+    end
+
+    @registry[saw.nick] = reg
+  end
 end
 plugin = SeenPlugin.new
 plugin.register("seen")

data/data/rbot/plugins/shortenurls.rb

@@ -15,7 +15,7 @@ require "shorturl"
 require "uri"
 
 class ShortenURLs < Plugin
-  # starting from about shorturl 0.8.4, the WWW module is not defined 
+  # 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',

data/data/rbot/plugins/slashdot.rb

@@ -31,7 +31,20 @@ class SlashdotPlugin < Plugin
     if arts.length > 1
       tits = []
       arts.each { |el|
-        artitle = (el/"div.generaltitle").first.to_html.ircify_html
+        # see if the div tag with generaltitle class is present
+        artitle = (el/"div.generaltitle").first
+        if artitle
+          tits << artitle.to_html.ircify_html
+          next
+        end
+        # otherwise, check for skin+datitle a tags
+        datitle = (el/"a.datitle").first
+        next unless datitle
+        skin = (el/"a.skin").first
+        artitle = [
+          skin ? skin.innerHTML.ircify_html : nil,
+          datitle.innerHTML.ircify_html
+        ].compact.join(" ")
         tits << artitle
       }
       content = tits.join(" | ")
@@ -49,7 +62,7 @@ class SlashdotPlugin < Plugin
       @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
@@ -86,7 +99,7 @@ class SlashdotPlugin < Plugin
       m.reply "search for #{search} failed"
     end
   end
-  
+
   def slashdot(m, params)
     debug params.inspect
     max = params[:limit].to_i
@@ -110,12 +123,12 @@ class SlashdotPlugin < Plugin
     max = 8 if max > 8
     matches = Array.new
     doc.elements.each("*/story") {|e|
-      matches << [ e.elements["title"].text, 
-                   e.elements["author"].text, 
+      matches << [ e.elements["title"].text,
+                   e.elements["author"].text,
                    e.elements["time"].text.gsub(/\d{4}-(\d{2})-(\d{2})/, "\\2/\\1").gsub(/:\d\d$/, "") ]
       done += 1
       break if done >= max
-    } 
+    }
     if oneline
       m.reply matches.collect{|mat| mat[0]}.join(" | ")
     else

data/data/rbot/plugins/spotify.rb

@@ -0,0 +1,78 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: spotify plugin for rbot
+#
+# Author:: Raine Virta <raine.virta@gmail.com>
+#
+# Copyright:: (C) 2009 Raine Virta
+#
+# License:: GPL v2
+
+class SpotifyPlugin < Plugin
+  def initialize
+    super
+
+    unless Object.const_defined?('Spotify')
+      raise 'Spotify module not found (lib_spotify plugin probably not enabled)'
+    end
+  end
+
+  def help(plugin, topic)
+    _("spotify plugin - usage: spotify <spotify>, spotify artist <artist>, spotify album <album>")
+  end
+
+  def search(m, params)
+    method = params[:method] || 'track'
+    begin
+      result = Spotify.search(method, params[:query].to_s)
+    rescue
+      m.reply "problems connecting to Spotify"
+    end
+
+    if result.nil?
+      m.reply "no results"
+      return
+    end
+
+    case method
+    when 'track'
+      reply = _("%{b}%{artist}%{b} – %{track}") % {
+        :artist => result.artist.name,
+        :track => result.name,
+        :b => Bold
+      }
+
+      if result.album.released
+        reply << _(" [%{u}%{album}%{u}, %{released}]") % {
+          :released => result.album.released,
+          :album => result.album.name,
+          :u => Underline
+        }
+      else
+        reply << _(" [%{u}%{album}%{u}]") % { :album => result.album.name, :u => Underline }
+      end
+
+      reply << _(" — %{url}") % { :url => result.url }
+    when 'artist'
+      reply = _("%{b}%{artist}%{b} — %{url}") % {
+        :b => Bold,
+        :artist => result.name,
+        :url => result.url
+      }
+    when 'album'
+      reply = _("%{b}%{artist}%{b} – %{u}%{album}%{u} — %{url}") % {
+        :b => Bold,
+        :u => Underline,
+        :artist => result.artist.name,
+        :album => result.name,
+        :url => result.url
+      }
+    end
+
+    m.reply reply
+  end
+end
+
+plugin = SpotifyPlugin.new
+plugin.map 'spotify [:method] *query', :action => :search, :requirements => { :method => /track|artist|album/ }

data/data/rbot/plugins/theyfightcrime.rb

@@ -1,21 +1,21 @@
 HE = [
 
     ['a superhumanly strong','an underprivileged','a globe-trotting','an impetuous','a shy','a suave','a notorious','a one-legged','an all-American','a short-sighted','an otherworldly','a hate-fuelled','a scrappy','an unconventional','a jaded','a leather-clad','a fiendish','a Nobel prize-winning','a suicidal','a maverick','a bookish','an old-fashioned','a witless','a lounge-singing','a war-weary','a scarfaced','a gun-slinging','an obese','a time-tossed','a benighted','an uncontrollable','an immortal','an oversexed','a world-famous','an ungodly','a fast talking','a deeply religious','a lonely','a sword-wielding','a genetically engineered'],
-    
+
     ['white trash','zombie','shark-wrestling','playboy','guitar-strumming','Jewish',
     'sweet-toothed','bohemian','crooked','chivalrous','moralistic','amnesiac','devious','drug-addicted',
     'voodoo','Catholic','overambitious','coffee-fuelled','pirate','misogynist','skateboarding',
     'arachnophobic','Amish','small-town','Republican','one-eyed','gay','guerilla','vegetarian',
     'dishevelled','alcoholic','flyboy','ninja','albino','hunchbacked','neurotic','umbrella-wielding',
     'native American','soccer-playing','day-dreaming'],
-    
+
     ['grifter','stage actor','paramedic','gentleman spy','jungle king','hairdresser',
     'photographer','ex-con','vagrant','filmmaker','werewolf','senator','romance novelist','shaman','cop',
     'rock star','farmboy','cat burglar','cowboy','cyborg','inventor','assassin','boxer','dog-catcher',
     'master criminal','gangster','firefighter','househusband','dwarf','librarian','paranormal investigator',
     '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''',
@@ -42,19 +42,19 @@ SHE = [
     'a plucky','a sarcastic','a psychotic','a hard-bitten','a manipulative','an orphaned','a cosmopolitan',
     'a chain-smoking','a cold-hearted','a warm-hearted','a sharp-shooting','an enchanted','a wealthy','a pregnant',
     'a mentally unstable','a virginal','a brilliant','a disco-crazy','a provocative','an artistic'],
-    
+
     ['tempestuous', 'Buddhist', 'foul-mouthed', 'nymphomaniac', 'green-skinned', 'impetuous', 'African-American','punk','hypochondriac','junkie','blonde','goth','insomniac','gypsy','mutant','renegade', 'tomboy','French-Canadian','motormouth','belly-dancing','communist','hip-hop','thirtysomething',
     'cigar-chomping','extravagent','out-of-work','Bolivian','mute','cat-loving','snooty','wisecracking',
     'red-headed','winged','kleptomaniac','antique-collecting','psychic','gold-digging','bisexual','paranoid',
     'streetsmart'],
-    
+
     ['archaeologist','pearl diver','mechanic','detective','hooker','femme fatale',
     'former first lady','barmaid','fairy princess','magician''s assistant','schoolgirl','college professor',
     'angel','bounty hunter','opera singer','cab driver','soap star','doctor','politician','lawyer','nun',
     'snake charmer','journalist','bodyguard','vampire','stripper','Valkyrie','wrestler','mermaid','single mother',
     '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',
@@ -80,14 +80,14 @@ TITLE = [
     'BLOOD', 'HARD', 'STEEL', 'TERMINAL', 'HOT', 'COLD', 'TOTAL', 'PROGNOSIS:',
     'BURNING', 'FAST', 'PAINFUL', 'MISSION:', 'DEADLY', 'PARTIAL',
     'RAGING', 'CORDIAL'],
-    
+
     ['DECISION', 'INCISION', 'CONCLUSION',
     'CONCUSSION', 'HEAT', 'FIRE', 'RECOIL', 'INSTINCT', 'DESIRE', 'WEAPON',
     'BADGE', 'DEED', 'JUSTICE', 'HEAT', 'VENGEANCE', 'RECESSION', 'COMBUSTION',
     'JUDGEMENT', 'WARNING', 'AWAKENING', 'JURISDICTION', 'BASIS', 'FINDINGS',
     'INJECTION', 'REJECTION', 'REMISSION', 'DIGESTION', 'IDENTITY',
     'DISPERSION'],
-    
+
     ['II','III','IV','V','VI','VII','VIII','IX','X','XI','XII','XIII','XIV','XV','XVI','XVII','XVIII','XIX']
 
 ]
@@ -96,11 +96,11 @@ class TheyFightCrime < Plugin
     def help(plugin, topic="")
         "movieplot => generate a random movie scenario. movietitle => generate a random movie title."
     end
-    
+
     def get_random_things(thing_array)
         thing_array.map { |things| things.pick_one }
     end
-    
+
     def movieplot(m, params)
         he_things = get_random_things(HE)
         she_things = get_random_things(SHE)