feat: support vpn mcp server (#206)

Author: Yuan2025801

server/mcp_server_vpn/README.md

@@ -0,0 +1,18 @@
+[project]
+name = "mcp-server-vpn"
+version = "1.0.0"
+description = "VPN MCP Server"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "mcp>=1.10.0",
+    "volcengine-python-sdk>=3.0.1",
+    "volcengine>=1.0.101",
+]
+
+[project.scripts]
+mcp-server-vpn = "mcp_server_vpn.main:main"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"

server/mcp_server_vpn/README_zh.md

@@ -0,0 +1,122 @@
+# This file was autogenerated by uv via the following command:
+#    uv pip compile pyproject.toml -o requirements.txt
+annotated-types==0.7.0
+    # via pydantic
+anyio==4.9.0
+    # via
+    #   httpx
+    #   mcp
+    #   sse-starlette
+    #   starlette
+attrs==25.3.0
+    # via
+    #   jsonschema
+    #   referencing
+beautifulsoup4==4.13.4
+    # via google
+certifi==2025.4.26
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
+    #   volcengine-python-sdk
+charset-normalizer==3.4.2
+    # via requests
+click==8.2.0
+    # via uvicorn
+decorator==5.2.1
+    # via retry
+google==3.0.0
+    # via volcengine
+h11==0.16.0
+    # via
+    #   httpcore
+    #   uvicorn
+httpcore==1.0.9
+    # via httpx
+httpx==0.28.1
+    # via mcp
+httpx-sse==0.4.0
+    # via mcp
+idna==3.10
+    # via
+    #   anyio
+    #   httpx
+    #   requests
+jsonschema==4.25.0
+    # via mcp
+jsonschema-specifications==2025.4.1
+    # via jsonschema
+mcp==1.12.2
+    # via mcp-server-vpn (pyproject.toml)
+protobuf==3.19.6
+    # via volcengine
+py==1.11.0
+    # via retry
+pycryptodome==3.9.9
+    # via volcengine
+pydantic==2.11.4
+    # via
+    #   mcp
+    #   pydantic-settings
+pydantic-core==2.33.2
+    # via pydantic
+pydantic-settings==2.9.1
+    # via mcp
+python-dateutil==2.9.0.post0
+    # via volcengine-python-sdk
+python-dotenv==1.1.0
+    # via pydantic-settings
+python-multipart==0.0.20
+    # via mcp
+pytz==2020.5
+    # via volcengine
+referencing==0.36.2
+    # via
+    #   jsonschema
+    #   jsonschema-specifications
+requests==2.32.4
+    # via volcengine
+retry==0.9.2
+    # via volcengine
+rpds-py==0.26.0
+    # via
+    #   jsonschema
+    #   referencing
+six==1.17.0
+    # via
+    #   python-dateutil
+    #   volcengine
+    #   volcengine-python-sdk
+sniffio==1.3.1
+    # via anyio
+soupsieve==2.7
+    # via beautifulsoup4
+sse-starlette==2.3.5
+    # via mcp
+starlette==0.46.2
+    # via
+    #   mcp
+    #   sse-starlette
+typing-extensions==4.13.2
+    # via
+    #   anyio
+    #   beautifulsoup4
+    #   pydantic
+    #   pydantic-core
+    #   referencing
+    #   typing-inspection
+typing-inspection==0.4.0
+    # via
+    #   pydantic
+    #   pydantic-settings
+urllib3==2.4.0
+    # via
+    #   requests
+    #   volcengine-python-sdk
+uvicorn==0.34.3
+    # via mcp
+volcengine==1.0.101
+    # via mcp-server-vpn (pyproject.toml)
+volcengine-python-sdk==3.0.1
+    # via mcp-server-vpn (pyproject.toml)

server/mcp_server_vpn/pyproject.toml

@@ -0,0 +1,19 @@
+from .base import BaseApi
+from .vpn import VPNClient
+from .models import (
+    DescribeVpnConnectionAttributesResponse,
+    DescribeVpnGatewayAttributesResponse,
+    DescribeVpnConnectionsResponse,
+    DescribeVpnGatewaysResponse,
+    DescribeCustomerGatewaysResponse,
+)
+
+__all__ = [
+    "BaseApi",
+    "VPNClient",
+    "DescribeVpnConnectionAttributesResponse",
+    "DescribeVpnGatewayAttributesResponse",
+    "DescribeVpnConnectionsResponse",
+    "DescribeVpnGatewaysResponse",
+    "DescribeCustomerGatewaysResponse",
+]

