[Onboard][Part Two] Mcp Server for Azure DocumentDB (Database)

.github/CODEOWNERS

@@ -166,6 +166,12 @@
 # ServiceLabel: %tools-Docker
 # ServiceOwners:                       @conniey @microsoft/azure-mcp
 
+# PRLabel: %tools-DocumentDb
+/tools/Azure.Mcp.Tools.DocumentDb/     @xingfan-git @microsoft/azure-mcp
+
+# ServiceLabel: %tools-DocumentDb
+# ServiceOwners:                       @xingfan-git
+
 # ServiceLabel: %tools-Eclipse
 # ServiceOwners:                       @srnagar @microsoft/azure-mcp
 

Directory.Packages.props

@@ -83,6 +83,7 @@
     <PackageVersion Include="Microsoft.Extensions.Hosting" Version="10.0.2" />
     <PackageVersion Include="ModelContextProtocol" Version="1.0.0" />
     <PackageVersion Include="ModelContextProtocol.AspNetCore" Version="1.0.0" />
+    <PackageVersion Include="MongoDB.Driver" Version="3.2.0" />
     <PackageVersion Include="MySqlConnector" Version="2.4.0" />
     <PackageVersion Include="Npgsql" Version="10.0.1" />
     <PackageVersion Include="YamlDotNet" Version="16.3.0" />

eng/scripts/New-BuildInfo.ps1

@@ -170,6 +170,53 @@ function CheckVariable($name) {
     return $value
 }
 
+function Test-ProjectUsesMongoDbDriver {
+    param(
+        [string] $ProjectPath
+    )
+
+    if (!(Test-Path $ProjectPath)) {
+        return $false
+    }
+
+    $projectContent = Get-Content $ProjectPath -Raw
+    return $projectContent -match '<PackageReference\s+Include="MongoDB\.Driver"'
+}
+
+function Test-ServerHasMongoDbDependency {
+    param(
+        [System.IO.FileInfo] $ServerProject
+    )
+
+    $serverProjectDirectory = Split-Path $ServerProject.FullName -Parent
+    $serverProjectContent = Get-Content $ServerProject.FullName -Raw
+    $projectReferenceMatches = [regex]::Matches($serverProjectContent, '<ProjectReference\s+Include="([^"]+)"')
+
+    foreach ($projectReferenceMatch in $projectReferenceMatches) {
+        $projectReference = $projectReferenceMatch.Groups[1].Value
+        if (-not $projectReference) {
+            continue
+        }
+
+        $expandedReference = $projectReference.Replace('$(RepoRoot)', $RepoRoot)
+        $expandedReference = $expandedReference.Replace('$(MSBuildThisFileDirectory)', "$serverProjectDirectory/")
+        $expandedReference = $expandedReference.Replace('\', '/')
+
+        if (-not [System.IO.Path]::IsPathRooted($expandedReference)) {
+            $expandedReference = (Join-Path $serverProjectDirectory $expandedReference).Replace('\', '/')
+        }
+
+        $referencedProjects = @(Get-ChildItem -Path $expandedReference -File -ErrorAction SilentlyContinue)
+        foreach ($referencedProject in $referencedProjects) {
+            if (Test-ProjectUsesMongoDbDriver -ProjectPath $referencedProject.FullName) {
+                return $true
+            }
+        }
+    }
+
+    return $false
+}
+
 $windowsPool = CheckVariable 'WINDOWSPOOL'
 $linuxPool = CheckVariable 'LINUXPOOL'
 $linuxArm64Pool = CheckVariable 'LINUXARM64POOL'
@@ -387,6 +434,13 @@ function Get-ServerDetails {
             $version.PrereleaseNumber = $BuildId
         }
 
+        # Check if this server depends on MongoDB.Driver (incompatible with IL trimming)
+        $hasMongoDbDependency = Test-ServerHasMongoDbDependency -ServerProject $serverProject
+
+        if ($hasMongoDbDependency) {
+            Write-Host "Server $serverName depends on DocumentDb (with MongoDB.Driver) - trimming will be disabled" -ForegroundColor Yellow
+        }
+
         # Calculate VSIX version based on server version
         $vsixVersion = $null
         $vsixIsPrerelease = $false
@@ -473,7 +527,8 @@ function Get-ServerDetails {
                         architecture = $arch
                         extension = $os.extension
                         native = $false
-                        trimmed = $true
+                        # Disable trimming for servers with MongoDB.Driver dependency (uses extensive reflection)
+                        trimmed = !$hasMongoDbDependency
                     }
                 }
             }
