Replace http.Server by a generic ListenAndServer

Signed-off-by: Alvaro Saurin <[email protected]>

Author: inercia

server/sse.go

@@ -51,6 +51,12 @@ func (s *sseSession) Initialized() bool {
 
 var _ ClientSession = (*sseSession)(nil)
 
+// ListenAndServer is a type that implements ListenAndServe and Shutdown
+type ListenAndServer interface {
+	ListenAndServe() error
+	Shutdown(ctx context.Context) error
+}
+
 // SSEServer implements a Server-Sent Events (SSE) based MCP server.
 // It provides real-time communication capabilities over HTTP using the SSE protocol.
 type SSEServer struct {
@@ -61,7 +67,7 @@ type SSEServer struct {
 	messageEndpoint              string
 	sseEndpoint                  string
 	sessions                     sync.Map
-	srv                          *http.Server
+	srv                          ListenAndServer
 	contextFunc                  SSEContextFunc
 
 	keepAlive         bool
@@ -131,7 +137,7 @@ func WithSSEEndpoint(endpoint string) SSEOption {
 }
 
 // WithHTTPServer sets the HTTP server instance
-func WithHTTPServer(srv *http.Server) SSEOption {
+func WithHTTPServer(srv ListenAndServer) SSEOption {
 	return func(s *SSEServer) {
 		s.srv = srv
 	}