add guarded function support, bump version

README.md

@@ -42,7 +42,7 @@ Efx follows the following principles:
 To use `Efx` in your project, add this to your dependencies in `mix.ex`:
 
 ```elixir
-{:efx, "~> 0.2.1"}
+{:efx, "~> 0.2.2"}
 ```
 
 If you want to have proper formatting of the `Efx.defeffect/2` macro, you can add

lib/efx.ex

@@ -189,7 +189,7 @@ defmodule Efx do
   end
 
   defmacro defeffect(fun, do_block) do
-    {name, ctx, args} = fun
+    {name, ctx, args} = extract_fun(fun)
     args = ensure_list(args)
     module = __CALLER__.module
 
@@ -239,6 +239,14 @@ defmodule Efx do
     end
   end
 
+  defp extract_fun({:when, _ctx, [fun, _when_condition]}) do
+    fun
+  end
+
+  defp extract_fun(fun) do
+    fun
+  end
+
   @spec spec_name({any(), any(), list()}) :: name :: atom()
   defp spec_name({_, _, a}) do
     a = ensure_list(a)

mix.exs

@@ -1,7 +1,7 @@
 defmodule Efx.MixProject do
   use Mix.Project
 
-  @version "0.2.1"
+  @version "0.2.2"
   @github_page "https://github.com/bravobike/efx"
 
   def project do

test/efx_case_test.exs

@@ -30,6 +30,11 @@ defmodule EfxCaseTest do
       assert EfxExample.multi_fun(:a) == :a
     end
 
+    test "on guarded funs" do
+      bind(EfxExample, :guarded_fun, fn arg -> arg end)
+      assert EfxExample.guarded_fun(:a) == :a
+    end
+
     test "works with expected number of calls" do
       bind(EfxExample, :get, fn -> [] end, calls: 1)
       assert EfxExample.get() == []

test/support/efx_example.ex

@@ -29,6 +29,24 @@ defmodule EfxCase.EfxExample do
     string
   end
 
+  # we use this to see if functions with guards
+  # work properly at compile time
+  @spec guarded_fun(any()) :: any()
+  defeffect guarded_fun(a) when a in [:a, :b] do
+    {a, :a_or_b}
+  end
+
+  @spec guarded_fun(any()) :: any()
+  defeffect guarded_fun(a) do
+    a
+  end
+
+  # we use this to see if one liners work without getting 
+  # deformed by formatter
+
+  @spec one_liner(any()) :: any()
+  defeffect one_liner(a), do: a
+
   @spec without_parens :: atom
   defeffect without_parens do
     :ok