Redesign cli and client (#135)

* Redesign cli and client

Author: Agent-Hellboy

docs/getting-started/examples.md

@@ -19,6 +19,16 @@ mcp-fuzzer --mode tools --protocol http --endpoint http://localhost:8000 --runs
 mcp-fuzzer --mode tools --protocol http --endpoint http://localhost:8000 --runs 10 --timeout 60.0
 ```
 
+#### Single Tool Fuzzing
+
+```bash
+# Fuzz only a specific tool
+mcp-fuzzer --mode tool --tool analyze_repository --protocol http --endpoint http://localhost:8000 --runs 20
+
+# Fuzz a specific tool with both phases
+mcp-fuzzer --mode tool --tool generate_terraform --phase both --protocol http --endpoint http://localhost:8000 --runs 15
+```
+
 #### Protocol Fuzzing
 
 ```bash

docs/getting-started/getting-started.md

@@ -43,7 +43,7 @@ First, ensure you have an MCP server running. You can use any of these transport
 #### Tool Fuzzing (Default Mode)
 
 ```bash
-# Basic tool fuzzing
+# Basic tool fuzzing (all tools)
 mcp-fuzzer --mode tools --protocol http --endpoint http://localhost:8000 --runs 10
 
 # With verbose output
@@ -53,6 +53,16 @@ mcp-fuzzer --mode tools --protocol http --endpoint http://localhost:8000 --runs
 mcp-fuzzer --mode tools --protocol stdio --endpoint "python test_server.py" --runs 5 --enable-safety-system
 ```
 
+#### Single Tool Fuzzing
+
+```bash
+# Fuzz only a specific tool
+mcp-fuzzer --mode tool --tool analyze_repository --protocol http --endpoint http://localhost:8000 --runs 20
+
+# Fuzz a specific tool with both phases
+mcp-fuzzer --mode tool --tool generate_terraform --phase both --protocol http --endpoint http://localhost:8000 --runs 15
+```
+
 #### Protocol Fuzzing
 
 ```bash

examples/custom_websocket_transport.py

@@ -159,7 +159,9 @@ async def send_request(
                     response = json.loads(response_text)
                     if isinstance(response, dict) and response.get("id") == request_id:
                         logger.debug(
-                            f"Received WebSocket response for {method} (id={request_id})"
+                            "Received WebSocket response for %s (id=%s)",
+                            method,
+                            request_id,
                         )
                         if "error" in response:
                             error_msg = f"Server error: {response['error']}"
@@ -278,11 +280,19 @@ async def _stream_request(
                             timeout=self.timeout
                         )
                         response = json.loads(response_text)
-                        req_id = payload.get("id") if isinstance(payload, dict) else None
-                        if req_id is None or (isinstance(response, dict) and response.get("id") == req_id):
+                        req_id = (
+                            payload.get("id") if isinstance(payload, dict) else None
+                        )
+                        if req_id is None or (
+                            isinstance(response, dict)
+                            and response.get("id") == req_id
+                        ):
                             yield response
                         else:
-                            logger.debug("Ignoring out-of-band WebSocket message during stream: %s", response)
+                            logger.debug(
+                                "Ignoring extra WebSocket stream message: %s",
+                                response,
+                            )
                     except asyncio.TimeoutError:
                         logger.debug("WebSocket stream timeout")
                         break
@@ -369,5 +379,11 @@ def _factory(first_arg: str, **kwargs):
 
     print("WebSocket transport example loaded successfully!")
     print("Usage in MCP Fuzzer:")
-    print("  transport = create_transport('websocket://localhost:8080/mcp')  # custom scheme")
-    print("  transport = create_transport('ws://localhost:8080/mcp')        # direct ws URL")
+    print(
+        "  transport = create_transport('websocket://localhost:8080/mcp')"
+        "  # custom scheme"
+    )
+    print(
+        "  transport = create_transport('ws://localhost:8080/mcp')"
+        "        # direct ws URL"
+    )