add timeout for waiting ktlint process

Author: vincentlauvlwj

ktorm-ksp-compiler/src/main/kotlin/org/ktorm/ksp/compiler/KtormProcessorProvider.kt

@@ -65,7 +65,8 @@ public class KtormProcessorProvider : SymbolProcessorProvider {
                 val fileSpec = FileGenerator.generate(table, environment)
 
                 // Beautify the generated code.
-                val formattedCode = formatter.format(fileSpec.toString())
+                val fileName = fileSpec.packageName.replace('.', '/') + "/" + fileSpec.name + ".kt"
+                val formattedCode = formatter.format(fileName, fileSpec.toString())
 
                 // Output the formatted code.
                 val dependencies = Dependencies(false, *table.getDependencyFiles().toTypedArray())
@@ -88,7 +89,7 @@ public class KtormProcessorProvider : SymbolProcessorProvider {
         } catch (_: NoClassDefFoundError) {
         }
 
-        return CodeFormatter { code -> code }
+        return CodeFormatter { _, code -> code }
     }
 
     private fun TableMetadata.getDependencyFiles(): List<KSFile> {

ktorm-ksp-compiler/src/main/kotlin/org/ktorm/ksp/compiler/formatter/CodeFormatter.kt

@@ -24,5 +24,5 @@ internal fun interface CodeFormatter {
     /**
      * Format the generated code to the community recommended coding style.
      */
-    fun format(code: String): String
+    fun format(fileName: String, code: String): String
 }

ktorm-ksp-compiler/src/main/kotlin/org/ktorm/ksp/compiler/formatter/KtLintCodeFormatter.kt

@@ -38,7 +38,7 @@ internal class KtLintCodeFormatter(val environment: SymbolProcessorEnvironment)
         )
     )
 
-    override fun format(code: String): String {
+    override fun format(fileName: String, code: String): String {
         try {
             // Manually fix some code styles before formatting.
             val snippet = code

ktorm-ksp-compiler/src/main/kotlin/org/ktorm/ksp/compiler/formatter/StandaloneKtLintCodeFormatter.kt

@@ -18,19 +18,25 @@ package org.ktorm.ksp.compiler.formatter
 
 import com.google.devtools.ksp.processing.SymbolProcessorEnvironment
 import java.io.File
+import java.util.concurrent.TimeUnit
 
 internal class StandaloneKtLintCodeFormatter(val environment: SymbolProcessorEnvironment) : CodeFormatter {
+    private val logger = environment.logger
     private val command = buildCommand()
 
     init {
-        environment.logger.info("[ktorm-ksp-compiler] init ktlint formatter with command: ${command.joinToString(" ")}")
+        logger.info("[ktorm-ksp-compiler] init ktlint formatter with command: ${command.joinToString(" ")}")
     }
 
-    override fun format(code: String): String {
+    override fun format(fileName: String, code: String): String {
         try {
             val p = ProcessBuilder(command).start()
             p.outputStream.bufferedWriter(Charsets.UTF_8).use { it.write(preprocessCode(code)) }
-            p.waitFor()
+
+            if (!p.waitFor(30, TimeUnit.SECONDS)) {
+                logger.info("[ktorm-ksp-compiler] ktlint execution timeout, skip code formatting for file: $fileName")
+                return code
+            }
 
             val formattedCode = p.inputStream.bufferedReader(Charsets.UTF_8).use { it.readText() }
             if (p.exitValue() == 0) {
@@ -43,12 +49,12 @@ internal class StandaloneKtLintCodeFormatter(val environment: SymbolProcessorEnv
                 } else {
                     // Exit exceptionally.
                     val msg = p.errorStream.bufferedReader(Charsets.UTF_8).use { it.readText() }
-                    environment.logger.error("[ktorm-ksp-compiler] ktlint exit with code: ${p.exitValue()}\n$msg")
+                    logger.error("[ktorm-ksp-compiler] ktlint exit with code: ${p.exitValue()}\n$msg")
                     return code
                 }
             }
         } catch (e: Throwable) {
-            environment.logger.exception(e)
+            logger.exception(e)
             return code
         }
     }