simple runner for 2022

2022/aoc.cabal

@@ -25,7 +25,7 @@ source-repository head
 
 library
   exposed-modules:
-      Lib
+      Aoc.Day1
   other-modules:
       Paths_aoc
   hs-source-dirs:

2022/app/Main.hs

@@ -2,32 +2,18 @@
 module Main (main) where
 
 import System.Environment (getArgs)
-import Data.Ord 
-import Data.List (sortBy)
-import Data.Text (pack, splitOn, unpack)
-    
-parseInput :: String -> [[Integer]]
-parseInput = (map (map read . words . unpack)) . splitOn "\n\n" . pack
-
-inputSums :: String -> [Integer]
-inputSums = map sum . parseInput
-
-part1 :: String -> Integer
-part1 = maximum . inputSums
-
-part2 :: String -> Integer 
-part2 = sum . take 3 . sortDesc . inputSums
-  where
-    sortDesc = sortBy (comparing Down)
+import qualified Aoc.Day1
 
 main :: IO ()
 main = do
-  args <- getArgs
-  contents <- getContents
+  (dayS:partS:_) <- getArgs
+  let day = read dayS :: Integer
+  let part = read partS :: Integer
 
-  let f = case args of ("1":_) -> part1
-                       ("2":_) -> part2
-                       _ -> error "unknown part"
+  contents <- readFile ("input/" ++ (show day) ++ ".txt")
 
-  putStrLn $ show $ f contents
+  let f = case day of 1 -> Aoc.Day1.solve
+                      _ -> error "unknown day"
+
+  putStrLn $ show $ f part contents
 

2022/src/Aoc/Day1.hs

@@ -0,0 +1,24 @@
+{-# LANGUAGE OverloadedStrings #-}
+module Aoc.Day1 (solve) where
+
+import Data.Ord 
+import Data.List (sortBy)
+import Data.Text (pack, splitOn, unpack)
+
+parseInput :: String -> [[Integer]]
+parseInput = (map (map read . words . unpack)) . splitOn "\n\n" . pack
+
+inputSums :: String -> [Integer]
+inputSums = map sum . parseInput
+
+part1 :: String -> Integer
+part1 = maximum . inputSums
+
+part2 :: String -> Integer 
+part2 = sum . take 3 . sortDesc . inputSums
+  where
+    sortDesc = sortBy (comparing Down)
+
+solve :: Integer -> String -> Integer
+solve 1 input = part1 input
+solve 2 input = part2 input

2022/src/Lib.hs

@@ -1,6 +0,0 @@
-module Lib
-    ( someFunc
-    ) where
-
-someFunc :: IO ()
-someFunc = putStrLn "someFunc"