"""
File contains logic for creating a webhook listener, pulling from git, and updating all messages.
"""

import os

import nextcord
import uvicorn
from fastapi import FastAPI
from git import Repo
from nextcord.ext import commands

from config import DakinyConfig
from utils import REPO_DIR, update_clan_messages

app = FastAPI()
REPO_URL = "https://gitea.cheezers.org/imcompiling/dakiny-messages.git"


class ClanMessages(commands.Cog, name="Clan Messages"):
    """
    Cog to be loaded into the bot. Used to be able to have a fastapi post listener and access to the bot at
    the same time.
    """

    def __init__(self, bot: commands.Bot):
        self.bot = bot
        self.config: DakinyConfig = self.bot.config
        self.repo_setup()

    @commands.Cog.listener()
    async def on_ready(self):
        """
        Triggers when the bot is in ready state. Adds the webhook listener to FastAPI and serves.
        """
        await self.update_messages()
        app.add_api_route("/messages/", self.update_messages, methods=["POST"])
        config = uvicorn.Config(app=app, port=5000, log_level="info")
        server = uvicorn.Server(config)
        await server.serve()

    async def update_messages(self):
        """
        Pulls from git and then calls method to update all the discord messages as described by the
        files in the git repo.
        """

        self.origin.pull()
        guild: nextcord.Guild = self.bot.get_guild(self.config.params.guild_id)

        await update_clan_messages(guild)

        return {"message": "Successfully updated all discord messages."}

    def repo_setup(self):
        """
        Checks if the repo exists and creates repo object, clones if it does not.
        """
        # check if repo has been cloned. open repo object
        if os.path.isdir(REPO_DIR):
            self.repo = Repo(REPO_DIR)
        else:
            self.repo = Repo.clone_from(REPO_URL, REPO_DIR)

        self.origin = self.repo.remote("origin")

        if self.config.params.git_branch == "test":
            self.repo.create_head("test", self.origin.refs.test).set_tracking_branch(
                self.origin.refs.test
            ).checkout()
        else:
            self.repo.create_head("main", self.origin.refs.main).set_tracking_branch(
                self.origin.refs.main
            ).checkout()

        self.origin.pull()


# This function will be called when this extension is loaded.
# It is necessary to add these functions to the bot.
def setup(bot: commands.Bot):
    """Adds function to bot"""
    bot.add_cog(ClanMessages(bot))