"ssh and su" automation sample with Ruby and except

ssh_and_su_example.rb

# -*- coding: utf-8 -*-

require 'pty'
require 'expect'

=begin

ssh & su example for expect.rb

variables

c   : config hash
cmd : command what you want to exec
      (you can recieve with Highline.ask)

=end

def exec_as_root( cmd, c )
  $expect_verbose = true
  host = c['prompt_host']

  PTY.spawn( ssh_cmdline( c ) ) { |r, w|
    w.sync = true

    # store host info ?
    r.expect( /\(yes\/no\)\?/, 3 ) { |s|
      w.puts "yes"
    }
    r.expect( /password: $/ ) {
      w.puts c['pass']
      sleep 2
    }
    r.expect( "#c['user']@#{host}" ) {
      w.puts "su"
    }
    r.expect( /^Password: $/ ) {
      w.puts c['root_pass']
    }
    r.expect( "root@#{host}" ) {
      w.puts cmd
    }
    r.expect( "root@#{host}" ) {
      w.puts "exit"
    }
  }
end

def ssh_cmdline( c )
  return "ssh -p #{c['port']} #{c['user']}@#{c['host']}"
end