module Main (main) where

import System.Environment.Blank ( getArgs )
import Lib (task1, task2)

main :: IO ()
main = do
  args <- getArgs
  case args of
    ["1", file] -> do
      input <- readFile file
      result <- task1 input
      print result
    ["2", file] -> do
      input <- readFile file
      result <- task2 input
      print result
    _ -> error "Usage: <1|2> <input file>"