feat: timeout for bot

Author: WorldObservationLog

extras/telegram_bot/config.example.toml

@@ -18,6 +18,7 @@ allowed_types = ["song", "album", "playlist"]
 max_tracks = 20
 max_song_duration_sec = 900
 max_total_duration_sec = 18000
+task_timeout_sec = 480
 
 [user_default]
 language = "follow-user"

extras/telegram_bot/src/config.py

@@ -26,6 +26,7 @@ class LimitsSettings(BaseModel):
     max_tracks: int = 100
     max_song_duration_sec: int = 1200
     max_total_duration_sec: int = 18000
+    task_timeout_sec: int = 480
 
 
 class UserDefaultSettings(BaseModel):

extras/telegram_bot/src/handlers/download.py

@@ -452,8 +452,18 @@ async def process_song(song_id: str, storefront: str, p_index=None):
 
     # Spawn Rip Tasks
     if actual_songs_to_rip:
+        async def rip_with_timeout(s_obj, cd, fl, pd, pl):
+            import asyncio
+            try:
+                await asyncio.wait_for(
+                    ripper.rip_song(s_obj, cd, fl, parent_done=pd, playlist=pl),
+                    timeout=bot_config.limits.task_timeout_sec
+                )
+            except asyncio.TimeoutError:
+                pass
+                
         for song_obj, p_index in actual_songs_to_rip:
-            safely_create_task(ripper.rip_song(song_obj, codec, flags, parent_done=parent_done, playlist=p_index))
+            safely_create_task(rip_with_timeout(song_obj, codec, flags, parent_done, p_index))
     else:
         # Everything was cached
         completed_event.set()

src/rip.py

@@ -207,6 +207,11 @@ async def rip_song(self, url: Song, codec: str, flags: Flags = Flags(),
             task.logger.logger.exception(f"Error processing song: {e}")
             task.update_status(Status.FAILED)
             task.error = e
+        except asyncio.CancelledError:
+            task.logger.logger.warning("Task processing timed out or was cancelled")
+            task.update_status(Status.FAILED)
+            task.error = Exception("Task execution timed out")
+            raise
         finally:
             await self.download_manager.unregister_task(task)
             task.update_status(task.status)  # Ensure status is set