import os

import discord
from dotenv import load_dotenv

load_dotenv()

from honcho import Client as HonchoClient
from honcho import LRUCache

intents = discord.Intents.default()
intents.messages = True
intents.message_content = True

CACHE = LRUCache(50) # Support 50 concurrent active conversations cached in memory 

honcho = HonchoClient("http://localhost:8000")

bot = discord.Bot(intents=intents)

def get_or_create(user_id, location_id) -> int:
    key = f"{user_id}+{location_id}"
    session_id = CACHE.get(key)
    if session_id is None:
        session = honcho.create_session(user_id, location_id)
        print(session)
        session_id = session["id"]
        CACHE.put(key, session_id)
    return session_id

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

@bot.event
async def on_message(message):
    if message.author == bot.user:
        return

    user_id = f"discord_{str(message.author.id)}"
    location_id=str(message.channel.id)
    # key = f"{user_id}+{location_id}"

    session_id = get_or_create(user_id, location_id)

    inp = message.content
    honcho.create_message_for_session(user_id, session_id, True, inp)
    async with message.channel.typing():
        await message.channel.send("Fake LLM Message")

    honcho.create_message_for_session(user_id, session_id, False, "Fake LLM Message")

@bot.slash_command(name = "restart", description = "Restart the Conversation")
async def restart(ctx):
    user_id=f"discord_{str(ctx.author.id)}"
    location_id=str(ctx.channel_id)
    key = f"{user_id}+{location_id}"
    session_id = CACHE.get(key)
    if session_id is not None:
        honcho.delete_session(user_id, session_id)
    session = honcho.create_session(user_id, location_id)
    session_id = session["id"]
    CACHE.put(key, session_id)

    msg = "Great! The conversation has been restarted. What would you like to talk about?"
    honcho.create_message_for_session(user_id, session_id, False, msg)
    await ctx.respond(msg)

bot.run(os.environ["BOT_TOKEN"])