use anyhow::Result;
use kodegen_server_http::{run_http_server, Managers, RouterSet, register_tool};
use rmcp::handler::server::router::{prompt::PromptRouter, tool::ToolRouter};
#[tokio::main]
async fn main() -> Result<()> {
run_http_server("prompt", |_config, _tracker| Box::pin(async move {
let mut tool_router = ToolRouter::new();
let mut prompt_router = PromptRouter::new();
let managers = Managers::new();
let manager = kodegen_tools_prompt::PromptManager::new();
manager.init().await?;
use kodegen_tools_prompt::*;
(tool_router, prompt_router) = register_tool(
tool_router,
prompt_router,
AddPromptTool::with_manager(manager.clone()),
);
(tool_router, prompt_router) = register_tool(
tool_router,
prompt_router,
EditPromptTool::with_manager(manager.clone()),
);
(tool_router, prompt_router) = register_tool(
tool_router,
prompt_router,
DeletePromptTool::with_manager(manager.clone()),
);
(tool_router, prompt_router) = register_tool(
tool_router,
prompt_router,
GetPromptTool::with_manager(manager.clone()),
);
Ok(RouterSet::new(tool_router, prompt_router, managers))
}))
.await
}