import aiohttp

from .recognise import RecognisedService


async def get_url_after_redirect(url: str) -> str:
    async with aiohttp.ClientSession() as session:
        async with session.head(url, allow_redirects=True) as resp:
            return str(resp.url)


async def get_id(recognised: RecognisedService):
    if recognised.name == "yt":
        return (
            recognised.parse_result.path.replace("/", "")
            if (recognised.parse_result.netloc.endswith("youtu.be"))
            else recognised.parse_result.query.split("=")[1].split("&")[0]
        )

    elif recognised.name == "spot":
        if recognised.parse_result.netloc.endswith("open.spotify.com"):
            return recognised.parse_result.path.split("/")[2]
        else:
            url = await get_url_after_redirect(recognised.parse_result.geturl())
            return url.split("/")[-1].split("?")[0]

    elif recognised.name == "deez":
        if recognised.parse_result.netloc.endswith("deezer.com"):
            return recognised.parse_result.path.split("/")[-1]
        else:
            url = await get_url_after_redirect(recognised.parse_result.geturl())
            return url.split("/")[-1].split("?")[0]

    elif recognised.name == "sc":
        if not recognised.parse_result.netloc.startswith("on"):
            return recognised.parse_result.geturl()
        return await get_url_after_redirect(recognised.parse_result.geturl())