brand-master/backend/app/routers/contact.py

from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session
from typing import List, Optional
from app.database.database import get_db
from app.models import ContactMessage, User
from app.schemas.contact import ContactMessageCreate, ContactMessageResponse, ContactMessageUpdate
from app.services.auth import get_current_admin_user, get_current_user
from app.services.email import send_contact_notification_to_admin, send_admin_response_to_customer
from app.config import settings

router = APIRouter(prefix="/api/contact", tags=["contact"])


@router.post("", response_model=ContactMessageResponse)
def send_contact_message(message: ContactMessageCreate, db: Session = Depends(get_db)):
    """Public endpoint - anyone can send a contact message"""
    message_data = message.model_dump() if hasattr(message, 'model_dump') else message.dict()
    db_message = ContactMessage(**message_data)
    db.add(db_message)
    db.commit()
    db.refresh(db_message)
    
    # Send email notification to admin
    try:
        send_contact_notification_to_admin(
            admin_email=settings.admin_email,
            customer_name=db_message.full_name,
            customer_email=db_message.email,
            subject=db_message.subject,
            message=db_message.message,
            phone=db_message.phone
        )
    except Exception as e:
        print(f"Failed to send admin notification email: {e}")
    
    return db_message


# Admin endpoints
admin_router = APIRouter(prefix="/api/admin/contact-messages", tags=["admin-contact"])


@admin_router.get("", response_model=List[ContactMessageResponse])
def get_all_messages(
    status: Optional[str] = None,
    is_read: Optional[bool] = None,
    skip: int = 0,
    limit: int = 100,
    db: Session = Depends(get_db),
    admin: User = Depends(get_current_admin_user)
):
    """Get all contact messages with optional filters"""
    query = db.query(ContactMessage)
    
    if status:
        query = query.filter(ContactMessage.status == status)
    if is_read is not None:
        query = query.filter(ContactMessage.is_read == is_read)
    
    messages = query.order_by(ContactMessage.created_at.desc()).offset(skip).limit(limit).all()
    return messages


@admin_router.get("/unread-count")
def get_unread_count(
    db: Session = Depends(get_db),
    admin: User = Depends(get_current_admin_user)
):
    """Get count of unread messages"""
    count = db.query(ContactMessage).filter(ContactMessage.is_read == False).count()
    return {"unread_count": count}


@admin_router.get("/{message_id}", response_model=ContactMessageResponse)
def get_message(
    message_id: int,
    db: Session = Depends(get_db),
    admin: User = Depends(get_current_admin_user)
):
    """Get a single contact message by ID"""
    message = db.query(ContactMessage).filter(ContactMessage.id == message_id).first()
    if not message:
        raise HTTPException(status_code=404, detail="Message not found")
    return message


@admin_router.put("/{message_id}", response_model=ContactMessageResponse)
def update_message(
    message_id: int,
    message_update: ContactMessageUpdate,
    db: Session = Depends(get_db),
    admin: User = Depends(get_current_admin_user)
):
    """Update contact message status/notes"""
    message = db.query(ContactMessage).filter(ContactMessage.id == message_id).first()
    if not message:
        raise HTTPException(status_code=404, detail="Message not found")
    
    update_data = message_update.model_dump(exclude_unset=True) if hasattr(message_update, 'model_dump') else message_update.dict(exclude_unset=True)
    
    # Check if admin_notes is being added/updated
    send_email_to_customer = False
    if 'admin_notes' in update_data and update_data['admin_notes']:
        # Only send if there's actually content and it's different from before
        if not message.admin_notes or update_data['admin_notes'] != message.admin_notes:
            send_email_to_customer = True
    
    for field, value in update_data.items():
        setattr(message, field, value)
    
    db.commit()
    db.refresh(message)
    
    # Send email to customer if admin responded
    if send_email_to_customer:
        try:
            send_admin_response_to_customer(
                customer_email=message.email,
                customer_name=message.full_name,
                original_subject=message.subject,
                admin_notes=message.admin_notes
            )
        except Exception as e:
            print(f"Failed to send customer response email: {e}")
    
    return message


@admin_router.delete("/{message_id}")
def delete_message(
    message_id: int,
    db: Session = Depends(get_db),
    admin: User = Depends(get_current_admin_user)
):
    """Delete a contact message"""
    message = db.query(ContactMessage).filter(ContactMessage.id == message_id).first()
    if not message:
        raise HTTPException(status_code=404, detail="Message not found")
    
    db.delete(message)
    db.commit()
    return {"message": "Contact message deleted successfully"}


# User/Customer endpoints
user_router = APIRouter(prefix="/api/my-messages", tags=["user-messages"])


@user_router.get("", response_model=List[ContactMessageResponse])
def get_my_messages(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """Get all contact messages sent by the current user"""
    messages = db.query(ContactMessage).filter(
        ContactMessage.email == current_user.email
    ).order_by(ContactMessage.created_at.desc()).all()
    return messages


@user_router.get("/{message_id}", response_model=ContactMessageResponse)
def get_my_message(
    message_id: int,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """Get a specific contact message by ID (only if sent by current user)"""
    message = db.query(ContactMessage).filter(
        ContactMessage.id == message_id,
        ContactMessage.email == current_user.email
    ).first()
    
    if not message:
        raise HTTPException(status_code=404, detail="Message not found")
    
    return message