from mkv.chapter import get_intro_outro
from audio.audio_utils import extract_audio_segment_tmp, get_audio_streams
from audio.audio_cross_correlation import get_audio_offset
from pathlib import Path
import numpy as np
from typing import Tuple

def check_intro_outro_sync(file: Path, length: float = 90.0, uncertainty: float = 0.1, max_offset: float = 0.1) -> Tuple[bool, float|None, int|None]:
    intro_outro = get_intro_outro(file, length)
    if intro_outro[0][0]<intro_outro[1][0]:
        intro = intro_outro[0]
        outro = intro_outro[1]
    else:
        intro = intro_outro[1]
        outro = intro_outro[0]

    ### Cut away any possible part that is not of the intro/outro
    intro = (intro[0]+intro[1]*(uncertainty/2), intro[1]-intro[1]*uncertainty)
    outro = (outro[0]+outro[1]*(uncertainty/2), outro[1]-outro[1]*uncertainty)

    ### Get the intro and outro Segments
    audio_streams = get_audio_streams(file)
    intros : list[Path] = []
    outros : list[Path] = []
    # for i in audio_streams:
    #     intros.append(extract_audio_segment_tmp(file, intro, None, i))
    #     outros.append(extract_audio_segment_tmp(file, outro, None, i))
    intros.append(extract_audio_segment_tmp(file, intro, None, audio_streams[0]))
    outros.append(extract_audio_segment_tmp(file, outro, None, audio_streams[0]))
    intros.append(extract_audio_segment_tmp(file, intro, None, audio_streams[-1]))
    outros.append(extract_audio_segment_tmp(file, outro, None, audio_streams[-1]))

    ### correlate
    intro_offsets = []
    outro_offsets = []
    for intro_a, outro_a in zip(intros, outros):
        idx = intros.index(intro_a)
        odx = outros.index(outro_a)
        for intro_b, outro_b in zip(intros[idx+1:], outros[odx+1:]):
            intro_offset = get_audio_offset(intro_a, intro_b, intro[1])
            intro_offsets.append(intro_offset)
            # print(f"intro_offset between Audioline {intros.index(intro_a)} and {intros.index(intro_b)} is {intro_offset}")
            outro_offset = get_audio_offset(outro_a, outro_b, outro[1])
            outro_offsets.append(outro_offset)
            # print(f"outro_offset between Audioline {outros.index(outro_a)} and {outros.index(outro_b)} is {outro_offset}")
            # print(f"Offset of offsets: {(outro_offset - intro_offset)}")
    mean_offset = float(np.mean([intro_offsets[0], outro_offsets[0]]))

    # print(f"Mean Intro Offset: {round(np.mean(intro_offsets), 1)}")
    # print(f"Mean Outro Offset: {round(np.mean(outro_offsets), 1)}")
    # print(f"Offset of mean offsets: {round(np.mean(outro_offsets) - np.mean(intro_offsets)), 3}")

    if np.abs(np.mean(outro_offsets) - np.mean(intro_offsets)) < max_offset:
        return True, mean_offset, audio_streams[-1]
    return False, None, None