Contents

class Server < Screwcap::Base
  def initialize(opts = {})
    super
    self.__options = opts
    self.__name = opts[:name]
    self.__servers = opts[:servers]
    self.__user = opts[:user]
    self.__options[:keys] = [self.__options.delete(:key)] if self.__options[:key]

    if self.__options[:address] and self.__options[:addresses].nil?
      self.__addresses = [self.__options.delete(:address)] 
    else
      self.__addresses = self.__options[:addresses]
    end

    validate

    self
  end

  def __with_connection_for(address, &block)
    if self.__options[:gateway]
      gateway = self.__servers.select {|s| s.__options[:is_gateway] == true }.find {|s| s.__name == self.__options[:gateway] }
      gateway.__get_gateway_connection.ssh(address, self.__user, options_for_net_ssh) do |ssh|
        yield ssh
      end
    else
      Net::SSH.start(address, self.__user, options_for_net_ssh) do |ssh|
        yield ssh
      end
    end
  end

  def __upload_to!(address, local, remote)
    self.__with_connection_for(address) {|ssh| ssh.scp.upload! local, remote }
  end

  protected

  def __get_gateway_connection
    self.__connection ||= Net::SSH::Gateway.new(self.__addresses.first, self.__user, self.__options.reject {|k,v| [:user,:addresses, :gateway, :name, :servers, :is_gateway].include?(k)})
  end

  private

  def validate
    raise Screwcap::InvalidServer, "Please specify an address for the server #{self.__options[:name]}." if self.__addresses.blank?
    raise Screwcap::InvalidServer, "Please specify a username to use for the server #{self.__name}." if self.__user.nil?
    raise Screwcap::InvalidServer, "A gateway can have only one address" if self.__addresses.size > 1 and self.__options[:is_gateway] == true
  end

  def options_for_net_ssh
    self.__options.reject {|k,v| [:user,:addresses, :gateway, :is_gateway, :name, :silent, :servers].include?(k)}
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
screwcap-0.3.2	lib/screwcap/server.rb
screwcap-0.3.1	lib/screwcap/server.rb
screwcap-0.3	lib/screwcap/server.rb