]> git.netwichtig.de Git - user/henk/code/ruby/rbot.git/blobdiff - data/rbot/plugins/lastfm.rb
dict.rb plugin: fixes the oxford parser so it doesn't spit out a bunch of extra crap
[user/henk/code/ruby/rbot.git] / data / rbot / plugins / lastfm.rb
index 4ec79510873861dee5ba494f5a4702a0a2475236..2b58bbd4f022121546d0049b73329777978e2e68 100644 (file)
@@ -5,16 +5,22 @@
 #
 # Author:: Jeremy Voorhis
 # Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
+# Author:: Casey Link <unnamedrambler@gmail.com>
 #
 # Copyright:: (C) 2005 Jeremy Voorhis
 # Copyright:: (C) 2007 Giuseppe Bilotta
+# Copyright:: (C) 2008 Casey Link
 #
 # License:: GPL v2
 
 class ::LastFmEvent
+  SELECTOR = /<tr class="vevent.*?<\/tr>/m
   # matches are:
   # 1. day 2. moth 3. year 4. url_who 5. who 6. url_where 7. where 8. how_many
-  REGEXP = /<tr class="vevent\s+\w+\s+\S+?-(\d\d)-(\d\d)-(\d\d\d\d)\s*">.*?<a class="url summary" href="(\/event\/\d+)">(.*?)<\/a>.*?<a href="(\/venue\/\d+)">(.*?)<\/a>.*?<td class="attendance">(.*?)<\/td>\s+<\/tr>/m
+  # TODO festival have TWO dates  -------+
+  # TODO event type -------------+       |
+  #                              V       V
+  MATCHER = /<tr class="vevent\s+\w+\s+(?:\S+\s+)?\S+?-(\d\d)-(\d\d)-(\d\d\d\d)\s*">.*?<a class="url summary" href="(\/event\/\d+)">(.*?)<\/a>.*?<a href="(\/venue\/\d+)">(.*?)<\/a>.*?<td>(?:(.*?) attending\s+)?.*?<\/td>\s+<\/tr>/m
   attr_accessor :url, :date, :artist, :location, :attendance
   def initialize(url, date, artist, location, attendance)
     @url = url
@@ -36,15 +42,27 @@ class ::LastFmEvent
 end
 
 class LastFmPlugin < Plugin
