Refactor Day6 part 1 to use a struct (will re-use in part 2)

2024/day6.exs

@@ -1,24 +1,30 @@
 #!/usr/bin/env elixir
+defmodule Patrol do
+  defstruct obstacles: MapSet.new(),
+            visited: MapSet.new(),
+            x: 0,
+            y: 0,
+            max_x: nil,
+            max_y: nil,
+            dir_x: 0,
+            dir_y: -1
+end
+
 defmodule Day6 do
-  def part1({max_x, max_y, {x, y}, obstacles}) do
+  def part1(%Patrol{} = patrol) when patrol.x == patrol.max_x or patrol.y == patrol.max_y do
-    walk(x, y, max_x, max_y, obstacles, MapSet.new(), 0, -1)
+    MapSet.size(patrol.visited)
-    |> MapSet.size()
   end
 
-  def walk(x, y, max_x, max_y, _obstacles, visited, _x_dir, _) when x == max_x or y == max_y do
+  def part1(%Patrol{} = patrol) do
-    visited
+    next_x = patrol.x + patrol.dir_x
-  end
+    next_y = patrol.y + patrol.dir_y
 
-  def walk(x, y, max_x, max_y, obstacles, visited, x_dir, y_dir) do
+    if MapSet.member?(patrol.obstacles, {next_x, next_y}) do
-    next_x = x + x_dir
+      {dir_x, dir_y} = turn_right(patrol.dir_x, patrol.dir_y)
-    next_y = y + y_dir
+      part1(%Patrol{patrol | dir_x: dir_x, dir_y: dir_y})
-
-    if MapSet.member?(obstacles, {next_x, next_y}) do
-      {x_dir, y_dir} = turn_right(x_dir, y_dir)
-      walk(x, y, max_x, max_y, obstacles, visited, x_dir, y_dir)
     else
-      visited = MapSet.put(visited, {x, y})
+      visited = MapSet.put(patrol.visited, {patrol.x, patrol.y})
-      walk(next_x, next_y, max_x, max_y, obstacles, visited, x_dir, y_dir)
+      part1(%Patrol{patrol | x: next_x, y: next_y, visited: visited})
     end
   end
 
@@ -35,6 +41,7 @@ defmodule Day6 do
   def input do
     with [input_filename] <- System.argv(),
          {:ok, input} <- File.read(input_filename) do
+      {_x, max_x, max_y, {x, y}, obstacles} =
         for <<char::binary-1 <- input>>, reduce: {0, 0, 0, nil, MapSet.new()} do
           {x, max_x, y, guard, obstacles} ->
             case char do
@@ -44,7 +51,8 @@ defmodule Day6 do
               "\n" -> {0, x, y + 1, guard, obstacles}
             end
         end
-      |> Tuple.delete_at(0)
+
+      %Patrol{x: x, y: y, max_x: max_x, max_y: max_y, obstacles: obstacles}
     else
       _ -> :error
     end