module Day1 (
  day1,
  findElfWithHighestCalorieCount,
  findCaloriesOfTop3Elves
) where

import qualified Data.Text as T
import Day1.Internal

day1 :: IO ()
day1 = do
  putStrLn "Day1"
  calorieCount <- day1_1
  putStrLn ("Highest calorie count = " ++ calorieCount)
  caloriesOfTop3Elves <- day1_2
  putStrLn ("Calories of top 3 elves = " ++ caloriesOfTop3Elves)


day1_1 :: IO String
day1_1 = do
  input <- readFile "ressources/day01-input"
  return (show $ findElfWithHighestCalorieCount $ T.pack input)

day1_2 :: IO String
day1_2 = do
  input <- readFile "ressources/day01-input"
  return (show $ findCaloriesOfTop3Elves $ T.pack input)


findCaloriesOfTop3Elves :: T.Text -> CalorieCount
findCaloriesOfTop3Elves  =
  calories . sumCalories . take 3 . sortElves . parseElfInput
  where
    sumCalories :: [Elf] -> Int
    sumCalories = sum . map getCalories

findElfWithHighestCalorieCount :: T.Text -> Elf
findElfWithHighestCalorieCount = head . sortElves . parseElfInput