47fe278667
The built-in middleware sequences will now be hardcoded onto Vagrant::Action. Other plugins can hook into these sequences to provide verification and so on. So the VirtualBox plugin will hook into that action sequence and add verification.
76 lines
2.4 KiB
Ruby
76 lines
2.4 KiB
Ruby
require "vagrant/util/platform"
|
|
|
|
module Vagrant
|
|
module Action
|
|
module Builtin
|
|
# This middleware will download a remote box and add it to the
|
|
# given box collection.
|
|
class BoxAdd
|
|
def initialize(app, env)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
# Instantiate the downloader
|
|
downloader = download_klass(env[:box_url]).new(env[:ui])
|
|
env[:ui].info I18n.t("vagrant.actions.box.download.with",
|
|
:class => downloader.class.to_s)
|
|
|
|
# Download the box to a temporary path. We store the temporary
|
|
# path as an instance variable so that the `#recover` method can
|
|
# access it.
|
|
@temp_path = env[:tmp_path].join("box" + Time.now.to_i.to_s)
|
|
File.open(@temp_path, Vagrant::Util::Platform.tar_file_options) do |f|
|
|
downloader.download!(env[:box_url], f)
|
|
end
|
|
|
|
# Add the box
|
|
env[:ui].info I18n.t("vagrant.actions.box.add.adding", :name => env[:box_name])
|
|
begin
|
|
env[:box_collection].add(@temp_path, env[:box_name])
|
|
rescue Vagrant::Errors::BoxUpgradeRequired
|
|
# Upgrade the box
|
|
env[:box_collection].upgrade(env[:box_name])
|
|
|
|
# Try adding it again
|
|
retry
|
|
end
|
|
|
|
# Call the 'recover' method in all cases to clean up the
|
|
# downloaded temporary file.
|
|
recover(env)
|
|
|
|
# Carry on!
|
|
@app.call(env)
|
|
end
|
|
|
|
def download_klass(url)
|
|
# This is hardcoded for now. In the future I'd like to make this
|
|
# pluginnable as well.
|
|
classes = [Downloaders::HTTP, Downloaders::File]
|
|
|
|
# Find the class to use.
|
|
classes.each_index do |i|
|
|
klass = classes[i]
|
|
|
|
# Use the class if it matches the given URI or if this
|
|
# is the last class...
|
|
return klass if classes.length == (i + 1) || klass.match?(url)
|
|
end
|
|
|
|
# If no downloader knows how to download this file, then we
|
|
# raise an exception.
|
|
raise Errors::BoxDownloadUnknownType
|
|
end
|
|
|
|
def recover(env)
|
|
if @temp_path && File.exist?(@temp_path)
|
|
env[:ui].info I18n.t("vagrant.actions.box.download.cleaning")
|
|
File.unlink(@temp_path)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|