require 'pathname'
require 'uuid'
require 'rspec/core/formatters/base_formatter'

module AllureRSpec

  class Formatter < RSpec::Core::Formatters::BaseFormatter

    ALLOWED_LABELS = [:feature, :story, :severity, :language, :framework]

    def example_failed(example)
      AllureRSpec::Builder.stop_test(
          example.metadata[:example_group][:description_args].first,
          example.metadata[:description],
          example.metadata[:execution_result].merge(
              :caller => example.metadata[:caller],
              :exception => example.metadata[:execution_result][:exception]
          )
      )
      super
    end

    def example_group_finished(group)
      AllureRSpec::Builder.stop_suite(group.metadata[:example_group][:description_args].first)
      super
    end

    def example_group_started(group)
      AllureRSpec::Builder.start_suite(group.metadata[:example_group][:description_args].first, labels(group))
      super
    end

    def example_passed(example)
      AllureRSpec::Builder.stop_test(
          example.metadata[:example_group][:description_args].first,
          example.metadata[:description],
          example.metadata[:execution_result].merge(:caller => example.metadata[:caller])
      )
      super
    end

    def example_pending(example)
      super
    end

    def example_started(example)
      suite = example.metadata[:example_group][:description_args].first
      test = example.metadata[:description]
      AllureRSpec::Builder.start_test(suite, test, labels(example))
      super
    end

    def start(example_count)
      dir = Pathname.new(AllureRSpec::Config.output_dir)
      FileUtils.rm_rf(dir)
    end

    def stop
      AllureRSpec::Builder.each_suite_build do |suite, xml|
        dir = Pathname.new(AllureRSpec::Config.output_dir)
        FileUtils.mkdir_p(dir)
        out_file = dir.join("#{UUID.new.generate}-testsuite.xml")
        File.open(out_file, 'w+') do |file|
          file.write(xml)
        end

        xsd = Nokogiri::XML::Schema(File.read(Pathname.new(File.dirname(__FILE__)).join("../../allure-model.xsd")))
        doc = Nokogiri::XML(File.read(out_file))

        xsd.validate(doc).each do |error|
          $stderr.puts error.message
        end
      end
      super
    end

    private

    def labels(example_or_group)
      ALLOWED_LABELS.
          map { |label| [label, example_or_group.metadata[label]] }.
          find_all { |value| !value[1].nil? }.
          inject({}) { |res, value| res.merge(value[0] => value[1]) }
    end

  end
end