game-discord-bot/app.py

#!/usr/bin/env python3
from config import config
from lib import *

import discord
import logging
logger = logging.getLogger('discord.gamebot')
logger.setLevel(logging.INFO)

intents = discord.Intents(messages=True, message_content=True)
client = discord.Client(intents=intents)


@client.event
async def on_ready():
    logger.info(f'Logged in as {client.user}')


@client.event
async def on_message(message):
    # Only allow chats in DM or whitelisted channels
    # TODO: FIX
    if not (isinstance(message.channel, discord.channel.DMChannel)) and not (message.channel.id in config['bot']['channels']):
        print(isinstance(message.channel, discord.channel.DMChannel))
        print(message.channel.id)
        return
    # print(message.channel.name)
    # Handle own messages
    if message.author == client.user:
        return

    # Handle whitelist
    if message.author.name not in config['bot']['whitelist']:
        logger.error(
            f"Unauthorized command by {message.author.name}. message: \"{message.content}\"")
        await message.channel.send(f"Sorry {message.author.mention}, you are not authorized to do that.")
        return

    if message.content == f"{client.user.mention} ping":
        logger.info(f"Ping by {message.author.name}")
        await message.channel.send(f"{message.author.mention} 👋")
        return

    # if message.content == f"{client.user.mention} version":
    #    logger.info(f"Version by {message.author.name}")
    #    await message.channel.send(f"{message.author.mention} ")
    #    return

    # Handle @Bot mention
    messagearray = message.content.split()
    if messagearray[0] == client.user.mention:
        messagearray.pop(0)
    else:
        return

    if messagearray[0] == "help":
        logger.info(f"Help by {message.author.name}")
        await message.channel.send(BuildHelpList())
        return
    if messagearray[0] in ["list", "info"]:
        logger.info(f"{messagearray[0]} by {message.author.name}")
        await message.channel.send(BuildGameList())
        return

    if messagearray[0] in ["start", "stop", "restart"] and len(messagearray) >= 2:
        game = GetGameByGameSlug(messagearray[1])
        if not game:
            logger.error(
                f"{messagearray[0]} for invalid Game \"{messagearray[1]}\" by {message.author.name}")
            await message.add_reaction("❌")
            await message.channel.send(f"Invalid game: {messagearray[1]}")
            return
        logger.info(
            f"{messagearray[0]} by {message.author.name}, Game: {messagearray[1]}")
        success = HandleSystemdService(game, messagearray[0])

        # :white_check_mark: :x:
        if success:
            logger.info(
                f"{messagearray[0]} by {message.author.name}, Game: {messagearray[1]} was SUCESSFUL")
            await message.add_reaction("✅")
        else:
            logger.error(
                f"{messagearray[0]} by {message.author.name}, Game: {messagearray[1]} FAILED!")
            await message.add_reaction("❌")
            await message.channel.send(f"Sorry {message.author.mention}, that didn't work. Ping @lauralani")

    if messagearray[0] == "status":
        status = BuildSingleGameStatus(messagearray[1])
        if not status:
            logger.error(
                f"Status for invalid Game {messagearray[1]} by {message.author.name}")
            await message.add_reaction("❌")
            await message.channel.send(f"Invalid game: {messagearray[1]}")
            return
        logger.info(
            f"Status for Game {messagearray[1]} by {message.author.name}")
        await message.channel.send(status)
        return


client.run(config['bot']['token'])