From fb11e3646799c97e66752005ab288b3d0326eaac Mon Sep 17 00:00:00 2001
From: Shadowfacts <me@shadowfacts.net>
Date: Wed, 24 Jun 2020 15:42:56 -0400
Subject: [PATCH] Fix crash when displaying search results

---
 .../Search/SearchResultsViewController.swift     | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/Tusker/Screens/Search/SearchResultsViewController.swift b/Tusker/Screens/Search/SearchResultsViewController.swift
index d91522f5..4e51c686 100644
--- a/Tusker/Screens/Search/SearchResultsViewController.swift
+++ b/Tusker/Screens/Search/SearchResultsViewController.swift
@@ -134,14 +134,18 @@ class SearchResultsViewController: EnhancedTableViewController {
             var snapshot = NSDiffableDataSourceSnapshot<Section, Item>()
             
             self.mastodonController.persistentContainer.performBatchUpdates({ (context, addAccounts, addStatuses) in
-                oldSnapshot.itemIdentifiers(inSection: .accounts).forEach { (item) in
-                    guard case let .account(id) = item else { return }
-                    self.mastodonController.persistentContainer.account(for: id, in: context)?.decrementReferenceCount()
+                if oldSnapshot.indexOfSection(.accounts) != nil {
+                    oldSnapshot.itemIdentifiers(inSection: .accounts).forEach { (item) in
+                        guard case let .account(id) = item else { return }
+                        self.mastodonController.persistentContainer.account(for: id, in: context)?.decrementReferenceCount()
+                    }
                 }
                 
-                oldSnapshot.itemIdentifiers(inSection: .statuses).forEach { (item) in
-                    guard case let .status(id, _) = item else { return }
-                    self.mastodonController.persistentContainer.status(for: id, in: context)?.decrementReferenceCount()
+                if oldSnapshot.indexOfSection(.statuses) != nil {
+                    oldSnapshot.itemIdentifiers(inSection: .statuses).forEach { (item) in
+                        guard case let .status(id, _) = item else { return }
+                        self.mastodonController.persistentContainer.status(for: id, in: context)?.decrementReferenceCount()
+                    }
                 }
                 
                 if self.onlySections.contains(.accounts) && !results.accounts.isEmpty {