-  BotConfig.register BotConfigIntegerValue.new('lastfm.max_events',
+  Config.register Config::IntegerValue.new('lastfm.max_events',
     :default => 25, :validate => Proc.new{|v| v > 1},
     :desc => "Maximum number of events to display.")
-  BotConfig.register BotConfigIntegerValue.new('lastfm.default_events',
+  Config.register Config::IntegerValue.new('lastfm.default_events',
     :default => 3, :validate => Proc.new{|v| v > 1},
     :desc => "Default number of events to display.")
 
   LASTFM = "http://www.last.fm"
 
+  def initialize
+    super
+    class << @registry
+      def store(val)
+        val
+      end
+      def restore(val)
+        val
+      end
+    end
+  end
+
   def help(plugin, topic="")
     case (topic.intern rescue nil)
     when :event, :events
@@ -55,8 +73,14 @@ class LastFmPlugin < Plugin
       "lastfm track <name> => show information on track/song <name> from last.fm [not implemented yet]"
     when :album
       "lastfm album <name> => show information on album <name> from last.fm [not implemented yet]"
+    when :now
+      "lastfm now [<user>] => show the now playing track from last.fm"
+    when :set
+      "lastfm set <user> => associate your current irc nick with a last.fm user"
+    when :who
+      "lastfm who [<nick>] => show who <nick> is at last.fm. if <nick> is empty, show who you are at lastfm."
     else
-      "lastfm <function> <user> => lastfm data for <user> on last.fm where <function> in [recenttracks, topartists, topalbums, toptracks, tags, friends, neighbors]. other topics: events, artist, group, song, track, album"
+      "lastfm => show your now playing track at lastfm. lastfm <function> [<user>] => lastfm data for <user> on last.fm where <function> in [recenttracks, topartists, topalbums, toptracks, tags, friends, neighbors]. other topics: events, artist, group, song, track, album, now, set, who"
     end
   end
 
@@ -70,67 +94,94 @@ class LastFmPlugin < Plugin
     page = nil
     spec = location ? "in #{location}" : "by #{artist}"
     query = location ? "?findloc=#{CGI.escape(location)}" : "?s=#{CGI.escape(artist)}&findloc="
-    Thread.new {
-      begin
-        page = @bot.httputil.get LASTFM + "/events/" + query
-        if page
-          events = Array.new
-          disp_events = Array.new
-
-          pre_events = page.scan(LastFmEvent::REGEXP)
-          # debug pre_events.inspect
+    begin
+      page = @bot.httputil.get LASTFM + "/events/" + query
+      if page
+        events = Array.new
+        disp_events = Array.new
+
+        pre_events = page.scan(LastFmEvent::SELECTOR)
+        # debug pre_events.inspect
+        if pre_events.empty?
+          # We may not find any even because the page gives a list
+          # of locations instead. In this case, retry with the first of
+          # these location
+          if page.match(/<a href="(\/events\/\?l=[^"]+)">/)
+            debug "Rechecking with #{$1}"
+            page = @bot.httputil.get(LASTFM+$1)
+            debug page
+            pre_events = page.scan(LastFmEvent::SELECTOR) if page
+            debug pre_events
+          end
           if pre_events.empty?
-            # We may not find any even because the page gives a list
-            # of locations instead. In this case, retry with the first of
-            # these location
-            if page.match(/<a href="(\/events\/\?l=[^"]+)">/)
-              debug "Rechecking with #{$1}"
-              page = @bot.httputil.get(LASTFM+$1)
-              pre_events = page.scan(LastFmEvent::REGEXP) if page
-            end
-            if pre_events.empty?
-              m.reply "No events found #{spec}, sorry"
-              Thread.exit
-            end
+            m.reply "No events found #{spec}, sorry"
+            return
+          end
+        end
+        pre_events.each { |s| s.scan(LastFmEvent::MATCHER) { |day, month, year, url_who, who, url_where, where, how_many|
+          date = Time.utc(year.to_i, month.to_i, day.to_i)
+          url = LASTFM + url_who
+          if who.match(/<strong>(.*?)<\/strong>(.+)?/)
+            artist = Bold + $1.ircify_html + Bold
+            artist << ", " << $2.ircify_html if $2
+          else
+            debug "who: #{who.inspect}"
+            artist = who.ircify_html
           end
-          pre_events.each { |day, month, year, url_who, who, url_where, where, how_many|
-            date = Time.utc(year.to_i, month.to_i, day.to_i)
-            url = LASTFM + url_who
-            if who.match(/<strong>(.*?)<\/strong>(.+)?/)
-              artist = Bold + $1.ircify_html + Bold
-              artist << ", " << $2.ircify_html if $2
-            else
-              debug "who: #{who.inspect}"
-              artist = who.ircify_html
-            end
-            if where.match(/<strong>(.*?)<\/strong>(?:<br\s*\/>(.+)?)?/)
-              loc = Bold + $1.ircify_html + Bold
-              loc << ", " << $2.ircify_html if $2
-            else
-              debug where.inspect
-              loc = where.ircify_html
-            end
-            attendance = how_many.ircify_html
-            events << LastFmEvent.new(url, date, artist, loc, attendance)
-          }
-          # debug events.inspect
-
-          events[0...num].each { |event|
-            disp_events << event.to_s
-          }
-          m.reply disp_events.join(' | '), :split_at => /\s+\|\s+/
+          if where.match(/<strong>(.*?)<\/strong>(?:<br\s*\/>(.+)?)?/)
+            loc = Bold + $1.ircify_html + Bold
+            loc << ", " << $2.ircify_html if $2
+          else
+            debug where.inspect
+            loc = where.ircify_html
+          end
+          attendance = how_many ? how_many.ircify_html : ''
+          events << LastFmEvent.new(url, date, artist, loc, attendance)
+        } }
+        # debug events.inspect
+
+        events[0...num].each { |event|
+          disp_events << event.to_s
+        }
+        m.reply disp_events.join(' | '), :split_at => /\s+\|\s+/
+      else
+        m.reply "No events found #{spec}"
+        return
+      end
+    rescue Exception => e
+      m.reply "I had problems looking for events #{spec}"
+      error e.inspect
+      debug e.backtrace.join("\n")
+      debug page[0...10*1024] if page
+      return
+    end
+  end
+
+  def now_playing(m, params)
+    opts = { :cache => false }
+    user = nil
+    if params[:who] then
+      user = params[:who].to_s
+    elsif @registry.has_key? ( m.sourcenick ) then
+      user = @registry[ m.sourcenick ]
+    else
+      # m.reply "I don't know who you are on last.fm. Use 'lastfm set username' to identify yourself."
+      # return
+      user = m.sourcenick
+    end
+    page = nil
+    begin
+      page = @bot.httputil.get("#{LASTFM}/user/#{user}", opts)
+      if page
+        if page.match(/class="nowListening">\s*<td class="subject">\s*<a href="\/music.*?">(.*)<\/a>\s*<\/td/)
+          m.reply "#{user} is jammin to #{$1.ircify_html}"
         else
-          m.reply "No events found #{spec}"
-          Thread.exit
+          m.reply "#{user} isn't listening to anything right now."
         end
-      rescue Exception => e
-        m.reply "I had problems looking for events #{spec}"
-        error e.inspect
-        debug e.backtrace.join("\n")
-        debug page[0...10*1024] if page
-        Thread.exit
       end
-    }
+    rescue
+      m.reply "I had problems getting #{user}'s current info"
+    end
   end
 
   def find_artist(m, params)
