[JPS] Improve JPS dumb mode

For the correct work of the compiler, some maps should be fulfilled by the build system. In the first implementation of dumb mode for JPS all maps were disabled. Now flag was introduced to determine IC behavior: to collect data needed for the compiler's code generation and disable maps and IC analysis

^KT-65043 Fixed

build-common/src/org/jetbrains/kotlin/incremental/AbstractIncrementalCache.kt

@@ -49,7 +49,7 @@ interface IncrementalCacheCommon {
  */
 abstract class AbstractIncrementalCache<ClassName>(
     workingDir: File,
-    icContext: IncrementalCompilationContext,
+    val icContext: IncrementalCompilationContext,
 ) : BasicMapsOwner(workingDir), IncrementalCacheCommon {
     companion object {
         private const val CLASS_ATTRIBUTES = "class-attributes"
@@ -229,6 +229,8 @@ abstract class AbstractIncrementalCache<ClassName>(
     }
 
     override fun updateComplementaryFiles(dirtyFiles: Collection<File>, expectActualTracker: ExpectActualTrackerImpl) {
+        if (icContext.useCompilerMapsOnly && this is IncrementalJvmCache) return
+
         dirtyFiles.forEach {
             complementaryFilesMap.remove(it)
         }

build-common/src/org/jetbrains/kotlin/incremental/IncrementalCompilationContext.kt

@@ -25,6 +25,7 @@ class IncrementalCompilationContext(
     val trackChangesInLookupCache: Boolean = false,
     val icFeatures: IncrementalCompilationFeatures = IncrementalCompilationFeatures.DEFAULT_CONFIGURATION,
     val fragmentContext: FragmentContext? = null,
+    val useCompilerMapsOnly: Boolean = false
 ) {
     @Deprecated("This constructor is scheduled to be removed. KSP is using it")
     constructor(

build-common/src/org/jetbrains/kotlin/incremental/IncrementalJsCache.kt

@@ -41,7 +41,7 @@ import java.io.File
 
 open class IncrementalJsCache(
     cachesDir: File,
-    private val icContext: IncrementalCompilationContext,
+    icContext: IncrementalCompilationContext,
     serializerProtocol: SerializerExtensionProtocol,
 ) : AbstractIncrementalCache<FqName>(cachesDir, icContext) {
     companion object {

build-common/src/org/jetbrains/kotlin/incremental/IncrementalJvmCache.kt

@@ -129,7 +129,7 @@ open class IncrementalJvmCache(
             sourceFiles.forEach {
                 sourceToClassesMap.append(it, className)
             }
-            internalNameToSource[className.internalName] = sourceFiles
+            if (!icContext.useCompilerMapsOnly) internalNameToSource[className.internalName] = sourceFiles
         }
 
         if (kotlinClassInfo.classId.isLocal) return
@@ -142,9 +142,11 @@ open class IncrementalJvmCache(
                 packagePartMap.addPackagePart(className)
 
                 protoMap.process(kotlinClassInfo, changesCollector)
+                if (!icContext.useCompilerMapsOnly) {
                     constantsMap.process(kotlinClassInfo, changesCollector)
                     inlineFunctionsMap.process(kotlinClassInfo, changesCollector)
                 }
+            }
             KotlinClassHeader.Kind.MULTIFILE_CLASS -> {
                 val partNames = kotlinClassInfo.classHeaderData.toList()
                 check(partNames.isNotEmpty()) { "Multifile class has no parts: $className" }
@@ -158,6 +160,8 @@ open class IncrementalJvmCache(
                 }
                 protoMap.remove(className, changesCollector)
                 classFqNameToSourceMap.remove(className.fqNameForClassNameWithoutDollars)
+                if (!icContext.useCompilerMapsOnly) {
+
                     classAttributesMap.remove(className.fqNameForClassNameWithoutDollars)
                     internalNameToSource.remove(className.internalName)
 
@@ -165,24 +169,31 @@ open class IncrementalJvmCache(
                     constantsMap.process(kotlinClassInfo, changesCollector)
                     inlineFunctionsMap.process(kotlinClassInfo, changesCollector)
                 }
+            }
             KotlinClassHeader.Kind.MULTIFILE_CLASS_PART -> {
                 if (sourceFiles != null) {
                     assert(sourceFiles.size == 1) { "Multifile class part from several source files: $sourceFiles" }
                 }
                 packagePartMap.addPackagePart(className)
                 partToMultifileFacade[className] = kotlinClassInfo.multifileClassName!!
-
                 protoMap.process(kotlinClassInfo, changesCollector)
+                if (!icContext.useCompilerMapsOnly) {
                     constantsMap.process(kotlinClassInfo, changesCollector)
                     inlineFunctionsMap.process(kotlinClassInfo, changesCollector)
                 }
+            }
             KotlinClassHeader.Kind.CLASS -> {
+                if (!icContext.useCompilerMapsOnly) {
                     addToClassStorage(kotlinClassInfo.protoData as ClassProtoData, sourceFiles?.let { sourceFiles.single() })
+                }
 
                 protoMap.process(kotlinClassInfo, changesCollector)
+
+                if (!icContext.useCompilerMapsOnly) {
                     constantsMap.process(kotlinClassInfo, changesCollector)
                     inlineFunctionsMap.process(kotlinClassInfo, changesCollector)
                 }
+            }
             KotlinClassHeader.Kind.UNKNOWN, KotlinClassHeader.Kind.SYNTHETIC_CLASS -> {
             }
         }
@@ -228,10 +239,15 @@ open class IncrementalJvmCache(
 
     fun saveJavaClassProto(source: File?, serializedJavaClass: SerializedJavaClass, collector: ChangesCollector) {
         val jvmClassName = JvmClassName.byClassId(serializedJavaClass.classId)
+
+        if (!icContext.useCompilerMapsOnly) {
             javaSourcesProtoMap.process(jvmClassName, serializedJavaClass, collector)
+        }
         source?.let { sourceToClassesMap.append(source, jvmClassName) }
+        if (!icContext.useCompilerMapsOnly) {
             addToClassStorage(serializedJavaClass.toProtoData(), source)
 //        collector.addJavaProto(ClassProtoData(proto, nameResolver))
+        }
         dirtyOutputClassesMap.notDirty(jvmClassName)
     }
 
@@ -280,13 +296,17 @@ open class IncrementalJvmCache(
             packagePartMap.remove(it)
             multifileFacadeToParts.remove(it)
             partToMultifileFacade.remove(it)
+            if (!icContext.useCompilerMapsOnly) {
                 constantsMap.remove(it)
                 inlineFunctionsMap.remove(it)
                 internalNameToSource.remove(it.internalName)
                 javaSourcesProtoMap.remove(it, changesCollector)
             }
+        }
 
+        if (!icContext.useCompilerMapsOnly) {
             removeAllFromClassStorage(dirtyClasses.map { it.fqNameForClassNameWithoutDollars }, changesCollector)
+        }
         dirtyOutputClassesMap.clear()
     }
 

jps/jps-plugin/src/org/jetbrains/kotlin/jps/build/KotlinBuilder.kt

@@ -499,7 +499,7 @@ class KotlinBuilder : ModuleLevelBuilder(BuilderCategory.SOURCE_PROCESSOR) {
 
         val generatedFiles = getGeneratedFiles(context, chunk, environment.outputItemsCollector)
 
-        markDirtyComplementaryMultifileClasses(generatedFiles, kotlinContext, incrementalCaches, fsOperations)
+        if (!isKotlinBuilderInDumbMode) markDirtyComplementaryMultifileClasses(generatedFiles, kotlinContext, incrementalCaches, fsOperations)
 
         val kotlinTargets = kotlinContext.targetsBinding
         for ((target, outputItems) in generatedFiles) {
@@ -530,7 +530,7 @@ class KotlinBuilder : ModuleLevelBuilder(BuilderCategory.SOURCE_PROCESSOR) {
             environment
         )
 
-        if (isKotlinBuilderInDumbMode || !representativeTarget.isIncrementalCompilationEnabled) {
+        if (!representativeTarget.isIncrementalCompilationEnabled) {
             return OK
         }
 
@@ -552,6 +552,7 @@ class KotlinBuilder : ModuleLevelBuilder(BuilderCategory.SOURCE_PROCESSOR) {
                 )
             }
 
+            if (!isKotlinBuilderInDumbMode) {
                 updateLookupStorage(lookupTracker, kotlinContext.lookupStorageManager, kotlinDirtyFilesHolder)
 
                 if (!isChunkRebuilding) {
@@ -563,6 +564,7 @@ class KotlinBuilder : ModuleLevelBuilder(BuilderCategory.SOURCE_PROCESSOR) {
                     )
                 }
             }
+        }
 
         return OK
     }

jps/jps-plugin/src/org/jetbrains/kotlin/jps/build/KotlinCompileContext.kt

@@ -84,7 +84,8 @@ class KotlinCompileContext(val jpsContext: CompileContext) {
 
     val icContext = IncrementalCompilationContext(
         pathConverterForSourceFiles = fileToPathConverter,
-        pathConverterForOutputFiles = fileToPathConverter
+        pathConverterForOutputFiles = fileToPathConverter,
+        useCompilerMapsOnly = KotlinBuilder.isKotlinBuilderInDumbMode
     )
 
     val lookupStorageManager = JpsLookupStorageManager(dataManager, icContext)