make keccak module configurable

README.md

@@ -19,6 +19,14 @@ Documentation can be generated with [ExDoc](https://github.com/elixir-lang/ex_do
 and published on [HexDocs](https://hexdocs.pm). Once published, the docs can
 be found at [https://hexdocs.pm/ex_abi](https://hexdocs.pm/ex_abi).
 
+## Confiiguration
+
+The default keccak library is set to `ex_keccak` but that can be ovveriden for a different libary:
+
+```elixir
+config :ex_abi, keccak_module: KeccakEx
+```
+
 ## Usage
 
 ### Encoding

lib/abi/function_selector.ex

@@ -7,6 +7,8 @@ defmodule ABI.FunctionSelector do
   require Integer
   require Logger
 
+  @keccak_module Application.get_env(:ex_abi, :keccak_module, ExKeccak)
+
   @type type ::
           {:uint, integer()}
           | :bool
@@ -404,7 +406,7 @@ defmodule ABI.FunctionSelector do
   defp add_method_id(selector) do
     signature = encode(selector)
 
-    case ExKeccak.hash_256(signature) do
+    case @keccak_module.hash_256(signature) do
       <<method_id::binary-size(4), _::binary>> ->
         %{selector | method_id: method_id}
 
@@ -416,7 +418,7 @@ defmodule ABI.FunctionSelector do
   defp add_event_id(selector) do
     signature = encode(selector)
 
-    %{selector | method_id: ExKeccak.hash_256(signature)}
+    %{selector | method_id: @keccak_module.hash_256(signature)}
   end
 
   defp get_types(function_selector) do

lib/abi/type_encoder.ex

@@ -7,6 +7,8 @@ defmodule ABI.TypeEncoder do
 
   alias ABI.FunctionSelector
 
+  @keccak_module Application.get_env(:ex_abi, :keccak_module, ExKeccak)
+
   @doc """
   Encodes the given data based on the function selector.
 
@@ -248,7 +250,7 @@ defmodule ABI.TypeEncoder do
     kec =
       function_selector
       |> FunctionSelector.encode()
-      |> ExKeccak.hash_256()
+      |> @keccak_module.hash_256()
 
     # Take first four bytes
     <<init::binary-size(4), _rest::binary>> = kec

mix.exs

@@ -33,7 +33,7 @@ defmodule ABI.Mixfile do
       {:dialyxir, "~> 1.4", only: [:dev, :test], runtime: false},
       {:ex_doc, "~> 0.24", only: :dev, runtime: false},
       {:jason, "~> 1.4"},
-      {:ex_keccak, "~> 0.7.5"},
+      {:ex_keccak, "~> 0.7.5", optional: true},
       {:propcheck, "~> 1.4", only: [:test, :dev]}
     ]
   end