79 lines
2.2 KiB
Ruby
79 lines
2.2 KiB
Ruby
module Vagrant
|
|
class Commands
|
|
# Reload the environment. This is almost equivalent to the {up} command
|
|
# except that it doesn't import the VM and do the initialize bootstrapping
|
|
# of the instance. Instead, it forces a shutdown (if its running) of the
|
|
# VM, updates the metadata (shared folders, forwarded ports), restarts
|
|
# the VM, and then reruns the provisioning if enabled.
|
|
class SSH < Base
|
|
Base.subcommand "ssh", self
|
|
description "SSH into the currently running environment"
|
|
|
|
def execute(args=[])
|
|
args = parse_options(args)
|
|
if !options[:execute].empty?
|
|
vms = args[0] ? {args[0] => env.vms[args[0].to_sym]} : env.vms
|
|
vms.each do |name, vm|
|
|
ssh_execute(name, vm)
|
|
end
|
|
else
|
|
ssh_connect(args[0])
|
|
end
|
|
end
|
|
|
|
def ssh_execute(name, vm)
|
|
if vm.nil?
|
|
error_and_exit(:unknown_vm, :vm => name)
|
|
return # for tests
|
|
elsif !vm.created?
|
|
error_and_exit(:environment_not_created)
|
|
return
|
|
end
|
|
|
|
vm.ssh.execute do |ssh|
|
|
options[:execute].each do |command|
|
|
vm.env.logger.info("Execute: #{command}")
|
|
ssh.exec!(command) do |channel, type, data|
|
|
# TODO: Exit status checking?
|
|
vm.env.logger.info("#{type}: #{data}")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def ssh_connect(name)
|
|
if name.nil? && env.multivm?
|
|
if env.primary_vm.nil?
|
|
error_and_exit(:ssh_multivm)
|
|
return # for tests
|
|
end
|
|
end
|
|
|
|
vm = name.nil? ? env.primary_vm : env.vms[name.to_sym]
|
|
if vm.nil?
|
|
error_and_exit(:unknown_vm, :vm => name)
|
|
return # for tests
|
|
end
|
|
|
|
if !vm.created?
|
|
error_and_exit(:environment_not_created)
|
|
return
|
|
else
|
|
vm.ssh.connect
|
|
end
|
|
end
|
|
|
|
def options_spec(opts)
|
|
opts.banner = "Usage: vagrant ssh [--execute COMMAND]"
|
|
|
|
# Defaults
|
|
options[:execute] = []
|
|
|
|
opts.on("-e", "--execute COMMAND", "A command to execute. Multiple -e's may be specified.") do |value|
|
|
options[:execute] << value
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|