orion/app/api/v1/vendor/email_templates.py

# app/api/v1/vendor/email_templates.py
"""
Vendor email template override endpoints.

Allows vendors to customize platform email templates with their own content.
Platform-only templates (billing, subscription) cannot be overridden.

Vendor Context: Uses token_vendor_id from JWT token (authenticated vendor API pattern).
"""

import logging
from typing import Any

from fastapi import APIRouter, Depends
from pydantic import BaseModel, EmailStr, Field
from sqlalchemy.orm import Session

from app.api.deps import get_current_vendor_api
from app.core.database import get_db
from app.services.email_service import EmailService
from app.services.email_template_service import EmailTemplateService
from app.services.vendor_service import vendor_service
from models.database.user import User

router = APIRouter(prefix="/email-templates")
logger = logging.getLogger(__name__)


# =============================================================================
# SCHEMAS
# =============================================================================


class VendorTemplateUpdate(BaseModel):
    """Schema for creating/updating a vendor template override."""

    subject: str = Field(..., min_length=1, max_length=500)
    body_html: str = Field(..., min_length=1)
    body_text: str | None = None
    name: str | None = Field(None, max_length=255)


class TemplatePreviewRequest(BaseModel):
    """Schema for previewing a template."""

    language: str = "en"
    variables: dict[str, Any] = {}


class TemplateTestRequest(BaseModel):
    """Schema for sending a test email."""

    to_email: EmailStr
    language: str = "en"
    variables: dict[str, Any] = {}


# =============================================================================
# ENDPOINTS
# =============================================================================


@router.get("")
def list_overridable_templates(
    current_user: User = Depends(get_current_vendor_api),
    db: Session = Depends(get_db),
):
    """
    List all email templates that the vendor can customize.

    Returns platform templates with vendor override status.
    Platform-only templates (billing, subscription) are excluded.
    """
    vendor_id = current_user.token_vendor_id
    service = EmailTemplateService(db)
    return service.list_overridable_templates(vendor_id)


@router.get("/{code}")
def get_template(
    code: str,
    current_user: User = Depends(get_current_vendor_api),
    db: Session = Depends(get_db),
):
    """
    Get a specific template with all language versions.

    Returns platform template details and vendor overrides for each language.
    """
    vendor_id = current_user.token_vendor_id
    service = EmailTemplateService(db)
    return service.get_vendor_template(vendor_id, code)


@router.get("/{code}/{language}")
def get_template_language(
    code: str,
    language: str,
    current_user: User = Depends(get_current_vendor_api),
    db: Session = Depends(get_db),
):
    """
    Get a specific template for a specific language.

    Returns vendor override if exists, otherwise platform template.
    """
    vendor_id = current_user.token_vendor_id
    service = EmailTemplateService(db)
    return service.get_vendor_template_language(vendor_id, code, language)


@router.put("/{code}/{language}")
def update_template_override(
    code: str,
    language: str,
    template_data: VendorTemplateUpdate,
    current_user: User = Depends(get_current_vendor_api),
    db: Session = Depends(get_db),
):
    """
    Create or update a vendor template override.

    Creates a vendor-specific version of the email template.
    The platform template remains unchanged.
    """
    vendor_id = current_user.token_vendor_id
    service = EmailTemplateService(db)

    result = service.create_or_update_vendor_override(
        vendor_id=vendor_id,
        code=code,
        language=language,
        subject=template_data.subject,
        body_html=template_data.body_html,
        body_text=template_data.body_text,
        name=template_data.name,
    )
    db.commit()

    return result


@router.delete("/{code}/{language}")
def delete_template_override(
    code: str,
    language: str,
    current_user: User = Depends(get_current_vendor_api),
    db: Session = Depends(get_db),
):
    """
    Delete a vendor template override.

    Reverts to using the platform default template for this language.
    """
    vendor_id = current_user.token_vendor_id
    service = EmailTemplateService(db)
    service.delete_vendor_override(vendor_id, code, language)
    db.commit()

    return {
        "message": "Template override deleted - reverted to platform default",
        "code": code,
        "language": language,
    }


@router.post("/{code}/preview")
def preview_template(
    code: str,
    preview_data: TemplatePreviewRequest,
    current_user: User = Depends(get_current_vendor_api),
    db: Session = Depends(get_db),
):
    """
    Preview a template with sample variables.

    Uses vendor override if exists, otherwise platform template.
    """
    vendor_id = current_user.token_vendor_id
    vendor = vendor_service.get_vendor_by_id(db, vendor_id)
    service = EmailTemplateService(db)

    # Add branding variables
    variables = {
        **_get_sample_variables(code),
        **preview_data.variables,
        "platform_name": "Wizamart",
        "vendor_name": vendor.name if vendor else "Your Store",
        "support_email": vendor.contact_email if vendor else "support@wizamart.com",
    }

    return service.preview_vendor_template(
        vendor_id=vendor_id,
        code=code,
        language=preview_data.language,
        variables=variables,
    )


@router.post("/{code}/test")
def send_test_email(
    code: str,
    test_data: TemplateTestRequest,
    current_user: User = Depends(get_current_vendor_api),
    db: Session = Depends(get_db),
):
    """
    Send a test email using the template.

    Uses vendor override if exists, otherwise platform template.
    """
    vendor_id = current_user.token_vendor_id
    vendor = vendor_service.get_vendor_by_id(db, vendor_id)

    # Build test variables
    variables = {
        **_get_sample_variables(code),
        **test_data.variables,
        "platform_name": "Wizamart",
        "vendor_name": vendor.name if vendor else "Your Store",
        "support_email": vendor.contact_email if vendor else "support@wizamart.com",
    }

    try:
        email_svc = EmailService(db)
        email_log = email_svc.send_template(
            template_code=code,
            to_email=test_data.to_email,
            variables=variables,
            vendor_id=vendor_id,
            language=test_data.language,
        )

        if email_log.status == "sent":
            return {
                "success": True,
                "message": f"Test email sent to {test_data.to_email}",
            }
        else:
            return {
                "success": False,
                "message": email_log.error_message or "Failed to send email",
            }
    except Exception as e:
        logger.exception(f"Failed to send test email: {e}")
        return {
            "success": False,
            "message": str(e),
        }


# =============================================================================
# HELPERS
# =============================================================================


def _get_sample_variables(template_code: str) -> dict[str, Any]:
    """Get sample variables for testing templates."""
    samples = {
        "signup_welcome": {
            "first_name": "John",
            "company_name": "Acme Corp",
            "email": "john@example.com",
            "vendor_code": "acme",
            "login_url": "https://example.com/login",
            "trial_days": "14",
            "tier_name": "Business",
        },
        "order_confirmation": {
            "customer_name": "Jane Doe",
            "order_number": "ORD-12345",
            "order_total": "€99.99",
            "order_items_count": "3",
            "order_date": "2024-01-15",
            "shipping_address": "123 Main St, Luxembourg City, L-1234",
        },
        "password_reset": {
            "customer_name": "John Doe",
            "reset_link": "https://example.com/reset?token=abc123",
            "expiry_hours": "1",
        },
        "team_invite": {
            "invitee_name": "Jane",
            "inviter_name": "John",
            "vendor_name": "Acme Corp",
            "role": "Admin",
            "accept_url": "https://example.com/accept",
            "expires_in_days": "7",
        },
    }
    return samples.get(template_code, {})