awslabs.elasticache-mcp-server 0.1.1__py3-none-any.whl

awslabs/__init__.py

@@ -0,0 +1,16 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# This file is part of the awslabs namespace.
+# It is intentionally minimal to support PEP 420 namespace packages.

awslabs/elasticache_mcp_server/__init__.py

@@ -0,0 +1,17 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""awslabs.elasticache-mcp-server"""
+
+__version__ = '0.1.0'

awslabs/elasticache_mcp_server/common/__init__.py

@@ -0,0 +1,15 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Common package for ElastiCache MCP server."""

awslabs/elasticache_mcp_server/common/connection.py

@@ -0,0 +1,117 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Connection management for AWS services used by MCP Server."""
+
+import boto3
+import os
+from botocore.config import Config
+from typing import Any, Optional
+
+
+class BaseConnectionManager:
+    """Base class for AWS service connection managers."""
+
+    _client: Optional[Any] = None
+    _service_name: str = ''  # Must be overridden by subclasses
+    _env_prefix: str = ''  # Must be overridden by subclasses
+
+    @classmethod
+    def get_connection(cls) -> Any:
+        """Get or create an AWS service client connection with retry capabilities.
+
+        Returns:
+            boto3.client: An AWS service client configured with retries
+        """
+        if cls._client is None:
+            # Get AWS configuration from environment
+            aws_profile = os.environ.get('AWS_PROFILE', 'default')
+            aws_region = os.environ.get('AWS_REGION', 'us-east-1')
+
+            # Configure retry settings
+            max_retries = int(os.environ.get(f'{cls._env_prefix}_MAX_RETRIES', '3'))
+            retry_mode = os.environ.get(f'{cls._env_prefix}_RETRY_MODE', 'standard')
+            connect_timeout = int(os.environ.get(f'{cls._env_prefix}_CONNECT_TIMEOUT', '5'))
+            read_timeout = int(os.environ.get(f'{cls._env_prefix}_READ_TIMEOUT', '10'))
+
+            # Create boto3 config with retry settings
+            config = Config(
+                retries={'max_attempts': max_retries, 'mode': retry_mode},
+                connect_timeout=connect_timeout,
+                read_timeout=read_timeout,
+                # Configure custom user agent to identify requests from LLM/MCP
+                user_agent_extra='MCP/ElastiCacheServer',
+            )
+
+            # Initialize AWS client with session and config
+            # so that if user changes credential, it will be reflected immediately in the next call
+            session = boto3.Session(profile_name=aws_profile, region_name=aws_region)
+            cls._client = session.client(service_name=cls._service_name, config=config)
+
+        return cls._client
+
+    @classmethod
+    def close_connection(cls) -> None:
+        """Close the AWS service client connection."""
+        if cls._client is not None:
+            cls._client.close()
+            cls._client = None
+
+
+class ElastiCacheConnectionManager(BaseConnectionManager):
+    """Manages connection to ElastiCache using boto3."""
+
+    _client: Optional[Any] = None
+    _service_name = 'elasticache'
+    _env_prefix = 'ELASTICACHE'
+
+
+class EC2ConnectionManager(BaseConnectionManager):
+    """Manages connection to EC2 using boto3."""
+
+    _client: Optional[Any] = None
+    _service_name = 'ec2'
+    _env_prefix = 'EC2'
+
+
+class CloudWatchLogsConnectionManager(BaseConnectionManager):
+    """Manages connection to CloudWatch Logs using boto3."""
+
+    _client: Optional[Any] = None
+    _service_name = 'logs'
+    _env_prefix = 'CLOUDWATCH_LOGS'
+
+
+class FirehoseConnectionManager(BaseConnectionManager):
+    """Manages connection to Kinesis Firehose using boto3."""
+
+    _client: Optional[Any] = None
+    _service_name = 'firehose'
+    _env_prefix = 'FIREHOSE'
+
+
+class CostExplorerConnectionManager(BaseConnectionManager):
+    """Manages connection to AWS Cost Explorer using boto3."""
+
+    _client: Optional[Any] = None
+    _service_name = 'ce'
+    _env_prefix = 'COST_EXPLORER'
+
+
+class CloudWatchConnectionManager(BaseConnectionManager):
+    """Manages connection to CloudWatch using boto3."""
+
+    _client: Optional[Any] = None
+    _service_name = 'cloudwatch'
+    _env_prefix = 'CLOUDWATCH'

awslabs/elasticache_mcp_server/common/decorators.py

@@ -0,0 +1,41 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Decorators for ElastiCache MCP Server."""
+
+from functools import wraps
+from typing import Any, Callable
+
+
+def handle_exceptions(func: Callable) -> Callable:
+    """Decorator to handle exceptions in ElastiCache operations.
+
+    Wraps the function in a try-catch block and returns any exceptions
+    in a standardized error format.
+
+    Args:
+        func: The function to wrap
+
+    Returns:
+        The wrapped function that handles exceptions
+    """
+
+    @wraps(func)
+    async def wrapper(*args: Any, **kwargs: Any):
+        try:
+            return await func(*args, **kwargs)
+        except Exception as e:
+            return {'error': str(e)}
+
+    return wrapper

