90 lines
2.5 KiB
Ruby
90 lines
2.5 KiB
Ruby
require 'optparse'
|
|
|
|
module VagrantPlugins
|
|
module CommandSnapshot
|
|
module Command
|
|
class Root < Vagrant.plugin("2", :command)
|
|
def self.synopsis
|
|
"manages snapshots: saving, restoring, etc."
|
|
end
|
|
|
|
def initialize(argv, env)
|
|
super
|
|
|
|
@main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
|
|
|
|
@subcommands = Vagrant::Registry.new
|
|
@subcommands.register(:save) do
|
|
require_relative "save"
|
|
Save
|
|
end
|
|
|
|
@subcommands.register(:restore) do
|
|
require_relative "restore"
|
|
Restore
|
|
end
|
|
|
|
@subcommands.register(:delete) do
|
|
require_relative "delete"
|
|
Delete
|
|
end
|
|
|
|
@subcommands.register(:list) do
|
|
require_relative "list"
|
|
List
|
|
end
|
|
|
|
@subcommands.register(:push) do
|
|
require_relative "push"
|
|
Push
|
|
end
|
|
|
|
@subcommands.register(:pop) do
|
|
require_relative "pop"
|
|
Pop
|
|
end
|
|
end
|
|
|
|
def execute
|
|
if @main_args.include?("-h") || @main_args.include?("--help")
|
|
# Print the help for all the commands.
|
|
return help
|
|
end
|
|
|
|
# If we reached this far then we must have a subcommand. If not,
|
|
# then we also just print the help and exit.
|
|
command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
|
|
return help if !command_class || !@sub_command
|
|
@logger.debug("Invoking command class: #{command_class} #{@sub_args.inspect}")
|
|
|
|
# Initialize and execute the command class
|
|
command_class.new(@sub_args, @env).execute
|
|
end
|
|
|
|
# Prints the help out for this command
|
|
def help
|
|
opts = OptionParser.new do |opts|
|
|
opts.banner = "Usage: vagrant snapshot <subcommand> [<args>]"
|
|
opts.separator ""
|
|
opts.separator "Available subcommands:"
|
|
|
|
# Add the available subcommands as separators in order to print them
|
|
# out as well.
|
|
keys = []
|
|
@subcommands.each { |key, value| keys << key.to_s }
|
|
|
|
keys.sort.each do |key|
|
|
opts.separator " #{key}"
|
|
end
|
|
|
|
opts.separator ""
|
|
opts.separator "For help on any individual subcommand run `vagrant snapshot <subcommand> -h`"
|
|
end
|
|
|
|
@env.ui.info(opts.help, prefix: false)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|