Choose word length with command line argument

app/Main.hs

@@ -140,11 +140,11 @@ loop = do
 runLoop :: GuessState -> IO GuessState
 runLoop = execStateT (H.runInputT H.defaultSettings loop)
 
-loadAndRun :: [FilePath] -> IO ()
+loadAndRun :: Int -> [FilePath] -> IO ()
-loadAndRun files = do
+loadAndRun n files = do
   wordlists <- mapM readFile files
   let wordlist = concatMap lines wordlists
-      cmap = buildClassMap (Just 5) wordlist
+      cmap = buildClassMap (Just n) wordlist
       g = guessState cmap
 
   let wordCount = numberjust $ show $ length wordlist
@@ -179,13 +179,19 @@ loadAndRun file = do
   --  - quit -> quit the program (alias: q)
 -}
 
+wrongArgs :: IO ()
+wrongArgs = do
+  name <- getProgName
+  putStrLn "  USAGE:"
+  putStrLn $ name ++ " <word length> <dictionary files>"
+  die "Error: No word length or dictionary given."
+
 main :: IO ()
 main = do
   args <- getArgs
   case args of
-    [] -> do
+    [] -> wrongArgs
-      name <- getProgName
+    [_] -> wrongArgs
-      putStrLn "  USAGE:"
+    (n:files) -> case readMaybe n of
-      putStrLn $ name ++ " <dictionary>"
+      Nothing     -> wrongArgs
-      die "Error: No dictionary given."
+      Just number -> loadAndRun number files
-    files -> loadAndRun files