Tdarr-Local/Tdarr_Plugin_gabehf_Make_First_Audio_In_Lang_Default.js

/* eslint max-len: 0, no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
const details = () => ({
    id: 'Tdarr_Plugin_gabehf_Make_First_Audio_In_Lang_Default',
    Stage: 'Pre-processing',
    Name: 'Make First Audio In The Preferred Language Default',
    Type: 'Audio',
    Operation: 'Transcode',
    Description:
      'Sets the first audio stream in preferred language as default, removing all other defaults if processed. \\n',
    Version: '1.0',
    Tags: 'pre-processing,ffmpeg,audio only,configurable',
    Inputs: [{
      name: 'preferred_language',
      type: 'string',
      defaultValue: 'jpn',
      inputUI: {
        type: 'text',
      },
      tooltip:
        'Your preferred language code(s) in ISO 639-2 language scheme, separated by a comma. Only the first track that matches any of these language codes will be made default. \\n Default: eng,en',
    }],
  });
  
  // eslint-disable-next-line @typescript-eslint/no-unused-vars
  const plugin = (file, librarySettings, inputs, otherArguments) => {
    const lib = require('../methods/lib')();
    // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign
    inputs = lib.loadDefaultValues(inputs, details);
    var languages = []
    if (inputs.preferred_language == "") {
        languages = ["jpn"]; //these languages should be kept, named according to ISO 639-2 language scheme
    } else {
        languages = inputs.preferred_language.toLowerCase().split(","); //these languages should be kept, named according to ISO 639-2 language scheme
    }

    const response = {
      processFile: false,
      preset: '',
      container: `.${file.container}`,
      handBrakeMode: false,
      FFmpegMode: true,
      reQueueAfter: false,
      infoLog: '',
    };
  
    // Check if file is a video. If it isn't then exit plugin.
    if (file.fileMedium !== 'video') {
      // eslint-disable-next-line no-console
      console.log('File is not video');
      response.infoLog += '☒File is not video \n';
      response.processFile = false;
      return response;
    }
  
    // Set up required variables.
    let ffmpegCommandInsert = '';
    let audioIdx = 0;
    let firstAudioInLangIdx = -1;
    let convert = false;
    let defaults = [] // list of tracks currently set as default

    // 0 jpn (default), 1 eng (default)
    // [true, true]
    // firstAudioInLangIdx = 0
    
    for (let i = 0; i < file.ffProbeData.streams.length; i++) {
      try {
        if (file.ffProbeData.streams[i].codec_type.toLowerCase() === 'audio') {
          defaults[audioIdx] = file.ffProbeData.streams[i].disposition.default === 1
          // if we find an audio in the lang
          if (languages.includes(file.ffProbeData.streams[i].tags.language)) {
            // mark the id of the first track in lang
            if (firstAudioInLangIdx === -1) {
              response.infoLog += `☒Audio stream 0:a:${audioIdx} set as firstAudioInLangIdx \n`
              firstAudioInLangIdx = audioIdx;
            }
          }
          audioIdx += 1;
        }
        } catch (err) {
            // Error
        }
    }

    // remove overlapping default(s)
    for (let i = 0; i < defaults.length; i++) {
      if (firstAudioInLangIdx === -1) {
        // we never found an audio track in lang, just break
        response.infoLog += `☒No audio stream in preferred lang was found \n`
        break;
      } else if (firstAudioInLangIdx === i && !defaults[i]) {
        convert = true;
        ffmpegCommandInsert += `-disposition:a:${i} default `;
        response.infoLog += `☒Audio stream 0:a:${i} is the first track in preferred language and not currently default; setting as default. \n`;
      } else if (firstAudioInLangIdx === i && defaults[i]) {
        response.infoLog += `☒Audio stream 0:a:${i} is the first track in preferred language and alrady default. \n`;
      } else if (defaults[i]) {
        convert = true;
        ffmpegCommandInsert += `-disposition:a:${i} 0 `;
        response.infoLog += `☒Audio stream 0:a:${i} is overlapping with new default audio track; removing disposition. \n`;
      }
    }
  
    // Convert file if convert variable is set to true.
    if (convert === true) {
      response.processFile = true;
      response.preset = `, -map 0 ${ffmpegCommandInsert}-c copy -max_muxing_queue_size 9999`;
      response.container = `.${file.container}`;
      response.reQueueAfter = true;
    } else {
      response.processFile = false;
      response.infoLog += "☑File doesn't contain audio tracks that require modification.\n";
    }
    return response;
  };
  module.exports.details = details;
  module.exports.plugin = plugin;