X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Flastfm.rb;h=2b58bbd4f022121546d0049b73329777978e2e68;hb=26d2e36ff08de4e3bbf510616d4c8dac6da9d6d0;hp=4ec79510873861dee5ba494f5a4702a0a2475236;hpb=b5f14172069c02320202bb799095d8f2813ec988;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git diff --git a/data/rbot/plugins/lastfm.rb b/data/rbot/plugins/lastfm.rb index 4ec79510..2b58bbd4 100644 --- a/data/rbot/plugins/lastfm.rb +++ b/data/rbot/plugins/lastfm.rb @@ -5,16 +5,22 @@ # # Author:: Jeremy Voorhis # Author:: Giuseppe "Oblomov" Bilotta +# Author:: Casey Link # # Copyright:: (C) 2005 Jeremy Voorhis # Copyright:: (C) 2007 Giuseppe Bilotta +# Copyright:: (C) 2008 Casey Link # # License:: GPL v2 class ::LastFmEvent + SELECTOR = /.*?(.*?)<\/a>.*?(.*?)<\/a>.*?(.*?)<\/td>\s+<\/tr>/m + # TODO festival have TWO dates -------+ + # TODO event type -------------+ | + # V V + MATCHER = /.*?(.*?)<\/a>.*?(.*?)<\/a>.*?(?:(.*?) 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 => show information on track/song from last.fm [not implemented yet]" when :album "lastfm album => show information on album from last.fm [not implemented yet]" + when :now + "lastfm now [] => show the now playing track from last.fm" + when :set + "lastfm set => associate your current irc nick with a last.fm user" + when :who + "lastfm who [] => show who is at last.fm. if is empty, show who you are at lastfm." else - "lastfm => lastfm data for on last.fm where 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 [] => lastfm data for on last.fm where 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(//) + 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(//) - 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>(.+)?/) + 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>(.+)?/) - 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>(?:(.+)?)?/) - 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>(?:(.+)?)?/) + 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*\s*(.*)<\/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>/m) + if page.match(/
(.*?)<\/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