Class: Vagrant::Config::Loader

Inherits:
Object
  • Object
show all
Defined in:
lib/vagrant/config/loader.rb

Overview

This class is responsible for loading Vagrant configuration, usually in the form of Vagrantfiles.

Loading works by specifying the sources for the configuration as well as the order the sources should be loaded. Configuration set later always overrides those set earlier; this is how configuration "scoping" is implemented.

Instance Method Summary collapse

Constructor Details

#initialize(versions, version_order) ⇒ Loader

Initializes a configuration loader.

Parameters:

  • versions (Registry)

    A registry of the available versions and their associated loaders.

  • version_order (Array)

    An array of the order of the versions in the registry. This is used to determine if upgrades are necessary. Additionally, the last version in this order is always considered the "current" version.



26
27
28
29
30
31
32
33
# File 'lib/vagrant/config/loader.rb', line 26

def initialize(versions, version_order)
  @logger        = Log4r::Logger.new("vagrant::config::loader")
  @config_cache  = {}
  @proc_cache    = {}
  @sources       = {}
  @versions      = versions
  @version_order = version_order
end

Instance Method Details

#load(order) ⇒ Object

This loads the configuration sources in the given order and returns an actual configuration object that is ready to be used.

Parameters:

  • order (Array<Symbol>)

    The order of configuration to load.

Returns:

  • (Object)

    The configuration object. This is different for each configuration version.



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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
# File 'lib/vagrant/config/loader.rb', line 87

