require 'rubygems'

unless ENV['RSPEC']
  require 'chef'
  require 'chef/handler'
end

require 'slack-notifier'

class StatusNotifierHandler < Chef::Handler

  DEFAULT_FAILED_MESSAGE  = '"Failure on #{node.name}: #{run_status.formatted_exception}"'.freeze
  DEFAULT_SUCCESS_MESSAGE = '"Chef run succesfully on #{node.name}"'.freeze

  def initialize(slack_params, hipchat_params, custom_message_params = {})
    @slack_params   = slack_params
    @hipchat_params = hipchat_params
    @custom_message_params = custom_message_params
  end

  def report
    if run_status.failed?
      msg = failed_message
      status = :failed
    else
      msg = success_message
      status = :success
    end

    send_to_slack(node.name, status, msg)
    send_to_hipchat(msg)
  end

  private

  def eval_message_for(message)
    return nil if message.nil?
    eval(message)
  end

  def failed_message
    eval_message_for(@custom_message_params[:failed_message] || DEFAULT_FAILED_MESSAGE)
  end

  def success_message
    eval_message_for(@custom_message_params[:success_message] || DEFAULT_SUCCESS_MESSAGE)
  end

  def send_to_hipchat(msg)
    return unless @hipchat_params[:enabled]
    hipchat[@hipchat_params[:room_name]].send(@hipchat_params[:username], msg, :notify => @hipchat_params[:notify])
  end

  def send_to_slack(node_name, status, msg)
    return unless @slack_params[:enabled]
    slack.ping '', attachments: [slack_attachment(node_name, status, msg)], channel: @slack_params[:channel], username: @slack_params[:username]
  end

  def hipchat
    return {}
  end

  def slack
    @slack ||= Slack::Notifier.new(@slack_params[:webhook_url])
  end

  def slack_attachment(node_name, status, msg)
    color = (status == :failed)? "#ff0000" : "#36a64f"
    {
      fallback: "Chef status, #{msg}",
      color: "#{color}",
      author_name: "Fai's Slave (#{node_name})",
      title: "status: #{status}",
      text: "#{msg}"
    }
  end
end