rbx-trepanning 0.0.4-universal-rubinius-1.2 → 0.0.6-universal-rubinius-1.2

data/io/input.rb

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-# Copyright (C) 2010 Rocky Bernstein <rockyb@rubyforge.net>
+# Copyright (C) 2010, 2011 Rocky Bernstein <rockyb@rubyforge.net>
 
 # Debugger user/command-oriented input possibly attached to IO-style
 # input or GNU Readline.
@@ -14,10 +14,14 @@ class Trepan
   # input or GNU Readline.
   class UserInput < Trepan::InputBase
 
+    @@readline_finalized = false
+
     def initialize(inp, opts={})
-      @opts      = DEFAULT_OPTS.merge(opts)
-      @input     = inp || STDIN
-      @eof       = false
+      @opts         = DEFAULT_OPTS.merge(opts)
+      @input        = inp || STDIN
+      @eof          = false
+      @line_edit    = @opts[:line_edit]
+      @use_readline = opts[:readline]
     end
 
     def closed?; @input.closed? end
@@ -28,18 +32,21 @@ class Trepan
     end
 
     # Read a line of input. EOFError will be raised on EOF.  
-    # 
-    # Note that we don't support prompting first. Instead, arrange
-    # to call Trepan::Output.write() first with the prompt. 
-    def readline
-      # FIXME we don't do command completion.
+    def readline(prompt='')
       raise EOFError if eof?
       begin 
-        line = @opts[:line_edit] ? Readline.readline : @input.gets
-        @eof = !line
-      rescue
+        if @line_edit && @use_readline
+          line = Readline.readline(prompt, true)
+        else
+          line = @input.gets
+          end
+      rescue EOFError
+      rescue => e
+        puts $!.backtrace
+        puts "Exception caught #{e.inspect}"
         @eof = true
       end
+      @eof = !line
       raise EOFError if eof?
       return line
     end
@@ -53,17 +60,33 @@ class Trepan
       def open(inp=nil, opts={})
         inp ||= STDIN
         inp = File.new(inp, 'r') if inp.is_a?(String)
-        opts[:line_edit] = false unless
+        opts[:line_edit] = @line_edit = 
           inp.respond_to?(:isatty) && inp.isatty && Trepan::GNU_readline?
         self.new(inp, opts)
       end
+
+      def finalize
+        if defined?(RbReadline) && !@@readline_finalized
+          RbReadline.rl_cleanup_after_signal()
+          RbReadline.rl_deprep_terminal()
+          @@readline_finalized = true
+        end
+      end
     end
   end
 end
 
 def Trepan::GNU_readline?
   begin
-    require 'readline'
+    return @use_readline unless @use_readline.nil?
+    @use_readline ||= require 'rb-readline.rb'
+    if @use_readline
+      # Returns current line buffer
+      def Readline.line_buffer
+        RbReadline.rl_line_buffer
+      end
+      at_exit { Trepan::UserInput::finalize }
+    end
     return true
   rescue LoadError
     return false