Merge pull request #143 from joreilly/mcp_server_cleanup

mcp server config cleanup

Author: joreilly

mcp-server/src/main/kotlin/server.kt

@@ -1,17 +1,10 @@
 import dev.johnoreilly.climatetrace.data.ClimateTraceRepository
 import dev.johnoreilly.climatetrace.di.initKoin
-import io.ktor.server.application.*
 import io.ktor.server.cio.*
 import io.ktor.server.engine.*
-import io.ktor.server.routing.*
-import io.ktor.server.sse.*
-import io.ktor.util.collections.*
 import io.ktor.utils.io.streams.*
 import io.modelcontextprotocol.kotlin.sdk.*
-import io.modelcontextprotocol.kotlin.sdk.server.Server
-import io.modelcontextprotocol.kotlin.sdk.server.ServerOptions
-import io.modelcontextprotocol.kotlin.sdk.server.SseServerTransport
-import io.modelcontextprotocol.kotlin.sdk.server.StdioServerTransport
+import io.modelcontextprotocol.kotlin.sdk.server.*
 import kotlinx.coroutines.Job
 import kotlinx.coroutines.runBlocking
 import kotlinx.io.asSink
@@ -118,33 +111,10 @@ fun `run mcp server using stdio`() {
  * @param port The port number on which the SSE server should be started.
  */
 fun `run sse mcp server`(port: Int): Unit = runBlocking {
-    val servers = ConcurrentMap<String, Server>()
-
     val server = configureServer()
     embeddedServer(CIO, host = "0.0.0.0", port = port) {
-        install(SSE)
-        routing {
-            sse("/sse") {
-                val transport = SseServerTransport("/message", this)
-
-                servers[transport.sessionId] = server
-
-                server.onClose {
-                    servers.remove(transport.sessionId)
-                }
-
-                server.connect(transport)
-            }
-            post("/message") {
-                val sessionId: String = call.request.queryParameters["sessionId"]!!
-                val transport = servers[sessionId]?.transport as? SseServerTransport
-                if (transport == null) {
-                    call.respond("Session not found", null)
-                    return@post
-                }
-
-                transport.handlePostMessage(call)
-            }
+        mcp {
+            server
         }
     }.start(wait = true)
 }