Skip to content

Commit e1231f1

Browse files
committed
Get content after we check cancellation
1 parent cf9cbd8 commit e1231f1

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

Core/Sources/Service/SuggestionCommandHandler/PseudoCommandHandler.swift

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,15 @@ struct PseudoCommandHandler {
4242

4343
@WorkspaceActor
4444
func generateRealtimeSuggestions(sourceEditor: SourceEditor?) async {
45-
// Can't use handler if content is not available.
46-
guard
47-
let editor = await getEditorContent(sourceEditor: sourceEditor),
48-
let filespace = await getFilespace(),
45+
guard let filespace = await getFilespace(),
4946
let (workspace, _) = try? await Service.shared.workspacePool
5047
.fetchOrCreateWorkspaceAndFilespace(fileURL: filespace.fileURL) else { return }
48+
49+
if Task.isCancelled { return }
50+
51+
// Can't use handler if content is not available.
52+
guard let editor = await getEditorContent(sourceEditor: sourceEditor)
53+
else { return }
5154

5255
let fileURL = filespace.fileURL
5356
let presenter = PresentInWindowSuggestionPresenter()
@@ -359,6 +362,7 @@ extension PseudoCommandHandler {
359362
guard let filespace = await getFilespace(),
360363
let sourceEditor = sourceEditor ?? XcodeInspector.shared.focusedEditor
361364
else { return nil }
365+
if Task.isCancelled { return nil }
362366
let content = sourceEditor.getContent()
363367
let uti = filespace.codeMetadata.uti ?? ""
364368
let tabSize = filespace.codeMetadata.tabSize ?? 4

0 commit comments

Comments
 (0)