Fix defpartial and defcurry do not support zero arity functions w…

…ithout parentheses witchcrafters#45
florius0 · Jun 22, 2022 · 78ea435 · 78ea435
1 parent 2e8b6d5
commit 78ea435
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/lib/quark/curry.ex b/lib/quark/curry.ex
@@ -100,4 +100,5 @@ defmodule Quark.Curry do
   end
 
   defp wrap([], body), do: body
+  defp wrap(nil, body), do: body
 end
diff --git a/lib/quark/partial.ex b/lib/quark/partial.ex
@@ -48,7 +48,11 @@ defmodule Quark.Partial do
       #=> 3
 
   """
-  defmacro defpartial({fun_name, ctx, args}, do: body) do
+  defmacro defpartial({fun_name, ctx, nil}, do: body), do: defpartial_quote({fun_name, ctx, []}, do: body)
+
+  defmacro defpartial({fun_name, ctx, args}, do: body), do: defpartial_quote({fun_name, ctx, args}, do: body)
+
+  defp defpartial_quote({fun_name, ctx, args}, do: body) do
     quote do
       defcurry unquote({fun_name, ctx, args}), do: unquote(body)
       unquote do: Enum.map(args_scan(args), &rehydrate(fun_name, ctx, &1))