class RubyMemcheck::Configuration

def command(*args)
  [
    # On some Rubies, not setting the stack size to be ulimited causes
    # Valgrind to report the following error:
    #   Invalid write of size 1
    #     reserve_stack (thread_pthread.c:845)
    #     ruby_init_stack (thread_pthread.c:871)
    #     main (main.c:48)
    "ulimit -s unlimited && ",
    valgrind,
    valgrind_options,
    valgrind_suppression_files.map { |f| "--suppressions=#{f}" },
    valgrind_generate_suppressions ? "--gen-suppressions=all" : "",
    ruby,
    args,
  ].flatten.join(" ")
end

def get_valgrind_suppression_files(dir)
  dir = File.expand_path(dir)
  full_ruby_version = "#{RUBY_ENGINE}-#{RUBY_VERSION}.#{RUBY_PATCHLEVEL}"
  versions = [full_ruby_version]
  (0..3).reverse_each { |i| versions << full_ruby_version.split(".")[0, i].join(".") }
  versions << RUBY_ENGINE
  versions.map do |version|
    old_format_files = Dir[File.join(dir, "#{binary_name}_#{version}.supp")]
    unless old_format_files.empty?
      warn("ruby_memcheck: please migrate your suppression filenames from " \
        "`gem_name_ruby-3.1.0.supp` to `ruby-3.1.0.supp` (drop the gem name from the filename)")
    end
    old_format_files + Dir[File.join(dir, "#{version}.supp")]
  end.flatten
end

def initialize(
  binary_name:,
  ruby: FileUtils::RUBY,
  valgrind: DEFAULT_VALGRIND,
  valgrind_options: DEFAULT_VALGRIND_OPTIONS,
  valgrind_suppressions_dir: DEFAULT_VALGRIND_SUPPRESSIONS_DIR,
  valgrind_generate_suppressions: false,
  skipped_ruby_functions: DEFAULT_SKIPPED_RUBY_FUNCTIONS,
  valgrind_xml_dir: Dir.mktmpdir,
  output_io: $stderr
)
  @binary_name = binary_name
  @ruby = ruby
  @valgrind = valgrind
  @valgrind_options = valgrind_options
  @valgrind_suppression_files =
    get_valgrind_suppression_files(File.join(__dir__, "../../suppressions")) +
    get_valgrind_suppression_files(valgrind_suppressions_dir)
  @valgrind_generate_suppressions = valgrind_generate_suppressions
  @skipped_ruby_functions = skipped_ruby_functions
  @output_io = output_io
  if valgrind_xml_dir
    valgrind_xml_dir = File.expand_path(valgrind_xml_dir)
    FileUtils.mkdir_p(valgrind_xml_dir)
    @valgrind_xml_dir = valgrind_xml_dir
    @valgrind_options += [
      "--xml=yes",
      # %p will be replaced with the PID
      # This prevents forking and shelling out from generating a corrupted XML
      # See --log-file from https://valgrind.org/docs/manual/manual-core.html
      "--xml-file=#{File.join(valgrind_xml_dir, "%p.out")}",
    ]
  end
end