diff --git a/lib/irb.rb b/lib/irb.rb index 16c03e964f..98bcac016d 100644 --- a/lib/irb.rb +++ b/lib/irb.rb @@ -439,6 +439,8 @@ module IRB # Evaluates input for this session. def eval_input + last_error = nil + @scanner.set_prompt do |ltype, indent, continue, line_no| if ltype @@ -488,7 +490,7 @@ module IRB signal_status(:IN_EVAL) do begin line.untaint - @context.evaluate(line, line_no) + @context.evaluate(line, line_no, exception: last_error) output_value if @context.echo? exc = nil rescue Interrupt => exc @@ -497,6 +499,7 @@ module IRB rescue Exception => exc end if exc + last_error = exc handle_exception(exc) end end diff --git a/lib/irb/context.rb b/lib/irb/context.rb index 1a06ebfad1..b82aaea6a6 100644 --- a/lib/irb/context.rb +++ b/lib/irb/context.rb @@ -376,8 +376,12 @@ module IRB @debug_level > 0 end - def evaluate(line, line_no) # :nodoc: + def evaluate(line, line_no, exception: nil) # :nodoc: @line_no = line_no + if exception + line = "begin ::Kernel.raise _; rescue _.class; #{line}; end" + @workspace.local_variable_set(:_, exception) + end set_last_value(@workspace.evaluate(self, line, irb_path, line_no)) end diff --git a/test/irb/test_context.rb b/test/irb/test_context.rb index 35141a41a4..1faf2b373d 100644 --- a/test/irb/test_context.rb +++ b/test/irb/test_context.rb @@ -40,5 +40,14 @@ module TestIRB assert_same(obj, @context.last_value) assert_same(obj, @context.evaluate('_', 1)) end + + def test_evaluate_with_exception + assert_nil(@context.evaluate("$!", 1)) + e = assert_raise_with_message(RuntimeError, 'foo') { + @context.evaluate("raise 'foo'", 1) + } + assert_equal('foo', e.message) + assert_same(e, @context.evaluate('$!', 1, exception: e)) + end end end