Class: Rbcli::RemoteExec

Inherits:
Object
  • Object
show all
Defined in:
lib/rbcli/features/remote_exec.rb

Instance Method Summary collapse

Constructor Details

#initialize(cmd, connection_string, remote_identity, params, args, global_opts, config) ⇒ RemoteExec

Returns a new instance of RemoteExec.



27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/rbcli/features/remote_exec.rb', line 27

def initialize cmd, connection_string, remote_identity, params, args, global_opts, config
  @cmd = cmd
  @params = params.to_json.gsub("\"", "\\\"")
  @args = args.to_json.gsub("\"", "\\\"")
  @global_opts = global_opts.to_json.gsub("\"", "\\\"")
  @config = config.to_json.gsub("\"", "\\\"")

  @remote_host = parse_connection connection_string, remote_identity

  ## SCRIPT COMMANDS
  if !@cmd.script.nil?
    @execution_block = lambda do |ssh|
      # Set paths
      tmpdir = ssh.exec! 'mktemp -d /tmp/rbcli.XXXXXXXXXXXX'
      remote_script_path = "#{tmpdir.strip}/script.sh"
      local_rbclilib_path = "#{File.dirname(__FILE__)}/../../../lib-sh/lib-rbcli.sh"
      remote_rbclilib_path = "#{tmpdir.strip}/lib-rbcli.sh"

      # Upload scripts
      ssh.scp.upload @cmd.script.path, remote_script_path
      ssh.scp.upload local_rbclilib_path, remote_rbclilib_path
      ssh.exec! "chmod 0700 #{remote_script_path}"

      # We need to test for JQ -- if it is not present we ask the user for the sudo password
      jq_found = ssh.exec!('which jq').exitstatus == 0
      if jq_found
        sudopw = nil
      else
        print "JQ not found on remote system. Please enter sudo password to install it, or leave blank if no password needed: "
        sudopw = gets.chomp
      end

      # Since we need to sudo, we need to send the sudo password when prompted
      #result = ''
      channel = ssh.open_channel do |channel, success|
        channel.on_data do |channel, data|
          print data.to_s
          if !jq_found and data =~ /^\[sudo\] password for /
            channel.send_data "#{sudopw}\n"
            # else
            #   result += data.to_s
          end
        end
        channel.request_pty
        channel.exec "__RBCLI_PARAMS=\"#{@params}\" __RBCLI_ARGS=\"#{@args}\" __RBCLI_GLOBAL=\"#{@global_opts}\" __RBCLI_CONFIG=\"#{@config}\" source #{remote_script_path}"
        channel.wait
      end
      channel.wait
      #puts result

      # Cleanup
      ssh.exec! "rm -rf #{tmpdir}"
    end

    ## EXTERN COMMANDS
  elsif !@cmd.extern.nil?
    @execution_block = lambda do |ssh|
      channel = ssh.open_channel do |channel, success|
        channel.on_data do |channel, data|
          print data.to_s
        end
        channel.request_pty
        channel.exec "__RBCLI_PARAMS=\"#{@params}\" __RBCLI_ARGS=\"#{@args}\" __RBCLI_GLOBAL=\"#{@global_opts}\" __RBCLI_CONFIG=\"#{@config}\" __RBCLI_MYVARS=\"#{@myvars}\" #{@cmd.extern.path}"
        channel.wait
      end
      channel.wait
    end

    ## STANDARD COMMANDS
  else
    @execution_block = lambda do |ssh|
      raise Exception.new "Warning: Remote SSH Execution does not yetexist for Standard Ruby commands. Please use a Script or External command to use this feature."
    end
  end

end

Instance Method Details

#run ⇒ Object 



104
105
106
107
108
109
110
111
112
113
114
115
116
117
 
# File 'lib/rbcli/features/remote_exec.rb', line 104

def run

  case @remote_host[:credstype]
  when :keyfile
    Net::SSH.start(@remote_host[:hostname], @remote_host[:username], port: @remote_host[:port], keys: @remote_host[:creds], keys_only: true, &@execution_block)
  when :keytext
    Net::SSH.start(@remote_host[:hostname], @remote_host[:username], port: @remote_host[:port], key_data: @remote_host[:creds], keys_only: true, &@execution_block)
  when :password
    Net::SSH.start(@remote_host[:hostname], @remote_host[:username], port: @remote_host[:port], password: @remote_host[:creds], keys_only: false, &@execution_block)
  else
    raise Exception.new "Invalid SSH Connection: No credentials specified for host #{@remote_host[:hostname]}"
  end

end