import os.path

from exception.danmaku import DanmakuException
from workflow.danmaku import get_file_start, diff_danmaku_files, danmaku_to_subtitle
from workflow.video import multi_gpu_encode_video_with_subtitles


def do_workflow(video_file, danmaku_base_file, *danmaku_files):
    if not os.path.exists(danmaku_base_file):
        ...
    result = []
    try:
        start_ts = get_file_start(danmaku_base_file)
    except DanmakuException:
        print("基准弹幕文件异常，跳过")
        return
    result.append(danmaku_to_subtitle(danmaku_base_file, 0))
    for danmaku_file in danmaku_files:
        try:
            bias = diff_danmaku_files(danmaku_base_file, danmaku_file)
            if bias < -600:
                print("弹幕文件", danmaku_file, "反向偏移超过10分钟")
                continue
            result.append(danmaku_to_subtitle(danmaku_file, bias))
        except DanmakuException:
            print("弹幕文件", danmaku_file, "异常")
            continue
    print(result)
    multi_gpu_encode_video_with_subtitles(video_file, result, start_ts)
    # clean files
    for file in result:
        if os.path.isfile(file):
            os.remove(file)