Ariella/commands.py

import discord
from discord import app_commands
from discord.ext import commands
import aiosqlite
import os
import sys
import subprocess
from gdpr import check_consent, give_consent, revoke_consent


class ModCommands(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @app_commands.command(name="consent",
                          description="Give consent to store data")
    async def consent(self, interaction: discord.Interaction):
        await give_consent(interaction.user.id)
        await interaction.response.send_message(
            "Consent given to store your data."
        )

    @app_commands.command(name="revoke_consent",
                          description="Revoke consent to store data")
    async def revoke_consent(self, interaction: discord.Interaction):
        await revoke_consent(interaction.user.id)
        await interaction.response.send_message(
            "Consent revoked and your data has been deleted."
        )

    @app_commands.command(name="addnote",
                          description="Add a note to a user and optionally "
                                      "add strikes")
    async def add_note(self, interaction: discord.Interaction,
                       user: discord.User, note: str, strikes: int = 0):
        if not await check_consent(user.id):
            await interaction.response.send_message(
                f"{user.name} has not given consent to store data."
            )
            return

        async with aiosqlite.connect("ariella.db") as db:
            cursor = await db.execute(
                "SELECT notes, strikes FROM user_notes WHERE user_id = ?",
                (user.id,)
            )
            row = await cursor.fetchone()
            if row:
                notes = row[0] + "\n" + note
                current_strikes = row[1] + strikes
                await db.execute(
                    "UPDATE user_notes SET notes = ?,"
                    "strikes = ? WHERE user_id = ?",
                    (notes, current_strikes, user.id)
                )
            else:
                await db.execute(
                    "INSERT INTO user_notes (user_id, notes, strikes) "
                    "VALUES (?, ?, ?)",
                    (user.id, note, strikes)
                )
            await db.commit()
        await interaction.response.send_message(
            f"Note added for {user.name}: {note}. Strikes: {strikes}"
        )

    @app_commands.command(name="warn",
                          description="Warn a user")
    async def warn_user(self, interaction: discord.Interaction,
                        user: discord.User, reason: str):
        if not await check_consent(user.id):
            await interaction.response.send_message(
                f"{user.name} has not given consent to store data."
            )
            return

        async with aiosqlite.connect("ariella.db") as db:
            cursor = await db.execute(
                "SELECT strikes FROM user_notes WHERE user_id = ?",
                (user.id,)
            )
            row = await cursor.fetchone()
            if row:
                strikes = row[0] + 1
                await db.execute(
                    "UPDATE user_notes SET strikes = ? WHERE user_id = ?",
                    (strikes, user.id)
                )
            else:
                strikes = 1
                await db.execute(
                    "INSERT INTO user_notes (user_id, notes, strikes) "
                    "VALUES (?, ?, ?)",
                    (user.id, "", strikes)
                )
            await db.commit()
        await interaction.response.send_message(
            f"User {user.name} warned for: {reason}. They now have {strikes} "
            f"strikes."
        )
        await user.send(
            f"You have been warned for: {reason}. You now have {strikes} "
            f"strikes."
        )

    @app_commands.command(name="removestrikes",
                          description="Remove strikes from a user")
    async def remove_strikes(self, interaction: discord.Interaction,
                             user: discord.User, strikes: int):
        if not await check_consent(user.id):
            await interaction.response.send_message(
                f"{user.name} has not given consent to store data."
            )
            return

        async with aiosqlite.connect("ariella.db") as db:
            cursor = await db.execute(
                "SELECT strikes FROM user_notes WHERE user_id = ?",
                (user.id,)
            )
            row = await cursor.fetchone()
            if row:
                current_strikes = max(row[0] - strikes, 0)
                await db.execute(
                    "UPDATE user_notes SET strikes = ? WHERE user_id = ?",
                    (current_strikes, user.id)
                )
                await db.commit()
                await interaction.response.send_message(
                    f"Removed {strikes} strikes from {user.name}. "
                    f"They now have {current_strikes} strikes."
                )
            else:
                await interaction.response.send_message(
                    f"No strikes found for {user.name}."
                )

    @app_commands.command(name="checknotes",
                          description="Check notes and strikes of a user")
    async def check_notes(self, interaction: discord.Interaction,
                          user: discord.User):
        if not await check_consent(user.id):
            await interaction.response.send_message(
                f"{user.name} has not given consent to store data."
            )
            return

        async with aiosqlite.connect("ariella.db") as db:
            cursor = await db.execute(
                "SELECT notes, strikes FROM user_notes WHERE user_id = ?",
                (user.id,)
            )
            row = await cursor.fetchone()
            if row:
                notes, strikes = row
                await interaction.response.send_message(
                    f"Notes for {user.name}: {notes}\nStrikes: {strikes}"
                )
            else:
                await interaction.response.send_message(
                    f"No notes found for {user.name}."
                )

    @app_commands.command(name="update",
                          description="Update Ariellia to the latest version")
    @commands.is_owner()
    async def update(self, interaction: discord.Interaction):
        await interaction.response.send_message("Updating the bot...")
        # Pull latest changes from the repository
        repo_dir = os.path.dirname(os.path.abspath(__file__))
        subprocess.run(["git", "-C", repo_dir, "pull"])
        # Restart the bot
        await interaction.followup.send("Restarting the bot...")
        os.execv(sys.executable, ['python'] + sys.argv)

    @app_commands.command(name="help",
                          description="List all commands")
    async def help_command(self, interaction: discord.Interaction):
        commands = self.bot.tree.walk_commands()
        help_text = "Here are the available commands:\n"
        for command in commands:
            help_text += f"/{command.name} - {command.description}\n"
        await interaction.response.send_message(help_text)


async def setup(bot):
    await bot.add_cog(ModCommands(bot))