module CopHelper

def _investigate(cop, processed_source)
  forces = RuboCop::Cop::Force.all.each_with_object([]) do |klass, instances|
    next unless cop.join_force?(klass)
    instances << klass.new([cop])
  end
  commissioner =
    RuboCop::Cop::Commissioner.new([cop], forces, raise_error: true)
  commissioner.investigate(processed_source)
  commissioner
end

def autocorrect_source(source, file = nil)
  RuboCop::Formatter::DisabledConfigFormatter.config_to_allow_offenses = {}
  RuboCop::Formatter::DisabledConfigFormatter.detected_styles = {}
  cop.instance_variable_get(:@options)[:auto_correct] = true
  processed_source = parse_source(source, file)
  _investigate(cop, processed_source)
  corrector =
    RuboCop::Cop::Corrector.new(processed_source.buffer, cop.corrections)
  corrector.rewrite
end

def autocorrect_source_file(source)
  Tempfile.open('tmp') { |f| autocorrect_source(source, f) }
end

def autocorrect_source_with_loop(source, file = nil)
  cnt = 0
  loop do
    cop.instance_variable_set(:@corrections, [])
    new_source = autocorrect_source(source, file)
    return new_source if new_source == source
    source = new_source
    cnt += 1
    if cnt > RuboCop::Runner::MAX_ITERATIONS
      raise RuboCop::Runner::InfiniteCorrectionLoop.new(file, [])
    end
  end
end

def inspect_source(source, file = nil)
  RuboCop::Formatter::DisabledConfigFormatter.config_to_allow_offenses = {}
  RuboCop::Formatter::DisabledConfigFormatter.detected_styles = {}
  processed_source = parse_source(source, file)
  raise 'Error parsing example code' unless processed_source.valid_syntax?
  _investigate(cop, processed_source)
end

def inspect_source_file(source)
  Tempfile.open('tmp') { |f| inspect_source(source, f) }
end

def parse_source(source, file = nil)
  if file&.respond_to?(:write)
    file.write(source)
    file.rewind
    file = file.path
  end
  RuboCop::ProcessedSource.new(source, ruby_version, file)
end