Fixed media description editing

1 files changed, 28 insertions, 1 deletions

diff --git a/Services/MediaService.cs b/Services/MediaService.cs
index 05dd5b0..a8b0406 100644
--- a/Services/MediaService.cs
+++ b/Services/MediaService.cs
@@ -3,6 +3,11 @@
 namespace HyperBooru.Services;
 
 public interface IMediaService {
+    public void SetDescription(
+        Media media,
+        string? shortDescription,
+        string? longDescription);
+
     public void SetIngest(Media media, bool ingest);
 }
 
@@ -18,11 +23,33 @@ public class MediaService : IMediaService {
             .First(td => td.Guid == HBContext.IngestTag);
 
         if(ingest)
-            media.Tags.Add(new() { TagDefinition = ingestTag });
+            media.Tags.Add(new(ingestTag));
         else
             media.Tags.Remove(
                 media.Tags.First(t => t.TagDefinition.Guid == ingestTag.Guid));
 
         db.SaveChanges();
     }
+
+    public void SetDescription(
+        Media media,
+        string? shortDescription,
+        string? longDescription) {
+
+        using var db = dbFactory.CreateDbContext();
+        var m = db.Media.First(m => m.Guid == media.Guid);
+
+        shortDescription = shortDescription?.Trim();
+        longDescription  = longDescription?.Trim();
+
+        if(string.IsNullOrEmpty(shortDescription))
+            shortDescription = null;
+        if(string.IsNullOrEmpty(longDescription))
+            longDescription = null;
+
+        m.ShortDescription = shortDescription;
+        m.LongDescription  = longDescription;
+
+        db.SaveChanges();
+    }
 }