Add invoke_dsc_resource() tool to MCP

Author: Steve Lee (POWERSHELL HE/HIM) (from Dev Box)

dsc/Cargo.lock

@@ -22,7 +22,7 @@ indicatif = { version = "0.18" }
 jsonschema = { version = "0.33", default-features = false }
 path-absolutize = { version = "3.1" }
 regex = "1.11"
-rmcp = { version = "0.6", features = [
+rmcp = { version = "0.7", features = [
     "server",
     "macros",
     "transport-io",

dsc/Cargo.toml

@@ -66,15 +66,18 @@ serverStopped = "MCP server stopped"
 failedToCreateRuntime = "Failed to create async runtime: %{error}"
 serverWaitFailed = "Failed to wait for MCP server: %{error}"
 
+[mcp.invoke_dsc_resource]
+resourceNotFound = "Resource type '%{resource}' does not exist"
+
 [mcp.list_dsc_functions]
 invalidNamePattern = "Invalid function name pattern '%{pattern}'"
 
 [mcp.list_dsc_resources]
 resourceNotAdapter = "The resource '%{adapter}' is not a valid adapter"
-adapterNotFound = "Adapter '%{adapter}' not found"
+adapterNotFound = "Adapter '%{adapter}' does not exist"
 
 [mcp.show_dsc_resource]
-resourceNotFound = "Resource type '%{type_name}' not found"
+resourceNotFound = "Resource type '%{type_name}' does not exist"
 
 [resolve]
 processingInclude = "Processing Include input"

dsc/locales/en-us.toml

@@ -0,0 +1,102 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+use crate::mcp::mcp_server::McpServer;
+use dsc_lib::{
+    configure::config_doc::ExecutionKind,
+    dscresources::{
+        dscresource::Invoke,
+        invoke_result::{
+            ExportResult,
+            GetResult,
+            SetResult,
+            TestResult,
+        },
+    },
+    DscManager,
+};
+use rmcp::{ErrorData as McpError, Json, tool, tool_router, handler::server::wrapper::Parameters};
+use rust_i18n::t;
+use schemars::JsonSchema;
+use serde::{Deserialize, Serialize};
+use tokio::task;
+
+#[derive(Deserialize, JsonSchema)]
+pub enum DscOperation {
+    Get,
+    Set,
+    Test,
+    Export,
+}
+
+#[derive(Serialize, JsonSchema)]
+pub enum ResourceOperationResult {
+    GetResult(GetResult),
+    SetResult(SetResult),
+    TestResult(TestResult),
+    ExportResult(ExportResult),
+}
+
+#[derive(Deserialize, JsonSchema)]
+pub struct InvokeDscResourceRequest {
+    #[schemars(description = "The operation to perform on the DSC resource")]
+    pub operation: DscOperation,
+    #[schemars(description = "The type name of the DSC resource to invoke")]
+    pub resource_type: String,
+    #[schemars(description = "The properties to pass to the DSC resource as JSON.  Must match the resource JSON schema from `show_dsc_resource` tool.")]
+    pub properties_json: String,
+}
+
+#[tool_router(router = invoke_dsc_resource_router, vis = "pub")]
+impl McpServer {
+    #[tool(
+        description = "Invoke a DSC resource operation (Get, Set, Test, Export) with specified properties in JSON format",
+        annotations(
+            title = "Invoke a DSC resource operation (Get, Set, Test, Export) with specified properties in JSON format",
+            read_only_hint = false,
+            destructive_hint = true,
+            idempotent_hint = true,
+            open_world_hint = true,
+        )
+    )]
+    pub async fn invoke_dsc_resource(&self, Parameters(InvokeDscResourceRequest { operation, resource_type, properties_json }): Parameters<InvokeDscResourceRequest>) -> Result<Json<ResourceOperationResult>, McpError> {
+        let result = task::spawn_blocking(move || {
+            let mut dsc = DscManager::new();
+            let Some(resource) = dsc.find_resource(&resource_type, None) else {
+                return Err(McpError::invalid_request(t!("mcp.invoke_dsc_resource.resourceNotFound", resource = resource_type), None));
+            };
+            match operation {
+                DscOperation::Get => {
+                    let result = match resource.get(&properties_json) {
+                        Ok(res) => res,
+                        Err(e) => return Err(McpError::internal_error(e.to_string(), None)),
+                    };
+                    Ok(ResourceOperationResult::GetResult(result))
+                },
+                DscOperation::Set => {
+                    let result = match resource.set(&properties_json, false, &ExecutionKind::Actual) {
+                        Ok(res) => res,
+                        Err(e) => return Err(McpError::internal_error(e.to_string(), None)),
+                    };
+                    Ok(ResourceOperationResult::SetResult(result))
+                },
+                DscOperation::Test => {
+                    let result = match resource.test(&properties_json) {
+                        Ok(res) => res,
+                        Err(e) => return Err(McpError::internal_error(e.to_string(), None)),
+                    };
+                    Ok(ResourceOperationResult::TestResult(result))
+                },
+                DscOperation::Export => {
+                    let result = match resource.export(&properties_json) {
+                        Ok(res) => res,
+                        Err(e) => return Err(McpError::internal_error(e.to_string(), None)),
+                    };
+                    Ok(ResourceOperationResult::ExportResult(result))
+                }
+            }
+        }).await.map_err(|e| McpError::internal_error(e.to_string(), None))??;
+
+        Ok(Json(result))
+    }
+}

dsc/src/mcp/invoke_dsc_resource.rs

@@ -20,7 +20,11 @@ impl McpServer {
     #[must_use]
     pub fn new() -> Self {
         Self {
-            tool_router: Self::list_dsc_resources_router() + Self::show_dsc_resource_router() + Self::list_dsc_functions_router(),
+            tool_router:
+                Self::invoke_dsc_resource_router()
+                + Self::list_dsc_resources_router()
+                + Self::show_dsc_resource_router()
+                + Self::list_dsc_functions_router(),
         }
     }
 }

dsc/src/mcp/mcp_server.rs

@@ -9,6 +9,7 @@ use rmcp::{
 };
 use rust_i18n::t;
 
+pub mod invoke_dsc_resource;
 pub mod list_dsc_functions;
 pub mod list_dsc_resources;
 pub mod mcp_server;

dsc/src/mcp/mod.rs

@@ -71,6 +71,7 @@ Describe 'Tests for MCP server' {
         }
 
         $tools = @{
+            'invoke_dsc_resource' = $false
             'list_dsc_functions' = $false
             'list_dsc_resources' = $false
             'show_dsc_resource' = $false