summaryrefslogtreecommitdiff
path: root/lib/leap_cli/commands/shell.rb
blob: 5c16f92acc25ea0cd3f0af2a924789f390d2eb70 (plain)
1
2
3
4
5
6
7
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
module LeapCli; module Commands

  desc 'Log in to the specified node with an interactive shell.'
  arg_name 'NAME' #, :optional => false, :multiple => false
  command :ssh do |c|
    c.action do |global_options,options,args|
      exec_ssh(:ssh, args)
    end
  end

  desc 'Log in to the specified node with an interactive shell using mosh.'
  arg_name 'NAME'
  command :mosh do |c|
    c.action do |global_options,options,args|
      exec_ssh(:mosh, args)
    end
  end

  private

  def exec_ssh(cmd, args)
    node = get_node_from_args(args)
    options = [
      "-o 'HostName=#{node.ip_address}'",
      "-o 'HostKeyAlias=#{node.name}'",
      "-o 'GlobalKnownHostsFile=#{path(:known_hosts)}'",
      "-o 'StrictHostKeyChecking=yes'"
    ]
    if node.vagrant?
      options << "-i #{vagrant_ssh_key_file}"
    end
    username = 'root'
    # the echo sets the terminal title. it would be better to do this on the server
    ssh = "ssh -l #{username} -p #{node.ssh.port} #{options.join(' ')}"
    if cmd == :ssh
      command = "#{ssh} #{node.name}"
    elsif cmd == :mosh
      command = "mosh --ssh \"#{ssh}\" #{node.name}"
    end
    log 2, command
    title = "echo -n \"\\033]0;#{username}@#{node.domain.full}\007\""
    exec "#{title} && #{command}"
  end

end; end