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)

        tempfile = Tempfile.new("bootstrap")

        local_tempfile = tempfile.path
        remote_tempfile = ::File.join("/", "tmp", ::File.basename(local_tempfile))

        ::File.open(local_tempfile, 'w') do |file|
          file.puts(content)
          file.respond_to?(:flush) and file.flush
        end

        self.upload(local_tempfile, remote_tempfile)

        command = Array.new
        command << %(sudo) if (options[:use_sudo] == true)
        command << %(/bin/bash)
        command << remote_tempfile
        command = command.join(' ')

        result = self.exec(command, options)
        result
      end

    end

  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
ztk-1.6.2	lib/ztk/ssh/bootstrap.rb
ztk-1.6.1	lib/ztk/ssh/bootstrap.rb