From f3e94ba6b2491c2caf1bc5e6fd8c831ac2a3ea8e Mon Sep 17 00:00:00 2001 From: Sebas Date: Tue, 23 Jun 2026 16:12:00 +0000 Subject: [PATCH] Stage 1: MCP-Server aus dtrack-stdio in dtrack-tools verschoben --- crates/dtrack-tools/src/lib.rs | 92 ++++++++++++++++++++++++++++++++++ 1 file changed, 92 insertions(+) create mode 100644 crates/dtrack-tools/src/lib.rs diff --git a/crates/dtrack-tools/src/lib.rs b/crates/dtrack-tools/src/lib.rs new file mode 100644 index 0000000..27c58a4 --- /dev/null +++ b/crates/dtrack-tools/src/lib.rs @@ -0,0 +1,92 @@ +use dtrack_core::DtrackClient; +use rmcp::{ + handler::server::router::tool::ToolRouter, + handler::server::wrapper::Parameters, + model::{ + CallToolResult, Content, Implementation, ProtocolVersion, ServerCapabilities, ServerInfo, + }, + tool, tool_handler, tool_router, + ErrorData as McpError, ServerHandler, +}; +use schemars::JsonSchema; +use serde::Deserialize; + +/// MCP-Server fuer Dependency-Track. Haelt den DT-Client + den vom Macro +/// generierten ToolRouter. +/// +/// Bewusst transport-unabhaengig: `dtrack-stdio` und `dtrack-http` bauen +/// denselben Server und haengen nur ein anderes Frontend (stdio bzw. HTTP) +/// davor. +#[derive(Clone)] +pub struct DtrackServer { + client: DtrackClient, + tool_router: ToolRouter, +} + +/// Argument fuer projektbezogene Tools. +#[derive(Debug, Deserialize, JsonSchema)] +pub struct ProjectArg { + /// UUID des Projekts (aus `list_projects`). + pub uuid: String, +} + +#[tool_router] +impl DtrackServer { + pub fn new(client: DtrackClient) -> Self { + Self { + client, + tool_router: Self::tool_router(), + } + } + + #[tool(description = "Listet aktive Dependency-Track-Projekte auf (Name, Version, UUID).")] + async fn list_projects(&self) -> Result { + let body = self + .client + .list_projects() + .await + .map_err(|e| McpError::internal_error(e.to_string(), None))?; + Ok(CallToolResult::success(vec![Content::text(body)])) + } + + #[tool(description = "Findings (Schwachstellen) eines Projekts anhand seiner UUID.")] + async fn project_findings( + &self, + Parameters(arg): Parameters, + ) -> Result { + let body = self + .client + .project_findings(&arg.uuid) + .await + .map_err(|e| McpError::internal_error(e.to_string(), None))?; + Ok(CallToolResult::success(vec![Content::text(body)])) + } + + #[tool(description = "Aktuelle Sicherheits-Metriken eines Projekts anhand seiner UUID.")] + async fn project_metrics( + &self, + Parameters(arg): Parameters, + ) -> Result { + let body = self + .client + .project_metrics(&arg.uuid) + .await + .map_err(|e| McpError::internal_error(e.to_string(), None))?; + Ok(CallToolResult::success(vec![Content::text(body)])) + } +} + +#[tool_handler] +impl ServerHandler for DtrackServer { + fn get_info(&self) -> ServerInfo { + ServerInfo { + protocol_version: ProtocolVersion::V_2025_06_18, + capabilities: ServerCapabilities::builder().enable_tools().build(), + server_info: Implementation::from_build_env(), + instructions: Some( + "Read-only Zugriff auf Dependency-Track (Projekte, Findings, Metriken)." + .to_string(), + ), + } + } +}