require "cgi"

module Teaspoon
  module Formatters
    class JunitFormatter < Base
      protected

      def log_runner(result)
        log_line(%{<?xml version="1.0" encoding="UTF-8"?>})
        log_line(%{<testsuites name="Teaspoon">})
        log_line(%{<testsuite name="#{escape(@suite_name)}" tests="#{@total_count}" time="#{result.start}">})
      end

      def log_suite(result)
        log_end_suite
        log_line(%{<testsuite name="#{escape(result.label)}">})
      end

      def log_passing_spec(result)
        log_junit_spec(suite: result.suite, label: result.label)
      end

      def log_pending_spec(result)
        log_junit_spec(suite: result.suite, label: result.label) do
          log_line(%{  <skipped/>})
        end
      end

      def log_failing_spec(result)
        log_junit_spec(suite: result.suite, label: result.label) do
          log_line(%{  <failure type="AssertionFailed">#{cdata(result.message)}</failure>})
        end
      end

      def log_result(_result)
        log_end_suite
      end

      def log_coverage(message)
        properties = "<properties>#{cdata(message)}</properties>"
        log_line(%{<testsuite name="Coverage summary" tests="0">\n#{properties}\n</testsuite>})
      end

      def log_threshold_failure(message)
        log_line(%{<testsuite name="Coverage thresholds" tests="1">})
        log_junit_spec(suite: "Coverage thresholds", label: "were not met") do
          log_line(%{  <failure type="AssertionFailed">#{cdata(message)}</failure>})
        end
        log_line(%{</testsuite>})
      end

      def log_complete(_failure_count)
        log_line(%{</testsuite>\n</testsuites>})
      end

      private

      def log_end_suite
        log_line(%{</testsuite>}) if @last_suite
      end

      def log_junit_spec(opts, &_block)
        log_line(%{<testcase classname="#{escape(opts[:suite])}" name="#{escape(opts[:label])}">})
        yield if block_given?
        log_line(%{<system-out>#{cdata(@stdout)}</system-out>}) unless @stdout.blank?
        log_line(%{</testcase>})
      end

      def escape(str)
        CGI::escapeHTML(str)
      end

      def cdata(str)
        "\n<![CDATA[\n#{str.gsub(/\n$/, '')}\n]]>\n"
      end
    end
  end
end