From 64fa3c7a6bc083773cd6d1c77acf031bd9649312 Mon Sep 17 00:00:00 2001
From: Jake Bailey <5341706+jakebailey@users.noreply.github.com>
Date: Mon, 3 Jun 2024 12:15:02 -0700
Subject: [PATCH] Check cancellation token in isolatedDeclarations codefix
 (#58755)

---
 src/services/codefixes/fixMissingTypeAnnotationOnExports.ts | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/services/codefixes/fixMissingTypeAnnotationOnExports.ts b/src/services/codefixes/fixMissingTypeAnnotationOnExports.ts
index 70ceb0c31eb..440fc4d05c4 100644
--- a/src/services/codefixes/fixMissingTypeAnnotationOnExports.ts
+++ b/src/services/codefixes/fixMissingTypeAnnotationOnExports.ts
@@ -254,6 +254,8 @@ function withContext<T>(
     };
 
     function addTypeAnnotation(span: TextSpan) {
+        context.cancellationToken.throwIfCancellationRequested();
+
         const nodeWithDiag = getTokenAtPosition(sourceFile, span.start);
 
         const expandoFunction = findExpandoFunction(nodeWithDiag);
@@ -331,6 +333,8 @@ function withContext<T>(
     }
 
     function addInlineAssertion(span: TextSpan): DiagnosticOrDiagnosticAndArguments | undefined {
+        context.cancellationToken.throwIfCancellationRequested();
+
         const nodeWithDiag = getTokenAtPosition(sourceFile, span.start);
         const expandoFunction = findExpandoFunction(nodeWithDiag);
         // No inline assertions for expando members
@@ -406,6 +410,8 @@ function withContext<T>(
     }
 
     function extractAsVariable(span: TextSpan): DiagnosticOrDiagnosticAndArguments | undefined {
+        context.cancellationToken.throwIfCancellationRequested();
+
         const nodeWithDiag = getTokenAtPosition(sourceFile, span.start);
         const targetNode = findBestFittingNode(nodeWithDiag, span) as Expression;
         if (!targetNode || isValueSignatureDeclaration(targetNode) || isValueSignatureDeclaration(targetNode.parent)) return;