6ddba4f7b3
Prior to this commit, the vagrant snapshot plugin would save snapshots with existing names which lead to duplicate snapshot names being saved. This commit fixes that by checking to see if the given snapshot name already exists and if so, fails telling the user the given snapshot name already exists. If a user passes a --force flag, vagrant will first delete the existing snapshot, and take a new one with the given name.
61 lines
1.9 KiB
Ruby
61 lines
1.9 KiB
Ruby
require 'optparse'
|
|
|
|
module VagrantPlugins
|
|
module CommandSnapshot
|
|
module Command
|
|
class Save < Vagrant.plugin("2", :command)
|
|
def execute
|
|
options = {}
|
|
options[:force] = false
|
|
|
|
opts = OptionParser.new do |o|
|
|
o.banner = "Usage: vagrant snapshot save [options] [vm-name] <name>"
|
|
o.separator ""
|
|
o.separator "Take a snapshot of the current state of the machine. The snapshot"
|
|
o.separator "can be restored via `vagrant snapshot restore` at any point in the"
|
|
o.separator "future to get back to this exact machine state."
|
|
o.separator ""
|
|
o.separator "Snapshots are useful for experimenting in a machine and being able"
|
|
o.separator "to rollback quickly."
|
|
|
|
o.on("-f", "--force", "Replace snapshot without confirmation") do |f|
|
|
options[:force] = f
|
|
end
|
|
end
|
|
|
|
# Parse the options
|
|
argv = parse_options(opts)
|
|
return if !argv
|
|
if argv.empty? || argv.length > 2
|
|
raise Vagrant::Errors::CLIInvalidUsage,
|
|
help: opts.help.chomp
|
|
end
|
|
|
|
name = argv.pop
|
|
with_target_vms(argv) do |vm|
|
|
if !vm.provider.capability?(:snapshot_list)
|
|
vm.ui.info(I18n.t("vagrant.commands.snapshot.not_supported"))
|
|
next
|
|
end
|
|
|
|
snapshot_list = vm.provider.capability(:snapshot_list)
|
|
|
|
if !snapshot_list.include? name
|
|
vm.action(:snapshot_save, snapshot_name: name)
|
|
elsif options[:force]
|
|
# not a unique snapshot name
|
|
vm.action(:snapshot_delete, snapshot_name: name)
|
|
vm.action(:snapshot_save, snapshot_name: name)
|
|
else
|
|
raise Vagrant::Errors::SnapshotConflictFailed
|
|
end
|
|
end
|
|
|
|
# Success, exit status 0
|
|
0
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|