Expose logger statistics

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 {
 
+    val errorCount: Int
+    val warningCount: Int
+    val infoCount: Int
+    val loggingCount: Int
+    val 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

@@ -37,8 +37,6 @@ import com.google.devtools.ksp.symbol.*
 import com.google.devtools.ksp.symbol.impl.KSObjectCacheManager
 import com.google.devtools.ksp.symbol.impl.java.KSFileJavaImpl
 import com.google.devtools.ksp.symbol.impl.kotlin.KSFileImpl
-import org.jetbrains.kotlin.incremental.isJavaFile
-import org.jetbrains.kotlin.incremental.isKotlinFile
 import org.jetbrains.kotlin.psi.KtFile
 import org.jetbrains.kotlin.resolve.BindingContext
 import org.jetbrains.kotlin.resolve.BindingTrace

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

@@ -33,29 +33,49 @@ class MessageCollectorBasedKSPLogger(private val messageCollector: MessageCollec
         const val PREFIX = "[ksp] "
     }
 
+    override var errorCount = 0
+        private set
+
+    override var warningCount = 0
+        private set
+
+    override var infoCount = 0
+        private set
+
+    override var loggingCount = 0
+        private set
+
+    override var exceptionCount = 0
+        private set
+
     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

@@ -32,6 +32,7 @@ class ErrorTypeProcessor : AbstractTestProcessor() {
     }
 
     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