# bots.py
import os
import json
import asyncio
import sys
import time
import traceback
import globals
from functools import partial
import twitchio.ext

from discord.ext import commands
from dotenv import load_dotenv

from globals import logger

from bot_discord import DiscordBot
from bot_twitch import TwitchBot

#from modules.db import init_db_connection, run_db_operation
#from modules.db import ensure_quotes_table, ensure_users_table, ensure_chatlog_table, checkenable_db_fk

from modules import db, utility

# Load environment variables
load_dotenv()

# Clear previous current-run logfile
logger.reset_curlogfile()

# Load bot configuration
config_data = globals.Constants.config_data

###############################
# Main Event Loop
###############################

async def main():
    global discord_bot, twitch_bot, db_conn

    # Log initial start
    logger.info("--------------- BOT STARTUP ---------------")
    # Before creating your DiscordBot/TwitchBot, initialize DB
    db_conn = globals.init_db_conn()

    try: # Ensure FKs are enabled
        db.checkenable_db_fk(db_conn)
    except Exception as e:
        logger.warning(f"Unable to ensure Foreign keys are enabled: {e}")

    # auto-create the quotes table if it doesn't exist
    tables = {
        "Bot events table": partial(db.ensure_bot_events_table, db_conn),
        "Quotes table": partial(db.ensure_quotes_table, db_conn),
        "Users table": partial(db.ensure_users_table, db_conn),
        "Platform_Mapping table": partial(db.ensure_platform_mapping_table, db_conn),
        "Chatlog table": partial(db.ensure_chatlog_table, db_conn),
        "Howls table": partial(db.ensure_userhowls_table, db_conn),
        "Discord activity table": partial(db.ensure_discord_activity_table, db_conn),
        "Account linking table": partial(db.ensure_link_codes_table, db_conn),
        "Community events table": partial(db.ensure_community_events_table, db_conn)
    }

    try:
        for table, func in tables.items():
            func()  # Call the function with db_conn and log already provided
            logger.debug(f"{table} ensured.")
    except Exception as e:
        logger.fatal(f"Unable to ensure DB tables exist: {e}")

    logger.info("Initializing bots...")

    # Create both bots
    discord_bot = DiscordBot()
    twitch_bot = TwitchBot()

    # Log startup
    utility.log_bot_startup(db_conn)

    # Provide DB connection to both bots
    try:
        discord_bot.set_db_connection(db_conn)
        twitch_bot.set_db_connection(db_conn)
        logger.info(f"Initialized database connection to both bots")
    except Exception as e:
        logger.fatal(f"Unable to initialize database connection to one or both bots: {e}")

    logger.info("Starting Discord and Twitch bots...")

    discord_task = asyncio.create_task(discord_bot.run(os.getenv("DISCORD_BOT_TOKEN")))
    twitch_task = asyncio.create_task(twitch_bot.run())

    from modules.utility import dev_func
    enable_dev_func = False
    if enable_dev_func:
        dev_func_result = dev_func(db_conn, enable_dev_func)
        logger.debug(f"dev_func output: {dev_func_result}")

    await asyncio.gather(discord_task, twitch_task)
    #await asyncio.gather(discord_task)

if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        utility.log_bot_shutdown(db_conn, intent="User Shutdown")
    except Exception as e:
        error_trace = traceback.format_exc()
        logger.fatal(f"Fatal Error: {e}\n{error_trace}")
        utility.log_bot_shutdown(db_conn)