Update screenshots of saves when saving

backend/endpoints/saves.py

@@ -95,7 +95,7 @@ async def add_save(
         else:
             scanned_screenshot.rom_id = rom.id
             scanned_screenshot.user_id = request.user.id
-            db_screenshot = db_screenshot_handler.add_screenshot(
+            db_screenshot_handler.add_screenshot(
                 screenshot=scanned_screenshot
             )
 
@@ -154,6 +154,36 @@ async def update_save(request: Request, id: int) -> SaveSchema:
         db_save = db_save_handler.update_save(
             db_save.id, {"file_size_bytes": saveFile.size}
         )
+
+    screenshotFile: UploadFile | None = data.get("screenshotFile", None)  # type: ignore
+    if screenshotFile and screenshotFile.filename:
+        screenshots_path = fs_asset_handler.build_screenshots_file_path(
+            user=request.user, platform_fs_slug=db_save.rom.platform_slug
+        )
+
+        fs_asset_handler.write_file(file=screenshotFile, path=screenshots_path)
+
+        # Scan or update screenshot
+        scanned_screenshot = scan_screenshot(
+            file_name=screenshotFile.filename,
+            user=request.user,
+            platform_fs_slug=db_save.rom.platform_slug,
+        )
+        db_screenshot = db_screenshot_handler.get_screenshot_by_filename(
+            rom_id=db_save.rom.id, user_id=request.user.id, file_name=screenshotFile.filename
+        )
+        if db_screenshot:
+            db_screenshot = db_screenshot_handler.update_screenshot(
+                db_screenshot.id,
+                {"file_size_bytes": scanned_screenshot.file_size_bytes},
+            )
+        else:
+            scanned_screenshot.rom_id = db_save.rom.id
+            scanned_screenshot.user_id = request.user.id
+            db_screenshot_handler.add_screenshot(
+                screenshot=scanned_screenshot
+            )
+
 
     # Set the last played time for the current user
     rom_user = db_rom_handler.get_rom_user(db_save.rom_id, request.user.id)

frontend/src/services/api/save.ts

@@ -44,12 +44,17 @@ async function uploadSaves({
 async function updateSave({
   save,
   saveFile,
+  screenshotFile,
 }: {
   save: SaveSchema;
   saveFile: File;
+  screenshotFile?: File;
 }): Promise<{ data: SaveSchema }> {
   const formData = new FormData();
   formData.append("saveFile", saveFile);
+  if (screenshotFile) {
+    formData.append("screenshotFile", screenshotFile);
+  }
 
   return api.put(`/saves/${save.id}`, formData);
 }

frontend/src/views/Player/EmulatorJS/utils.ts

@@ -72,6 +72,12 @@ export async function saveSave({
         saveFile: new File([saveFile], save.file_name, {
           type: "application/octet-stream",
         }),
+        screenshotFile:
+          screenshotFile && save.screenshot
+            ? new File([screenshotFile], save.screenshot.file_name, {
+                type: "application/octet-stream",
+              })
+            : undefined,
       });
 
       // Update the save in the rom object