Merge pull request #38 from srobo/feature/mentionable

Make teams mentionable

src/sr/discord_bot/bot.py

@@ -29,6 +29,7 @@
     delete_team,
     export_team,
     create_voice,
+    repair_permissions,
     create_team_channel,
 )
 from sr.discord_bot.commands.stats import (
@@ -77,6 +78,7 @@ def __init__(
         team.add_command(create_voice)
         team.add_command(create_team_channel)
         team.add_command(export_team)
+        team.add_command(repair_permissions)
         self.tree.add_command(team, guild=self.guild)
         stats = Stats()
         stats.add_command(post_stats)

src/sr/discord_bot/commands/team.py

@@ -76,6 +76,7 @@ async def new_team(interaction: discord.interactions.Interaction["BotClient"], t
     role = await guild.create_role(
         reason=TEAM_CREATED_REASON + interaction.user.name,
         name=role_name,
+        mentionable=True,
     )
     channel = await guild.create_text_channel(
         reason=TEAM_CREATED_REASON + interaction.user.name,
@@ -255,3 +256,41 @@ async def export_team(
         output = output + await _export_team(tla, only_teams, guild, interaction)
     output = output + "\n```"
     await interaction.followup.send(content=output, ephemeral=True)
+
+
+def _repair_permissions_status_msg(cur_team: int, num_teams: int) -> str:
+    return f"Repairing permissions... {cur_team}/{num_teams} teams processed"
+
+
+@group.command(  # type:ignore[arg-type]
+    name='repair-permissions',
+    description='Reset channel and team permissions',
+)
+async def repair_permissions(interaction: discord.interactions.Interaction["BotClient"]) -> None:
+    guild: discord.Guild | None = interaction.guild
+    if guild is None:
+        raise app_commands.NoPrivateMessage()
+
+    team_roles = [role for role in guild.roles if role.name.startswith(ROLE_PREFIX)]
+
+    await interaction.response.defer(
+        thinking=True,
+        ephemeral=True,
+    )
+    await interaction.edit_original_response(content=_repair_permissions_status_msg(0, len(team_roles)))
+
+    for index, role in enumerate(team_roles):
+        tla = role.name.removeprefix(ROLE_PREFIX)
+        channels = [
+            channel
+            for channel in guild.channels
+            if channel.name.startswith(TEAM_CHANNEL_PREFIX + tla)
+        ]
+        channel_permissions = permissions(interaction.client, role)
+        for channel in channels:
+            for channel_role, overwrites in channel_permissions.items():
+                await channel.set_permissions(target=channel_role, overwrite=overwrites)
+        await role.edit(mentionable=True)
+        await interaction.edit_original_response(content=_repair_permissions_status_msg(index + 1, len(team_roles)))
+
+    await interaction.edit_original_response(content="Repairing permissions... done!")