X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Flastfm.rb;h=09ea380305ae8166dcf1007a88c8731042001768;hb=b0c7f5f17b65043a796f8b3f4dc27cbfa287e087;hp=63935da15e0179feb02f90b3562314ac2379e710;hpb=7b35a6bfcd69da8b308dd1bc8cb0b9218caf4504;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git diff --git a/data/rbot/plugins/lastfm.rb b/data/rbot/plugins/lastfm.rb index 63935da1..09ea3803 100644 --- a/data/rbot/plugins/lastfm.rb +++ b/data/rbot/plugins/lastfm.rb @@ -1,25 +1,198 @@ -require 'open-uri' +#-- vim:sw=2:et +#++ +# +# :title: lastfm plugin for rbot +# +# Author:: Jeremy Voorhis +# Author:: Giuseppe "Oblomov" Bilotta +# +# Copyright:: (C) 2005 Jeremy Voorhis +# Copyright:: (C) 2007 Giuseppe Bilotta +# +# License:: GPL v2 -# plugin submitted by Jeremy Voorhis (jvoorhis) +class ::LastFmEvent + # matches are: + # 1. day 2. moth 3. year 4. url_who 5. who 6. url_where 7. where 8. how_many + REGEXP = /.*?(.*?)<\/a>.*?(.*?)<\/a>.*?(.*?)<\/td>\s+<\/tr>/m + attr_accessor :url, :date, :artist, :location, :attendance + def initialize(url, date, artist, location, attendance) + @url = url + @date = date + @artist = artist + @location = location + @attendance = attendance + end + + def compact_display + if @attendance.empty? + return "%s %s @ %s %s" % [@date.strftime("%a %b, %d %Y"), @artist, @location, @url] + else + return "%s %s @ %s (%s) %s" % [@date.strftime("%a %b, %d %Y"), @artist, @location, @attendance, @url] + end + end + alias :to_s :compact_display + +end class LastFmPlugin < Plugin + Config.register Config::IntegerValue.new('lastfm.max_events', + :default => 25, :validate => Proc.new{|v| v > 1}, + :desc => "Maximum number of events to display.") + 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 help(plugin, topic="") - "lastfm => lastfm data for on last.fm where in [recenttracks, topartists, topalbums, toptracks, tags, friends, neighbors]" + case (topic.intern rescue nil) + when :event, :events + "lastfm [] events in => show information on events in or near . lastfm [] events by => show information on events by . The number of events that can be displayed is optional, defaults to #{@bot.config['lastfm.default_events']} and cannot be higher than #{@bot.config['lastfm.max_events']}" + when :artist, :group + "lastfm artist => show information on artist/group from last.fm" + when :song, :track + "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]" + 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" + end end - def do_lastfm (m, params) + def find_event(m, params) + num = params[:num] || @bot.config['lastfm.default_events'] + num = num.to_i.clip(1, @bot.config['lastfm.max_events']) + + location = artist = nil + location = params[:location].to_s if params[:location] + artist = params[:who].to_s if params[:who] + page = nil + spec = location ? "in #{location}" : "by #{artist}" + query = location ? "?findloc=#{CGI.escape(location)}" : "?s=#{CGI.escape(artist)}&findloc=" begin - if params[:action] == "neighbors" || params[:action] == "neighbours" then - params[:action]="neighbours" + 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 + 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" + return + end + 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+/ + else + m.reply "No events found #{spec}" + return end - data = open("http://ws.audioscrobbler.com/1.0/user/#{params[:user]}/#{params[:action]}.txt") - m.reply "#{params[:action]} for #{params[:user]}:" - m.reply data.to_a[0..3].map{|l| l.split(',')[-1].chomp}.join(", ") + 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 find_artist(m, params) + artist = params[:who].to_s + page = nil + begin + esc = URI.escape(CGI.escape(artist)) + page = @bot.httputil.get "#{LASTFM}/music/#{esc}" + if page + if page.match(/

(.*?)<\/a><\/h1>/) + url = LASTFM + $1 + title = $2.ircify_html + else + raise "No URL/Title found for #{artist}" + 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) + wiki = $1.ircify_html + end + + m.reply "%s : %s\n%s" % [title, url, wiki], :overlong => :truncate + else + m.reply "no data found on #{artist}" + return + end + rescue Exception => e + m.reply "I had problems looking for #{artist}" + error e.inspect + debug e.backtrace.join("\n") + debug page[0...10*1024] if page + return + end + end + + def find_track(m, params) + m.reply "not implemented yet, sorry" + end + + def find_album(m, params) + m.reply "not implemented yet, sorry" + end + + def lastfm(m, params) + action = params[:action].intern + action = :neighbours if action == :neighbors + user = params[:user] + 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 #{params[:action]} for #{params[:user]} (is #{params[:user]} a user?)" + m.reply "could not find #{action} for #{user} (is #{user} a user?)" end end end plugin = LastFmPlugin.new -plugin.map 'lastfm :action :user', :action => 'do_lastfm' +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 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', :thread => true