import discord, io, traceback, json, os, asyncio

from colorama import Fore, Style, init

init(autoreset=True)

from core.database import Servers
from core.files import Data
from core.embeds import Embeds

commands = discord.ext.commands

settings = Data("settings").json_read()

class Bumps(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.config = Data("config").yaml_read()

    @commands.guild_only()
    @commands.cooldown(1, settings['cooldown'], commands.BucketType.guild)
    @commands.command()
    async def bump(self, ctx):
        server = Servers(ctx.guild.id)
        guild = ctx.guild
        prefix = Servers(guild.id).getPrefix() if Servers(guild.id).hasPrefix else self.config["prefix"]
        
        if not server.get():
            ctx.command.reset_cooldown(ctx)
            return await ctx.send(embed=Embeds(f"You must setup this server first! Use `{prefix}setup` to do so!").error())

        wait_msg = await ctx.send(embed=discord.Embed(
            title="⏳ Bumping...",
            description="Please wait while we're sending out the bumps!",
            color=discord.Color.orange()
        ))

        server.update(icon_url=str(ctx.guild.icon_url_as(static_format="png")), server_name=ctx.guild.name)

        servers = Servers().get_all()

        success, fail = 0, 0

        invite = await invite_channel.create_invite(max_uses=0, max_age=0, unique=False)

        embed = discord.Embed(
            title=guild.name,
            description=server.get()['description'],
            color=discord.Color(value=server.get()['color']),
            url=invite.url
        )

        embed.add_field(name="🌍 Members", value=len(guild.members))
        embed.add_field(name="🤣 Emojis", value=f"{len(guild.emojis)}/{guild.emoji_limit}")
        embed.add_field(name="💎 Boost Tier", value=f"Tier {guild.premium_tier} ({guild.premium_subscription_count} Boosts)")
        embed.add_field(name="👑 Owner", value=str(guild.owner))
        embed.add_field(name="🔗 Invite", value=f"[Click to join!]({invite.url})")
        embed.set_thumbnail(url=guild.icon_url_as(static_format="png"))
        embed.set_footer(text=f"Powered by • {self.config['bot_name']}")

        for entry in servers:
            try:
                webhook = await self.bot.fetch_webhook(entry['webhook'])
                invite_channel = self.bot.get_channel(entry['invite'])

                await webhook.send(
                    username=self.config['bot_name'],
                    avatar_url=self.bot.user.avatar_url,
                    embed=embed
                )

                success += 1
            except Exception as e:
                error = f"{e}"
                value = io.StringIO().getvalue()
                print(f"{Fore.RED}[ERROR] {Style.RESET_ALL}{error}\n{Fore.YELLOW}Error was recorded in {Fore.RED}error.log")
                with open("error.log", "w+") as f:
                    f.write(f"{value}{traceback.format_exc()}")


                with open("cache_data.json", "w+") as f:
                    json.dump(entry, f, indent=4)

                Servers(entry['_id']).delete()

                try:
                    await self.bot.get_guild(entry['_id']).owner.send(embed=discord.Embed(
                        title="⚠️ Server Removed ⚠️",
                        description="Your server was removed from the database because it caused an error! Make sure I have permission to `Manage Webhooks` and `Create Instant Invites`! I've attached your server info below.",
                        color=discord.Color.red()
                    ))

                except: pass

                fail += 1

                os.remove("cache_data.json")
        await wait_msg.delete()            

        done_message = await ctx.send(embed=discord.Embed(
            title="⏫ Server Bumped",
            description=f"Your server was bumped to `{success+fail}` servers!\n✅ There were `{success}` successful bumps!\n❎ There were `{fail}` failed ones, they got booted from the Database!",
            color=discord.Color.green()
        )
        .set_footer(text=f"Powered by • {self.config['bot_name']}"))

        if settings["show_motd"]:
            await asyncio.sleep(settings["show_motd_wait"])
            return await done_message.edit(embed=discord.Embed(
                title="🗞️ Message Of The Day 🗞️",
                description=Data("motd").read(),
                color=discord.Color.green()
            ))
        else:
            return

def setup(bot):
    bot.add_cog(Bumps(bot))