Sha256: 33c94b4d9f747bf1b5f27cda82493c7dbdec9bb2faaa07d6d95c7518b4bf4333
Contents?: true
Size: 1.89 KB
Versions: 8
Compression:
Stored size: 1.89 KB
Contents
module ZTK
class SSH
# SSH Bootstrap Functionality
module Bootstrap
require 'tempfile'
# SSH Bootstrap
#
# Renders the *content* string into a file on the remote host and proceeds
# to execute it via */bin/bash*. Sudo is prefixed by default, but can be
# disabled.
#
# @example Sample Bootstrap, assuming the @ssh variable is an instance of ZTK::SSH connected to a host.
# @ssh.bootstrap(IO.read("bootstrap.sh"))
#
# @example
# @ssh.bootstrap("apt-get -y upgrade")
#
# @param [String] content What to render out to the bootstrap file we will
# execute.
# @param [Hash] options The options hash. This will also accept options
# for #exec in order to better control the bootstrapping execution.
# @option options [String] :use_sudo True if we should execute the
# bootstrap via sudo; False to execute it as the defined user.
def bootstrap(content, options={})
options = {
:silence => true,
:use_sudo => true
}.merge(options)
bootstrap_tempfile = Tempfile.new("bootstrap")
remote_tempfile = ::File.join("", "tmp", ::File.basename(bootstrap_tempfile.path.dup))
bootstrap_tempfile.close!
local_tempfile = Tempfile.new("tempfile-local")
local_tempfile.puts(content)
local_tempfile.respond_to?(:flush) and local_tempfile.flush
command = Array.new
command << %(sudo) if (options[:use_sudo] == true)
command << %(/bin/bash)
command << remote_tempfile
command = command.join(' ')
result = nil
ZTK::RescueRetry.try(:tries => 3, :on_retry => method(:on_retry)) do
self.upload(local_tempfile.path, remote_tempfile)
result = self.exec(command, options)
end
local_tempfile.close!
result
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems