module Minitest::Reporters

def self.choose_reporters(console_reporters, env)
  if env["TM_PID"]
    [RubyMateReporter.new]
  elsif env["RM_INFO"] || env["TEAMCITY_VERSION"]
    [RubyMineReporter.new]
  else
    Array(console_reporters)
  end
end

def self.use!(console_reporters = ProgressReporter.new, env = ENV, backtrace_filter = ExtensibleBacktraceFilter.default_filter)
  use_runner!(console_reporters, env)
  Minitest.backtrace_filter = backtrace_filter
  unless defined?(@@loaded)
    use_around_test_hooks!
    use_old_activesupport_fix!
    @@loaded = true
  end
end

def self.use_around_test_hooks!
  Minitest::Test.class_eval do
    def run_with_hooks(*args)
      if defined?(Minitest::Reporters) && reporters = Minitest::Reporters.reporters
        reporters.each { |r| r.before_test(self) }
        result = run_without_hooks(*args)
        reporters.each { |r| r.after_test(self) }
        result
      else
        run_without_hooks(*args)
      end
    end
    alias_method :run_without_hooks, :run
    alias_method :run, :run_with_hooks
  end
end

def self.use_old_activesupport_fix!
  if defined?(ActiveSupport::VERSION) && ActiveSupport::VERSION::MAJOR < 4
    require "minitest/old_activesupport_fix"
  end
end

def self.use_runner!(console_reporters, env)
  self.reporters = choose_reporters(console_reporters, env)
end

def run_with_hooks(*args)
  if defined?(Minitest::Reporters) && reporters = Minitest::Reporters.reporters
    reporters.each { |r| r.before_test(self) }
    result = run_without_hooks(*args)
    reporters.each { |r| r.after_test(self) }
    result
  else
    run_without_hooks(*args)
  end
end