Merge pull request #1 from danbeldev/develop

feat: add afterHandle to Dispatcher for post-processing responses

Author: danbeldev

alice-ktx/src/test/kotlin/com/github/alice/ktx/middleware/MiddlewareTest.kt

@@ -8,10 +8,10 @@ import org.junit.Test
 class MiddlewareTest {
 
     @Test
-    fun `should register inner middleware via DSL`()  {
+    fun `should register inner middleware via DSL`() {
         val dispatcher = Dispatcher()
 
-        dispatcher.innerMiddleware { response {  } }
+        dispatcher.innerMiddleware { response { } }
 
         Assert.assertEquals(1, dispatcher.middlewares[MiddlewareType.INNER]?.size)
     }
@@ -20,8 +20,17 @@ class MiddlewareTest {
     fun `should register outer middleware via DSL`() {
         val dispatcher = Dispatcher()
 
-        dispatcher.outerMiddleware { response {  } }
+        dispatcher.outerMiddleware { response { } }
 
         Assert.assertEquals(1, dispatcher.middlewares[MiddlewareType.OUTER]?.size)
     }
+
+    @Test
+    fun `should register after handle via DSL`() {
+        val dispatcher = Dispatcher()
+
+        dispatcher.afterHandle { _, res -> res }
+
+        Assert.assertEquals(1, dispatcher.postMiddlewares.size)
+    }
 }

examples/src/main/kotlin/com/github/examples/AfterHandle.kt

@@ -0,0 +1,33 @@
+package com.github.examples
+
+import com.github.alice.ktx.dispatch
+import com.github.alice.ktx.handlers.impl.request
+import com.github.alice.ktx.middleware.afterHandle
+import com.github.alice.ktx.models.response.response
+import com.github.alice.ktx.skill
+import com.github.alice.ktx.webhook.impl.ktorWebhookServer
+
+fun main() {
+    skill {
+        webhookServer = ktorWebhookServer {
+            port = 8080
+            path = "/alice"
+        }
+
+        dispatch {
+            afterHandle { processRequestEnvironment, messageResponse ->
+                messageResponse.copy(
+                    response = messageResponse.response?.copy(
+                        text = messageResponse.response?.text + processRequestEnvironment.message.version
+                    )
+                )
+            }
+
+            request {
+                response {
+                    text = "test after handel, version alice: "
+                }
+            }
+        }
+    }.run()
+}