traia-iatp 0.1.29__py3-none-any.whl

traia_iatp/mcp/templates/server.py.j2

@@ -0,0 +1,175 @@
+#!/usr/bin/env python3
+"""
+{{ api_name }} MCP Server - FastMCP with D402 Transport Wrapper
+
+Uses FastMCP from official MCP SDK with D402MCPTransport wrapper for HTTP 402.
+
+Architecture:
+- FastMCP for tool decorators and Context objects
+- D402MCPTransport wraps the /mcp route for HTTP 402 interception
+- Proper HTTP 402 status codes (not JSON-RPC wrapped)
+
+Generated from OpenAPI: {{ docs_url }}
+
+Environment Variables:
+{% if requires_auth %}- {{ api_key_env_var }}: Server's internal API key (for paid requests)
+{% endif %}- SERVER_ADDRESS: Payment address (IATP wallet contract)
+- MCP_OPERATOR_PRIVATE_KEY: Operator signing key
+- D402_TESTING_MODE: Skip facilitator (default: true)
+"""
+
+import os
+import logging
+import sys
+from typing import Dict, Any, Optional
+from datetime import datetime
+
+import requests
+from retry import retry
+from dotenv import load_dotenv
+import uvicorn
+
+load_dotenv()
+
+# Configure logging
+logging.basicConfig(
+    level=os.getenv("LOG_LEVEL", "INFO").upper(),
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger('{{ api_slug }}_mcp')
+
+# FastMCP from official SDK
+from mcp.server.fastmcp import FastMCP, Context
+from starlette.requests import Request
+from starlette.responses import JSONResponse
+
+# D402 payment protocol - using Starlette middleware
+from traia_iatp.d402.starlette_middleware import D402PaymentMiddleware
+from traia_iatp.d402.mcp_middleware import require_payment_for_tool, get_active_api_key
+from traia_iatp.d402.payment_introspection import extract_payment_configs_from_mcp
+from traia_iatp.d402.types import TokenAmount, TokenAsset, EIP712Domain
+
+# Configuration
+STAGE = os.getenv("STAGE", "MAINNET").upper()
+PORT = int(os.getenv("PORT", "8000"))
+SERVER_ADDRESS = os.getenv("SERVER_ADDRESS")
+if not SERVER_ADDRESS:
+    raise ValueError("SERVER_ADDRESS required for payment protocol")
+
+{% if requires_auth %}
+API_KEY = os.getenv("{{ api_key_env_var }}")
+if not API_KEY:
+    logger.warning(f"⚠️  {{ api_key_env_var }} not set - payment required for all requests")
+{% else %}
+API_KEY = None
+{% endif %}
+
+logger.info("="*80)
+logger.info(f"{{ api_name }} MCP Server (FastMCP + D402 Wrapper)")
+logger.info(f"API: {{ api_url }}")
+logger.info(f"Payment: {SERVER_ADDRESS}")
+{% if requires_auth %}
+logger.info(f"API Key: {'✅' if API_KEY else '❌ Payment required'}")
+{% endif %}
+logger.info("="*80)
+
+# Create FastMCP server
+mcp = FastMCP("{{ api_name }} MCP Server")
+
+logger.info(f"✅ FastMCP server created")
+
+# ============================================================================
+# TOOL IMPLEMENTATIONS
+# ============================================================================
+# Tool implementations will be added here by endpoint_implementer_crew
+# Each tool will use the @mcp.tool() and @require_payment_for_tool() decorators
+# TODO: Add your API-specific functions here
+
+# ============================================================================
+# APPLICATION SETUP WITH STARLETTE MIDDLEWARE
+# ============================================================================
+
+def create_app_with_middleware():
+    """
+    Create Starlette app with d402 payment middleware.
+    
+    Strategy:
+    1. Get FastMCP's Starlette app via streamable_http_app()
+    2. Extract payment configs from @require_payment_for_tool decorators
+    3. Add Starlette middleware with extracted configs
+    4. Single source of truth - no duplication!
+    """
+    logger.info("🔧 Creating FastMCP app with middleware...")
+    
+    # Get FastMCP's Starlette app
+    app = mcp.streamable_http_app()
+    logger.info(f"✅ Got FastMCP Starlette app")
+    
+    # Extract payment configs from decorators (single source of truth!)
+    tool_payment_configs = extract_payment_configs_from_mcp(mcp, SERVER_ADDRESS)
+    logger.info(f"📊 Extracted {len(tool_payment_configs)} payment configs from @require_payment_for_tool decorators")
+    
+    # Add D402 payment middleware with extracted configs
+    app.add_middleware(
+        D402PaymentMiddleware,
+        tool_payment_configs=tool_payment_configs,
+        server_address=SERVER_ADDRESS,
+        {% if requires_auth %}
+        requires_auth=True,  # Extracts API keys + checks payment
+        internal_api_key=API_KEY,  # Server's internal key (for Mode 2: paid access)
+        {% else %}
+        requires_auth=False,  # Only checks payment
+        {% endif %}
+        testing_mode=os.getenv("D402_TESTING_MODE", "true").lower() == "true",
+        facilitator_url=os.getenv("D402_FACILITATOR_URL"),
+        facilitator_api_key=os.getenv("D402_FACILITATOR_API_KEY")
+    )
+    logger.info("✅ Added D402PaymentMiddleware")
+    {% if requires_auth %}
+    logger.info("   - Auth extraction: Enabled")
+    logger.info("   - Dual mode: API key OR payment")
+    {% else %}
+    logger.info("   - Payment-only mode")
+    {% endif %}
+    
+    # Add health check endpoint (bypasses middleware)
+    @app.route("/health", methods=["GET"])
+    async def health_check(request: Request) -> JSONResponse:
+        """Health check endpoint for container orchestration."""
+        return JSONResponse(
+            content={
+                "status": "healthy",
+                "service": "{{ api_slug }}-mcp-server",
+                "timestamp": datetime.now().isoformat()
+            }
+        )
+    logger.info("✅ Added /health endpoint")
+    
+    return app
+
+if __name__ == "__main__":
+    logger.info("="*80)
+    logger.info(f"Starting {{ api_name }} MCP Server")
+    logger.info("="*80)
+    logger.info("Architecture:")
+    logger.info("  1. D402PaymentMiddleware intercepts requests")
+    {% if requires_auth %}
+    logger.info("     - Extracts API keys from Authorization header")
+    logger.info("     - Checks payment → HTTP 402 if no API key AND no payment")
+    {% else %}
+    logger.info("     - Checks payment → HTTP 402 if missing")
+    {% endif %}
+    logger.info("  2. FastMCP processes valid requests with tool decorators")
+    logger.info("="*80)
+    
+    # Create app with middleware
+    app = create_app_with_middleware()
+    
+    # Run with uvicorn
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=PORT,
+        log_level=os.getenv("LOG_LEVEL", "info").lower()
+    )
+