class Rails::TestUnit::Runner

def attach_before_load_options(opts)
  opts.on("--warnings", "-w", "Run with Ruby warnings enabled") { }
  opts.on("-e", "--environment ENV", "Run tests in the ENV environment") { }
end

def compose_filter(runnable, filter)
  filter = normalize_declarative_test_filter(filter)
  if filters.any? { |_, lines| lines.any? }
    CompositeFilter.new(runnable, filter, filters)
  else
    filter
  end
end

def default_test_exclude_glob
  ENV["DEFAULT_TEST_EXCLUDE"] || "test/{system,dummy,fixtures}/**/*_test.rb"
end

def default_test_glob
  ENV["DEFAULT_TEST"] || "test/**/*_test.rb"
end

def extract_filters(argv)
  # Extract absolute and relative paths but skip -n /.*/ regexp filters.
  argv.filter_map do |path|
    next unless path_argument?(path)
    path = path.tr("\\", "/")
    case
    when /(:\d+(-\d+)?)+$/.match?(path)
      file, *lines = path.split(":")
      filters << [ file, lines ]
      file
    when Dir.exist?(path)
      "#{path}/**/*_test.rb"
    else
      filters << [ path, [] ]
      path
    end
  end
end

def list_tests(patterns)
  tests = Rake::FileList[patterns.any? ? patterns : default_test_glob]
  tests.exclude(default_test_exclude_glob) if patterns.empty?
  tests.exclude(%r{test/isolation/assets/node_modules})
  tests
end

def load_tests(argv)
  patterns = extract_filters(argv)
  tests = list_tests(patterns)
  tests.to_a.each do |path|
    abs_path = File.expand_path(path)
    require abs_path
  rescue LoadError => exception
    if exception.path == abs_path
      all_tests = list_tests([default_test_glob])
      corrections = DidYouMean::SpellChecker.new(dictionary: all_tests).correct(path)
      if corrections.empty?
        raise exception
      end
      raise(InvalidTestError.new(path, DidYouMean::Formatter.message_for(corrections)), cause: nil)
    else
      raise
    end
  end
end

def normalize_declarative_test_filter(filter)
  if filter.is_a?(String)
    if regexp_filter?(filter)
      # Minitest::Spec::DSL#it does not replace whitespace in method
      # names, so match unmodified method names as well.
      filter = filter.gsub(/\s+/, "_").delete_suffix("/") + "|" + filter.delete_prefix("/")
    elsif !filter.start_with?("test_")
      filter = "test_#{filter.gsub(/\s+/, "_")}"
    end
  end
  filter
end

def parse_options(argv)
  # Perform manual parsing and cleanup since option parser raises on unknown options.
  env_index = argv.index("--environment") || argv.index("-e")
  if env_index
    argv.delete_at(env_index)
    environment = argv.delete_at(env_index).strip
  end
  ENV["RAILS_ENV"] = environment || "test"
  w_index = argv.index("--warnings") || argv.index("-w")
  $VERBOSE = argv.delete_at(w_index) if w_index
end

def path_argument?(arg)
  PATH_ARGUMENT_PATTERN.match?(arg)
end

def regexp_filter?(arg)
  arg.start_with?("/") && arg.end_with?("/")
end

def run(argv = [])
  load_tests(argv)
  require "active_support/testing/autorun"
end

def run_from_rake(test_command, argv = [])
  # Ensure the tests run during the Rake Task action, not when the process exits
  success = system("rails", test_command, *argv, *Shellwords.split(ENV["TESTOPTS"] || ""))
  success || exit(false)
end