import asyncio
import hashlib
import logging
import os
import tempfile
from Utils import calcDownloadPath
import pathlib
import pysubs2
from pysubs2.time import ms_to_times
from pysubs2.formats.substation import SubstationFormat

LOG = logging.getLogger('player.subtitle')


def ms_to_timestamp(ms: int) -> str:
    ms = max(0, ms)
    h, m, s, ms = ms_to_times(ms)
    cs = ms // 10
    return f"{h:01d}:{m:02d}:{s:02d}.{cs:02d}"


SubstationFormat.ms_to_timestamp = ms_to_timestamp


class Subtitle:
    allowedExtensions: tuple[str] = (".srt", ".vtt", ".ass",)

    async def make(self, path: str, file: str) -> bytes:
        realFile: str = calcDownloadPath(basePath=path, folder=file, createPath=False)

        rFile = pathlib.Path(realFile)

        if not rFile.exists():
            raise Exception(f"File '{file}' does not exist.")

        if not rFile.suffix in self.allowedExtensions:
            raise Exception(f"File '{file}' subtitle type is not supported.")

        if rFile.suffix == ".vtt":
            subData = ''
            with open(realFile, 'r') as f:
                subData = f.read()

            return subData

        subs = pysubs2.load(path=str(rFile))

        if len(subs.events) < 1:
            raise Exception(f"No subtitle events were found in '{rFile}'.")

        if len(subs.events) < 2:
            return subs.to_string('vtt')

        if subs.events[0].end == subs.events[len(subs.events)-1].end:
            subs.events.pop(0)

        return subs.to_string('vtt')