bot.py

import sys

import discord
from commands import command, admin
from game import *
import config
import interface


# ============ Local functions ============


async def init_setup(init_game_list=False):
    """ Log ready message, check server roles/channels setup """
    startup_msg = "=========================BOT STARTUP========================="
    print(startup_msg)

    database_verified = None

    for guild in client.guilds:
        print("Connected to server: ", guild.name, " ServerID: ", guild.id, " Game Categories: ", config.GAME_CATEGORIES)
        for category_name in config.GAME_CATEGORIES:
            if init_game_list is False:
                await admin.create_game_category(guild, client.user, category_name)

            category = discord.utils.get(guild.categories, name=category_name)
            print("category =", category_name, "->", category)
            if not category:
                continue

            # game_list.add_game(category.id,Game(category, interface.ConsoleInterface(category)))
            game_list.add_game(category.id, Game(category, interface.DiscordInterface(category, client)))

            await admin.send_text_to_channel(category, startup_msg, config.LOBBY_CHANNEL)
            await admin.send_text_to_channel(category, startup_msg, config.GAMEPLAY_CHANNEL)

            game = game_list.get_game(category.id)

            valid_database = database_verified if database_verified is not None else await game.database.verify_init()
            database_verified = valid_database
            if not valid_database:
                verify_database_failed_msg = "Verify database failed, please check `GITHUB_GIST_TOKEN` and `GITHUB_GIST_ID_URL` or remove/comment them in .env file"
                print(verify_database_failed_msg)
                await admin.send_text_to_channel(category, verify_database_failed_msg, config.GAMEPLAY_CHANNEL)


async def process_message(discord_client, message):
    if message.content.strip().startswith(config.BOT_PREFIX):
        game = game_list.get_game(message.channel.category.id)
        if game is None:
            # Init game_list
            await init_setup(True)
            game = game_list.get_game(message.channel.category.id)

        await command.process_command(discord_client, game, message)


def verify_ok(_):
    return True


# ============ Test Discord server =======
async def test_bot(game, guild):
    print("\n\n\n=====================================")
    print("------------ Bot testing ------------")
    print(guild.name)
    # Test admin/player commands
    # await command.test_commands(guild)
    # Test game commands
    await game.test_game()
    # await game.test_game()  # Rerun second time

    print("------------ End bot testing ------------")

# ============ Discord server ============
# We need to enable intents to access guild.members list
# details: https://discordpy.readthedocs.io/en/latest/intents.html#member-intent
intents = discord.Intents.all()  # Recent change in discord need set this to all()
intents.members = True
client = discord.Client(intents=intents)
game_list = GameList()


@client.event
async def on_ready():
    await init_setup()
    print("The bot is ready")

    # Uncomment to run test
    # server_id = config.DISCORD_TESTING_SERVER_ID  # Running test on Nhim's server
    # server_id = config.DISCORD_DEPLOY_SERVER_ID  # Running test on DNH ma sói bot's server
    # guild = client.get_guild(server_id)
    # category = discord.utils.get(guild.categories, name=config.GAME_CATEGORIES[0])
    # await test_bot(game_list.get_game(category.id), guild)


@client.event
async def on_message(message):
    if verify_ok(message):
        await process_message(client, message)  # loop through all commands and do action on first command that match


if __name__ == '__main__':
    if not config.DISCORD_TOKEN:
        print("Use must setup DISCORD_TOKEN in .env file")
        sys.exit(1)
    # keep_alive() # Uncomment to keep the bot alive
    client.run(config.DISCORD_TOKEN)