43 lines
1.1 KiB
Ruby
43 lines
1.1 KiB
Ruby
module Vagrant
|
|
module Command
|
|
class SSHCommand < Base
|
|
desc "SSH into the currently running Vagrant environment."
|
|
class_option :execute, :type => :string, :default => false, :aliases => "-e"
|
|
register "ssh"
|
|
|
|
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 "Execute: #{options[:execute]}"
|
|
ssh.exec!(options[:execute]) do |channel, type, data|
|
|
ssh_vm.env.ui.info "#{data}"
|
|
end
|
|
end
|
|
end
|
|
|
|
def ssh_connect
|
|
raise VMNotCreatedError.new("The VM must be created for SSH to work. Please run `vagrant up`.") if !ssh_vm.created?
|
|
ssh_vm.ssh.connect
|
|
end
|
|
|
|
def ssh_vm
|
|
@ssh_vm ||= begin
|
|
vm = self.name.nil? && env.multivm? ? env.primary_vm
|
|
raise MultiVMTargetRequired.new("A target or primary VM must be specified for SSH in a multi-vm environment.") if !vm && target_vms.length > 1
|
|
vm = target_vms.first if !vm
|
|
vm
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|