@@ -148,7 +199,7 @@ class LastFmPlugin < Plugin
         end
 
         wiki = "This artist doesn't have a description yet. You can help by writing it: #{url}/+wiki?action=edit"
-        if page.match(/<div class="wikiAbstract">(.*?)<\/div>/m)
+        if page.match(/<div (?:class|id)="wikiAbstract">(.*?)<\/div>/m)
           wiki = $1.ircify_html
         end
 
@@ -174,29 +225,65 @@ class LastFmPlugin < Plugin
     m.reply "not implemented yet, sorry"
   end
 
+  def set_user(m, params)
+    user = params[:who].to_s
+    nick = m.sourcenick
+    @registry[ nick ] = user
+    m.reply "Ok, I'll remember that #{nick} is #{user} at last.fm"
+  end
+
+  def get_user(m, params)
+    nick = ""
+    if params[:who] then
+      nick = params[:who].to_s
+    else 
+      nick = m.sourcenick
+    end
+    if @registry.has_key?( nick ) then
+      user = @registry[ nick ]
+      m.reply "#{nick} is #{user} at last.fm"
+    else
+      m.reply "Sorry, I don't know who #{nick} is at last.fm"
+    end
+  end
+
   def lastfm(m, params)
     action = params[:action].intern
     action = :neighbours if action == :neighbors
-    user = params[:user]
-    Thread.new {
-      begin
-        data = @bot.httputil.get("http://ws.audioscrobbler.com/1.0/user/#{user}/#{action}.txt")
-        m.reply "#{action} for #{user}:"
-        m.reply data.to_a[0..3].map{|l| l.split(',',2)[-1].chomp}.join(", ")
-      rescue
-        m.reply "could not find #{action} for #{user} (is #{user} a user?)"
-      end
-    }
+    user = nil
+    if params[:user] then
+      user = params[:user].to_s
+    elsif @registry.has_key? ( m.sourcenick ) then
+      user = @registry[ m.sourcenick ]
+    else
+      # m.reply "I don't know who you are on last.fm. Use 'lastfm set username' to identify yourself."
+      # return
+      user = m.sourcenick
+    end
+    begin
+      data = @bot.httputil.get("http://ws.audioscrobbler.com/1.0/user/#{user}/#{action}.txt")
+      m.reply "#{action} for #{user}:"
+      m.reply data.to_a[0..3].map{|l| l.split(',',2)[-1].chomp}.join(", ")
+    rescue
+      m.reply "could not find #{action} for #{user} (is #{user} a user?)"
+    end
   end
 end
 
 plugin = LastFmPlugin.new
-plugin.map 'lastfm [:num] event[s] in *location', :action => :find_event, :requirements => { :num => /\d+/ }
-plugin.map 'lastfm [:num] event[s] by *who', :action => :find_event, :requirements => { :num => /\d+/ }
-plugin.map 'lastfm [:num] event[s] [for] *who', :action => :find_event, :requirements => { :num => /\d+/ }
-plugin.map 'lastfm artist *who', :action => :find_artist
-plugin.map 'lastfm group *who', :action => :find_artist
+plugin.map 'lastfm [:num] event[s] in *location', :action => :find_event, :requirements => { :num => /\d+/ }, :thread => true
+plugin.map 'lastfm [:num] event[s] by *who', :action => :find_event, :requirements => { :num => /\d+/ }, :thread => true
+plugin.map 'lastfm [:num] event[s] [for] *who', :action => :find_event, :requirements => { :num => /\d+/ }, :thread => true
+plugin.map 'lastfm artist *who', :action => :find_artist, :thread => true
+plugin.map 'lastfm group *who', :action => :find_artist, :thread => true
+plugin.map 'lastfm now *who', :action => :now_playing, :thread => true
+plugin.map 'lastfm now', :action => :now_playing, :thread => true
 plugin.map 'lastfm track *dunno', :action => :find_track
 plugin.map 'lastfm song *dunno', :action => :find_track
 plugin.map 'lastfm album *dunno', :action => :find_album
-plugin.map 'lastfm :action *user'
+plugin.map 'lastfm set *who', :action => :set_user, :thread => true
+plugin.map 'lastfm who *who', :action => :get_user, :thread => true
+plugin.map 'lastfm who', :action => :get_user, :thread => true
+plugin.map 'lastfm :action *user', :thread => true
+plugin.map 'lastfm :action', :thread => true
+plugin.map 'lastfm', :action => :now_playing, :thread => true