from dataclasses import dataclass
from typing import Awaitable, Callable, Literal
from urllib.parse import ParseResult, urlparse

from bot.modules.database import db
from bot.modules.database.db import DBDict
from bot.modules.deezer import deezer
from bot.modules.soundcloud import soundcloud
from bot.modules.spotify import spotify
from bot.modules.youtube import youtube


@dataclass
class RecognisedService:
    name: Literal["yt", "spot", "deez", "sc"]
    db_table: DBDict
    by_id_func: Callable | Awaitable
    parse_result: ParseResult


def recognise_music_service(url: str) -> RecognisedService | None:
    url = urlparse(url)
    if url.netloc.endswith("youtube.com") or url.netloc.endswith("youtu.be"):
        return RecognisedService(
            name="yt",
            db_table=db.youtube,
            by_id_func=youtube.songs.from_id,
            parse_result=url,
        )
    elif url.netloc.endswith("open.spotify.com") or url.netloc.endswith("spotify.link"):
        return RecognisedService(
            name="spot",
            db_table=db.spotify,
            by_id_func=spotify.songs.from_id,
            parse_result=url,
        )
    elif url.netloc.endswith("deezer.page.link") or url.netloc.endswith("deezer.com"):
        return RecognisedService(
            name="deez",
            db_table=db.deezer,
            by_id_func=deezer.songs.from_id,
            parse_result=url,
        )
    elif url.netloc.endswith("soundcloud.com"):
        return RecognisedService(
            name="sc",
            db_table=db.soundcloud,
            by_id_func=soundcloud.songs.from_url,
            parse_result=url,
        )
    else:
        return None