destreamer/src/VideoUtils.ts

import { ApiClient } from './ApiClient';
import { promptUser } from './CommandLineParser';
import { logger } from './Logger';
import { Video, Session } from './Types';

import { AxiosResponse } from 'axios';
import fs from 'fs';
import { parse as parseDuration, Duration } from 'iso8601-duration';
import path from 'path';
import sanitizeWindowsName from 'sanitize-filename';

function publishedDateToString(date: string): string {
    const dateJs: Date = new Date(date);
    const day: string = dateJs.getDate().toString().padStart(2, '0');
    const month: string = (dateJs.getMonth() + 1).toString(10).padStart(2, '0');

    return `${dateJs.getFullYear()}-${month}-${day}`;
}


function publishedTimeToString(date: string): string {
    const dateJs: Date = new Date(date);
    const hours: string = dateJs.getHours().toString();
    const minutes: string = dateJs.getMinutes().toString();
    const seconds: string = dateJs.getSeconds().toString();

    return `${hours}:${minutes}:${seconds}`;
}


function isoDurationToString(time: string): string {
    const duration: Duration = parseDuration(time);

    return `${duration.hours ?? '00'}:${duration.minutes ?? '00'}:${duration.seconds?.toFixed(0) ?? '00'}`;
}


function durationToTotalChunks(duration: string): number {
    const durationObj: any = parseDuration(duration);
    const hrs: number = durationObj.hours ?? 0;
    const mins: number = durationObj.minutes ?? 0;
    const secs: number = Math.ceil(durationObj.seconds ?? 0);

    return (hrs * 60) + mins + (secs / 60);
}


export async function getVideoInfo(videoGuid: string, session: Session, subtitles?: boolean): Promise<Video> {
    // template elements
    let title: string;
    let duration: string;
    let publishDate: string;
    let publishTime: string;
    let author: string;
    let authorEmail: string;
    let uniqueId: string;
    // final video path (here for consistency with typedef)
    const outPath = '';
    // ffmpeg magic (abstraction of FFmpeg timemark)
    let totalChunks: number;
    // various sources
    let playbackUrl: string;
    let posterImageUrl: string;
    let captionsUrl: string | undefined;

    const apiClient: ApiClient = ApiClient.getInstance(session);

    let response: AxiosResponse<any> | undefined =
        await apiClient.callApi('videos/' + videoGuid + '?$expand=creator', 'get');

    title = sanitizeWindowsName(response?.data['name']);

    duration = isoDurationToString(response?.data.media['duration']);

    publishDate = publishedDateToString(response?.data['publishedDate']);

    publishTime = publishedTimeToString(response?.data['publishedDate']);

    author = response?.data['creator'].name;

    authorEmail = response?.data['creator'].mail;

    uniqueId = '#' + videoGuid.split('-')[0];

    totalChunks = durationToTotalChunks(response?.data.media['duration']);

    playbackUrl = response?.data['playbackUrls']
        .filter((item: { [x: string]: string; }) =>
            item['mimeType'] == 'application/vnd.apple.mpegurl')
        .map((item: { [x: string]: string }) => {
            return item['playbackUrl'];
        })[0];

    posterImageUrl = response?.data['posterImage']['medium']['url'];

    if (subtitles) {
        let captions: AxiosResponse<any> | undefined = await apiClient.callApi(`videos/${videoGuid}/texttracks`, 'get');

        if (!captions?.data.value.length) {
            captionsUrl = undefined;
        }
        else if (captions?.data.value.length === 1) {
            logger.info(`Found subtitles for ${title}. \n`);
            captionsUrl = captions?.data.value.pop().url;
        }
        else {
            const index: number = promptUser(captions.data.value.map((item: { language: string; autoGenerated: string; }) => {
                return `[${item.language}] autogenerated: ${item.autoGenerated}`;
            }));
            captionsUrl = captions.data.value[index].url;
        }
    }

    return {
        title: title,
        duration: duration,
        publishDate: publishDate,
        publishTime: publishTime,
        author: author,
        authorEmail: authorEmail,
        uniqueId: uniqueId,
        outPath: outPath,
        totalChunks: totalChunks,
        playbackUrl: playbackUrl,
        posterImageUrl: posterImageUrl,
        captionsUrl: captionsUrl
    };
}


export function createUniquePath(video: Video, outDir: string, template: string, format: string, skip?: boolean): Video {

    let title: string = template;
    let finalTitle: string;
    const elementRegEx = RegExp(/{(.*?)}/g);
    let match = elementRegEx.exec(template);

    while (match) {
        let value = video[match[1] as keyof Video] as string;
        title = title.replace(match[0], value);
        match = elementRegEx.exec(template);
    }

    let i = 0;
    finalTitle = title;

    while (!skip && fs.existsSync(path.join(outDir, finalTitle + '.' + format))) {
        finalTitle = `${title}.${++i}`;
    }

    video.outPath = path.join(outDir, finalTitle + '.' + format);

    return video;
}