Refactor to make things more reusable

aoc.cabal

@@ -11,7 +11,10 @@ executable aoc
     default-language: GHC2024
     default-extensions:
         BlockArguments
+        DuplicateRecordFields
         MultiWayIf
+        NoFieldSelectors
+        OverloadedRecordDot
         ViewPatterns
     ghc-options:
         -Wall

app/Main.hs

@@ -2,12 +2,29 @@ module Main (main) where
 
 import Control.Monad.State
 import Data.Bifunctor
-import Data.Traversable
 import Text.Read (readMaybe)
 
 main :: IO ()
 main = do
-    Just input <-
+    runPuzzle puzzle1
+
+runPuzzle :: Puzzle a -> IO ()
+runPuzzle p = do
+    Just input <- p.parse <$> readFile ("inputs/examples/" <> show p.number)
+    putStrLn $ p.part1 input
+    putStrLn $ p.part2 input
+data Puzzle input = Puzzle
+    { number :: Word
+    , parse :: String -> Maybe input
+    , part1 :: input -> String
+    , part2 :: input -> String
+    }
+
+puzzle1 :: Puzzle [(Direction, Inc)]
+puzzle1 =
+    Puzzle
+        { number = 1
+        , parse =
             traverse
                 ( \case
                     'L' : (readMaybe -> Just i) -> Just (L, Inc i)
@@ -15,17 +32,18 @@ main = do
                     _ -> Nothing
                 )
                 . lines
-            <$> readFile "inputs/examples/1"
+        , part1 =
-    print
+            show
                 . sum
                 . flip evalState 50
-        $ for input \(d, i) -> state \p ->
+                . traverse \(d, i) -> state \p ->
                     let (_, p') = step i d p
                      in (Count if p' == 0 then 1 else 0, p')
-    print
+        , part2 =
+            show
                 . sum
                 . flip evalState 50
-        $ for input \(d, i) -> state \p ->
+                . traverse \(d, i) -> state \p ->
                     let (c, p') = step i d p
                         c' = case d of
                             R -> abs c
@@ -35,6 +53,7 @@ main = do
                                     | p' == 0 -> abs c + 1
                                     | otherwise -> abs c
                      in (c', p')
+        }
 
 data Direction = L | R
     deriving (Eq, Ord, Show)