"""Demo script for routing and port forward tools.
This script demonstrates the usage of the routing and port forward tools
added in Task 15. It shows how to:
- List traffic routes
- Get route details
- List port forwards
- Get port forward details
Run this script to test the new tools interactively.
"""
import asyncio
import json
import sys
from pathlib import Path
# Add parent directory to path for imports
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
from unifi_mcp.tools.security import (
ListTrafficRoutesTool,
GetRouteDetailsTool,
ListPortForwardsTool,
GetPortForwardDetailsTool,
)
from unifi_mcp.unifi_client import UniFiClient
from unifi_mcp.config.loader import load_config
async def demo_routing_tools():
"""Demonstrate routing and port forward tools."""
print("=" * 80)
print("UniFi MCP Server - Routing and Port Forward Tools Demo")
print("=" * 80)
print()
# Load configuration
print("Loading configuration...")
try:
config = load_config()
print("✓ Configuration loaded successfully")
except Exception as e:
print(f"✗ Failed to load configuration: {e}")
return
print()
# Create UniFi client
print("Connecting to UniFi controller...")
client = UniFiClient(config.unifi)
try:
await client.connect()
print(f"✓ Connected to {config.unifi.host}")
except Exception as e:
print(f"✗ Failed to connect: {e}")
return
print()
print("=" * 80)
# Demo 1: List Traffic Routes
print("\n1. LIST TRAFFIC ROUTES")
print("-" * 80)
list_routes_tool = ListTrafficRoutesTool()
try:
result = await list_routes_tool.invoke(client, {})
if result.get("success"):
routes = result["data"]
total = result.get("total", len(routes))
print(f"Found {total} traffic route(s)")
print()
if routes:
for i, route in enumerate(routes, 1):
print(f"Route {i}:")
print(f" ID: {route['id']}")
print(f" Name: {route['name']}")
print(f" Enabled: {route['enabled']}")
print(f" Type: {route['type']}")
print(f" Destination: {route['destination_network']}")
print(f" Next Hop: {route['next_hop']}")
print(f" Distance: {route['distance']}")
print(f" Interface: {route['interface']}")
print()
else:
print("No traffic routes configured")
else:
print(f"Error: {result.get('error', {}).get('message', 'Unknown error')}")
except Exception as e:
print(f"Error listing routes: {e}")
# Demo 2: Get Route Details (if routes exist)
print("\n2. GET ROUTE DETAILS")
print("-" * 80)
get_route_tool = GetRouteDetailsTool()
try:
# First get the list to find a route ID
result = await list_routes_tool.invoke(client, {})
if result.get("success") and result["data"]:
route_id = result["data"][0]["id"]
print(f"Getting details for route: {route_id}")
print()
details_result = await get_route_tool.invoke(client, {"route_id": route_id})
if details_result.get("success"):
route = details_result["data"]
print("Route Details:")
print(json.dumps(route, indent=2))
else:
print(f"Error: {details_result.get('error', {}).get('message', 'Unknown error')}")
else:
print("No routes available to get details for")
except Exception as e:
print(f"Error getting route details: {e}")
# Demo 3: List Port Forwards
print("\n3. LIST PORT FORWARDS")
print("-" * 80)
list_forwards_tool = ListPortForwardsTool()
try:
result = await list_forwards_tool.invoke(client, {})
if result.get("success"):
forwards = result["data"]
total = result.get("total", len(forwards))
print(f"Found {total} port forward(s)")
print()
if forwards:
for i, forward in enumerate(forwards, 1):
print(f"Port Forward {i}:")
print(f" ID: {forward['id']}")
print(f" Name: {forward['name']}")
print(f" Enabled: {forward['enabled']}")
print(f" Protocol: {forward['protocol']}")
print(f" External Port: {forward['external_port']}")
print(f" Destination: {forward['destination_ip']}:{forward['destination_port']}")
print(f" Source: {forward['source']}")
print(f" Logging: {forward['log']}")
print()
else:
print("No port forwards configured")
else:
print(f"Error: {result.get('error', {}).get('message', 'Unknown error')}")
except Exception as e:
print(f"Error listing port forwards: {e}")
# Demo 4: Get Port Forward Details (if forwards exist)
print("\n4. GET PORT FORWARD DETAILS")
print("-" * 80)
get_forward_tool = GetPortForwardDetailsTool()
try:
# First get the list to find a forward ID
result = await list_forwards_tool.invoke(client, {})
if result.get("success") and result["data"]:
forward_id = result["data"][0]["id"]
print(f"Getting details for port forward: {forward_id}")
print()
details_result = await get_forward_tool.invoke(client, {"forward_id": forward_id})
if details_result.get("success"):
forward = details_result["data"]
print("Port Forward Details:")
print(json.dumps(forward, indent=2))
else:
print(f"Error: {details_result.get('error', {}).get('message', 'Unknown error')}")
else:
print("No port forwards available to get details for")
except Exception as e:
print(f"Error getting port forward details: {e}")
# Demo 5: Filter enabled only
print("\n5. LIST ENABLED PORT FORWARDS ONLY")
print("-" * 80)
try:
result = await list_forwards_tool.invoke(client, {"enabled_only": True})
if result.get("success"):
forwards = result["data"]
total = result.get("total", len(forwards))
print(f"Found {total} enabled port forward(s)")
print()
if forwards:
for i, forward in enumerate(forwards, 1):
print(f"{i}. {forward['name']} - {forward['protocol']} "
f"{forward['external_port']} → "
f"{forward['destination_ip']}:{forward['destination_port']}")
else:
print("No enabled port forwards")
else:
print(f"Error: {result.get('error', {}).get('message', 'Unknown error')}")
except Exception as e:
print(f"Error listing enabled port forwards: {e}")
print()
print("=" * 80)
print("Demo completed!")
print("=" * 80)
# Close client
await client.close()
if __name__ == "__main__":
asyncio.run(demo_routing_tools())
