Contents

module Pair
  class Session
    class Tunnel
      attr_accessor :options,  :tunnel
      private       :options=, :tunnel=

      # E.g. options:
      # {
      #   "host_port"  => 22,
      #   "host_login" => "bjeanes",
      #   "port"       => 2222,
      #   "bastion" => {
      #     "host"       => "bastion.pairmill.com",
      #     "ip_address" => "12.12.12.12",
      #     "ssh_port"   => 22,
      #     "host_user"  => "host",
      #     "join_user"  => "join",
      #   }
      # }
      def initialize(options)
        self.options = options
      end

      def open
        self.tunnel = IO.popen(ssh_command)
        puts "SSH tunnel started (PID = #{tunnel.pid})" if $-d

        if block_given?
          yield
          close
        end
      end

      def close
        @tunnel_closed ||= begin
          puts "Closing connection..."
          Process.kill("INT", tunnel.pid)
          Process.wait
          true
        end
      end

      private

      def ssh_command
        options = []

        options << "-nqT" # Run no command, be quiet, don't allocate pseudo-terminal
        options << "-l #{bastion["host_user"]}"
        options << "-p #{bastion["ssh_port"]}" unless bastion["ssh_port"] == 22
        options << "-R #{port}:localhost:#{host_port}"

        "ssh #{bastion["host"]} #{options.join(" ")}"
      end

      def method_missing(method, *args, &block)
        options[method.to_s] || super(method, *args, &block)
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
pair-0.0.4	lib/pair/session/tunnel.rb
pair-0.0.3	lib/pair/session/tunnel.rb
pair-0.0.2	lib/pair/session/tunnel.rb