import discord
import os
import argparse
from pathlib import Path

parser = argparse.ArgumentParser(
    prog="Scrapist",
    description="Simple emoji extractor for Discord."
)
parser.add_argument('-t', '--discord_token', help="user or bot token that has access to the Discord server you'll be exporting emoji from. can also be provided via the `$DISCORD_TOKEN` env variable")
parser.add_argument('-g', '--guild_id', help="ID of the Discord server you wish to extract emoji from", required=True)
parser.add_argument('-o', '--out_dir', help="path where emojis will be downloaded to", required=True)
args = parser.parse_args()

discord_token = os.getenv("DISCORD_TOKEN")
if args.discord_token is not None:
    discord_token = args.discord_token
if discord_token is None:
    exit("No Discord token provided")

guild_id = args.guild_id
out_dir = args.out_dir

class Scrapist(discord.Client):
    async def on_ready(self):
        print('Logged in as', self.user)

        guild = self.get_guild(int(guild_id))
        guild_name = guild.name

        # Create output directories if they don't exist
        Path(f'{out_dir}/{guild_name}/emoji').mkdir(parents=True, exist_ok=True)
        Path(f'{out_dir}/{guild_name}/animated').mkdir(parents=True, exist_ok=True)
        Path(f'{out_dir}/{guild_name}/sticker').mkdir(parents=True, exist_ok=True)
                
        emotes = await guild.fetch_emojis()
        for emote in emotes:
            if emote.animated:
                filename = f"{emote.name}.webp"
                await emote.save(f'{out_dir}/{guild_name}/animated/{filename}')
            else:
                filename = f"{emote.name}.png"
                await emote.save(f'{out_dir}/{guild_name}/emoji/{filename}')
            print(f"downloaded {filename}")
        stickers = await guild.fetch_stickers()
        for sticker in stickers:
            filename = f"{sticker.name}.png"
            await sticker.save(f'{out_dir}/{guild_name}/sticker/{filename}')
            print(f"downloaded {sticker.name}")
        
        await self.close()

client = Scrapist()
client.run(discord_token)