GH-43 Support mount with receiver (Resolve #43)

routing-core/src/main/kotlin/io/javalin/community/routing/JavalinRoutingExtensions.kt

@@ -1,12 +1,16 @@
 package io.javalin.community.routing
 
 import io.javalin.Javalin
+import io.javalin.config.RouterConfig
 import io.javalin.http.Handler
 import io.javalin.http.HandlerType
 import io.javalin.router.Endpoint
 import io.javalin.router.InternalRouter
+import io.javalin.router.JavalinDefaultRouting
+import io.javalin.router.RoutingApiInitializer
 import io.javalin.router.RoutingSetupScope
 import io.javalin.security.RouteRole
+import java.util.function.Consumer
 
 class JavalinRoutingExtensions(private val javalin: Javalin) {
 
@@ -57,6 +61,18 @@ fun InternalRouter.registerRoute(route: Route, path: String, handler: Handler, v
     }
 }
 
+fun RouterConfig.mount(setup: RoutingSetupScope<JavalinDefaultRouting>): RouterConfig = also {
+    mount(Consumer {
+        setup.invokeAsSamWithReceiver(it)
+    })
+}
+
+fun <SETUP> RouterConfig.mount(initializer: RoutingApiInitializer<SETUP>, setup: RoutingSetupScope<SETUP> = RoutingSetupScope {}): RouterConfig = also {
+    mount(initializer, Consumer {
+        setup.invokeAsSamWithReceiver(it)
+    })
+}
+
 fun <SETUP> RoutingSetupScope<SETUP>.invokeAsSamWithReceiver(receiver: SETUP) {
     with(this) { receiver.setup() }
 }

routing-core/src/test/kotlin/io/javalin/community/routing/JavalinRoutesTest.kt

@@ -51,4 +51,13 @@ class JavalinRoutesTest {
         }
     }
 
+    @Test
+    fun `should enable sam receiver for mount`() {
+        Javalin.create { cfg ->
+            cfg.router.mount {
+                get("/") { it.result("Hello World!") }
+            }
+        }
+    }
+
 }