feat: add requester tracking

discord_sound_streamer/datastore/models/commands.py

@@ -1,10 +1,11 @@
 from dataclasses import dataclass
 from datetime import datetime
 
-from hikari import Snowflake
+from hikari import Snowflake, User
 
 
 @dataclass
 class LastCommandValue:
     executed_at: datetime
     channel_id: Snowflake
+    triggered_by: User

discord_sound_streamer/services/commands.py

@@ -12,7 +12,11 @@ async def update_last_command_time(ctx: tanjun.abc.Context) -> None:
     if ctx.guild_id:
         commands_operations.set_last_command(
             ctx.guild_id,
-            LastCommandValue(executed_at=datetime.now(UTC), channel_id=ctx.channel_id),
+            LastCommandValue(
+                executed_at=datetime.now(UTC),
+                channel_id=ctx.channel_id,
+                triggered_by=ctx.author,
+            ),
         )
         logger.info(f"Updated last command time for guild {ctx.guild_id}")
 

discord_sound_streamer/services/embed.py

@@ -14,6 +14,10 @@
 }
 
 
+def _mention(user_id: int) -> str:
+    return f"<@{user_id}>"
+
+
 def _format_timedelta(ms: int) -> str:
     if ms > 2**31 - 1:
         return "∞"
@@ -48,7 +52,12 @@ def _apply_track_list_to_embed(embed: Embed, tracks: List[AudioTrack]) -> None:
             value=_build_track_link(track),
             inline=True,
         )
-        embed.add_field(name="Uploader", value=track.author, inline=True)
+        if track.requester:
+            embed.add_field(
+                name=track.author, value=_mention(track.requester), inline=True
+            )
+        else:
+            embed.add_field(name="Uploader", value=track.author, inline=True)
         embed.add_field(
             name="Length",
             value=_format_timedelta(track.duration),
@@ -77,6 +86,8 @@ def build_track_embed(
         inline=True,
     )
     embed.add_field(name="Link", value=_build_track_link(track), inline=True)
+    if track.requester != 0:
+        embed.add_field(name="Requester", value=_mention(track.requester), inline=True)
     return embed
 
 

discord_sound_streamer/services/play.py

@@ -100,7 +100,7 @@ async def _play_tracks(
         #         wait=wait_fixed(0.25),
         #     ):
         #         with attempt:
-        player.add(track)
+        player.add(track, requester=author_id)
         # except RetryError as re:
         #     logger.exception(re)