class UrbanPlugin < Plugin
  URBAN = 'http://www.urbandictionary.com/define.php?term='

  def help( plugin, topic="")
    "urban [word] [n]: give the [n]th definition of [word] from urbandictionary.com. urbanday: give the word-of-the-day at urban"
  end

  def get_def(m, word, n = nil)
    n = n.to_i if n
    u = URBAN + CGI.escape(word)
    u += '&skip=' + n.to_s if n
    s = @bot.httputil.get(u)

    notfound = s.match %r{<div style="color: #669FCE"><i>.*?</i> isn't defined}

    if s.sub!(%r{<div class="pager"><b>(\d+)</b>\s*definition.*$}m, '')
      total = $1.to_i
    else
      total = 1
    end

    n = total if n && n > total

    rv = Array.new

    s.scan(%r{<td class="def_number"[^>]*>(\d+)\.</td>.*?<td class="def_word">(?:<a.*?>)?([^>]+)(?:</a>)?</td>.*?<div class="def_p">.*?<p>(.+?)</p>.*?<p style=".*?>(.+?)</p>}m) do |a1, a2, a3, a4|
      rv << (
        "#{Bold}#{a2} (#{a1}/#{total})#{Bold}: " +
        a3.ircify_html(:limit => 300) + " " +
        "<i>#{a4}</i>".ircify_html(:limit => 100)
      ) unless (n && n != a1.to_i) || rv.size >= 3
    end

    if notfound
      if rv.empty?
        m.reply "#{word} not found"
      else
        m.reply "#{word} not found. maybe you mean:"
        rv.each { |s| m.reply s }
      end
    else
      rv.each { |s| m.reply s }
    end
  end

  def urban(m, params)
    words = params[:words].to_s
    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
    get_def(m, words, params[:n])
  end

  def uotd(m, params)
    home = @bot.httputil.get("http://www.urbandictionary.com/daily.php")
    if home.nil?
      m.reply "Couldn't get the urban dictionary word of the day"
      return
    end
    home.match(%r{href="/define.php\?term=.*?">(.*?)<})
    wotd = $1
    debug "Urban word of the day: #{wotd}"
    if !wotd
      m.reply "Couldn't get the urban dictionary word of the day"
      return
    end
    get_def(m, wotd, 1)
  end
end

plugin = UrbanPlugin.new
plugin.map "urban *words :n", :requirements => { :n => /^-?\d+$/ }, :action => 'urban'
plugin.map "urban [*words]", :action => 'urban'
plugin.map "urbanday", :action => 'uotd'