Finished refactor to FastMCP and use more best practices

Author: kenliao94

.pre-commit-config.yaml

@@ -0,0 +1,16 @@
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.7.3
+    hooks:
+      - id: ruff
+        args: [--fix]
+      - id: ruff-format
+
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.5.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+      - id: check-added-large-files
+      - id: check-toml

CHANGELOG.md

@@ -0,0 +1,15 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [2.0.0] - 2025-04-30
+
+### Added
+- Initial release of RabbitMQ MCP Server
+- Support for basic queue and exchange operations
+- FastMCP integration for improved performance and maintainability
+- Ruff linting and formatting configuration
+- Testing infrastructure with pytest

README.md

@@ -3,6 +3,15 @@
 
 A [Model Context Protocol](https://www.anthropic.com/news/model-context-protocol) server implementation for RabbitMQ. Enabling MCP client to interact with queues and topics hosted in a RabbitMQ instance.
 
+## Features
+
+- Connect to RabbitMQ instances
+- Publish messages to queues and exchanges
+- List queues and exchanges
+- Get detailed information about queues and exchanges
+- Delete queues and exchanges
+- Purge messages from queues
+
 ## Running locally with the Claude desktop app
 
 ### Installing via Smithery
@@ -22,7 +31,7 @@ https://smithery.ai/server/@kenliao94/mcp-server-rabbitmq
 - On MacOS: `~/Library/Application\ Support/Claude/claude_desktop_config.json`
 - On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
 
-```
+```json
 {
     "mcpServers": {
       "rabbitmq": {
@@ -49,3 +58,38 @@ https://smithery.ai/server/@kenliao94/mcp-server-rabbitmq
 ```
 4. Install and open the [Claude desktop app](https://claude.ai/download).
 5. Try asking Claude to do a read/write operation of some sort to confirm the setup (e.g. ask it to publish a message to a queue). If there are issues, use the Debugging tools provided in the MCP documentation [here](https://modelcontextprotocol.io/docs/tools/debugging).
+
+## Development
+
+### Setup Development Environment
+
+```bash
+# Clone the repository
+git clone https://github.com/kenliao94/mcp-server-rabbitmq.git
+cd mcp-server-rabbitmq
+
+# Install pre-commit hooks
+pre-commit install
+```
+
+### Running Tests
+
+```bash
+pytest
+```
+
+### Code Quality
+
+This project uses ruff for linting and formatting:
+
+```bash
+# Run linter
+ruff check .
+
+# Run formatter
+ruff format .
+```
+
+## License
+
+This project is licensed under the Apache License 2.0 - see the LICENSE file for details.

mcp_server_rabbitmq/__init__.py

@@ -0,0 +1 @@
+"""RabbitMQ MCP Server package."""

src/mcp_server_rabbitmq/admin.py renamed to mcp_server_rabbitmq/admin.py

@@ -1,6 +1,8 @@
-import requests
-from typing import Optional, List, Dict
 import base64
+from typing import Dict, List, Optional
+
+import requests
+
 
 class RabbitMQAdmin:
     def __init__(self, host: str, port: int, username: str, password: str, use_tls: bool):
@@ -30,37 +32,37 @@ def list_exchanges(self) -> List[Dict]:
 
     def get_queue_info(self, queue: str, vhost: str = "/") -> Dict:
         """Get detailed information about a specific queue"""
-        vhost_encoded = requests.utils.quote(vhost, safe='')
+        vhost_encoded = requests.utils.quote(vhost, safe="")
         response = self._make_request("GET", f"queues/{vhost_encoded}/{queue}")
         return response.json()
 
     def delete_queue(self, queue: str, vhost: str = "/") -> None:
         """Delete a queue"""
         validate_rabbitmq_name(queue, "Queue name")
-        vhost_encoded = requests.utils.quote(vhost, safe='')
+        vhost_encoded = requests.utils.quote(vhost, safe="")
         self._make_request("DELETE", f"queues/{vhost_encoded}/{queue}")
 
     def purge_queue(self, queue: str, vhost: str = "/") -> None:
         """Remove all messages from a queue"""
         validate_rabbitmq_name(queue, "Queue name")
-        vhost_encoded = requests.utils.quote(vhost, safe='')
+        vhost_encoded = requests.utils.quote(vhost, safe="")
         self._make_request("DELETE", f"queues/{vhost_encoded}/{queue}/contents")
 
     def get_exchange_info(self, exchange: str, vhost: str = "/") -> Dict:
         """Get detailed information about a specific exchange"""
-        vhost_encoded = requests.utils.quote(vhost, safe='')
+        vhost_encoded = requests.utils.quote(vhost, safe="")
         response = self._make_request("GET", f"exchanges/{vhost_encoded}/{exchange}")
         return response.json()
 
     def delete_exchange(self, exchange: str, vhost: str = "/") -> None:
         """Delete an exchange"""
         validate_rabbitmq_name(exchange, "Exchange name")
-        vhost_encoded = requests.utils.quote(vhost, safe='')
+        vhost_encoded = requests.utils.quote(vhost, safe="")
         self._make_request("DELETE", f"exchanges/{vhost_encoded}/{exchange}")
 
     def get_bindings(self, queue: Optional[str] = None, exchange: Optional[str] = None, vhost: str = "/") -> List[Dict]:
         """Get bindings, optionally filtered by queue or exchange"""
-        vhost_encoded = requests.utils.quote(vhost, safe='')
+        vhost_encoded = requests.utils.quote(vhost, safe="")
         if queue:
             validate_rabbitmq_name(queue, "Queue name")
             response = self._make_request("GET", f"queues/{vhost_encoded}/{queue}/bindings")

src/mcp_server_rabbitmq/connection.py renamed to mcp_server_rabbitmq/connection.py

@@ -1,6 +1,7 @@
-import pika
 import ssl
-from typing import Optional
+
+import pika
+
 
 class RabbitMQConnection:
     def __init__(self, host: str, port: int, username: str, password: str, use_tls: bool):
@@ -10,7 +11,7 @@ def __init__(self, host: str, port: int, username: str, password: str, use_tls:
 
         if use_tls:
             ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLSv1_2)
-            ssl_context.set_ciphers('ECDHE+AESGCM:!ECDSA')
+            ssl_context.set_ciphers("ECDHE+AESGCM:!ECDSA")
             self.parameters.ssl_options = pika.SSLOptions(context=ssl_context)
 
     def get_channel(self) -> tuple[pika.BlockingConnection, pika.channel.Channel]:
@@ -22,7 +23,7 @@ def validate_rabbitmq_name(name: str, field_name: str) -> None:
     """Validate RabbitMQ queue/exchange names"""
     if not name or not name.strip():
         raise ValueError(f"{field_name} cannot be empty")
-    if not all(c.isalnum() or c in '-_.:' for c in name):
+    if not all(c.isalnum() or c in "-_.:" for c in name):
         raise ValueError(f"{field_name} can only contain letters, digits, hyphen, underscore, period, or colon")
     if len(name) > 255:
         raise ValueError(f"{field_name} must be less than 255 characters")

mcp_server_rabbitmq/constant.py

@@ -0,0 +1,3 @@
+"""Constants for the RabbitMQ MCP server."""
+
+MCP_SERVER_VERSION = "2.0.0"

src/mcp_server_rabbitmq/handlers.py renamed to mcp_server_rabbitmq/handlers.py

@@ -1,7 +1,9 @@
-from .connection import RabbitMQConnection
-from .admin import RabbitMQAdmin
 from typing import List
 
+from .admin import RabbitMQAdmin
+from .connection import RabbitMQConnection
+
+
 def handle_enqueue(rabbitmq: RabbitMQConnection, queue: str, message: str):
     connection, channel = rabbitmq.get_channel()
     channel.queue_declare(queue)
@@ -16,11 +18,11 @@ def handle_fanout(rabbitmq: RabbitMQConnection, exchange: str, message: str):
 
 def handle_list_queues(rabbitmq_admin: RabbitMQAdmin) -> List[str]:
     result = rabbitmq_admin.list_queues()
-    return [queue['name'] for queue in result]
+    return [queue["name"] for queue in result]
 
 def handle_list_exchanges(rabbitmq_admin: RabbitMQAdmin) -> List[str]:
     result = rabbitmq_admin.list_exchanges()
-    return [exchange['name'] for exchange in result]
+    return [exchange["name"] for exchange in result]
 
 def handle_get_queue_info(rabbitmq_admin: RabbitMQAdmin, queue: str, vhost: str = "/") -> dict:
     return rabbitmq_admin.get_queue_info(queue, vhost)