From a13fa323b76d3815f9c92e5bb44c38ba12f322e9 Mon Sep 17 00:00:00 2001
From: Michael Adler <therisen06@gmail.com>
Date: Thu, 19 Dec 2024 18:47:04 +0100
Subject: [PATCH] Day 17, Part 2: WIP

---
 src/day17/day17.lua      | 66 +++++++++++++++++++++-------------------
 src/day17/day17_spec.lua | 14 +++++++--
 2 files changed, 47 insertions(+), 33 deletions(-)

diff --git a/src/day17/day17.lua b/src/day17/day17.lua
index 39f38b5..d7f60bf 100644
--- a/src/day17/day17.lua
+++ b/src/day17/day17.lua
@@ -5,35 +5,7 @@ local bit = require("bit")
 local bxor = bit.bxor
 local pow = math.pow
 
---- @param input string
-M.solve = function(input)
-    local prog = {}
-    local prog_count = 0
-    local reg_A, reg_B, reg_C
-
-    local parse = function()
-        local i = 1
-        for line in input:gmatch("[^\r\n]+") do
-            if i == 1 then
-                local num = line:match("Register A: (%d+)")
-                reg_A = tonumber(num)
-            elseif i == 2 then
-                local num = line:match("Register B: (%d+)")
-                reg_B = tonumber(num)
-            elseif i == 3 then
-                local num = line:match("Register C: (%d+)")
-                reg_C = tonumber(num)
-            else
-                for num in line:gmatch("(%d+)") do
-                    prog[prog_count] = tonumber(num)
-                    prog_count = prog_count + 1
-                end
-            end
-            i = i + 1
-        end
-    end
-    parse()
-
+local function run_prog(prog, prog_count, reg_A, reg_B, reg_C)
     local function combo_operand(operand)
         --  Combo operands 0 through 3 represent literal values 0 through 3.
         if operand >= 0 and operand <= 3 then
@@ -116,9 +88,41 @@ M.solve = function(input)
         ip = ip + 2
         ::continue::
     end
-    local part1 = table.concat(output, ",")
+    return output
+end
+
+--- @param input string
+M.solve = function(input)
+    local parse = function()
+        local prog = {}
+        local prog_count = 0
+        local reg_A, reg_B, reg_C
+        local i = 1
+        for line in input:gmatch("[^\r\n]+") do
+            if i == 1 then
+                local num = line:match("Register A: (%d+)")
+                reg_A = tonumber(num)
+            elseif i == 2 then
+                local num = line:match("Register B: (%d+)")
+                reg_B = tonumber(num)
+            elseif i == 3 then
+                local num = line:match("Register C: (%d+)")
+                reg_C = tonumber(num)
+            else
+                for num in line:gmatch("(%d+)") do
+                    prog[prog_count] = tonumber(num)
+                    prog_count = prog_count + 1
+                end
+            end
+            i = i + 1
+        end
+        return prog, prog_count, reg_A, reg_B, reg_C
+    end
+    local prog, prog_count, original_reg_A, reg_B, reg_C = parse()
+    local part1 = table.concat(run_prog(prog, prog_count, original_reg_A, reg_B, reg_C), ",")
 
-    return part1, 0
+    local part2 = 0
+    return part1, part2
 end
 
 return M
diff --git a/src/day17/day17_spec.lua b/src/day17/day17_spec.lua
index 7c4140a..60dddd7 100644
--- a/src/day17/day17_spec.lua
+++ b/src/day17/day17_spec.lua
@@ -8,12 +8,22 @@ Register C: 0
 
 Program: 0,1,5,4,3,0
 ]]
-        local part1, part2 = day17.solve(input)
+        local part1, _ = day17.solve(input)
         it("part1", function()
             assert.are.equal("4,6,3,5,6,3,5,2,1,0", part1)
         end)
+    end)
+
+    describe("example 2", function()
+        local input = [[Register A: 2024
+Register B: 0
+Register C: 0
+
+Program: 0,3,5,4,3,0
+]]
+        local _, part2 = day17.solve(input)
         it("part2", function()
-            assert.are.equal(0, part2)
+            assert.are.equal(117440, part2)
         end)
     end)
 end)