63 lines
2.0 KiB
Ruby
63 lines
2.0 KiB
Ruby
require "log4r"
|
|
|
|
module Vagrant
|
|
module Easy
|
|
# Base class for all easy commands. This contains the basic code
|
|
# that knows how to run the easy commands.
|
|
class CommandBase < Vagrant::Command::Base
|
|
# This is the command that this easy command responds to
|
|
attr_reader :command
|
|
|
|
# This is called by the {EasyCommand.create} method when creating
|
|
# an easy command to set the invocation command.
|
|
def self.configure(name, &block)
|
|
# We use class-level instance variables so that each class has
|
|
# its own single command/runner. If we use class variables then this
|
|
# whole base sharse a single one.
|
|
@command = name
|
|
@runner = block
|
|
end
|
|
|
|
def initialize(*args, &block)
|
|
if self.class == CommandBase
|
|
raise "CommandBase must not be instantiated directly. Please subclass."
|
|
end
|
|
|
|
# Let the regular command state setup
|
|
super
|
|
|
|
# Get the command we're listening to and the block we're invoking
|
|
# when we get that command, do some basic validation.
|
|
@command = self.class.instance_variable_get(:@command)
|
|
@runner = self.class.instance_variable_get(:@runner)
|
|
if !@command || !@runner
|
|
raise ArgumentError, "CommandBase requires both a command and a runner"
|
|
end
|
|
|
|
@logger = Log4r::Logger.new("vagrant::easy_command::#{@command}")
|
|
end
|
|
|
|
def execute
|
|
# Build up a basic little option parser
|
|
opts = OptionParser.new do |opts|
|
|
opts.banner = "Usage: vagrant #{@command}"
|
|
end
|
|
|
|
# Parse the options
|
|
argv = parse_options(opts)
|
|
return if !argv
|
|
|
|
# Run the action for each VM.
|
|
@logger.info("Running easy command: #{@command}")
|
|
with_target_vms(argv) do |vm|
|
|
@logger.debug("Running easy command for VM: #{vm.name}")
|
|
@runner.call(Operations.new(vm))
|
|
end
|
|
|
|
# Exit status 0 every time for now
|
|
0
|
|
end
|
|
end
|
|
end
|
|
end
|