Refactor cube extraction logic in Day2 module

samjowen · Dec 27, 2023 · 71f1c72 · 71f1c72
1 parent 1548c62
commit 71f1c72
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/lib/day2/day2.ex b/lib/day2/day2.ex
@@ -25,21 +25,24 @@ defmodule Day2 do
 
     cube_list = String.split(set_string, ",")
 
-    Enum.reduce(cube_list, initial_cube_map, fn x, acc ->
-      case String.contains?(x, "red") do
-        true -> Map.put(acc, :red, extract_integer(x))
-        false -> acc
-      end
+    Enum.reduce(cube_list, initial_cube_map, fn cube, acc ->
+      acc
       |> (fn acc ->
-            case String.contains?(x, "green") do
-              true -> Map.put(acc, :green, extract_integer(x))
-              false -> acc
+            case String.contains?(cube, "red") do
+              true -> Map.put(acc, :red, extract_integer(cube))
+              _ -> acc
             end
           end).()
       |> (fn acc ->
-            case String.contains?(x, "blue") do
-              true -> Map.put(acc, :blue, extract_integer(x))
-              false -> acc
+            case String.contains?(cube, "green") do
+              true -> Map.put(acc, :green, extract_integer(cube))
+              _ -> acc
+            end
+          end).()
+      |> (fn acc ->
+            case String.contains?(cube, "blue") do
+              true -> Map.put(acc, :blue, extract_integer(cube))
+              _ -> acc
             end
           end).()
     end)