import asyncio
import contextlib
import logging.config
from argparse import ArgumentParser
from logging import Logger
from os import getpid, makedirs
from pathlib import Path

# Import required for uvicorn
from libbot.utils import config_get

from classes.pycord_bot import PycordBot
from modules.logging_utils import get_logging_config, get_logger
from modules.migrator import migrate_database
from modules.scheduler import scheduler

makedirs(Path("logs/"), exist_ok=True)

logging.config.dictConfig(get_logging_config())

logger: Logger = get_logger(__name__)

# Declare the parser that retrieves the command line arguments
parser: ArgumentParser = ArgumentParser(
    prog="QuizBot"
)

# Add a switch argument --migrate to be parsed...
parser.add_argument("--migrate", action="store_true")
# parser.add_argument("--downgrade", action="store_true")
# parser.add_argument("--confirm", action="store_true")

# ...and parse the arguments we added
args = parser.parse_args()

# Try to import the module that improves performance
# and ignore errors when module is not installed
with contextlib.suppress(ImportError):
    import uvloop

    uvloop.install()


async def main():
    # Perform migration if command line argument was provided
    if args.migrate:
        migrate_database()

    # if args.downgrade:
    #     if not args.confirm:
    #         logger.warning(
    #             "Argument --downgrade has been provided but this operation may include irreversible changes and destroy existing data in both database and configuration. It's not recommended to do this without backing everything up. Retry this command providing an additional argument --confirm to accept the risks and explicitly execute the downgrade."
    #         )
    #         exit(1)
    #
    #     logger.info("Downgrading...")
    #     downgrade_database()
    #     exit()

    bot = PycordBot(scheduler=scheduler)

    bot.load_extension("cogs")

    try:
        await bot.start(config_get("bot_token", "bot"))
    except KeyboardInterrupt:
        logger.warning("Forcefully shutting down with PID %s...", getpid())
        await bot.close()


if __name__ == "__main__":
    asyncio.create_task(main())