import os

from discord.ext import commands
from dotenv import load_dotenv

import random

max_len = 6

with open('text-faust.txt', 'r', encoding='utf-8') as f:
    training_text = f.read()

def train(text, max_len=5):
    model = dict()
    for i in range(1, max_len+1):
        model[i] = dict()
        for k in range(len(text) - i - 1):
            if not text[k:k+i] in model[i]:
                 model[i][text[k:k+i]] = dict()
            if text[k+i] in model[i][text[k:k+i]]:
                model[i][text[k:k+i]][text[k+i]] += 1
            else:
                model[i][text[k:k+i]][text[k+i]] = 1
    return model

def query(model, previous):
    for l in range(len(previous), 0, -1):
        if previous[-l:] not in model[l]:
            continue
        freqs = model[l][previous[-l:]]
        return random.choices(list(freqs.keys()), list(freqs.values()))[0]
        #return random.choices(list(freqs.keys()))[0]

model = train(training_text, max_len=max_len)

load_dotenv()
TOKEN = os.getenv('DISCORD_TOKEN')

bot = commands.Bot(command_prefix='-')

@bot.event
async def on_ready():
    print('We have logged in as {0.user}'.format(bot))

@bot.command(name='isthis')
async def isthis_command(ctx, person, schmetterling, frage):
    await ctx.send(f"https://api.memegen.link/images/pigeon/{person}/{schmetterling}/{frage}.png")

@bot.command(name='pictures')
async def pictures_command(ctx, a, auch_a, person):
    await ctx.send(f"https://api.memegen.link/images/same/{a}/{auch_a}/{person}.png")

@bot.command(name='drake')
async def drake_command(ctx, no, yes):
    await ctx.send(f"https://api.memegen.link/images/drake/{no}/{yes}.jpg")

@bot.command(name='obiwan')
async def obiwan_command(ctx, zitat):
    await ctx.send(f"https://api.memegen.link/images/chosen/{zitat}.png")

@bot.command(name='picard')
async def picard_command(ctx, oben, unten):
    await ctx.send(f"https://api.memegen.link/images/facepalm/{oben}/{unten}.png")

@bot.command(name='umdreh')
async def umdreh_command(ctx, yay, kerl, nay):
    await ctx.send(f"https://api.memegen.link/images/db/{yay}/{kerl}/{nay}.png")

@bot.command(name='changemind')
async def changemind_command(ctx, these):
    await ctx.send(f"https://api.memegen.link/images/cmm/{these}.png")

@bot.command(name='boromir')
async def boromir_command(ctx, notsimply, mordor):
    await ctx.send(f"https://api.memegen.link/images/mordor/{notsimply}/{mordor}.png")

@bot.command(name='brains')
async def brains_command(ctx, eins, zwei, drei, pfier):
    await ctx.send(f"https://api.memegen.link/images/gb/{eins}/{zwei}/{drei}/{pfier}.png")

@bot.command(name='summonbeleidiger')
async def test(ctx):
    await ctx.send("+beleidiger")

@bot.command(name='faust')
async def test(ctx, current_text):
    n_lines = 500
    for _ in range(n_lines):
        current_text += query(model, current_text[-6:])

    await ctx.send(current_text)

@bot.command(name='faustmeme')
async def test(ctx, current_text):
    current_text = current_text.replace('-', ' ')
    n_lines = 100
    for _ in range(n_lines):
        current_text += query(model, current_text[-6:])

    current_text = current_text.replace('-', '--').replace('\n', '-').replace(' ', '-')
    await ctx.send(f"https://api.memegen.link/images/chosen/{current_text}.png""")

bot.run(TOKEN)