class Gem::Ext::CargoBuilder

# File rubygems/ext/cargo_builder.rb, line 9
def initialize(spec)
  require_relative "../command"
  require_relative "cargo_builder/link_flag_converter"

  @spec = spec
  @runner = self.class.method(:run)
  @profile = :release
end

# File rubygems/ext/cargo_builder.rb, line 18
def build(_extension, dest_path, results, args = [], lib_dir = nil, cargo_dir = Dir.pwd)
  require "fileutils"
  require "shellwords"

  build_crate(dest_path, results, args, cargo_dir)
  validate_cargo_build!(dest_path)
  rename_cdylib_for_ruby_compatibility(dest_path)
  finalize_directory(dest_path, lib_dir, cargo_dir)
  results
end

# File rubygems/ext/cargo_builder.rb, line 29
def build_crate(dest_path, results, args, cargo_dir)
  env = build_env
  cmd = cargo_command(cargo_dir, dest_path, args)
  runner.call cmd, results, "cargo", cargo_dir, env

  results
end

# File rubygems/ext/cargo_builder.rb, line 37
def build_env
  build_env = rb_config_env
  build_env["RUBY_STATIC"] = "true" if ruby_static? && ENV.key?("RUBY_STATIC")
  cfg = "--cfg=rb_sys_gem --cfg=rubygems --cfg=rubygems_#{Gem::VERSION.tr(".", "_")}"
  build_env["RUSTFLAGS"] = [ENV["RUSTFLAGS"], cfg].compact.join(" ")
  build_env
end

# File rubygems/ext/cargo_builder.rb, line 45
def cargo_command(cargo_dir, dest_path, args = [])
  manifest = File.join(cargo_dir, "Cargo.toml")
  cargo = ENV.fetch("CARGO", "cargo")

  cmd = []
  cmd += [cargo, "rustc"]
  cmd += ["--crate-type", "cdylib"]
  cmd += ["--target", ENV["CARGO_BUILD_TARGET"]] if ENV["CARGO_BUILD_TARGET"]
  cmd += ["--target-dir", dest_path]
  cmd += ["--manifest-path", manifest]
  cmd += ["--lib"]
  cmd += ["--profile", profile.to_s]
  cmd += ["--locked"]
  cmd += Gem::Command.build_args
  cmd += args
  cmd += ["--"]
  cmd += [*cargo_rustc_args(dest_path)]
  cmd
end

# File rubygems/ext/cargo_builder.rb, line 166
def cargo_crate_name
  spec.metadata.fetch("cargo_crate_name", spec.name).tr("-", "_")
end

# File rubygems/ext/cargo_builder.rb, line 158
def cargo_dylib_path(dest_path)
  prefix = so_ext == "dll" ? "" : "lib"
  path_parts = [dest_path]
  path_parts << ENV["CARGO_BUILD_TARGET"] if ENV["CARGO_BUILD_TARGET"]
  path_parts += ["release", "#{prefix}#{cargo_crate_name}.#{so_ext}"]
  File.join(*path_parts)
end

# File rubygems/ext/cargo_builder.rb, line 73
def cargo_rustc_args(dest_dir)
  [
    *linker_args,
    *mkmf_libpath,
    *rustc_dynamic_linker_flags(dest_dir),
    *rustc_lib_flags(dest_dir),
    *platform_specific_rustc_args(dest_dir),
  ]
end

# File rubygems/ext/cargo_builder.rb, line 193
def darwin_target?
  makefile_config("target_os").include?("darwin")
end

# File rubygems/ext/cargo_builder.rb, line 152
def final_extension_path(dest_path)
  dylib_path = cargo_dylib_path(dest_path)
  dlext_name = "#{spec.name}.#{makefile_config("DLEXT")}"
  dylib_path.gsub(File.basename(dylib_path), dlext_name)
end

# File rubygems/ext/cargo_builder.rb, line 268
def finalize_directory(dest_path, lib_dir, extension_dir)
  require "fileutils"
  require "tempfile"

  ext_path = final_extension_path(dest_path)

  begin
    tmp_dest = Dir.mktmpdir(".gem.", extension_dir)

    # Some versions of `mktmpdir` return absolute paths, which will break make
    # if the paths contain spaces.
    #
    # As such, we convert to a relative path.
    tmp_dest_relative = get_relative_path(tmp_dest.clone, extension_dir)

    full_tmp_dest = File.join(extension_dir, tmp_dest_relative)

    # TODO: remove in RubyGems 4
    if Gem.install_extension_in_lib && lib_dir
      FileUtils.mkdir_p lib_dir
      FileUtils.cp_r ext_path, lib_dir, remove_destination: true
    end

    FileUtils::Entry_.new(full_tmp_dest).traverse do |ent|
      destent = ent.class.new(dest_path, ent.rel)
      destent.exist? || FileUtils.mv(ent.path, destent.path)
    end
  ensure
    FileUtils.rm_rf tmp_dest if tmp_dest
  end
end

# File rubygems/ext/cargo_builder.rb, line 300
def get_relative_path(path, base)
  path[0..base.length - 1] = "." if path.start_with?(base)
  path
end

# File rubygems/ext/cargo_builder.rb, line 185
def ldflag_to_link_modifier(arg)
  LinkFlagConverter.convert(arg)
end

# File rubygems/ext/cargo_builder.rb, line 125
def libruby_args(dest_dir)
  libs = makefile_config(ruby_static? ? "LIBRUBYARG_STATIC" : "LIBRUBYARG_SHARED")
  raw_libs = Shellwords.split(libs)
  raw_libs.flat_map {|l| ldflag_to_link_modifier(l) }
