Fix improper handling of text/plain responses

Add the ability to enable request/response debugging

Author: rdbatch

src/main/kotlin/com/batchofcode/runtimelocal/config/EnvConfig.kt

@@ -2,4 +2,5 @@ package com.batchofcode.runtimelocal.config
 
 object EnvConfig {
     val port = System.getProperty("runtime.port")?.toInt() ?: System.getenv("RUNTIME_PORT")?.toInt() ?: 9000
+    val debug = System.getProperty("runtime.debug")?.toBoolean() ?: System.getenv("RUNTIME_DEBUG")?.toBoolean() ?: false
 }

src/main/kotlin/com/batchofcode/runtimelocal/handler/InvocationHandler.kt

@@ -9,6 +9,7 @@ import org.http4k.core.Status.Companion.BAD_REQUEST
 import org.http4k.core.Status.Companion.OK
 import org.http4k.format.Jackson.json
 import org.http4k.lens.Header
+import org.http4k.lens.LensFailure
 import org.http4k.lens.string
 import org.http4k.routing.RoutingHttpHandler
 import org.http4k.routing.bind
@@ -44,7 +45,12 @@ object InvocationHandler {
     }
 
     fun response(): HttpHandler = handler@{
-        val requestBody = Body.json().toLens()(it)
+        val requestBody = try {
+            Body.json().toLens()(it).toString()
+        }
+        catch (ex: LensFailure) {
+            Body.string(ContentType.TEXT_PLAIN).toLens()(it)
+        }
         val traceIdHeader = Header.optional("_X_AMZN_TRACE_ID")(it)
         val requestId = it.path("requestId") ?: return@handler Response(BAD_REQUEST).with(
             Body.string(ContentType.TEXT_PLAIN).toLens() of "Missing Request ID"

src/main/kotlin/com/batchofcode/runtimelocal/handler/RootHandler.kt

@@ -2,14 +2,21 @@ package com.batchofcode.runtimelocal.handler
 
 import com.batchofcode.runtimelocal.config.EnvConfig
 import org.http4k.core.then
+import org.http4k.filter.DebuggingFilters
 import org.http4k.filter.ServerFilters
 import org.http4k.server.ApacheServer
 import org.http4k.server.asServer
 
 fun main() {
     val port = EnvConfig.port
 
-    ServerFilters.CatchAll()
+    var filterChain = ServerFilters.CatchAll()
+    if (EnvConfig.debug) {
+        println("REQUEST/RESPONSE DEBUG MODE ENABLED")
+        filterChain = filterChain.then(DebuggingFilters.PrintRequestAndResponse())
+    }
+
+    filterChain
         .then(Routes())
         .asServer(ApacheServer(port)).start()