bangumi.py

import json
import logging
import re
import time
from typing import Optional

from sqlalchemy.sql import func
from sqlmodel import Session, and_, delete, false, or_, select

from module.models import Bangumi, BangumiUpdate

logger = logging.getLogger(__name__)


def _normalize_group_name(group: str | None) -> str:
    """Normalize group name for comparison by removing common separators."""
    if not group:
        return ""
    # Remove common separators (&, ×, _, -) and normalize to lowercase
    return re.sub(r"[&×_\-]", "", group).lower().strip()


def _groups_are_similar(group1: str | None, group2: str | None) -> bool:
    """
    Check if two group names are similar enough to be considered the same group.

    Handles cases like:
    - "LoliHouse" vs "LoliHouse&动漫国字幕组"
    - "字幕组A" vs "字幕组A×字幕组B"
    """
    if not group1 or not group2:
        return False

    # Exact match or substring match (one contains the other)
    if group1 == group2 or group1 in group2 or group2 in group1:
        return True

    # Normalized comparison - check if core group names overlap
    norm1 = _normalize_group_name(group1)
    norm2 = _normalize_group_name(group2)
    return norm1 in norm2 or norm2 in norm1


def _get_aliases_list(bangumi: Bangumi) -> list[str]:
    """Get the list of title aliases from a bangumi's title_aliases JSON field."""
    if not bangumi.title_aliases:
        return []
    try:
        aliases = json.loads(bangumi.title_aliases)
        return aliases if isinstance(aliases, list) else []
    except (json.JSONDecodeError, TypeError):
        return []


def _set_aliases_list(bangumi: Bangumi, aliases: list[str]) -> None:
    """Set the title aliases JSON field from a list."""
    if not aliases:
        bangumi.title_aliases = None
    else:
        # Remove duplicates while preserving order
        unique_aliases = list(dict.fromkeys(aliases))
        bangumi.title_aliases = json.dumps(unique_aliases, ensure_ascii=False)


# Module-level TTL cache for search_all results
_bangumi_cache: list[Bangumi] | None = None
_bangumi_cache_time: float = 0
_BANGUMI_CACHE_TTL: float = 300.0  # 5 minutes - extended from 60s to reduce DB queries


def _invalidate_bangumi_cache():
    global _bangumi_cache, _bangumi_cache_time
    _bangumi_cache = None
    _bangumi_cache_time = 0


