Support proService clean up

Author: intitni

Core/Sources/Service/ScheduledCleaner.swift

@@ -7,16 +7,9 @@ import Workspace
 import XcodeInspector
 
 public final class ScheduledCleaner {
-    let workspacePool: WorkspacePool
-    let guiController: GraphicalUserInterfaceController
+    weak var service: Service?
 
-    init(
-        workspacePool: WorkspacePool,
-        guiController: GraphicalUserInterfaceController
-    ) {
-        self.workspacePool = workspacePool
-        self.guiController = guiController
-    }
+    init() {}
 
     func start() {
         Task { @ServiceActor in
@@ -38,6 +31,8 @@ public final class ScheduledCleaner {
 
     @ServiceActor
     func cleanUp() async {
+        guard let service else { return }
+        
         let workspaceInfos = XcodeInspector.shared.xcodes.reduce(
             into: [
                 XcodeAppInstanceInspector.WorkspaceIdentifier:
@@ -53,18 +48,18 @@ public final class ScheduledCleaner {
                 }
             }
         }
-        for (url, workspace) in workspacePool.workspaces {
+        for (url, workspace) in service.workspacePool.workspaces {
             if workspace.isExpired, workspaceInfos[.url(url)] == nil {
                 Logger.service.info("Remove idle workspace")
                 _ = await Task { @MainActor in
-                    guiController.viewStore.send(
+                    service.guiController.viewStore.send(
                         .promptToCodeGroup(.discardExpiredPromptToCode(documentURLs: Array(
                             workspace.filespaces.keys
                         )))
                     )
                 }.result
                 await workspace.cleanUp(availableTabs: [])
-                workspacePool.removeWorkspace(url: url)
+                service.workspacePool.removeWorkspace(url: url)
             } else {
                 let tabs = (workspaceInfos[.url(url)]?.tabs ?? [])
                     .union(workspaceInfos[.unknown]?.tabs ?? [])
@@ -77,7 +72,7 @@ public final class ScheduledCleaner {
                     ) {
                         Logger.service.info("Remove idle filespace")
                         _ = await Task { @MainActor in
-                            guiController.viewStore.send(
+                            service.guiController.viewStore.send(
                                 .promptToCodeGroup(.discardExpiredPromptToCode(documentURLs: [url]))
                             )
                         }.result
@@ -87,11 +82,16 @@ public final class ScheduledCleaner {
                 await workspace.cleanUp(availableTabs: tabs)
             }
         }
+        
+        #if canImport(ProService)
+        await service.proService.cleanUp(workspaceInfos: workspaceInfos)
+        #endif
     }
 
     @ServiceActor
     public func closeAllChildProcesses() async {
-        for (_, workspace) in workspacePool.workspaces {
+        guard let service else { return }
+        for (_, workspace) in service.workspacePool.workspaces {
             await workspace.terminateSuggestionService()
         }
     }

Core/Sources/Service/Service.swift

@@ -38,7 +38,7 @@ public final class Service {
     private init() {
         @Dependency(\.workspacePool) var workspacePool
 
-        scheduledCleaner = .init(workspacePool: workspacePool, guiController: guiController)
+        scheduledCleaner = .init()
         workspacePool.registerPlugin { SuggestionServiceWorkspacePlugin(workspace: $0) }
         self.workspacePool = workspacePool
         globalShortcutManager = .init(guiController: guiController)
@@ -52,6 +52,8 @@ public final class Service {
             ProService()
         }
         #endif
+        
+        scheduledCleaner.service = self
     }
 
     @MainActor