@@ -498,7 +553,7 @@ function Get-ServerDetails {
                 architecture = $additionalPlatform.architecture
                 extension = $os.extension
                 native = $additionalPlatform.native
-                trimmed = $additionalPlatform.trimmed
+                trimmed = $additionalPlatform.trimmed -and !$hasMongoDbDependency
                 specialPurpose = $additionalPlatform.specialPurpose
             }
         }

servers/Azure.Mcp.Server/Azure.Mcp.Server.slnx

@@ -179,6 +179,14 @@
     <Project Path="../../tools/Azure.Mcp.Tools.DeviceRegistry/tests/Azure.Mcp.Tools.DeviceRegistry.UnitTests/Azure.Mcp.Tools.DeviceRegistry.UnitTests.csproj" />
     <Project Path="../../tools/Azure.Mcp.Tools.DeviceRegistry/tests/Azure.Mcp.Tools.DeviceRegistry.LiveTests/Azure.Mcp.Tools.DeviceRegistry.LiveTests.csproj" />
   </Folder>
+  <Folder Name="/tools/Azure.Mcp.Tools.DocumentDb/" />
+  <Folder Name="/tools/Azure.Mcp.Tools.DocumentDb/src/">
+    <Project Path="../../tools/Azure.Mcp.Tools.DocumentDb/src/Azure.Mcp.Tools.DocumentDb.csproj" />
+  </Folder>
+  <Folder Name="/tools/Azure.Mcp.Tools.DocumentDb/tests/">
+    <Project Path="../../tools/Azure.Mcp.Tools.DocumentDb/tests/Azure.Mcp.Tools.DocumentDb.LiveTests/Azure.Mcp.Tools.DocumentDb.LiveTests.csproj" />
+    <Project Path="../../tools/Azure.Mcp.Tools.DocumentDb/tests/Azure.Mcp.Tools.DocumentDb.UnitTests/Azure.Mcp.Tools.DocumentDb.UnitTests.csproj" />
+  </Folder>
   <Folder Name="/tools/Azure.Mcp.Tools.EventGrid/" />
   <Folder Name="/tools/Azure.Mcp.Tools.EventGrid/src/">
     <Project Path="../../tools/Azure.Mcp.Tools.EventGrid/src/Azure.Mcp.Tools.EventGrid.csproj" />

servers/Azure.Mcp.Server/README.md

@@ -963,6 +963,18 @@ Example prompts that generate Azure CLI commands:
 * "Get Azure Data Explorer databases in cluster 'mycluster'"
 * "Sample 10 rows from table 'StormEvents' in Azure Data Explorer database 'db1'"
 
+### 🗄️ Azure DocumentDB (with MongoDB compatibility)
+
+* "List indexes for collection 'items' in DocumentDB database 'test'"
+* "Create an index on field 'category' for collection 'items' in DocumentDB database 'test'"
+* "Drop index 'category_1' from collection 'items' in DocumentDB database 'test'"
+* "Show index statistics for collection 'items' in DocumentDB database 'test'"
+* "Show current DocumentDB operations"
+* "List all databases in DocumentDB"
+* "Get statistics for database 'mydb'"
+* "Get details for database 'analytics' in DocumentDB"
+* "Drop database 'testdb'"
+
 ### 📣 Azure Event Grid
 
 * "List all Event Grid topics in subscription 'my-subscription'"

