Initial commit

discord-bot/src/playback_service.py

@@ -0,0 +1,192 @@
+import time
+import discord
+from src.models import BotResponse, BotStatus, MessageType, PlaybackInformation
+from src.my_voice_client import get_voice_client
+from src.song_queue import (
+    get_current_metadata,
+    get_queue_status,
+    handle_song_end,
+    has_current_song,
+    move_to_last_song_in_queue,
+    set_current_song_start_time,
+    set_queue_position,
+)
+
+pause_offset = -1
+
+
+def after_playing(error):
+    if error:
+        print(f"Error during playback: {error}")
+    fileName, duration, start_time = get_current_metadata()
+    print(f"Finished playing {fileName}")
+
+    fileName, duration, start_time = get_current_metadata()
+    current_playing_time = time.time() - start_time
+
+    if current_playing_time > (duration - 1):
+        # song ended
+        handle_song_end()
+        if has_current_song():
+            print("start next song")
+            play_current_song()
+        else:
+            print("end of queue")
+    else:
+        print("not changing song because it is still playing")
+
+
+def change_playback_position(position: int):
+    fileName, duration, start_time = get_current_metadata()
+    voice_client = get_voice_client()
+    if voice_client and voice_client.is_playing():
+        voice_client.pause()
+        audio = discord.FFmpegPCMAudio(
+            source=fileName, before_options=f"-ss {position}"
+        )
+        voice_client.play(audio, after=after_playing)
+        set_current_song_start_time(time.time() - position)
+        return {"status": "Playback position changed"}
+    else:
+        print("cannot change position, no song playing")
+        return None
+
+
+def play_current_song():
+    if has_current_song():
+        fileName, duration, start_time = get_current_metadata()
+        start_time_now()
+        get_voice_client().play(
+            discord.FFmpegPCMAudio(source=fileName), after=after_playing
+        )
+
+
+def get_status():
+    voice_client = get_voice_client()
+    if voice_client and voice_client.is_playing():
+        return BotStatus.PLAYING
+    return BotStatus.IDLE
+
+
+def get_playback_info():
+    fileName, duration, start_time = get_current_metadata()
+    voice_client = get_voice_client()
+    if voice_client and voice_client.is_playing():
+        elapsed_time = time.time() - start_time
+
+        return PlaybackInformation(
+            file_name=fileName,
+            current_position=elapsed_time,
+            duration=duration,
+        )
+    else:
+        return None
+
+
+def handle_message(data) -> BotResponse:
+    if "action" not in data:
+        response = BotResponse(
+            message_type=MessageType.ERROR,
+            status=get_status(),
+            error="Invalid request, action is required",
+        )
+        return response
+
+    if data["action"] == "set_playback":
+        if "position" not in data:
+            response = BotResponse(
+                message_type=MessageType.ERROR,
+                status=get_status(),
+                error="Invalid request, position is required",
+            )
+            return response
+
+        result = change_playback_position(data["position"])
+        if result:
+            response = BotResponse(
+                message_type=MessageType.MESSAGE,
+                status=get_status(),
+                message="position changed",
+            )
+            return response
+        else:
+            response = BotResponse(
+                message_type=MessageType.ERROR,
+                status=get_status(),
+                error="unable to change position",
+            )
+            return response
+    elif data["action"] == "set_position":
+        if "position" not in data:
+            response = BotResponse(
+                message_type=MessageType.ERROR,
+                status=get_status(),
+                error="Invalid request, position is required",
+            )
+            return response
+        set_queue_position(data["position"])
+        get_voice_client().stop()
+        play_current_song()
+        info = get_playback_info()
+        response = BotResponse(
+            message_type=MessageType.PLAYBACK_INFORMATION,
+            status=BotStatus.PLAYING if info else BotStatus.IDLE,
+            playback_information=info,
+            song_queue=get_queue_status(),
+        )
+        return response
+
+    elif data["action"] == "get_playback_info":
+        if not has_current_song():
+            return BotResponse(
+                message_type=MessageType.PLAYBACK_INFORMATION,
+                status=BotStatus.IDLE,
+                playback_information=None,
+                song_queue=get_queue_status(),
+            )
+        info = get_playback_info()
+        response = BotResponse(
+            message_type=MessageType.PLAYBACK_INFORMATION,
+            status=BotStatus.PLAYING if info else BotStatus.IDLE,
+            playback_information=info,
+            song_queue=get_queue_status(),
+        )
+        return response
+
+
+def get_filename_and_starttime():
+    fileName, duration, start_time = get_current_metadata()
+    return fileName, start_time
+
+
+def start_time_now():
+    set_current_song_start_time(time.time())
+
+
+def handle_new_song_on_queue():
+    if not has_current_song():
+        move_to_last_song_in_queue()
+        if has_current_song():
+            play_current_song()
+        else:
+            print("moving to the last song did not put us on a current song")
+    else:
+        print("not moving to new song because there is current song")
+
+
+def pause_song():
+    global pause_offset
+    voice_client = get_voice_client()
+    if voice_client and voice_client.is_playing():
+        fileName, duration, start_time = get_current_metadata()
+        pause_offset = time.time() - start_time
+        voice_client.pause()
+
+
+def unpause_song():
+    global pause_offset
+    voice_client = get_voice_client()
+    if voice_client and voice_client.is_playing():
+        voice_client.resume()
+        set_current_song_start_time(time.time() - pause_offset)
+        pause_offset = -1