Class: Rosh

Inherits:
Object
  • Object
show all
Defined in:
lib/rosh.rb,
lib/rosh/version.rb

Constant Summary collapse

VERSION = 
'0.8.0'

Instance Method Summary collapse

Constructor Details

#initialize(*args) ⇒ Rosh

Returns a new instance of Rosh.



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/rosh.rb', line 8

def initialize(*args)
  @interval = 3
  @ssh_opts = []
  alive_interval = 5
  @escape = '^t'
  OptionParser.new("test").tap do |opt|
    opt.banner = 'Usage: rosh [options] hostname [session-name]'
    opt.on('-a alive-interval'){|v| alive_interval = v.to_i}
    opt.on('-e escape'){|v| @escape = v}
    opt.on('-I interval'){|v| @interval = v.to_f}
    opt.on('-V'){|v| @verbose = true}
  end.parse! args
  @host, @name = *args, :default
  abort 'hostname is required' if @host == :default
  @ssh_opts << "-o ServerAliveInterval=#{alive_interval}"
  @ssh_opts << "-o ServerAliveCountMax=1"

  # check ~/.ssh/config to resolve alias name
  config = Net::SSH::Config.for(@host)
  if @verbose
    puts "ssh-config: #{config}"
  end
  @host = config[:host_name] if config[:host_name]
  @ssh_opts << "-l #{config[:user]}" if config[:user]
  @ssh_opts << "-p #{config[:port]}" if config[:port]
  @ssh_opts << "-J #{config[:proxy].jump_proxies}" if config[:proxy]
  if keys = config[:keys]
    keys.each{|k| @ssh_opts << "-i #{k}"}
  end
  if @verbose
    puts "host: #{@host}"
    puts "name: #{@name}"
    puts "interval: #{@interval}"
    puts "alive_interval: #{alive_interval}"
    puts "options: #{@ssh_opts*' '}"
  end
  @first_try = true
end

Instance Method Details

#connect ⇒ Object 



47
48
49
50
51
52
53
54
55
 
# File 'lib/rosh.rb', line 47

def connect
  cmd = ["ssh", *@ssh_opts, resolv,
    '-t', "'screen -rx #{@name}'", '2>/dev/null']*' '
  if @verbose
    puts "connecting to #{@host}..."
    puts cmd
  end
  reconnect until system cmd
end

#reconnect ⇒ Object 



57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
# File 'lib/rosh.rb', line 57

def reconnect
  if @first_try
    if !sh('-p 0 -X echo ok', '2>&1 >/dev/null')
      print "creating new screen session #{@name}... "
      if sh %{-c /dev/null -e "#{@escape*2}" -dm} and
        sh '-p 0 -X eval "stuff STY=\\040screen\\015"'
        puts "done."
      else
        puts "failed."
      end
    end
    @first_try = false
  else
    sleep [@last_try - Time.now + @interval, 0].max if @last_try
    puts "reconnecting..."
    @last_try = Time.now
  end
end