class BangumiDatabase:
    def __init__(self, session: Session):
        self.session = session

    def find_semantic_duplicate(self, data: Bangumi) -> Optional[Bangumi]:
        """
        Find existing bangumi that semantically matches the new one.

        This handles cases where subtitle groups change naming mid-season.
        A semantic match requires:
        - Same official_title
        - Same dpi (resolution)
        - Same subtitle type
        - Same source
        - Similar group_name (one contains the other)

        Returns the matching Bangumi if found, None otherwise.
        """
        statement = select(Bangumi).where(
            and_(
                Bangumi.official_title == data.official_title,
                Bangumi.deleted == false(),
            )
        )
        candidates = self.session.execute(statement).scalars().all()

        for candidate in candidates:
            is_exact_duplicate = (
                candidate.title_raw == data.title_raw
                and candidate.group_name == data.group_name
            )
            if is_exact_duplicate:
                continue

            is_semantic_match = (
                candidate.dpi == data.dpi
                and candidate.subtitle == data.subtitle
                and candidate.source == data.source
                and _groups_are_similar(candidate.group_name, data.group_name)
            )
            if is_semantic_match:
                logger.debug(
                    "[Database] Found semantic duplicate: '%s' matches "
                    "existing '%s' (official: %s)",
                    data.title_raw,
                    candidate.title_raw,
                    data.official_title,
                )
                return candidate

        return None

    def add_title_alias(
        self, bangumi_id: int, new_title_raw: str, auto_commit: bool = True
    ) -> bool:
        """
        Add a new title_raw alias to an existing bangumi.

        This allows a single bangumi entry to match multiple naming patterns.
        """
        bangumi = self.session.get(Bangumi, bangumi_id)
        if not bangumi:
            logger.warning(
                f"[Database] Cannot add alias: bangumi id {bangumi_id} not found"
            )
            return False

        # Don't add if it's the same as the main title_raw
        if bangumi.title_raw == new_title_raw:
            return False

        # Get existing aliases and add the new one
        aliases = _get_aliases_list(bangumi)
        if new_title_raw in aliases:
            return False  # Already exists

        aliases.append(new_title_raw)
        _set_aliases_list(bangumi, aliases)

        self.session.add(bangumi)
        if auto_commit:
            self.session.commit()
            _invalidate_bangumi_cache()
        logger.info(
            f"[Database] Added alias '{new_title_raw}' to bangumi '{bangumi.official_title}' "
            f"(id: {bangumi_id})"
        )
        return True

    def get_all_title_patterns(self, bangumi: Bangumi) -> list[str]:
        """Get all title patterns for matching (title_raw + all aliases)."""
        patterns = [bangumi.title_raw]
        patterns.extend(_get_aliases_list(bangumi))
        return patterns

    def _is_duplicate(self, data: Bangumi) -> bool:
        """Check if a bangumi rule already exists based on title_raw and group_name."""
        statement = select(Bangumi).where(
            and_(
                Bangumi.title_raw == data.title_raw,
                Bangumi.group_name == data.group_name,
            )
        )
        result = self.session.execute(statement)
        return result.scalar_one_or_none() is not None

    def add(self, data: Bangumi) -> bool:
        if self._is_duplicate(data):
            logger.debug(
                "[Database] Skipping duplicate: %s (%s)",
                data.official_title,
                data.group_name,
            )
            return False

        # Check for semantic duplicate (same anime, different naming pattern)
        semantic_match = self.find_semantic_duplicate(data)
        if semantic_match:
            # Add as alias instead of creating new entry
            self.add_title_alias(semantic_match.id, data.title_raw)
            logger.info(
                f"[Database] Merged '{data.title_raw}' as alias to existing "
                f"'{semantic_match.title_raw}' (official: {data.official_title})"
            )
            return False  # Return False since we didn't add a new entry

        self.session.add(data)
        self.session.commit()
        _invalidate_bangumi_cache()
        logger.debug("[Database] Insert %s into database.", data.official_title)
        return True

    def add_all(self, datas: list[Bangumi]) -> int:
        """Add multiple bangumi, skipping duplicates. Returns count of added items."""
        if not datas:
            return 0

        # Batch query: get all existing (title_raw, group_name) combinations in one query
        # This replaces N individual _is_duplicate() calls with a single SELECT
        keys_to_check = [(d.title_raw, d.group_name) for d in datas]
        conditions = [
            and_(Bangumi.title_raw == tr, Bangumi.group_name == gn)
            for tr, gn in keys_to_check
        ]
        if conditions:
            statement = select(Bangumi.title_raw, Bangumi.group_name).where(
                or_(*conditions)
            )
            result = self.session.execute(statement)
            existing = set(result.all())
        else:
            existing = set()

        # Filter out exact duplicates
        to_add = [d for d in datas if (d.title_raw, d.group_name) not in existing]

        # Check for semantic duplicates and add as aliases
        semantic_merged = 0
        really_to_add = []
        for d in to_add:
            semantic_match = self.find_semantic_duplicate(d)
            if semantic_match:
                # Add as alias instead of creating new entry (defer commit)
                self.add_title_alias(semantic_match.id, d.title_raw, auto_commit=False)
                semantic_merged += 1
                logger.info(
                    f"[Database] Merged '{d.title_raw}' as alias to existing "
                    f"'{semantic_match.title_raw}' (official: {d.official_title})"
                )
            else:
                really_to_add.append(d)

        # Also deduplicate within the batch itself
        seen = set()
        unique_to_add = []
        for d in really_to_add:
            key = (d.title_raw, d.group_name)
            if key not in seen:
                seen.add(key)
                unique_to_add.append(d)

        if not unique_to_add:
            if semantic_merged > 0:
                self.session.commit()
                _invalidate_bangumi_cache()
                logger.debug(
                    "[Database] %s bangumi merged as aliases, " "rest were duplicates.",
                    semantic_merged,
                )
            else:
                logger.debug(
                    "[Database] All %s bangumi already exist, skipping.",
                    len(datas),
                )
            return 0

        self.session.add_all(unique_to_add)
        self.session.commit()
        _invalidate_bangumi_cache()
        skipped = len(datas) - len(unique_to_add) - semantic_merged
        if skipped > 0 or semantic_merged > 0:
            logger.debug(
                "[Database] Insert %s bangumi, "
                "skipped %s duplicates, merged %s as aliases.",
                len(unique_to_add),
                skipped,
                semantic_merged,
            )
        else:
            logger.debug(
                "[Database] Insert %s bangumi into database.",
                len(unique_to_add),
            )
        return len(unique_to_add)

    def update(self, data: Bangumi | BangumiUpdate, _id: int = None) -> bool:
        if _id and isinstance(data, BangumiUpdate):
            db_data = self.session.get(Bangumi, _id)
        elif isinstance(data, Bangumi):
            db_data = self.session.get(Bangumi, data.id)
        else:
            return False
        if not db_data:
            return False
        bangumi_data = data.model_dump(exclude_unset=True)
        for key, value in bangumi_data.items():
            setattr(db_data, key, value)
        self.session.add(db_data)
        self.session.commit()
        _invalidate_bangumi_cache()
        logger.debug("[Database] Update %s", data.official_title)
        return True

    def update_all(self, datas: list[Bangumi]):
        self.session.add_all(datas)
        self.session.commit()
        _invalidate_bangumi_cache()
        logger.debug("[Database] Update %s bangumi.", len(datas))

    def update_rss(self, title_raw: str, rss_set: str):
        statement = select(Bangumi).where(Bangumi.title_raw == title_raw)
        result = self.session.execute(statement)
        bangumi = result.scalar_one_or_none()
        if bangumi:
            bangumi.rss_link = rss_set
            bangumi.added = False
            self.session.add(bangumi)
            self.session.commit()
            _invalidate_bangumi_cache()
            logger.debug("[Database] Update %s rss_link to %s.", title_raw, rss_set)

    def update_poster(self, title_raw: str, poster_link: str):
        statement = select(Bangumi).where(Bangumi.title_raw == title_raw)
        result = self.session.execute(statement)
        bangumi = result.scalar_one_or_none()
        if bangumi:
            bangumi.poster_link = poster_link
            self.session.add(bangumi)
            self.session.commit()
            _invalidate_bangumi_cache()
            logger.debug(
                "[Database] Update %s poster_link to %s.", title_raw, poster_link
            )

    def delete_one(self, _id: int):
        statement = select(Bangumi).where(Bangumi.id == _id)
        result = self.session.execute(statement)
        bangumi = result.scalar_one_or_none()
        if bangumi:
            self.session.delete(bangumi)
            self.session.commit()
            _invalidate_bangumi_cache()
            logger.debug("[Database] Delete bangumi id: %s.", _id)

    def delete_all(self):
        statement = delete(Bangumi)
        self.session.execute(statement)
        self.session.commit()
        _invalidate_bangumi_cache()

    def search_all(self) -> list[Bangumi]:
        global _bangumi_cache, _bangumi_cache_time
        now = time.time()
        if (
            _bangumi_cache is not None
            and (now - _bangumi_cache_time) < _BANGUMI_CACHE_TTL
        ):
            return _bangumi_cache
        statement = select(Bangumi)
        result = self.session.execute(statement)
        bangumis = list(result.scalars().all())
        # Expunge objects from session to prevent DetachedInstanceError when
        # cached objects are accessed from a different session/request context
        for b in bangumis:
            self.session.expunge(b)
        _bangumi_cache = bangumis
        _bangumi_cache_time = now
        return _bangumi_cache

    def search_id(self, _id: int) -> Optional[Bangumi]:
        statement = select(Bangumi).where(Bangumi.id == _id)
        bangumi = self.session.execute(statement).scalar_one_or_none()
        if bangumi is None:
            logger.warning(f"[Database] Cannot find bangumi id: {_id}.")
            return None
        logger.debug("[Database] Find bangumi id: %s.", _id)
        return bangumi

    def search_ids(self, ids: list[int]) -> list[Bangumi]:
        """Batch lookup multiple bangumi by their IDs."""
        if not ids:
            return []
        statement = select(Bangumi).where(Bangumi.id.in_(ids))
        result = self.session.execute(statement)
        return list(result.scalars().all())

    def match_poster(self, bangumi_name: str) -> str:
        statement = select(Bangumi).where(
            func.instr(bangumi_name, Bangumi.official_title) > 0
        )
        data = self.session.execute(statement).scalar_one_or_none()
        return data.poster_link if data else ""

    def match_list(self, torrent_list: list, rss_link: str) -> list:
        match_datas = self.search_all()
        if not match_datas:
            return torrent_list

        # Build index for O(1) lookup after regex match
        # Include both title_raw and all aliases
        title_index: dict[str, Bangumi] = {}
        for m in match_datas:
            # Add main title_raw (skip if None to avoid TypeError in sorted())
            if m.title_raw:
                title_index[m.title_raw] = m
            # Add all aliases
            for alias in _get_aliases_list(m):
                if alias:
                    title_index[alias] = m

        # Build compiled regex pattern for fast substring matching
        # Sort by length descending so longer (more specific) matches are found first
        sorted_titles = sorted(title_index.keys(), key=len, reverse=True)
        # Escape special regex characters and join with alternation
        pattern = "|".join(re.escape(title) for title in sorted_titles)
        title_regex = re.compile(pattern)

        unmatched = []
        rss_updated = set()
        for torrent in torrent_list:
            match = title_regex.search(torrent.name)
            if match:
                matched_title = match.group(0)
                match_data = title_index[matched_title]
                # Use the bangumi's main title_raw for rss_updated tracking
                if (
                    rss_link not in match_data.rss_link
                    and match_data.title_raw not in rss_updated
                ):
                    match_data = self.session.merge(match_data)
                    match_data.rss_link += f",{rss_link}"
                    match_data.added = False
                    rss_updated.add(match_data.title_raw)
            else:
                unmatched.append(torrent)
        # Batch commit all rss_link updates
        if rss_updated:
            self.session.commit()
            _invalidate_bangumi_cache()
            logger.debug(
                "[Database] Batch updated rss_link for %s bangumi.",
                len(rss_updated),
            )
        return unmatched

    def match_torrent(self, torrent_name: str) -> Optional[Bangumi]:
        """
        Match torrent name to a bangumi, checking both title_raw and title_aliases.

        Returns the bangumi with the longest matching pattern for specificity.
        """
        match_datas = self.search_all()
        if not match_datas:
            return None

        best_match: Optional[Bangumi] = None
        best_match_len = 0

        for bangumi in match_datas:
            if bangumi.deleted:
                continue

            # Check all patterns (title_raw + aliases)
            patterns = self.get_all_title_patterns(bangumi)
            for pattern in patterns:
                if pattern in torrent_name:
                    # Prefer longer matches (more specific)
                    if len(pattern) > best_match_len:
                        best_match = bangumi
                        best_match_len = len(pattern)

        return best_match

    def not_complete(self) -> list[Bangumi]:
        condition = select(Bangumi).where(
            and_(Bangumi.eps_collect == false(), Bangumi.deleted == false())
        )
        result = self.session.execute(condition)
        return list(result.scalars().all())

    def not_added(self) -> list[Bangumi]:
        conditions = select(Bangumi).where(
            or_(
                Bangumi.added == 0,
                Bangumi.rule_name.is_(None),
                Bangumi.save_path.is_(None),
            )
        )
        result = self.session.execute(conditions)
        return list(result.scalars().all())

    def disable_rule(self, _id: int):
        statement = select(Bangumi).where(Bangumi.id == _id)
        result = self.session.execute(statement)
        bangumi = result.scalar_one_or_none()
        if bangumi:
            bangumi.deleted = True
            self.session.add(bangumi)
            self.session.commit()
            _invalidate_bangumi_cache()
            logger.debug("[Database] Disable rule %s.", bangumi.title_raw)

    def search_rss(self, rss_link: str) -> list[Bangumi]:
        statement = select(Bangumi).where(func.instr(rss_link, Bangumi.rss_link) > 0)
        result = self.session.execute(statement)
        return list(result.scalars().all())

    def archive_one(self, _id: int) -> bool:
        """Set archived=True for the given bangumi."""
        bangumi = self.session.get(Bangumi, _id)
        if not bangumi:
            logger.warning(f"[Database] Cannot archive bangumi id: {_id}, not found.")
            return False
        bangumi.archived = True
        self.session.add(bangumi)
        self.session.commit()
        _invalidate_bangumi_cache()
        logger.debug("[Database] Archived bangumi id: %s.", _id)
        return True

    def unarchive_one(self, _id: int) -> bool:
        """Set archived=False for the given bangumi."""
        bangumi = self.session.get(Bangumi, _id)
        if not bangumi:
            logger.warning(f"[Database] Cannot unarchive bangumi id: {_id}, not found.")
            return False
        bangumi.archived = False
        self.session.add(bangumi)
        self.session.commit()
        _invalidate_bangumi_cache()
        logger.debug("[Database] Unarchived bangumi id: %s.", _id)
        return True

    def match_by_save_path(self, save_path: str) -> Optional[Bangumi]:
        """Find bangumi by save_path to get offset.

        Tries exact match first, then falls back to matching with/without trailing slashes
        and different path separators.

        Note: When multiple subscriptions share the same save_path (e.g., different RSS
        sources for the same anime), this returns the first match. Use match_torrent()
        for more accurate matching when torrent_name is available.
        """
        if not save_path:
            return None

        # Try exact match first
        statement = select(Bangumi).where(
            and_(Bangumi.save_path == save_path, Bangumi.deleted == false())
        )
        result = self.session.execute(statement)
        bangumi = result.scalars().first()
        if bangumi:
            return bangumi

        # Normalize the input path and try variations
        normalized = save_path.replace("\\", "/").rstrip("/")
        variations = [
            normalized,
            normalized + "/",
            save_path.rstrip("/"),
            save_path.rstrip("\\"),
        ]
        # Remove duplicates while preserving order
        seen = {save_path}
        unique_variations = []
        for v in variations:
            if v not in seen:
                seen.add(v)
                unique_variations.append(v)

        for variant in unique_variations:
            statement = select(Bangumi).where(
                and_(Bangumi.save_path == variant, Bangumi.deleted == false())
            )
            result = self.session.execute(statement)
            bangumi = result.scalars().first()
            if bangumi:
                return bangumi

        return None

    def get_needs_review(self) -> list[Bangumi]:
        """Get all bangumi that need review for offset mismatch."""
        statement = select(Bangumi).where(
            and_(
                Bangumi.needs_review == True,  # noqa: E712
                Bangumi.deleted == false(),
            )
        )
        result = self.session.execute(statement)
        return list(result.scalars().all())

    def get_active_for_scan(self) -> list[Bangumi]:
        """Get all active (non-deleted, non-archived) bangumi for offset scanning."""
        statement = select(Bangumi).where(
            and_(
                Bangumi.deleted == false(),
                Bangumi.archived == false(),
            )
        )
        result = self.session.execute(statement)
        return list(result.scalars().all())

    def set_needs_review(
        self,
        _id: int,
        reason: str,
        suggested_season_offset: int | None = None,
        suggested_episode_offset: int | None = None,
    ) -> bool:
        """Mark a bangumi as needing review with suggested offsets.

        Args:
            _id: The bangumi ID
            reason: Human-readable reason for the review
            suggested_season_offset: Suggested season offset value
            suggested_episode_offset: Suggested episode offset value
        """
        bangumi = self.session.get(Bangumi, _id)
        if not bangumi:
            return False
        bangumi.needs_review = True
        bangumi.needs_review_reason = reason
        bangumi.suggested_season_offset = suggested_season_offset
        bangumi.suggested_episode_offset = suggested_episode_offset
        self.session.add(bangumi)
        self.session.commit()
        _invalidate_bangumi_cache()
        logger.debug(
            "[Database] Marked bangumi id %s as needs_review: %s "
            "(suggested: season=%s, episode=%s)",
            _id,
            reason,
            suggested_season_offset,
            suggested_episode_offset,
        )
        return True

    def clear_needs_review(self, _id: int) -> bool:
        """Clear the needs_review flag and suggested offsets for a bangumi."""
        bangumi = self.session.get(Bangumi, _id)
        if not bangumi:
            return False
        bangumi.needs_review = False
        bangumi.needs_review_reason = None
        bangumi.suggested_season_offset = None
        bangumi.suggested_episode_offset = None
        self.session.add(bangumi)
        self.session.commit()
        _invalidate_bangumi_cache()
        logger.debug("[Database] Cleared needs_review for bangumi id %s", _id)
        return True

    def set_weekday(self, _id: int, weekday: int | None) -> bool:
        """Set air_weekday and weekday_locked for manual calendar assignment."""
        bangumi = self.session.get(Bangumi, _id)
        if not bangumi:
            return False
        if weekday is not None:
            bangumi.air_weekday = weekday
            bangumi.weekday_locked = True
        else:
            bangumi.air_weekday = None
            bangumi.weekday_locked = False
        self.session.add(bangumi)
        self.session.commit()
        _invalidate_bangumi_cache()
        logger.debug(
            "[Database] Set weekday=%s, locked=%s for bangumi id %s",
            weekday,
            bangumi.weekday_locked,
            _id,
        )
        return True