#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json
import argparse


def load_dict(dict_file):
    try :
        with open(dict_file, "r") as dico:
            liste = json.load(dico)
        return liste
    except:
        print("Impossible d'ouvrir le fichier" + fp)
        exit(2)


def noDouble(mots):
    # Fonction d'élimination de mots contenant plusieurs occurences de lettres
    ndList = []
    for mot in mots:
        llist = []
        for l in mot:
            if not l in llist:
                llist.append(l)
        if len(mot) == len(llist):
            ndList.append(mot)
    return(ndList)

def firstTryFilter(mots, max_mots):
    '''
    First Try : sélectionne des mots ayant des chances de valider ou invalider
    l'une des lettres les plus utilisée en français
    '''
    bestLetters = ['E', 'A', 'I', 'R', 'S', 'N', 'T', 'O', 'L', 'U']
    bestWords = []
    for mot in mots:
        for bestLetter in bestLetters:
            if bestLetter in mot:
                bestWords.append(mot)
            bestWords = list(set(bestWords))
    bestWords = noDouble(bestWords)  # Autant maximiser les chances
    bestWords = bestWords[0:max_mots]
    return(bestWords)

# Fonction de comparaison des lettres d'un mot, avec celle de la pattern
# Plus nécessaire actuellement mais peut être dans une prochaine vie
'''
def patternCheck(pattern, mot, l):
    goodPosCounter = 0
    for pos in range(len(pattern)):
        print("Checking : " + l + "against : " + pattern[pos] + " and " + mot[pos])
        if pattern[pos] == l and mot[pos] == l:
            goodPosCounter +=1
            print("checked")
    if goodPosCounter == pattern.count(l):
        return(0)
    else:
        return(1)
'''

def resolv_fl(liste, firstletter):
    # On ne garde que les mots qui commence par FirstLetter
    return [mot for mot in liste if mot[0] == firstletter]


def resolv_len(liste, lenght):
    # On ne garde que les mots qui ont la bonne longueur
    return [mot for mot in liste if len(mot) == lenght]


# Pattern : élimination des mots ne satisfaisant pas la pattern
#if pattern:
#    patternCount = len(pattern) - pattern.count(".")
#    for mot in goodLen:
#        patternEval = 0
#        for l in range(len(mot)):
#            if mot[l] == pattern[l]:
#                patternEval += 1
#        if patternEval == patternCount :   
#            patternFilter.append(mot)
#
# Known Letters : élimination des mots ne comprenant pas les lettres validées (hors pattern)
#if kl:
#    if patternFilter:
#        mots = patternFilter
#    else:
#        mots = goodLen
#    for mot in mots:
#        validate = 0
#        for l in kl:
#            if l in mot:
#                validate += 1
#        if validate == len(kl):
#            klFilter.append(mot)

# Bad Letters : élimination des mots contenant des lettres invalidées (hors pattern)
#if bl:
#    if klFilter:
#        mots = klFilter
#    elif patternFilter:
#        mots = patternFilter
#    else:
#        mots = goodLen
#    for mot in mots:
#        invalidate = 0
#        for l in bl:
#            if l in mot and l in pattern:
#                if pattern.count(l) != mot.count(l):
#                    invalidate += 1
#            elif l in mot:
#                invalidate += 1
#        if invalidate == 0:
#            blFilter.append(mot)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description='Cheating with SUTOM'
    )
    parser.add_argument(
        '-d',
        '--dictionary',
        help='Dictionnaire des mots du jeu',
        required=True
    )
    parser.add_argument(
        '-fl',
        '--firstLetter',
        help='Première lettre',
        default=False,
        required=True
    )
    parser.add_argument(
        '-nb',
        '--nbLetters',
        type=int,
        help='Nombre de lettres, un entier',
        default=False,
        required=True
    )
    parser.add_argument(
        '-kl',
        '--knownLetters',
        default=False,
        help='Lettres connues, hors première, sans espaces : -kl BE'
    )
    parser.add_argument(
        '-bl',
        '--badLetters',
        default=False,
        help='Lettres non valides, sans espaces : -bl AKL'
    )
    parser.add_argument(
        '-p',
        '--pattern',
        default=False,
        help='Placement, avec des "." pour les inconnues : -p A...T.I.'
    )
    parser.add_argument(
        '-nd',
        '--noDoubleLetters',
        help="Option pour ne chercher que des mots ne contenant qu'une occurence de chaque lettre",
        default=False,
        action="store_true"
    )
    parser.add_argument(
        '-ft',
        '--firstTry',
        default=False,
        type=int,
        help="Sélectionne des mots commençants par firstLetter et composés de ft voyelles différentes: -ft 4",
    )

    args = parser.parse_args()
            
    if args.pattern and args.nbLetters and len(pattern) != nb:
        print("Tu as merdé ta pattern petit scarabé !")
        exit(2)
    
    # Known Letters
    kl = args.knownLetters
    
    # Bad Letters
    bl = args.badLetters
    
    liste = load_dict(args.dictionary)
    liste = resolv_len(liste, args.nbLetters)
    liste = resolv_fl(liste, args.firstLetter)
    if args.noDoubleLetters:
        liste = noDouble(liste)
    if args.firstTry:
        liste = firstTryFilter(liste, args.firstTry)
        if len(liste) == 1:
            print(f"Essaie : \"{liste[0]}\"")
        elif len(liste) > 1:
            print(f"Mots à tester : {liste}")
        else:
            print("Hmmm... Je crois qu'on s'est perdu...")
        exit(0)
    if len(liste) > 1:
        print(liste)
        print(f"Il y a {len(liste)} mots dans la liste")
    elif len(liste) == 1:
        print(f"La solution est \"{liste[0]}\".")
    else:
        print("Hmmm... Je crois qu'on s'est perdu...")