servers/Azure.Mcp.Server/changelog-entries/1773124572664.yaml

@@ -0,0 +1,4 @@
+pr: 1968
+changes:
+  - section: "Features Added"
+    description: "Added mcp tools for managing Azure DocumentDB (with MongoDB compatibility) index"

servers/Azure.Mcp.Server/changelog-entries/1773579786664.yaml

@@ -0,0 +1,3 @@
+changes:
+  - section: "Features Added"
+    description: "Added mcp tools for managing Azure DocumentDB (with MongoDB compatibility) database"

servers/Azure.Mcp.Server/docs/azmcp-commands.md

@@ -1694,6 +1694,57 @@ azmcp deviceregistry namespace list --subscription <subscription> \
                                     [--resource-group <resource-group>]
 ```
 
+### Azure DocumentDB (with MongoDB compatibility) Operations
+
+```bash
+# List all indexes on a collection
+# ❌ Destructive | ✅ Idempotent | ❌ OpenWorld | ✅ ReadOnly | ❌ Secret | ❌ LocalRequired
+azmcp documentdb index list indexes --connection-string <connection-string> \
+                                    --db-name <db-name> \
+                                    --collection-name <collection-name>
+
+# Create an index on a collection
+# ✅ Destructive | ❌ Idempotent | ❌ OpenWorld | ❌ ReadOnly | ❌ Secret | ❌ LocalRequired
+azmcp documentdb index create index --connection-string <connection-string> \
+                                    --db-name <db-name> \
+                                    --collection-name <collection-name> \
+                                    --keys <json-index-keys> \
+                                    [--options <json-index-options>]
+
+# Drop an index from a collection
+# ✅ Destructive | ❌ Idempotent | ❌ OpenWorld | ❌ ReadOnly | ❌ Secret | ❌ LocalRequired
+azmcp documentdb index drop index --connection-string <connection-string> \
+                                  --db-name <db-name> \
+                                  --collection-name <collection-name> \
+                                  --index-name <index-name>
+
+# Get index statistics for a collection
+# ❌ Destructive | ✅ Idempotent | ❌ OpenWorld | ✅ ReadOnly | ❌ Secret | ❌ LocalRequired
+azmcp documentdb index index stats --connection-string <connection-string> \
+                                   --db-name <db-name> \
+                                   --collection-name <collection-name>
+
+# Get current DocumentDB operations
+# ❌ Destructive | ✅ Idempotent | ❌ OpenWorld | ✅ ReadOnly | ❌ Secret | ❌ LocalRequired
+azmcp documentdb index current ops --connection-string <connection-string> \
+                                   [--ops <json-filter>]
+
+# List all databases or inspect a single database
+# ❌ Destructive | ✅ Idempotent | ❌ OpenWorld | ✅ ReadOnly | ❌ Secret | ❌ LocalRequired
+azmcp documentdb database list databases --connection-string <connection-string> \
+                                         [--db-name <db-name>]
+
+# Get statistics for a database
+# ❌ Destructive | ✅ Idempotent | ❌ OpenWorld | ✅ ReadOnly | ❌ Secret | ❌ LocalRequired
+azmcp documentdb database db stats --connection-string <connection-string> \
+                                   --db-name <db-name>
+
+# Drop a database
+# ✅ Destructive | ❌ Idempotent | ❌ OpenWorld | ❌ ReadOnly | ❌ Secret | ❌ LocalRequired
+azmcp documentdb database drop database --connection-string <connection-string> \
+                                        --db-name <db-name>
+```
+
 ### Azure Event Grid Operations
 
 ```bash

servers/Azure.Mcp.Server/docs/e2eTestPrompts.md

@@ -356,6 +356,28 @@ This file contains prompts used for end-to-end testing to ensure each tool is in
 | deviceregistry_namespace_list | List Device Registry namespaces in resource group <resource_group_name> |
 | deviceregistry_namespace_list | What Device Registry namespaces do I have in my Azure subscription? |
 
