use crate::mcp::server::AzureMcpServer;
use hyper_util::{
rt::{TokioExecutor, TokioIo},
server::conn::auto::Builder,
service::TowerToHyperService,
};
use rmcp::transport::streamable_http_server::{
StreamableHttpService, session::local::LocalSessionManager,
};
pub async fn run_server(server: AzureMcpServer, port: u16) -> std::io::Result<()> {
let service = TowerToHyperService::new(StreamableHttpService::new(
move || Ok(server.clone()),
LocalSessionManager::default().into(),
Default::default(),
));
let addr = format!("0.0.0.0:{}", port);
let listener = tokio::net::TcpListener::bind(&addr).await?;
println!("Listening on http://{}", addr);
loop {
let (stream, _) = listener.accept().await?;
let io = TokioIo::new(stream);
let service = service.clone();
tokio::spawn(async move {
if let Err(err) = Builder::new(TokioExecutor::default())
.serve_connection(io, service)
.await
{
eprintln!("Error serving connection: {:?}", err);
}
});
}
}
