# retoor <retoor@molodetz.nl>

import logging
from typing import Annotated
from fastapi import Depends, APIRouter, Request
from devplacepy.models import AiCorrectionForm
from devplacepy.config import DEFAULT_CORRECTION_PROMPT
from devplacepy.database import get_table
from devplacepy.responses import action_result
from devplacepy.utils import clear_user_cache
from devplacepy.services.audit import record as audit

from devplacepy.routers.profile._shared import resolve_customization_target
from devplacepy.dependencies import json_or_form

logger = logging.getLogger(__name__)
router = APIRouter()

@router.post("/{username}/ai-correction")
async def set_ai_correction(
    request: Request,
    username: str,
    data: Annotated[AiCorrectionForm, Depends(json_or_form(AiCorrectionForm))],
):
    target, denied = resolve_customization_target(request, username)
    if denied is not None:
        return denied
    enabled = 1 if data.enabled else 0
    sync = 1 if data.sync else 0
    prompt = data.prompt.strip() or DEFAULT_CORRECTION_PROMPT
    get_table("users").update(
        {
            "uid": target["uid"],
            "ai_correction_enabled": enabled,
            "ai_correction_prompt": prompt,
            "ai_correction_sync": sync,
        },
        ["uid"],
    )
    clear_user_cache(target["uid"])
    logger.info(
        f"AI correction {'enabled' if enabled else 'disabled'} for {target['username']}"
    )
    audit.record(
        request,
        "profile.ai_correction",
        target_type="user",
        target_uid=target["uid"],
        target_label=target["username"],
        new_value=enabled,
        summary=(
            f"{'enabled' if enabled else 'disabled'} AI content correction "
            f"for {target['username']}"
        ),
        links=[audit.target("user", target["uid"], target["username"])],
    )
    url = f"/profile/{target['username']}"
    return action_result(
        request,
        url,
        data={
            "url": url,
            "enabled": bool(enabled),
            "sync": bool(sync),
            "prompt": prompt,
        },
    )