76 lines
2.1 KiB
Ruby
76 lines
2.1 KiB
Ruby
require 'optparse'
|
|
|
|
module VagrantPlugins
|
|
module CommandPackage
|
|
class Command < Vagrant.plugin("1", :command)
|
|
def execute
|
|
options = {}
|
|
|
|
opts = OptionParser.new do |opts|
|
|
opts.banner = "Usage: vagrant package [vm-name] [--base name] [--output name.box]"
|
|
opts.separator " [--include one,two,three] [--vagrantfile file]"
|
|
|
|
opts.separator ""
|
|
|
|
opts.on("--base NAME", "Name of a VM in virtualbox to package as a base box") do |b|
|
|
options[:base] = b
|
|
end
|
|
|
|
opts.on("--output NAME", "Name of the file to output") do |o|
|
|
options[:output] = o
|
|
end
|
|
|
|
opts.on("--include x,y,z", Array, "Additional files to package with the box.") do |i|
|
|
options[:include] = i
|
|
end
|
|
|
|
opts.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)
|
|
vm = Vagrant::VM.new(options[:base], @env, @env.config.global, :base => true)
|
|
raise Vagrant::Errors::BaseVMNotFound, :name => options[:base] if !vm.created?
|
|
@logger.debug("Packaging base VM: #{vm.name}")
|
|
package_vm(vm, options)
|
|
end
|
|
|
|
def package_target(name, options)
|
|
with_target_vms(name, :single_target => true) do |vm|
|
|
raise Vagrant::Errors::VMNotCreatedError if !vm.created?
|
|
@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
|
|
|
|
vm.package(opts)
|
|
end
|
|
end
|
|
end
|
|
end
|