Rye::Box
The Rye::Box class represents a machine. All system commands are made through this class.
rbox = Rye::Box.new('filibuster') rbox.hostname # => filibuster rbox.uname # => FreeBSD rbox.uptime # => 20:53 up 1 day, 1:52, 4 users
You can also run local commands through SSH
rbox = Rye::Box.new('localhost') rbox.hostname # => localhost rbox.uname(:a) # => Darwin vanya 9.6.0 ...
Methods
Public Class
Public Instance
- ==
- []
- add_keys
- batch
- cd
- connect
- current_umask
- current_working_directory
- debug?
- disable_quiet_mode
- disable_safe_mode
- disable_sudo
- disconnect
- enable_quiet_mode
- enable_safe_mode
- enable_sudo
- error?
- exception_hook
- exception_hook=
- getenv
- guess_user_home
- host
- host=
- host_key
- impltype
- impltype=
- info?
- inspect
- instance_exec
- interactive_ssh
- keys
- method_missing
- nickname
- nickname=
- opts
- opts=
- ostype
- ostype=
- post_command_hook
- post_command_hook=
- pre_command_hook
- pre_command_hook=
- preview_command
- quiet
- quietly
- remove_keys
- root?
- rye_pty
- rye_shell
- safe
- safe?
- safely
- setenv
- ssh_config_options
- stash
- stash=
- stdout_hook
- stdout_hook=
- sudo
- sudo?
- switch_user
- templates
- templates?
- to_s
- umask=
- unsafely
- user
- via
- via=
- via?
- via_hop
Included modules
Classes and Modules
Public Instance Aliases
add_env | -> | setenv |
add_key | -> | add_keys |
execute | -> | method_missing |
remove_key | -> | remove_keys |
wildly | -> | unsafely |
Public Class methods
-
host The hostname to connect to. Default: localhost.
-
user The username to connect as. Default: SSH config file or current shell user.
-
opts a hash of optional arguments.
The opts hash excepts the following keys:
-
:safe => should Rye be safe? Default: true
-
:port => remote server ssh port. Default: SSH config file or 22
-
:keys => one or more private key file paths (passwordless login)
-
:via => the Rye::Hop to access this host through
-
:info => an IO object to print Rye::Box command info to. Default: nil
-
:debug => an IO object to print Rye::Box debugging info to. Default: nil
-
:error => an IO object to print Rye::Box errors to. Default: STDERR
-
:getenv => pre-fetch host environment variables? (default: true)
-
:password => the user's password (ignored if there's a valid private key)
-
:templates => the template engine to use for uploaded files. One of: :erb (default)
-
:sudo => Run all commands via sudo (default: false)
-
:password_prompt => Show a password prompt on auth failure (default: true)
NOTE: opts can also contain any parameter supported by Net::SSH.start that is not already mentioned above.
# File lib/rye/box.rb, line 114 def initialize(host='localhost', opts={}) ssh_opts = ssh_config_options(host) @rye_exception_hook = {} @rye_host = host if opts[:user] @rye_user = opts[:user] else @rye_user = ssh_opts[:user] || Rye.sysinfo.user end # These opts are use by Rye::Box and also passed to Net::SSH @rye_opts = { :safe => true, :port => ssh_opts[:port], :keys => Rye.keys, :via => nil, :info => nil, :debug => nil, :error => STDERR, :getenv => true, :templates => :erb, :quiet => false, :password_prompt => true }.merge(opts) # Close the SSH session before Ruby exits. This will do nothing # if disconnect has already been called explicitly. at_exit { self.disconnect } # Properly handle whether the opt :via is a +Rye::Hop+ or a +String+ via_hop(@rye_opts.delete(:via)) # @rye_opts gets sent to Net::SSH so we need to remove the keys # that are not meant for it. @rye_safe, @rye_debug = @rye_opts.delete(:safe), @rye_opts.delete(:debug) @rye_info, @rye_error = @rye_opts.delete(:info), @rye_opts.delete(:error) @rye_getenv = {} if @rye_opts.delete(:getenv) # Enable getenv with a hash @rye_ostype, @rye_impltype = @rye_opts.delete(:ostype), @rye_opts.delete(:impltype) @rye_quiet, @rye_sudo = @rye_opts.delete(:quiet), @rye_opts.delete(:sudo) @rye_templates = @rye_opts.delete(:templates) @rye_password_prompt = @rye_opts.delete(:password_prompt) # Store the state of the terminal @rye_stty_save = `stty -g`.chomp rescue nil unless @rye_templates.nil? require @rye_templates.to_s # should be :erb end @rye_opts[:logger] = Logger.new(@rye_debug) if @rye_debug # Enable Net::SSH debugging @rye_opts[:paranoid] ||= true unless @rye_safe == false # See Net::SSH.start @rye_opts[:keys] = [@rye_opts[:keys]].flatten.compact # Just in case someone sends a true value rather than IO object @rye_debug = STDERR if @rye_debug == true || DEBUG @rye_error = STDERR if @rye_error == true @rye_info = STDOUT if @rye_info == true # Add the given private keys to the keychain that will be used for @rye_host add_keys(@rye_opts[:keys]) # We don't want Net::SSH to handle the keypairs. This may change # but for we're letting ssh-agent do it. # TODO: Check if this should ot should not be enabled. #@rye_opts.delete(:keys) # From: capistrano/lib/capistrano/cli.rb STDOUT.sync = true # so that Net::SSH prompts show up debug "ssh-agent info: #{Rye.sshagent_info.inspect}" debug @rye_opts.inspect end
Public Instance methods
Compares itself with the other box. If the hostnames are the same, this will return true. Otherwise false.
# File lib/rye/box.rb, line 407 def ==(other) @rye_host == other.host end
Change the current working directory (sort of).
I haven't been able to wrangle Net::SSH to do my bidding. "My bidding" in this case, is maintaining an open channel between commands. I'm using Net::SSH::Connection::Session#exec for all commands which is like a funky helper method that opens a new channel each time it's called. This seems to be okay for one-off commands but changing the directory only works for the channel it's executed in. The next time exec is called, there's a new channel which is back in the default (home) directory.
Long story short, the work around is to maintain the current directory locally and send it with each command.
rbox.pwd # => /home/rye ($ pwd ) rbox['/usr/bin'].pwd # => /usr/bin ($ cd /usr/bin && pwd) rbox.pwd # => /usr/bin ($ cd /usr/bin && pwd)
# File lib/rye/box.rb, line 245 def [](fpath=nil) if fpath.nil? || fpath.index('/') == 0 @rye_current_working_directory = fpath else # Append to non-absolute paths if @rye_current_working_directory newpath = File.join(@rye_current_working_directory, fpath) @rye_current_working_directory = newpath else @rye_current_working_directory = fpath end end debug "CWD: #{@rye_current_working_directory}" self end
Add one or more private keys to the list of key paths.
-
keys is a list of file paths to private keys
Returns the instance of Box
# File lib/rye/box.rb, line 322 def add_keys(*keys) @rye_opts[:keys] ||= [] @rye_opts[:keys] += keys.flatten.compact @rye_opts[:keys].uniq! self # MUST RETURN self end
Execute a block in the context of an instance of Rye::Box.
rbox = Rye::Box.new rbox.batch do ls :l uname :a end
OR
rbox.batch(&block)
The batch can also accept arguments.
rbox.batch('path/2/file') do |file| ls :l file end
Returns the return value of the block.
# File lib/rye/box.rb, line 550 def batch(*args, &block) self.instance_exec(*args, &block) end
Like [] except it returns an empty Rye::Rap object to mimick a regular command method. Call with nil key (or no arg) to reset.
# File lib/rye/box.rb, line 263 def cd(fpath=nil) Rye::Rap.new(self[fpath]) end
Open an SSH session with +@rye_host+. This called automatically when you the first comamnd is run if it's not already connected. Raises a Rye::NoHost exception if +@rye_host+ is not specified. Will attempt a password login up to 3 times if the initial authentication fails.
-
reconnect Disconnect first if already connected. The default
is true. When set to false, connect will do nothing if already connected.
# File lib/rye/box.rb, line 649 def connect(reconnect=true) raise Rye::NoHost unless @rye_host return if @rye_ssh && !reconnect disconnect if @rye_ssh if @rye_via debug "Opening connection to #{@rye_host} as #{@rye_user}, via #{@rye_via.host}" else debug "Opening connection to #{@rye_host} as #{@rye_user}" end highline = HighLine.new # Used for password prompt retried = 0 @rye_opts[:keys].compact! # A quick fix in Windows. TODO: Why is there a nil? begin if @rye_via # tell the +Rye::Hop+ what and where to setup, # it returns the local port used @rye_localport = @rye_via.fetch_port(@rye_host, @rye_opts[:port].nil? ? 22 : @rye_opts[:port] ) debug "fetched localport #{@rye_localport}" @rye_ssh = Net::SSH.start("localhost", @rye_user, @rye_opts.merge(:port => @rye_localport) || {}) else @rye_ssh = Net::SSH.start(@rye_host, @rye_user, @rye_opts || {}) end rescue Net::SSH::HostKeyMismatch => ex STDERR.puts ex.message print "\a" if @rye_info # Ring the bell raise ex rescue Net::SSH::AuthenticationFailed => ex print "\a" if retried == 0 && @rye_info # Ring the bell once retried += 1 @rye_opts[:auth_methods] ||= [] # Raise Net::SSH::AuthenticationFailed if publickey is the # only auth method if @rye_opts[:auth_methods] == ["publickey"] raise ex elsif @rye_password_prompt && (STDIN.tty? && retried <= 3) STDERR.puts "Passwordless login failed for #{@rye_user}" @rye_opts[:password] = highline.ask("Password: ") { |q| q.echo = '' }.strip @rye_opts[:auth_methods].push *['keyboard-interactive', 'password'] retry else raise ex end end # We add :auth_methods (a Net::SSH joint) to force asking for a # password if the initial (key-based) authentication fails. We # need to delete the key from @rye_opts otherwise it lingers until # the next connection (if we switch_user is called for example). @rye_opts.delete :auth_methods if @rye_opts.has_key?(:auth_methods) self end
The most recent valud for umask (or 0022)
# File lib/rye/box.rb, line 77 def current_umask; @rye_current_umask; end
# File lib/rye/box.rb, line 74 def current_working_directory; @rye_current_working_directory; end
# File lib/rye/box.rb, line 71 def disable_quiet_mode; @rye_quiet = false; end
# File lib/rye/box.rb, line 67 def disable_safe_mode; @rye_safe = false; end
# File lib/rye/box.rb, line 49 def disable_sudo; @rye_sudo = false; end
Close the SSH session with +@rye_host+. This is called automatically at exit if the connection is open.
# File lib/rye/box.rb, line 706 def disconnect return unless @rye_ssh && !@rye_ssh.closed? begin if @rye_ssh.busy?; info "Is something still running? (ctrl-C to exit)" Timeout::timeout(10) do @rye_ssh.loop(0.3) { @rye_ssh.busy?; } end end debug "Closing connection to #{@rye_ssh.host}" @rye_ssh.close if @rye_via debug "disconnecting Hop #{@rye_via.host}" @rye_via.disconnect end rescue SystemCallError, Timeout::Error => ex error "Rye::Box: Disconnect timeout (#{ex.message})" debug ex.backtrace rescue Interrupt debug "Exiting..." end end
# File lib/rye/box.rb, line 70 def enable_quiet_mode; @rye_quiet = true; end
# File lib/rye/box.rb, line 66 def enable_safe_mode; @rye_safe = true; end
Supply a block to be called whenever there's an Exception. It's called with 1 argument: the exception class. If the exception block returns :retry, the command will be executed again.
e.g.
rbox.exception_hook(CommandNotFound) do |ex| STDERR.puts "An error occurred: #{ex.class}" choice = Annoy.get_user_input('(S)kip (R)etry (A)bort: ') if choice == 'R' :retry elsif choice == 'S' # do nothing else exit # ! end end
# File lib/rye/box.rb, line 526 def exception_hook(klass, &block) @rye_exception_hook[klass] = block if block @rye_exception_hook[klass] end
A Hash. The keys are exception classes, the values are Procs to execute
# File lib/rye/box.rb, line 90 def exception_hook=(val); @rye_exception_hook = val; end
Returns the hash containing the parsed output of "env" on the remote machine. If the initialize option :getenv was set to false, this will return an empty hash. This is a lazy loaded method so it fetches the remote envvars the first time this method is called.
puts rbox.getenv['HOME'] # => "/home/gloria" (remote)
NOTE: This method should not raise an exception under normal circumstances.
# File lib/rye/box.rb, line 367 def getenv(key=nil) if @rye_getenv && @rye_getenv.empty? && self.can?(:env) vars = self.quietly { env } rescue [] vars.each do |nvpair| # Parse "GLORIA_HOME=/gloria/lives/here" into a name/value # pair. The regexp ensures we split only at the 1st = sign n, v = nvpair.scan(/\A([\w_-]+?)=(.+)\z/).flatten @rye_getenv[n] = v end end key.nil? ? @rye_getenv : @rye_getenv[key.to_s] end
Uses the output of "useradd -D" to determine the default home directory. This returns a GUESS rather than the a user's real home directory. Currently used only by authorize_keys_remote. Only useful before you've logged in. Otherwise check $HOME
# File lib/rye/box.rb, line 421 def guess_user_home(other_user=nil) this_user = other_user || opts[:user] @rye_guessed_homes ||= {} # A simple cache. return @rye_guessed_homes[this_user] if @rye_guessed_homes.has_key?(this_user) # Some junk to determine where user home directories are by default. # We're relying on the command "useradd -D" so this may not work on # different Linuxen and definitely won't work on Windows. # This code will be abstracted out once I find a decent home for it. # /etc/default/useradd, HOME=/home OR useradd -D # /etc/adduser.config, DHOME=/home OR ?? user_defaults = {} ostmp = self.ostype ostmp &&= ostype.to_s if ostmp == "sunos" #nv.scan(/([\w_-]+?)=(.+?)\s/).each do |n, v| # n = 'HOME' if n == 'basedir' # user_defaults[n.upcase] = v.strip #end # In Solaris, useradd -D says the default home path is /home # but that directory is not writable. See: http://bit.ly/IJDD0 user_defaults['HOME'] = '/export/home' elsif ostmp == "darwin" user_defaults['HOME'] = '/Users' elsif ostmp == "windows" user_defaults['HOME'] = 'C:/Documents and Settings' else raw = self.quietly { useradd(:D) } rescue [] raw = ["HOME=/home"] if raw.nil? || raw.empty? raw.each do |nv| n, v = nv.scan(/\A([\w_-]+?)=(.+)\z/).flatten user_defaults[n] = v end end @rye_guessed_homes[this_user] = "#{user_defaults['HOME']}/#{this_user}" end
Returns the host SSH keys for this box
# File lib/rye/box.rb, line 412 def host_key raise "No host" unless @rye_host Rye.remote_host_keys(@rye_host) end
# File lib/rye/box.rb, line 85 def impltype=(val); @rye_impltype = val; end
# File lib/rye/box.rb, line 397 def inspect %{#<%s:%s name=%s cwd=%s umask=%s env=%s safe=%s opts=%s keys=%s>} % [self.class.to_s, self.host, self.nickname, @rye_current_working_directory, @rye_current_umask, (@rye_current_environment_variables || '').inspect, self.safe, self.opts.inspect, self.keys.inspect] end
# File lib/rye/box.rb, line 615 def instance_exec(*args, &block) # !> method redefined; discarding old instance_exec mname = "__instance_exec_#{Thread.current.object_id.abs}_#{object_id.abs}" InstanceExecHelper.module_eval{ define_method(mname, &block) } begin ret = send(mname, *args) ensure InstanceExecHelper.module_eval{ undef_method(mname) } rescue nil end ret end
If STDIN.tty? is true (i.e. if we're connected to a terminal with a human at the helm), this will open an SSH connection via the regular SSH command (via a call to system). This requires the SSH command-line executable (ssh).
If STDIN.tty? is false or run is false, this will return the SSH command (a String) that would have been run.
NOTE: As of Rye 0.9 you can run interactive sessions with rye by calling any shell method without arguments.
e.g.
rbox = Rye::Box.new 'somemachine' rbox.bash
TODO: refactor to use net_ssh_exec! in 0.9
# File lib/rye/box.rb, line 308 def interactive_ssh(run=true) debug "interactive_ssh with keys: #{@rye_opts[:keys].inspect}" run = false unless STDIN.tty? args = [] @rye_opts[:keys].each { |key| args.push *[:i, key] } args << "#{@rye_user}@#{@rye_host}" cmd = Rye.prepare_command("ssh", args) return cmd unless run system(cmd) end
A handler for undefined commands. Raises Rye::CommandNotFound exception.
# File lib/rye/box.rb, line 464 def method_missing(cmd, *args, &block) if cmd == :to_ary super elsif @rye_safe ex = Rye::CommandNotFound.new(cmd.to_s) raise ex unless @rye_exception_hook.has_key? ex.class @rye_exception_hook[Rye::CommandNotFound].call ex else if block.nil? run_command cmd, *args else ex = Rye::CommandNotFound.new(cmd.to_s) raise ex unless @rye_exception_hook.has_key? ex.class end end end
# File lib/rye/box.rb, line 64 def nickname=(val); @rye_nickname = val; end
Return the value of uname in lowercase This is a temporary fix. We can use SysInfo for this, upload it, execute it directly, parse the output.
# File lib/rye/box.rb, line 344 def ostype return @rye_ostype if @rye_ostype # simple cache os = self.quietly { uname.first } rescue nil os ||= 'unknown' os &&= os.downcase @rye_ostype = os end
Supply a block to be called after every command. It's called with one argument: an instance of Rye::Rap.
When this block is supplied, the command does not raise an exception when the exit code is greater than 0 (the typical behavior) so the block needs to check the Rye::Rap object to determine whether an exception should be raised.
# File lib/rye/box.rb, line 634 def post_command_hook(&block) @rye_post_command_hook = block if block @rye_post_command_hook end
# File lib/rye/box.rb, line 88 def post_command_hook=(val); @rye_post_command_hook = val; end
Supply a block to be called before every command. It's called with three arguments: command name, an Array of arguments, user name, hostname e.g.
rbox.pre_command_hook do |cmd,args,user,host| ... end
# File lib/rye/box.rb, line 494 def pre_command_hook(&block) @rye_pre_command_hook = block if block @rye_pre_command_hook end
# File lib/rye/box.rb, line 86 def pre_command_hook=(val); @rye_pre_command_hook = val; end
Returns the command an arguments as a String.
# File lib/rye/box.rb, line 483 def preview_command(*args) prep_args(*args).join(' ') end
Like batch, except it enables quiet mode before executing the block. After executing the block, quiet mode is returned back to whichever state it was previously in. In other words, this method won't enable quiet mode if it was already disabled.
In quiet mode, the pre and post command hooks are not called. This is used internally when calling commands like ls to check whether a file path exists (to prevent polluting the logs).
# File lib/rye/box.rb, line 584 def quietly(*args, &block) previous_state = @rye_quiet enable_quiet_mode ret = self.instance_exec *args, &block @rye_quiet = previous_state ret end
Remove one or more private keys fromt he list of key paths.
-
keys is a list of file paths to private keys
Returns the instance of Box
# File lib/rye/box.rb, line 333 def remove_keys(*keys) @rye_opts[:keys] ||= [] @rye_opts[:keys] -= keys.flatten.compact @rye_opts[:keys].uniq! self # MUST RETURN self end
See unsafely (except in reverse)
# File lib/rye/box.rb, line 568 def safely(*args, &block) previous_state = @rye_safe enable_safe_mode ret = self.instance_exec *args, &block @rye_safe = previous_state ret end
Add an environment variable. n and v are the name and value. Returns the instance of Rye::Box
# File lib/rye/box.rb, line 382 def setenv(n, v) debug "Adding env: #{n}=#{v}" debug "prev value: #{@rye_getenv[n]}" @rye_getenv[n] = v (@rye_current_environment_variables ||= {})[n] = v self end
Parse SSH config files for use with Net::SSH
# File lib/rye/box.rb, line 190 def ssh_config_options(host) return Net::SSH::Config.for(host) end
Returns the current value of the stash +@rye_stash+
# File lib/rye/box.rb, line 53 def stash; @rye_stash; end
Store a value to the stash +@rye_stash+
# File lib/rye/box.rb, line 63 def stash=(val); @rye_stash = val; end
Supply a block to be called every time a command receives STDOUT data.
e.g.
rbox.stdout_hook do |content| ... end
# File lib/rye/box.rb, line 505 def stdout_hook(&block) @rye_stdout_hook = block if block @rye_stdout_hook end
# File lib/rye/box.rb, line 87 def stdout_hook=(val); @rye_stdout_hook = val; end
Like batch, except it enables sudo mode before executing the block. If the user is already root, this has no effect. Otherwise all commands executed in the block will run via sudo.
If no block is specified then sudo is called just like a regular command.
# File lib/rye/box.rb, line 598 def sudo(*args, &block) if block.nil? run_command('sudo', args); else previous_state = @rye_sudo enable_sudo ret = self.instance_exec *args, &block @rye_sudo = previous_state ret end end
Reconnect as another user. This is different from su= which executes subsequent commands via +su -c COMMAND USER+.
-
newuser The username to reconnect as
NOTE: if there is an open connection, it's disconnected but not reconnected because it's possible it wasn't connected yet in the first place (if you create the instance with default settings for example)
# File lib/rye/box.rb, line 283 def switch_user(newuser) return if newuser.to_s == self.user.to_s @rye_opts ||= {} @rye_user = newuser disconnect end
# File lib/rye/box.rb, line 46 def templates?; !@rye_templates.nil?; end
Returns +user@rye_host+
# File lib/rye/box.rb, line 395 def to_s; '%s@rye_%s' % [user, @rye_host]; end
Change the current umask (sort of -- works the same way as cd) The default umask is 0022
# File lib/rye/box.rb, line 269 def umask=(val='0022') @rye_current_umask = val self end
Like batch, except it disables safe mode before executing the block. After executing the block, safe mode is returned back to whichever state it was previously in. In other words, this method won't enable safe mode if it was already disabled.
# File lib/rye/box.rb, line 558 def unsafely(*args, &block) previous_state = @rye_safe disable_safe_mode ret = self.instance_exec *args, &block @rye_safe = previous_state ret end
-
hops Rye::Hop objects will be added directly
to the set. Hostnames will be used to create new instances of Rye::Hop h1 = Rye::Hop.new "host1" h1.via_hop "host2", :user => "service_user"
OR
h1 = Rye::Hop.new "host1" h2 = Rye::Hop.new "host2" h1.via_hop h2
# File lib/rye/box.rb, line 205 def via_hop(*args) args = args.flatten.compact if args.first.nil? return @rye_via elsif args.first.is_a?(Rye::Hop) @rye_via = args.first elsif args.first.is_a?(String) hop = args.shift if args.first.is_a?(Hash) @rye_via = Rye::Hop.new(hop, args.first.merge( :debug => @rye_debug, :info => @rye_info, :error => @rye_error) ) else @rye_via = Rye::Hop.new(hop) end end disconnect self end