get collections for detailed rom info

backend/endpoints/responses/rom.py

@@ -5,6 +5,7 @@
 from typing import NotRequired, get_type_hints
 
 from endpoints.responses.assets import SaveSchema, ScreenshotSchema, StateSchema
+from endpoints.responses.collection import CollectionSchema
 from fastapi import Request
 from fastapi.responses import StreamingResponse
 from handler.metadata.igdb_handler import IGDBMetadata
@@ -147,6 +148,7 @@ class DetailedRomSchema(RomSchema):
     user_states: list[StateSchema] = Field(default_factory=list)
     user_screenshots: list[ScreenshotSchema] = Field(default_factory=list)
     user_notes: list[UserNotesSchema] = Field(default_factory=list)
+    collections: list[CollectionSchema] = Field(default_factory=list)
 
     @classmethod
     def from_orm_with_request(cls, db_rom: Rom, request: Request) -> DetailedRomSchema:
@@ -169,6 +171,9 @@ def from_orm_with_request(cls, db_rom: Rom, request: Request) -> DetailedRomSche
             for s in db_rom.screenshots
             if s.user_id == user_id
         ]
+        rom.collections = [
+            CollectionSchema.model_validate(c) for c in db_rom.get_collections()
+        ]
 
         return rom
 

backend/handler/database/roms_handler.py

@@ -179,6 +179,20 @@ def get_sibling_roms(
             )
         ).all()
 
+    @begin_session
+    def get_rom_collections(
+        self, rom_id: int, session: Session = None
+    ) -> list[Collection]:
+        return (
+            session.scalars(
+                select(Collection)
+                .filter(Collection.roms.contains([rom_id]))
+                .order_by(Collection.name.asc())
+            )
+            .unique()
+            .all()
+        )
+
     @begin_session
     def update_rom(self, id: int, data: dict, session: Session = None) -> Rom:
         return session.execute(

backend/models/rom.py

@@ -11,6 +11,7 @@
 
 if TYPE_CHECKING:
     from models.assets import Save, Screenshot, State
+    from models.collection import Collection
     from models.platform import Platform
     from models.user import User
 
@@ -103,6 +104,11 @@ def get_sibling_roms(self) -> list[Rom]:
 
         return db_rom_handler.get_sibling_roms(self)
 
+    def get_collections(self) -> list[Collection]:
+        from handler.database import db_rom_handler
+
+        return db_rom_handler.get_rom_collections(self.id)
+
     # Metadata fields
     @property
     def alternative_names(self) -> list[str]: