X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Flastfm.rb;h=28279881721124f07ca65853e16993c4b0c99dec;hb=4440a6d38f74c169e5b3e0e83b0fa44bea820a33;hp=46b140fff86c43e2599ba87e790a097faf3eab7e;hpb=899c42e1b879cb5b1e6f8203cd7ff53ed6d1a45b;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git diff --git a/data/rbot/plugins/lastfm.rb b/data/rbot/plugins/lastfm.rb index 46b140ff..28279881 100644 --- a/data/rbot/plugins/lastfm.rb +++ b/data/rbot/plugins/lastfm.rb @@ -11,9 +11,14 @@ # # License:: GPL v2 -require 'open-uri' - class ::LastFmEvent + SELECTOR = /.*?(.*?)<\/a>.*?(.*?)<\/a>.*?(?:(.*?) attending\s+)?.*?<\/td>\s+<\/tr>/m attr_accessor :url, :date, :artist, :location, :attendance def initialize(url, date, artist, location, attendance) @url = url @@ -35,17 +40,17 @@ 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 help(plugin, topic="") - case topic.intern + 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 @@ -68,27 +73,32 @@ class LastFmPlugin < Plugin 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 location - esc = URI.escape(location) - page = @bot.httputil.get "#{LASTFM}/events/?findloc=#{esc}" - else - esc = URI.escape(artist) - page = @bot.httputil.get "#{LASTFM}/events?s=#{esc}&findloc=" - end - + page = @bot.httputil.get LASTFM + "/events/" + query if page events = Array.new disp_events = Array.new - # matches are: - # 1. day 2. moth 3. year 4. url_who 5. who 6. url_where 7. where 8. how_many - pre_events = page.scan(/.*?(.*?)<\/a>.*?(.*?)<\/a>.*?(.*?)<\/td>\s+<\/tr>/m) + pre_events = page.scan(LastFmEvent::SELECTOR) # debug pre_events.inspect if pre_events.empty? - m.reply "No events found #{spec}, sorry" + # 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? + m.reply "No events found #{spec}, sorry" + return + end end - pre_events.each { |day, month, year, url_who, who, url_where, where, how_many| + 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>(.+)?/) @@ -105,9 +115,9 @@ class LastFmPlugin < Plugin debug where.inspect loc = where.ircify_html end - attendance = how_many.ircify_html + attendance = how_many ? how_many.ircify_html : '' events << LastFmEvent.new(url, date, artist, loc, attendance) - } + } } # debug events.inspect events[0...num].each { |event| @@ -131,7 +141,7 @@ class LastFmPlugin < Plugin artist = params[:who].to_s page = nil begin - esc = URI.escape(artist) + esc = URI.escape(CGI.escape(artist)) page = @bot.httputil.get "#{LASTFM}/music/#{esc}" if page if page.match(/

(.*?)<\/a><\/h1>/) @@ -142,7 +152,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 @@ -173,7 +183,7 @@ class LastFmPlugin < Plugin action = :neighbours if action == :neighbors user = params[:user] begin - data = open("http://ws.audioscrobbler.com/1.0/user/#{user}/#{action}.txt") + 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 @@ -183,12 +193,12 @@ class LastFmPlugin < Plugin 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 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 :action *user', :thread => true