import json
import os.path
import time

from entity.ffmpeg import FfmpegTask
import logging

from util import ffmpeg, oss

logger = logging.getLogger('biz/ffmpeg')


def parse_ffmpeg_task(task_info, template_info):
    tasks = []
    # 中间片段
    task_params_str = task_info.get("taskParams", "{}")
    task_params = json.loads(task_params_str)
    for part in template_info.get("video_parts"):
        source = parse_video(part.get('source'), task_params, template_info)
        if not source:
            logger.warning("no video found for part: " + str(part))
            continue
        only_if = part.get('only_if', '')
        if only_if:
            if not check_placeholder_exist(only_if, task_params):
                logger.info("because only_if exist, placeholder: %s not exist, skip part: %s", only_if, part)
                continue
        sub_ffmpeg_task = FfmpegTask(source)
        sub_ffmpeg_task.annexb = True
        sub_ffmpeg_task.frame_rate = template_info.get("frame_rate", 25)
        for effect in part.get('effects', []):
            sub_ffmpeg_task.add_effect(effect)
        for lut in part.get('filters', []):
            sub_ffmpeg_task.add_lut(os.path.join(template_info.get("local_path"), lut))
        for audio in part.get('audios', []):
            sub_ffmpeg_task.add_audios(os.path.join(template_info.get("local_path"), audio))
        for overlay in part.get('overlays', []):
            sub_ffmpeg_task.add_overlay(os.path.join(template_info.get("local_path"), overlay))
        tasks.append(sub_ffmpeg_task)
    output_file = "out_" + str(time.time()) + ".mp4"
    task = FfmpegTask(tasks, output_file=output_file)
    overall = template_info.get("overall_template")
    task.center_cut = template_info.get("crop_mode", None)
    task.frame_rate = template_info.get("frame_rate", 25)
    if overall.get('source', ''):
        source = parse_video(overall.get('source'), task_params, template_info)
        task.add_inputs(source)
    for effect in overall.get('effects', []):
        task.add_effect(effect)
    for lut in overall.get('filters', []):
        task.add_lut(os.path.join(template_info.get("local_path"), lut))
    for audio in overall.get('audios', []):
        task.add_audios(os.path.join(template_info.get("local_path"), audio))
    for overlay in overall.get('overlays', []):
        task.add_overlay(os.path.join(template_info.get("local_path"), overlay))
    return task


def parse_video(source, task_params, template_info):
    print(source)
    if source.startswith('PLACEHOLDER_'):
        placeholder_id = source.replace('PLACEHOLDER_', '')
        new_sources = task_params.get(placeholder_id, [])
        if type(new_sources) is list:
            if len(new_sources) == 0:
                logger.debug("no video found for placeholder: " + placeholder_id)
                return None
            else:
                # TODO: Random Pick / Policy Pick
                new_sources = new_sources[0].get("url")
        if new_sources.startswith("http"):
            _, source_name = os.path.split(new_sources)
            oss.download_from_oss(new_sources, source_name)
            return source_name
        return new_sources
    return os.path.join(template_info.get("local_path"), source)


def check_placeholder_exist(placeholder_id, task_params):
    if placeholder_id in task_params:
        new_sources = task_params.get(placeholder_id, [])
        if type(new_sources) is list:
            if len(new_sources) == 0:
                return False
            else:
                return True
        return True
    return False

def start_ffmpeg_task(ffmpeg_task):
   for task in ffmpeg_task.analyze_input_render_tasks():
       result = start_ffmpeg_task(task)
       if not result:
           return False
   ffmpeg_task.correct_task_type()
   return ffmpeg.start_render(ffmpeg_task)


def clear_task_tmp_file(ffmpeg_task):
    for task in ffmpeg_task.analyze_input_render_tasks():
        clear_task_tmp_file(task)
    try:
        if "template" not in ffmpeg_task.get_output_file():
            os.remove(ffmpeg_task.get_output_file())
            logger.info("delete tmp file: " + ffmpeg_task.get_output_file())
        else:
            logger.info("skip delete template file: " + ffmpeg_task.get_output_file())
    except OSError:
        logger.warning("delete tmp file failed: " + ffmpeg_task.get_output_file())
        return False
    return True


def probe_video_info(ffmpeg_task):
    # 获取视频长度宽度和时长
    return ffmpeg.probe_video_info(ffmpeg_task.get_output_file())