-require 'cgi'
-begin
- require 'rubyful_soup'
-rescue
- warning "could not load rubyful_soup, urban dictionary disabled"
- warning "please get it from http://www.crummy.com/software/RubyfulSoup/"
- warning "or install it via gem"
- return
-end
-require 'uri/common'
-
class UrbanPlugin < Plugin
def help( plugin, topic="")
- "~urban [word] [n]. Give the [n]th definition of [word] from urbandictionary.com."
+ "urban [word] [n]: give the [n]th definition of [word] from urbandictionary.com. urbanday: give the word-of-the-day at urban"
end
- def privmsg( m )
+ def urban(m, params)
+ words = params[:words].to_s
+ n = params[:n].nil? ? 1 : params[:n].to_i rescue 1
- unless(m.params && m.params.length > 0)
- m.reply "incorrect usage: " + help(m.plugin)
+ if words.empty?
+ resp = @bot.httputil.head('http://www.urbandictionary.com/random.php',
+ :max_redir => -1,
+ :cache => false)
+ if resp.code == "302" && (loc = resp['location'])
+ words = URI.unescape(loc.match(/define.php\?term=(.*)$/)[1]) rescue nil
+ end
+ end
+ # we give a very high 'skip' because this will allow us to get the number of definitions by retrieving the previous definition
+ uri = "http://www.urbanwap.com/search.php?term=#{CGI.escape words}&skip=65536"
+ page = @bot.httputil.get(uri)
+ if page.nil?
+ m.reply "Couldn't retrieve an urban dictionary definition of #{words}"
return
end
-
- paramArray = m.params.split(' ')
- definitionN = 0
- if m.params == 'random' then
- uri = URI.parse( "http://www.urbandictionary.com/random.php" )
- else
- if( paramArray.last.to_i != 0 ) then
- definitionN = paramArray.last.to_i - 1
- query = m.params.chomp( paramArray.last )
- query.rstrip!
- else
- query = m.params
- end
- uri = URI.parse( "http://www.urbandictionary.com/define.php?term=#{ URI.escape query}" )
+ if page =~ / is undefined<\/card><\/wml>/
+ m.reply "There is no urban dictionary definition of #{words}"
+ return
end
-
- soup = BeautifulSoup.new( @bot.httputil.get( uri ) )
- if titleNavi = soup.find_all( 'td', :attrs => { 'class' => 'def_word' } )[0] then
- title = titleNavi.contents
- results = soup.find_all( 'div', :attrs => { 'class' => 'def_p' } )
- debug PP.pp(results,'')
- output = Array.new
- if results[definitionN] then
- results[definitionN].p.contents.each { |s| output.push( strip_tags( s.to_s ) ) }
- m.reply "\002#{title}\002 - #{output}"
- else
- m.reply "#{query} does not have #{definitionN + 1} definitions."
- end
+ if page =~ /&skip=(\d+)">prev<\/a>/
+ numdefs = $1.to_i + 1
else
- m.reply "#{m.params} not found."
+ numdefs = 1
+ end
+ n = numdefs + n + 1 if n < 0
+ if n > numdefs
+ m.reply "Urban dictionary only has #{numdefs} definitions for '#{words}'"
+ n = numdefs
end
+ if n < numdefs
+ uri = "http://www.urbanwap.com/search.php?term=#{CGI.escape words}&skip=#{n-1}"
+ page = @bot.httputil.get(uri)
+ if page.nil?
+ case n % 10
+ when 1
+ ord = 'st'
+ when 2
+ ord = 'nd'
+ when 3
+ ord = 'rd'
+ else
+ ord = 'th'
+ end
+ m.reply "Couldn't retrieve the #{n}#{ord} urban dictionary definition of #{words}"
+ return
+ end
+ end
+ m.reply "#{get_def(page)} (#{n}/#{numdefs})"
+ end
+ def get_def(text)
+ # Start by removing the prev/home/next links
+ t = text.gsub(/(?:<a href[^>]*>prev<\/a> )?<a href[^>]*>home<\/a>(?: <a href[^>]*>next<\/a>)?/,'')
+ # Close up paragraphs
+ t.gsub!(/<\/?p>/, ' ')
+ t.gsub!("\n", ' ')
+ # Reverse headings
+ t.gsub!(/<\/?b>/,"#{Reverse}")
+ # Enbolden links
+ t.gsub!(/<\/?a(?: [^>]*)?>/,"#{Bold}")
+ # Reverse examples
+ t.gsub!(/<\/?(?:i|em)>/,"#{Underline}")
+ # Clear anything else
+ t.gsub!(/<.*?>/, '')
+
+ Utils.decode_html_entities t.strip
end
- def strip_tags(html)
- html.gsub(/<.+?>/,'').
- gsub(/&/,'&').
- gsub(/"/,'"').
- gsub(/</,'<').
- gsub(/>/,'>').
- gsub(/&ellip;/,'...').
- gsub(/'/, "'").
- gsub("\n",'')
+ def uotd(m, params)
+ home = @bot.httputil.get("http://www.urbanwap.com/")
+ if home.nil?
+ m.reply "Couldn't get the urban dictionary word of the day"
+ return
+ end
+ home.match(/Word of the Day: <a href="(.*?)">.*?<\/a>/)
+ wotd = $1
+ debug "Urban word of the day: #{wotd}"
+ page = @bot.httputil.get(wotd)
+ if page.nil?
+ m.reply "Couldn't get the urban dictionary word of the day"
+ else
+ m.reply get_def(page)
+ end
end
end
plugin = UrbanPlugin.new
-plugin.register( "urban" )
+plugin.map "urban *words :n", :requirements => { :n => /^-?\d+$/ }, :action => 'urban'
+plugin.map "urban [*words]", :action => 'urban'
+plugin.map "urbanday", :action => 'uotd'
+