86e2b78997
Prior to this commit, the package actions would create a temp dir in the process of packaging and compressing a Vagrant box. This commit ensures that the temp dir is removed once the command has completed so that it doesn't leave around lots of temp directories.
98 lines
2.8 KiB
Ruby
98 lines
2.8 KiB
Ruby
require 'optparse'
|
|
require 'securerandom'
|
|
|
|
module VagrantPlugins
|
|
module CommandPackage
|
|
class Command < Vagrant.plugin("2", :command)
|
|
def self.synopsis
|
|
"packages a running vagrant environment into a box"
|
|
end
|
|
|
|
def execute
|
|
options = {}
|
|
|
|
opts = OptionParser.new do |o|
|
|
o.banner = "Usage: vagrant package [options] [name|id]"
|
|
o.separator ""
|
|
o.separator "Options:"
|
|
o.separator ""
|
|
|
|
o.on("--base NAME", "Name of a VM in VirtualBox to package as a base box (VirtualBox Only)") do |b|
|
|
options[:base] = b
|
|
end
|
|
|
|
o.on("--output NAME", "Name of the file to output") do |output|
|
|
options[:output] = output
|
|
end
|
|
|
|
o.on("--include FILE,FILE..", Array, "Comma separated additional files to package with the box") do |i|
|
|
options[:include] = i
|
|
end
|
|
|
|
o.on("--vagrantfile FILE", "Vagrantfile to package with the box") do |v|
|
|
options[:vagrantfile] = v
|
|
end
|
|
end
|
|
|
|
# Parse the options
|
|
argv = parse_options(opts)
|
|
return if !argv
|
|
|
|
@logger.debug("package options: #{options.inspect}")
|
|
if options[:base]
|
|
package_base(options)
|
|
else
|
|
package_target(argv[0], options)
|
|
end
|
|
|
|
# Success, exit status 0
|
|
0
|
|
end
|
|
|
|
protected
|
|
|
|
def package_base(options)
|
|
# XXX: This whole thing is hardcoded and very temporary. The whole
|
|
# `vagrant package --base` process is deprecated for something much
|
|
# better in the future. We just hardcode this to keep VirtualBox working
|
|
# for now.
|
|
provider = Vagrant.plugin("2").manager.providers[:virtualbox]
|
|
tmp_data_directory = File.join(@env.tmp_path, SecureRandom.uuid)
|
|
FileUtils.mkdir_p(tmp_data_directory)
|
|
begin
|
|
vm = Vagrant::Machine.new(
|
|
options[:base],
|
|
:virtualbox, provider[0], nil, provider[1],
|
|
@env.vagrantfile.config,
|
|
Pathname.new(tmp_data_directory), nil,
|
|
@env, @env.vagrantfile, true)
|
|
@logger.debug("Packaging base VM: #{vm.name}")
|
|
package_vm(vm, options)
|
|
ensure
|
|
FileUtils.rm_rf(tmp_data_directory)
|
|
end
|
|
end
|
|
|
|
def package_target(name, options)
|
|
with_target_vms(name, single_target: true) do |vm|
|
|
@logger.debug("Packaging VM: #{vm.name}")
|
|
package_vm(vm, options)
|
|
end
|
|
end
|
|
|
|
def package_vm(vm, options)
|
|
opts = options.inject({}) do |acc, data|
|
|
k,v = data
|
|
acc["package.#{k}"] = v
|
|
acc
|
|
end
|
|
|
|
env = vm.action(:package, opts)
|
|
temp_dir = env["export.temp_dir"]
|
|
ensure
|
|
FileUtils.rm_rf(temp_dir) if temp_dir
|
|
end
|
|
end
|
|
end
|
|
end
|