Normalize captures (#21)

.github/workflows/ci.yml

@@ -10,6 +10,7 @@ jobs:
     name: Test on Ubuntu (Elixir ${{ matrix.elixir }}, OTP ${{ matrix.otp }})
     runs-on: ubuntu-20.04
     strategy:
+      fail-fast: false
       matrix:
         elixir:
           - '1.13.4'

lib/beam_file/normalizer.ex

@@ -76,6 +76,13 @@ defmodule BeamFile.Normalizer do
     end
   end
 
+  def normalize({:capture, meta, nil} = ast, :code) do
+    case Keyword.get(meta, :counter) do
+      {Capture, count} -> {:"capture#{count}", meta, nil}
+      _else -> ast
+    end
+  end
+
   def normalize({:super, meta, args}, target) do
     case Keyword.has_key?(meta, :super) do
       true ->

test/beam_file_test.exs

@@ -596,6 +596,11 @@ defmodule BeamFileTest do
       assert code <> "\n" == TestSupport.fixture("comps.exs")
     end
 
+    test "returns elixir code for the Capture module" do
+      assert {:ok, code} = BeamFile.elixir_code(Capture)
+      assert code <> "\n" == TestSupport.fixture("capture.exs")
+    end
+
     test "returns an error for invalid binary" do
       assert BeamFile.elixir_code(<<0, 0, 7>>) == {:error, {:not_a_beam_file, <<0, 0, 7>>}}
     end

test/fixtures/1.13.4/capture.exs

@@ -0,0 +1,9 @@
+defmodule Elixir.Capture do
+  def fun do
+    fn x1, x2 -> :erlang.+(:erlang.*(x1, x2), x1) end
+  end
+
+  def double do
+    fn x1 -> :erlang.+(x1, x1) end
+  end
+end

test/fixtures/1.14.5/capture.exs

@@ -0,0 +1,9 @@
+defmodule Elixir.Capture do
+  def fun do
+    fn x1, x2 -> :erlang.+(:erlang.*(x1, x2), x1) end
+  end
+
+  def double do
+    fn x1 -> :erlang.+(x1, x1) end
+  end
+end

test/fixtures/1.15.8/capture.exs

@@ -0,0 +1,9 @@
+defmodule Elixir.Capture do
+  def fun do
+    fn x1, x2 -> :erlang.+(:erlang.*(x1, x2), x1) end
+  end
+
+  def double do
+    fn x1 -> :erlang.+(x1, x1) end
+  end
+end

test/fixtures/1.16.3/capture.exs

@@ -0,0 +1,9 @@
+defmodule Elixir.Capture do
+  def fun do
+    fn x1, x2 -> :erlang.+(:erlang.*(x1, x2), x1) end
+  end
+
+  def double do
+    fn x1 -> :erlang.+(x1, x1) end
+  end
+end

test/fixtures/1.17.2/capture.exs

@@ -0,0 +1,9 @@
+defmodule Elixir.Capture do
+  def fun do
+    fn capture3, capture4 -> :erlang.+(:erlang.*(capture3, capture4), capture3) end
+  end
+
+  def double do
+    fn capture5 -> :erlang.+(capture5, capture5) end
+  end
+end

test/fixtures/capture.ex

@@ -0,0 +1,9 @@
+defmodule Capture do
+  def fun do
+    &(&1 * &2 + &1)
+  end
+
+  def double do
+    &(&1 + &1)
+  end
+end