From 54e3b5d64a4be7fea437e5ff27d7300b07e45a4a Mon Sep 17 00:00:00 2001
From: Mathieu Mitchell <mmitchell@iweb.com>
Date: Thu, 1 Sep 2016 12:27:58 -0400
Subject: [PATCH] Generate and provide an SSH config file to shell script being
 run

Currently, shell scripts being invoked have no way of finding what
machine is being provisioned, or what are the connection informations to
use. Trying to invoke ``vagrant ssh-config`` in the inline script is not
possible because Vagrant locks the machine.

This commit generates an "ssh-config" file (similar to the ssh-config
target) as a temporary file and provides the path to it in the
VAGRANT_SSH_CONFIGFILE environment variable.
---
 lib/vagrant-host-shell/provisioner.rb | 54 ++++++++++++++++++++-------
 1 file changed, 41 insertions(+), 13 deletions(-)

diff --git a/lib/vagrant-host-shell/provisioner.rb b/lib/vagrant-host-shell/provisioner.rb
index 791ca1d..8e55147 100644
--- a/lib/vagrant-host-shell/provisioner.rb
+++ b/lib/vagrant-host-shell/provisioner.rb
@@ -1,21 +1,49 @@
 module VagrantPlugins::HostShell
   class Provisioner < Vagrant.plugin('2', :provisioner)
     def provision
-      result = Vagrant::Util::Subprocess.execute(
-        'bash',
-        '-c',
-        config.inline,
-        :notify => [:stdout, :stderr],
-        :workdir => config.cwd,
-        :env => {PATH: ENV["VAGRANT_OLD_ENV_PATH"]},
-      ) do |io_name, data|
-        @machine.env.ui.info "[#{io_name}] #{data}"
-      end
+      ssh_info = @machine.ssh_info
+      raise Vagrant::Errors::SSHNotReady if ssh_info.nil?
 
-      if config.abort_on_nonzero && !result.exit_code.zero?      
-        raise VagrantPlugins::HostShell::Errors::NonZeroStatusError.new(config.inline, result.exit_code)  
-      end
+      variables = {
+        host_key: machine.name,
+        ssh_host: ssh_info[:host],
+        ssh_port: ssh_info[:port],
+        ssh_user: ssh_info[:username],
+        keys_only: ssh_info[:keys_only],
+        paranoid: ssh_info[:paranoid],
+        private_key_path: ssh_info[:private_key_path],
+        log_level: ssh_info[:log_level],
+        forward_agent: ssh_info[:forward_agent],
+        forward_x11:   ssh_info[:forward_x11],
+        proxy_command: ssh_info[:proxy_command],
+        ssh_command:   ssh_info[:ssh_command],
+        forward_env:   ssh_info[:forward_env],
+      }
+
+      template = "commands/ssh_config/config"
+      ssh_config = Vagrant::Util::TemplateRenderer.render(template, variables)
 
+      Tempfile.open("#{machine.name}") do |f|
+        f.write(ssh_config)
+        f.flush()
+        result = Vagrant::Util::Subprocess.execute(
+          'bash',
+          '-c',
+          config.inline,
+          :notify => [:stdout, :stderr],
+          :workdir => config.cwd,
+          :env => {
+            PATH: ENV["VAGRANT_OLD_ENV_PATH"],
+            VAGRANT_SSH_CONFIGFILE: f.path,
+          },
+        ) do |io_name, data|
+          @machine.env.ui.info "[#{io_name}] #{data}"
+        end
+
+        if config.abort_on_nonzero && !result.exit_code.zero?
+          raise VagrantPlugins::HostShell::Errors::NonZeroStatusError.new(config.inline, result.exit_code)
+        end
+      end
     end
   end
 end