from fastapi import APIRouter, HTTPException, Depends import traceback from fastapi.responses import JSONResponse from starlette.status import HTTP_500_INTERNAL_SERVER_ERROR from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials from backend.application.email_sender_hub import EmailSenderHub from common.token.token_manager import TokenManager from pydantic import BaseModel from typing import List router = APIRouter() security = HTTPBearer() token_manager = TokenManager() email_sender_hub = EmailSenderHub() # Define the request body schema class EmailSenderSetRequest(BaseModel): email_senders: List[str] class EmailSenderAddRequest(BaseModel): new_senders: List[str] class EmailSenderRemoveRequest(BaseModel): emails_to_remove: List[str] # check credentials for admin and tenant def admin_only(credentials: HTTPAuthorizationCredentials = Depends(security)): try: payload = token_manager.decode_token(credentials.credentials) # handle token generated by structure Authentication service if "subject" in payload: # Authentication format: payload.subject.role role = payload.get("subject", {}).get("role") else: # local generated format: payload.role role = payload.get("role") # according to AdministrativeRole enum defined in authentication service, ADMINISTRATOR = 8 if role not in ["admin", 8]: raise HTTPException(status_code=403, detail="Admin access required") return payload except Exception: raise HTTPException(status_code=401, detail="Invalid token") def tenant_only(credentials: HTTPAuthorizationCredentials = Depends(security)): try: payload = token_manager.decode_token(credentials.credentials) # handle token generated by structure Authentication service if "subject" in payload: tenant_id = payload.get("subject", {}).get("id") user_id = payload.get("subject", {}).get("id") role = payload.get("subject", {}).get("role") else: tenant_id = payload.get("tenant_id") user_id = payload.get("id") role = payload.get("role") if not tenant_id: raise HTTPException(status_code=403, detail="Tenant access required") # according to AdministrativeRole enum defined in authentication service, BUSINESS = 2 if role not in [2, "tenant"]: raise HTTPException(status_code=403, detail="Tenant access required") payload["tenant_id"] = tenant_id payload["user_id"] = user_id payload["is_admin"] = False return payload except Exception: raise HTTPException(status_code=401, detail="Invalid token") # Web API # Get email senders for tenant @router.get( "/email_senders/get", dependencies=[Depends(tenant_only)], operation_id="get_email_senders", summary="Get email senders for tenant", description="Retrieve the list of email senders configured for the current tenant", response_description="List of email sender addresses" ) async def get_email_senders(payload: dict = Depends(tenant_only)): try: tenant_id = payload.get("tenant_id") result = await email_sender_hub.get_email_senders(tenant_id) return JSONResponse( content={"success": True, "email_senders": result}, status_code=200 ) except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) except Exception as e: raise HTTPException(status_code=500, detail="Failed to get email senders") # Set email senders for tenant @router.post( "/email_senders/set", dependencies=[Depends(tenant_only)], operation_id="set_email_senders", summary="Set email senders for tenant", description="Set the complete list of email senders for the specified tenant", response_description="Success/failure response in setting email senders" ) async def set_email_senders(request: EmailSenderSetRequest, payload: dict = Depends(tenant_only)): try: tenant_id = payload.get("tenant_id") result = await email_sender_hub.set_email_senders(tenant_id, request.email_senders) return JSONResponse( content=result, status_code=200 ) except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) except Exception as e: import traceback traceback.print_exc() raise HTTPException(status_code=500, detail=f"Failed to set email senders: {str(e)}") # Add email senders to tenant @router.post( "/email_senders/add", dependencies=[Depends(tenant_only)], operation_id="add_email_senders", summary="Add email senders to tenant", description="Add new email senders to the existing list for the specified tenant", response_description="Success/failure response in adding email senders" ) async def add_email_senders(request: EmailSenderAddRequest, payload: dict = Depends(tenant_only)): try: tenant_id = payload.get("tenant_id") result = await email_sender_hub.add_email_senders(tenant_id, request.new_senders) return JSONResponse( content=result, status_code=200 ) except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) except Exception as e: import traceback traceback.print_exc() raise HTTPException(status_code=500, detail=f"Failed to add email senders: {str(e)}") # Remove email senders from tenant @router.delete( "/email_senders/remove", dependencies=[Depends(tenant_only)], operation_id="remove_email_senders", summary="Remove email senders from tenant", description="Remove specific email senders from the tenant's list", response_description="Success/failure response in removing email senders" ) async def remove_email_senders(request: EmailSenderRemoveRequest, payload: dict = Depends(tenant_only)): try: tenant_id = payload.get("tenant_id") result = await email_sender_hub.remove_email_senders(tenant_id, request.emails_to_remove) return JSONResponse( content=result, status_code=200 ) except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) except Exception as e: import traceback traceback.print_exc() raise HTTPException(status_code=500, detail=f"Failed to remove email senders: {str(e)}") # Clear all email senders for tenant @router.delete( "/email_senders/clear", dependencies=[Depends(tenant_only)], operation_id="clear_email_senders", summary="Clear all email senders for tenant", description="Remove all email senders from the current tenant's list", response_description="Success/failure response in clearing email senders" ) async def clear_email_senders(payload: dict = Depends(tenant_only)): try: tenant_id = payload.get("tenant_id") result = await email_sender_hub.clear_email_senders(tenant_id) return JSONResponse( content=result, status_code=200 ) except Exception as e: raise HTTPException(status_code=500, detail="Failed to clear email senders") # Delete email sender configuration for tenant @router.delete( "/email_senders/delete/{tenant_id}", dependencies=[Depends(admin_only)], operation_id="delete_email_sender", summary="Delete email sender configuration for tenant", description="Completely delete the email sender configuration for the specified tenant", response_description="Success/failure response in deleting email sender configuration" ) async def delete_email_sender(tenant_id: str): try: result = await email_sender_hub.delete_email_sender(tenant_id) return JSONResponse( content=result, status_code=200 ) except Exception as e: raise HTTPException(status_code=500, detail="Failed to delete email sender")