work around to make ksp fail execution on error or exception log event

Author: neetopia

api/src/main/kotlin/com/google/devtools/ksp/processing/KSPLogger.kt

@@ -22,6 +22,12 @@ import com.google.devtools.ksp.symbol.KSNode
 
 interface KSPLogger {
 
+    var errorCount: Int
+    var warningCount: Int
+    var infoCount: Int
+    var loggingCount: Int
+    var exceptionCount: Int
+
     fun logging(message: String, symbol: KSNode? = null)
     fun info(message: String, symbol: KSNode? = null)
     fun warn(message: String, symbol: KSNode? = null)

compiler-plugin/src/main/kotlin/com/google/devtools/ksp/KotlinSymbolProcessingExtension.kt

@@ -107,7 +107,11 @@ abstract class AbstractKotlinSymbolProcessingExtension(val options: KspOptions,
         bindingTrace: BindingTrace,
         files: Collection<KtFile>
     ): AnalysisResult? {
-        return AnalysisResult.success(BindingContext.EMPTY, module, shouldGenerateCode = false)
+        return if (logger.errorCount > 0 || logger.exceptionCount > 0) {
+            AnalysisResult.success(bindingTrace.bindingContext, module, shouldGenerateCode = false)
+        } else {
+            AnalysisResult.success(BindingContext.EMPTY, module, shouldGenerateCode = false)
+        }
     }
 
     private var annotationProcessingComplete = false

compiler-plugin/src/main/kotlin/com/google/devtools/ksp/processing/impl/MessageCollectorBasedKSPLogger.kt

@@ -33,29 +33,40 @@ class MessageCollectorBasedKSPLogger(private val messageCollector: MessageCollec
         const val PREFIX = "[ksp] "
     }
 
+    override var errorCount = 0
+    override var warningCount = 0
+    override var infoCount = 0
+    override var loggingCount = 0
+    override var exceptionCount = 0
+
     private fun convertMessage(message: String, symbol: KSNode?): String =
         when (val location = symbol?.location) {
             is FileLocation -> "$PREFIX${location.filePath}:${location.lineNumber}: $message"
             is NonExistLocation, null -> "$PREFIX$message"
         }
 
     override fun logging(message: String, symbol: KSNode?) {
+        loggingCount++
         messageCollector.report(CompilerMessageSeverity.LOGGING, convertMessage(message, symbol))
     }
 
     override fun info(message: String, symbol: KSNode?) {
+        infoCount++
         messageCollector.report(CompilerMessageSeverity.INFO, convertMessage(message, symbol))
     }
 
     override fun warn(message: String, symbol: KSNode?) {
+        warningCount++
         messageCollector.report(CompilerMessageSeverity.WARNING, convertMessage(message, symbol))
     }
 
     override fun error(message: String, symbol: KSNode?) {
+        errorCount++
         messageCollector.report(CompilerMessageSeverity.ERROR, convertMessage(message, symbol))
     }
 
     override fun exception(e: Throwable) {
+        exceptionCount++
         val writer = StringWriter()
         e.printStackTrace(PrintWriter(writer))
         messageCollector.report(CompilerMessageSeverity.EXCEPTION, writer.toString())

compiler-plugin/src/main/kotlin/com/google/devtools/ksp/processor/ErrorTypeProcessor.kt

@@ -19,6 +19,8 @@
 package com.google.devtools.ksp.processor
 
 import com.google.devtools.ksp.getClassDeclarationByName
+import com.google.devtools.ksp.processing.CodeGenerator
+import com.google.devtools.ksp.processing.KSPLogger
 import com.google.devtools.ksp.processing.Resolver
 import com.google.devtools.ksp.processing.impl.ResolverImpl
 import com.google.devtools.ksp.symbol.KSPropertyDeclaration
@@ -31,7 +33,12 @@ class ErrorTypeProcessor : AbstractTestProcessor() {
         return result
     }
 
+    override fun init(options: Map<String, String>, kotlinVersion: KotlinVersion, codeGenerator: CodeGenerator, logger: KSPLogger) {
+        logger.error("Test Error")
+    }
+
     override fun process(resolver: Resolver) {
+
         val classC = resolver.getClassDeclarationByName(resolver.getKSNameFromString("C"))!!
         val errorAtTop = classC.declarations.single { it.simpleName.asString() == "errorAtTop" } as KSPropertyDeclaration
         val errorInComponent = classC.declarations.single { it.simpleName.asString() == "errorInComponent" } as KSPropertyDeclaration