""" Subscription routes for API Gateway - Direct subscription endpoints New URL Pattern Architecture: - Registration: /registration/payment-setup, /registration/complete, /registration/state/{state_id} - Tenant Subscription: /tenants/{tenant_id}/subscription/* - Setup Intents: /setup-intents/{setup_intent_id}/verify - Payment Customers: /payment-customers/create - Plans: /plans (public) """ from fastapi import APIRouter, Request, Response, HTTPException, Path from fastapi.responses import JSONResponse import httpx import logging from typing import Optional from app.core.config import settings from app.core.header_manager import header_manager logger = logging.getLogger(__name__) router = APIRouter() # ================================================================ # PUBLIC ENDPOINTS (No Authentication) # ================================================================ @router.api_route("/plans", methods=["GET", "OPTIONS"]) async def proxy_plans(request: Request): """Proxy plans request to tenant service""" target_path = "/plans" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/plans/{tier}", methods=["GET", "OPTIONS"]) async def proxy_plan_details(request: Request, tier: str = Path(...)): """Proxy specific plan details request to tenant service""" target_path = f"/plans/{tier}" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/plans/{tier}/features", methods=["GET", "OPTIONS"]) async def proxy_plan_features(request: Request, tier: str = Path(...)): """Proxy plan features request to tenant service""" target_path = f"/plans/{tier}/features" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/plans/{tier}/limits", methods=["GET", "OPTIONS"]) async def proxy_plan_limits(request: Request, tier: str = Path(...)): """Proxy plan limits request to tenant service""" target_path = f"/plans/{tier}/limits" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/plans/compare", methods=["GET", "OPTIONS"]) async def proxy_plan_compare(request: Request): """Proxy plan comparison request to tenant service""" target_path = "/plans/compare" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # REGISTRATION FLOW ENDPOINTS (No Tenant Context) # ================================================================ @router.api_route("/registration/payment-setup", methods=["POST", "OPTIONS"]) async def proxy_registration_payment_setup(request: Request): """Proxy registration payment setup request to tenant service""" target_path = "/api/v1/registration/payment-setup" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/registration/complete", methods=["POST", "OPTIONS"]) async def proxy_registration_complete(request: Request): """Proxy registration completion request to tenant service""" target_path = "/api/v1/registration/complete" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/registration/state/{state_id}", methods=["GET", "OPTIONS"]) async def proxy_registration_state(request: Request, state_id: str = Path(...)): """Proxy registration state request to tenant service""" target_path = f"/api/v1/registration/state/{state_id}" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # TENANT SUBSCRIPTION STATUS ENDPOINTS # ================================================================ @router.api_route("/tenants/{tenant_id}/subscription/status", methods=["GET", "OPTIONS"]) async def proxy_subscription_status(request: Request, tenant_id: str = Path(...)): """Proxy subscription status request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/status" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/details", methods=["GET", "OPTIONS"]) async def proxy_subscription_details(request: Request, tenant_id: str = Path(...)): """Proxy subscription details request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/details" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/tier", methods=["GET", "OPTIONS"]) async def proxy_subscription_tier(request: Request, tenant_id: str = Path(...)): """Proxy subscription tier request to tenant service (cached)""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/tier" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/limits", methods=["GET", "OPTIONS"]) async def proxy_subscription_limits(request: Request, tenant_id: str = Path(...)): """Proxy subscription limits request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/limits" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/usage", methods=["GET", "OPTIONS"]) async def proxy_subscription_usage(request: Request, tenant_id: str = Path(...)): """Proxy subscription usage request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/usage" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/features/{feature}", methods=["GET", "OPTIONS"]) async def proxy_subscription_feature(request: Request, tenant_id: str = Path(...), feature: str = Path(...)): """Proxy subscription feature check request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/features/{feature}" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # SUBSCRIPTION MANAGEMENT ENDPOINTS # ================================================================ @router.api_route("/tenants/{tenant_id}/subscription/cancel", methods=["POST", "OPTIONS"]) async def proxy_subscription_cancel(request: Request, tenant_id: str = Path(...)): """Proxy subscription cancellation request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/cancel" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/reactivate", methods=["POST", "OPTIONS"]) async def proxy_subscription_reactivate(request: Request, tenant_id: str = Path(...)): """Proxy subscription reactivation request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/reactivate" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/validate-upgrade/{new_plan}", methods=["GET", "OPTIONS"]) async def proxy_validate_upgrade(request: Request, tenant_id: str = Path(...), new_plan: str = Path(...)): """Proxy plan upgrade validation request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/validate-upgrade/{new_plan}" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/upgrade", methods=["POST", "OPTIONS"]) async def proxy_subscription_upgrade(request: Request, tenant_id: str = Path(...)): """Proxy subscription upgrade request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/upgrade" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # QUOTA & LIMIT CHECK ENDPOINTS # ================================================================ @router.api_route("/tenants/{tenant_id}/subscription/limits/locations", methods=["GET", "OPTIONS"]) async def proxy_location_limits(request: Request, tenant_id: str = Path(...)): """Proxy location limits check request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/limits/locations" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/limits/products", methods=["GET", "OPTIONS"]) async def proxy_product_limits(request: Request, tenant_id: str = Path(...)): """Proxy product limits check request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/limits/products" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/limits/users", methods=["GET", "OPTIONS"]) async def proxy_user_limits(request: Request, tenant_id: str = Path(...)): """Proxy user limits check request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/limits/users" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/limits/recipes", methods=["GET", "OPTIONS"]) async def proxy_recipe_limits(request: Request, tenant_id: str = Path(...)): """Proxy recipe limits check request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/limits/recipes" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/limits/suppliers", methods=["GET", "OPTIONS"]) async def proxy_supplier_limits(request: Request, tenant_id: str = Path(...)): """Proxy supplier limits check request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/limits/suppliers" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # PAYMENT MANAGEMENT ENDPOINTS # ================================================================ @router.api_route("/tenants/{tenant_id}/subscription/payment-method", methods=["GET", "POST", "OPTIONS"]) async def proxy_payment_method(request: Request, tenant_id: str = Path(...)): """Proxy payment method request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/payment-method" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/tenants/{tenant_id}/subscription/invoices", methods=["GET", "OPTIONS"]) async def proxy_invoices(request: Request, tenant_id: str = Path(...)): """Proxy invoices request to tenant service""" target_path = f"/api/v1/tenants/{tenant_id}/subscription/invoices" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # SETUP INTENT VERIFICATION # ================================================================ @router.api_route("/setup-intents/{setup_intent_id}/verify", methods=["GET", "OPTIONS"]) async def proxy_setup_intent_verify(request: Request, setup_intent_id: str = Path(...)): """Proxy SetupIntent verification request to tenant service""" target_path = f"/api/v1/setup-intents/{setup_intent_id}/verify" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # PAYMENT CUSTOMER MANAGEMENT # ================================================================ @router.api_route("/payment-customers/create", methods=["POST", "OPTIONS"]) async def proxy_payment_customer_create(request: Request): """Proxy payment customer creation request to tenant service""" target_path = "/api/v1/payment-customers/create" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # USAGE FORECAST ENDPOINTS # ================================================================ @router.api_route("/usage-forecast", methods=["GET", "OPTIONS"]) async def proxy_usage_forecast(request: Request): """Proxy usage forecast request to tenant service""" target_path = "/api/v1/usage-forecast" return await _proxy_to_tenant_service(request, target_path) @router.api_route("/usage-forecast/track-usage", methods=["POST", "OPTIONS"]) async def proxy_track_usage(request: Request): """Proxy track usage request to tenant service""" target_path = "/api/v1/usage-forecast/track-usage" return await _proxy_to_tenant_service(request, target_path) # ================================================================ # PROXY HELPER FUNCTIONS # ================================================================ async def _proxy_to_tenant_service(request: Request, target_path: str): """Proxy request to tenant service""" return await _proxy_request(request, target_path, settings.TENANT_SERVICE_URL) async def _proxy_request(request: Request, target_path: str, service_url: str): """Generic proxy function with enhanced error handling""" # Handle OPTIONS requests directly for CORS if request.method == "OPTIONS": return Response( status_code=200, headers={ "Access-Control-Allow-Origin": settings.CORS_ORIGINS_LIST, "Access-Control-Allow-Methods": "GET, POST, PUT, DELETE, OPTIONS", "Access-Control-Allow-Headers": "Content-Type, Authorization, X-Tenant-ID", "Access-Control-Allow-Credentials": "true", "Access-Control-Max-Age": "86400" } ) try: url = f"{service_url}{target_path}" # Use unified HeaderManager for consistent header forwarding headers = header_manager.get_all_headers_for_proxy(request) # Debug logging user_context = getattr(request.state, 'user', None) service_context = getattr(request.state, 'service', None) if user_context: logger.info(f"Forwarding subscription request to {url} with user context: user_id={user_context.get('user_id')}, email={user_context.get('email')}, subscription_tier={user_context.get('subscription_tier', 'not_set')}") elif service_context: logger.debug(f"Forwarding subscription request to {url} with service context: service_name={service_context.get('service_name')}, user_type=service") else: logger.warning(f"No user or service context available when forwarding subscription request to {url}") # Get request body if present body = None if request.method in ["POST", "PUT", "PATCH"]: body = await request.body() # Add query parameters params = dict(request.query_params) timeout_config = httpx.Timeout( connect=30.0, read=60.0, write=30.0, pool=30.0 ) async with httpx.AsyncClient(timeout=timeout_config) as client: response = await client.request( method=request.method, url=url, headers=headers, content=body, params=params ) # Handle different response types if response.headers.get("content-type", "").startswith("application/json"): try: content = response.json() except: content = {"message": "Invalid JSON response from service"} else: content = response.text return JSONResponse( status_code=response.status_code, content=content ) except Exception as e: logger.error(f"Unexpected error proxying subscription request to {service_url}{target_path}: {e}") raise HTTPException( status_code=500, detail="Internal gateway error" )