module CORL::Mixin::Machine::SSH

# File lib/core/mixin/machine/ssh.rb, line 99
def close_ssh_session
  Util::SSH.close_session(node.public_ip, node.user)
end

# File lib/core/mixin/machine/ssh.rb, line 9
def init_ssh_session(reset = false, tries = 5, sleep_secs = 5)
  ssh_wait_for_ready
  
  success = true
      
  begin
    Util::SSH.session(node.public_ip, node.user, node.ssh_port, node.private_key, reset)
          
  rescue Net::SSH::HostKeyMismatch => error
    error.remember_host!
    sleep 0.2
    reset = true
    retry

  rescue Errno::ECONNREFUSED, Net::SSH::ConnectionTimeout, Net::SSH::Disconnect => error     
    if tries > 1
      sleep(sleep_secs)
      
      tries -= 1
      reset  = true
      retry
    else
      success = false
    end
    
  rescue => error
    ui.warn(error)
    success = false
  end
  success
end

# File lib/core/mixin/machine/ssh.rb, line 43
def ssh_download(remote_path, local_path, options = {}, &code)
  config  = Config.ensure(options)
  success = false
  
  begin
    if init_ssh_session
      Util::SSH.download(node.public_ip, node.user, remote_path, local_path, config.export) do |name, received, total|
        code.call(name, received, total) if code
      end
      success = true
    end  
  rescue => error
    ui.error(error.message)
  end
  
  success
end

# File lib/core/mixin/machine/ssh.rb, line 83
def ssh_exec(commands, options = {}, &code)
  config  = Config.ensure(options)
  results = nil
  
  if commands
    if init_ssh_session
      results = Util::SSH.exec(node.public_ip, node.user, commands) do |type, command, data|
        code.call(type, command, data) if code  
      end
    end
  end
  results
end

# File lib/core/mixin/machine/ssh.rb, line 105
def ssh_terminal(user, options = {})
  Util::SSH.terminal(node.public_ip, user, Config.ensure(options).export)
end

# File lib/core/mixin/machine/ssh.rb, line 63
def ssh_upload(local_path, remote_path, options = {}, &code)
  config  = Config.ensure(options)
  success = false
  
  begin
    if init_ssh_session
      Util::SSH.upload(node.public_ip, node.user, local_path, remote_path, config.export) do |name, sent, total|
        code.call(name, sent, total) if code
      end
      success = true
    end  
  rescue => error
    ui.error(error.message)
  end
  
  success  
end

# File lib/core/mixin/machine/ssh.rb, line 112
def ssh_wait_for_ready
  # Override in class if needed (see Fog Machine provider)
end