Merge pull request #173 from omgate234/feat/audio-image-fetch

feat: handle audio image CRUD endpoints

Author: 0xrohitgarg

backend/director/entrypoint/api/routes.py

@@ -125,6 +125,32 @@ def get_video_or_all(collection_id, video_id):
         return videodb.get_video(video_id)
     else:
         return videodb.get_videos()
+    
+@videodb_bp.route(
+    "/collection/<collection_id>/audio", defaults={"audio_id": None}, methods=["GET"]
+)
+@videodb_bp.route("/collection/<collection_id>/audio/<audio_id>", methods=["GET"])
+def get_audio_or_all(collection_id, audio_id, **kwargs):
+    """Get a video by ID or all videos in a collection."""
+    videodb = VideoDBHandler(collection_id)
+    if audio_id:
+        return videodb.get_audio(audio_id)
+    else:
+        return videodb.get_audios()
+
+
+@videodb_bp.route(
+    "/collection/<collection_id>/image", defaults={"image_id": None}, methods=["GET"]
+)
+@videodb_bp.route("/collection/<collection_id>/image/<image_id>", methods=["GET"])
+def get_image_or_all(collection_id, image_id, **kwargs):
+    """Get a video by ID or all videos in a collection."""
+    videodb = VideoDBHandler(collection_id)
+    if image_id:
+        return videodb.get_image(image_id)
+    else:
+        return videodb.get_images()
+
 
 
 @videodb_bp.route("/collection/<collection_id>/video/<video_id>", methods=["DELETE"])
@@ -140,6 +166,32 @@ def delete_video(collection_id, video_id):
         return {"message": str(e)}, 500
 
 
+@videodb_bp.route("/collection/<collection_id>/audio/<audio_id>", methods=["DELETE"])
+def delete_audio(collection_id, audio_id):
+    """Delete a audio by ID from a specific collection."""
+    try:
+        if not audio_id:
+            return {"message": "Video ID is required"}, 400
+        videodb = VideoDBHandler(collection_id)
+        result = videodb.delete_audio(audio_id)
+        return result, 200
+    except Exception as e:
+        return {"message": str(e)}, 500
+
+
+@videodb_bp.route("/collection/<collection_id>/image/<image_id>", methods=["DELETE"])
+def delete_image(collection_id, image_id):
+    """Delete a image by ID from a specific collection."""
+    try:
+        if not image_id:
+            return {"message": "Video ID is required"}, 400
+        videodb = VideoDBHandler(collection_id)
+        result = videodb.delete_image(image_id)
+        return result, 200
+    except Exception as e:
+        return {"message": str(e)}, 500
+
+
 @videodb_bp.route(
     "/collection/<collection_id>/image/<image_id>/generate_url", methods=["GET"]
 )
@@ -157,6 +209,22 @@ def generate_image_url(collection_id, image_id):
     except Exception as e:
         return {"message": str(e)}, 500
 
+@videodb_bp.route(
+    "/collection/<collection_id>/audio/<audio_id>/generate_url", methods=["GET"]
+)
+def generate_audio_url(collection_id, audio_id):
+    try:
+        if not collection_id:
+            return {"message": "Collection ID is required"}, 400
+
+        if not audio_id:
+            return {"message": "Audio ID is required"}, 400
+
+        videodb = VideoDBHandler(collection_id)
+        result = videodb.generate_audio_url(audio_id)
+        return result, 200
+    except Exception as e:
+        return {"message": str(e)}, 500
 
 @videodb_bp.route("/collection/<collection_id>/upload", methods=["POST"])
 def upload_video(collection_id):

backend/director/handler.py

@@ -148,7 +148,7 @@ def __init__(self, collection_id="default"):
     def upload(self, source, source_type="url", media_type="video", name=None):
         return self.videodb_tool.upload(source, source_type, media_type, name)
 
-    def get_collection(self):
+    def get_collection(self, collection_id):
         """Get a collection by ID."""
         return self.videodb_tool.get_collection()
 
@@ -170,10 +170,40 @@ def delete_video(self, video_id):
         """Delete a specific video by its ID."""
         return self.videodb_tool.delete_video(video_id)
 
