src/Main.idr

@@ -2,8 +2,9 @@ module Main
 
 import Data.SortedMap
 import Data.Problem
+import Utils
+
 import Language.Reflection
-import System
 
 import AllDays
 
@@ -31,13 +32,11 @@ getInput = go ""
 
 main : IO ()
 main = do
-  sol <- case !getArgs of
-           [_, s] => pure s
-           _ => putStr "Solution: " >> getLine
-  let Just pr = parseProblem sol
-        | Nothing => putStrLn "ERROR: Invalid solution"
-      Just func = lookup pr solMap
-        | Nothing => putStrLn "ERROR: This part is not implemented yet!"
+  putStr "Solution: "
+  Just pr <- parseProblem <$> getLine
+    | Nothing => putStrLn "Invalid solution ID"
+  let Just func = lookup pr solMap
+    | Nothing => putStrLn "This part is not implemented yet!"
 
   putStrLn "Input (end with \\):"
   input <- trim <$> getInput