X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Flastfm.rb;h=4ec79510873861dee5ba494f5a4702a0a2475236;hb=eea3e8b5cf1bd6c16dbae2bfa62e271bca4ac9a6;hp=358137d281e414cc59d0521269e5bdb588748eac;hpb=796d473d40b3a7912a4cef4134a9f96de92e19c2;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git
diff --git a/data/rbot/plugins/lastfm.rb b/data/rbot/plugins/lastfm.rb
index 358137d2..4ec79510 100644
--- a/data/rbot/plugins/lastfm.rb
+++ b/data/rbot/plugins/lastfm.rb
@@ -11,9 +11,10 @@
#
# License:: GPL v2
-require 'open-uri'
-
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
@@ -35,13 +36,19 @@ class ::LastFmEvent
end
class LastFmPlugin < Plugin
+ BotConfig.register BotConfigIntegerValue.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',
+ :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 "
+ "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
@@ -54,75 +61,83 @@ class LastFmPlugin < Plugin
end
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}"
- 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
-
- 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)
- # debug pre_events.inspect
- if pre_events.empty?
- m.reply "No events found #{spec}, sorry"
- 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
+ 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
+ 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
end
- attendance = how_many.ircify_html
- events << LastFmEvent.new(url, date, artist, loc, attendance)
- }
- # debug events.inspect
-
- events[0..2].each { |event|
- disp_events << event.to_s
- }
- m.reply disp_events.join(' | ')
- else
- m.reply "No events found #{spec}"
- return
+ 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}"
+ Thread.exit
+ 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 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(artist)
+ esc = URI.escape(CGI.escape(artist))
page = @bot.httputil.get "#{LASTFM}/music/#{esc}"
if page
if page.match(/ | |