# encoding: utf-8
require "logstash/namespace"
require "logstash/outputs/http"

# This output allows you to write events to https://www.hipchat.com/[HipChat].
#
class LogStash::Outputs::HipChat < LogStash::Outputs::Base

  config_name "hipchat"

  # The HipChat authentication token.
  config :token, :validate => :string, :required => true

  # The ID or name of the room.
  config :room_id, :validate => :string, :required => true

  # The name the message will appear be sent from.
  config :from, :validate => :string, :default => "logstash"

  # Whether or not this message should trigger a notification for people in the room.
  config :trigger_notify, :validate => :boolean, :default => false

  # Background color for message.
  # HipChat currently supports one of "yellow", "red", "green", "purple",
  # "gray", or "random". (default: yellow)
  config :color, :validate => :string, :default => "yellow"

  # Message format to send, event tokens are usable here.
  config :format, :validate => :string, :default => "%{message}"

  public
  def register
    require "ftw"
    require "uri"

    @agent = FTW::Agent.new

    @url = "https://api.hipchat.com/v1/rooms/message?auth_token=" + @token
    @content_type = "application/x-www-form-urlencoded"
  end # def register

  public
  def receive(event)
    return unless output?(event)

    hipchat_data = Hash.new
    hipchat_data['room_id'] = event.sprintf(@room_id)
    hipchat_data['from']    = @from
    hipchat_data['color']   = @color
    hipchat_data['notify']  = @trigger_notify ? "1" : "0"
    hipchat_data['message'] = event.sprintf(@format)

    @logger.debug("HipChat data", :hipchat_data => hipchat_data)

    begin
      request = @agent.post(@url)
      request["Content-Type"] = @content_type
      request.body = encode(hipchat_data)

      response = @agent.execute(request)

      # Consume body to let this connection be reused
      rbody = ""
      response.read_body { |c| rbody << c }
      #puts rbody
    rescue Exception => e
      @logger.warn("Unhandled exception", :request => request, :response => response, :exception => e, :stacktrace => e.backtrace)
    end
  end # def receive

  # shamelessly lifted this from the LogStash::Outputs::Http, I'd rather put this
  # in a common place for both to use, but unsure where that place is or should be
  def encode(hash)
    return hash.collect do |key, value|
      CGI.escape(key) + "=" + CGI.escape(value)
    end.join("&")
  end # def encode

end # class LogStash::Outputs::HipChat