Merge branch 'feature/fix-codeium-cancellation-error' into develop

Author: intitni

Core/Sources/Service/SuggestionPresenter/PresentInWindowSuggestionPresenter.swift

@@ -28,6 +28,7 @@ struct PresentInWindowSuggestionPresenter {
 
     func presentError(_ error: Error) {
         if error is CancellationError { return }
+        if let urlError = error as? URLError, urlError.code == URLError.cancelled { return }
         Task { @MainActor in
             let controller = GraphicalUserInterfaceController.shared.suggestionWidget
             controller.presentError(error.localizedDescription)

Core/Sources/SuggestionWidget/SuggestionWidgetController.swift

@@ -275,7 +275,7 @@ public final class SuggestionWidgetController: NSObject {
 
 public extension SuggestionWidgetController {
     func suggestCode(fileURL: URL) {
-        widgetViewModel.isProcessing = false
+        widgetViewModel.endIsProcessing()
         Task {
             if let suggestion = await dataSource?.suggestionForFile(at: fileURL) {
                 suggestionPanelViewModel.content = .suggestion(suggestion)
@@ -285,24 +285,28 @@ public extension SuggestionWidgetController {
     }
 
     func discardSuggestion(fileURL: URL) {
-        widgetViewModel.isProcessing = false
+        widgetViewModel.endIsProcessing()
         Task {
             await updateContentForActiveEditor(fileURL: fileURL)
         }
     }
 
     func markAsProcessing(_ isProcessing: Bool) {
-        widgetViewModel.isProcessing = isProcessing
+        if isProcessing {
+            widgetViewModel.markIsProcessing()
+        } else {
+            widgetViewModel.endIsProcessing()
+        }
     }
 
     func presentError(_ errorDescription: String) {
         suggestionPanelViewModel.content = .error(errorDescription)
         suggestionPanelViewModel.isPanelDisplayed = true
-        widgetViewModel.isProcessing = false
+        widgetViewModel.endIsProcessing()
     }
 
     func presentChatRoom(fileURL: URL) {
-        widgetViewModel.isProcessing = false
+        widgetViewModel.endIsProcessing()
         Task {
             if let chat = await dataSource?.chatForFile(at: fileURL) {
                 chatWindowViewModel.chat = chat
@@ -343,14 +347,14 @@ public extension SuggestionWidgetController {
     }
 
     func closeChatRoom(fileURL: URL) {
-        widgetViewModel.isProcessing = false
+        widgetViewModel.endIsProcessing()
         Task {
             await updateContentForActiveEditor(fileURL: fileURL)
         }
     }
 
     func presentPromptToCode(fileURL: URL) {
-        widgetViewModel.isProcessing = false
+        widgetViewModel.endIsProcessing()
         Task {
             if let provider = await dataSource?.promptToCodeForFile(at: fileURL) {
                 suggestionPanelViewModel.content = .promptToCode(provider)
@@ -367,7 +371,7 @@ public extension SuggestionWidgetController {
     }
 
     func discardPromptToCode(fileURL: URL) {
-        widgetViewModel.isProcessing = false
+        widgetViewModel.endIsProcessing()
         Task {
             await updateContentForActiveEditor(fileURL: fileURL)
         }

Core/Sources/SuggestionWidget/WidgetView.swift

@@ -6,9 +6,40 @@ import SwiftUI
 
 @MainActor
 final class WidgetViewModel: ObservableObject {
+    struct IsProcessingCounter {
+        var expirationDate: TimeInterval
+    }
+
+    private var isProcessingCounters = [IsProcessingCounter]()
+    private var cleanupIsProcessingCounterTask: Task<Void, Error>?
     @Published var isProcessing: Bool
     @Published var currentFileURL: URL?
 
+    func markIsProcessing(date: Date = Date()) {
+        let deadline = date.timeIntervalSince1970 + 20
+        isProcessingCounters.append(IsProcessingCounter(expirationDate: deadline))
+        isProcessing = true
+        
+        cleanupIsProcessingCounterTask?.cancel()
+        cleanupIsProcessingCounterTask = Task { [weak self] in
+            try await Task.sleep(nanoseconds: 20 * 1_000_000_000)
+            try Task.checkCancellation()
+            Task { @MainActor [weak self] in
+                guard let self else { return }
+                isProcessingCounters.removeAll()
+                isProcessing = false
+            }
+        }
+    }
+
+    func endIsProcessing(date: Date = Date()) {
+        if !isProcessingCounters.isEmpty {
+            isProcessingCounters.removeFirst()
+        }
+        isProcessingCounters.removeAll(where: { $0.expirationDate < date.timeIntervalSince1970 })
+        isProcessing = !isProcessingCounters.isEmpty
+    }
+
     init(isProcessing: Bool = false) {
         self.isProcessing = isProcessing
     }