+## Azure DocumentDB (with MongoDB compatibility)
+
+| Tool Name | Test Prompt |
+|:----------|:----------|
+| documentdb_index_list_indexes | List indexes for collection <collection-name> in DocumentDB database <db-name> |
+| documentdb_index_list_indexes | Show me all indexes on collection <collection-name> in database <db-name> |
+| documentdb_index_create_index | Create an index on collection <collection-name> in DocumentDB database <db-name> using keys <keys> |
+| documentdb_index_create_index | Add a DocumentDB index for collection <collection-name> in database <db-name> with keys <keys> and options <options> |
+| documentdb_index_drop_index | Drop index <index-name> from collection <collection-name> in DocumentDB database <db-name> |
+| documentdb_index_drop_index | Remove the <index-name> index from DocumentDB collection <collection-name> in database <db-name> |
+| documentdb_index_index_stats | Show index statistics for collection <collection-name> in DocumentDB database <db-name> |
+| documentdb_index_index_stats | Get DocumentDB index stats for collection <collection-name> in database <db-name> |
+| documentdb_index_current_ops | Show current DocumentDB operations |
+| documentdb_index_current_ops | Get current DocumentDB operations filtered by <ops> |
+| documentdb_database_db_stats | Get statistics for database <db-name> |
+| documentdb_database_db_stats | Show me stats for DocumentDB database <db-name> |
+| documentdb_database_drop_database | Drop database <db-name> |
+| documentdb_database_drop_database | Delete the database <db-name> from DocumentDB |
+| documentdb_database_list_databases | List all databases in DocumentDB |
+| documentdb_database_list_databases | Show me all DocumentDB databases |
+| documentdb_database_list_databases | Get details for database <db-name> in DocumentDB |
+
 ## Azure Event Grid
 
 | Tool Name | Test Prompt |

servers/Azure.Mcp.Server/src/Program.cs

@@ -104,6 +104,7 @@ private static IAreaSetup[] RegisterAreas()
             new Azure.Mcp.Tools.AzureTerraformBestPractices.AzureTerraformBestPracticesSetup(),
             new Azure.Mcp.Tools.Deploy.DeploySetup(),
             new Azure.Mcp.Tools.DeviceRegistry.DeviceRegistrySetup(),
+            new Azure.Mcp.Tools.DocumentDb.DocumentDbSetup(),
             new Azure.Mcp.Tools.EventGrid.EventGridSetup(),
             new Azure.Mcp.Tools.Acr.AcrSetup(),
             new Azure.Mcp.Tools.Advisor.AdvisorSetup(),

servers/Azure.Mcp.Server/src/Resources/consolidated-tools.json

@@ -181,6 +181,142 @@
         "cosmos_database_container_item_query"
       ]
     },
