This commit is contained in:
@@ -0,0 +1,86 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
module Bundler
|
||||
class EnvironmentPreserver
|
||||
INTENTIONALLY_NIL = "BUNDLER_ENVIRONMENT_PRESERVER_INTENTIONALLY_NIL"
|
||||
BUNDLER_KEYS = %w[
|
||||
BUNDLE_BIN_PATH
|
||||
BUNDLE_GEMFILE
|
||||
BUNDLER_VERSION
|
||||
BUNDLER_SETUP
|
||||
GEM_HOME
|
||||
GEM_PATH
|
||||
MANPATH
|
||||
PATH
|
||||
RB_USER_INSTALL
|
||||
RUBYLIB
|
||||
RUBYOPT
|
||||
].map(&:freeze).freeze
|
||||
BUNDLER_PREFIX = "BUNDLER_ORIG_"
|
||||
|
||||
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?
|
||||
env[@prefix + key] ||= value
|
||||
else
|
||||
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?
|
||||
if value_original == INTENTIONALLY_NIL
|
||||
env.delete(key)
|
||||
else
|
||||
env[key] = value_original
|
||||
end
|
||||
env.delete(@prefix + key)
|
||||
end
|
||||
env
|
||||
end
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user