#!/usr/bin/python

import string
from collections import defaultdict

import dictionary


def analyze_dictionary(wordle_dictionary):
    # Word Position, Letter: Count
    position_matches = [
        defaultdict(int),
        defaultdict(int),
        defaultdict(int),
        defaultdict(int),
        defaultdict(int),
    ]

    # Letter: Count
    word_matches = defaultdict(int)

    for word in wordle_dictionary:
        unique_letters = set(word)
        for letter in unique_letters:
            word_matches[letter] += 1

        for index in range(len(word)):
            letter = word[index]
            assert letter in string.ascii_lowercase
            position_matches[index][letter] += 1

    return position_matches, word_matches


def score_word(position_matches, word_matches, word):
    result = 0

    # Green Score
    for index in range(len(word)):
        letter = word[index]
        green_score = position_matches[index][letter]
        result += 10 * green_score

    # Yellow Score
    for letter in set(word):
        yellow_score = word_matches[letter]
        result += yellow_score

    return result


def scored_words(wordle_dictionary=dictionary.words()):
    # word: score
    scored_words = dict()

    position_matches, word_matches = analyze_dictionary(wordle_dictionary)

    for word in wordle_dictionary:
        scored_words[word] = score_word(position_matches, word_matches, word)

    result = list(wordle_dictionary)
    result.sort(key=lambda word: scored_words[word], reverse=True)

    return [(word, scored_words[word]) for word in result]


if __name__ == "__main__":
    consts = constraints.Constraints()
    consts.update_graded(
        "cares",
        [
            solver.LetterResult.INCORRECT,
            solver.LetterResult.INCORRECT,
            solver.LetterResult.INCORRECT,
            solver.LetterResult.INCORRECT,
            solver.LetterResult.INCORRECT,
        ],
    )

    valid_words = consts.filter_words(dictionary.words())

    print("Top: ", solver.scored_words(valid_words)[:5])