import pytest
from sqlalchemy import select

from backend.app.celery_app import celery
from backend.app.models import ImageRecord, MLSettings, TagHead, TagKind
from backend.app.services.tag_service import TagService

pytestmark = pytest.mark.integration


@pytest.fixture(autouse=True)
def eager():
    celery.conf.task_always_eager = True
    yield
    celery.conf.task_always_eager = False


async def _img(db, sha="s" * 64):
    img = ImageRecord(
        path=f"/images/{sha}.jpg", sha256=sha, size_bytes=1,
        mime="image/jpeg", width=1, height=1,
        origin="imported_filesystem", integrity_status="unknown",
    )
    db.add(img)
    await db.commit()
    return img


@pytest.mark.asyncio
async def test_get_suggestions(client, db):
    # Suggestions come from a trained head now (Camie/centroid removed): an image
    # whose embedding aligns with the head surfaces that concept.
    s = (await db.execute(select(MLSettings).where(MLSettings.id == 1))).scalar_one()
    img = ImageRecord(
        path="/images/headsug.jpg", sha256="h" * 64, size_bytes=1,
        mime="image/jpeg", width=1, height=1, origin="imported_filesystem",
        integrity_status="unknown", siglip_embedding=[3.0] + [0.0] * 1151,
    )
    db.add(img)
    await db.flush()
    tag = await TagService(db).find_or_create("sword", TagKind.general)
    db.add(TagHead(
        tag_id=tag.id, embedding_version=s.embedder_model_version,
        weights=[1.0] + [0.0] * 1151, bias=0.0, suggest_threshold=0.5,
        auto_apply_threshold=None, n_pos=10, n_neg=30,
        ap=0.8, precision_cv=0.9, recall=0.6,
    ))
    await db.commit()
    resp = await client.get(f"/api/images/{img.id}/suggestions")
    assert resp.status_code == 200
    body = await resp.get_json()
    general = body["by_category"].get("general", [])
    s2 = next(x for x in general if x["canonical_tag_id"] == tag.id)
    assert s2["source"] == "head"
    assert s2["above_threshold"] is True   # ~0.73 clears the 0.5 suggest cut


@pytest.mark.asyncio
async def test_accept_requires_tag_id(client, db):
    img = await _img(db)
    resp = await client.post(
        f"/api/images/{img.id}/suggestions/accept", json={}
    )
    assert resp.status_code == 400


@pytest.mark.asyncio
async def test_accept_applies_tag_to_image(client, db):
    # Camie/allowlist retired (#1189): accept applies the tag to THIS image
    # (source='ml_accepted', a head-training positive) — no bulk allowlist
    # fan-out anymore.
    from backend.app.models.tag import image_tag

    img = await _img(db)
    tag = await TagService(db).find_or_create("AcceptMe", TagKind.character)
    await db.commit()
    resp = await client.post(
        f"/api/images/{img.id}/suggestions/accept", json={"tag_id": tag.id}
    )
    assert resp.status_code == 200
    assert (await resp.get_json())["accepted"] is True
    src = (await db.execute(
        select(image_tag.c.source)
        .where(image_tag.c.image_record_id == img.id)
        .where(image_tag.c.tag_id == tag.id)
    )).scalar_one()
    assert src == "ml_accepted"


@pytest.mark.asyncio
async def test_dismiss(client, db):
    img = await _img(db)
    tag = await TagService(db).find_or_create("DismissMe", TagKind.general)
    await db.commit()
    resp = await client.post(
        f"/api/images/{img.id}/suggestions/dismiss", json={"tag_id": tag.id}
    )
    assert resp.status_code == 204


@pytest.mark.asyncio
async def test_undismiss_reverses_rejection(client, db):
    img = await _img(db)
    tag = await TagService(db).find_or_create("UndismissMe", TagKind.general)
    await db.commit()
    await client.post(
        f"/api/images/{img.id}/suggestions/dismiss", json={"tag_id": tag.id}
    )
    resp = await client.post(
        f"/api/images/{img.id}/suggestions/undismiss", json={"tag_id": tag.id}
    )
    assert resp.status_code == 204
    # Idempotent: un-rejecting again (nothing to clear) is still a 204.
    resp2 = await client.post(
        f"/api/images/{img.id}/suggestions/undismiss", json={"tag_id": tag.id}
    )
    assert resp2.status_code == 204
