summaryrefslogtreecommitdiff
path: root/data/rbot/plugins/lastfm.rb
blob: 4ec79510873861dee5ba494f5a4702a0a2475236 (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
#-- 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
  # matches are:
  # 1. day 2. moth 3. year 4. url_who 5. who 6. url_where 7. where 8. how_many
  REGEXP = /<tr class="vevent\s+\w+\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 class="attendance">(.*?)<\/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
  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 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="
    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(/<a href="(\/events\/\?l=[^"]+)">/)
              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
          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>(.*?)<\/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.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
    }
  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="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]
    Thread.new {
      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+/ }
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 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'