module Autotest::GnomeNotify

  # Time notification will be displayed before disappearing automatically
  EXPIRATION_IN_SECONDS = 2
  ERROR_STOCK_ICON = "gtk-dialog-error"
  SUCCESS_STOCK_ICON = "gtk-dialog-info"
  PENDING_STOCK_ICON = "gtk-dialog-warning"

  Autotest.add_hook :ran_command do |at|
    if at.results.empty? # break out gracefully if a test file throws an uncaught exception
      notify ERROR_STOCK_ICON, "Syntax error", ""
    else
      output = format_input( at.results )
      display_notification( output )
    end
  end

  private

  def self.notify stock_icon, title, message
    options = "-t #{EXPIRATION_IN_SECONDS * 1000} -i #{stock_icon}"
    system "notify-send #{options} '#{title}' '#{message}'"
  end

  # isolate the example, failure, pending line like "12 examples, 4 failures, 5 pending"
  # and remove color codes
  def self.format_input( input )
    output = input.detect{ |result| result.include? "example"}
    output = remove_color_codes( output )
  end

  def self.display_notification( output )
    examples, failures, pending = output.split(", ")
    if failures.to_i > 0
      notify ERROR_STOCK_ICON, "Tests Failed", output
    elsif pending.to_i > 0
      notify PENDING_STOCK_ICON, "Tests Pending", output
    else
      notify SUCCESS_STOCK_ICON, "Tests Passed", output
    end
  end


  def self.remove_color_codes( string )
    string.match( /m(.+)\e/)[1]
  end

end