Provide pry's command prompt after eval

BetterErrors · Aug 5, 2017 · 404c31f · 404c31f
1 parent b02636b
commit 404c31f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/better_errors/pry/repl/pry.rb b/lib/better_errors/pry/repl/pry.rb
@@ -61,7 +61,9 @@ def send_input(str)
       end
 
       def prompt
-        if indent = @pry.instance_variable_get(:@indent) and !indent.indent_level.empty?
+        if @pry.respond_to? :select_prompt
+          [@pry.select_prompt, ""]
+        elsif indent = @pry.instance_variable_get(:@indent) and !indent.indent_level.empty?
           ["..", indent.indent_level]
         else
           [">>", ""]