Curried functions in Elixir

curry.ex

defmodule Curry do
  defmacro curry([{name, arity}]) do
    (arity - 1)
    |> Range.new(0)
    |> Enum.map(fn arity ->
      arguments =
        Range.new(0, arity)
        |> Enum.filter(&(&1 > 0))
        |> Enum.map(&"a#{&1}")
        |> Enum.map(&String.to_atom/1)
        |> Enum.map(&Macro.var(&1, nil))

      case arguments do
        [] ->
          quote do
            def unquote(name)() do
              &unquote(name)(&1)
            end
          end

        arguments ->
          quote do
            def unquote(name)(unquote_splicing(arguments)) do
              &unquote(name)(unquote_splicing(arguments), &1)
            end
          end
      end
    end)
  end
end

defmodule Math do
  import Curry

  def add(a, b), do: a + b

  curry add: 2
end