from fastapi import APIRouter

from app.schemas.common import ApiResponse
from app.services.sensitive_words_service import get_sensitive_words_payload, save_sensitive_words_payload

router = APIRouter(tags=["sensitive_words"])


@router.get("/sensitive-words", response_model=ApiResponse)
def get_sensitive_words() -> ApiResponse:
    return ApiResponse(data=get_sensitive_words_payload())


@router.put("/sensitive-words", response_model=ApiResponse)
def update_sensitive_words(payload: dict) -> ApiResponse:
    data = save_sensitive_words_payload(payload)
    return ApiResponse(message=f"已保存 {data['total']} 个敏感词", data=data)