require 'snarl' if RUBY_PLATFORM =~ /mswin/

module TestNotifier
  # create a .test_notifier at your home
  # directory and save the images as passed.png,
  # failure.png and error.png for custom images
  PASSED_IMAGE = "passed.png"
  FAILURE_IMAGE = "failure.png"
  ERROR_IMAGE = "error.png"
  FAILURE_TITLE = "FAILED"
  PASSED_TITLE = "Passed"
  
  RSPEC_REGEX = /(\d+) examples?, (\d+) failures?(, (\d+) pendings?)?/
  TEST_UNIT_REGEX = /(\d+)\stests,\s(\d+)\sassertions,\s(\d+)\sfailures,\s(\d+)\serrors/
  
  def self.notify(image, title, message)
    image ||= "none.png"
    
    custom_image = File.join(File.expand_path("~/.test_notifier"), "image", image)
    image = File.exists?(custom_image) ? custom_image : File.join(File.dirname(__FILE__), "test_notifier", "icons", image)

    if RUBY_PLATFORM =~ /darwin/
      system("growlnotify -n test_notifier --image #{image} -p 2 -m \"#{message}\" -t \"#{title}\"")
    elsif RUBY_PLATFORM =~ /mswin/
      Snarl.show_message(title, message, image)
    elsif RUBY_PLATFORM =~ /linux/
      system("notify-send -i #{image} #{title} \"#{message}\"")
    end
  end
  
  def self.rspec?(content)
    (RSPEC_REGEX =~ content)
  end
  
  def self.notification_for_rspec(content)
    matches, *output = *content.to_s.match(RSPEC_REGEX)
    output = output.map {|i| i.to_i }
    e, f, none, p = output
    
    if f > 0
      # test has failed
      title = FAILURE_TITLE
      image = FAILURE_IMAGE
    elsif e > 0
      # everything's ok
      title = PASSED_TITLE
      image = PASSED_IMAGE
    else
      # no examples
      return
    end
    
    message = "#{e} examples, #{f} failures, #{p} pending"
    notify(image, title, message)
  end
  
  def self.notification_for_test_unit(content)
    matches, *output = *content.to_s.match(TEST_UNIT_REGEX)
    output = output.map {|i| i.to_i }
    t, a, f, e = output
    
    if f > 0 || e > 0
      # test has failed or raised an error
      title = FAILURE_TITLE
      image = e > 0 ? ERROR_IMAGE : FAILURE_IMAGE
    elsif a > 0
      # everything's ok
      title = PASSED_TITLE
      image = PASSED_IMAGE
    else
      # no assertions
      return
    end
    
    message = "#{t} tests, #{a} assertions, #{f} failures, #{e} errors"
    notify(image, title, message)
  end
end