For now make sure module resolution cache usage doesnt go past program creation

Author: sheetalkamat

src/compiler/moduleNameResolver.ts

@@ -780,9 +780,9 @@ export function resolvePackageNameToPackageJson(
     containingDirectory: string,
     options: CompilerOptions,
     host: ModuleResolutionHost,
-    cache: ModuleResolutionCache | undefined,
+    cache: PackageJsonInfoCache | undefined,
 ): PackageJsonInfo | undefined {
-    const moduleResolutionState = getTemporaryModuleResolutionState(cache?.getPackageJsonInfoCache(), host, options);
+    const moduleResolutionState = getTemporaryModuleResolutionState(cache, host, options);
 
     return forEachAncestorDirectory(containingDirectory, ancestorDirectory => {
         if (getBaseFileName(ancestorDirectory) !== "node_modules") {
@@ -2193,7 +2193,7 @@ export function getEntrypointsFromPackageJsonInfo(
     packageJsonInfo: PackageJsonInfo,
     options: CompilerOptions,
     host: GetPackageJsonEntrypointsHost,
-    cache: ModuleResolutionCache | undefined,
+    cache: PackageJsonInfoCache | undefined,
     resolveJs?: boolean,
 ): string[] | false {
     if (!resolveJs && packageJsonInfo.contents.resolvedEntrypoints !== undefined) {
@@ -2205,7 +2205,7 @@ export function getEntrypointsFromPackageJsonInfo(
     let entrypoints: string[] | undefined;
     const extensions = Extensions.TypeScript | Extensions.Declaration | (resolveJs ? Extensions.JavaScript : 0);
     const features = getNodeResolutionFeatures(options);
-    const loadPackageJsonMainState = getTemporaryModuleResolutionState(cache?.getPackageJsonInfoCache(), host, options);
+    const loadPackageJsonMainState = getTemporaryModuleResolutionState(cache, host, options);
     loadPackageJsonMainState.conditions = getConditions(options);
     loadPackageJsonMainState.requestContainingDirectory = packageJsonInfo.packageDirectory;
     const mainResolution = loadNodeModuleFromDirectoryWorker(

src/compiler/program.ts

@@ -1869,14 +1869,16 @@ export function createProgram(rootNamesOrOptions: readonly string[] | CreateProg
     resolvedLibProcessing = undefined;
     resolvedModulesProcessing = undefined;
     resolvedTypeReferenceDirectiveNamesProcessing = undefined;
+    const packageJsonCache = moduleResolutionCache?.getPackageJsonInfoCache();
+    moduleResolutionCache = undefined;
 
     const program: Program = {
         getRootFileNames: () => rootNames,
         getSourceFile,
         getSourceFileByPath,
         getSourceFiles: () => files,
         getMissingFilePaths: () => missingFilePaths!, // TODO: GH#18217
-        getModuleResolutionCache: () => moduleResolutionCache,
+        getPackageJsonInfoCache: () => packageJsonCache,
         getFilesByNameMap: () => filesByName,
         getCompilerOptions: () => options,
         getSyntacticDiagnostics,

src/compiler/types.ts

@@ -4686,7 +4686,7 @@ export interface Program extends ScriptReferenceHost {
      */
     getMissingFilePaths(): readonly Path[];
     /** @internal */
-    getModuleResolutionCache(): ModuleResolutionCache | undefined;
+    getPackageJsonInfoCache(): PackageJsonInfoCache | undefined;
     /** @internal */
     getFilesByNameMap(): Map<string, SourceFile | false | undefined>;
 

src/server/project.ts

@@ -832,6 +832,11 @@ export abstract class Project implements LanguageServiceHost, ModuleResolutionHo
         return this.resolutionCache.getModuleResolutionCache();
     }
 
+    /** @internal */
+    getPackageJsonInfoCache() {
+        return this.resolutionCache.getModuleResolutionCache().getPackageJsonInfoCache();
+    }
+
     /** @internal */
     resolveTypeReferenceDirectiveReferences<T extends string | FileReference>(
         typeDirectiveReferences: readonly T[],
@@ -2631,7 +2636,7 @@ export class AutoImportProviderProject extends Project {
                     hostProject.currentDirectory,
                     compilerOptions,
                     host,
-                    program.getModuleResolutionCache(),
+                    program.getPackageJsonInfoCache(),
                 );
                 if (packageJson) {
                     const entrypoints = getRootNamesFromPackageJson(packageJson, program, symlinkCache);
@@ -2651,7 +2656,7 @@ export class AutoImportProviderProject extends Project {
                             directory,
                             compilerOptions,
                             host,
-                            program.getModuleResolutionCache(),
+                            program.getPackageJsonInfoCache(),
                         );
                         if (typesPackageJson) {
                             const entrypoints = getRootNamesFromPackageJson(typesPackageJson, program, symlinkCache);
@@ -2691,7 +2696,7 @@ export class AutoImportProviderProject extends Project {
                 packageJson,
                 compilerOptions,
                 host,
-                program.getModuleResolutionCache(),
+                program.getPackageJsonInfoCache(),
                 resolveJs,
             );
             if (entrypoints) {
@@ -2842,8 +2847,8 @@ export class AutoImportProviderProject extends Project {
     }
 
     /** @internal */
-    override getModuleResolutionCache() {
-        return this.hostProject.getCurrentProgram()?.getModuleResolutionCache();
+    override getPackageJsonInfoCache() {
+        return this.hostProject.getPackageJsonInfoCache();
     }
 }
 

src/server/session.ts

@@ -1595,7 +1595,7 @@ export class Session<TMessage = string> implements EventSender {
             if (nodeModulesPathParts && fileName.lastIndexOf(nodeModulesPathPart) === nodeModulesPathParts.topLevelNodeModulesIndex) {
                 // Second check ensures the fileName only contains one `/node_modules/`. If there's more than one I give up.
                 const packageDirectory = fileName.substring(0, nodeModulesPathParts.packageRootIndex);
-                const packageJsonCache = project.getModuleResolutionCache()?.getPackageJsonInfoCache();
+                const packageJsonCache = project.getPackageJsonInfoCache();
                 const compilerOptions = project.getCompilationSettings();
                 const packageJson = getPackageScopeForPath(getNormalizedAbsolutePath(packageDirectory + "/package.json", project.getCurrentDirectory()), getTemporaryModuleResolutionState(packageJsonCache, project, compilerOptions));
                 if (!packageJson) return undefined;
@@ -1606,7 +1606,7 @@ export class Session<TMessage = string> implements EventSender {
                     packageJson,
                     { moduleResolution: ModuleResolutionKind.Node10 },
                     project,
-                    project.getModuleResolutionCache(),
+                    packageJsonCache,
                 );
                 // This substring is correct only because we checked for a single `/node_modules/` at the top.
                 const packageNamePathPart = fileName.substring(

src/services/utilities.ts

@@ -2463,7 +2463,7 @@ export function createModuleSpecifierResolutionHost(program: Program, host: Lang
         useCaseSensitiveFileNames: maybeBind(host, host.useCaseSensitiveFileNames),
         getSymlinkCache: maybeBind(host, host.getSymlinkCache) || program.getSymlinkCache,
         getModuleSpecifierCache: maybeBind(host, host.getModuleSpecifierCache),
-        getPackageJsonInfoCache: () => program.getModuleResolutionCache()?.getPackageJsonInfoCache(),
+        getPackageJsonInfoCache: () => program.getPackageJsonInfoCache(),
         getGlobalTypingsCacheLocation: maybeBind(host, host.getGlobalTypingsCacheLocation),
         redirectTargetsMap: program.redirectTargetsMap,
         getProjectReferenceRedirect: fileName => program.getProjectReferenceRedirect(fileName),