"""评测台：查询用户 segment 是否已完成回忆录 Phase1（topic_category 已写入）。"""

from __future__ import annotations

from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from app.features.conversation.models import Conversation, Segment
from app.features.evaluation.errors import (
    EvaluationBadRequestError,
    EvaluationNotFoundError,
)
from app.features.evaluation.schemas import MemoirPhase1ReadyOut


class MemoirReadinessService:
    def __init__(self, db: AsyncSession) -> None:
        self._db = db

    async def memoir_phase1_ready_for_segments(
        self,
        *,
        conversation_id: str,
        segment_ids: list[str],
    ) -> MemoirPhase1ReadyOut:
        cid = (conversation_id or "").strip()
        if not cid:
            raise EvaluationBadRequestError("conversation_id is required")
        ids = [s.strip() for s in segment_ids if (s or "").strip()]
        if not ids:
            raise EvaluationBadRequestError("segment_ids is required")

        conv = await self._db.get(Conversation, cid)
        if not conv or conv.deleted_at is not None:
            raise EvaluationNotFoundError("conversation not found")

        stmt = select(Segment).where(
            Segment.id.in_(ids),
            Segment.conversation_id == cid,
        )
        result = await self._db.execute(stmt)
        rows = list(result.scalars().all())
        found_ids = {s.id for s in rows}
        missing = [i for i in ids if i not in found_ids]
        if missing:
            raise EvaluationBadRequestError(
                "segment not in conversation: " + ", ".join(missing[:5])
                + ("…" if len(missing) > 5 else "")
            )

        pending = [s.id for s in rows if s.topic_category is None]
        ready = len(pending) == 0
        return MemoirPhase1ReadyOut(
            ready=ready,
            checked_segment_ids=ids,
            pending_segment_ids=pending,
        )