require 'delegate' module Fog module SSH def self.new(address, username, options = {}) if Fog.mocking? Fog::SSH::Mock.new(address, username, options) else Fog::SSH::Real.new(address, username, options) end end class Mock def self.data @data ||= Hash.new do |hash, key| hash[key] = [] end end def self.reset @data= nil end def initialize(address, username, options) @address = address @username = username @options = options end def run(commands, &blk) self.class.data[@address] << {:commands => commands, :username => @username, :options => @options} end end class Real def initialize(address, username, options) require 'net/ssh' key_manager = Net::SSH::Authentication::KeyManager.new(nil, options) unless options[:key_data] || options[:keys] || options[:password] || key_manager.agent raise ArgumentError.new(':key_data, :keys, :password or a loaded ssh-agent is required to initialize SSH') end if options[:key_data] || options[:keys] options[:keys_only] = true #Explicitly set these so net-ssh doesn't add the default keys #as seen at https://github.com/net-ssh/net-ssh/blob/master/lib/net/ssh/authentication/session.rb#L131-146 options[:keys] = [] unless options[:keys] options[:key_data] = [] unless options[:key_data] end @address = address @username = username @options = { :paranoid => false }.merge(options) end def run(commands, &blk) commands = [*commands] results = [] begin Net::SSH.start(@address, @username, @options) do |ssh| commands.each do |command| result = Result.new(command) ssh.open_channel do |ssh_channel| ssh_channel.request_pty ssh_channel.exec(command) do |channel, success| unless success raise "Could not execute command: #{command.inspect}" end channel.on_data do |ch, data| result.stdout << data yield [data, ''] if blk end channel.on_extended_data do |ch, type, data| next unless type == 1 result.stderr << data yield ['', data] if blk end channel.on_request('exit-status') do |ch, data| result.status = data.read_long end channel.on_request('exit-signal') do |ch, data| result.status = 255 end end end ssh.loop results << result end end rescue Net::SSH::HostKeyMismatch => exception exception.remember_host! sleep 0.2 retry end results end end class Result attr_accessor :command, :stderr, :stdout, :status def display_stdout data = stdout.split("\r\n") if data.is_a?(String) Formatador.display_line(data) elsif data.is_a?(Array) Formatador.display_lines(data) end end def display_stderr Formatador.display_line(stderr.split("\r\n")) end def initialize(command) @command = command @stderr = '' @stdout = '' end end end end