Tdarr_Plugins/Community/Tdarr_Plugin_MC93_Migz6OrderStreams.js

function details() {
    return {
        id: "Tdarr_Plugin_MC93_Migz6OrderStreams",
        Stage: "Pre-processing",
        Name: "Migz-Order Streams",
        Type: "Streams",
        Operation: "Order",
        Description: `Orders streams into Video first, then Audio (2ch, 6ch, 8ch) and finally Subtitles. \n\n`,
        Version: "1.1",
        Link: "https://github.com/HaveAGitGat/Tdarr_Plugins/blob/master/Community/Tdarr_Plugin_MC93_Migz6OrderStreams.js",
        Tags: 'pre-processing,ffmpeg,'
    }
}

function plugin(file) {
    var response = {
        processFile: false,
        preset: '',
        container: '.' + file.container,
        handBrakeMode: false,
        FFmpegMode: true,
        infoLog: '',
    }

    // Set up required variables.
    var ffmpegCommandInsert = ''
    var videoIdx = 0
    var audioIdx = 0
    var audio2Idx = 0
    var audio6Idx = 0
    var audio8Idx = 0
    var subtitleIdx = 0
    var convert = false

    // Go through each stream in the file.
    for (var i = 0; i < file.ffProbeData.streams.length; i++) {
        try {
			// Check if stream is video.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "video") {
				// Check if audioIdx or subtitleIdx do NOT equal 0, if they do then it means a audio or subtitle track has already appeared before the video track so file needs to be organized.
                if (audioIdx != "0" || subtitleIdx != "0") {
                    convert = true
                    response.infoLog += "☒ Video not first. \n"
                }
				// Increment videoIdx.
                videoIdx++
            }

            // Check if stream is audio.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "audio") {
				// Check if subtitleIdx does NOT equal 0, if it does then it means a subtitle track has already appeared before an audio track so file needs to be organized.
                if (subtitleIdx != "0") {
                    convert = true
                    response.infoLog += "☒ Audio not second. \n"
                }
				// Increment audioIdx.
                audioIdx++
				
				// Check if audio track is 2 channel.
                if (file.ffProbeData.streams[i].channels == "2") {
					// Check if audio6Idx or audio8Idx do NOT equal 0, if they do then it means a 6 channel or 8 channel audio track has already appeared before the 2 channel audio track so file needs to be organized.
                    if (audio6Idx != "0" || audio8Idx != "0") {
                        convert = true
                        response.infoLog += "☒ Audio 2ch not first. \n"
                    }
					// Increment audio2Idx.
                    audio2Idx++
                }
				// Check if audio track is 6 channel.
                if (file.ffProbeData.streams[i].channels == "6") {
					// Check if audio8Idx does NOT equal 0, if it does then it means a 8 channel audio track has already appeared before the 6 channel audio track so file needs to be organized.
                    if (audio8Idx != "0") {
                        convert = true
                        response.infoLog += "☒ Audio 6ch not second. \n"
                    }
					// Increment audio6Idx.
                    audio6Idx++
                }
				
				// Check if audio track is 8 channel.
                if (file.ffProbeData.streams[i].channels == "8") {
					// Increment audio8Idx.
                    audio8Idx++
                }
            }

            // Check if stream is subtitle.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "subtitle") {
				// Increment subtitleIdx
                subtitleIdx++
            }
        } catch (err) {}
    }

    // Go through each stream in the file.
    for (var i = 0; i < file.ffProbeData.streams.length; i++) {
        try {
			// Check if stream is video AND is not a mjpeg.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "video" && file.ffProbeData.streams[i].codec_name.toLowerCase() != "mjpeg") {
                ffmpegCommandInsert += `-map 0:${i} `
            }
        } catch (err) {}
    }

    // Go through each stream in the file.
    for (var i = 0; i < file.ffProbeData.streams.length; i++) {
        try {
			// Check if stream is audio AND 2 channel.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "audio" && file.ffProbeData.streams[i].channels == "2") {
                ffmpegCommandInsert += `-map 0:${i} `
            }
        } catch (err) {}
    }

    // Go through each stream in the file.
    for (var i = 0; i < file.ffProbeData.streams.length; i++) {
        try {
			// Check if stream is audio AND 6 channel.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "audio" && file.ffProbeData.streams[i].channels == "6") {
                ffmpegCommandInsert += `-map 0:${i} `
            }
        } catch (err) {}
    }

    // Go through each stream in the file.
    for (var i = 0; i < file.ffProbeData.streams.length; i++) {
        try {
			// Check if stream is audio AND 8 channel.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "audio" && file.ffProbeData.streams[i].channels == "8") {
                ffmpegCommandInsert += `-map 0:${i} `
            }
        } catch (err) {}
    }

    // Go through each stream in the file.
    for (var i = 0; i < file.ffProbeData.streams.length; i++) {
        try {
			// Check if stream is audio AND not 2, 6 or 8 channel.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "audio" && file.ffProbeData.streams[i].channels != "2" && file.ffProbeData.streams[i].channels != "6" && file.ffProbeData.streams[i].channels != "8") {
                ffmpegCommandInsert += `-map 0:${i} `
            }
        } catch (err) {}
    }

    // Go through each stream in the file.
    for (var i = 0; i < file.ffProbeData.streams.length; i++) {
        try {
			// Check if stream is subtitle.
            if (file.ffProbeData.streams[i].codec_type.toLowerCase() == "subtitle") {
                ffmpegCommandInsert += `-map 0:${i} `
            }
        } catch (err) {}
    }

    // Convert file if convert variable is set to true.
    if (convert == true) {
        response.processFile = true;
        response.preset = `,${ffmpegCommandInsert} -c copy -max_muxing_queue_size 4096`
        response.reQueueAfter = true;
        response.infoLog += "☒ Streams are out of order, reorganizing streams. Video, Audio, Subtitles. \n"
    } else {
        response.infoLog += "☑ Streams are in expected order. \n "
        response.processFile = false;
    }
    return response

}
module.exports.details = details;
module.exports.plugin = plugin;