fix: Searchkit memory leaks (#1735)

Author: tom-ludwig

CodeEdit.xcodeproj/project.xcworkspace/xcshareddata/swiftpm/Package.resolved

@@ -22,7 +22,7 @@ extension SearchIndexer {
         }
 
         return modifyIndexQueue.sync {
-            SKIndexAddDocumentWithText(index, document.takeUnretainedValue(), text as CFString, canReplace)
+            SKIndexAddDocumentWithText(index, document.takeRetainedValue(), text as CFString, canReplace)
         }
     }
 
@@ -66,7 +66,7 @@ extension SearchIndexer {
         let mime = mimeType ?? self.detectMimeType(fileURL)
 
         return modifyIndexQueue.sync {
-            SKIndexAddDocument(index, document.takeUnretainedValue(), mime as CFString?, canReplace)
+            SKIndexAddDocument(index, document.takeRetainedValue(), mime as CFString?, canReplace)
         }
     }
 
@@ -107,7 +107,7 @@ extension SearchIndexer {
     /// - Returns: `true` if the document was successfully removed, `false` otherwise. 
     /// **Note:** If the document didn't exist, this also returns `true`.
     public func removeDocument(url: URL) -> Bool {
-        let document = SKDocumentCreateWithURL(url as CFURL).takeUnretainedValue()
+        let document = SKDocumentCreateWithURL(url as CFURL).takeRetainedValue()
         return self.remove(document: document)
     }
 

CodeEdit/Features/Documents/Indexer/SearchIndexer+Add.swift

@@ -64,25 +64,24 @@ extension SearchIndexer {
 
         var isLeaf = true
 
-        let iterator = SKIndexDocumentIteratorCreate(index, inParentDocument).takeUnretainedValue()
+        let iterator = SKIndexDocumentIteratorCreate(index, inParentDocument).takeRetainedValue()
         while let skDocument = SKIndexDocumentIteratorCopyNext(iterator) {
             isLeaf = false
-            self.addLeafURLs(index: index, inParentDocument: skDocument.takeUnretainedValue(), docs: &docs)
+            self.addLeafURLs(index: index, inParentDocument: skDocument.takeRetainedValue(), docs: &docs)
         }
-
         if isLeaf, inParentDocument != nil,
            kSKDocumentStateNotIndexed != SKIndexGetDocumentState(index, inParentDocument) {
-            if let temp = SKDocumentCopyURL(inParentDocument) {
-                let baseURL = temp.takeUnretainedValue() as URL
-                let documentID = SKIndexGetDocumentID(index, inParentDocument)
-                docs.append(
-                    DocumentID(
-                        url: baseURL,
-                        document: inParentDocument!,
-                        documentID: documentID
-                    )
+            let url = SKDocumentCopyURL(inParentDocument).takeRetainedValue()
+
+            let documentID = SKIndexGetDocumentID(index, inParentDocument)
+            docs.append(
+                DocumentID(
+                    url: url as URL,
+                    document: inParentDocument!,
+                    documentID: documentID
                 )
-            }
+            )
+
         }
     }
 

CodeEdit/Features/Documents/Indexer/SearchIndexer+InternalMethods.swift

@@ -93,7 +93,7 @@ extension SearchIndexer {
 
             let partialResult: [SearchResult] = zip(urls[0..<foundCount], scores)
                 .compactMap { (cfurl, score) -> SearchResult? in
-                    guard let url  = cfurl?.takeUnretainedValue() as URL? else {
+                    guard let url  = cfurl?.takeRetainedValue() as URL? else {
                         return nil
                     }
 

CodeEdit/Features/Documents/Indexer/SearchIndexer+ProgressiveSearch.swift

@@ -67,11 +67,11 @@ extension SearchIndexer {
 
         var result = [TermCount]()
 
-        let document = SKDocumentCreateWithURL(url as CFURL).takeUnretainedValue()
+        let document = SKDocumentCreateWithURL(url as CFURL).takeRetainedValue()
         let documentID = SKIndexGetDocumentID(index, document)
 
         guard let termVals = SKIndexCopyTermIDArrayForDocumentID(index, documentID),
-              let terms = termVals.takeUnretainedValue() as? [CFIndex] else {
+              let terms = termVals.takeRetainedValue() as? [CFIndex] else {
             return []
         }
 

CodeEdit/Features/Documents/Indexer/SearchIndexer+Terms.swift

@@ -67,7 +67,7 @@ public class SearchIndexer {
     private(set) lazy var stopWords: Set<String> = {
         var stopWords: Set<String> = []
         if let index = self.index,
-           let properties = SKIndexGetAnalysisProperties(self.index).takeUnretainedValue() as? [String: Any],
+           let properties = SKIndexGetAnalysisProperties(self.index).takeRetainedValue() as? [String: Any],
            let newStopWords = properties[kSKStopWords as String] as? Set<String> {
             stopWords = newStopWords
         }