make navtree use the same sourcefile as native impl

Author: zardoy

typescript/src/decorateProxy.ts

@@ -172,7 +172,7 @@ export const decorateLanguageService = (
             // so we forced to communicate via fs
             const config = JSON.parse(ts.sys.readFile(require('path').join(__dirname, '../../plugin-config.json'), 'utf8') ?? '{}')
             proxy.getNavigationTree = fileName => {
-                if (c('patchOutline') || config.patchOutline) return getNavTreeItems(info, fileName, config.outline)
+                if (c('patchOutline') || config.patchOutline) return getNavTreeItems(languageService, languageServiceHost, fileName, config.outline)
                 return languageService.getNavigationTree(fileName)
             }
         }

typescript/src/getPatchedNavTree.ts

@@ -160,13 +160,17 @@ const getPatchedNavModule = (additionalFeatures: AdditionalFeatures): { getNavig
 
 let navModule: { getNavigationTree: any }
 
-export const getNavTreeItems = (info: ts.server.PluginCreateInfo, fileName: string, additionalFeatures: AdditionalFeatures) => {
+export const getNavTreeItems = (
+    languageService: ts.LanguageService,
+    languageServiceHost: ts.LanguageServiceHost,
+    fileName: string,
+    additionalFeatures: AdditionalFeatures,
+) => {
     if (!navModule) navModule = getPatchedNavModule(additionalFeatures)
-    const program = info.languageService.getProgram()
-    if (!program) throw new Error('no program')
-    // sourceFile is unreliable to use here, as it's not the same as the one used within original getNavigationTree
-    const sourceFile = program?.getSourceFile(fileName)
+    const sourceFile =
+        (languageService as import('typescript-full').LanguageService).getNonBoundSourceFile?.(fileName) ??
+        languageService.getProgram()!.getSourceFile(fileName)
     if (!sourceFile) throw new Error('no sourceFile')
 
-    return navModule.getNavigationTree(sourceFile, getCancellationToken(info.languageServiceHost))
+    return navModule.getNavigationTree(sourceFile, getCancellationToken(languageServiceHost))
 }

typescript/test/other.spec.ts

@@ -57,7 +57,7 @@ test('Patched navtree (outline)', () => {
             </Notification>
         }
     `)
-    const navTreeItems: ts.NavigationTree = getNavTreeItems({ languageService, languageServiceHost: {} } as any, entrypoint, {
+    const navTreeItems: ts.NavigationTree = getNavTreeItems(languageService, {} as any, entrypoint, {
         arraysTuplesNumberedItems: false,
     })
     const simplify = (items: ts.NavigationTree[]) => {