Refactor diagnostic & result reporters...

to better reflect actual roles

compiler/cli/src/org/jetbrains/kotlin/cli/common/fir/FirDiagnosticsCompilerResultsReporter.kt

@@ -8,12 +8,9 @@ package org.jetbrains.kotlin.cli.common.fir
 import org.jetbrains.kotlin.AbstractKtSourceElement
 import org.jetbrains.kotlin.KtLightSourceElement
 import org.jetbrains.kotlin.KtPsiSourceElement
-import org.jetbrains.kotlin.cli.common.messages.AnalyzerWithCompilerReport
-import org.jetbrains.kotlin.cli.common.messages.CompilerMessageSourceLocation
-import org.jetbrains.kotlin.cli.common.messages.MessageCollector
-import org.jetbrains.kotlin.cli.common.messages.MessageUtil
+import org.jetbrains.kotlin.cli.common.messages.*
 import org.jetbrains.kotlin.diagnostics.*
-import org.jetbrains.kotlin.diagnostics.impl.BaseDiagnosticReporter
+import org.jetbrains.kotlin.diagnostics.impl.BaseDiagnosticsCollector
 import org.jetbrains.kotlin.diagnostics.rendering.RootDiagnosticRendererFactory
 import java.io.Closeable
 import java.io.File
@@ -29,10 +26,26 @@ object FirDiagnosticsCompilerResultsReporter {
         return hasErrors
     }
 
