support for structs

Author: mindreframer

lib/atomic_map.ex

@@ -1,5 +1,13 @@
 defmodule AtomicMap do
   def convert(v, opts \\ [])
+
+  def convert(struct=%{__struct__: type}, opts) do
+    struct
+    |> Map.from_struct
+    |> convert(opts)
+    |> Map.put(:__struct__, type)
+  end
+
   def convert(map, opts) when is_map(map) do
     safe = Keyword.get(opts, :safe, true)
     map |> Enum.reduce(%{}, fn({k,v}, acc)->

test/atomic_map_test.exs

@@ -2,6 +2,10 @@ defmodule AtomicMapTest do
   use ExUnit.Case
   doctest AtomicMap
 
+  defmodule MyStruct do
+    defstruct first: nil, second: nil
+  end
+
   test "works with maps" do
     input    = %{"a" => 2, "b" => %{"c" => 4}}
     expected = %{a: 2, b: %{c: 4}}
@@ -38,6 +42,18 @@ defmodule AtomicMapTest do
     assert AtomicMap.convert(input) == expected
   end
 
+  test "works with structs" do
+    input    = %AtomicMapTest.MyStruct{first: [%{"a" => 1}]}
+    expected = %AtomicMapTest.MyStruct{first: [%{a: 1}]}
+    assert AtomicMap.convert(input) == expected
+  end
+
+  test "works with nested structs" do
+    input    = %AtomicMapTest.MyStruct{first: [%AtomicMapTest.MyStruct{first: %{"b" => 1}}]}
+    expected = %AtomicMapTest.MyStruct{first: [%AtomicMapTest.MyStruct{first: %{b: 1}}]}
+    assert AtomicMap.convert(input) == expected
+  end
+
   test "raises for not existing atoms" do
     assert_raise ArgumentError, fn ->
       input = %{"a" => 2, "b" => %{"c" => 4}, "__not___existing__" => 5}