# Security threat level plugin for rbot
# by Robin Kearney (robin@riviera.org.uk)
#
# inspired by elliots fascination with the us
# threat level.
#
# again a dirty hack but it works, just...
#

require 'uri/common'

class ThreatPlugin < Plugin

  def help(plugin, topic="")
    "threat => prints out the current threat level as reported by http://www.dhs.gov/"
  end

  def privmsg(m)
    color = ""
    red = "\x0304" # severe
    orange = "\x0307" # high
    yellow = "\x0308" # elevated
    blue = "\x0312" # guarded
    green = "\x0303" # low
    black = "\x0301" # default

    page = @bot.httputil.get URI.parse("http://www.dhs.gov/")

    if page =~ /\"Current National Threat Level is (.*?)\"/
      state = $1
      case state
      when "severe"
        color = red
      when "high"
        color = orange
      when "elevated"
        color = yellow
      when "guarded"
        color = blue
      when "low"
        color = green
      else
        color = black
      end

      m.reply color + "Today " + m.sourcenick + " the threat level is " + state.capitalize
    else
      m.reply "I was unable to retrieve the threat level"
    end

    return
  end

end
plugin = ThreatPlugin.new
plugin.register("threat")