main.py

import logging, discord, os
import discord.ext.tasks as tasks
import discord.ext.commands as commands
from dotenv import load_dotenv
from mcstatus import MinecraftServer
from mcstatus.pinger import PingResponse
from typing import Optional, TypeVar

logging.basicConfig(level=logging.INFO)

# Rust style unwrap method that throws if value is None
T = TypeVar("T")
def unwrap(value: Optional[T]) -> T:
    if value is not None:
        return value
    else:
        raise TypeError()

load_dotenv()
DISCORD_TOKEN = unwrap(os.getenv("DISCORD_TOKEN"))
SERVER_HOST = unwrap(os.getenv("SERVER_HOST"))
BOT_PREFIX = unwrap(os.getenv("BOT_PREFIX"))
COLOR_HEX = int(unwrap(os.getenv("COLOR_HEX")), 16) # Make sure to convert to int with base 16 (hex)

server = MinecraftServer.lookup(SERVER_HOST)
state: Optional[PingResponse] = None

bot = commands.Bot(command_prefix = BOT_PREFIX)

@bot.event
async def on_ready():
    logging.info(f"Connected to discord")
    update_state.start()

# Command that sends the current server status as an embed
@bot.command(
    aliases=["STATUS", "s", "S"],
    description="Show server status"
)
async def status(ctx: commands.Context):
    logging.info(f"Received status command from {ctx.author}")
    global state
    embed = discord.Embed(
        title = "Status", 
        description = f"{state.players.online} out of {state.players.max} currently online." if state is not None else "Server is offline", 
        color = COLOR_HEX
    )
    
    if state is not None and state.players.sample is not None:
        embed.add_field(
            name = "Players",
            # List comprehension to get all the players in a nice looking string.
            value = ", ".join([player.name for player in state.players.sample]))

    await ctx.reply(embed=embed)

# Loop that updates the server state and sets the bots presence to match
@tasks.loop(seconds=60)
async def update_state():
    logging.info("Updating state")
    global state
    try:
        state = server.status()
        if state.players.online == 0 and state.players.max == 0:
            state = None
    except Exception as e:
        logging.warn(e)
        state = None
    
    
    await bot.change_presence(
        activity=discord.Activity(
            type = discord.ActivityType.watching, 
            name = f"{state.players.online} out of {state.players.max} players" if state is not None else "an offline server"
        ))

bot.run(DISCORD_TOKEN)