multi-downloader-nx_mirror/modules/merger.js

const iso639 = require('iso-639');
const argv = require('../funi').argv;

/**
 * @param {Array<object>} videoAndAudio 
 * @param {Array<object>} onlyVid 
 * @param {Array<object>} onlyAuido
 * @param {Array<object>} subtitles 
 * @param {string} output
 * @returns {string}
 */
const buildCommandFFmpeg = (videoAndAudio, onlyVid, onlyAuido, subtitles, output) => {
    let args = [];
    let metaData = [];

    let index = 0;
    let hasVideo = false;
    for (let vid of videoAndAudio) {
        args.push(`-i "${vid.path}"`)
        if (!hasVideo) {
            metaData.push(`-map ${index}`)
            metaData.push(`-metadata:s:a:${index} language=${getLanguageCode(vid.lang, vid.lang)}`)
            metaData.push(`-metadata:s:v:${index} title="[Funimation]"`)
            hasVideo = true
        } else {
            metaData.push(`-map ${index}:a`)
            metaData.push(`-metadata:s:a:${index} language=${getLanguageCode(vid.lang, vid.lang)}`)
        }
        index++;
    }

    for (let vid of onlyVid) {
        if (!hasVideo) {
            args.push(`-i "${vid.path}"`)
            metaData.push(`-map ${index}`)
            metaData.push(`-metadata:s:a:${index} language=${getLanguageCode(vid.lang, vid.lang)}`)
            metaData.push(`-metadata:s:v:${index} title="[Funimation]"`)
            hasVideo = true
            index++;
        }
    }

    for (let aud of onlyAuido) {
        args.push(`-i "${aud.path}"`)
        metaData.push(`-map ${index}`)
        metaData.push(`-metadata:s:a:${index} language=${getLanguageCode(aud.lang, aud.lang)}`)
        index++;
    }

    for (let index in subtitles) {
        let sub = subtitles[index];
        args.push(`-i "${sub.file}"`);
    }

    args.push(...subtitles.map((_, subIndex) => `-map ${subIndex + index}`));
    args.push(...metaData)
    args.push(
        '-c:v copy',
        '-c:a copy',
        '-c:s mov_text',
        '-c:s ass'
    );
    args.push(...subtitles.map((sub, index) => `-metadata:s:${index + 2} language=${getLanguageCode(sub.language)}`));
    args.push(`"${output}"`);
    return args.join(' ');
};

/**
 * @param {string} videoFile 
 * @param {object} audioFile 
 * @param {Array<object>} subtitles 
 * @returns {string}
 */
const buildCommandMkvMerge = (videoFile, audioSettings, subtitles, output) => {
    let args = [];
    args.push(`-o "${output}"`);
    args.push(
        '--no-date',
        '--disable-track-statistics-tags',
        '--engage no_variable_data',
        '--track-name 0:[Funimation]'
    );

    if (audioSettings.uri) {
        args.push(
            '--video-tracks 0',
            '--no-audio'
        );
        args.push(`"${videoFile}"`);
        args.push(`--language 0:${getLanguageCode(audioSettings.language, argv.todo ? 'jpn' : 'eng')}`);
        args.push(
            '--no-video',
            '--audio-tracks 0'
        );
        args.push(`"${audioSettings.uri}"`);
    } else{
        args.push(`--language 1:${argv.todo ? 'jpn' : 'eng'}`);
        args.push(
            '--video-tracks 0',
            '--audio-tracks 1'
        );
        args.push(`"${videoFile}"`);
    }

    if(subtitles.length > 0){
        for (let index in subtitles) {
            let subObj = subtitles[index];
            args.push('--language',`${/*parseInt(index) + (audioSettings.uri ? 2 : 1)*/0}:${getLanguageCode(subObj.language)}`);
            args.push(`"${subObj.file}"`);
        }
    } else {
        args.push(
            '--no-subtitles',
            '--no-attachments'
        );
    }

    return args.join(' ');
};

const getLanguageCode = (from, _default = 'eng') => {
    for (let lang in iso639.iso_639_2) {
        let langObj = iso639.iso_639_2[lang];
        if (Object.prototype.hasOwnProperty.call(langObj, '639-1') && langObj['639-1'] === from) {
            return langObj['639-2'];
        }
    }
    return _default;
};

module.exports = {
    buildCommandFFmpeg,
    getLanguageCode,
    buildCommandMkvMerge
};