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

   1 class UrbanPlugin < Plugin
   2
   3   def help( plugin, topic="")
   4     "urban [word] [n]: give the [n]th definition of [word] from urbandictionary.com. urbanday: give the word-of-the-day at urban"
   5   end
   6
   7   def urban(m, params)
   8     words = params[:words].to_s
   9     n = params[:n].nil? ? 1 : params[:n].to_i rescue 1
  10
  11     if words.empty?
  12       resp = @bot.httputil.head('http://www.urbandictionary.com/random.php',
  13                                :max_redir => -1,
  14                                :cache => false)
  15       if resp.code == "302" && (loc = resp['location'])
  16         words = URI.unescape(loc.match(/define.php\?term=(.*)$/)[1]) rescue nil
  17       end
  18     end
  19     # we give a very high 'skip' because this will allow us to get the number of definitions by retrieving the previous definition
  20     uri = "http://www.urbanwap.com/search.php?term=#{CGI.escape words}&skip=65536"
  21     page = @bot.httputil.get(uri)
  22     if page.nil?
  23       m.reply "Couldn't retrieve an urban dictionary definition of #{words}"
  24       return
  25     end
  26     if page =~ / is undefined<\/card><\/wml>/
  27       m.reply "There is no urban dictionary definition of #{words}"
  28       return
  29     end
  30     if page =~ /&amp;skip=(\d+)">prev<\/a>/
  31       numdefs = $1.to_i + 1
  32     else
  33       numdefs = 1
  34     end
  35     n = numdefs + n + 1 if n < 0
  36     if n > numdefs
  37       m.reply "Urban dictionary only has #{numdefs} definitions for '#{words}'"
  38       n = numdefs
  39     end
  40     if n < numdefs
  41       uri = "http://www.urbanwap.com/search.php?term=#{CGI.escape words}&skip=#{n-1}"
  42       page = @bot.httputil.get(uri)
  43       if page.nil?
  44         case n % 10
  45         when 1
  46           ord = 'st'
  47         when 2
  48           ord = 'nd'
  49         when 3
  50           ord = 'rd'
  51         else
  52           ord = 'th'
  53         end
  54         m.reply "Couldn't retrieve the #{n}#{ord} urban dictionary definition of #{words}"
  55         return
  56       end
  57     end
  58     m.reply "#{get_def(page)} (#{n}/#{numdefs})"
  59   end
  60
  61   def get_def(text)
  62     # Start by removing the prev/home/next links
  63     t = text.gsub(/(?:<a href[^>]*>prev<\/a> )?<a href[^>]*>home<\/a>(?: <a href[^>]*>next<\/a>)?/,'')
  64     # Close up paragraphs
  65     t.gsub!(/<\/?p>/, ' ')
  66     t.gsub!("\n", ' ')
  67     # Reverse headings
  68     t.gsub!(/<\/?b>/,"#{Reverse}")
  69     # Enbolden links
  70     t.gsub!(/<\/?a(?: [^>]*)?>/,"#{Bold}")
  71     # Reverse examples
  72     t.gsub!(/<\/?(?:i|em)>/,"#{Underline}")
  73     # Clear anything else
  74     t.gsub!(/<.*?>/, '')
  75
  76     Utils.decode_html_entities t.strip
  77   end
  78
  79   def uotd(m, params)
  80     home = @bot.httputil.get("http://www.urbanwap.com/")
  81     if home.nil?
  82       m.reply "Couldn't get the urban dictionary word of the day"
  83       return
  84     end
  85     home.match(/Word of the Day: <a href="(.*?)">.*?<\/a>/)
  86     wotd = $1
  87     debug "Urban word of the day: #{wotd}"
  88     page = @bot.httputil.get(wotd)
  89     if page.nil?
  90       m.reply "Couldn't get the urban dictionary word of the day"
  91     else
  92       m.reply get_def(page)
  93     end
  94   end
  95 end
  96
  97 plugin = UrbanPlugin.new
  98 plugin.map "urban *words :n", :requirements => { :n => /^-?\d+$/ }, :action => 'urban'
  99 plugin.map "urban [*words]", :action => 'urban'
 100 plugin.map "urbanday", :action => 'uotd'
 101