Adds tools to fetch user and team info (#31)

* Implement a wrapper to let the LLMs fetch information about the current user

* Implement a wrapper to let the LLMs to fetch information about the teams members

* Implement 2 new tools that leverage the newly introduced wrappers to get the current user and list team members

* Switch to use organization get_profile, avoid duplication of functionalities

* Improve CSV serialization error handling robustness

Author: danielealbano

src/azure/mod.rs

@@ -5,4 +5,5 @@ pub mod models;
 pub mod organizations;
 pub mod projects;
 pub mod tags;
+pub mod teams;
 pub mod work_items;

src/azure/teams.rs

@@ -0,0 +1,41 @@
+use crate::azure::client::{AzureDevOpsClient, AzureError};
+use reqwest::Method;
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct TeamMember {
+    pub identity: TeamMemberIdentity,
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct TeamMemberIdentity {
+    #[serde(rename = "displayName")]
+    pub display_name: String,
+    #[serde(rename = "uniqueName")]
+    pub unique_name: String,
+    pub id: String,
+}
+
+#[derive(Debug, Deserialize)]
+struct TeamMembersResponse {
+    value: Vec<TeamMember>,
+}
+
+impl AzureDevOpsClient {
+    pub async fn list_team_members(
+        &self,
+        organization: &str,
+        project: &str,
+        team_id: &str,
+    ) -> Result<Vec<TeamMember>, AzureError> {
+        let path = format!(
+            "projects/{}/teams/{}/members?api-version=7.1",
+            project, team_id
+        );
+        let response: TeamMembersResponse = self
+            .org_request(organization, Method::GET, &path, None::<&String>)
+            .await?;
+
+        Ok(response.value)
+    }
+}

src/mcp/server.rs

@@ -432,6 +432,23 @@ struct GetTeamIterationsArgs {
     team_id: String,
 }
 
+#[derive(Deserialize, JsonSchema)]
+struct ListTeamMembersArgs {
+    /// AzDO org name
+    #[serde(deserialize_with = "deserialize_non_empty_string")]
+    organization: String,
+    /// AzDO project name
+    #[serde(deserialize_with = "deserialize_non_empty_string")]
+    project: String,
+    /// Team ID or name
+    team_id: String,
+}
+
+#[derive(Deserialize, JsonSchema)]
+struct GetCurrentUserArgs {
+    // No parameters needed
+}
+
 #[derive(Deserialize, JsonSchema)]
 struct GetWorkItemArgs {
     /// AzDO org name
@@ -821,6 +838,102 @@ impl AzureMcpServer {
         )]))
     }
 
+    #[tool(description = "List team members")]
+    async fn azdo_list_team_members(
+        &self,
+        args: Parameters<ListTeamMembersArgs>,
+    ) -> Result<CallToolResult, McpError> {
+        log::info!("Tool invoked: azdo_list_team_members");
+        let members = self
+            .client
+            .list_team_members(&args.0.organization, &args.0.project, &args.0.team_id)
+            .await
+            .map_err(|e| McpError {
+                code: ErrorCode(-32000),
+                message: e.to_string().into(),
+                data: None,
+            })?;
+
+        let mut wtr = csv::WriterBuilder::new()
+            .has_headers(false)
+            .from_writer(vec![]);
+
+        for member in members {
+            wtr.write_record(&[member.identity.display_name, member.identity.unique_name])
+                .map_err(|e| McpError {
+                    code: ErrorCode(-32000),
+                    message: format!("Failed to write CSV: {}", e).into(),
+                    data: None,
+                })?;
+        }
+
+        wtr.flush().map_err(|e| McpError {
+            code: ErrorCode(-32000),
+            message: format!("Failed to flush CSV: {}", e).into(),
+            data: None,
+        })?;
+
+        let csv_bytes = wtr.into_inner().map_err(|e| McpError {
+            code: ErrorCode(-32000),
+            message: format!("Failed to get CSV bytes: {}", e).into(),
+            data: None,
+        })?;
+
+        let data = String::from_utf8(csv_bytes).map_err(|e| McpError {
+            code: ErrorCode(-32000),
+            message: format!("Failed to convert CSV to string: {}", e).into(),
+            data: None,
+        })?;
+
+        Ok(CallToolResult::success(vec![Content::text(data)]))
+    }
+
+    #[tool(description = "Get current user profile")]
+    async fn azdo_get_current_user(
+        &self,
+        _args: Parameters<GetCurrentUserArgs>,
+    ) -> Result<CallToolResult, McpError> {
+        log::info!("Tool invoked: azdo_get_current_user");
+        let profile = organizations::get_profile(&self.client)
+            .await
+            .map_err(|e| McpError {
+                code: ErrorCode(-32000),
+                message: e.to_string().into(),
+                data: None,
+            })?;
+
+        let mut wtr = csv::WriterBuilder::new()
+            .has_headers(false)
+            .from_writer(vec![]);
+
+        wtr.write_record(&[profile.display_name, profile.email_address])
+            .map_err(|e| McpError {
+                code: ErrorCode(-32000),
+                message: format!("Failed to write CSV: {}", e).into(),
+                data: None,
+            })?;
+
+        wtr.flush().map_err(|e| McpError {
+            code: ErrorCode(-32000),
+            message: format!("Failed to flush CSV: {}", e).into(),
+            data: None,
+        })?;
+
+        let csv_bytes = wtr.into_inner().map_err(|e| McpError {
+            code: ErrorCode(-32000),
+            message: format!("Failed to get CSV bytes: {}", e).into(),
+            data: None,
+        })?;
+
+        let data = String::from_utf8(csv_bytes).map_err(|e| McpError {
+            code: ErrorCode(-32000),
+            message: format!("Failed to convert CSV to string: {}", e).into(),
+            data: None,
+        })?;
+
+        Ok(CallToolResult::success(vec![Content::text(data)]))
+    }
+
     #[tool(description = "List AzDO organizations")]
     async fn azdo_list_organizations(
         &self,