Fixed #2960: formatter fails with $0

crystal-lang · Jul 6, 2016 · d30f29c · d30f29c
1 parent e8b1b84
commit d30f29c
Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/spec/compiler/formatter/formatter_spec.cr b/spec/compiler/formatter/formatter_spec.cr
@@ -391,6 +391,8 @@ describe Crystal::Formatter do
   assert_format "$? = 1", "$? = 1"
   assert_format "$1", "$1"
   assert_format "$1.bar", "$1.bar"
+  assert_format "$0", "$0"
+  assert_format "$0.bar", "$0.bar"
 
   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
@@ -889,6 +889,14 @@ module Crystal
     end
 
     def visit(node : Path)
+      # This is the case of $0, which generates `PROGRAM_NAME` by the parser
+      if @token.type == :GLOBAL_MATCH_DATA_INDEX
+        write "$"
+        write @token.value
+        next_token
+        return false
+      end
+
       check_open_paren
 
       # Sometimes the :: is not present because the parser generates ::Nil, for example