aboutsummaryrefslogtreecommitdiffstats
path: root/lib/bundler/environment_preserver.rb
blob: a891f4854d1e4424ea7d31f4820e790740059107 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
# frozen_string_literal: true
module Bundler
  class EnvironmentPreserver
    # @param env [ENV]
    # @param keys [Array<String>]
    def initialize(env, keys)
      @original = env.to_hash
      @keys = keys
      @prefix = "BUNDLER_ORIG_"
    end

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

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