File: prompter.rb

package info (click to toggle)
libnet-ssh-ruby 1.1.2-1
  • links: PTS
  • area: main
  • in suites: lenny
  • size: 3,472 kB
  • ctags: 2,465
  • sloc: ruby: 10,848; makefile: 17
file content (73 lines) | stat: -rw-r--r-- 2,059 bytes parent folder | download | duplicates (3)
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
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
#--
# =============================================================================
# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
# All rights reserved.
#
# This source file is distributed as part of the Net::SSH Secure Shell Client
# library for Ruby. This file (and the library as a whole) may be used only as
# allowed by either the BSD license, or the Ruby license (or, by association
# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
# distribution for the texts of these licenses.
# -----------------------------------------------------------------------------
# net-ssh website : http://net-ssh.rubyforge.org
# project website: http://rubyforge.org/projects/net-ssh
# =============================================================================
#++

unless defined?( HAS_TERMIOS )
  begin
    require 'termios'
    HAS_TERMIOS = true
  rescue LoadError
    HAS_TERMIOS = false
  end
end

module Net
  module SSH
    module Util

      # A simple class for prompting a user for input.
      class Prompter

        # Displays the prompt and then waits for the user to enter a response.
        # If the 'termios' library is available, the user's input will not be
        # echoed to the terminal. The user's response will be returned, with
        # any trailing newline chomped.
        def password( prompt=nil )
          if $stdin.tty?
            echo false
            print prompt if prompt
          end

          $stdin.gets.chomp
        ensure
          if $stdin.tty?
            echo true
            puts
          end
        end

        if HAS_TERMIOS
          def echo( enable )
            term = Termios::getattr( $stdin )

            if enable
              term.c_lflag |= ( Termios::ECHO | Termios::ICANON )
            else
              term.c_lflag &= ~Termios::ECHO
            end

            Termios::setattr( $stdin, Termios::TCSANOW, term )
          end
        else
          def echo( enable )
          end
        end
        private :echo

      end

    end
  end
end