main.py

import discord
from discord.ext import commands, tasks
from utils import serversettings as Settings, queues as Queues, update
import os
from config import config 
import asyncio
import json
import sys
import nacl
import subprocess



# Initialize Discord bot with command prefix and intents
bot = commands.Bot(command_prefix=commands.when_mentioned_or(""), intents=discord.Intents.all())
tree = bot.tree
is_windows = os.name == 'nt'

### LOAD COGS ###
async def load_cogs():
    cogs = config.get_cogs()
    for cog in cogs:
        try:
            await bot.load_extension(f"cogs.{cog}")
            print(f"[+] Successfully loaded {cog}")
        except Exception as e:
            print(f"[-] An error occurred while loading {cog}: {e}")

### CHECK FOR UPDATES EVERY HOUR ###
async def update_msg():
        if len(sys.argv) > 1:
            if sys.argv[1] == 'updated':
                try:
                    embed = discord.Embed(title="Update complete!", description="Your bot has been updated to the latest version.", color=discord.Color.blue())
                    app_info = await bot.application_info()
                    owner = app_info.owner
                    await owner.send(embed=embed)
                except Exception as e:
                    print(f"[-] An error occurred while sending the update message: {e}")   

@tasks.loop(hours=1)
async def check_for_updates(): 
    
    updates = update.check_upd(is_windows)
    # If updates are available, send a message to the bot owner
    if updates:
        app_info = await bot.application_info()
        owner = app_info.owner
        embed = discord.Embed(title="Updates are available", description=updates, color=discord.Color.green())
        message = await owner.send(embed=embed)
        # add reaction to the message to allow the owner to update the bot
        await message.add_reaction('✅')
        await message.add_reaction('❌')
        try:
            reaction, _ = await bot.wait_for('reaction_add', check=lambda reaction, user: user == owner and reaction.message == message)
            if reaction.emoji == '✅':
                update.update(is_windows)
                await bot.close()
                sys.exit(0)
            elif reaction.emoji == '❌':
                await owner.send('[-] Update declined. Bot will not be updated.')
                await asyncio.sleep(24*3600.0)
        except asyncio.CancelledError:
            pass
        except asyncio.TimeoutError:
            await owner.send('[-] Update declined. Bot will not be updated.')            
            await asyncio.sleep(24*3600.0)
        except Exception as e:     
            pass       

### ON READY EVENT ###
@bot.event
async def on_ready():
    # Load cogs
    await load_cogs()
    await update_msg()
    print(f'[+] Booted {bot.user}...')
    await bot.change_presence(activity=discord.Activity(type=discord.ActivityType.listening, name="/play <song>"), status=discord.Status.dnd)
    # Reset queues and fetch settings for all guilds
    queues = {}
    settings = Settings.get_settings_all()

    # Initialize settings for all guilds
    try:
        await Settings.check_guilds(bot.guilds)
        print('[+] Successfully initialized bot settings')
    except Exception as e:
        print('[!] Error initializing bot settings: ', e)

    # Initialize queues for all guilds
    try:
        for guild in bot.guilds:
            queues[str(guild.id)] = []
        with open(config.queues, 'w') as f:
            json.dump(queues, f, indent=4)
            print('[+] Successfully initialized queues')
    except Exception as e:
        print('[!] Error initializing queues: ', e)

    # if playlists are enabled, initialize playlists for all guilds
    playlist_cog = bot.get_cog('Playlist')
    if playlist_cog is not None:
        print('[+] Playlists cog found, initializing playlists')

        try:
            playlist_cog.initialize_playlists(bot.guilds)
            print('[+] Successfully initialized playlists')
        except Exception as e:
            print('[!] Error initializing playlists: ', e)

    # Sync the tree for all guilds
    for guild in bot.guilds:
        try:
            await tree.sync(guild=guild)
            print(f'[+] Successfully synced guild {guild.name}')
        except Exception as e:
            print(f'[!] Error syncing guild {guild.name}: {e}')

    # Print URL for inviting the bot to a server
    oauth_url = discord.utils.oauth_url(bot.application_id, permissions=discord.Permissions(permissions=8))
    print(f'[+] Invite URL: {oauth_url}')
    
    await check_for_updates.start()


### INITIALIZE GUILD SETTINGS AND QUEUES ON GUILD JOIN ###
@bot.event
async def on_guild_join(guild):
    Settings.set_guild_settings(guild.id, Settings.default_settings)
    Queues.update_queue(guild.id, [])
    print(f'[+] Joined {guild.name} with id {guild.id}')
    print(f'[+] Successfully initialized config/serversettings.json for {guild.name}')
    print(f'[+] Successfully initialized config/queues.json for {guild.name}')

### HANDLE ON MESSAGE EVENT ###
@bot.event
async def on_message(message):
    if message.guild is not None:
        prefix = Settings.get_settings(message.guild.id)['prefix']
        if message.content.startswith(prefix):
            message.content = message.content[len(prefix):]
            await bot.process_commands(message)
    else:
        await bot.process_commands(message)

### HANDLE ON COMMAND ERROR EVENT ###
@bot.event
async def on_command_error(ctx, error):
    if isinstance(error, commands.MissingPermissions):
        await ctx.send(f":x: You need to have `{', '.join(error.missing_permissions)}` permissions to use this command.", ephemeral=True)
        return
    if isinstance(error, commands.CommandNotFound):
        return
    if isinstance(error, commands.CheckFailure):
        await ctx.send(":x: You do not have permission to use this command.")
        return
    if isinstance(error, commands.MissingRequiredArgument):
        await ctx.send(":x: Missing required argument.")
        return
    if isinstance(error, commands.BadArgument):
        await ctx.send(":x: Invalid argument.")
        return
    if isinstance(error, commands.CommandOnCooldown):
        await ctx.send(f":x: Command on cooldown. Try again in {error.retry_after:.2f} seconds.")
        return
    if isinstance(error, commands.CommandInvokeError):
        await ctx.send(f":x: An error occurred while executing the command.: {error}")
        return

    raise error


def main(*args):

    # Check for updates
    if args:
        if args[0] == 'updated':
            print("[+] Successfully updated to the latest version!")
    else:
        if update.check_upd(is_windows): 
            update.update(is_windows)
            sys.exit(0)

    # Initialize the bot        
    try:
        config.init()
        bot.run(config.bot_token)
    except Exception as e:
        print(f"[-] An error occurred while running the bot: {e}")
        return


if __name__ == '__main__':
    args = sys.argv[1:]
    main(*args)