Sha256: 64bd1725ece94c0c3c74ba836a183a8941c9405792daa6f371e3e365bf613aa9

Contents?: true

Size: 1.94 KB

Versions: 63

Compression:

Stored size: 1.94 KB

Contents

# frozen_string_literal: true

module Bundler
  class EnvironmentPreserver
    INTENTIONALLY_NIL = "BUNDLER_ENVIRONMENT_PRESERVER_INTENTIONALLY_NIL".freeze
    BUNDLER_KEYS = %w[
      BUNDLE_BIN_PATH
      BUNDLE_GEMFILE
      BUNDLER_VERSION
      GEM_HOME
      GEM_PATH
      MANPATH
      PATH
      RB_USER_INSTALL
      RUBYLIB
      RUBYOPT
    ].map(&:freeze).freeze
    BUNDLER_PREFIX = "BUNDLER_ORIG_".freeze

    def self.from_env
      new(env_to_hash(ENV), BUNDLER_KEYS)
    end

    def self.env_to_hash(env)
      to_hash = env.to_hash
      return to_hash unless Gem.win_platform?

      to_hash.each_with_object({}) {|(k,v), a| a[k.upcase] = v }
    end

    # @param env [Hash]
    # @param keys [Array<String>]
    def initialize(env, keys)
      @original = env
      @keys = keys
      @prefix = BUNDLER_PREFIX
    end

    # Replaces `ENV` with the bundler environment variables backed up
    def replace_with_backup
      unless Gem.win_platform?
        ENV.replace(backup)
        return
      end

      # Fallback logic for Windows below to workaround
      # https://bugs.ruby-lang.org/issues/16798. Can be dropped once all
      # supported rubies include the fix for that.

      ENV.clear

      backup.each {|k, v| ENV[k] = v }
    end

    # @return [Hash]
    def backup
      env = @original.clone
      @keys.each do |key|
        value = env[key]
        if !value.nil? && !value.empty?
          env[@prefix + key] ||= value
        elsif value.nil?
          env[@prefix + key] ||= INTENTIONALLY_NIL
        end
      end
      env
    end

    # @return [Hash]
    def restore
      env = @original.clone
      @keys.each do |key|
        value_original = env[@prefix + key]
        next if value_original.nil? || value_original.empty?
        if value_original == INTENTIONALLY_NIL
          env.delete(key)
        else
          env[key] = value_original
        end
        env.delete(@prefix + key)
      end
      env
    end
  end
end

Version data entries

63 entries across 63 versions & 4 rubygems

Version Path
scrapbook-0.3.2 vendor/ruby/2.7.0/gems/bundler-2.3.5/lib/bundler/environment_preserver.rb
bundler-2.3.27 lib/bundler/environment_preserver.rb
rubygems-update-3.3.27 bundler/lib/bundler/environment_preserver.rb
scrapbook-0.3.1 vendor/ruby/2.7.0/gems/bundler-2.3.5/lib/bundler/environment_preserver.rb
rubygems-update-3.2.34 bundler/lib/bundler/environment_preserver.rb
bundler-2.2.34 lib/bundler/environment_preserver.rb
tdiary-5.2.4 vendor/bundle/ruby/3.1.0/gems/bundler-2.3.7/lib/bundler/environment_preserver.rb
rubygems-update-3.3.26 bundler/lib/bundler/environment_preserver.rb
bundler-2.3.26 lib/bundler/environment_preserver.rb
rubygems-update-3.3.25 bundler/lib/bundler/environment_preserver.rb
bundler-2.3.25 lib/bundler/environment_preserver.rb
rubygems-update-3.3.24 bundler/lib/bundler/environment_preserver.rb
bundler-2.3.24 lib/bundler/environment_preserver.rb
rubygems-update-3.3.23 bundler/lib/bundler/environment_preserver.rb
bundler-2.3.23 lib/bundler/environment_preserver.rb
rubygems-update-3.3.22 bundler/lib/bundler/environment_preserver.rb
bundler-2.3.22 lib/bundler/environment_preserver.rb
rubygems-update-3.3.21 bundler/lib/bundler/environment_preserver.rb
bundler-2.3.21 lib/bundler/environment_preserver.rb
rubygems-update-3.3.20 bundler/lib/bundler/environment_preserver.rb