summaryrefslogtreecommitdiff
path: root/data/rbot/plugins/lastfm.rb
blob: bc7f819c8f0065343962935e8140673e303dd8b0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
#-- vim:sw=2:et
#++
#
# :title: lastfm plugin for rbot
#
# Author:: Jeremy Voorhis
# Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
#
# Copyright:: (C) 2005 Jeremy Voorhis
# Copyright:: (C) 2007 Giuseppe Bilotta
#
# License:: GPL v2

class ::LastFmEvent
  SELECTOR = /<tr class="vevent.*?<\/tr>/m
  # matches are:
  # 1. day 2. moth 3. year 4. url_who 5. who 6. url_where 7. where 8. how_many
  # TODO festival have TWO dates  -------+
  # TODO event type -------------+       |
  #                              V       V
  MATCHER = /<tr class="vevent\s+\w+\s+(?:\S+\s+)?\S+?-(\d\d)-(\d\d)-(\d\d\d\d)\s*">.*?<a class="url summary" href="(\/event\/\d+)">(.*?)<\/a>.*?<a href="(\/venue\/\d+)">(.*?)<\/a>.*?<td>(?:(.*?) attending\s+)?.*?<\/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="")
    case (topic.intern rescue nil)
    when :event, :events
      "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 #{@bot.config['lastfm.default_events']} and cannot be higher than #{@bot.config['lastfm.max_events']}"
    when :artist, :group
      "lastfm artist <name> => show information on artist/group <name> from last.fm"
    when :song, :track
      "lastfm track <name> => show information on track/song <name> from last.fm [not implemented yet]"
    when :album
      "lastfm album <name> => show information on album <name> from last.fm [not implemented yet]"
    else
      "lastfm <function> <user> => lastfm data for <user> on last.fm where <function> in [recenttracks, topartists, topalbums, toptracks, tags, friends, neighbors]. other topics: events, artist, group, song, track, album"
    end
  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}"
    query = location ? "?findloc=#{CGI.escape(location)}" : "?s=#{CGI.escape(artist)}&findloc="
    begin
      page = @bot.httputil.get LASTFM + "/events/" + query
      if page
        events = Array.new
        disp_events = Array.new

        pre_events = page.scan(LastFmEvent::SELECTOR)
        # 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(/<a href="(\/events\/\?l=[^"]+)">/)
            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 { |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>(.*?)<\/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>(.*?)<\/strong>(?:<br\s*\/>(.+)?)?/)
            loc = Bold + $1.ircify_html + Bold
            loc << ", " << $2.ircify_html if $2
          else
            debug where.inspect
            loc = where.ircify_html
          end
          attendance = how_many ? 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
    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 now_playing(m, params)
    opts = { :cache => false }
    user = params[:who].to_s
    page = nil
    begin
      page = @bot.httputil.get("#{LASTFM}/user/#{user}", opts)
      if page
        if page.match(/class="nowListening">\s*<td class="subject">\s*<a href="\/music.*?">(.*)<\/a>\s*<\/td/)
          m.reply "#{user} is jammin to #{$1.ircify_html}"
        else
          m.reply "#{user} isn't listening to anything right now."
        end
      end
    rescue
      m.reply "I had problems getting #{user}'s current info"
    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(/<h1 class="h1artist"><a href="([^"]+)">(.*?)<\/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 (?:class|id)="wikiAbstract">(.*?)<\/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 #{action} for #{user} (is #{user} a user?)"
    end
  end
end

plugin = LastFmPlugin.new
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 now *who', :action => :now_playing, :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