aw-bot/bot/commands.py

from typing import Literal
import re

import discord
from discord import app_commands

from database import add_pattern, add_user_to_blacklist, is_user_blacklisted

from bot.config import update_patterns, message_patterns
from bot.utils import compile_stats, fetch_game_stats, perform_search

GUILD_ID = 1110531063161299074

BOT_LOG = 1112049391482703873
GENERAL_CHANNEL = 1110531063744303138


async def setup(bot):
    async def on_tree_error(
        interaction: discord.Interaction, error: app_commands.AppCommandError
    ):
        if isinstance(error, app_commands.CommandOnCooldown):
            return await interaction.response.send_message(
                f"Command is currently on cooldown! Try again in **{error.retry_after:.2f}** seconds!"
            )
        elif isinstance(error, app_commands.MissingPermissions):
            return await interaction.response.send_message(
                "You are missing permissions to use that"
            )
        else:
            raise error

    bot.tree.on_error = on_tree_error

    @bot.tree.command(
        name="add_pattern",
        description="Add a new message pattern to the database.",
        guild=discord.Object(id=GUILD_ID),
    )
    @app_commands.checks.has_permissions(administrator=True)
    async def add_pattern_cmd(
        interaction: discord.Interaction, regex: str, response: str
    ):
        """Slash command to add a new message pattern to the database."""
        add_pattern(regex, response)
        update_patterns(regex, response)
        await interaction.response.send_message(
            f"Pattern added!\n**Regex:** `{regex}`\n**Response:** `{response}`"
        )

    @bot.tree.command(
        name="add_to_blacklist",
        description="Add a user to the blacklist.",
        guild=discord.Object(id=GUILD_ID),
    )
    @app_commands.checks.has_permissions(administrator=True)
    async def add_to_blacklist_cmd(
        interaction: discord.Interaction, user: discord.User, reason: str
    ):
        """Slash command to add a user to the blacklist."""
        add_user_to_blacklist(user.id, reason)
        await interaction.response.send_message(
            f"User **{user.name}** has been added to the blacklist.\n**Reason:** `{reason}`"
        )

    @bot.tree.command(
        name="search",
        description="Search for servers by hostname or IP.",
        guild=discord.Object(id=GUILD_ID),
    )
    async def slash_search(interaction: discord.Interaction, query: str):
        results = await perform_search(query)
        await interaction.response.send_message(results)

    @app_commands.checks.cooldown(1, 60, key=lambda i: (i.guild_id, i.user.id))
    @bot.tree.command(
        name="stats",
        description="Get stats for a specific game or all games",
        guild=discord.Object(id=GUILD_ID),
    )
    async def stats(
        interaction: discord.Interaction, game: Literal["iw4", "s1", "iw6", "t7", "all"]
    ):
        if game == "all":
            stats_message = await compile_stats()
        else:
            data = await fetch_game_stats(game)
            if data:
                stats_message = f"**Stats for {game.upper()}:**\n"
                count_servers = data.get("countServers", "N/A")
                count_players = data.get("countPlayers", "N/A")
                stats_message += f"Total Servers: {count_servers}\n"
                stats_message += f"Total Players: {count_players}\n"
            else:
                stats_message = "Failed to fetch game stats. Please try again later."

        await interaction.response.send_message(stats_message, ephemeral=True)

    @bot.tree.command(
        name="meme",
        description="Check if the input matches any predefined memess.",
        guild=discord.Object(id=GUILD_ID),
    )
    async def meme(interaction: discord.Interaction, input: str):
        """
        Slash command to check if the input matches any predefined patterns.
        """
        # Check if the user is blacklisted
        if is_user_blacklisted(interaction.user.id):
            await interaction.response.send_message(
                "You are blacklisted from using this command.", ephemeral=True
            )
            return

        # Check if any of the patterns match the input
        for pattern in message_patterns:
            if re.search(pattern["regex"], input, re.IGNORECASE):
                response = pattern["response"]
                reply_message = await interaction.response.send_message(
                    response, ephemeral=False
                )
                # Add a reaction to the reply message (if the user decides to delete it)
                await reply_message.add_reaction("\U0000274C")
                break
        else:
            await interaction.response.send_message(
                "No matching patterns found.", ephemeral=True
            )

    await bot.tree.sync(guild=discord.Object(id=GUILD_ID))  # Force sync

    print("Commands extension loaded!")