[JPS] Properly register files to compile

JPS graph needs to know if Kotlin JPS tried to compile some files. Previously, we registered these files only iff compilation succeeded. Now we register them also in case when the compiler fails

Author: Aleksei.Cherepanov

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

@@ -464,9 +464,7 @@ class KotlinBuilder : ModuleLevelBuilder(BuilderCategory.SOURCE_PROCESSOR) {
 
         cleanJsOutputs(context, kotlinChunk, incrementalCaches, kotlinDirtyFilesHolder)
 
-        if (LOG.isDebugEnabled) {
-            LOG.debug("Compiling files: ${kotlinDirtyFilesHolder.allDirtyFiles}")
-        }
+
 
         val reportService = JpsStatisticsReportService.getFromContext(context)
         reportService.reportCompilerArguments(chunk, kotlinChunk)
@@ -494,6 +492,7 @@ class KotlinBuilder : ModuleLevelBuilder(BuilderCategory.SOURCE_PROCESSOR) {
             LOG.info("Compiled with errors")
             return ABORT
         } else {
+            JavaBuilderUtil.registerSuccessfullyCompiled(context, kotlinDirtyFilesHolder.allDirtyFiles)
             LOG.info("Compiled successfully")
         }
 
@@ -640,11 +639,23 @@ class KotlinBuilder : ModuleLevelBuilder(BuilderCategory.SOURCE_PROCESSOR) {
             }
         }
 
+        registerFilesToCompile(dirtyFilesHolder, context)
         val isDoneSomething = representativeTarget.compileModuleChunk(commonArguments, dirtyFilesHolder, environment, buildMetricReporter)
 
         return if (isDoneSomething) environment.outputItemsCollector else null
     }
 
+    private fun registerFilesToCompile(
+        dirtyFilesHolder: KotlinDirtySourceFilesHolder,
+        context: CompileContext,
+    ) {
+        val allDirtyFiles = dirtyFilesHolder.allDirtyFiles
+        if (LOG.isDebugEnabled) {
+            LOG.debug("Compiling files: $allDirtyFiles")
+        }
+        JavaBuilderUtil.registerFilesToCompile(context, allDirtyFiles)
+    }
+
     private fun createCompileEnvironment(
         context: CompileContext,
         kotlinModuleBuilderTarget: KotlinModuleBuildTarget<*>,

jps/jps-plugin/src/org/jetbrains/kotlin/jps/targets/KotlinJvmModuleBuildTarget.kt

@@ -422,10 +422,6 @@ class KotlinJvmModuleBuildTarget(kotlinContext: KotlinCompileContext, jpsModuleB
             }
         }
         // important: in jps-dependency-graph you can't register additional dependencies after [callback.associate].
-
-        val allCompiled = dirtyFilesHolder.allDirtyFiles
-        JavaBuilderUtil.registerFilesToCompile(localContext, allCompiled)
-        JavaBuilderUtil.registerSuccessfullyCompiled(localContext, allCompiled)
     }
 
     private fun processInlineConstTracker(inlineConstTracker: InlineConstTrackerImpl, sourceFile: File, output: GeneratedJvmClass, callback: Backend) {