Azure Omni-Tool MCP Server

A Model Context Protocol (MCP) server in TypeScript that acts as an intelligent bridge between natural language requests and Azure CLI execution.

Features

✅ Plan/Execute Flow - Review commands before execution
✅ Safety Guardrails - Shell injection detection, destructive command warnings
✅ Audit Trail - Operator email tagging for traceability
✅ Retry Logic - Exponential backoff for transient failures
✅ Caching - LRU cache with configurable TTL
✅ Tenant Scoping - Configure tenant/subscription via environment
✅ Azure Service Adapters - Type-safe access to 8 Azure services

Architecture Overview

flowchart TB
    subgraph Client["🖥️ Client Layer"]
        LLM[LLM / AI Agent]
    end

    subgraph MCP["⚙️ MCP Server"]
        direction TB
        Entry[index.ts]
        
        subgraph Tools["Tools"]
            T1[manage_azure_resources]
            T2[get_azure_context]
            T3[azure_service]
        end
        
        subgraph Lib["Core Libraries"]
            Auth[auth.ts]
            Cache[cache.ts]
            CLI[cli-executor.ts]
            Retry[retry.ts]
            Safety[safety.ts]
            Audit[audit.ts]
        end
        
        subgraph Services["Service Adapters"]
            S1[StorageService]
            S2[CosmosService]
            S3[SearchService]
            S4[KustoService]
            S5[MonitorService]
            S6[AppConfigService]
            S7[KeyVaultService]
            S8[PostgresService]
        end
    end

    subgraph Azure["☁️ Azure"]
        AzCLI[Azure CLI]
        AzAPI[Azure APIs]
    end

    LLM -->|MCP Protocol| Entry
    Entry --> Tools
    Tools --> Lib
    Tools --> Services
    Services --> Lib
    Lib --> AzCLI
    Auth --> AzAPI

Request Flow

sequenceDiagram
    participant C as Client
    participant M as MCP Server
    participant S as Safety
    participant E as CLI Executor
    participant A as Azure

    C->>M: Tool Request
    M->>S: Validate Input
    alt Unsafe Command
        S-->>M: Block + Warning
        M-->>C: Error Response
    else Safe
        S-->>M: Approved
        M->>E: Execute Command
        E->>A: az CLI call
        A-->>E: Response
        E-->>M: Result + Parse
        M-->>C: Structured Output
    end

Plan/Execute Flow

flowchart LR
    A[LLM Client] -->|Natural Language| B[MCP Server]
    B --> C{execute_now?}
    C -->|false| D[Return Plan]
    C -->|true| E[Execute CLI]
    E --> F{Success?}
    F -->|Yes| G[Return Output]
    F -->|No| H[Return Error + Analysis]
    H -->|Feedback Loop| A

Quick Start

1. Install Dependencies

npm install

2. Configure Environment

cp .env.example .env
# Edit .env with your settings

3. Build & Run

npm run build
npm start

MCP Client Configuration

{
  "mcpServers": {
    "azure-omni-tool": {
      "command": "node",
      "args": ["path/to/Azure-mcp/dist/index.js"]
    }
  }
}

Tools

manage_azure_resources

Plan and execute Azure CLI commands with safety checks.

| Argument | Type | Description | |----------|------|-------------| | command | string | Azure CLI command | | explanation | string | Why this command was chosen | | execute_now | boolean | false = plan, true = execute |

get_azure_context

Query Azure environment with caching.

| Query Type | Description | |------------|-------------| | subscriptions | List accessible subscriptions | | resource_groups | List resource groups | | resources | List resources | | custom | Custom KQL via Resource Graph |

azure_service

Interact with specific Azure services.

| Service | Actions | |---------|---------| | storage | list, listContainers, listBlobs, getContainer, listTables, queryTable | | cosmos | list, listDatabases, listContainers, query, getContainer | | search | list, listIndexes, getIndex, query, getService | | kusto | list, listDatabases, listTables, getSchema, sample, query | | monitor | list, getWorkspace, listTables, query, listMetrics, getMetrics | | appconfig | list, getStore, listKeyValues, getKeyValue, setKeyValue, lock, unlock | | keyvault | list, getVault, listKeys, getKey, createKey, listSecrets, getSecret, listCertificates | | postgres | list, getServer, listDatabases, listParameters, getParameter, listTables, getTableSchema, query |

Environment Variables

| Variable | Description | Default | |----------|-------------|---------| | AZURE_TENANT_ID | Azure tenant for scoping | - | | AZURE_SUBSCRIPTION_ID | Default subscription | - | | OPERATOR_EMAIL | Email for audit trail | - | | OPERATOR_NAME | Operator name | - | | LOG_LEVEL | Logging level | info | | ENABLE_CACHE | Enable query caching | true | | CACHE_TTL_SECONDS | Cache duration | 300 | | CACHE_CLEANUP_INTERVAL_MS | Cache cleanup interval | 60000 | | MAX_RETRIES | Retry attempts | 3 | | RETRY_DELAY_MS | Base retry delay | 1000 | | COMMAND_TIMEOUT_MS | CLI timeout | 120000 | | AZURE_MCP_INCLUDE_PRODUCTION_CREDENTIALS | Enable Managed Identity | false |

Project Structure

Azure-mcp/
├── src/
│   ├── index.ts                 # MCP server entry
│   ├── lib/
│   │   ├── auth.ts              # Azure credential management
│   │   ├── audit.ts             # Audit trail with correlation IDs
│   │   ├── cache.ts             # LRU cache with TTL
│   │   ├── cli-executor.ts      # Azure CLI wrapper
│   │   ├── config.ts            # Environment config
│   │   ├── logger.ts            # Structured JSON logging
│   │   ├── retry.ts             # Exponential backoff
│   │   ├── safety.ts            # Input sanitization
│   │   └── types.ts             # Shared types
│   ├── services/
│   │   ├── base-service.ts      # Abstract service base
│   │   ├── storage.ts           # Azure Storage
│   │   ├── cosmos.ts            # Cosmos DB
│   │   ├── search.ts            # AI Search
│   │   ├── kusto.ts             # Data Explorer
│   │   ├── monitor.ts           # Monitor / Log Analytics
│   │   ├── appconfig.ts         # App Configuration
│   │   ├── keyvault.ts          # Key Vault
│   │   ├── postgres.ts          # PostgreSQL Flexible Server
│   │   └── index.ts             # Service factory
│   └── tools/
│       ├── azure-manager.ts     # Plan/Execute tool
│       ├── context-retriever.ts # Context queries
│       └── service-tool.ts      # Service adapter tool
├── .env.example
├── package.json
└── tsconfig.json

Prerequisites

• Node.js >= 18.0.0
• Azure CLI installed and authenticated (az login)

License

MIT