def load(order)
  @logger.info("Loading configuration in order: #{order.inspect}")

  unknown_sources = @sources.keys - order
  if !unknown_sources.empty?
    @logger.warn("Unknown config sources: #{unknown_sources.inspect}")
  end

  # Get the current version config class to use
  current_version      = @version_order.last
  current_config_klass = @versions.get(current_version)

  # This will hold our result
  result = current_config_klass.init

  # Keep track of the warnings and errors that may come from
  # upgrading the Vagrantfiles
  warnings = []
  errors   = []

  if !@sources[:root].nil? && @sources[:root].eql?(@sources[:home])
    # Vagrants home dir is set to the same dir as its project directory
    # so we don't want to load and merge the same Vagrantfile config
    # and execute its settings/procs twice
    #
    # Note: This protection won't work if there are two separate but
    # identical Vagrantfiles in the home and project dir
    @logger.info("Duplicate Vagrantfile config objects detected in :root and :home.")
    @sources.delete(:home)
    @logger.info("Removed :home config from being loaded")
  end

  order.each do |key|
    next if !@sources.key?(key)

    @sources[key].each do |version, proc|
      if !@config_cache.key?(proc)
        @logger.debug("Loading from: #{key} (evaluating)")

        # Get the proper version loader for this version and load
        version_loader = @versions.get(version)
        begin
          version_config = version_loader.load(proc)
        rescue NameError => e
          line = "(unknown)"
          path = "(unknown)"
          if e.backtrace && e.backtrace[0]
            backtrace_tokens = e.backtrace[0].split(":")
            path = e.backtrace.first.slice(0, e.backtrace.first.rindex(':')).rpartition(':').first
            backtrace_tokens.each do |part|
              if part =~ /\d+/
                line = part.to_i
                break
              end
            end
          end

          raise Errors::VagrantfileNameError,
            path: path,
            line: line,
            message: e.message.sub(/' for .*$/, "'")
        end

        # Store the errors/warnings associated with loading this
        # configuration. We'll store these for later.
        version_warnings = []
        version_errors   = []

        # If this version is not the current version, then we need
        # to upgrade to the latest version.
        if version != current_version
          @logger.debug("Upgrading config from version #{version} to #{current_version}")
          version_index = @version_order.index(version)
          current_index = @version_order.index(current_version)

          (version_index + 1).upto(current_index) do |index|
            next_version = @version_order[index]
            @logger.debug("Upgrading config to version #{next_version}")

            # Get the loader of this version and ask it to upgrade
            loader = @versions.get(next_version)
            upgrade_result = loader.upgrade(version_config)

            this_warnings = upgrade_result[1]
            this_errors   = upgrade_result[2]
            @logger.debug("Upgraded to version #{next_version} with " +
                          "#{this_warnings.length} warnings and " +
                          "#{this_errors.length} errors")

            # Append loading this to the version warnings and errors
            version_warnings += this_warnings
            version_errors   += this_errors

            # Store the new upgraded version
            version_config = upgrade_result[0]
          end
        end

        # Cache the loaded configuration along with any warnings
        # or errors so that they can be retrieved later.
        @config_cache[proc] = [version_config, version_warnings, version_errors]
      else
        @logger.debug("Loading from: #{key} (cache)")
      end

      # Merge the configurations
      cache_data = @config_cache[proc]
      result = current_config_klass.merge(result, cache_data[0])

      # Append the total warnings/errors
      warnings += cache_data[1]
      errors   += cache_data[2]
    end
  end

  @logger.debug("Configuration loaded successfully, finalizing and returning")
  [current_config_klass.finalize(result), warnings, errors]
end

#partial_load(key) ⇒ Object

This method is used for doing partial loads of the Vagrantfile. It will load the contents of a single location and return the config. No merging is performed and no finalization is applied.

@note: This will load either version, but we assume a v2 result @todo(spox): check version and raise error on v1

Parameters:

  • key (Symbol)

    name of location

Returns:

  • (Object)

    configuration

Raises:

  • (KeyError)


215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
# File 'lib/vagrant/config/loader.rb', line 215

def partial_load(key)
  raise KeyError,
        "Unknown path key provided (#{key})" if !@sources.key?(key)

  version, proc = @sources[key].first
  @logger.debug("Loading from: #{key} (evaluating)")

  # Get the proper version loader for this version and load
  version_loader = @versions.get(version)
  raise KeyError,
        "Failed to create loader for requested version: #{version}" if version_loader.nil?

  begin
    version_config = version_loader.load(proc)
  rescue NameError => e
    line = "(unknown)"
    path = "(unknown)"
    if e.backtrace && e.backtrace[0]
      backtrace_tokens = e.backtrace[0].split(":")
      path = e.backtrace.first.slice(0, e.backtrace.first.rindex(':')).rpartition(':').first
      backtrace_tokens.each do |part|
        if part =~ /\d+/
          line = part.to_i
          break
        end
      end
    end

    raise Errors::VagrantfileNameError,
          path: path,
          line: line,
          message: e.message.sub(/' for .*$/, "'") + "\n#{e.backtrace.join("\n")}" + "\nVersion #{version.inspect} loader: #{version_loader.inspect} versions: #{@versions.inspect}"
  end
  version_config
end

#set(name, sources) ⇒ Object

Set the configuration data for the given name.

The name should be a symbol and must uniquely identify the data being given.

data can either be a path to a Ruby Vagrantfile or a Proc directly. data can also be an array of such values.

At this point, no configuration is actually loaded. Note that calling set multiple times with the same name will override any previously set values. In this way, the last set data for a given name wins.



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
# File 'lib/vagrant/config/loader.rb', line 46

def set(name, sources)
  # Sources should be an array
  sources = [sources] if !sources.kind_of?(Array)

  reliably_inspected_sources = sources.reduce({}) { |accum, source|
    begin
      accum[source] = source.inspect
    rescue Encoding::CompatibilityError
      accum[source] = "<!Vagrant failed to call #inspect source with object id #{source.object_id} and class #{source.class} due to a string encoding error>"
    end

    accum
  }

  @logger.info("Set #{name.inspect} = #{reliably_inspected_sources.values}")

  # Gather the procs for every source, since that is what we care about.
  procs = []
  sources.each do |source|
    if !@proc_cache.key?(source)
      # Load the procs for this source and cache them. This caching
      # avoids the issue where a file may have side effects when loading
      # and loading it multiple times causes unexpected behavior.
      @logger.debug("Populating proc cache for #{reliably_inspected_sources[source]}")
      @proc_cache[source] = procs_for_source(source, reliably_inspected_sources)
    end

    # Add on to the array of procs we're going to use
    procs.concat(@proc_cache[source])
  end

  # Set this source by name.
  @sources[name] = procs
end