server/mcp_server_vpn/requirements.txt

@@ -0,0 +1,76 @@
+import json
+
+from volcengine.ApiInfo import ApiInfo
+from typing import Any
+from volcengine import Credentials
+from volcengine import ServiceInfo
+from volcengine.base import Service
+
+
+class BaseApi(Service.Service):
+    def __init__(self, region, endpoint, api_info, service, ak, sk):
+        """init function.
+        :param region:   region of request
+        :param api_info: an object of volcengine.ApiInfo.ApiInfo()
+        :param endpoint: endpoint of top gateway
+        :param service:  a specific service name registered on top gateway
+        :param ak:       account ak
+        :param sk:       account ak
+        """
+        self.connection_timeout = 10
+        self.socket_timeout = 10
+        self.schema = 'https'
+        self.header = dict()
+        self.header["Content-Type"] = "application/json"
+        self.endpoint = endpoint
+
+        self.credential = Credentials.Credentials(ak, sk, service, region)
+        self.service_info = ServiceInfo.ServiceInfo(
+            self.endpoint,
+            self.header,
+            self.credential,
+            self.connection_timeout,
+            self.socket_timeout,
+            self.schema,
+        )
+        self.api_info = api_info
+        Service.Service.__init__(self, self.service_info, self.api_info)
+
+    @staticmethod
+    def to_params(obj: Any) -> dict[str, Any]:
+        """Convert a request model to a parameters dict, dropping None values.
+
+        ``volcengine-python-sdk`` request models expose ``attribute_map``
+        describing how Python attribute names map to request parameter keys. This
+        helper applies that mapping so that callers can use snake_case attribute
+        names while the API receives the expected camel case keys.
+        """
+
+        if hasattr(obj, "to_dict"):
+            data = obj.to_dict()
+        elif hasattr(obj, "model_dump"):
+            data = obj.model_dump()
+        elif isinstance(obj, dict):
+            data = obj
+        else:
+            data = getattr(obj, "__dict__", {})
+
+        attr_map = getattr(obj, "attribute_map", None)
+        if isinstance(attr_map, dict):
+            data = {attr_map.get(k, k): v for k, v in data.items()}
+
+        return {k: v for k, v in data.items() if v is not None}
+
+    def get(self, action, params, doseq=0):
+        res = super(BaseApi, self).get(action, params, doseq)
+        try:
+            res_json = json.loads(res)
+        except Exception as e:
+            raise Exception("res body is not json, %s, %s" % (e, res))
+        if "ResponseMetadata" not in res_json:
+            raise Exception(
+                "ResponseMetadata not in resp body, action %s, resp %s" % (action, res)
+            )
+        elif "Error" in res_json["ResponseMetadata"]:
+            raise Exception("%s failed, %s" % (action, res))
+        return res_json

server/mcp_server_vpn/src/mcp_server_vpn/__init__.py

@@ -0,0 +1,56 @@
+from pydantic import BaseModel, ConfigDict
+from pydantic.version import VERSION
+
+
+class BaseResponseModel(BaseModel):
+    """Generic response model allowing extra fields."""
+
+    if VERSION.startswith("2"):
+        model_config = ConfigDict(
+            extra="allow",
+            arbitrary_types_allowed=True,
+        )
+    else:
+        class Config:
+            extra = "allow"
+            arbitrary_types_allowed = True
+
+
+class DescribeVpnConnectionAttributesResponse(BaseResponseModel):
+    pass
+
+
+class DescribeVpnGatewayAttributesResponse(BaseResponseModel):
+    pass
+
+
+class DescribeVpnConnectionsResponse(BaseResponseModel):
+    pass
+
+
+class DescribeVpnGatewaysResponse(BaseResponseModel):
+    pass
+
+
+class DescribeVpnGatewayRouteAttributesResponse(BaseResponseModel):
+    pass
+
+
+class DescribeVpnGatewayRoutesResponse(BaseResponseModel):
+    pass
+
+
+class DescribeCustomerGatewaysResponse(BaseResponseModel):
+    pass
+
+
+class DescribeSslVpnClientCertAttributesResponse(BaseResponseModel):
+    pass
+
+
+class DescribeSslVpnClientCertsResponse(BaseResponseModel):
+    pass
+
+
+class DescribeSslVpnServersResponse(BaseResponseModel):
+    pass