Sha256: 38643af338c63c95716b97847f7df50826544ffee0cb7f9aee5592db1023f9f7
Contents?: true
Size: 1.76 KB
Versions: 6
Compression:
Stored size: 1.76 KB
Contents
module Utils
module LitaPuppet
# Utility methods for doing things over SSH
module SSH
# Only really used for the over_ssh method to determine success or failure
def calculate_result(output, exception)
result = {}
if exception
result[:exception] = exception
else
result[:exit_status] = output.exit_status
result[:stdout] = output.stdout
result[:stderr] = output.stderr
end
result
end
# Intelligently do some things over SSH
def over_ssh(opts = {})
raise 'MissingSSHHost' unless opts[:host]
raise 'MissingSSHUser' unless opts[:user]
opts[:timeout] ||= 300 # default to a 5 minute timeout
remote = Rye::Box.new(
opts[:host],
user: opts[:user],
auth_methods: ['publickey'],
password_prompt: false,
error: STDOUT # send STDERR to STDOUT for things that actually print
)
exception = nil
# Getting serious about not crashing Lita...
output = begin
# pass our host back to the user to work with
Timeout.timeout(opts[:timeout]) { yield remote }
rescue Rye::Err, StandardError => e
exception = e
ensure
remote.disconnect
end
calculate_result(output, exception)
end
# Provides a super simple way to just run a single command over ssh
def simple_ssh_command(host, user, command, timeout = 300)
over_ssh(host: host, user: user, timeout: timeout) do |server|
server.cd '/tmp'
# Need to use sudo
server.enable_sudo
# scary...
server.disable_safe_mode
server.execute command
end
end
end
end
end
Version data entries
6 entries across 6 versions & 1 rubygems