From f5b51642777ea8ab9ac63fd0a48ef768faf86d95 Mon Sep 17 00:00:00 2001
From: Christiaan Bloemendaal <github@tnrd.net>
Date: Wed, 31 Jan 2024 23:03:50 +0100
Subject: [PATCH] fix: null reference on extra elements

---
 .../Database/Documents/RequestedChapterDocument.cs         | 7 ++++++-
 .../Database/Documents/RequestedMangaDocument.cs           | 7 ++++++-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/Mangarr.Stack/Database/Documents/RequestedChapterDocument.cs b/src/Mangarr.Stack/Database/Documents/RequestedChapterDocument.cs
index b73f553..bdae6b6 100644
--- a/src/Mangarr.Stack/Database/Documents/RequestedChapterDocument.cs
+++ b/src/Mangarr.Stack/Database/Documents/RequestedChapterDocument.cs
@@ -8,7 +8,7 @@ public class RequestedChapterDocument : DocumentBase<RequestedChapterDocument>,
     public const int CurrentVersion = 1;
 
     // ReSharper disable once FieldCanBeMadeReadOnly.Local
-    [BsonExtraElements] private Dictionary<string, object> _extraElements = null!;
+    [BsonExtraElements] private Dictionary<string, object>? _extraElements;
 
     public required string RequestedMangaId { get; set; }
     public required string ChapterId { get; set; } = null!;
@@ -27,6 +27,11 @@ void ISupportInitialize.BeginInit()
 
     void ISupportInitialize.EndInit()
     {
+        if (_extraElements == null)
+        {
+            return;
+        }
+
         if (!_extraElements.TryGetValue("CreationDate", out object? creationDate))
         {
             return;
diff --git a/src/Mangarr.Stack/Database/Documents/RequestedMangaDocument.cs b/src/Mangarr.Stack/Database/Documents/RequestedMangaDocument.cs
index 040f9c5..a94e932 100644
--- a/src/Mangarr.Stack/Database/Documents/RequestedMangaDocument.cs
+++ b/src/Mangarr.Stack/Database/Documents/RequestedMangaDocument.cs
@@ -8,7 +8,7 @@ public class RequestedMangaDocument : DocumentBase<RequestedMangaDocument>, ISup
     public const int CurrentVersion = 1;
 
     // ReSharper disable once FieldCanBeMadeReadOnly.Local
-    [BsonExtraElements] private Dictionary<string, object> _extraElements = null!;
+    [BsonExtraElements] private Dictionary<string, object>? _extraElements;
 
     public required int SearchId { get; set; }
     public required string SourceId { get; set; } = null!;
@@ -26,6 +26,11 @@ void ISupportInitialize.BeginInit()
 
     void ISupportInitialize.EndInit()
     {
+        if (_extraElements == null)
+        {
+            return;
+        }
+
         if (!_extraElements.TryGetValue("CreationDate", out object? creationDate))
         {
             return;