import itertools

def wordlist_generator(base_words, years, chars):
    """
    Gera combinações de senhas baseadas em palavras-chave, anos e caracteres.
    """
    print(f"[*] Gerando wordlist baseada em {len(base_words)} termos...")

    # Dicionário simples de Leet Speak
    leet_map = {'a': '4', 'e': '3', 'i': '1', 'o': '0', 's': '5', 't': '7'}

    results = set()

    for word in base_words:
        # 1. Adiciona a palavra original, maiúscula e capitalizada
        variations = [word, word.upper(), word.capitalize()]

        # 2. Gera uma versão Leet Speak básica
        leet_word = "".join([leet_map.get(c.lower(), c) for c in word])
        variations.append(leet_word)

        for v in variations:
            results.add(v)
            # 3. Combina com anos
            for year in years:
                results.add(f"{v}{year}")
                results.add(f"{v}_{year}")

            # 4. Combina com caracteres especiais
            for char in chars:
                results.add(f"{v}{char}")
                results.add(f"{v}{year}{char}")

    return sorted(list(results))

def main():
    print("=" * 60)
    print("Módulo: Wordlist Generator | Projeto: 30 Lessons in Python")
    print("=" * 60)

    # Dados coletados na fase de Reconhecimento (OSINT)
    termos = ["Salvador", "Bahia", "Hunter", "Mystique"]
    anos = ["2025", "2026"]
    especiais = ["!", "@", "#", "$"]

    wordlist = wordlist_generator(termos, anos, especiais)

    # Salva em arquivo para uso em outras ferramentas (Hydra, Hashcat, etc)
    with open("custom_wordlist.txt", "w") as f:
        for password in wordlist:
            f.write(f"{password}\n")

    print(f"\n[+] Wordlist gerada com sucesso!")
    print(f"[+] Total de combinações: {len(wordlist)}")
    print(f"[+] Arquivo salvo como: custom_wordlist.txt")
    print(f"[*] Exemplos gerados: {wordlist[:5]} ... {wordlist[-5:]}")

if __name__ == "__main__":
    main()