+    def delete_image(self, image_id):
+        """Delete a specific image by its ID."""
+        return self.videodb_tool.delete_image(image_id)
+
+    def delete_audio(self, video_id):
+        """Delete a specific audio by its ID."""
+        return self.videodb_tool.delete_audio(video_id)
+
     def get_videos(self):
         """Get all videos in a collection."""
         return self.videodb_tool.get_videos()
 
+    def get_audio(self, audio_id):
+        """Get a audio by ID."""
+        return self.videodb_tool.get_audio(audio_id)
+
+    def get_audios(self):
+        """Get all audios in a collection."""
+        return self.videodb_tool.get_audios()
+
+    def generate_audio_url(self, audio_id):
+        return self.videodb_tool.generate_audio_url(audio_id=audio_id)
+
+    def delete_audio(self, audio_id):
+        return self.videodb_tool.delete_audio(audio_id)
+
+    def get_image(self, image_id):
+        """Get a image by ID."""
+        return self.videodb_tool.get_image(image_id)
+
+    def get_images(self):
+        """Get all images in a collection."""
+        return self.videodb_tool.get_images()
+
     def generate_image_url(self, image_id):
         return self.videodb_tool.generate_image_url(image_id=image_id)
 

backend/director/tools/videodb_tool.py

@@ -53,6 +53,20 @@ def get_image(self, image_id: str = None):
         except Exception as e:
             raise Exception(f"Failed to fetch image with ID {image_id}: {e}")
 
+    def get_images(self):
+        """Get all images in a collection."""
+        images = self.collection.get_images()
+        return [
+            {
+                "id": image.id,
+                "collection_id": image.collection_id,
+                "name": image.name,
+                "url": image.url,
+                "type": "image",
+            }
+            for image in images
+        ]
+
     def create_collection(self, name, description=""):
         """Create a new collection with the given name and description."""
         if not name:
@@ -140,6 +154,7 @@ def get_videos(self):
                 "stream_url": video.stream_url,
                 "length": video.length,
                 "thumbnail_url": video.thumbnail_url,
+                "type": "video",
             }
             for video in videos
         ]
@@ -161,12 +176,18 @@ def get_audios(self):
         return [
             {
                 "id": audio.id,
+                "collection_id": audio.collection_id,
                 "name": audio.name,
                 "length": audio.length,
+                "type": "audio",
             }
             for audio in audios
         ]
 
+    def generate_audio_url(self, audio_id):
+        audio = self.collection.get_audio(audio_id)
+        return audio.generate_url()
+
     def generate_image_url(self, image_id):
         image = self.collection.get_image(image_id)
         return image.generate_url()
@@ -320,3 +341,58 @@ def add_subtitle(self, video_id, style: SubtitleStyle = SubtitleStyle()):
         video = self.collection.get_video(video_id)
         stream_url = video.add_subtitle(style)
         return stream_url
+
+    def delete_audio(self, audio_id):
+        """Delete a specific audio by its ID."""
+        if not audio_id:
+            raise ValueError("Audio ID is required to delete a audio.")
+        try:
+            audio = self.collection.get_audio(audio_id)
+            if not audio:
+                raise ValueError(
+                    f"Audio with ID {audio_id} not found in collection {self.collection.id}."
+                )
+
+            audio.delete()
+            return {
+                "success": True,
+                "message": f"Video {audio.id} deleted successfully",
+            }
+        except ValueError as ve:
+            logging.error(f"ValueError while deleting video: {ve}")
+            raise ve
+        except Exception as e:
+            logging.exception(
+                f"Unexpected error occurred while deleting video {audio_id}"
+            )
+            raise Exception(
+                "An unexpected error occurred while deleting the video. Please try again later."
+            )
+
+    def delete_image(self, image_id):
+        """Delete a specific image by its ID."""
+        if not image_id:
+            raise ValueError("Image ID is required to delete a image.")
+        try:
+            image = self.collection.get_image(image_id)
+            if not image:
+                raise ValueError(
+                    f"Image with ID {image_id} not found in collection {self.collection.id}."
+                )
+
+            image.delete()
+            return {
+                "success": True,
+                "message": f"Image {image_id} deleted successfully",
+            }
+        except ValueError as ve:
+            logging.error(f"ValueError while deleting video: {ve}")
+            raise ve
+        except Exception as e:
+            logging.exception(
+                f"Unexpected error occurred while deleting video {image_id}"
+            )
+            raise Exception(
+                "An unexpected error occurred while deleting the video. Please try again later."
+            )
+