import { video_info } from '.';
import { LiveStreaming, Stream } from './classes/LiveStream';

export enum StreamType {
    Arbitrary = 'arbitrary',
    Raw = 'raw',
    OggOpus = 'ogg/opus',
    WebmOpus = 'webm/opus',
    Opus = 'opus'
}

interface InfoData {
    LiveStreamData: {
        isLive: boolean;
        dashManifestUrl: string;
        hlsManifestUrl: string;
    };
    html5player: string;
    format: any[];
    video_details: any;
}

function parseAudioFormats(formats: any[]) {
    const result: any[] = [];
    formats.forEach((format) => {
        const type = format.mimeType as string;
        if (type.startsWith('audio')) {
            format.codec = type.split('codecs="')[1].split('"')[0];
            format.container = type.split('audio/')[1].split(';')[0];
            result.push(format);
        }
    });
    return result;
}

export async function stream(url: string, cookie?: string): Promise<Stream | LiveStreaming> {
    const info = await video_info(url, cookie);
    const final: any[] = [];
    let type: StreamType;
    if (
        info.LiveStreamData.isLive === true &&
        info.LiveStreamData.hlsManifestUrl !== null &&
        info.video_details.durationInSec === '0'
    ) {
        return new LiveStreaming(
            info.LiveStreamData.dashManifestUrl,
            info.format[info.format.length - 1].targetDurationSec,
            info.video_details.url
        );
    }

    const audioFormat = parseAudioFormats(info.format);
    const opusFormats = filterFormat(audioFormat, 'opus');

    if (opusFormats.length === 0) {
        type = StreamType.Arbitrary;
        if (audioFormat.length === 0) {
            final.push(info.format[info.format.length - 1]);
        } else {
            final.push(audioFormat[audioFormat.length - 1]);
        }
    } else {
        type = StreamType.WebmOpus;
        final.push(opusFormats[opusFormats.length - 1]);
    }

    return new Stream(
        final[0].url,
        type,
        info.video_details.durationInSec,
        Number(final[0].contentLength),
        info.video_details.url,
        cookie as string
    );
}

export async function stream_from_info(info: InfoData, cookie?: string): Promise<Stream | LiveStreaming> {
    const final: any[] = [];
    let type: StreamType;
    if (
        info.LiveStreamData.isLive === true &&
        info.LiveStreamData.hlsManifestUrl !== null &&
        info.video_details.durationInSec === '0'
    ) {
        return new LiveStreaming(
            info.LiveStreamData.dashManifestUrl,
            info.format[info.format.length - 1].targetDurationSec,
            info.video_details.url
        );
    }

    const audioFormat = parseAudioFormats(info.format);
    const opusFormats = filterFormat(audioFormat, 'opus');

    if (opusFormats.length === 0) {
        type = StreamType.Arbitrary;
        if (audioFormat.length === 0) {
            final.push(info.format[info.format.length - 1]);
        } else {
            final.push(audioFormat[audioFormat.length - 1]);
        }
    } else {
        type = StreamType.WebmOpus;
        final.push(opusFormats[opusFormats.length - 1]);
    }

    return new Stream(
        final[0].url,
        type,
        info.video_details.durationInSec,
        Number(final[0].contentLength),
        info.video_details.url,
        cookie as string
    );
}

function filterFormat(formats: any[], codec: string) {
    const result: any[] = [];
    formats.forEach((format) => {
        if (format.codec === codec) result.push(format);
    });
    return result;
}