From a8a34f7f96a02d668c136415b29c2d622a2546e5 Mon Sep 17 00:00:00 2001
From: Jesse Plamondon-Willard <github@jplamondonw.com>
Date: Sat, 24 Feb 2018 20:24:42 -0500
Subject: fix event error in rare cases

---
 src/SMAPI/Framework/Events/ManagedEventBase.cs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'src/SMAPI/Framework/Events/ManagedEventBase.cs')

diff --git a/src/SMAPI/Framework/Events/ManagedEventBase.cs b/src/SMAPI/Framework/Events/ManagedEventBase.cs
index cc4d89ec..7e42d613 100644
--- a/src/SMAPI/Framework/Events/ManagedEventBase.cs
+++ b/src/SMAPI/Framework/Events/ManagedEventBase.cs
@@ -55,7 +55,7 @@ namespace StardewModdingAPI.Framework.Events
         protected void AddTracking(TEventHandler handler, IEnumerable<TEventHandler> invocationList)
         {
             this.SourceMods[handler] = this.ModRegistry.GetFromStack();
-            this.CachedInvocationList = invocationList.ToArray();
+            this.CachedInvocationList = invocationList?.ToArray() ?? new TEventHandler[0];
         }
 
         /// <summary>Remove tracking for an event handler.</summary>
@@ -63,8 +63,9 @@ namespace StardewModdingAPI.Framework.Events
         /// <param name="invocationList">The updated event invocation list.</param>
         protected void RemoveTracking(TEventHandler handler, IEnumerable<TEventHandler> invocationList)
         {
-            this.SourceMods.Remove(handler);
-            this.CachedInvocationList = invocationList.ToArray();
+            this.CachedInvocationList = invocationList?.ToArray() ?? new TEventHandler[0];
+            if(!this.CachedInvocationList.Contains(handler)) // don't remove if there's still a reference to the removed handler (e.g. it was added twice and removed once)
+                this.SourceMods.Remove(handler);
         }
 
         /// <summary>Log an exception from an event handler.</summary>
-- 
cgit