import datetime
import os
import argparse
import subprocess
from hashlib import md5
from typing import Union

from bs4 import BeautifulSoup

from config import DANMAKU_FACTORY_EXEC, VIDEO_RESOLUTION, DANMAKU_SPEED, DEFAULT_FONT_NAME, DANMAKU_FONT_SIZE
from exception.danmaku import NoDanmakuException, DanmakuFormatErrorException
from util.file import check_file_exist


def get_file_start(file: Union[os.PathLike[str], str]) -> float:
    with open(file, "r", encoding="utf-8") as f:
        soup = BeautifulSoup("".join(f.readlines()), "lxml")
        danmaku_item = soup.find("d")
        if danmaku_item is None:
            # 没有弹幕？
            raise NoDanmakuException()
        danmaku_info = danmaku_item["p"]
        split_info = danmaku_info.split(",")
        if len(split_info) < 5:
            raise DanmakuFormatErrorException()
        bias_sec = float(split_info[0])
        bias_ts_ms = int(split_info[4])
        return bias_ts_ms / 1000 - bias_sec


def diff_danmaku_files(base_file: Union[os.PathLike[str], str], file: Union[os.PathLike[str], str]) -> float:
    return get_file_start(file) - get_file_start(base_file)


def danmaku_to_subtitle(file: Union[os.PathLike[str], str], time_shift: float):
    new_subtitle_name = md5(file.encode("utf-8")).hexdigest() + ".ass"
    process = subprocess.Popen((
        DANMAKU_FACTORY_EXEC, "--ignore-warnings",
        "-r", str(VIDEO_RESOLUTION), "-s", str(DANMAKU_SPEED), "-f", "5",
        "-S", str(DANMAKU_FONT_SIZE), "-N", str(DEFAULT_FONT_NAME), "--showmsgbox", "FALSE",
        "-O", "255", "-L", "1", "-D", "0",
        "-o", "ass", new_subtitle_name, "-i", file, "-t", str(time_shift)
    ))
    process.wait()
    return new_subtitle_name


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("base", help="以此为标准")
    parser.add_argument("file", nargs="+", help="需要对齐的文件")
    args = parser.parse_args()
    check_file_exist(args.base)
    base_start_ts = get_file_start(args.base)
    base_start = datetime.datetime.fromtimestamp(base_start_ts)
    print("[+] 基准文件[{:s}]，开始时间：{:.3f}，时间：{}".format(args.base, base_start_ts, base_start))
    for _file in args.file:
        check_file_exist(_file)
        file_start_ts = get_file_start(_file)
        diff_sec = file_start_ts - base_start_ts
        print("[+] 待调整文件[{:s}]，开始时间：{:.3f}，偏差：{:.3f}".format(_file, file_start_ts, diff_sec))