Add permanent streaming command

1 files changed, 24 insertions, 1 deletions

diff --git a/bot/exts/moderation/stream.py b/bot/exts/moderation/stream.py
index af0633bdb..46c68f74b 100644
--- a/bot/exts/moderation/stream.py
+++ b/bot/exts/moderation/stream.py
@@ -26,7 +26,7 @@ class Stream(commands.Cog):
     async def _remove_streaming_permission(self, schedule_user: discord.Member) -> None:
         """Remove streaming permission from Member."""
         await self._delete_from_redis(schedule_user.id)
-        await schedule_user.remove_roles(discord.Object(Roles.video), reason="Temporary streaming access revoked")
+        await schedule_user.remove_roles(discord.Object(Roles.video), reason="Streaming access revoked")
 
     async def _add_to_redis_cache(self, user_id: int, timestamp: float) -> None:
         """Adds 'task' to redis cache."""
@@ -86,6 +86,29 @@ class Stream(commands.Cog):
         duration = format_infraction_with_duration(str(duration))
         await ctx.send(f"{Emojis.check_mark} {user.mention} can now stream until {duration}.")
 
+    @commands.command(aliases=("pstream",))
+    @commands.has_any_role(*STAFF_ROLES)
+    async def permanentstream(
+            self,
+            ctx: commands.Context,
+            user: discord.Member,
+            *_
+    ) -> None:
+        """Permanently give user a streaming permission."""
+        # Check if user already has streaming permission
+        already_allowed = any(Roles.video == role.id for role in user.roles)
+        if already_allowed:
+            if user.id in self.scheduler:
+                self.scheduler.cancel(user.id)
+                await self._delete_from_redis(user.id)
+                await ctx.send(f"{Emojis.check_mark} Moved temporary permission to permanent")
+                return
+            await ctx.send(f"{Emojis.cross_mark} This user can already stream.")
+            return
+
+        await user.add_roles(discord.Object(Roles.video), reason="Permanent streaming access granted")
+        await ctx.send(f"{Emojis.check_mark} {user.mention} can now stream forever")
+
     @commands.command(aliases=("unstream", ))
     @commands.has_any_role(*STAFF_ROLES)
     async def revokestream(