-    fun reportToCollector(diagnosticsReporter: BaseDiagnosticReporter, messageCollector: MessageCollector): Boolean {
+    fun reportToMessageCollector(diagnosticsCollector: BaseDiagnosticsCollector, messageCollector: MessageCollector): Boolean {
+        return reportByFile(diagnosticsCollector) { diagnostic, location ->
+            reportDiagnosticToMessageCollector(diagnostic, location, messageCollector)
+        }
+    }
+
+    fun throwFirstErrorAsException(
+        diagnosticsCollector: BaseDiagnosticsCollector, messageRenderer: MessageRenderer = MessageRenderer.PLAIN_RELATIVE_PATHS
+    ): Boolean {
+        return reportByFile(diagnosticsCollector) { diagnostic, location ->
+            throwErrorDiagnosticAsException(diagnostic, location, messageRenderer)
+        }
+    }
+
+    fun reportByFile(
+        diagnosticsCollector: BaseDiagnosticsCollector, report: (KtDiagnostic, CompilerMessageSourceLocation) -> Unit
+    ): Boolean {
         var hasErrors = false
-        for (filePath in diagnosticsReporter.diagnosticsByFilePath.keys) {
-            for (diagnostic in diagnosticsReporter.diagnosticsByFilePath[filePath].orEmpty().sortedWith(SingleFileDiagnosticComparator)) {
+        for (filePath in diagnosticsCollector.diagnosticsByFilePath.keys) {
+            for (diagnostic in diagnosticsCollector.diagnosticsByFilePath[filePath].orEmpty().sortedWith(SingleFileDiagnosticComparator)) {
                 var filePositionFinder: SequentialFilePositionFinder? = null
                 try {
                     when {
@@ -49,7 +62,7 @@ object FirDiagnosticsCompilerResultsReporter {
                             MessageUtil.createMessageLocation(filePath, position.lineContent, position.line, position.column, -1, -1)
                         }
                     }?.let { location ->
-                        reportDiagnostic(diagnostic, location, messageCollector)
+                        report(diagnostic, location)
                         hasErrors = hasErrors || diagnostic.severity == Severity.ERROR
                     }
                 } finally {
@@ -75,12 +88,12 @@ object FirDiagnosticsCompilerResultsReporter {
     private fun reportDiagnostic(diagnostic: KtDiagnostic, reporter: MessageCollector): Boolean {
         if (!diagnostic.isValid) return false
         diagnostic.location()?.let {
-            reportDiagnostic(diagnostic, it, reporter)
+            reportDiagnosticToMessageCollector(diagnostic, it, reporter)
         }
         return diagnostic.severity == Severity.ERROR
     }
 
-    private fun reportDiagnostic(
+    private fun reportDiagnosticToMessageCollector(
         diagnostic: KtDiagnostic,
         location: CompilerMessageSourceLocation,
         reporter: MessageCollector
@@ -90,6 +103,19 @@ object FirDiagnosticsCompilerResultsReporter {
         reporter.report(severity, renderer.render(diagnostic), location)
     }
 
+    private fun throwErrorDiagnosticAsException(
+        diagnostic: KtDiagnostic,
+        location: CompilerMessageSourceLocation,
+        messageRenderer: MessageRenderer
+    ) {
+        if (diagnostic.severity == Severity.ERROR) {
+            val severity = AnalyzerWithCompilerReport.convertSeverity(diagnostic.severity)
+            val renderer = RootDiagnosticRendererFactory(diagnostic)
+            val diagnosticText = messageRenderer.render(severity, renderer.render(diagnostic), location)
+            throw IllegalStateException("${diagnostic.factory.name}: $diagnosticText")
+        }
+    }
+
     private fun KtDiagnostic.location(): CompilerMessageSourceLocation? = when (val element = element) {
         is KtPsiSourceElement -> element.location(this)
         is KtLightSourceElement -> element.location(this)

compiler/cli/src/org/jetbrains/kotlin/cli/jvm/compiler/FirKotlinToJvmBytecodeCompiler.kt

@@ -24,7 +24,7 @@ import org.jetbrains.kotlin.codegen.CodegenFactory
 import org.jetbrains.kotlin.codegen.state.GenerationState
 import org.jetbrains.kotlin.config.*
 import org.jetbrains.kotlin.diagnostics.DiagnosticReporterFactory
-import org.jetbrains.kotlin.diagnostics.impl.BaseDiagnosticReporter
+import org.jetbrains.kotlin.diagnostics.impl.BaseDiagnosticsCollector
 import org.jetbrains.kotlin.fir.*
 import org.jetbrains.kotlin.fir.backend.Fir2IrResult
 import org.jetbrains.kotlin.fir.backend.jvm.FirJvmBackendClassResolver
@@ -133,7 +133,7 @@ object FirKotlinToJvmBytecodeCompiler {
             performanceManager?.notifyAnalysisFinished()
         }
         if (firResult == null) {
-            FirDiagnosticsCompilerResultsReporter.reportToCollector(diagnosticsReporter, messageCollector)
+            FirDiagnosticsCompilerResultsReporter.reportToMessageCollector(diagnosticsReporter, messageCollector)
             return null
         }
 
@@ -153,7 +153,7 @@ object FirKotlinToJvmBytecodeCompiler {
             diagnosticsReporter
         )
 
-        FirDiagnosticsCompilerResultsReporter.reportToCollector(diagnosticsReporter, messageCollector)
+        FirDiagnosticsCompilerResultsReporter.reportToMessageCollector(diagnosticsReporter, messageCollector)
 
         performanceManager?.notifyIRGenerationFinished()
         performanceManager?.notifyGenerationFinished()
@@ -168,7 +168,7 @@ object FirKotlinToJvmBytecodeCompiler {
         val fir: List<FirFile>
     )
 
-    private fun CompilationContext.runFrontend(ktFiles: List<KtFile>, diagnosticsReporter: BaseDiagnosticReporter): FirResult? {
+    private fun CompilationContext.runFrontend(ktFiles: List<KtFile>, diagnosticsReporter: BaseDiagnosticsCollector): FirResult? {
         @Suppress("NAME_SHADOWING")
         var ktFiles = ktFiles
         val syntaxErrors = ktFiles.fold(false) { errorsFound, ktFile ->
@@ -287,7 +287,7 @@ object FirKotlinToJvmBytecodeCompiler {
         fir2IrResult: Fir2IrResult,
         extensions: JvmGeneratorExtensionsImpl,
         session: FirSession,
-        diagnosticsReporter: BaseDiagnosticReporter
+        diagnosticsReporter: BaseDiagnosticsCollector
     ): GenerationState {
         val (moduleFragment, symbolTable, components) = fir2IrResult
         val dummyBindingContext = NoScopeRecordCliBindingTrace().bindingContext

compiler/cli/src/org/jetbrains/kotlin/cli/jvm/compiler/KotlinToJVMBytecodeCompiler.kt

@@ -375,7 +375,7 @@ object KotlinToJVMBytecodeCompiler {
             ),
             environment.messageCollector
         )
-        FirDiagnosticsCompilerResultsReporter.reportToCollector(diagnosticsReporter, environment.messageCollector)
+        FirDiagnosticsCompilerResultsReporter.reportToMessageCollector(diagnosticsReporter, environment.messageCollector)
 
         ProgressIndicatorAndCompilationCanceledStatus.checkCanceled()
         return state