summaryrefslogtreecommitdiff
path: root/spec/bundler/support/rubygems_version_manager.rb
blob: c174c461f01915236648a23926f82bf813478b86 (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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
# frozen_string_literal: true

require_relative "options"
require_relative "env"
require_relative "subprocess"

class RubygemsVersionManager
  include Spec::Options
  include Spec::Env
  include Spec::Subprocess

  def initialize(source)
    @source = source
  end

  def switch
    return if use_system?

    assert_system_features_not_loaded!

    switch_local_copy_if_needed

    reexec_if_needed
  end

  def assert_system_features_not_loaded!
    at_exit do
      rubylibdir = RbConfig::CONFIG["rubylibdir"]

      rubygems_path = rubylibdir + "/rubygems"
      rubygems_default_path = rubygems_path + "/defaults"

      bundler_path = rubylibdir + "/bundler"

      bad_loaded_features = $LOADED_FEATURES.select do |loaded_feature|
        (loaded_feature.start_with?(rubygems_path) && !loaded_feature.start_with?(rubygems_default_path)) ||
          loaded_feature.start_with?(bundler_path)
      end

      errors = if bad_loaded_features.any?
        all_commands_output + "the following features were incorrectly loaded:\n#{bad_loaded_features.join("\n")}"
      end

      raise errors if errors
    end
  end

  private

  def use_system?
    @source.nil?
  end

  def reexec_if_needed
    return unless rubygems_unrequire_needed?

    require "rbconfig"

    cmd = [RbConfig.ruby, $0, *ARGV].compact

    ENV["RUBYOPT"] = opt_add("-I#{File.join(local_copy_path, "lib")}", opt_remove("--disable-gems", ENV["RUBYOPT"]))

    exec(ENV, *cmd)
  end

  def switch_local_copy_if_needed
    return unless local_copy_switch_needed?

    git("checkout #{target_tag}", local_copy_path)

    ENV["RGV"] = local_copy_path
  end

  def rubygems_unrequire_needed?
    require "rubygems"
    !$LOADED_FEATURES.include?(File.join(local_copy_path, "lib/rubygems.rb"))
  end

  def local_copy_switch_needed?
    !source_is_path? && target_tag != local_copy_tag
  end

  def target_tag
    @target_tag ||= resolve_target_tag
  end

  def local_copy_tag
    git("rev-parse --abbrev-ref HEAD", local_copy_path)
  end

  def local_copy_path
    @local_copy_path ||= resolve_local_copy_path
  end

  def resolve_local_copy_path
    return expanded_source if source_is_path?

    rubygems_path = File.join(source_root, "tmp/rubygems")

    unless File.directory?(rubygems_path)
      git("clone .. #{rubygems_path}", source_root)
    end

    rubygems_path
  end

  def source_is_path?
    File.directory?(expanded_source)
  end

  def expanded_source
    @expanded_source ||= File.expand_path(@source, source_root)
  end

  def source_root
    @source_root ||= File.expand_path(ruby_core? ? "../../.." : "../..", __dir__)
  end

  def resolve_target_tag
    return "v#{@source}" if @source.match?(/^\d/)

    @source
  end
end