import vocab
import random

"""
Play the game like this:
    > python
    >>> import wordle_v2
    >>> wordle_v2.process_guess("chant")
    C__N_
    >>> wordle_v2.process_guess("store")
    _____
    >>> wordle_v2.process_guess("claim")
    CL_i_
    etc...
"""

random_index = random.randrange(0, len(vocab.words))
hidden = vocab.words[random_index]

def process_letter(char, i):
    """
    If the guess is in the right place, return upper case letter
    If the guess is not in the right place, return lower case letter
    If the guess isn't in the word at all, return "_"
    """
    if (hidden[i] == char):
        return char.upper()
    elif (hidden.count(char) > 0): # letter is in word, just not at this index
        return char.lower()
    else:
        return "_"

def process_guess(guess):
    if len(guess) != 5:
        print("your guess must be 5 letters long")
    elif not guess.isalpha():
        print("your guess must be all letters!")
    elif vocab.words.count(guess) == 0:
        print("your guess must be a valid word!")
    elif guess == hidden:
        print("YOU WON!!!!")
    else:
        feedback = ""
        feedback = feedback + process_letter(guess[0], 0)
        feedback = feedback + process_letter(guess[1], 1)
        feedback = feedback + process_letter(guess[2], 2)
        feedback = feedback + process_letter(guess[3], 3)
        feedback = feedback + process_letter(guess[4], 4)
        print(feedback)