46 lines
1.4 KiB
Ruby
46 lines
1.4 KiB
Ruby
module VagrantPlugins
|
|
module DockerProvider
|
|
module Action
|
|
class PrepareSSH
|
|
def initialize(app, env)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
# If we aren't using a host VM, then don't worry about it
|
|
return @app.call(env) if !env[:machine].provider.host_vm?
|
|
|
|
env[:machine].ui.output(I18n.t(
|
|
"docker_provider.ssh_through_host_vm"))
|
|
|
|
# Modify the SSH info to be the host VM's info
|
|
env[:ssh_info] = env[:machine].provider.host_vm.ssh_info
|
|
|
|
# Modify the SSH options for when we `vagrant ssh`...
|
|
ssh_opts = env[:ssh_opts] || {}
|
|
|
|
# Build the command we'll execute within the Docker host machine:
|
|
ssh_command = env[:machine].communicate.container_ssh_command
|
|
if !Array(ssh_opts[:extra_args]).empty?
|
|
ssh_command << " #{Array(ssh_opts[:extra_args]).join(" ")}"
|
|
end
|
|
|
|
# Modify the SSH options for the original command:
|
|
# Append "-t" to force a TTY allocation
|
|
ssh_opts[:extra_args] = ["-t"]
|
|
# Enable Agent forwarding when requested for the target VM
|
|
if env[:machine].ssh_info[:forward_agent]
|
|
ssh_opts[:extra_args] << "-o ForwardAgent=yes"
|
|
end
|
|
ssh_opts[:extra_args] << ssh_command
|
|
|
|
# Set the opts
|
|
env[:ssh_opts] = ssh_opts
|
|
|
|
@app.call(env)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|