+    {
+      "name": "inspect_azure_documentdb_indexes_and_diagnostics",
+      "description": "Inspect Azure DocumentDB collection indexes, index statistics, and current operations by supplying a connection string for each request.",
+      "toolMetadata": {
+        "destructive": {
+          "value": false,
+          "description": "This tool performs only additive updates without deleting or modifying existing resources."
+        },
+        "idempotent": {
+          "value": true,
+          "description": "Running this operation multiple times with the same arguments produces the same result without additional effects."
+        },
+        "openWorld": {
+          "value": false,
+          "description": "This tool's domain of interaction is closed and well-defined, limited to a specific set of entities (like memory access)."
+        },
+        "readOnly": {
+          "value": true,
+          "description": "This tool only performs read operations without modifying any state or data."
+        },
+        "secret": {
+          "value": false,
+          "description": "This tool does not handle sensitive or secret information."
+        },
+        "localRequired": {
+          "value": false,
+          "description": "This tool is available in both local and remote server modes."
+        }
+      },
+      "mappedToolList": [
+        "documentdb_index_list_indexes",
+        "documentdb_index_index_stats",
+        "documentdb_index_current_ops"
+      ]
+    },
+    {
+      "name": "manage_azure_documentdb_indexes",
+      "description": "Create or drop indexes in Azure DocumentDB collections by supplying a connection string for each request.",
+      "toolMetadata": {
+        "destructive": {
+          "value": true,
+          "description": "This tool may delete or modify existing resources in its environment."
+        },
+        "idempotent": {
+          "value": false,
+          "description": "Running this operation multiple times with the same arguments may have additional effects or produce different results."
+        },
+        "openWorld": {
+          "value": false,
+          "description": "This tool's domain of interaction is closed and well-defined, limited to a specific set of entities (like memory access)."
+        },
+        "readOnly": {
+          "value": false,
+          "description": "This tool may modify its environment and perform write operations (create, update, delete)."
+        },
+        "secret": {
+          "value": false,
+          "description": "This tool does not handle sensitive or secret information."
+        },
+        "localRequired": {
+          "value": false,
+          "description": "This tool is available in both local and remote server modes."
+        }
+      },
+      "mappedToolList": [
+        "documentdb_index_create_index",
+        "documentdb_index_drop_index"
+      ]
+    },
+    {
+      "name": "get_azure_documentdb_database_details",
+      "description": "List DocumentDB databases, inspect a specific database, and retrieve database statistics including collection counts and storage usage.",
+      "toolMetadata": {
+        "destructive": {
+          "value": false,
+          "description": "This tool performs only additive updates without deleting or modifying existing resources."
+        },
+        "idempotent": {
+          "value": true,
+          "description": "Running this operation multiple times with the same arguments produces the same result without additional effects."
+        },
+        "openWorld": {
+          "value": false,
+          "description": "This tool's domain of interaction is closed and well-defined, limited to a specific set of entities (like memory access)."
+        },
+        "readOnly": {
+          "value": true,
+          "description": "This tool only performs read operations without modifying any state or data."
+        },
+        "secret": {
+          "value": false,
+          "description": "This tool does not handle sensitive or secret information."
+        },
+        "localRequired": {
+          "value": false,
+          "description": "This tool is available in both local and remote server modes."
+        }
+      },
+      "mappedToolList": [
+        "documentdb_database_list_databases",
+        "documentdb_database_db_stats"
+      ]
+    },
+    {
+      "name": "delete_azure_documentdb_databases",
+      "description": "Delete DocumentDB databases by dropping a database and all of its collections and data.",
+      "toolMetadata": {
+        "destructive": {
+          "value": true,
+          "description": "This tool may delete or modify existing resources in its environment."
+        },
+        "idempotent": {
+          "value": false,
+          "description": "Running this operation multiple times with the same arguments may have additional effects or produce different results."
+        },
+        "openWorld": {
+          "value": false,
+          "description": "This tool's domain of interaction is closed and well-defined, limited to a specific set of entities (like memory access)."
+        },
+        "readOnly": {
+          "value": false,
+          "description": "This tool may modify its environment and perform write operations (create, update, delete)."
+        },
+        "secret": {
+          "value": false,
+          "description": "This tool does not handle sensitive or secret information."
+        },
+        "localRequired": {
+          "value": false,
+          "description": "This tool is available in both local and remote server modes."
+        }
+      },
+      "mappedToolList": [
+        "documentdb_database_drop_database"
+      ]
+    },
     {
       "name": "create_azure_sql_databases_and_servers",
       "description": "Create new Azure SQL databases and SQL servers with configurable performance tiers and settings.",

tools/Azure.Mcp.Tools.DocumentDb/src/AssemblyInfo.cs

@@ -0,0 +1,6 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+using System.Runtime.CompilerServices;
+
+[assembly: InternalsVisibleTo("Azure.Mcp.Tools.DocumentDb.UnitTests")]