perf: auto import cache for svelte-kit language service proxy (#2513)

#2464

getPackageJsonsVisibleToFile and getGlobalTypingsCacheLocation are singleton so this should definitely be fine. The proxy language service has the exact list of entry files and the auto-typing also doesn't add any export so it should be fine to just proxy these methods.

Author: jasonlyu123

packages/typescript-plugin/src/language-service/sveltekit.ts

@@ -10,6 +10,22 @@ interface KitSnapshot {
     addedCode: InternalHelpers.AddedCode[];
 }
 
+declare module 'typescript/lib/tsserverlibrary' {
+    interface LanguageServiceHost {
+        /** @internal */ getCachedExportInfoMap?(): unknown;
+        /** @internal */ getModuleSpecifierCache?(): unknown;
+        /** @internal */ getGlobalTypingsCacheLocation?(): string | undefined;
+        /** @internal */ getSymlinkCache?(files: readonly ts.SourceFile[]): unknown;
+        /** @internal */ getPackageJsonsVisibleToFile?(
+            fileName: string,
+            rootDir?: string
+        ): readonly unknown[];
+        /** @internal */ getPackageJsonAutoImportProvider?(): ts.Program | undefined;
+
+        /** @internal*/ getModuleResolutionCache?(): ts.ModuleResolutionCache;
+    }
+}
+
 const cache = new WeakMap<
     ts.server.PluginCreateInfo,
     {
@@ -719,6 +735,51 @@ function getProxiedLanguageService(info: ts.server.PluginCreateInfo, ts: _ts, lo
             ? (...args: Parameters<NonNullable<ts.LanguageServiceHost['realpath']>>) =>
                   originalLanguageServiceHost.realpath!(...args)
             : undefined;
+
+        getProjectReferences = originalLanguageServiceHost.getProjectReferences
+            ? () => originalLanguageServiceHost.getProjectReferences!()
+            : undefined;
+
+        getParsedCommandLine = originalLanguageServiceHost.getParsedCommandLine
+            ? (fileName: string) => originalLanguageServiceHost.getParsedCommandLine!(fileName)
+            : undefined;
+
+        getCachedExportInfoMap = originalLanguageServiceHost.getCachedExportInfoMap
+            ? () => originalLanguageServiceHost.getCachedExportInfoMap!()
+            : undefined;
+
+        getModuleSpecifierCache = originalLanguageServiceHost.getModuleSpecifierCache
+            ? () => originalLanguageServiceHost.getModuleSpecifierCache!()
+            : undefined;
+
+        getGlobalTypingsCacheLocation = originalLanguageServiceHost.getGlobalTypingsCacheLocation
+            ? () => originalLanguageServiceHost.getGlobalTypingsCacheLocation!()
+            : undefined;
+
+        getSymlinkCache = originalLanguageServiceHost.getSymlinkCache
+            ? (...args: any[]) =>
+                  originalLanguageServiceHost.getSymlinkCache!(
+                      // @ts-ignore
+                      ...args
+                  )
+            : undefined;
+
+        getPackageJsonsVisibleToFile = originalLanguageServiceHost.getPackageJsonsVisibleToFile
+            ? (...args: any[]) =>
+                  originalLanguageServiceHost.getPackageJsonsVisibleToFile!(
+                      // @ts-ignore
+                      ...args
+                  )
+            : undefined;
+
+        getPackageJsonAutoImportProvider =
+            originalLanguageServiceHost.getPackageJsonAutoImportProvider
+                ? () => originalLanguageServiceHost.getPackageJsonAutoImportProvider!()
+                : undefined;
+
+        getModuleResolutionCache = originalLanguageServiceHost.getModuleResolutionCache
+            ? () => originalLanguageServiceHost.getModuleResolutionCache!()
+            : undefined;
     }
 
     // Ideally we'd create a full Proxy of the language service, but that seems to have cache issues