mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-28 16:20:34 +08:00
Alex Ott
e7515fd08e
* improve messages and UX * move log statement Co-authored-by: Andreas Köpf <andreas.koepf@provisio.com>
105 lines
3.8 KiB
Python
105 lines
3.8 KiB
Python
"""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)
|