class Gem::AtomicFileWriter

# File lib/rubygems/util/atomic_file_writer.rb, line 14
def self.open(file_name)
  temp_dir = File.dirname(file_name)
  require "tempfile" unless defined?(Tempfile)

  Tempfile.create(".#{File.basename(file_name)}", temp_dir) do |temp_file|
    temp_file.binmode
    return_value = yield temp_file
    temp_file.close

    original_permissions = if File.exist?(file_name)
      File.stat(file_name)
    else
      # If not possible, probe which are the default permissions in the
      # destination directory.
      probe_permissions_in(File.dirname(file_name))
    end

    # Set correct permissions on new file
    if original_permissions
      begin
        File.chown(original_permissions.uid, original_permissions.gid, temp_file.path)
        File.chmod(original_permissions.mode, temp_file.path)
      rescue Errno::EPERM, Errno::EACCES
        # Changing file ownership failed, moving on.
      end
    end

    # Overwrite original file with temp file
    File.rename(temp_file.path, file_name)
    return_value
  end
end