end

# File rubygems/ext/cargo_builder.rb, line 107
def linker_args
  cc_flag = Shellwords.split(makefile_config("CC"))
  linker = cc_flag.shift
  link_args = cc_flag.flat_map {|a| ["-C", "link-arg=#{a}"] }

  return mswin_link_args if linker == "cl"

  ["-C", "linker=#{linker}", *link_args]
end

# File rubygems/ext/cargo_builder.rb, line 259
def makefile_config(var_name)
  val = RbConfig::MAKEFILE_CONFIG[var_name]

  return unless val

  RbConfig.expand(val.dup)
end

# File rubygems/ext/cargo_builder.rb, line 207
def maybe_resolve_ldflag_variable(input_arg, dest_dir)
  var_matches = input_arg.match(/\$\((\w+)\)/)

  return input_arg unless var_matches

  var_name = var_matches[1]

  return input_arg if var_name.nil? || var_name.chomp.empty?

  case var_name
  # On windows, it is assumed that mkmf has setup an exports file for the
  # extension, so we have to to create one ourselves.
  when "DEFFILE"
    write_deffile(dest_dir)
  else
    RbConfig::CONFIG[var_name]
  end
end

# File rubygems/ext/cargo_builder.rb, line 197
def mingw_target?
  makefile_config("target_os").include?("mingw")
end

# File rubygems/ext/cargo_builder.rb, line 255
def mkmf_libpath
  ["-L", "native=#{makefile_config("libdir")}"]
end

# File rubygems/ext/cargo_builder.rb, line 189
def msvc_target?
  makefile_config("target_os").include?("msvc")
end

# File rubygems/ext/cargo_builder.rb, line 117
def mswin_link_args
  args = []
  args += ["-l", makefile_config("LIBRUBYARG_SHARED").chomp(".lib")]
  args += split_flags("LIBS").flat_map {|lib| ["-l", lib.chomp(".lib")] }
  args += split_flags("LOCAL_LIBS").flat_map {|lib| ["-l", lib.chomp(".lib")] }
  args
end

# File rubygems/ext/cargo_builder.rb, line 83
def platform_specific_rustc_args(dest_dir, flags = [])
  if mingw_target?
    # On mingw platforms, mkmf adds libruby to the linker flags
    flags += libruby_args(dest_dir)

    # Make sure ALSR is used on mingw
    # see https://github.com/rust-lang/rust/pull/75406/files
    flags += ["-C", "link-arg=-Wl,--dynamicbase"]
    flags += ["-C", "link-arg=-Wl,--disable-auto-image-base"]

    # If the gem is installed on a host with build tools installed, but is
    # run on one that isn't the missing libraries will cause the extension
    # to fail on start.
    flags += ["-C", "link-arg=-static-libgcc"]
  elsif darwin_target?
    # Ventura does not always have this flag enabled
    flags += ["-C", "link-arg=-Wl,-undefined,dynamic_lookup"]
  end

  flags
end

# File rubygems/ext/cargo_builder.rb, line 67
def rb_config_env
  result = {}
  RbConfig::CONFIG.each {|k, v| result["RBCONFIG_#{k}"] = v }
  result
end

# File rubygems/ext/cargo_builder.rb, line 138
def rename_cdylib_for_ruby_compatibility(dest_path)
  new_path = final_extension_path(dest_path)
  FileUtils.cp(cargo_dylib_path(dest_path), new_path)
  new_path
end

# File rubygems/ext/cargo_builder.rb, line 131
def ruby_static?
  return true if %w[1 true].include?(ENV["RUBY_STATIC"])

  makefile_config("ENABLE_SHARED") == "no"
end

# File rubygems/ext/cargo_builder.rb, line 170
def rustc_dynamic_linker_flags(dest_dir)
  split_flags("DLDFLAGS")
    .map {|arg| maybe_resolve_ldflag_variable(arg, dest_dir) }
    .compact
    .flat_map {|arg| ldflag_to_link_modifier(arg) }
end

# File rubygems/ext/cargo_builder.rb, line 177
def rustc_lib_flags(dest_dir)
  split_flags("LIBS").flat_map {|arg| ldflag_to_link_modifier(arg) }
end

# File rubygems/ext/cargo_builder.rb, line 242
def so_ext
  return RbConfig::CONFIG["SOEXT"] if RbConfig::CONFIG.key?("SOEXT")

  if win_target?
    "dll"
  elsif darwin_target?
    "dylib"
  else
    "so"
  end
end

# File rubygems/ext/cargo_builder.rb, line 181
def split_flags(var)
  Shellwords.split(RbConfig::CONFIG.fetch(var, ""))
end

# File rubygems/ext/cargo_builder.rb, line 144
def validate_cargo_build!(dir)
  dylib_path = cargo_dylib_path(dir)

  raise DylibNotFoundError, dir unless File.exist?(dylib_path)

  dylib_path
end

# File rubygems/ext/cargo_builder.rb, line 201
def win_target?
  target_platform = RbConfig::CONFIG["target_os"]
  !!Gem::WIN_PATTERNS.find {|r| target_platform =~ r }
end

# File rubygems/ext/cargo_builder.rb, line 226
def write_deffile(dest_dir)
  deffile_path = File.join(dest_dir, "#{spec.name}-#{RbConfig::CONFIG["arch"]}.def")
  export_prefix = makefile_config("EXPORT_PREFIX") || ""

  File.open(deffile_path, "w") do |f|
    f.puts "EXPORTS"
    f.puts "#{export_prefix.strip}Init_#{spec.name}"
  end

  deffile_path
end