zwift-offline/discord_bot.py

import asyncio
import aiohttp
import os
import threading

from configparser import ConfigParser

import discord

import zwift_offline


class DiscordBot(discord.Client):
    def __init__(self, intents, channel, welcome_msg, help_msg, announce):
        discord.Client.__init__(self, intents=intents)
        self.channel_id = channel
        self.welcome_msg = welcome_msg
        self.help_msg = help_msg
        self.announce = announce

    async def on_ready(self):
        self.channel = self.get_channel(self.channel_id)

    async def on_member_join(self, member):
        if self.welcome_msg:
            await self.channel.send('%s\n%s' % (member.mention, self.welcome_msg))

    async def on_message(self, message):
        if message.author.id == self.user.id:
            return
        if message.content == '!online':
            await message.channel.send('%s riders online' % len(zwift_offline.online))
        elif message.content == '!help' and self.help_msg:
            await message.channel.send(self.help_msg)
        elif message.content == '!ping':
            await message.channel.send('pong')
        elif message.channel == self.channel and not message.author.bot and not message.content.startswith('!'):
            zwift_offline.send_message(message.content, message.author.name)


class DiscordThread(threading.Thread):
    def __init__(self, config_file):
        threading.Thread.__init__(self)
        if not os.path.isfile(config_file):
            raise Exception("DiscordThread invoked without a configuration file")

        self.CONFIG = ConfigParser()
        SECTION = 'discord'
        self.CONFIG.read(config_file)
        self.token = self.CONFIG.get(SECTION, 'token')
        self.webhook = self.CONFIG.get(SECTION, 'webhook')
        self.channel = self.CONFIG.getint(SECTION, 'channel')
        self.welcome_msg = self.CONFIG.get(SECTION, 'welcome_message', fallback='')
        self.help_msg = self.CONFIG.get(SECTION, 'help_message', fallback='')
        self.announce = self.CONFIG.getboolean(SECTION, 'announce_players', fallback=False)

        self.intents = discord.Intents.default()
        self.intents.members = True
        self.intents.message_content = True
        self.loop = asyncio.get_event_loop()
        self.start()

    async def starter(self):
        self.discord_bot = DiscordBot(self.intents, self.channel, self.welcome_msg, self.help_msg, self.announce)
        await self.discord_bot.start(self.token)

    def run(self):
        try:
            self.loop.run_until_complete(self.starter())
        except Exception as exc:
            print('DiscordThread exception: %s' % repr(exc))

    async def webhook_send(self, message, sender):
        async with aiohttp.ClientSession() as session:
            await discord.Webhook.from_url(self.webhook, session=session).send(message, username=sender)

    def send_message(self, message, sender_id=None):
        if sender_id is not None:
            profile = zwift_offline.get_partial_profile(sender_id)
            sender = profile.first_name + ' ' + profile.last_name
        else:
            sender = 'Server'
        asyncio.run_coroutine_threadsafe(self.webhook_send(message, sender), self.loop)

    def change_presence(self, n):
        if n > 0:
            activity = discord.Game(name=f"{n} rider{'s'[:n>1]} online")
        else:
            activity = None
        asyncio.run_coroutine_threadsafe(self.discord_bot.change_presence(activity=activity), self.loop)