X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Flastfm.rb;h=57cf3ec14596078c0aec421f3eecb2a7b5d758c6;hb=bdaf120460c0d0b6be3a037beebf3bc05d8746fe;hp=67f8d5f78852d7eebb2ee3b8999baaabb0514d78;hpb=2b81b5d6ec06fbf9e32300eeb9caece8c26d33b7;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git diff --git a/data/rbot/plugins/lastfm.rb b/data/rbot/plugins/lastfm.rb index 67f8d5f7..57cf3ec1 100644 --- a/data/rbot/plugins/lastfm.rb +++ b/data/rbot/plugins/lastfm.rb @@ -6,10 +6,12 @@ # Author:: Jeremy Voorhis # Author:: Giuseppe "Oblomov" Bilotta # Author:: Casey Link +# Author:: Raine Virta # # Copyright:: (C) 2005 Jeremy Voorhis # Copyright:: (C) 2007 Giuseppe Bilotta # Copyright:: (C) 2008 Casey Link +# Copyright:: (C) 2009 Raine Virta # # License:: GPL v2 @@ -17,6 +19,8 @@ require 'rexml/document' require 'cgi' class ::LastFmEvent + attr_reader :attendance, :date + def initialize(hash) @url = hash[:url] @date = hash[:date] @@ -36,15 +40,26 @@ class ::LastFmEvent end def compact_display - if @attendance - return "%s %s @ %s (%s attending) %s" % [@date.strftime("%a %b, %d %Y"), @artist_string, @location, @attendance, @url] + unless @attendance.zero? + return "%s %s @ %s (%s attending) %s" % [@date.strftime("%a, %b %d"), @artist_string, @location, @attendance, @url] end - return "%s %s @ %s %s" % [@date.strftime("%a %b, %d %Y"), @artist_string, @location, @url] + return "%s %s @ %s %s" % [@date.strftime("%a, %b %d"), @artist_string, @location, @url] end alias :to_s :compact_display end +define_structure :LastFmVenue, :id, :city, :street, :postal, :country, :name, :url, :lat, :long +class ::Struct::LastFmVenue + def to_s + str = self.name.dup + if self.country + str << " (" << [self.city, self.country].compact.join(", ") << ")" + end + str + end +end + class LastFmPlugin < Plugin include REXML Config.register Config::IntegerValue.new('lastfm.max_events', @@ -53,6 +68,18 @@ class LastFmPlugin < Plugin Config.register Config::IntegerValue.new('lastfm.default_events', :default => 3, :validate => Proc.new{|v| v > 1}, :desc => "Default number of events to display.") + Config.register Config::IntegerValue.new('lastfm.max_shouts', + :default => 5, :validate => Proc.new{|v| v > 1}, + :desc => "Maximum number of user shouts to display.") + Config.register Config::IntegerValue.new('lastfm.default_shouts', + :default => 3, :validate => Proc.new{|v| v > 1}, + :desc => "Default number of user shouts to display.") + Config.register Config::IntegerValue.new('lastfm.max_user_data', + :default => 25, :validate => Proc.new{|v| v > 1}, + :desc => "Maximum number of user data entries (except events and shouts) to display.") + Config.register Config::IntegerValue.new('lastfm.default_user_data', + :default => 10, :validate => Proc.new{|v| v > 1}, + :desc => "Default number of user data entries (except events and shouts) to display.") APIKEY = "b25b959554ed76058ac220b7b2e0a026" APIURL = "http://ws.audioscrobbler.com/2.0/?api_key=#{APIKEY}&" @@ -70,9 +97,10 @@ class LastFmPlugin < Plugin end def help(plugin, topic="") + period = _(", where can be one of: 3|6|12 months, a year") 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 %{d} and cannot be higher than %{m}") % {:d => @bot.config['lastfm.default_events'], :m => @bot.config['lastfm.max_events']} + _("lastfm [] events in => show information on events in or near . lastfm [] events by => show information on events by . lastfm [] events at => show information on events at specific . The number of events that can be displayed is optional, defaults to %{d} and cannot be higher than %{m}. Append 'sort by [in order]' to sort events. Events can be sorted by attendance or date (default) in ascending or descending order.") % {:d => @bot.config['lastfm.default_events'], :m => @bot.config['lastfm.max_events']} when :artist _("lastfm artist => show information on artist from last.fm") when :album @@ -80,31 +108,113 @@ class LastFmPlugin < Plugin when :track _("lastfm track => search tracks matching on last.fm") when :now, :np - _("lastfm now [] => show the now playing track from last.fm. np [] does the same.") + _("lastfm now [] => show the now playing track from last.fm. np [] does the same.") when :set - _("lastfm set user => associate your current irc nick with a last.fm user. lastfm set verb => set your preferred now playing verb. default \"listening\" and \"listened\".") + _("lastfm set user => associate your current irc nick with a last.fm user. lastfm set verb , => set your preferred now playing/just played verbs. default \"is listening to\" and \"listened to\".") when :who - _("lastfm who [] => show who is at last.fm. if is empty, show who you are at lastfm.") + _("lastfm who [] => show who is on last.fm. if is empty, show who you are on lastfm.") when :compare - _("lastfm compare => show musical taste compatibility between nick1 and nick2.") + _("lastfm compare [] => show musical taste compatibility between nick1 (or user if omitted) and nick2") + when :shouts + _("lastfm shouts [] => show shouts to ") + when :friends + _("lastfm friends [] => show 's friends") + when :neighbors, :neighbours + _("lastfm neighbors [] => show people who share similar musical taste as ") + when :lovedtracks + _("lastfm loved[tracks] [] => show tracks that has loved") + when :recenttracks, :recentracks + _("lastfm recent[tracks] [] => show tracks that has recently played") + when :topalbums + _("lastfm topalbums [] [over ] => show 's top albums%{p}") % { :p => period } + when :topartists + _("lastfm topartists [] [over ] => show 's top artists%{p}") % { :p => period } + when :toptracks + _("lastfm toptracks [] [over ] => show 's top tracks%{p}") % { :p => period } + when :weeklyalbumchart + _("lastfm weeklyalbumchart [] => show 's weekly album chart") + when :weeklyartistchart + _("lastfm weeklyartistchart [] => show 's weekly artist chart") + when :weeklytrackchart + _("lastfm weeklyartistchart [] => show 's weekly track chart") else - _("lastfm [] => show your or 's now playing track at lastfm. np [] => same as 'lastfm'. other topics: events, artist, album, track, now, set, who, compare") + _("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") end end + # TODO allow searching by country etc. + # + # Options: name, limit + def search_venue_by(options) + params = {} + params[:venue] = CGI.escape(options[:name]) + options.delete(:name) + params.merge!(options) + + uri = "#{APIURL}method=venue.search&" + uri << params.to_a.map {|e| e.join("=")}.join("&") + + xml = @bot.httputil.get_response(uri) + doc = Document.new xml.body + results = [] + + doc.root.elements.each("results/venuematches/venue") do |v| + venue = LastFmVenue.new + venue.id = v.elements["id"].text.to_i + venue.url = v.elements["url"].text + venue.lat = v.elements["location/geo:point/geo:lat"].text.to_f + venue.long = v.elements["location/geo:point/geo:long"].text.to_f + venue.name = v.elements["name"].text + venue.city = v.elements["location/city"].text + venue.street = v.elements["location/street"].text + venue.postal = v.elements["location/postalcode"].text + venue.country = v.elements["location/country"].text + + results << venue + end + results + end + def find_events(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] + sort_by = params[:sort_by] || :date + sort_order = params[:sort_order] + sort_order = sort_order.to_sym unless sort_order.nil? - uri = nil - if artist == nil - uri = "#{APIURL}method=geo.getevents&location=#{CGI.escape location}" - else - uri = "#{APIURL}method=artist.getevents&artist=#{CGI.escape artist}" + location = params[:location] + artist = params[:who] + venue = params[:venue] + user = resolve_username(m, params[:user]) + + if location + uri = "#{APIURL}method=geo.getevents&location=#{CGI.escape location.to_s}" + emptymsg = _("no events found in %{location}") % {:location => location.to_s} + elsif venue + begin + venues = search_venue_by(:name => venue.to_s, :limit => 1) + rescue Exception => e + error e + m.reply _("an error occurred looking for venue %{venue}: %{e}") % { + :venue => venue.to_s, + :e => e.message + } + end + + if venues.empty? + m.reply _("no venue found matching %{venue}") % {:venue => venue.to_s} + return + end + venue = venues.first + uri = "#{APIURL}method=venue.getevents&venue=#{venue.id}" + emptymsg = _("no events found at %{venue}") % {:venue => venue.to_s} + elsif artist + uri = "#{APIURL}method=artist.getevents&artist=#{CGI.escape artist.to_s}" + emptymsg = _("no events found by %{artist}") % {:artist => artist.to_s} + elsif user + uri = "#{APIURL}method=user.getevents&user=#{CGI.escape user}" + emptymsg = _("%{user} is not attending any events") % {:user => user} end xml = @bot.httputil.get_response(uri) @@ -113,7 +223,7 @@ class LastFmPlugin < Plugin if doc.root and doc.root.attributes["status"] == "failed" m.reply doc.root.elements["error"].text else - m.reply _("Could not retrieve events") + m.reply _("could not retrieve events") end end disp_events = Array.new @@ -129,11 +239,7 @@ class LastFmPlugin < Plugin h[:date] = Time.utc(date[3].to_i, date[2], date[1].to_i) h[:desc] = e.elements["description"].text h[:url] = e.elements["url"].text - e.detect {|node| - if node.kind_of? Element and node.attributes["name"] == "attendance" then - h[:attendance] = node.text - end - } + h[:attendance] = e.elements["attendance"].text.to_i artists = Array.new e.elements.each("artists/artist"){ |a| artists << a.text @@ -141,6 +247,22 @@ class LastFmPlugin < Plugin h[:artists] = artists events << LastFmEvent.new(h) } + if events.empty? + m.reply emptymsg + return + end + + # sort order when sorted by date is ascending by default + # and descending when sorted by attendance + case sort_by.to_sym + when :attendance + events = events.sort_by { |e| e.attendance }.reverse + events.reverse! if [:ascending, :asc].include? sort_order + when :date + events = events.sort_by { |e| e.date } + events.reverse! if [:descending, :desc].include? sort_order + end + events[0...num].each { |event| disp_events << event.to_s } @@ -150,38 +272,29 @@ class LastFmPlugin < Plugin def tasteometer(m, params) opts = { :cache => false } - user1 = params[:user1].to_s - user2 = params[:user2].to_s + user1 = resolve_username(m, params[:user1]) + user2 = resolve_username(m, params[:user2]) xml = @bot.httputil.get_response("#{APIURL}method=tasteometer.compare&type1=user&type2=user&value1=#{CGI.escape user1}&value2=#{CGI.escape user2}", opts) doc = Document.new xml.body unless doc m.reply _("last.fm parsing failed") return end - if xml.class == Net::HTTPInternalServerError + if xml.class == Net::HTTPBadRequest if doc.root.elements["error"].attributes["code"] == "7" then error = doc.root.elements["error"].text error.match(/Invalid username: \[(.*)\]/); - if @registry.has_key? $1 and not params[:recurs] - if user1 == $1 - params[:user1] = @registry[ $1 ] - elsif user2 == $1 - params[:user2] = @registry[ $1 ] - end - params[:recurs] = true - tasteometer(m, params) - else - m.reply _("%{u} doesn't exist at last.fm. Perhaps you need to: lastfm set ") % {:u => baduser} - return - end + baduser = $1 + + m.reply _("%{u} doesn't exist on last.fm") % {:u => baduser} + return else - m.reply _("Bad: %{e}") % {:e => doc.root.element["error"].text} + m.reply _("error: %{e}") % {:e => doc.root.element["error"].text} return end end - now = artist = track = albumtxt = date = nil score = doc.root.elements["comparison/result/score"].text.to_f - rating = nil + artists = doc.root.get_elements("comparison/result/artists/artist").map { |e| e.elements["name"].text} case when score >= 0.9 rating = _("Super") @@ -196,19 +309,26 @@ class LastFmPlugin < Plugin else rating = _("Very Low") end - m.reply _("%{a}'s and %{b}'s musical compatibility rating is: %{r}") % {:a => user1, :b => user2, :r => rating} + + common_artists = unless artists.empty? + _(" and music they have in common includes: %{artists}") % { + :artists => Utils.comma_list(artists) } + else + nil + end + + m.reply _("%{a}'s and %{b}'s musical compatibility rating is %{bold}%{r}%{bold}%{common}") % { + :a => user1, + :b => user2, + :r => rating.downcase, + :bold => Bold, + :common => common_artists + } end def now_playing(m, params) opts = { :cache => false } - user = nil - if params[:who] - user = params[:who].to_s - elsif @registry.has_key? m.sourcenick - user = @registry[ m.sourcenick ] - else - user = m.sourcenick - end + user = resolve_username(m, params[:who]) xml = @bot.httputil.get_response("#{APIURL}method=user.getrecenttracks&user=#{CGI.escape user}", opts) doc = Document.new xml.body unless doc @@ -216,17 +336,13 @@ class LastFmPlugin < Plugin return end if xml.class == Net::HTTPBadRequest - if doc.root.elements["error"].text == "Invalid user name supplied" then - if @registry.has_key? user and not params[:recurs] - params[:who] = @registry[ user ] - params[:recurs] = true - now_playing(m, params) - else - m.reply "#{user} doesn't exist at last.fm. Perhaps you need to: lastfm set " - return - end + if doc.root.elements["error"].attributes["code"] == "6" then + m.reply _("%{user} doesn't exist on last.fm, perhaps they need to: lastfm user ") % { + :user => user + } + return else - m.reply _("Error %{e}") % {:e => doc.root.element["error"].text} + m.reply _("error: %{e}") % {:e => doc.root.element["error"].text} return end end @@ -240,10 +356,15 @@ class LastFmPlugin < Plugin artist = first.elements["artist"].text track = first.elements["name"].text albumtxt = first.elements["album"].text - album = "" - if albumtxt + album = if albumtxt year = get_album(artist, albumtxt)[2] - album = "[#{albumtxt}, #{year}] " if year + if year + _(" [%{albumtext}, %{year}]") % { :albumtext => albumtxt, :year => year } + else + _(" [%{albumtext}]") % { :albumtext => albumtxt } + end + else + nil end past = nil date = XPath.first(first, "//date") @@ -252,38 +373,64 @@ class LastFmPlugin < Plugin past = Time.at(time.to_i) end if now == "true" - verb = _("listening") + verb = _("is listening to") if @registry.has_key? "#{m.sourcenick}_verb_present" verb = @registry["#{m.sourcenick}_verb_present"] end - m.reply _("%{u} is %{v} to \"%{t}\" by %{a} %{b}") % {:u => user, :v => verb, :t => track, :a => artist, :b => album} + reply = _("%{u} %{v} \"%{t}\" by %{bold}%{a}%{bold}%{b}") % {:u => user, :v => verb, :t => track, :a => artist, :b => album, :bold => Bold} else - verb = _("listened") + verb = _("listened to") if @registry.has_key? "#{m.sourcenick}_verb_past" verb = @registry["#{m.sourcenick}_verb_past"] end ago = Utils.timeago(past) - m.reply _("%{u} %{v} to \"%{t}\" by %{a} %{b}%{p}") % {:u => user, :v => verb, :t => track, :a => artist, :b => album, :p => ago} + reply = _("%{u} %{v} \"%{t}\" by %{bold}%{a}%{bold}%{b} %{p};") % {:u => user, :v => verb, :t => track, :a => artist, :b => album, :p => ago, :bold => Bold} end + + if Object.const_defined?('Spotify') + if track = Spotify.search(:track, "#{artist} #{track}") + reply << _(" [%{u}%{url}%{u}]") % {:u => Underline, :url => track.url} + end + end + + reply << _(" -- see %{uri} for more") % { :uri => "http://www.last.fm/user/#{CGI.escape user}"} + m.reply reply end def find_artist(m, params) - xml = @bot.httputil.get("#{APIURL}method=artist.getinfo&artist=#{CGI.escape params[:artist].to_s}") - unless xml - m.reply _("I had problems getting info for %{a}.") % {:a => params[:artist]} + info_xml = @bot.httputil.get("#{APIURL}method=artist.getinfo&artist=#{CGI.escape params[:artist].to_s}") + unless info_xml + m.reply _("I had problems getting info for %{a}") % {:a => params[:artist]} return end - doc = Document.new xml - unless doc + info_doc = Document.new info_xml + unless info_doc m.reply _("last.fm parsing failed") return end - first = doc.root.elements["artist"] + tags_xml = @bot.httputil.get("#{APIURL}method=artist.gettoptags&artist=#{CGI.escape params[:artist].to_s}") + tags_doc = Document.new tags_xml + + first = info_doc.root.elements["artist"] artist = first.elements["name"].text - playcount = first.elements["stats"].elements["playcount"].text - listeners = first.elements["stats"].elements["listeners"].text + url = first.elements["url"].text + stats = {} + %w(playcount listeners).each do |e| + t = first.elements["stats/#{e}"].text + stats[e.to_sym] = t.gsub(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1,") + end summary = first.elements["bio"].elements["summary"].text - m.reply _("%{b}%{a}%{b} has been played %{c} times and is being listened to by %{l} people.") % {:b => Bold, :a => artist, :c => playcount, :l => listeners} + similar = first.get_elements("similar/artist").map { |a| + _("%{b}%{a}%{b}") % { :a => a.elements["name"].text, :b => Bold } } + tags = tags_doc.root.get_elements("toptags/tag")[0..4].map { |t| + _("%{u}%{t}%{u}") % { :t => t.elements["name"].text, :u => Underline } } + reply = _("%{b}%{a}%{b} <%{u}> has been played %{b}%{c}%{b} times and is being listened to by %{b}%{l}%{b} people") % { + :b => Bold, :a => artist, :u => url, :c => stats[:playcount], :l => stats[:listeners] } + reply << _(". Tagged as: %{t}") % { + :t => tags.join(", "), :b => Bold } unless tags.empty? + reply << _(". Similar artists: %{s}") % { + :s => similar.join(", "), :b => Bold } unless similar.empty? + m.reply reply m.reply summary.ircify_html end @@ -291,7 +438,7 @@ class LastFmPlugin < Plugin track = params[:track].to_s xml = @bot.httputil.get("#{APIURL}method=track.search&track=#{CGI.escape track}") unless xml - m.reply _("I had problems getting info for %{a}.") % {:a => track} + m.reply _("I had problems getting info for %{a}") % {:a => track} return end debug xml @@ -323,6 +470,44 @@ class LastFmPlugin < Plugin end end + def find_venue(m, params) + venue = params[:venue].to_s + venues = search_venue_by(:name => venue, :limit => 1) + venue = venues.last + + if venues.empty? + m.reply "sorry, can't find such venue" + return + end + + reply = _("%{b}%{name}%{b}, %{street}, %{u}%{city}%{u}, %{country}, see %{url} for more info") % { + :u => Underline, :b => Bold, :name => venue.name, :city => venue.city, :street => venue.street, + :country => venue.country, :url => venue.url + } + + if venue.street && venue.city + maps_uri = "http://maps.google.com/maps?q=#{venue.street},+#{venue.city}" + maps_uri << ",+#{venue.postal}" if venue.postal + elsif venue.lat && venue.long + maps_uri = "http://maps.google.com/maps?q=#{venue.lat},+#{venue.long}" + else + m.reply reply + return + end + + maps_uri << "+(#{venue.name.gsub(" ", "%A0")})" + + begin + require "shorturl" + maps_uri = ShortURL.shorten(CGI.escape(maps_uri)) + rescue LoadError => e + error e + end + + reply << _(" and %{maps} for maps") % { :maps => maps_uri, :b => Bold } + m.reply reply + end + def get_album(artist, album) xml = @bot.httputil.get("#{APIURL}method=album.getinfo&artist=#{CGI.escape artist}&album=#{CGI.escape album}") unless xml @@ -352,14 +537,14 @@ class LastFmPlugin < Plugin return end year = "(#{album[2]}) " unless album[2] == nil - m.reply _("The album \"%{a}\" by %{r} %{y}has been played %{c} times.") % {:a => album[1], :r => album[0], :y => year, :c => album[3]} + m.reply _("the album \"%{a}\" by %{r} %{y}has been played %{c} times") % {:a => album[1], :r => album[0], :y => year, :c => album[3]} end def set_user(m, params) user = params[:who].to_s nick = m.sourcenick @registry[ nick ] = user - m.reply _("Ok, I'll remember that %{n} is %{u} at last.fm") % {:n => nick, :u => user} + m.reply _("okay, I'll remember that %{n} is %{u} on last.fm") % {:n => nick, :u => user} end def set_verb(m, params) @@ -368,7 +553,7 @@ class LastFmPlugin < Plugin key = "#{m.sourcenick}_verb_" @registry[ "#{key}past" ] = past @registry[ "#{key}present" ] = present - m.reply _("Ok, I'll remember that %{n} prefers %{r} and %{p}.") % {:n => m.sourcenick, :p => past, :r => present} + m.reply _("okay, I'll remember that %{n} prefers \"%{r}\" and \"%{p}\"") % {:n => m.sourcenick, :p => past, :r => present} end def get_user(m, params) @@ -380,61 +565,233 @@ class LastFmPlugin < Plugin end if @registry.has_key? nick user = @registry[ nick ] - m.reply "#{nick} is #{user} at last.fm" + m.reply _("%{nick} is %{user} on last.fm") % { + :nick => nick, + :user => user + } else - m.reply _("Sorry, I don't know who %{n} is at last.fm perhaps you need to: lastfm set ") % {:n => nick} + m.reply _("sorry, I don't know who %{n} is on last.fm, perhaps they need to: lastfm set user ") % {:n => nick} end end - # TODO this user data retrieval should be upgraded to API 2.0 but it would need separate parsing - # for each dataset, or almost def lastfm(m, params) - action = params[:action].intern - action = :neighbours if action == :neighbors - action = :recenttracks if action == :recentracks - action = :topalbums if action == :topalbum - action = :topartists if action == :topartist - action = :toptags if action == :toptag - user = nil - if params[:user] then - user = params[:user].to_s - elsif @registry.has_key? m.sourcenick - user = @registry[ m.sourcenick ] + action = case params[:action] + when "neighbors" then "neighbours" + when "recentracks", "recent" then "recenttracks" + when "loved" then "lovedtracks" + when /^weekly(track|album|artist)s$/ + "weekly#{$1}chart" + when "events" + find_events(m, params) + return + else + params[:action] + end.to_sym + + if action == :shouts + num = params[:num] || @bot.config['lastfm.default_shouts'] + num = num.to_i.clip(1, @bot.config['lastfm.max_shouts']) else - # m.reply "I don't know who you are on last.fm. Use 'lastfm set username' to identify yourself." - # return - user = m.sourcenick + num = params[:num] || @bot.config['lastfm.default_user_data'] + num = num.to_i.clip(1, @bot.config['lastfm.max_user_data']) + end + + user = resolve_username(m, params[:user]) + uri = "#{APIURL}method=user.get#{action}&user=#{CGI.escape user}" + + if period = params[:period] + period_uri = (period.last == "year" ? "12month" : period.first + "month") + uri << "&period=#{period_uri}" 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?). perhaps you need to: lastfm set " + res = @bot.httputil.get_response(uri) + raise _("no response body") unless res.body + rescue Exception => e + m.reply _("I had problems accessing last.fm: %{e}") % {:e => e.message} + return end + doc = Document.new(res.body) + unless doc + m.reply _("last.fm parsing failed") + return + end + + case res + when Net::HTTPBadRequest + if doc.root and doc.root.attributes["status"] == "failed" + m.reply "error: " << doc.root.elements["error"].text.downcase + end + return + end + + seemore = _("; see %{uri} for more") + case action + when :friends + friends = doc.root.get_elements("friends/user").map do |u| + u.elements["name"].text + end + + if friends.empty? + reply = _("%{user} has no friends :(") + elsif friends.length <= num + reply = _("%{user} has %{total} friends: %{friends}") + else + reply = _("%{user} has %{total} friends, including %{friends}%{seemore}") + end + m.reply reply % { + :user => user, + :total => friends.size, + :friends => Utils.comma_list(friends.shuffle[0, num]), + :seemore => seemore % { :uri => "http://www.last.fm/user/#{CGI.escape user}/friends" } + } + when :lovedtracks + loved = doc.root.get_elements("lovedtracks/track").map do |track| + [track.elements["artist/name"].text, track.elements["name"].text].join(" - ") + end + loved_prep = loved.shuffle[0, num].to_enum(:each_with_index).collect { |e,i| (i % 2).zero? ? Underline+e+Underline : e } + + if loved.empty? + reply = _("%{user} has not loved any tracks") + elsif loved.length <= num + reply = _("%{user} has loved %{total} tracks: %{tracks}") + else + reply = _("%{user} has loved %{total} tracks, including %{tracks}%{seemore}") + end + + m.reply reply % { + :user => user, + :total => loved.size, + :tracks => Utils.comma_list(loved_prep), + :seemore => seemore % { :uri => "http://www.last.fm/user/#{CGI.escape user}/library/loved" } + } + when :neighbours + nbrs = doc.root.get_elements("neighbours/user").map do |u| + u.elements["name"].text + end + + if nbrs.empty? + reply = _("no one seems to share %{user}'s musical taste") + elsif nbrs.length <= num + reply = _("%{user}'s musical neighbours are %{nbrs}") + else + reply = _("%{user}'s musical neighbours include %{nbrs}%{seemore}") + end + m.reply reply % { + :user => user, + :nbrs => Utils.comma_list(nbrs.shuffle[0, num]), + :seemore => seemore % { :uri => "http://www.last.fm/user/#{CGI.escape user}/neighbours" } + } + when :recenttracks + tracks = doc.root.get_elements("recenttracks/track").map do |track| + [track.elements["artist"].text, track.elements["name"].text].join(" - ") + end + + counts = [] + tracks.each do |track| + if t = counts.assoc(track) + counts[counts.rindex(t)] = [track, t[-1] += 1] + else + counts << [track, 1] + end + end + + tracks_prep = counts[0, num].to_enum(:each_with_index).map do |e,i| + str = (i % 2).zero? ? Underline+e[0]+Underline : e[0] + str << " (%{i} times%{m})" % { + :i => e.last, + :m => counts.size == 1 ? _(" or more") : nil + } if e.last > 1 + str + end + + if tracks.empty? + m.reply _("%{user} hasn't played anything recently") % { :user => user } + else + m.reply _("%{user} has recently played %{tracks}") % + { :user => user, :tracks => Utils.comma_list(tracks_prep) } + end + when :shouts + shouts = doc.root.get_elements("shouts/shout") + if shouts.empty? + m.reply _("there are no shouts for %{user}") % { :user => user } + else + shouts[0, num].each do |shout| + m.reply _("<%{author}> %{body}") % { + :body => shout.elements["body"].text, + :author => shout.elements["author"].text, + } + end + end + when :toptracks, :topalbums, :topartists, :weeklytrackchart, :weeklyalbumchart, :weeklyartistchart + type = action.to_s.scan(/track|album|artist/).to_s + items = doc.root.get_elements("#{action}/#{type}").map do |item| + case action + when :weeklytrackchart, :weeklyalbumchart + format = "%{artist} - %{title} (%{bold}%{plays}%{bold})" + artist = item.elements["artist"].text + when :weeklyartistchart, :topartists + format = "%{artist} (%{bold}%{plays}%{bold})" + artist = item.elements["name"].text + when :toptracks, :topalbums + format = "%{artist} - %{title} (%{bold}%{plays}%{bold})" + artist = item.elements["artist/name"].text + end + + _(format) % { + :artist => artist, + :title => item.elements["name"].text, + :plays => item.elements["playcount"].text, + :bold => Bold + } + end + if items.empty? + m.reply _("%{user} hasn't played anything in this period of time") % { :user => user } + else + m.reply items[0, num].join(", ") + end + end + end + + def resolve_username(m, name) + name = m.sourcenick if name.nil? + @registry[name] or name end end +event_map_options = { + :action => :find_events, + :requirements => { + :num => /\d+/, + :sort_order => /(?:asc|desc)(?:ending)?/ + }, + :thread => true +} + plugin = LastFmPlugin.new -plugin.map 'lastfm [:num] event[s] in *location', :action => :find_events, :requirements => { :num => /\d+/ }, :thread => true -plugin.map 'lastfm [:num] event[s] by *who', :action => :find_events, :requirements => { :num => /\d+/ }, :thread => true -plugin.map 'lastfm [:num] event[s] [for] *who', :action => :find_events, :requirements => { :num => /\d+/ }, :thread => true -plugin.map 'lastfm now :who', :action => :now_playing, :thread => true -plugin.map 'lastfm now', :action => :now_playing, :thread => true -plugin.map 'np :who', :action => :now_playing, :thread => true +plugin.map 'lastfm [:num] event[s] in *location [sort[ed] by :sort_by] [[in] :sort_order [order]]', event_map_options.dup +plugin.map 'lastfm [:num] event[s] by *who [sort[ed] by :sort_by] [[in] :sort_order [order]]', event_map_options.dup +plugin.map 'lastfm [:num] event[s] at *venue [sort[ed] by :sort_by] [[in] :sort_order [order]]', event_map_options.dup +plugin.map 'lastfm [:num] event[s] [for] *who [sort[ed] by :sort_by] [[in] :sort_order [order]]', event_map_options.dup plugin.map 'lastfm artist *artist', :action => :find_artist, :thread => true plugin.map 'lastfm album *album [by *artist]', :action => :find_album plugin.map 'lastfm track *track', :action => :find_track, :thread => true -plugin.map 'lastfm set nick :who', :action => :set_user, :thread => true -plugin.map 'lastfm set user :who', :action => :set_user, :thread => true -plugin.map 'lastfm set username :who', :action => :set_user, :thread => true -plugin.map 'lastfm set verb :present :past', :action => :set_verb, :thread => true -plugin.map 'lastfm who :who', :action => :get_user, :thread => true -plugin.map 'lastfm who', :action => :get_user, :thread => true -plugin.map 'lastfm compare :user1 :user2', :action => :tasteometer, :thread => true -plugin.map 'np', :action => :now_playing, :thread => true -plugin.map "lastfm [user] :action [:user]", :thread => true, - :requirements => { :action => - /^(?:events|friends|neighbou?rs|playlists|recent?tracks|top(?:album|artist|tag)s?|weekly(?:album|artist|track)chart|weeklychartlist)$/ +plugin.map 'lastfm venue *venue', :action => :find_venue, :thread => true +plugin.map 'lastfm set user[name] :who', :action => :set_user, :thread => true +plugin.map 'lastfm set verb *present, *past', :action => :set_verb, :thread => true +plugin.map 'lastfm who [:who]', :action => :get_user, :thread => true +plugin.map 'lastfm compare to :user2', :action => :tasteometer, :thread => true +plugin.map 'lastfm compare [:user1] [to] :user2', :action => :tasteometer, :thread => true +plugin.map "lastfm [user] [:num] :action [:user]", :thread => true, + :requirements => { + :action => /^(?:events|shouts|friends|neighbou?rs|loved(?:tracks)?|recent(?:t?racks)?|top(?:album|artist|track)s?|weekly(?:albums?|artists?|tracks?)(?:chart)?)$/, + :num => /^\d+$/ +} +plugin.map 'lastfm [user] [:num] :action [:user] over [*period]', :thread => true, + :requirements => { + :action => /^(?:top(?:album|artist|track)s?)$/, + :period => /^(?:(?:3|6|12) months)|(?:a\s|1\s)?year$/, + :num => /^\d+$/ } -plugin.map "lastfm [:who]", :action => :now_playing, :thread => true +plugin.map 'lastfm [now] [:who]', :action => :now_playing, :thread => true +plugin.map 'np [:who]', :action => :now_playing, :thread => true