Ticket #40: urban.rb

# UrbanDictionary plugin for RubyBot
# (c) 2005 Dayne Broderson <broderson@gmail.com>
# bish0p in #modzer0 on irc.freenode.net
# Licensed under MIT License.

class UrbanPlugin < Plugin
        require 'soap/rpc/driver'
        def initialize
                super
                @key =  @bot.config['urban']
                @soap = SOAP::RPC::Driver.new('http://api.urbandictionary.com/soap')
                @soap.add_method('lookup', 'key', 'term')
                # returns an integer, the number of published definitions for that word
                @soap.add_method('count_definitions', 'key', 'term')
                # returns a list of words that surround the parameter in alphabetical order
                @soap.add_method('nearby', 'key', 'term')
                # returns Urban Word of the Day
                @soap.add_method('get_daily_word')
        end

  def help(plugin, topic="")
    "urban => look up a word in urban dictionary\n" + 
                "      no options gives word of the day\n" +
                "      a number provides that specific definition of the last word looked up"
  end

  def privmsg(m)
                if not @key 
                        m.reply "'urban' key is not set in ~/.rbot.conf.yaml"
                        m.reply "get one at http://www.urbandictionary.com/api.php"
                        return
                end
                if m.params
                        urban m, m.params
                else
                        urban m
                end
  end
 
  def urban(m, term=nil)
                if term.nil?
                        # slurp down the term of the day
                        daily = @soap.get_daily_word
                        term = daily.word
                        m.reply "UrbanDictionary Word of the Day is: #{term}"
                        result = @soap.lookup(@key, term)
                elsif term =~ /(\d+)$/
                        num = $1
                        term = @registry['urban.last']
                        m.reply "Getting definition #{num} of #{term}"
                        result = @soap.lookup(@key, term)
                        i = num.to_i - 1
                else
                        # slurp down the term asked for
                        result = @soap.lookup(@key,term)
                end

                unless result
                        m.reply "urban lookup failed"
                        return
                end
                        
                if result.size < 1
                        m.reply "no result found for '#{term}'"
                        return
                end

                @registry["urban.last"] = "#{term}"
                i = rand(result.size) if not i
                out = "(#{i+1}/#{result.size}): #{result[i].definition}"
                if result[i].author
                        out  = out + "\n - #{result[i].author}"
                else 
                        out = "definition # #{i} doesn't exist"
                end
                m.reply out
  end
end
plugin = UrbanPlugin.new
plugin.register("urban")