Add command for pinning/unpinning messages

src/sr/discord_bot/bot.py

@@ -21,6 +21,7 @@
     ANNOUNCE_CHANNEL_NAME,
     WELCOME_CATEGORY_NAME,
 )
+from sr.discord_bot.commands.pin import pin_message, unpin_message
 from sr.discord_bot.commands.join import join
 from sr.discord_bot.commands.logs import logs
 from sr.discord_bot.commands.team import (
@@ -87,6 +88,8 @@ def __init__(
         self.tree.add_command(stats, guild=self.guild)
         self.tree.add_command(join, guild=self.guild)
         self.tree.add_command(logs, guild=self.guild)
+        self.tree.add_command(pin_message, guild=self.guild)
+        self.tree.add_command(unpin_message, guild=self.guild)
         self.load_passwords()
         load_subscribed_messages(self)
 

src/sr/discord_bot/commands/pin.py

@@ -0,0 +1,49 @@
+from typing import cast, TYPE_CHECKING
+
+import discord
+from discord import app_commands
+
+if TYPE_CHECKING:
+    from sr.discord_bot.bot import BotClient
+
+
+@app_commands.command(  # type:ignore[arg-type]
+    name='pin',
+    description='Pin a message to the channel',
+)
+@app_commands.describe(
+    message_url='Link of the message to pin',
+)
+@app_commands.rename(message_url='message')
+async def pin_message(
+    interaction: discord.interactions.Interaction["BotClient"],
+    message_url: str,
+) -> None:
+    """Pin a message to the channel."""
+    channel = cast(discord.TextChannel, interaction.channel)
+    message_id = int(message_url.split('/')[-1])
+    message = await channel.fetch_message(message_id)
+    reason = "Pinned by " + interaction.user.mention
+    await message.pin(reason=reason)
+    await interaction.response.send_message(f"_{reason}_")
+
+
+@app_commands.command(  # type:ignore[arg-type]
+    name='unpin',
+    description='Unpin a message from the channel',
+)
+@app_commands.describe(
+    message_url='Link of the message to unpin',
+)
+@app_commands.rename(message_url='message')
+async def unpin_message(
+    interaction: discord.interactions.Interaction["BotClient"],
+    message_url: str,
+) -> None:
+    """Unpin a message from the channel."""
+    channel = cast(discord.TextChannel, interaction.channel)
+    message_id = int(message_url.split('/')[-1])
+    message = await channel.fetch_message(message_id)
+    reason = "Unpinned by " + interaction.user.mention
+    await message.unpin(reason=reason)
+    await interaction.response.send_message(f"_{reason}_")