Added correct error handling for non-existent media object when deleting mediadev

1 files changed, 5 insertions, 1 deletions

diff --git a/Services/MediaService.cs b/Services/MediaService.cs
index 2d1533c..0ec5130 100644
--- a/Services/MediaService.cs
+++ b/Services/MediaService.cs
@@ -3,6 +3,7 @@ using ImageMagick;
 using Microsoft.EntityFrameworkCore;
 using MimeDetective;
 using MimeDetective.Definitions;
+using System.Diagnostics.CodeAnalysis;
 using System.Security.Cryptography;
 using System.Text.RegularExpressions;
 
@@ -230,7 +231,10 @@ public class MediaService : IMediaService {
 
     public void Delete(Guid media) {
         using var db = dbFactory.CreateDbContext();
-        var m = db.Media.First(m => m.Guid == media);
+        var m = db.Media.FirstOrDefault(m => m.Guid == media);
+
+        if(m is null)
+            throw new ObjectNotFoundException([ media ]);
 
         var path = Path.Join(
             config.MediaBasePath,