mirror of
https://github.com/alterware/aw-bot.git
synced 2025-10-26 14:15:54 +00:00
98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
from typing import Literal
|
|
|
|
import discord
|
|
from discord import app_commands
|
|
|
|
from database import add_pattern, add_user_to_blacklist
|
|
|
|
from bot.config import update_patterns
|
|
from bot.utils import compile_stats, fetch_game_stats, perform_search
|
|
|
|
GUILD_ID = 1110531063161299074
|
|
|
|
|
|
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)
|
|
|
|
await bot.tree.sync(guild=discord.Object(id=GUILD_ID)) # Force sync
|
|
|
|
print("Commands extension loaded!")
|