use real cancellation token

Author: zardoy

typescript/src/getPatchedNavTree.ts

@@ -1,4 +1,4 @@
-import { isTs5, nodeModules } from './utils'
+import { getCancellationToken, isTs5, nodeModules } from './utils'
 import { createLanguageService } from './dummyLanguageService'
 import { getCannotFindCodes } from './utils/cannotFindCodes'
 
@@ -168,9 +168,5 @@ export const getNavTreeItems = (info: ts.server.PluginCreateInfo, fileName: stri
     const sourceFile = program?.getSourceFile(fileName)
     if (!sourceFile) throw new Error('no sourceFile')
 
-    const cancellationToken = info.languageServiceHost.getCompilerHost?.()?.getCancellationToken?.() ?? {
-        isCancellationRequested: () => false,
-        throwIfCancellationRequested: () => {},
-    }
-    return navModule.getNavigationTree(sourceFile, cancellationToken)
+    return navModule.getNavigationTree(sourceFile, getCancellationToken(info.languageServiceHost))
 }

typescript/src/utils.ts

@@ -167,6 +167,30 @@ export function addObjectMethodResultInterceptors<T extends Record<string, any>>
     }
 }
 
+// have absolutely no idea why such useful utility is not publicly available
+export const getChangesTracker = formatOptions => {
+    return new tsFull.textChanges.ChangeTracker(/* will be normalized by vscode anyway */ '\n', tsFull.formatting.getFormatContext(formatOptions, {}))
+}
+
+export const getCancellationToken = (languageServiceHost: ts.LanguageServiceHost) => {
+    let cancellationToken = languageServiceHost.getCancellationToken?.() as ts.CancellationToken | undefined
+    // if (!cancellationToken) {
+    //     debugger
+    // }
+    cancellationToken ??= {
+        isCancellationRequested: () => false,
+        throwIfCancellationRequested: () => {},
+    }
+    if (!cancellationToken.throwIfCancellationRequested) {
+        cancellationToken.throwIfCancellationRequested = () => {
+            if (cancellationToken!.isCancellationRequested()) {
+                throw new ts.OperationCanceledException()
+            }
+        }
+    }
+    return cancellationToken
+}
+
 const wordRangeAtPos = (text: string, position: number) => {
     const isGood = (pos: number) => {
         return /[-\w\d]/i.test(text.at(pos) ?? '')

typescript/src/workspaceSymbolSearch.ts

@@ -1,4 +1,5 @@
 import { GetConfig } from './types'
+import { getCancellationToken } from './utils'
 
 export default (proxy: ts.LanguageService, languageService: ts.LanguageService, c: GetConfig, languageServiceHost: ts.LanguageServiceHost) => {
     proxy.getNavigateToItems = (searchValue, maxResultCount, fileName, excludeDtsFiles) => {
@@ -8,10 +9,7 @@ export default (proxy: ts.LanguageService, languageService: ts.LanguageService,
         }
 
         const program = languageService.getProgram()!
-        const cancellationToken = languageServiceHost.getCompilerHost?.()?.getCancellationToken?.() ?? {
-            isCancellationRequested: () => false,
-            throwIfCancellationRequested: () => {},
-        }
+
         let sourceFiles = fileName ? [program.getSourceFile(fileName)!] : program.getSourceFiles()
         if (!fileName) {
             const excludes = tsFull.getRegularExpressionForWildcard(workspaceSymbolSearchExcludePatterns, '', 'exclude')?.slice(1)
@@ -23,8 +21,7 @@ export default (proxy: ts.LanguageService, languageService: ts.LanguageService,
         return tsFull.NavigateTo.getNavigateToItems(
             sourceFiles as any,
             program.getTypeChecker() as any,
-            // TODO! use real cancellationToken
-            cancellationToken,
+            getCancellationToken(languageServiceHost),
             searchValue,
             maxResultCount,
             excludeDtsFiles ?? false,