"""Main FastMCP server for Cox's Bazar AI Itinerary."""
import sys
import os
from pathlib import Path
# Add src directory to path if running directly
if __name__ == "__main__" or "mcp_server" not in sys.modules:
src_path = Path(__file__).parent.parent # This points to src/
if str(src_path) not in sys.path:
sys.path.insert(0, str(src_path))
from dotenv import load_dotenv
from fastmcp.server.providers import FileSystemProvider
from mcp_server.mcp_instance import mcp
from fastmcp.server.middleware.rate_limiting import RateLimitingMiddleware
#from fastmcp.server.middleware import AuthMiddleware
#from fastmcp.server.auth import require_auth
load_dotenv()
def main():
"""Run the MCP server."""
transport_name = os.environ.get("TRANSPORT_NAME") or "stdio"
server_port = int(os.environ.get("SERVER_PORT") or 8000)
server_host = os.environ.get("SERVER_HOST") or "0.0.0.0"
#Auto-register all MCP components (tools, prompts, resources, custom routes)
components_dir = Path(__file__).parent / "components"
custom_routes_dir = Path(__file__).parent / "config" / "custom_routes.py"
mcp.providers.append(FileSystemProvider(components_dir))
mcp.providers.append(FileSystemProvider(custom_routes_dir))
mcp.add_middleware(RateLimitingMiddleware(
max_requests_per_second=10.0,
burst_capacity=20,
get_client_id=lambda ctx: ctx.fastmcp_context.session_id if ctx.fastmcp_context else "unknown"
))
# global auth middleware
#mcp.middleware.append(AuthMiddleware(auth=require_auth))
if transport_name == "http" or transport_name == "streamable-http" or transport_name == "sse":
mcp.run(transport=transport_name, port=server_port, host=server_host)
else:
mcp.run(transport=transport_name)
if __name__ == "__main__":
main()
