Add files via upload

Author: mortensi

pyproject.toml

@@ -0,0 +1,12 @@
+[project]
+name = "redis-mcp-server"
+version = "0.1.0"
+description = "Redis MCP Server, by Redis"
+readme = "README.md"
+requires-python = ">=3.13"
+dependencies = [
+    "mcp[cli]>=1.5.0",
+    "redis>=5.2.1",
+    "pydantic-settings>=2.0.0",
+    "dotenv>=0.9.9",
+]

src/__init__.py

@@ -0,0 +1,14 @@
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+REDIS_CFG = {"host": os.getenv('REDIS_HOST', '127.0.0.1'),
+             "port": int(os.getenv('REDIS_PORT',6379)),
+             "password": os.getenv('REDIS_PWD',''),
+             "ssl": os.getenv('REDIS_SSL', False),
+             "ssl_ca_path": os.getenv('REDIS_CA_PATH',''),
+             "ssl_keyfile": os.getenv('REDIS_SSL_KEYFILE', ''),
+             "ssl_certfile": os.getenv('REDIS_SSL_CERTFILE', ''),
+             "ssl_cert_reqs": os.getenv('REDIS_CERT_REQS', ''),
+             "ssl_ca_certs": os.getenv('REDIS_CA_CERTS', '')}

src/common/__init__.py

@@ -0,0 +1,46 @@
+import sys
+from version import __version__
+import redis
+from redis import Redis
+from typing import Optional
+from common.config import REDIS_CFG
+
+class RedisConnectionManager:
+    _instance: Optional[Redis] = None
+
+    @classmethod
+    def get_connection(cls, decode_responses=True) -> Redis:
+        if cls._instance is None:
+            try:
+                print("Initializing Redis connection", file=sys.stderr)
+                cls._instance = redis.StrictRedis(
+                    host=REDIS_CFG["host"],
+                    port=REDIS_CFG["port"],
+                    password=REDIS_CFG["password"],
+                    ssl=REDIS_CFG["ssl"],
+                    ssl_ca_path=REDIS_CFG["ssl_ca_path"],
+                    decode_responses=decode_responses,
+                    max_connections=10,
+                    lib_name=f"redis-py(mcp-server_v{__version__})"
+                )
+
+            except redis.exceptions.ConnectionError:
+                print("Failed to connect to Redis server", file=sys.stderr)
+                raise
+            except redis.exceptions.AuthenticationError:
+                print("Authentication failed", file=sys.stderr)
+                raise
+            except redis.exceptions.TimeoutError:
+                print("Connection timed out", file=sys.stderr)
+                raise
+            except redis.exceptions.ResponseError as e:
+                print(f"Response error: {e}", file=sys.stderr)
+                raise
+            except redis.exceptions.RedisError as e:
+                print(f"Redis error: {e}", file=sys.stderr)
+                raise
+            except Exception as e:
+                print(f"Unexpected error: {e}", file=sys.stderr)
+                raise
+
+        return cls._instance

src/common/config.py

@@ -0,0 +1,8 @@
+from mcp.server.fastmcp import FastMCP
+
+# Initialize FastMCP server
+mcp = FastMCP(
+    "Redis MCP Server",
+    dependencies=["redis", "pydantic"]
+)
+

src/common/connection.py

@@ -0,0 +1,18 @@
+from common.connection import RedisConnectionManager
+from common.server import mcp
+import tools.server_management
+import tools.misc 
+import tools.redis_query_engine
+import tools.hash  
+
+class RedisMCPServer:
+    def __init__(self):
+        redis_client = RedisConnectionManager.get_connection()
+        print(redis_client.ping())
+
+    def run(self):
+        mcp.run(transport='stdio')
+
+if __name__ == "__main__":
+    server = RedisMCPServer()
+    server.run()

src/common/server.py

@@ -0,0 +1,92 @@
+from common.connection import RedisConnectionManager
+from redis.exceptions import RedisError
+from common.server import mcp
+
+@mcp.tool()
+async def hset(name: str, key: str, value: str) -> str:
+    """Set a field in a hash stored at key.
+    
+    Args:
+        name: The Redis hash key.
+        key: The field name inside the hash.
+        value: The value to set.
+    
+    Returns:
+        A success message or an error message.
+    """
+    try:
+        r = RedisConnectionManager.get_connection()
+        r.hset(name, key, value)
+        return f"Field '{key}' set successfully in hash '{name}'."
+    except RedisError as e:
+        return f"Error setting field '{key}' in hash '{name}': {str(e)}"
+
+@mcp.tool()
+async def hget(name: str, key: str) -> str:
+    """Get the value of a field in a Redis hash.
+    
+    Args:
+        name: The Redis hash key.
+        key: The field name inside the hash.
+    
+    Returns:
+        The field value or an error message.
+    """
+    try:
+        r = RedisConnectionManager.get_connection()
+        value = r.hget(name, key)
+        return value.decode() if value else f"Field '{key}' not found in hash '{name}'."
+    except RedisError as e:
+        return f"Error getting field '{key}' from hash '{name}': {str(e)}"
+
+@mcp.tool()
+async def hdel(name: str, key: str) -> str:
+    """Delete a field from a Redis hash.
+    
+    Args:
+        name: The Redis hash key.
+        key: The field name inside the hash.
+    
+    Returns:
+        A success message or an error message.
+    """
+    try:
+        r = RedisConnectionManager.get_connection()
+        deleted = r.hdel(name, key)
+        return f"Field '{key}' deleted from hash '{name}'." if deleted else f"Field '{key}' not found in hash '{name}'."
+    except RedisError as e:
+        return f"Error deleting field '{key}' from hash '{name}': {str(e)}"
+
+@mcp.tool()
+async def hgetall(name: str) -> dict:
+    """Get all fields and values from a Redis hash.
+    
+    Args:
+        name: The Redis hash key.
+    
+    Returns:
+        A dictionary of field-value pairs or an error message.
+    """
+    try:
+        r = RedisConnectionManager.get_connection()
+        hash_data = r.hgetall(name)
+        return {k: v for k, v in hash_data.items()} if hash_data else f"Hash '{name}' is empty or does not exist."
+    except RedisError as e:
+        return f"Error getting all fields from hash '{name}': {str(e)}"
+
+@mcp.tool()
+async def hexists(name: str, key: str) -> bool:
+    """Check if a field exists in a Redis hash.
+    
+    Args:
+        name: The Redis hash key.
+        key: The field name inside the hash.
+    
+    Returns:
+        True if the field exists, False otherwise.
+    """
+    try:
+        r = RedisConnectionManager.get_connection()
+        return r.hexists(name, key)
+    except RedisError as e:
+        return f"Error checking existence of field '{key}' in hash '{name}': {str(e)}"