guild_settings.py

"""Guild settings."""
import hikari
import lightbulb
from aiosqlite import Connection
from bot.db.schemas import GuildSettings
from bot.utils import mention
from lightbulb.utils import permissions_in
from loguru import logger

plugin = lightbulb.Plugin("GuildSettings")
plugin.add_checks(lightbulb.guild_only)
plugin.add_checks(lightbulb.has_guild_permissions(hikari.Permissions.MANAGE_GUILD))


@plugin.command
@lightbulb.command("settings", "Bot settings for the server.")
@lightbulb.implements(lightbulb.SlashCommandGroup)
async def settings(_: lightbulb.SlashContext) -> None:
    """Bot settings for the server."""
    # This will never execute because it is a group
    pass


@settings.child
@lightbulb.command("get", "Get all the guild settings.")
@lightbulb.implements(lightbulb.SlashSubCommand)
async def get(ctx: lightbulb.SlashContext) -> None:
    """Get one of or all the guild settings."""
    conn: Connection = ctx.bot.d.db
    assert ctx.guild_id is not None  # `guild_only` check

    async with conn.cursor() as cursor:
        # Get all settings
        await cursor.execute("SELECT * FROM guild_settings WHERE guild_id = ?", (ctx.guild_id,))
        row = await cursor.fetchone()

        if row is None:
            logger.warning(f"No guild settings for {ctx.guild_id}")
            await ctx.respond("No settings found for this guild.")
            return

        guild_settings = GuildSettings.parse_obj(row)

        # Respond with all
        # TODO: Embed
        await ctx.respond(
            f"""\
**Guild Settings**
`log_channel`: {
mention(guild_settings.log_channel_id, "channel")
if guild_settings.log_channel_id else 'not set'}
"""
        )


@settings.child
@lightbulb.option("channel", "The channel to use.", hikari.TextableGuildChannel)
@lightbulb.command("log_channel", "Set the channel that the bot logs task and label completions in.", ephemeral=True)
@lightbulb.implements(lightbulb.SlashSubCommand)
async def log_channel(ctx: lightbulb.SlashContext) -> None:
    """Set the channel that the bot logs task and label completions in."""
    channel: hikari.TextableGuildChannel = ctx.options.channel
    conn: Connection = ctx.bot.d.db
    assert ctx.guild_id is not None  # `guild_only` check

    # Check if the bot can send messages in that channel
    assert isinstance(channel, hikari.InteractionChannel)  # Slash commands are interactions
    me = ctx.bot.cache.get_me() or await ctx.bot.rest.fetch_my_user()
    own_member = ctx.bot.cache.get_member(ctx.guild_id, me.id) or await ctx.bot.rest.fetch_member(ctx.guild_id, me.id)

    # Get the channel from the cache if it is there, otherwise fetch it
    if (ch := ctx.bot.cache.get_guild_channel(channel.id)) is None:
        ch = {ch.id: ch for ch in await ctx.bot.rest.fetch_guild_channels(channel.id)}[channel.id]

    if not isinstance(ch, hikari.GuildTextChannel):
        await ctx.respond(f"{ch.mention} is not a text channel.")
        return

    # if the bot's permissions for this channel don't contain SEND_MESSAGE
    # This will also filter out categories and voice channels
    if not permissions_in(ch, own_member) & hikari.Permissions.SEND_MESSAGES:
        await ctx.respond(f"I don't have permission to send messages in {ch.mention}.")
        return

    await ctx.respond(f"Setting `log_channel` to {channel.mention}.")

    # update the database
    async with conn.cursor() as cursor:
        await cursor.execute(
            "INSERT OR REPLACE INTO guild_settings (guild_id, log_channel_id) VALUES (?, ?)",
            (ctx.guild_id, channel.id),
        )
    await conn.commit()
    logger.info(f"Updated `log_channel` for {ctx.guild_id} to {channel.id}.")


def load(bot: lightbulb.BotApp):
    """Add the plugin to the bot."""
    bot.add_plugin(plugin)


def unload(bot: lightbulb.BotApp):
    """Remove the plugin to the bot."""
    bot.remove_plugin(plugin)