Format: fix to format $1? (#5313)

Fixed #5310
crystal-lang · Nov 19, 2017 · 6027a94 · 6027a94
1 parent 818c438
commit 6027a94
Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/spec/compiler/formatter/formatter_spec.cr b/spec/compiler/formatter/formatter_spec.cr
@@ -414,6 +414,7 @@ describe Crystal::Formatter do
   assert_format "$1.bar", "$1.bar"
   assert_format "$0", "$0"
   assert_format "$0.bar", "$0.bar"
+  assert_format "$1?"
 
   assert_format "foo . is_a? ( Bar )", "foo.is_a?(Bar)"
   assert_format "foo . responds_to?( :bar )", "foo.responds_to?(:bar)"

diff --git a/src/compiler/crystal/tools/formatter.cr b/src/compiler/crystal/tools/formatter.cr
@@ -2061,7 +2061,7 @@ module Crystal
       column = @column
 
       # Special case: $1, $2, ...
-      if @token.type == :GLOBAL_MATCH_DATA_INDEX && node.name == "[]" && obj.is_a?(Global)
+      if @token.type == :GLOBAL_MATCH_DATA_INDEX && (node.name == "[]" || node.name == "[]?") && obj.is_a?(Global)
         write "$"
         write @token.value
         next_token