43 lines
1.1 KiB
Ruby
43 lines
1.1 KiB
Ruby
module Vagrant
|
|
module Command
|
|
class SSHCommand < NamedBase
|
|
class_option :execute, :type => :string, :default => false, :aliases => "-e"
|
|
register "ssh", "SSH into the currently running Vagrant environment."
|
|
|
|
def execute
|
|
if options[:execute]
|
|
ssh_execute
|
|
else
|
|
ssh_connect
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
def ssh_execute
|
|
ssh_vm.ssh.execute do |ssh|
|
|
ssh_vm.env.ui.info I18n.t("vagrant.commands.ssh.execute", :command => options[:execute])
|
|
ssh.exec!(options[:execute]) do |channel, type, data|
|
|
ssh_vm.env.ui.info "#{data}"
|
|
end
|
|
end
|
|
end
|
|
|
|
def ssh_connect
|
|
raise Errors::VMNotCreatedError.new if !ssh_vm.created?
|
|
raise Errors::VMNotRunningError.new if !ssh_vm.vm.running?
|
|
ssh_vm.ssh.connect
|
|
end
|
|
|
|
def ssh_vm
|
|
@ssh_vm ||= begin
|
|
vm = self.name.nil? && env.multivm? ? env.primary_vm : nil
|
|
raise Errors::MultiVMTargetRequired.new(:command => "ssh") if !vm && target_vms.length > 1
|
|
vm = target_vms.first if !vm
|
|
vm
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|