fix: handle prefix operator with newline (#22)

elixir-tools · Apr 11, 2024 · 4a2fcb6 · 4a2fcb6
1 parent d39f91d
commit 4a2fcb6
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/lib/spitfire.ex b/lib/spitfire.ex
@@ -593,7 +593,7 @@ defmodule Spitfire do
         current_precedence(parser)
       end
 
-    parser = next_token(parser)
+    parser = parser |> next_token() |> eat_eol()
     {rhs, parser} = parse_expression(parser, precedence, false, false, false)
 
     ast = {token, meta, [rhs]}

diff --git a/test/spitfire_test.exs b/test/spitfire_test.exs
@@ -101,6 +101,13 @@ defmodule SpitfireTest do
       '''
 
       assert Spitfire.parse(code) == s2q(code)
+
+      code = ~S'''
+      ^
+      foo
+      '''
+
+      assert Spitfire.parse(code) == s2q(code)
     end
 
     test "parses numbers" do
@@ -842,6 +849,11 @@ defmodule SpitfireTest do
         @foo bar()
         ''',
         ~s'''
+        @
+
+        __cursor__()
+        ''',
+        ~s'''
         @foo %{
           foo: :bar
         }