import glob
import re
from pathlib import Path
from urllib.parse import quote
from aiohttp.web import Response
from .ffprobe import ffprobe
from .Subtitle import Subtitle
from .Utils import calcDownloadPath, checkId, StreamingError


class Playlist:
    _url: str = None

    def __init__(self, url: str):
        self.url = url

    async def make(self, download_path: str, file: str) -> str | Response:
        rFile = Path(calcDownloadPath(basePath=download_path, folder=file, createPath=False))

        if not rFile.exists():
            possibleFile = checkId(download_path, rFile)
            if not possibleFile:
                raise StreamingError(f"File '{rFile}' does not exist.")

            return Response(
                status=302,
                headers={
                    "Location": f"{self.url}api/player/playlist/{quote(str(possibleFile).replace(download_path, '').strip('/'))}.m3u8"
                },
            )

        try:
            ff = await ffprobe(rFile)
        except UnicodeDecodeError:
            pass

        if "duration" not in ff.metadata:
            raise StreamingError(f"Unable to get '{rFile}' duration.")

        duration: float = float(ff.metadata.get("duration"))

        playlist = []
        playlist.append("#EXTM3U")

        subs = ""

        index = 0
        for item in self.getSideCarFiles(rFile):
            if item.suffix not in Subtitle.allowedExtensions:
                continue

            index += 1
            lang: str = "und"
            lg = re.search(r"\.(?P<lang>\w{2,3})\.\w{3}$", item.name)
            if lg:
                lang = lg.groupdict().get("lang")

            subs = ',SUBTITLES="subs"'

            url = f"{self.url}api/player/m3u8/subtitle/{quote(str(Path(file).with_name(item.name)))}.m3u8?duration={duration}"
            name = f"{item.suffix[1:].upper()} ({index}) - {lang}"
            playlist.append(
                f'#EXT-X-MEDIA:TYPE=SUBTITLES,GROUP-ID="subs",NAME="{name}",DEFAULT=NO,AUTOSELECT=NO,FORCED=NO,LANGUAGE="{lang}",URI="{url}"'
            )

        playlist.append(f"#EXT-X-STREAM-INF:PROGRAM-ID=1{subs}")
        playlist.append(f"{self.url}api/player/m3u8/video/{quote(file)}.m3u8")

        return "\n".join(playlist)

    def getSideCarFiles(self, file: Path) -> list[Path]:
        """
        Get sidecar files for the given file.

        :param file: File to get sidecar files for.
        :return: List of sidecar files.
        """
        files = []

        for sub_file in file.parent.glob(f"{glob.escape(file.stem)}.*"):
            if sub_file == file or sub_file.is_file() is False or sub_file.stem.startswith("."):
                continue

            files.append(sub_file)

        return files