"""
Author: Uyanide pywang0608@foxmail.com
Date: 2026-04-05 02:13:49
Description: Musixmatch metadata enricher (matcher.track.get by Spotify track ID)
"""

from typing import Optional

from loguru import logger

from .base import BaseEnricher
from ..authenticators.musixmatch import MusixmatchAuthenticator
from ..models import TrackMeta

_MUSIXMATCH_TRACK_MATCH_URL = (
    "https://apic-desktop.musixmatch.com/ws/1.1/matcher.track.get"
)


class MusixmatchSpotifyEnricher(BaseEnricher):
    """Fill title, artist, album, and length from Musixmatch using Spotify track ID."""

    def __init__(self, auth: MusixmatchAuthenticator) -> None:
        self.auth = auth

    @property
    def name(self) -> str:
        return "musixmatch"

    @property
    def provides(self) -> set[str]:
        return {"title", "artist", "album", "length"}

    async def enrich(self, track: TrackMeta) -> Optional[dict]:
        if not track.trackid:
            return None

        logger.debug(f"Musixmatch enricher: looking up trackid={track.trackid}")

        try:
            data = await self.auth.get_json(
                _MUSIXMATCH_TRACK_MATCH_URL,
                {"track_spotify_id": track.trackid},
            )
        except Exception as e:
            logger.warning(f"Musixmatch enricher: request failed: {e}")
            return None

        if data is None:
            return None

        body = data.get("message", {}).get("body")
        t = body.get("track") if isinstance(body, dict) else None
        if not isinstance(t, dict):
            logger.debug(
                f"Musixmatch enricher: no track data for trackid={track.trackid}"
            )
            return None

        updates: dict = {}
        if isinstance(t.get("track_name"), str) and t["track_name"]:
            updates["title"] = t["track_name"]
        if isinstance(t.get("artist_name"), str) and t["artist_name"]:
            updates["artist"] = t["artist_name"]
        if isinstance(t.get("album_name"), str) and t["album_name"]:
            updates["album"] = t["album_name"]
        if isinstance(t.get("track_length"), int) and t["track_length"] > 0:
            updates["length"] = t["track_length"] * 1000

        if updates:
            logger.debug(f"Musixmatch enricher: filled {list(updates.keys())}")
        return updates or None