blob: 5a85849eac60d456b94231a6de8f8ced318d629a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
{-# LANGUAGE ExistentialQuantification, RankNTypes, GADTs #-}
module Main where
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
|
