]> git.netwichtig.de Git - user/henk/code/ruby/rbot.git/blob - data/rbot/plugins/lastfm.rb
markov: refactor triplet learning
[user/henk/code/ruby/rbot.git] / data / rbot / plugins / lastfm.rb
1 #-- vim:sw=2:et
2 #++
3 #
4 # :title: lastfm plugin for rbot
5 #
6 # Author:: Jeremy Voorhis
7 # Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
8 # Author:: Casey Link <unnamedrambler@gmail.com>
9 # Author:: Raine Virta <rane@kapsi.fi>
10 #
11 # Copyright:: (C) 2005 Jeremy Voorhis
12 # Copyright:: (C) 2007 Giuseppe Bilotta
13 # Copyright:: (C) 2008 Casey Link
14 # Copyright:: (C) 2009 Raine Virta
15 #
16 # License:: GPL v2
17
18 require 'rexml/document'
19 require 'cgi'
20
21 class ::LastFmEvent
22   def initialize(hash)
23     @url = hash[:url]
24     @date = hash[:date]
25     @location = hash[:location]
26     @description = hash[:description]
27     @attendance = hash[:attendance]
28
29     @artists = hash[:artists]
30
31     if @artists.length > 10 #more than 10 artists and it floods
32       diff = @artists.length - 10
33       @artist_string = Bold + @artists[0..10].join(', ') + Bold
34       @artist_string << _(" and %{n} more...") % {:n => diff}
35     else
36       @artist_string = Bold + @artists.join(', ') + Bold
37     end
38   end
39
40   def compact_display
41    unless @attendance.zero?
42      return "%s %s @ %s (%s attending) %s" % [@date.strftime("%a, %b %d"), @artist_string, @location, @attendance, @url]
43    end
44    return "%s %s @ %s %s" % [@date.strftime("%a, %b %d"), @artist_string, @location, @url]
45   end
46   alias :to_s :compact_display
47
48 end
49
50 define_structure :LastFmVenue, :id, :city, :street, :postal, :country, :name, :url, :lat, :long
51
52 class LastFmPlugin < Plugin
53   include REXML
54   Config.register Config::IntegerValue.new('lastfm.max_events',
55     :default => 25, :validate => Proc.new{|v| v > 1},
56     :desc => "Maximum number of events to display.")
57   Config.register Config::IntegerValue.new('lastfm.default_events',
58     :default => 3, :validate => Proc.new{|v| v > 1},
59     :desc => "Default number of events to display.")
60   Config.register Config::IntegerValue.new('lastfm.max_shouts',
61     :default => 5, :validate => Proc.new{|v| v > 1},
62     :desc => "Maximum number of user shouts to display.")
63   Config.register Config::IntegerValue.new('lastfm.default_shouts',
64     :default => 3, :validate => Proc.new{|v| v > 1},
65     :desc => "Default number of user shouts to display.")
66   Config.register Config::IntegerValue.new('lastfm.max_user_data',
67     :default => 25, :validate => Proc.new{|v| v > 1},
68     :desc => "Maximum number of user data entries (except events and shouts) to display.")
69   Config.register Config::IntegerValue.new('lastfm.default_user_data',
70     :default => 10, :validate => Proc.new{|v| v > 1},
71     :desc => "Default number of user data entries (except events and shouts) to display.")
72
73   APIKEY = "b25b959554ed76058ac220b7b2e0a026"
74   APIURL = "http://ws.audioscrobbler.com/2.0/?api_key=#{APIKEY}&"
75
76   def initialize
77     super
78     class << @registry
79       def store(val)
80         val
81       end
82       def restore(val)
83         val
84       end
85     end
86   end
87
88   def help(plugin, topic="")
89     period = _(", where <period> can be one of: 3|6|12 months, a year")
90     case (topic.intern rescue nil)
91     when :event, :events
92       _("lastfm [<num>] events in <location> => show information on events in or near <location>. lastfm [<num>] events by <artist/group> => show information on events by <artist/group>. The number of events <num> that can be displayed is optional, defaults to %{d} and cannot be higher than %{m}") % {:d => @bot.config['lastfm.default_events'], :m => @bot.config['lastfm.max_events']}
93     when :artist
94       _("lastfm artist <name> => show information on artist <name> from last.fm")
95     when :album
96       _("lastfm album <name> => show information on album <name> from last.fm [not implemented yet]")
97     when :track
98       _("lastfm track <name> => search tracks matching <name> on last.fm")
99     when :now, :np
100       _("lastfm now [<nick>] => show the now playing track from last.fm. np [<nick>] does the same.")
101     when :set
102       _("lastfm set user <user> => associate your current irc nick with a last.fm user. lastfm set verb <present>, <past> => set your preferred now playing/just played verbs. default \"is listening to\" and \"listened to\".")
103     when :who
104       _("lastfm who [<nick>] => show who <nick> is on last.fm. if <nick> is empty, show who you are on lastfm.")
105     when :compare
106       _("lastfm compare [<nick1>] <nick2> => show musical taste compatibility between nick1 (or user if omitted) and nick2")
107     when :shouts
108       _("lastfm shouts [<nick>] => show shouts to <nick>")
109     when :friends
110       _("lastfm friends [<nick>] => show <nick>'s friends")
111     when :neighbors, :neighbours
112       _("lastfm neighbors [<nick>] => show people who share similar musical taste as <nick>")
113     when :lovedtracks
114       _("lastfm loved[tracks] [<nick>] => show tracks that <nick> has loved")
115     when :recenttracks, :recentracks
116       _("lastfm recent[tracks] [<nick>] => show tracks that <nick> has recently played")
117     when :topalbums
118       _("lastfm topalbums [<nick>] [over <period>] => show <nick>'s top albums%{p}") % { :p => period }
119     when :topartists
120       _("lastfm topartists [<nick>] [over <period>] => show <nick>'s top artists%{p}") % { :p => period }
121     when :toptracks
122       _("lastfm toptracks [<nick>] [over <period>] => show <nick>'s top tracks%{p}") % { :p => period }
123     when :weeklyalbumchart
124       _("lastfm weeklyalbumchart [<nick>] => show <nick>'s weekly album chart")
125     when :weeklyartistchart
126       _("lastfm weeklyartistchart [<nick>] => show <nick>'s weekly artist chart")
127     when :weeklytrackchart
128       _("lastfm weeklyartistchart [<nick>] => show <nick>'s weekly track chart")
129     else
130       _("last.fm plugin - topics: events, artist, album, track, now, set, who, compare, shouts, friends, neighbors, (loved|recent)tracks, top(albums|tracks|artists), weekly(album|artist|track)chart")
131     end
132   end
133
134   # TODO allow searching by country etc.
135   #
136   # Options: name, limit
137   def search_venue_by(options)
138     params = {}
139     params[:venue] = CGI.escape(options[:name])
140     options.delete(:name)
141     params.merge!(options)
142
143     uri = "#{APIURL}method=venue.search&"
144     uri << params.to_a.map {|e| e.join("=")}.join("&")
145
146     xml = @bot.httputil.get_response(uri)
147     doc = Document.new xml.body
148     results = []
149
150     doc.root.elements.each("results/venuematches/venue") do |v|
151       venue = LastFmVenue.new
152       venue.id      = v.elements["id"].text.to_i
153       venue.url     = v.elements["url"].text
154       venue.lat     = v.elements["location/geo:point/geo:lat"].text.to_f
155       venue.long    = v.elements["location/geo:point/geo:long"].text.to_f
156       venue.name    = v.elements["name"].text
157       venue.city    = v.elements["location/city"].text
158       venue.street  = v.elements["location/street"].text
159       venue.postal  = v.elements["location/postalcode"].text
160       venue.country = v.elements["location/country"].text
161
162       results << venue
163     end
164     results
165   end
166
167   def find_events(m, params)
168     num = params[:num] || @bot.config['lastfm.default_events']
169     num = num.to_i.clip(1, @bot.config['lastfm.max_events'])
170
171     location = params[:location]
172     artist = params[:who]
173     venue = params[:venue]
174     user = resolve_username(m, params[:user])
175
176     if location
177       uri = "#{APIURL}method=geo.getevents&location=#{CGI.escape location.to_s}"
178       emptymsg = _("no events found in %{location}") % {:location => location.to_s}
179     elsif venue
180       venues = search_venue_by(:name => venue.to_s, :limit => 1)
181       venue  = venues.first
182       uri = "#{APIURL}method=venue.getevents&venue=#{venue.id}"
183     elsif artist
184       uri = "#{APIURL}method=artist.getevents&artist=#{CGI.escape artist.to_s}"
185       emptymsg = _("no events found by %{artist}") % {:artist => artist.to_s}
186     elsif user
187       uri = "#{APIURL}method=user.getevents&user=#{CGI.escape user}"
188       emptymsg = _("%{user} is not attending any events") % {:user => user}
189     end
190     xml = @bot.httputil.get_response(uri)
191
192     doc = Document.new xml.body
193     if xml.class == Net::HTTPInternalServerError
194       if doc.root and doc.root.attributes["status"] == "failed"
195         m.reply doc.root.elements["error"].text
196       else
197         m.reply _("could not retrieve events")
198       end
199     end
200     disp_events = Array.new
201     events = Array.new
202     doc.root.elements.each("events/event"){ |e|
203       h = {}
204       h[:title] = e.elements["title"].text
205       venue = e.elements["venue"].elements["name"].text
206       city = e.elements["venue"].elements["location"].elements["city"].text
207       country =  e.elements["venue"].elements["location"].elements["country"].text
208       h[:location] = Underline + venue + Underline + " #{Bold + city + Bold}, #{country}"
209       date = e.elements["startDate"].text.split
210       h[:date] = Time.utc(date[3].to_i, date[2], date[1].to_i)
211       h[:desc] = e.elements["description"].text
212       h[:url] = e.elements["url"].text
213       h[:attendance] = e.elements["attendance"].text.to_i
214       artists = Array.new
215       e.elements.each("artists/artist"){ |a|
216         artists << a.text
217       }
218       h[:artists] = artists
219       events << LastFmEvent.new(h)
220     }
221     if events.empty?
222       m.reply emptymsg
223       return
224     end
225     events[0...num].each { |event|
226       disp_events << event.to_s
227     }
228     m.reply disp_events.join(' | '), :split_at => /\s+\|\s+/
229
230   end
231
232   def tasteometer(m, params)
233     opts = { :cache => false }
234     user1 = resolve_username(m, params[:user1])
235     user2 = resolve_username(m, params[:user2])
236     xml = @bot.httputil.get_response("#{APIURL}method=tasteometer.compare&type1=user&type2=user&value1=#{CGI.escape user1}&value2=#{CGI.escape user2}", opts)
237     doc = Document.new xml.body
238     unless doc
239       m.reply _("last.fm parsing failed")
240       return
241     end
242     if xml.class == Net::HTTPBadRequest
243       if doc.root.elements["error"].attributes["code"] == "7" then
244         error = doc.root.elements["error"].text
245         error.match(/Invalid username: \[(.*)\]/);
246         baduser = $1
247
248         m.reply _("%{u} doesn't exist on last.fm") % {:u => baduser}
249         return
250       else
251         m.reply _("error: %{e}") % {:e => doc.root.element["error"].text}
252         return
253       end
254     end
255     score = doc.root.elements["comparison/result/score"].text.to_f
256     artists = doc.root.get_elements("comparison/result/artists/artist").map { |e| e.elements["name"].text}
257     case
258       when score >= 0.9
259         rating = _("Super")
260       when score >= 0.7
261         rating = _("Very High")
262       when score >= 0.5
263         rating = _("High")
264       when score >= 0.3
265         rating = _("Medium")
266       when score >= 0.1
267         rating = _("Low")
268       else
269         rating = _("Very Low")
270     end
271
272     reply = _("%{a}'s and %{b}'s musical compatibility rating is %{bold}%{r}%{bold}") % {
273       :a => user1,
274       :b => user2,
275       :r => rating.downcase,
276       :bold => Bold
277     }
278
279     reply << _(" and music they have in common includes: %{artists}") % {
280       :artists => artists.join(", ")
281     } unless artists.empty?
282
283     m.reply reply
284   end
285
286   def now_playing(m, params)
287     opts = { :cache => false }
288     user = resolve_username(m, params[:who])
289     xml = @bot.httputil.get_response("#{APIURL}method=user.getrecenttracks&user=#{CGI.escape user}", opts)
290     doc = Document.new xml.body
291     unless doc
292       m.reply _("last.fm parsing failed")
293       return
294     end
295     if xml.class == Net::HTTPBadRequest
296       if doc.root.elements["error"].text == "Invalid user name supplied" then
297         m.reply _("%{user} doesn't exist on last.fm, perhaps they need to: lastfm user <username>") % {
298           :user => user
299         }
300         return
301       else
302         m.reply _("error: %{e}") % {:e => doc.root.element["error"].text}
303         return
304       end
305     end
306     now = artist = track = albumtxt = date = nil
307     unless doc.root.elements[1].has_elements?
308      m.reply _("%{u} hasn't played anything recently") % {:u => user}
309      return
310     end
311     first = doc.root.elements[1].elements[1]
312     now = first.attributes["nowplaying"]
313     artist = first.elements["artist"].text
314     track = first.elements["name"].text
315     albumtxt = first.elements["album"].text
316     album = ""
317     if albumtxt
318       year = get_album(artist, albumtxt)[2]
319       album = "[#{albumtxt}, #{year}]" if year
320     end
321     past = nil
322     date = XPath.first(first, "//date")
323     if date != nil
324       time = date.attributes["uts"]
325       past = Time.at(time.to_i)
326     end
327     if now == "true"
328        verb = _("is listening to")
329        if @registry.has_key? "#{m.sourcenick}_verb_present"
330          verb = @registry["#{m.sourcenick}_verb_present"]
331        end
332        reply = _("%{u} %{v} \"%{t}\" by %{a} %{b}") % {:u => user, :v => verb, :t => track, :a => artist, :b => album}
333     else
334       verb = _("listened to")
335        if @registry.has_key? "#{m.sourcenick}_verb_past"
336          verb = @registry["#{m.sourcenick}_verb_past"]
337        end
338       ago = Utils.timeago(past)
339       reply = _("%{u} %{v} \"%{t}\" by %{a} %{b} %{p}") % {:u => user, :v => verb, :t => track, :a => artist, :b => album, :p => ago}
340     end
341
342     reply << _("; see %{uri} for more") % { :uri => "http://www.last.fm/user/#{CGI.escape user}"}
343     m.reply reply
344   end
345
346   def find_artist(m, params)
347     info_xml = @bot.httputil.get("#{APIURL}method=artist.getinfo&artist=#{CGI.escape params[:artist].to_s}")
348     unless info_xml
349       m.reply _("I had problems getting info for %{a}") % {:a => params[:artist]}
350       return
351     end
352     info_doc = Document.new info_xml
353     unless info_doc
354       m.reply _("last.fm parsing failed")
355       return
356     end
357     tags_xml = @bot.httputil.get("#{APIURL}method=artist.gettoptags&artist=#{CGI.escape params[:artist].to_s}")
358     tags_doc = Document.new tags_xml
359
360     first = info_doc.root.elements["artist"]
361     artist = first.elements["name"].text
362     url = first.elements["url"].text
363     stats = {}
364     %w(playcount listeners).each do |e|
365       t = first.elements["stats/#{e}"].text
366       stats[e.to_sym] = t.gsub(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1,")
367     end
368     summary = first.elements["bio"].elements["summary"].text
369     similar = first.get_elements("similar/artist").map { |a|
370       _("%{b}%{a}%{b}") % { :a => a.elements["name"].text, :b => Bold } }
371     tags = tags_doc.root.get_elements("toptags/tag")[0..4].map { |t|
372       _("%{u}%{t}%{u}") % { :t => t.elements["name"].text, :u => Underline } }
373     reply = _("%{b}%{a}%{b} <%{u}> has been played %{b}%{c}%{b} times and is being listened to by %{b}%{l}%{b} people") % {
374       :b => Bold, :a => artist, :u => url, :c => stats[:playcount], :l => stats[:listeners] }
375     reply << _(". Tagged as: %{t}") % {
376       :t => tags.join(", "), :b => Bold } unless tags.empty?
377     reply << _(". Similar artists: %{s}") % {
378       :s => similar.join(", "), :b => Bold } unless similar.empty?
379     m.reply reply
380     m.reply summary.ircify_html
381   end
382
383   def find_track(m, params)
384     track = params[:track].to_s
385     xml = @bot.httputil.get("#{APIURL}method=track.search&track=#{CGI.escape track}")
386     unless xml
387       m.reply _("I had problems getting info for %{a}") % {:a => track}
388       return
389     end
390     debug xml
391     doc = Document.new xml
392     unless doc
393       m.reply _("last.fm parsing failed")
394       return
395     end
396     debug doc.root
397     results = doc.root.elements["results/opensearch:totalResults"].text.to_i rescue 0
398     if results > 0
399       begin
400         hits = []
401         doc.root.each_element("results/trackmatches/track") do |track|
402           hits << _("%{bold}%{t}%{bold} by %{bold}%{a}%{bold} (%{n} listeners)") % {
403             :t => track.elements["name"].text,
404             :a => track.elements["artist"].text,
405             :n => track.elements["listeners"].text,
406             :bold => Bold
407           }
408         end
409         m.reply hits.join(' -- '), :split_at => ' -- '
410       rescue
411         error $!
412         m.reply _("last.fm parsing failed")
413       end
414     else
415       m.reply _("track %{a} not found") % {:a => track}
416     end
417   end
418
419   def find_venue(m, params)
420     venue  = params[:venue].to_s
421     venues = search_venue_by(:name => venue, :limit => 1)
422     venue  = venues.last
423
424     if venues.empty?
425       m.reply "sorry, can't find such venue"
426       return
427     end
428
429     reply = _("%{b}%{name}%{b}, %{street}, %{u}%{city}%{u}, %{country}, see %{url} for more info") % {
430       :u => Underline, :b => Bold, :name => venue.name, :city => venue.city, :street => venue.street,
431       :country => venue.country, :url => venue.url
432     }
433
434     if venue.street && venue.city
435       maps_uri = "http://maps.google.com/maps?q=#{venue.street},+#{venue.city}"
436       maps_uri << ",+#{venue.postal}" if venue.postal
437     elsif venue.lat && venue.long
438       maps_uri = "http://maps.google.com/maps?q=#{venue.lat},+#{venue.long}"
439     else
440       m.reply reply
441       return
442     end
443
444     maps_uri << "+(#{venue.name.gsub(" ", "%A0")})"
445
446     begin
447       require "shorturl"
448       maps_uri = ShortURL.shorten(CGI.escape(maps_uri))
449     rescue LoadError => e
450       error e
451     end
452
453     reply << _(" and %{maps} for maps") % { :maps => maps_uri, :b => Bold }
454     m.reply reply
455   end
456
457   def get_album(artist, album)
458     xml = @bot.httputil.get("#{APIURL}method=album.getinfo&artist=#{CGI.escape artist}&album=#{CGI.escape album}")
459     unless xml
460       return [_("I had problems getting album info")]
461     end
462     doc = Document.new xml
463     unless doc
464       return [_("last.fm parsing failed")]
465     end
466     album = date = playcount = artist = date = year = nil
467     first = doc.root.elements["album"]
468     artist = first.elements["artist"].text
469     playcount = first.elements["playcount"].text
470     album = first.elements["name"].text
471     date = first.elements["releasedate"].text
472     unless date.strip.length < 2
473       year = date.strip.split[2].chop
474     end
475     result = [artist, album, year, playcount]
476     return result
477   end
478
479   def find_album(m, params)
480     album = get_album(params[:artist].to_s, params[:album].to_s)
481     if album.length == 1
482       m.reply _("I couldn't locate: \"%{a}\" by %{r}") % {:a => params[:album], :r => params[:artist]}
483       return
484     end
485     year = "(#{album[2]}) " unless album[2] == nil
486     m.reply _("the album \"%{a}\" by %{r} %{y}has been played %{c} times") % {:a => album[1], :r => album[0], :y => year, :c => album[3]}
487   end
488
489   def set_user(m, params)
490     user = params[:who].to_s
491     nick = m.sourcenick
492     @registry[ nick ] = user
493     m.reply _("okay, I'll remember that %{n} is %{u} on last.fm") % {:n => nick, :u => user}
494   end
495
496   def set_verb(m, params)
497     past = params[:past].to_s
498     present = params[:present].to_s
499     key = "#{m.sourcenick}_verb_"
500     @registry[ "#{key}past" ] = past
501     @registry[ "#{key}present" ] = present
502     m.reply _("okay, I'll remember that %{n} prefers \"%{r}\" and \"%{p}\"") % {:n => m.sourcenick, :p => past, :r => present}
503   end
504
505   def get_user(m, params)
506     nick = ""
507     if params[:who]
508       nick = params[:who].to_s
509     else
510       nick = m.sourcenick
511     end
512     if @registry.has_key? nick
513       user = @registry[ nick ]
514       m.reply _("%{nick} is %{user} on last.fm") % {
515         :nick => nick,
516         :user => user
517       }
518     else
519       m.reply _("sorry, I don't know who %{n} is on last.fm, perhaps they need to: lastfm set user <username>") % {:n => nick}
520     end
521   end
522
523   def lastfm(m, params)
524     action = case params[:action]
525     when "neighbors" then "neighbours"
526     when "recentracks", "recent" then "recenttracks"
527     when "loved" then "lovedtracks"
528     when /^weekly(track|album|artist)s$/
529       "weekly#{$1}chart"
530     when "events"
531       find_events(m, params)
532       return
533     else
534       params[:action]
535     end.to_sym
536
537     if action == :shouts
538       num = params[:num] || @bot.config['lastfm.default_shouts']
539       num = num.to_i.clip(1, @bot.config['lastfm.max_shouts'])
540     else
541       num = params[:num] || @bot.config['lastfm.default_user_data']
542       num = num.to_i.clip(1, @bot.config['lastfm.max_user_data'])
543     end
544
545     user = resolve_username(m, params[:user])
546     uri = "#{APIURL}method=user.get#{action}&user=#{CGI.escape user}"
547
548     if period = params[:period]
549       period_uri = (period.last == "year" ? "12month" : period.first + "month")
550       uri << "&period=#{period_uri}"
551     end
552
553     begin
554       res = @bot.httputil.get_response(uri)
555       raise _("no response body") unless res.body
556     rescue Exception => e
557         m.reply _("I had problems accessing last.fm: %{e}") % {:e => e.message}
558         return
559     end
560     doc = Document.new(res.body)
561     unless doc
562       m.reply _("last.fm parsing failed")
563       return
564     end
565
566     case res
567     when Net::HTTPBadRequest
568       if doc.root and doc.root.attributes["status"] == "failed"
569         m.reply "error: " << doc.root.elements["error"].text.downcase
570       end
571       return
572     end
573
574     seemore =  _("; see %{uri} for more")
575     case action
576     when :friends
577       friends = doc.root.get_elements("friends/user").map do |u|
578         u.elements["name"].text
579       end
580
581       if friends.empty?
582         reply = _("%{user} has no friends :(")
583       elsif friends.length <= num
584         reply = _("%{user} has %{total} friends: %{friends}")
585       else
586         reply = _("%{user} has %{total} friends, including %{friends}")
587         reply << seemore
588       end
589       m.reply reply % {
590         :user => user,
591         :total => friends.size,
592         :friends => friends.shuffle[0, num].join(", "),
593         :uri => "http://www.last.fm/user/#{CGI.escape user}/friends"
594       }
595     when :lovedtracks
596       loved = doc.root.get_elements("lovedtracks/track").map do |track|
597         [track.elements["artist/name"].text, track.elements["name"].text].join(" - ")
598       end
599       loved_prep = loved.shuffle[0, num].to_enum(:each_with_index).collect { |e,i| (i % 2).zero? ? Underline+e+Underline : e }
600
601       if loved.empty?
602         reply = _("%{user} has not loved any tracks")
603       elsif loved.length <= num
604         reply = _("%{user} has loved %{total} tracks: %{tracks}")
605       else
606         reply = _("%{user} has loved %{total} tracks, including %{tracks}")
607         reply << seemore
608       end
609       m.reply reply % {
610           :user => user,
611           :total => loved.size,
612           :tracks => loved_prep.join(", "),
613           :uri => "http://www.last.fm/user/#{CGI.escape user}/library/loved"
614         }
615     when :neighbours
616       nbrs = doc.root.get_elements("neighbours/user").map do |u|
617         u.elements["name"].text
618       end
619
620       if nbrs.empty?
621         reply = _("no one seems to share %{user}'s musical taste")
622       elsif nbrs.length <= num
623         reply = _("%{user}'s musical neighbours are %{nbrs}")
624       else
625         reply = _("%{user}'s musical neighbours include %{nbrs}")
626         reply << seemore
627       end
628       m.reply reply % {
629           :user  => user,
630           :nbrs  => nbrs.shuffle[0, num].join(", "),
631           :uri   => "http://www.last.fm/user/#{CGI.escape user}/neighbours"
632       }
633     when :recenttracks
634       tracks = doc.root.get_elements("recenttracks/track").map do |track|
635         [track.elements["artist"].text, track.elements["name"].text].join(" - ")
636       end
637
638       counts = []
639       tracks.each do |track|
640         if t = counts.assoc(track)
641           counts[counts.rindex(t)] = [track, t[-1] += 1]
642         else
643           counts << [track, 1]
644         end
645       end
646
647       tracks_prep = counts[0, num].to_enum(:each_with_index).map do |e,i|
648         str = (i % 2).zero? ? Underline+e[0]+Underline : e[0]
649         str << " (%{i} times%{m})" % {
650           :i => e.last,
651           :m => counts.size == 1 ? _(" or more") : nil
652         } if e.last > 1
653         str
654       end
655
656       if tracks.empty?
657         m.reply _("%{user} hasn't played anything recently") % { :user => user }
658       else
659         m.reply _("%{user} has recently played %{tracks}") %
660           { :user => user, :tracks => tracks_prep.join(", ") }
661       end
662     when :shouts
663       shouts = doc.root.get_elements("shouts/shout")
664       if shouts.empty?
665         m.reply _("there are no shouts for %{user}") % { :user => user }
666       else
667         shouts[0, num].each do |shout|
668           m.reply _("<%{author}> %{body}") % {
669             :body   => shout.elements["body"].text,
670             :author => shout.elements["author"].text,
671           }
672         end
673       end
674     when :toptracks, :topalbums, :topartists, :weeklytrackchart, :weeklyalbumchart, :weeklyartistchart
675       type  = action.to_s.scan(/track|album|artist/).to_s
676       items = doc.root.get_elements("#{action}/#{type}").map do |item|
677         case action
678         when :weeklytrackchart, :weeklyalbumchart
679           format = "%{artist} - %{title} (%{bold}%{plays}%{bold})"
680           artist = item.elements["artist"].text
681         when :weeklyartistchart, :topartists
682           format = "%{artist} (%{bold}%{plays}%{bold})"
683           artist = item.elements["name"].text
684         when :toptracks, :topalbums
685           format = "%{artist} - %{title} (%{bold}%{plays}%{bold})"
686           artist = item.elements["artist/name"].text
687         end
688
689         _(format) % {
690           :artist => artist,
691           :title  => item.elements["name"].text,
692           :plays  => item.elements["playcount"].text,
693           :bold   => Bold
694         }
695       end
696       if items.empty?
697         m.reply _("%{user} hasn't played anything in this period of time") % { :user => user }
698       else
699         m.reply items[0, num].join(", ")
700       end
701     end
702   end
703
704   def resolve_username(m, name)
705     name = m.sourcenick if name.nil?
706     @registry[name] or name
707   end
708 end
709
710 plugin = LastFmPlugin.new
711 plugin.map 'lastfm [:num] event[s] in *location', :action => :find_events, :requirements => { :num => /\d+/ }, :thread => true
712 plugin.map 'lastfm [:num] event[s] by *who', :action => :find_events, :requirements => { :num => /\d+/ }, :thread => true
713 plugin.map 'lastfm [:num] event[s] at *venue', :action => :find_events, :requirements => { :num => /\d+/ }, :thread => true
714 plugin.map 'lastfm [:num] event[s] [for] *who', :action => :find_events, :requirements => { :num => /\d+/ }, :thread => true
715 plugin.map 'lastfm artist *artist', :action => :find_artist, :thread => true
716 plugin.map 'lastfm album *album [by *artist]', :action => :find_album
717 plugin.map 'lastfm track *track', :action => :find_track, :thread => true
718 plugin.map 'lastfm venue *venue', :action => :find_venue, :thread => true
719 plugin.map 'lastfm set user[name] :who', :action => :set_user, :thread => true
720 plugin.map 'lastfm set verb *present, *past', :action => :set_verb, :thread => true
721 plugin.map 'lastfm who [:who]', :action => :get_user, :thread => true
722 plugin.map 'lastfm compare to :user2', :action => :tasteometer, :thread => true
723 plugin.map 'lastfm compare [:user1] [to] :user2', :action => :tasteometer, :thread => true
724 plugin.map "lastfm [user] [:num] :action [:user]", :thread => true,
725   :requirements => {
726     :action => /^(?:events|shouts|friends|neighbou?rs|loved(?:tracks)?|recent(?:t?racks)?|top(?:album|artist|track)s?|weekly(?:albums?|artists?|tracks?)(?:chart)?)$/,
727     :num => /^\d+$/
728 }
729 plugin.map 'lastfm [user] [:num] :action [:user] over [*period]', :thread => true,
730   :requirements => {
731     :action => /^(?:top(?:album|artist|track)s?)$/,
732     :period => /^(?:(?:3|6|12) months)|(?:a\s|1\s)?year$/,
733     :num => /^\d+$/
734 }
735 plugin.map 'lastfm [now] [:who]', :action => :now_playing, :thread => true
736 plugin.map 'np [:who]', :action => :now_playing, :thread => true