module Riot
  class Reporter
    attr_accessor :passes, :failures, :errors, :current_context
    def initialize
      @passes = @failures = @errors = 0
      @current_context = ""
    end

    def summarize(&block)
      started = Time.now
      yield
    ensure
      results(Time.now - started)
    end

    def describe_context(context); @current_context = context; end

    def report(description, response)
      code, result = *response
      case code
      when :pass then
        @passes += 1
        pass(description)
      when :fail then
        @failures += 1
        fail(description, result)
      when :error then
        @errors += 1
        error(description, result)
      end
    end

    def new(*args, &block)
      self
    end
  end # Reporter

  class IOReporter < Reporter
    attr_reader :writer
    def initialize(writer=STDOUT)
      super()
      @writer = writer
    end
    def say(message) writer.puts(message); end

    def results(time_taken)
      values = [passes, failures, errors, ("%0.6f" % time_taken)]
      say "\n%d passes, %d failures, %d errors in %s seconds" % values
    end

    def format_error(e)
      format = "    #{e.class.name} occured"
      format += "\n#{e.to_s}\n"
      e.backtrace.each { |line| format += "\n      at #{line}" }

      format
    end

    begin
      raise LoadError if ENV["TM_MODE"]
      require 'rubygems'
      require 'term/ansicolor'
      include Term::ANSIColor
    rescue LoadError
      def green(str); str; end
      alias :red :green
      alias :yellow :green
    end
  end

  class StoryReporter < IOReporter
    def describe_context(context)
      super
      say context.description
    end
    def pass(description) say "  + " + green(description); end
    def fail(description, message) say "  - " + yellow("#{description}: #{message}"); end
    def error(description, e) say "  ! " + red("#{description}: #{e.message}"); end
  end

  class VerboseStoryReporter < StoryReporter
    def error(description, e)
      super(description, e)
      say red(format_error(e))
    end
  end

  class DotMatrixReporter < IOReporter
    def pass(description); writer.write green("."); end

    def initialize(writer=STDOUT)
      super
      @details = []
    end
    def fail(description, message)
      writer.write yellow("F")
      @details << "FAILURE - #{current_context.description} #{description} => #{message}"
    end

    def error(description, e)
      writer.write red("E")
      @details << "ERROR - #{current_context.description} #{description} => #{format_error(e)}"
    end

    def results(time_taken)
      say "\n"
      @details.each { |detail| say detail }
      super
    end
  end

  class SilentReporter < Reporter
    def pass(description); end
    def fail(description, message); end
    def error(description, e); end
    def results(time_taken); end
  end
end # Riot