Harden media and follow-up handling

2026-07-11 03:16:11 -05:00
parent 9c8775d740
commit 0f9bd2fa78
3 changed files with 11 additions and 3 deletions
@@ -1,4 +1,5 @@
 using MongoDB.Driver;
+using MongoDB.Bson;
 using QuestionsApi.Models;

 namespace QuestionsApi.Services;
@@ -29,6 +30,11 @@ public class QuestionMediaStore
        return asset;
    }

-    public async Task<QuestionMediaAsset?> GetAsync(string id) =>
-        await _col.Find(asset => asset.Id == id).FirstOrDefaultAsync();
+    public async Task<QuestionMediaAsset?> GetAsync(string id)
+    {
+        if (!ObjectId.TryParse(id, out _))
+            return null;
+
+        return await _col.Find(asset => asset.Id == id).FirstOrDefaultAsync();
+    }
 }
@@ -1,4 +1,5 @@
 using MongoDB.Driver;
+using MongoDB.Bson;
 using QuestionsApi.Models;

 namespace QuestionsApi.Services;
@@ -102,6 +103,7 @@ public class QuestionStore
            }
            else
            {
+                question.Id = ObjectId.GenerateNewId().ToString();
                question.CreatedUtc = DateTime.UtcNow;
            }