Daemon plain file and plain text format support

Author: johndoe2013

kls_database.db

@@ -1,5 +1,6 @@
 package xyz.block.kotlinformatter
 
+import java.io.ByteArrayInputStream
 import java.io.RandomAccessFile
 import java.net.ServerSocket
 import java.net.Socket
@@ -132,6 +133,38 @@ class Daemon(
                   clientSocket.write(ExitCode.SUCCESS, response)
                 }
 
+                "format" -> {
+                  lastCommandTime = Instant.now()
+                  val fileArgs = messageParts.drop(1)
+                  if (fileArgs.isEmpty()) {
+                    clientSocket.write(ExitCode.FAILURE, "No files provided for format command")
+                  } else if (fileArgs.size == 1 && fileArgs[0] == "-") {
+                    // Process the code starting from a newline and ending at ASCII EOT (04)
+                    // Usage: (echo "format -"; echo <CODE>; echo -ne '\04\n') | nc localhost <PORT>
+                    // Or replace "echo <CODE>" to "cat" for standard input 
+                    val inputLines = mutableListOf<String>()
+                    while (true) {
+                      val line = reader.readLine() ?: break
+                      if (line.trim() == "\u0004") break
+                      inputLines.add(line)
+                    }
+                    val inputContent = inputLines.joinToString("\n")
+                    val inputStream = ByteArrayInputStream(inputContent.toByteArray())
+                    lateinit var formatted: String 
+                    val result = KotlinFormatter(
+                      files = listOf("-"),
+                      inputStream = inputStream,
+                      outputCallback = { formatted = it }
+                    ).format()
+                    val exitCode = if (result.hasFailure) ExitCode.FAILURE else ExitCode.SUCCESS
+                    clientSocket.write(exitCode, if (result.hasFailure) result.output else formatted)
+                  } else {
+                    val result = KotlinFormatter(files = fileArgs).format()
+                    val exitCode = if (result.hasFailure) ExitCode.FAILURE else ExitCode.SUCCESS
+                    clientSocket.write(exitCode, result.output)
+                  }
+                }
+
                 "pre-commit" -> {
                   lastCommandTime = Instant.now()
                   val files = messageParts.drop(1)
@@ -245,4 +278,4 @@ class Daemon(
     val LOCK_FILE_PATH = Paths.get(".kotlinformatter/kf.lock")
     val rootGitPath by lazy { Paths.get(GitProcessRunner.run("rev-parse", "--show-toplevel").trim()) }
   }
-}
+}

kotlin-format/src/main/kotlin/xyz/block/kotlinformatter/Daemon.kt

@@ -112,6 +112,54 @@ class DaemonTest {
     assertThat(secondDaemonStarted).isFalse()
   }
 
+  @Test
+  fun `format command with file arguments`() {
+    val testFile = File(tempDir.toFile(), "FormatTest.kt")
+    val unformatted = "fun  foo()=println(\"Hello\")"
+    val expectedFormatted = "fun foo() = println(\"Hello\")\n"
+    testFile.writeText(unformatted)
+
+    val daemonData = daemon.readLockFile()
+    assertThat(daemonData).isNotNull
+
+    // Send format command
+    Socket("localhost", daemonData!!.port).use { socket ->
+      socket.getOutputStream().write("format ${testFile.absolutePath}\n".toByteArray())
+      val reader = socket.getInputStream().bufferedReader()
+      val exitCode = reader.readLine()
+      assertThat(exitCode).isEqualTo(ExitCode.SUCCESS.toString())
+      val output = reader.readText()
+      assertThat(output).contains("✅ Formatted")
+    }
+
+    // Verify file content
+    val formattedContent = testFile.readText()
+    assertThat(formattedContent).isEqualTo(expectedFormatted)
+  }
+
+  @Test
+  fun `format command with 'stdin' mode`() {
+    val unformatted = "fun  foo()=println(\"Hello\")"
+    val expectedFormatted = "fun foo() = println(\"Hello\")\n"
+
+    val daemonData = daemon.readLockFile()
+    assertThat(daemonData).isNotNull
+
+    // Send format -
+    Socket("localhost", daemonData!!.port).use { socket ->
+      val outputStream = socket.getOutputStream()
+      outputStream.write("format -\n".toByteArray())
+      outputStream.write(unformatted.toByteArray())
+      outputStream.write("\n\u0004\n".toByteArray())
+
+      val reader = socket.getInputStream().bufferedReader()
+      val exitCode = reader.readLine()
+      assertThat(exitCode).isEqualTo(ExitCode.SUCCESS.toString())
+      val output = reader.readText()
+      assertThat(output).isEqualTo(expectedFormatted)
+    }
+  }
+
   @Test
   fun `handles pre-commit command`() {
     val testFile = File(tempDir.toFile(), "Test.kt")
@@ -330,4 +378,4 @@ class DaemonTest {
     val MAX_RUNTIME = Duration.ofSeconds(5)
     val TIMEOUT_CHECKER_INTERVAL = Duration.ofSeconds(1)
   }
-}
+}