git.oblomov.eu Git - rbot/blob - data/rbot/plugins/urban.rb

   1 class UrbanPlugin < Plugin
   2   URBAN = 'http://www.urbandictionary.com/define.php?term='
   3
   4   def help( plugin, topic="")
   5     "urban [word] [n]: give the [n]th definition of [word] from urbandictionary.com. urbanday: give the word-of-the-day at urban"
   6   end
   7
   8   def format_definition(total, num, word, desc, ex)
   9     "#{Bold}#{word} (#{num}/#{total})#{Bold}: " +
  10     desc.ircify_html(:limit => 300) + " " +
  11     "<i>#{ex}</i>".ircify_html(:limit => 100)
  12   end
  13
  14   def get_def(m, word, n = nil)
  15     n = n ? n.to_i : 1
  16     p = (n-1)/7 + 1
  17     u = URBAN + URI.escape(word)
  18     u += '&page=' + p.to_s if p > 1
  19     s = @bot.httputil.get(u)
  20     return m.reply("Couldn't get the urban dictionary definition for #{word}") if s.nil?
  21
  22     notfound = s.match %r{<i>.*?</i> isn't defined}
  23
  24     numpages = if s[%r{<div id='paginator'>.*?</div>}m]
  25       $&.scan(/\d+/).collect {|x| x.to_i}.max || 1
  26     else 1 end
  27
  28     rv = Array.new
  29     s.scan(%r{<div class='word'[^>]*>.*?<a class="index"[^>]*>.*?(\d+)\..*?</a>.*?<span>(.*?)</span>.*?<div class="definition">(.+?)</div>.*?<div class="example">(.+?)</div>}m) do |num, wrd, desc, ex|
  30       rv << [num.to_i, wrd.strip, desc.strip, ex.strip]
  31     end
  32
  33     maxnum = rv.collect {|x| x[0]}.max || 0
  34     return m.reply("#{Bold}#{word}#{Bold} not found") if rv.empty?
  35
  36     if notfound
  37       suggestions = rv.map { |str| Underline + str[1] + Underline }.uniq.join ', '
  38       m.reply "#{Bold}#{word}#{Bold} not found. maybe you mean #{suggestions}?"
  39       return
  40     end
  41
  42     answer = rv.find { |a| a[0] == n }
  43     answer ||= (n > maxnum ? rv.last : rv.first)
  44     m.reply format_definition((p == numpages ? maxnum : "#{(numpages-1)*7 + 1}+"), *answer)
  45   end
  46
  47   def urban(m, params)
  48     words = params[:words].to_s
  49     if words.empty?
  50       resp = @bot.httputil.head('http://www.urbandictionary.com/random.php',
  51                                :max_redir => -1,
  52                                :cache => false)
  53       return m.reply("Couldn't get a random urban dictionary word") if resp.nil?
  54       if resp.code == "302" && (loc = resp['location'])
  55         words = URI.unescape(loc.match(/define.php\?term=(.*)$/)[1]) rescue nil
  56       end
  57     end
  58     get_def(m, words, params[:n])
  59   end
  60
  61   def uotd(m, params)
  62     home = @bot.httputil.get("http://www.urbandictionary.com/daily.php")
  63     if home.nil?
  64       m.reply "Couldn't get the urban dictionary word of the day"
  65       return
  66     end
  67     home.match(%r{href="/define.php\?term=.*?">(.*?)<})
  68     wotd = $1
  69     debug "Urban word of the day: #{wotd}"
  70     if !wotd
  71       m.reply "Couldn't get the urban dictionary word of the day"
  72       return
  73     end
  74     get_def(m, wotd, 1)
  75   end
  76 end
  77
  78 plugin = UrbanPlugin.new
  79 plugin.map "urban *words :n", :requirements => { :n => /^-?\d+$/ }, :action => 'urban'
  80 plugin.map "urban [*words]", :action => 'urban'
  81 plugin.map "urbanday", :action => 'uotd'
  82