Tdarr_Plugins/Community/Tdarr_Plugin_MC93_Migz6OrderStreams.js

/* eslint-disable */
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.2",
    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 9999`;
    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;