1 files changed, 75 insertions, 1 deletions
diff --git a/src/Main.hs b/src/Main.hs
index 2d51f8d..5a85849 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -1,2 +1,76 @@
+{-# LANGUAGE ExistentialQuantification, RankNTypes, GADTs #-}
+module Main where
 
-main = putStrLn "Hello, World!"
+import Text.Printf
+import System.Environment
+import Data.Set (Set)
+import qualified Data.Set as Set
+import System.IO
+
+import Wordle
+import Greedy
+
+main :: IO ()
+main = do
+  argv <- getArgs
+
+  case argv of
+    (word:_) -> do
+      env <-
+          Environment (length word) .
+          Set.fromList .
+          filter (\w -> length w == length word) .
+          lines <$> readFile "nounlist.txt"
+
+      if word `notElem` wordList env
+        then putStrLn "Word not in wordList"
+        else runMain 6 env word (greedyAI env)
+
+    _ -> putStrLn "Need one argument"
+
+playerAI :: AI
+playerAI = AI () (\_ env -> do
+  putStr $ printf "%s\r" (replicate (wordLength env) '.')
+  hFlush stdout
+  getLine) (\_ _ -> return ())
+
+runMain :: Int -> Environment -> String -> AI -> IO ()
+runMain round env secretWord (AI state doGuess doUpdate) = do
+  if round == 0
+    then gameOver
+    else continue
+
+  where
+
+  gameOver = putStrLn "Gave Over"
+
+  continue = do
+
+    guess <- doGuess state env
+
+    if not (guess `Set.member` wordList env)
+      then do
+        putStrLn "Word not in wordList"
+        runMain round env secretWord (AI state doGuess doUpdate)
+
+      else do
+        let hints = Hints $ zipWith (\g w -> (
+                g,
+                if g == w
+                    then Correct
+                    else if g `elem` secretWord
+                      then Contains
+                      else DoesNotContain)) guess secretWord
+
+        putStrLn $ printHints hints
+        if guess == secretWord
+          then putStrLn "Correct!"
+          else do
+            newState <- doUpdate state hints
+            runMain (round - 1) env secretWord (AI newState doGuess doUpdate)
+
+        where printHints (Hints h) = flip concatMap h $ \(c, h) ->
+                (case h of
+                   Correct -> printf "\x1b[01;7;32m%c\x1b[0m"
+                   Contains -> printf "\x1b[01;7;33m%c\x1b[0m"
+                   _ -> printf "%c") c