diff options
| author | Nobuhiro IMAI <nov@yo.rim.or.jp> | 2020-09-22 09:06:43 +0900 |
|---|---|---|
| committer | aycabta <aycabta@gmail.com> | 2020-11-22 21:00:11 +0900 |
| commit | 5218f17737594867ffe5c2cd9c90b19c7378af05 (patch) | |
| tree | 8e91b19c17a9741ccef405b494dd8683e5f105ed | |
| parent | 7c879cbb3f24c742e0d3226c39d413abe7519205 (diff) | |
[ruby/irb] support more body argument for oneliner method definition
https://github.com/ruby/irb/commit/2ff1295533
| -rw-r--r-- | lib/irb/ruby-lex.rb | 12 | ||||
| -rw-r--r-- | test/irb/test_ruby_lex.rb | 2 |
2 files changed, 5 insertions, 9 deletions
diff --git a/lib/irb/ruby-lex.rb b/lib/irb/ruby-lex.rb index 7be77fd12e..c75d5ebf2e 100644 --- a/lib/irb/ruby-lex.rb +++ b/lib/irb/ruby-lex.rb @@ -317,14 +317,12 @@ class RubyLex if t[2] == '=' in_oneliner_def = :BODY end - elsif t[3].allbits?(Ripper::EXPR_END) + else if in_oneliner_def == :BODY # one-liner method definition indent -= 1 end in_oneliner_def = nil - else - in_oneliner_def = nil end end @@ -376,14 +374,12 @@ class RubyLex if t[2] == '=' in_oneliner_def = :BODY end - elsif t[3].allbits?(Ripper::EXPR_END) + else if in_oneliner_def == :BODY # one[-liner method definition depth_difference -= 1 end in_oneliner_def = nil - else - in_oneliner_def = nil end end @@ -451,7 +447,7 @@ class RubyLex if t[2] == '=' in_oneliner_def = :BODY end - elsif t[3].allbits?(Ripper::EXPR_END) + else if in_oneliner_def == :BODY # one-liner method definition if is_first_printable_of_line @@ -462,8 +458,6 @@ class RubyLex end end in_oneliner_def = nil - else - in_oneliner_def = nil end end diff --git a/test/irb/test_ruby_lex.rb b/test/irb/test_ruby_lex.rb index 14b43f5718..25f037ec3b 100644 --- a/test/irb/test_ruby_lex.rb +++ b/test/irb/test_ruby_lex.rb @@ -233,6 +233,8 @@ module TestIRB Row.new(%q( def bar0() = 3), nil, 2), Row.new(%q( def bar1(a) = a), nil, 2), Row.new(%q( def bar2(a, b) = a + b), nil, 2), + Row.new(%q( def bar3() = :s), nil, 2), + Row.new(%q( def bar4() = Time.now), nil, 2), Row.new(%q(end), 0, 0), ] |