Taking the actual opened tabs into account when cleaning up filespaces

Author: intitni

Core/Sources/AXExtension/AXUIElement.swift

@@ -28,10 +28,16 @@ public extension AXUIElement {
         try? copyValue(key: kAXDocumentAttribute)
     }
 
+    /// Label in Accessibility Inspector.
     var description: String {
         (try? copyValue(key: kAXDescriptionAttribute)) ?? ""
     }
 
+    /// Type in Accessibility Inspector.
+    var roleDescription: String {
+        (try? copyValue(key: kAXRoleDescriptionAttribute)) ?? ""
+    }
+    
     var label: String {
         (try? copyValue(key: kAXLabelValueAttribute)) ?? ""
     }
@@ -158,6 +164,29 @@ public extension AXUIElement {
         }
         return nil
     }
+    
+    func children(where match: (AXUIElement) -> Bool) -> [AXUIElement] {
+        var all = [AXUIElement]()
+        for child in children {
+            if match(child) { all.append(child) }
+        }
+        for child in children {
+            all.append(contentsOf: child.children(where: match))
+        }
+        return all
+    }
+    
+    func firstChild(where match: (AXUIElement) -> Bool) -> AXUIElement? {
+        for child in children {
+            if match(child) { return child }
+        }
+        for child in children {
+            if let target = child.firstChild(where: match) {
+                return target
+            }
+        }
+        return nil
+    }
 
     func visibleChild(identifier: String) -> AXUIElement? {
         for child in visibleChildren {

Core/Sources/Service/ScheduledCleaner.swift

@@ -1,3 +1,6 @@
+import ActiveApplicationMonitor
+import AppKit
+import AXExtension
 import Foundation
 
 public final class ScheduledCleaner {
@@ -6,22 +9,47 @@ public final class ScheduledCleaner {
         Task { @ServiceActor in
             while !Task.isCancelled {
                 try await Task.sleep(nanoseconds: 2 * 60 * 60 * 1_000_000_000)
+                let availableTabs = findAvailableOpenedTabs()
                 for (url, workspace) in workspaces {
                     if workspace.isExpired {
                         workspaces[url] = nil
                     } else {
                         // cleanup chats for unused files
                         let filespaces = workspace.filespaces
-                        for (url, filespace) in filespaces {
-                            if filespace.isExpired {
+                        for (url, _) in filespaces {
+                            if workspace.isFilespaceExpired(
+                                fileURL: url,
+                                availableTabs: availableTabs
+                            ) {
                                 WidgetDataSource.shared.cleanup(for: url)
                             }
                         }
                         // cleanup workspace
-                        workspace.cleanUp()
+                        workspace.cleanUp(availableTabs: availableTabs)
                     }
                 }
             }
         }
     }
+
+    func findAvailableOpenedTabs() -> Set<String> {
+        guard let xcode = ActiveApplicationMonitor.latestXcode else { return [] }
+        let app = AXUIElementCreateApplication(xcode.processIdentifier)
+        let windows = app.windows.filter { $0.identifier == "Xcode.WorkspaceWindow" }
+        guard !windows.isEmpty else { return [] }
+        var allTabs = Set<String>()
+        for window in windows {
+            guard let editArea = window.firstChild(where: { $0.description == "editor area" })
+            else { continue }
+            let tabBars = editArea.children { $0.description == "tab bar" }
+            for tabBar in tabBars {
+                let tabs = tabBar.children { $0.roleDescription == "tab" }
+                for tab in tabs {
+                    allTabs.insert(tab.title)
+                }
+            }
+        }
+        return allTabs
+    }
 }
+

Core/Sources/Service/Workspace.swift

@@ -325,6 +325,7 @@ extension Workspace {
     }
 
     func notifyOpenFile(filespace: Filespace) {
+        lastTriggerDate = Environment.now()
         Task {
             try await copilotSuggestionService?.notifyOpenTextDocument(
                 fileURL: filespace.fileURL,
@@ -334,6 +335,8 @@ extension Workspace {
     }
 
     func notifyUpdateFile(filespace: Filespace, content: String) {
+        filespace.refreshUpdateTime()
+        lastTriggerDate = Environment.now()
         Task {
             try await copilotSuggestionService?.notifyChangeTextDocument(
                 fileURL: filespace.fileURL,
@@ -343,23 +346,32 @@ extension Workspace {
     }
 
     func notifySaveFile(filespace: Filespace) {
+        filespace.refreshUpdateTime()
+        lastTriggerDate = Environment.now()
         Task {
             try await copilotSuggestionService?.notifySaveTextDocument(fileURL: filespace.fileURL)
         }
     }
 }
 
 extension Workspace {
-    func cleanUp() {
-        for (fileURL, filespace) in filespaces {
-            if filespace.isExpired {
+    func cleanUp(availableTabs: Set<String>) {
+        for (fileURL, _) in filespaces {
+            if isFilespaceExpired(fileURL: fileURL, availableTabs: availableTabs) {
                 Task {
                     try await copilotSuggestionService?.notifyCloseTextDocument(fileURL: fileURL)
                 }
                 filespaces[fileURL] = nil
             }
         }
     }
+    
+    func isFilespaceExpired(fileURL: URL, availableTabs: Set<String>) -> Bool {
+        let filename = fileURL.lastPathComponent
+        if availableTabs.contains(filename) { return false }
+        guard let filespace = filespaces[fileURL] else { return true }
+        return filespace.isExpired
+    }
 
     func cancelInFlightRealtimeSuggestionRequests() {
         for task in realtimeSuggestionRequests {