Sha256: 9e00cfbeec01bae81ced911a8663f9dfb0c3da1c869c3a6d84cf49917d114750
Contents?: true
Size: 1.82 KB
Versions: 4
Compression:
Stored size: 1.82 KB
Contents
require "log4r"
require "vagrant/util/platform"
require "vagrant/util/ssh"
require "vagrant/util/shell_quote"
module Vagrant
module Action
module Builtin
# This class will run a single command on the remote machine and will
# mirror the output to the UI. The resulting exit status of the command
# will exist in the `:ssh_run_exit_status` key in the environment.
class SSHRun
# For quick access to the `SSH` class.
include Vagrant::Util
def initialize(app, env)
@app = app
@logger = Log4r::Logger.new("vagrant::action::builtin::ssh_run")
end
def call(env)
# Grab the SSH info from the machine
info = env[:machine].ssh_info
# If the result is nil, then the machine is telling us that it is
# not yet ready for SSH, so we raise this exception.
raise Errors::SSHNotReady if info.nil?
info[:private_key_path] ||= []
if info[:private_key_path].empty?
raise Errors::SSHRunRequiresKeys
end
# Get the command and wrap it in a login shell
command = ShellQuote.escape(env[:ssh_run_command], "'")
command = "#{env[:machine].config.ssh.shell} -c '#{command}'"
# Execute!
opts = env[:ssh_opts] || {}
opts[:extra_args] ||= []
# Allow the user to specify a tty or non-tty manually, but if they
# don't then we default to a TTY
if !opts[:extra_args].include?("-t") && !opts[:extra_args].include?("-T")
opts[:extra_args] << "-t"
end
opts[:extra_args] << command
opts[:subprocess] = true
env[:ssh_run_exit_status] = Util::SSH.exec(info, opts)
# Call the next middleware
@app.call(env)
end
end
end
end
end
Version data entries
4 entries across 1 versions & 1 rubygems