Fix WorkspaceDocument Leak (#1808)

Author: thecoolwinter

CodeEdit.xcodeproj/project.pbxproj

@@ -13,7 +13,7 @@ final class AppDelegate: NSObject, NSApplicationDelegate, ObservableObject {
     private let updater = SoftwareUpdater()
 
     @Environment(\.openWindow)
-    private var openWindow
+    var openWindow
 
     func applicationDidFinishLaunching(_ notification: Notification) {
         setupServiceContainer()
@@ -209,8 +209,8 @@ final class AppDelegate: NSObject, NSApplicationDelegate, ObservableObject {
             defaults.removeObject(forKey: "openInCEFiles")
         }
 
-        DispatchQueue.main.asyncAfter(deadline: .now() + 1) {
-            self.checkForFilesToOpen()
+        DispatchQueue.main.asyncAfter(deadline: .now() + 1) { [weak self] in
+            self?.checkForFilesToOpen()
         }
     }
 

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

@@ -85,11 +85,11 @@ final class CEWorkspaceFile: Codable, Comparable, Hashable, Identifiable, Editor
     /// Returns a parent ``CEWorkspaceFile``.
     ///
     /// If the item already is the top-level ``CEWorkspaceFile`` this returns `nil`.
-    var parent: CEWorkspaceFile?
+    weak var parent: CEWorkspaceFile?
 
     private let fileDocumentSubject = PassthroughSubject<CodeFileDocument?, Never>()
 
-    var fileDocument: CodeFileDocument? {
+    weak var fileDocument: CodeFileDocument? {
         didSet {
             fileDocumentSubject.send(fileDocument)
         }

CodeEdit/AppDelegate.swift

@@ -387,6 +387,7 @@ final class CEWorkspaceFileManager {
     }
 
     deinit {
+        fsEventStream?.cancel()
         observers.removeAllObjects()
     }
 }

CodeEdit/Features/CEWorkspace/Models/CEWorkspaceFile.swift

@@ -10,20 +10,12 @@ import Combine
 
 /// The CodeEdit workspace settings model.
 final class CEWorkspaceSettings: ObservableObject {
-    @ObservedObject private var workspace: WorkspaceDocument
     @Published public var preferences: CEWorkspaceSettingsData = .init()
 
-    private var storeTask: AnyCancellable!
+    private var storeTask: AnyCancellable?
     private let fileManager = FileManager.default
 
-    private var folderURL: URL? {
-        guard let workspaceURL = workspace.fileURL else {
-            return nil
-        }
-
-        return workspaceURL
-            .appendingPathComponent(".codeedit", isDirectory: true)
-    }
+    private(set) var folderURL: URL?
 
     private var settingsURL: URL? {
         folderURL?
@@ -32,15 +24,23 @@ final class CEWorkspaceSettings: ObservableObject {
     }
 
     init(workspaceDocument: WorkspaceDocument) {
-        self.workspace = workspaceDocument
-
+        folderURL = workspaceDocument.fileURL?.appendingPathComponent(".codeedit", isDirectory: true)
         loadSettings()
 
         self.storeTask = self.$preferences.throttle(for: 2.0, scheduler: RunLoop.main, latest: true).sink {
             try? self.savePreferences($0)
         }
     }
 
+    func cleanUp() {
+        storeTask?.cancel()
+        storeTask = nil
+    }
+
+    deinit {
+        cleanUp()
+    }
+
     /// Load and construct ``CEWorkspaceSettings`` model from `.codeedit/settings.json`
     private func loadSettings() {
         if let settingsURL = settingsURL {

CodeEdit/Features/CEWorkspace/Models/CEWorkspaceFileManager.swift

@@ -11,8 +11,8 @@ import Combine
 
 /// A view that pops up a branch picker.
 struct ToolbarBranchPicker: View {
-    private var workspaceFileManager: CEWorkspaceFileManager?
-    private var sourceControlManager: SourceControlManager?
+    private weak var workspaceFileManager: CEWorkspaceFileManager?
+    private weak var sourceControlManager: SourceControlManager?
 
     @Environment(\.controlActiveState)
     private var controlActive

CodeEdit/Features/CEWorkspace/Models/CEWorkspaceSettings.swift

@@ -31,7 +31,7 @@ struct QuickActionsView: View {
 
     func callHandler(command: Command) {
         closePalette()
-        command.closureWrapper.call()
+        command.closureWrapper()
         selectedItem = nil
         state.commandQuery = ""
         state.filteredCommands = []