[ruby/irb] Move IO configuration to IRB::Irb

(https://github.com/ruby/irb/pull/681)

It shouldn't be `RubyLex`'s responsibility to handle IO. So this moves
the configuration to `IRB::Irb`.

https://github.com/ruby/irb/commit/daff750076
This commit is contained in:
Stan Lo 2023-08-13 16:23:00 +01:00 committed by git
parent 3af5badae8
commit 9099d62ac7
4 changed files with 766 additions and 731 deletions

View File

@ -537,7 +537,7 @@ module IRB
end
end
@scanner.configure_io(@context.io)
configure_io
@scanner.each_top_level_statement do |line, line_no, is_assignment|
signal_status(:IN_EVAL) do
@ -564,6 +564,58 @@ module IRB
end
end
def configure_io
if @context.io.respond_to?(:check_termination)
@context.io.check_termination do |code|
if Reline::IOGate.in_pasting?
rest = @scanner.check_termination_in_prev_line(code)
if rest
Reline.delete_text
rest.bytes.reverse_each do |c|
Reline.ungetc(c)
end
true
else
false
end
else
# Accept any single-line input for symbol aliases or commands that transform args
next true if @scanner.single_line_command?(code)
_tokens, _opens, terminated = @scanner.check_code_state(code)
terminated
end
end
end
if @context.io.respond_to?(:dynamic_prompt)
@context.io.dynamic_prompt do |lines|
lines << '' if lines.empty?
tokens = RubyLex.ripper_lex_without_warning(lines.map{ |l| l + "\n" }.join, context: @context)
line_results = IRB::NestingParser.parse_by_line(tokens)
tokens_until_line = []
line_results.map.with_index do |(line_tokens, _prev_opens, next_opens, _min_depth), line_num_offset|
line_tokens.each do |token, _s|
# Avoid appending duplicated token. Tokens that include "\n" like multiline tstring_content can exist in multiple lines.
tokens_until_line << token if token != tokens_until_line.last
end
continue = @scanner.should_continue?(tokens_until_line)
@scanner.prompt(next_opens, continue, line_num_offset)
end
end
end
if @context.io.respond_to?(:auto_indent) and @context.auto_indent_mode
@context.io.auto_indent do |lines, line_index, byte_pointer, is_newline|
next nil if lines == [nil] # Workaround for exit IRB with CTRL+d
next nil if !is_newline && lines[line_index]&.byteslice(0, byte_pointer)&.match?(/\A\s*\z/)
code = lines[0..line_index].map { |l| "#{l}\n" }.join
tokens = RubyLex.ripper_lex_without_warning(code, context: @context)
@scanner.process_indent_level(tokens, lines, line_index, is_newline)
end
end
end
def evaluate_line(line, line_no)
# Transform a non-identifier alias (@, $) or keywords (next, break)
command, args = line.split(/\s/, 2)

View File

@ -74,59 +74,6 @@ class RubyLex
@input = block
end
def configure_io(io)
@io = io
if @io.respond_to?(:check_termination)
@io.check_termination do |code|
if Reline::IOGate.in_pasting?
rest = check_termination_in_prev_line(code)
if rest
Reline.delete_text
rest.bytes.reverse_each do |c|
Reline.ungetc(c)
end
true
else
false
end
else
# Accept any single-line input for symbol aliases or commands that transform args
next true if single_line_command?(code)
_tokens, _opens, terminated = check_code_state(code)
terminated
end
end
end
if @io.respond_to?(:dynamic_prompt)
@io.dynamic_prompt do |lines|
lines << '' if lines.empty?
tokens = self.class.ripper_lex_without_warning(lines.map{ |l| l + "\n" }.join, context: @context)
line_results = IRB::NestingParser.parse_by_line(tokens)
tokens_until_line = []
line_results.map.with_index do |(line_tokens, _prev_opens, next_opens, _min_depth), line_num_offset|
line_tokens.each do |token, _s|
# Avoid appending duplicated token. Tokens that include "\n" like multiline tstring_content can exist in multiple lines.
tokens_until_line << token if token != tokens_until_line.last
end
continue = should_continue?(tokens_until_line)
prompt(next_opens, continue, line_num_offset)
end
end
end
if @io.respond_to?(:auto_indent) and @context.auto_indent_mode
@io.auto_indent do |lines, line_index, byte_pointer, is_newline|
next nil if lines == [nil] # Workaround for exit IRB with CTRL+d
next nil if !is_newline && lines[line_index]&.byteslice(0, byte_pointer)&.match?(/\A\s*\z/)
code = lines[0..line_index].map { |l| "#{l}\n" }.join
tokens = self.class.ripper_lex_without_warning(code, context: @context)
process_indent_level(tokens, lines, line_index, is_newline)
end
end
end
def set_prompt(&block)
@prompt = block
end
@ -240,7 +187,7 @@ class RubyLex
save_prompt_to_context_io([], false, 0)
# multiline
return @input.call if @io.respond_to?(:check_termination)
return @input.call if @context.io.respond_to?(:check_termination)
# nomultiline
code = ''
@ -270,7 +217,7 @@ class RubyLex
break unless code
if code != "\n"
code.force_encoding(@io.encoding)
code.force_encoding(@context.io.encoding)
yield code, @line_no, assignment_expression?(code)
end
@line_no += code.count("\n")

681
test/irb/test_irb.rb Normal file
View File

@ -0,0 +1,681 @@
# frozen_string_literal: true
require "irb"
require_relative "helper"
module TestIRB
class IrbIOConfigurationTest < TestCase
Row = Struct.new(:content, :current_line_spaces, :new_line_spaces, :indent_level)
class MockIO_AutoIndent
attr_reader :calculated_indent
def initialize(*params)
@params = params
@calculated_indent
end
def auto_indent(&block)
@calculated_indent = block.call(*@params)
end
end
class MockIO_DynamicPrompt
def initialize(params, &assertion)
@params = params
@assertion = assertion
end
def dynamic_prompt(&block)
result = block.call(@params)
@assertion.call(result)
end
end
def setup
@irb = build_irb
save_encodings
end
def teardown
restore_encodings
end
class AutoIndentationTest < IrbIOConfigurationTest
def test_auto_indent
input_with_correct_indents = [
[%q(def each_top_level_statement), 0, 2],
[%q( initialize_input), 2, 2],
[%q( catch(:TERM_INPUT) do), 2, 4],
[%q( loop do), 4, 6],
[%q( begin), 6, 8],
[%q( prompt), 8, 8],
[%q( unless l = lex), 8, 10],
[%q( throw :TERM_INPUT if @line == ''), 10, 10],
[%q( else), 8, 10],
[%q( @line_no += l.count("\n")), 10, 10],
[%q( next if l == "\n"), 10, 10],
[%q( @line.concat l), 10, 10],
[%q( if @code_block_open or @ltype or @continue or @indent > 0), 10, 12],
[%q( next), 12, 12],
[%q( end), 10, 10],
[%q( end), 8, 8],
[%q( if @line != "\n"), 8, 10],
[%q( @line.force_encoding(@io.encoding)), 10, 10],
[%q( yield @line, @exp_line_no), 10, 10],
[%q( end), 8, 8],
[%q( break if @io.eof?), 8, 8],
[%q( @line = ''), 8, 8],
[%q( @exp_line_no = @line_no), 8, 8],
[%q( ), nil, 8],
[%q( @indent = 0), 8, 8],
[%q( rescue TerminateLineInput), 6, 8],
[%q( initialize_input), 8, 8],
[%q( prompt), 8, 8],
[%q( end), 6, 6],
[%q( end), 4, 4],
[%q( end), 2, 2],
[%q(end), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_braces_on_their_own_line
input_with_correct_indents = [
[%q(if true), 0, 2],
[%q( [), 2, 4],
[%q( ]), 2, 2],
[%q(end), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_multiple_braces_in_a_line
input_with_correct_indents = [
[%q([[[), 0, 6],
[%q( ]), 4, 4],
[%q( ]), 2, 2],
[%q(]), 0, 0],
[%q([<<FOO]), 0, 0],
[%q(hello), 0, 0],
[%q(FOO), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_a_closed_brace_and_not_closed_brace_in_a_line
input_with_correct_indents = [
[%q(p() {), 0, 2],
[%q(}), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_symbols
input_with_correct_indents = [
[%q(:a), 0, 0],
[%q(:A), 0, 0],
[%q(:+), 0, 0],
[%q(:@@a), 0, 0],
[%q(:@a), 0, 0],
[%q(:$a), 0, 0],
[%q(:def), 0, 0],
[%q(:`), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_incomplete_coding_magic_comment
input_with_correct_indents = [
[%q(#coding:u), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_incomplete_encoding_magic_comment
input_with_correct_indents = [
[%q(#encoding:u), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_incomplete_emacs_coding_magic_comment
input_with_correct_indents = [
[%q(# -*- coding: u), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_incomplete_vim_coding_magic_comment
input_with_correct_indents = [
[%q(# vim:set fileencoding=u), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_mixed_rescue
input_with_correct_indents = [
[%q(def m), 0, 2],
[%q( begin), 2, 4],
[%q( begin), 4, 6],
[%q( x = a rescue 4), 6, 6],
[%q( y = [(a rescue 5)]), 6, 6],
[%q( [x, y]), 6, 6],
[%q( rescue => e), 4, 6],
[%q( raise e rescue 8), 6, 6],
[%q( end), 4, 4],
[%q( rescue), 2, 4],
[%q( raise rescue 11), 4, 4],
[%q( end), 2, 2],
[%q(rescue => e), 0, 2],
[%q( raise e rescue 14), 2, 2],
[%q(end), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_oneliner_method_definition
input_with_correct_indents = [
[%q(class A), 0, 2],
[%q( def foo0), 2, 4],
[%q( 3), 4, 4],
[%q( end), 2, 2],
[%q( def foo1()), 2, 4],
[%q( 3), 4, 4],
[%q( end), 2, 2],
[%q( def foo2(a, b)), 2, 4],
[%q( a + b), 4, 4],
[%q( end), 2, 2],
[%q( def foo3 a, b), 2, 4],
[%q( a + b), 4, 4],
[%q( end), 2, 2],
[%q( def bar0() = 3), 2, 2],
[%q( def bar1(a) = a), 2, 2],
[%q( def bar2(a, b) = a + b), 2, 2],
[%q( def bar3() = :s), 2, 2],
[%q( def bar4() = Time.now), 2, 2],
[%q(end), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_tlambda
input_with_correct_indents = [
[%q(if true), 0, 2, 1],
[%q( -> {), 2, 4, 2],
[%q( }), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_corresponding_syntax_to_keyword_do_in_class
input_with_correct_indents = [
[%q(class C), 0, 2, 1],
[%q( while method_name do), 2, 4, 2],
[%q( 3), 4, 4, 2],
[%q( end), 2, 2, 1],
[%q( foo do), 2, 4, 2],
[%q( 3), 4, 4, 2],
[%q( end), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_corresponding_syntax_to_keyword_do
input_with_correct_indents = [
[%q(while i > 0), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while true), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while ->{i > 0}.call), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while ->{true}.call), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while i > 0 do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while true do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while ->{i > 0}.call do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while ->{true}.call do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(foo do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(foo true do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(foo ->{true} do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(foo ->{i > 0} do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_corresponding_syntax_to_keyword_for
input_with_correct_indents = [
[%q(for i in [1]), 0, 2, 1],
[%q( puts i), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_corresponding_syntax_to_keyword_for_with_do
input_with_correct_indents = [
[%q(for i in [1] do), 0, 2, 1],
[%q( puts i), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_typing_incomplete_include_interpreted_as_keyword_in
input_with_correct_indents = [
[%q(module E), 0, 2, 1],
[%q(end), 0, 0, 0],
[%q(class A), 0, 2, 1],
[%q( in), 2, 2, 1] # scenario typing `include E`
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_bracket_corresponding_to_times
input_with_correct_indents = [
[%q(3.times { |i|), 0, 2, 1],
[%q( puts i), 2, 2, 1],
[%q(}), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_do_corresponding_to_times
input_with_correct_indents = [
[%q(3.times do |i|), 0, 2, 1],
[%q( puts i), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_bracket_corresponding_to_loop
input_with_correct_indents = [
['loop {', 0, 2, 1],
[' 3', 2, 2, 1],
['}', 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_do_corresponding_to_loop
input_with_correct_indents = [
[%q(loop do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_embdoc_indent
input_with_correct_indents = [
[%q(=begin), 0, 0, 0],
[%q(a), 0, 0, 0],
[%q( b), 1, 1, 0],
[%q(=end), 0, 0, 0],
[%q(if 1), 0, 2, 1],
[%q( 2), 2, 2, 1],
[%q(=begin), 0, 0, 0],
[%q(a), 0, 0, 0],
[%q( b), 1, 1, 0],
[%q(=end), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_heredoc_with_indent
if Gem::Version.new(RUBY_VERSION) < Gem::Version.new('2.7.0')
pend 'This test needs Ripper::Lexer#scan to take broken tokens'
end
input_with_correct_indents = [
[%q(<<~Q+<<~R), 0, 2, 1],
[%q(a), 2, 2, 1],
[%q(a), 2, 2, 1],
[%q( b), 2, 2, 1],
[%q( b), 2, 2, 1],
[%q( Q), 0, 2, 1],
[%q( c), 4, 4, 1],
[%q( c), 4, 4, 1],
[%q( R), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_oneliner_def_in_multiple_lines
input_with_correct_indents = [
[%q(def a()=[), 0, 2, 1],
[%q( 1,), 2, 2, 1],
[%q(].), 0, 0, 0],
[%q(to_s), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_broken_heredoc
input_with_correct_indents = [
[%q(def foo), 0, 2, 1],
[%q( <<~Q), 2, 4, 2],
[%q( Qend), 4, 4, 2],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_pasted_code_keep_base_indent_spaces
input_with_correct_indents = [
[%q( def foo), 0, 6, 1],
[%q( if bar), 6, 10, 2],
[%q( [1), 10, 12, 3],
[%q( ]+[["a), 10, 14, 4],
[%q(b" + `c), 0, 14, 4],
[%q(d` + /e), 0, 14, 4],
[%q(f/ + :"g), 0, 14, 4],
[%q(h".tap do), 0, 16, 5],
[%q( 1), 16, 16, 5],
[%q( end), 14, 14, 4],
[%q( ]), 12, 12, 3],
[%q( ]), 10, 10, 2],
[%q( end), 8, 6, 1],
[%q( end), 4, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_pasted_code_keep_base_indent_spaces_with_heredoc
input_with_correct_indents = [
[%q( def foo), 0, 6, 1],
[%q( if bar), 6, 10, 2],
[%q( [1), 10, 12, 3],
[%q( ]+[["a), 10, 14, 4],
[%q(b" + <<~A + <<-B + <<C), 0, 16, 5],
[%q( a#{), 16, 18, 6],
[%q( 1), 18, 18, 6],
[%q( }), 16, 16, 5],
[%q( A), 14, 16, 5],
[%q( b#{), 16, 18, 6],
[%q( 1), 18, 18, 6],
[%q( }), 16, 16, 5],
[%q( B), 14, 0, 0],
[%q(c#{), 0, 2, 1],
[%q(1), 2, 2, 1],
[%q(}), 0, 0, 0],
[%q(C), 0, 14, 4],
[%q( ]), 12, 12, 3],
[%q( ]), 10, 10, 2],
[%q( end), 8, 6, 1],
[%q( end), 4, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_heredoc_keep_indent_spaces
(1..4).each do |indent|
row = Row.new(' ' * indent, nil, [4, indent].max, 2)
lines = ['def foo', ' <<~Q', row.content]
assert_row_indenting(lines, row)
assert_indent_level(lines, row.indent_level)
end
end
private
def assert_row_indenting(lines, row)
actual_current_line_spaces = calculate_indenting(lines, false)
error_message = <<~MSG
Incorrect spaces calculation for line:
```
> #{lines.last}
```
All lines:
```
#{lines.join("\n")}
```
MSG
assert_equal(row.current_line_spaces, actual_current_line_spaces, error_message)
error_message = <<~MSG
Incorrect spaces calculation for line after the current line:
```
#{lines.last}
>
```
All lines:
```
#{lines.join("\n")}
```
MSG
actual_next_line_spaces = calculate_indenting(lines, true)
assert_equal(row.new_line_spaces, actual_next_line_spaces, error_message)
end
def assert_rows_with_correct_indents(rows_with_spaces, assert_indent_level: false)
lines = []
rows_with_spaces.map do |row|
row = Row.new(*row)
lines << row.content
assert_row_indenting(lines, row)
if assert_indent_level
assert_indent_level(lines, row.indent_level)
end
end
end
def assert_indent_level(lines, expected)
code = lines.map { |l| "#{l}\n" }.join # code should end with "\n"
_tokens, opens, _ = @irb.scanner.check_code_state(code)
indent_level = @irb.scanner.calc_indent_level(opens)
error_message = "Calculated the wrong number of indent level for:\n #{lines.join("\n")}"
assert_equal(expected, indent_level, error_message)
end
def calculate_indenting(lines, add_new_line)
lines = lines + [""] if add_new_line
last_line_index = lines.length - 1
byte_pointer = lines.last.length
mock_io = MockIO_AutoIndent.new(lines, last_line_index, byte_pointer, add_new_line)
@irb.context.auto_indent_mode = true
@irb.context.io = mock_io
@irb.configure_io
mock_io.calculated_indent
end
end
class DynamicPromptTest < IrbIOConfigurationTest
def test_endless_range_at_end_of_line
input_with_prompt = [
['001:0: :> ', %q(a = 3..)],
['002:0: :> ', %q()],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_heredoc_with_embexpr
input_with_prompt = [
['001:0:":* ', %q(<<A+%W[#{<<B)],
['002:0:":* ', %q(#{<<C+%W[)],
['003:0:":* ', %q(a)],
['004:2:]:* ', %q(C)],
['005:2:]:* ', %q(a)],
['006:0:":* ', %q(]})],
['007:0:":* ', %q(})],
['008:0:":* ', %q(A)],
['009:2:]:* ', %q(B)],
['010:1:]:* ', %q(})],
['011:0: :> ', %q(])],
['012:0: :> ', %q()],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_heredoc_prompt_with_quotes
input_with_prompt = [
["001:1:':* ", %q(<<~'A')],
["002:1:':* ", %q(#{foobar})],
["003:0: :> ", %q(A)],
["004:1:`:* ", %q(<<~`A`)],
["005:1:`:* ", %q(whoami)],
["006:0: :> ", %q(A)],
['007:1:":* ', %q(<<~"A")],
['008:1:":* ', %q(foobar)],
['009:0: :> ', %q(A)],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_backtick_method
input_with_prompt = [
['001:0: :> ', %q(self.`(arg))],
['002:0: :> ', %q()],
['003:0: :> ', %q(def `(); end)],
['004:0: :> ', %q()],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_dynamic_prompt
input_with_prompt = [
['001:1: :* ', %q(def hoge)],
['002:1: :* ', %q( 3)],
['003:0: :> ', %q(end)],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_dynamic_prompt_with_double_newline_breaking_code
input_with_prompt = [
['001:1: :* ', %q(if true)],
['002:2: :* ', %q(%)],
['003:1: :* ', %q(;end)],
['004:1: :* ', %q(;hello)],
['005:0: :> ', %q(end)],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_dynamic_prompt_with_multiline_literal
input_with_prompt = [
['001:1: :* ', %q(if true)],
['002:2:]:* ', %q( %w[)],
['003:2:]:* ', %q( a)],
['004:1: :* ', %q( ])],
['005:1: :* ', %q( b)],
['006:2:]:* ', %q( %w[)],
['007:2:]:* ', %q( c)],
['008:1: :* ', %q( ])],
['009:0: :> ', %q(end)],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_dynamic_prompt_with_blank_line
input_with_prompt = [
['001:1:]:* ', %q(%w[)],
['002:1:]:* ', %q()],
['003:0: :> ', %q(])],
]
assert_dynamic_prompt(input_with_prompt)
end
def assert_dynamic_prompt(input_with_prompt)
expected_prompt_list, lines = input_with_prompt.transpose
dynamic_prompt_executed = false
io = MockIO_DynamicPrompt.new(lines) do |prompt_list|
error_message = <<~EOM
Expected dynamic prompt:
#{expected_prompt_list.join("\n")}
Actual dynamic prompt:
#{prompt_list.join("\n")}
EOM
dynamic_prompt_executed = true
assert_equal(expected_prompt_list, prompt_list, error_message)
end
@irb.context.io = io
@irb.scanner.set_prompt do |ltype, indent, continue, line_no|
'%03d:%01d:%1s:%s ' % [line_no, indent, ltype, continue ? '*' : '>']
end
@irb.configure_io
assert dynamic_prompt_executed, "dynamic_prompt's assertions were not executed."
end
end
private
def build_irb
IRB.init_config(nil)
workspace = IRB::WorkSpace.new(TOPLEVEL_BINDING.dup)
IRB.conf[:VERBOSE] = false
IRB::Irb.new(workspace)
end
end
end

View File

@ -1,27 +1,10 @@
$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
require 'irb'
require 'rubygems'
require 'ostruct'
# frozen_string_literal: true
require "irb"
require_relative "helper"
module TestIRB
class RubyLexTest < TestCase
Row = Struct.new(:content, :current_line_spaces, :new_line_spaces, :indent_level)
class MockIO_AutoIndent
attr_reader :calculated_indent
def initialize(*params)
@params = params
@calculated_indent
end
def auto_indent(&block)
@calculated_indent = block.call(*@params)
end
end
def setup
save_encodings
end
@ -30,98 +13,6 @@ module TestIRB
restore_encodings
end
def calculate_indenting(lines, add_new_line)
lines = lines + [""] if add_new_line
last_line_index = lines.length - 1
byte_pointer = lines.last.length
context = build_context
context.auto_indent_mode = true
ruby_lex = RubyLex.new(context)
mock_io = MockIO_AutoIndent.new(lines, last_line_index, byte_pointer, add_new_line)
ruby_lex.configure_io(mock_io)
mock_io.calculated_indent
end
def assert_row_indenting(lines, row)
actual_current_line_spaces = calculate_indenting(lines, false)
error_message = <<~MSG
Incorrect spaces calculation for line:
```
> #{lines.last}
```
All lines:
```
#{lines.join("\n")}
```
MSG
assert_equal(row.current_line_spaces, actual_current_line_spaces, error_message)
error_message = <<~MSG
Incorrect spaces calculation for line after the current line:
```
#{lines.last}
>
```
All lines:
```
#{lines.join("\n")}
```
MSG
actual_next_line_spaces = calculate_indenting(lines, true)
assert_equal(row.new_line_spaces, actual_next_line_spaces, error_message)
end
def assert_rows_with_correct_indents(rows_with_spaces, assert_indent_level: false)
lines = []
rows_with_spaces.map do |row|
row = Row.new(*row)
lines << row.content
assert_row_indenting(lines, row)
if assert_indent_level
assert_indent_level(lines, row.indent_level)
end
end
end
def assert_indent_level(lines, expected, local_variables: [])
indent_level, _continue, _code_block_open = check_state(lines, local_variables: local_variables)
error_message = "Calculated the wrong number of indent level for:\n #{lines.join("\n")}"
assert_equal(expected, indent_level, error_message)
end
def assert_should_continue(lines, expected, local_variables: [])
_indent_level, continue, _code_block_open = check_state(lines, local_variables: local_variables)
error_message = "Wrong result of should_continue for:\n #{lines.join("\n")}"
assert_equal(expected, continue, error_message)
end
def assert_code_block_open(lines, expected, local_variables: [])
_indent_level, _continue, code_block_open = check_state(lines, local_variables: local_variables)
error_message = "Wrong result of code_block_open for:\n #{lines.join("\n")}"
assert_equal(expected, code_block_open, error_message)
end
def check_state(lines, local_variables: [])
context = build_context(local_variables)
code = lines.map { |l| "#{l}\n" }.join # code should end with "\n"
ruby_lex = RubyLex.new(context)
tokens, opens, terminated = ruby_lex.check_code_state(code)
indent_level = ruby_lex.calc_indent_level(opens)
continue = ruby_lex.should_continue?(tokens)
[indent_level, continue, !terminated]
end
def test_interpolate_token_with_heredoc_and_unclosed_embexpr
code = <<~'EOC'
+<<A-
@ -143,370 +34,6 @@ module TestIRB
end
end
def test_auto_indent
input_with_correct_indents = [
[%q(def each_top_level_statement), 0, 2],
[%q( initialize_input), 2, 2],
[%q( catch(:TERM_INPUT) do), 2, 4],
[%q( loop do), 4, 6],
[%q( begin), 6, 8],
[%q( prompt), 8, 8],
[%q( unless l = lex), 8, 10],
[%q( throw :TERM_INPUT if @line == ''), 10, 10],
[%q( else), 8, 10],
[%q( @line_no += l.count("\n")), 10, 10],
[%q( next if l == "\n"), 10, 10],
[%q( @line.concat l), 10, 10],
[%q( if @code_block_open or @ltype or @continue or @indent > 0), 10, 12],
[%q( next), 12, 12],
[%q( end), 10, 10],
[%q( end), 8, 8],
[%q( if @line != "\n"), 8, 10],
[%q( @line.force_encoding(@io.encoding)), 10, 10],
[%q( yield @line, @exp_line_no), 10, 10],
[%q( end), 8, 8],
[%q( break if @io.eof?), 8, 8],
[%q( @line = ''), 8, 8],
[%q( @exp_line_no = @line_no), 8, 8],
[%q( ), nil, 8],
[%q( @indent = 0), 8, 8],
[%q( rescue TerminateLineInput), 6, 8],
[%q( initialize_input), 8, 8],
[%q( prompt), 8, 8],
[%q( end), 6, 6],
[%q( end), 4, 4],
[%q( end), 2, 2],
[%q(end), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_braces_on_their_own_line
input_with_correct_indents = [
[%q(if true), 0, 2],
[%q( [), 2, 4],
[%q( ]), 2, 2],
[%q(end), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_multiple_braces_in_a_line
input_with_correct_indents = [
[%q([[[), 0, 6],
[%q( ]), 4, 4],
[%q( ]), 2, 2],
[%q(]), 0, 0],
[%q([<<FOO]), 0, 0],
[%q(hello), 0, 0],
[%q(FOO), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_a_closed_brace_and_not_closed_brace_in_a_line
input_with_correct_indents = [
[%q(p() {), 0, 2],
[%q(}), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_symbols
input_with_correct_indents = [
[%q(:a), 0, 0],
[%q(:A), 0, 0],
[%q(:+), 0, 0],
[%q(:@@a), 0, 0],
[%q(:@a), 0, 0],
[%q(:$a), 0, 0],
[%q(:def), 0, 0],
[%q(:`), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_endless_range_at_end_of_line
input_with_prompt = [
['001:0: :> ', %q(a = 3..)],
['002:0: :> ', %q()],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_heredoc_with_embexpr
input_with_prompt = [
['001:0:":* ', %q(<<A+%W[#{<<B)],
['002:0:":* ', %q(#{<<C+%W[)],
['003:0:":* ', %q(a)],
['004:2:]:* ', %q(C)],
['005:2:]:* ', %q(a)],
['006:0:":* ', %q(]})],
['007:0:":* ', %q(})],
['008:0:":* ', %q(A)],
['009:2:]:* ', %q(B)],
['010:1:]:* ', %q(})],
['011:0: :> ', %q(])],
['012:0: :> ', %q()],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_heredoc_prompt_with_quotes
input_with_prompt = [
["001:1:':* ", %q(<<~'A')],
["002:1:':* ", %q(#{foobar})],
["003:0: :> ", %q(A)],
["004:1:`:* ", %q(<<~`A`)],
["005:1:`:* ", %q(whoami)],
["006:0: :> ", %q(A)],
['007:1:":* ', %q(<<~"A")],
['008:1:":* ', %q(foobar)],
['009:0: :> ', %q(A)],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_backtick_method
input_with_prompt = [
['001:0: :> ', %q(self.`(arg))],
['002:0: :> ', %q()],
['003:0: :> ', %q(def `(); end)],
['004:0: :> ', %q()],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_incomplete_coding_magic_comment
input_with_correct_indents = [
[%q(#coding:u), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_incomplete_encoding_magic_comment
input_with_correct_indents = [
[%q(#encoding:u), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_incomplete_emacs_coding_magic_comment
input_with_correct_indents = [
[%q(# -*- coding: u), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_incomplete_vim_coding_magic_comment
input_with_correct_indents = [
[%q(# vim:set fileencoding=u), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_mixed_rescue
input_with_correct_indents = [
[%q(def m), 0, 2],
[%q( begin), 2, 4],
[%q( begin), 4, 6],
[%q( x = a rescue 4), 6, 6],
[%q( y = [(a rescue 5)]), 6, 6],
[%q( [x, y]), 6, 6],
[%q( rescue => e), 4, 6],
[%q( raise e rescue 8), 6, 6],
[%q( end), 4, 4],
[%q( rescue), 2, 4],
[%q( raise rescue 11), 4, 4],
[%q( end), 2, 2],
[%q(rescue => e), 0, 2],
[%q( raise e rescue 14), 2, 2],
[%q(end), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_oneliner_method_definition
input_with_correct_indents = [
[%q(class A), 0, 2],
[%q( def foo0), 2, 4],
[%q( 3), 4, 4],
[%q( end), 2, 2],
[%q( def foo1()), 2, 4],
[%q( 3), 4, 4],
[%q( end), 2, 2],
[%q( def foo2(a, b)), 2, 4],
[%q( a + b), 4, 4],
[%q( end), 2, 2],
[%q( def foo3 a, b), 2, 4],
[%q( a + b), 4, 4],
[%q( end), 2, 2],
[%q( def bar0() = 3), 2, 2],
[%q( def bar1(a) = a), 2, 2],
[%q( def bar2(a, b) = a + b), 2, 2],
[%q( def bar3() = :s), 2, 2],
[%q( def bar4() = Time.now), 2, 2],
[%q(end), 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents)
end
def test_tlambda
input_with_correct_indents = [
[%q(if true), 0, 2, 1],
[%q( -> {), 2, 4, 2],
[%q( }), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_corresponding_syntax_to_keyword_do_in_class
input_with_correct_indents = [
[%q(class C), 0, 2, 1],
[%q( while method_name do), 2, 4, 2],
[%q( 3), 4, 4, 2],
[%q( end), 2, 2, 1],
[%q( foo do), 2, 4, 2],
[%q( 3), 4, 4, 2],
[%q( end), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_corresponding_syntax_to_keyword_do
input_with_correct_indents = [
[%q(while i > 0), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while true), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while ->{i > 0}.call), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while ->{true}.call), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while i > 0 do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while true do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while ->{i > 0}.call do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(while ->{true}.call do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(foo do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(foo true do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(foo ->{true} do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
[%q(foo ->{i > 0} do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_corresponding_syntax_to_keyword_for
input_with_correct_indents = [
[%q(for i in [1]), 0, 2, 1],
[%q( puts i), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_corresponding_syntax_to_keyword_for_with_do
input_with_correct_indents = [
[%q(for i in [1] do), 0, 2, 1],
[%q( puts i), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_typing_incomplete_include_interpreted_as_keyword_in
input_with_correct_indents = [
[%q(module E), 0, 2, 1],
[%q(end), 0, 0, 0],
[%q(class A), 0, 2, 1],
[%q( in), 2, 2, 1] # scenario typing `include E`
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_bracket_corresponding_to_times
input_with_correct_indents = [
[%q(3.times { |i|), 0, 2, 1],
[%q( puts i), 2, 2, 1],
[%q(}), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_do_corresponding_to_times
input_with_correct_indents = [
[%q(3.times do |i|), 0, 2, 1],
[%q( puts i), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_bracket_corresponding_to_loop
input_with_correct_indents = [
['loop {', 0, 2, 1],
[' 3', 2, 2, 1],
['}', 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_do_corresponding_to_loop
input_with_correct_indents = [
[%q(loop do), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_local_variables_dependent_code
lines = ["a /1#/ do", "2"]
assert_indent_level(lines, 1)
@ -515,206 +42,6 @@ module TestIRB
assert_code_block_open(lines, false, local_variables: ['a'])
end
def test_embdoc_indent
input_with_correct_indents = [
[%q(=begin), 0, 0, 0],
[%q(a), 0, 0, 0],
[%q( b), 1, 1, 0],
[%q(=end), 0, 0, 0],
[%q(if 1), 0, 2, 1],
[%q( 2), 2, 2, 1],
[%q(=begin), 0, 0, 0],
[%q(a), 0, 0, 0],
[%q( b), 1, 1, 0],
[%q(=end), 0, 2, 1],
[%q( 3), 2, 2, 1],
[%q(end), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_heredoc_with_indent
if Gem::Version.new(RUBY_VERSION) < Gem::Version.new('2.7.0')
pend 'This test needs Ripper::Lexer#scan to take broken tokens'
end
input_with_correct_indents = [
[%q(<<~Q+<<~R), 0, 2, 1],
[%q(a), 2, 2, 1],
[%q(a), 2, 2, 1],
[%q( b), 2, 2, 1],
[%q( b), 2, 2, 1],
[%q( Q), 0, 2, 1],
[%q( c), 4, 4, 1],
[%q( c), 4, 4, 1],
[%q( R), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_oneliner_def_in_multiple_lines
input_with_correct_indents = [
[%q(def a()=[), 0, 2, 1],
[%q( 1,), 2, 2, 1],
[%q(].), 0, 0, 0],
[%q(to_s), 0, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_broken_heredoc
input_with_correct_indents = [
[%q(def foo), 0, 2, 1],
[%q( <<~Q), 2, 4, 2],
[%q( Qend), 4, 4, 2],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_heredoc_keep_indent_spaces
(1..4).each do |indent|
row = Row.new(' ' * indent, nil, [4, indent].max, 2)
lines = ['def foo', ' <<~Q', row.content]
assert_row_indenting(lines, row)
assert_indent_level(lines, row.indent_level)
end
end
class MockIO_DynamicPrompt
def initialize(params, &assertion)
@params = params
@assertion = assertion
end
def dynamic_prompt(&block)
result = block.call(@params)
@assertion.call(result)
end
end
def test_pasted_code_keep_base_indent_spaces
input_with_correct_indents = [
[%q( def foo), 0, 6, 1],
[%q( if bar), 6, 10, 2],
[%q( [1), 10, 12, 3],
[%q( ]+[["a), 10, 14, 4],
[%q(b" + `c), 0, 14, 4],
[%q(d` + /e), 0, 14, 4],
[%q(f/ + :"g), 0, 14, 4],
[%q(h".tap do), 0, 16, 5],
[%q( 1), 16, 16, 5],
[%q( end), 14, 14, 4],
[%q( ]), 12, 12, 3],
[%q( ]), 10, 10, 2],
[%q( end), 8, 6, 1],
[%q( end), 4, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def test_pasted_code_keep_base_indent_spaces_with_heredoc
input_with_correct_indents = [
[%q( def foo), 0, 6, 1],
[%q( if bar), 6, 10, 2],
[%q( [1), 10, 12, 3],
[%q( ]+[["a), 10, 14, 4],
[%q(b" + <<~A + <<-B + <<C), 0, 16, 5],
[%q( a#{), 16, 18, 6],
[%q( 1), 18, 18, 6],
[%q( }), 16, 16, 5],
[%q( A), 14, 16, 5],
[%q( b#{), 16, 18, 6],
[%q( 1), 18, 18, 6],
[%q( }), 16, 16, 5],
[%q( B), 14, 0, 0],
[%q(c#{), 0, 2, 1],
[%q(1), 2, 2, 1],
[%q(}), 0, 0, 0],
[%q(C), 0, 14, 4],
[%q( ]), 12, 12, 3],
[%q( ]), 10, 10, 2],
[%q( end), 8, 6, 1],
[%q( end), 4, 0, 0],
]
assert_rows_with_correct_indents(input_with_correct_indents, assert_indent_level: true)
end
def assert_dynamic_prompt(input_with_prompt)
expected_prompt_list, lines = input_with_prompt.transpose
context = build_context
ruby_lex = RubyLex.new(context)
dynamic_prompt_executed = false
io = MockIO_DynamicPrompt.new(lines) do |prompt_list|
error_message = <<~EOM
Expected dynamic prompt:
#{expected_prompt_list.join("\n")}
Actual dynamic prompt:
#{prompt_list.join("\n")}
EOM
dynamic_prompt_executed = true
assert_equal(expected_prompt_list, prompt_list, error_message)
end
ruby_lex.set_prompt do |ltype, indent, continue, line_no|
'%03d:%01d:%1s:%s ' % [line_no, indent, ltype, continue ? '*' : '>']
end
ruby_lex.configure_io(io)
assert dynamic_prompt_executed, "dynamic_prompt's assertions were not executed."
end
def test_dynamic_prompt
input_with_prompt = [
['001:1: :* ', %q(def hoge)],
['002:1: :* ', %q( 3)],
['003:0: :> ', %q(end)],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_dynamic_prompt_with_double_newline_breaking_code
input_with_prompt = [
['001:1: :* ', %q(if true)],
['002:2: :* ', %q(%)],
['003:1: :* ', %q(;end)],
['004:1: :* ', %q(;hello)],
['005:0: :> ', %q(end)],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_dynamic_prompt_with_multiline_literal
input_with_prompt = [
['001:1: :* ', %q(if true)],
['002:2:]:* ', %q( %w[)],
['003:2:]:* ', %q( a)],
['004:1: :* ', %q( ])],
['005:1: :* ', %q( b)],
['006:2:]:* ', %q( %w[)],
['007:2:]:* ', %q( c)],
['008:1: :* ', %q( ])],
['009:0: :> ', %q(end)],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_dynamic_prompt_with_blank_line
input_with_prompt = [
['001:1:]:* ', %q(%w[)],
['002:1:]:* ', %q()],
['003:0: :> ', %q(])],
]
assert_dynamic_prompt(input_with_prompt)
end
def test_literal_ends_with_space
assert_code_block_open(['% a'], true)
assert_code_block_open(['% a '], false)
@ -890,5 +217,33 @@ module TestIRB
IRB.conf[:VERBOSE] = false
IRB::Context.new(nil, workspace)
end
def assert_indent_level(lines, expected, local_variables: [])
indent_level, _continue, _code_block_open = check_state(lines, local_variables: local_variables)
error_message = "Calculated the wrong number of indent level for:\n #{lines.join("\n")}"
assert_equal(expected, indent_level, error_message)
end
def assert_should_continue(lines, expected, local_variables: [])
_indent_level, continue, _code_block_open = check_state(lines, local_variables: local_variables)
error_message = "Wrong result of should_continue for:\n #{lines.join("\n")}"
assert_equal(expected, continue, error_message)
end
def assert_code_block_open(lines, expected, local_variables: [])
_indent_level, _continue, code_block_open = check_state(lines, local_variables: local_variables)
error_message = "Wrong result of code_block_open for:\n #{lines.join("\n")}"
assert_equal(expected, code_block_open, error_message)
end
def check_state(lines, local_variables: [])
context = build_context(local_variables)
code = lines.map { |l| "#{l}\n" }.join # code should end with "\n"
ruby_lex = RubyLex.new(context)
tokens, opens, terminated = ruby_lex.check_code_state(code)
indent_level = ruby_lex.calc_indent_level(opens)
continue = ruby_lex.should_continue?(tokens)
[indent_level, continue, !terminated]
end
end
end