awslabs/elasticache_mcp_server/common/server.py

@@ -0,0 +1,30 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Common MCP server configuration."""
+
+from mcp.server.fastmcp import FastMCP
+
+
+mcp = FastMCP(
+    'awslabs.elasticache-mcp-server',
+    instructions="""AWS ElastiCache MCP Server provides tools for interacting with Amazon ElastiCache.
+    These tools allow you to describe and manage serverless caches in your AWS account.
+    You can use these capabilities to get information about cache configurations, endpoints, and more.""",
+    dependencies=[
+        'pydantic',
+        'loguru',
+        'boto3',
+    ],
+)

awslabs/elasticache_mcp_server/context.py

@@ -0,0 +1,39 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Context management for ElastiCache MCP Server."""
+
+
+class Context:
+    """Context class for ElastiCache MCP Server."""
+
+    _readonly = False
+
+    @classmethod
+    def initialize(cls, readonly: bool = False):
+        """Initialize the context.
+
+        Args:
+            readonly: Whether to run in readonly mode
+        """
+        cls._readonly = readonly
+
+    @classmethod
+    def readonly_mode(cls) -> bool:
+        """Check if the server is running in readonly mode.
+
+        Returns:
+            True if readonly mode is enabled, False otherwise
+        """
+        return cls._readonly

awslabs/elasticache_mcp_server/main.py

@@ -0,0 +1,52 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""awslabs elasticache MCP Server implementation."""
+
+import argparse
+from awslabs.elasticache_mcp_server.common.server import mcp
+from awslabs.elasticache_mcp_server.context import Context
+from awslabs.elasticache_mcp_server.tools import (  # noqa: F401 - imported for side effects to register tools
+    cc,
+    ce,
+    cw,
+    cwlogs,
+    firehose,
+    misc,
+    rg,
+    serverless,
+)
+from loguru import logger
+
+
+def main():
+    """Run the MCP server with CLI argument support."""
+    parser = argparse.ArgumentParser(
+        description='An AWS Labs Model Context Protocol (MCP) server for interacting with Amazon ElastiCache'
+    )
+    parser.add_argument(
+        '--readonly',
+        action=argparse.BooleanOptionalAction,
+        help='Prevents the MCP server from performing mutating operations',
+    )
+
+    args = parser.parse_args()
+    Context.initialize(args.readonly)
+
+    logger.info('Amazon ElastiCache MCP Server Started...')
+    mcp.run()
+
+
+if __name__ == '__main__':
+    main()

awslabs/elasticache_mcp_server/tools/__init__.py

@@ -0,0 +1,15 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Tools package for ElastiCache MCP server."""

awslabs/elasticache_mcp_server/tools/cc/__init__.py

@@ -0,0 +1,31 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Cache cluster tools for ElastiCache MCP server."""
+
+from .create import create_cache_cluster
+from .delete import delete_cache_cluster
+from .describe import describe_cache_clusters
+from .modify import modify_cache_cluster
+from .connect import connect_jump_host_cc, get_ssh_tunnel_command_cc, create_jump_host_cc
+
+__all__ = [
+    'create_cache_cluster',
+    'delete_cache_cluster',
+    'describe_cache_clusters',
+    'modify_cache_cluster',
+    'connect_jump_host_cc',
+    'get_ssh_tunnel_command_cc',
+    'create_jump_host_cc',
+]