diff options
Diffstat (limited to 'lib/pry/output.rb')
| -rw-r--r-- | lib/pry/output.rb | 22 |
1 files changed, 10 insertions, 12 deletions
diff --git a/lib/pry/output.rb b/lib/pry/output.rb index 6ecd8752..91373843 100644 --- a/lib/pry/output.rb +++ b/lib/pry/output.rb @@ -5,8 +5,8 @@ class Pry attr_reader :pry_instance def initialize(pry_instance) - @pry_instance = pry_instance - @boxed_io = pry_instance.config.output + @output = pry_instance.config.output + @color = pry_instance.config.color end def puts(*objs) @@ -24,7 +24,7 @@ class Pry def print(*objs) objs.each do |obj| - @boxed_io.print decolorize_maybe(obj.to_s) + @output.print decolorize_maybe(obj.to_s) end nil end @@ -32,27 +32,25 @@ class Pry alias write print def tty? - @boxed_io.respond_to?(:tty?) && @boxed_io.tty? + @output.respond_to?(:tty?) && @output.tty? end def method_missing(method_name, *args, &block) - if @boxed_io.respond_to?(method_name) - @boxed_io.__send__(method_name, *args, &block) + if @output.respond_to?(method_name) + @output.__send__(method_name, *args, &block) else super end end def respond_to_missing?(method_name, include_private = false) - @boxed_io.respond_to?(method_name, include_private) + @output.respond_to?(method_name, include_private) end def decolorize_maybe(str) - if pry_instance.config.color - str - else - Pry::Helpers::Text.strip_color str - end + return str if @color + + Pry::Helpers::Text.strip_color(str) end end end |