add Make First Audio in Lang Default

9 months ago · dd5022beb8
parent 60926b310f
commit dd5022beb8
1 changed files with 106 additions and 0 deletions
--- a/Tdarr_Plugin_gabehf_Make_First_Audio_In_Lang_Default.js
+++ b/Tdarr_Plugin_gabehf_Make_First_Audio_In_Lang_Default.js
@ -0,0 +1,106 @@
+/* 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:
+      'If no audio in the preferred language are made default already, this plugin sets the first audio stream in that language as default. \\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 convert = false;
+    let defaults = [] // list of tracks currently set as default
+    
+    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 (!convert && file.ffProbeData.streams[i].disposition.default === 0 && languages.includes(file.ffProbeData.streams[i].tags.language)) {
+                convert = true;
+                ffmpegCommandInsert += `-disposition:a:${audioIdx} default `;
+                response.infoLog += `☒Audio stream 0:a:${audioIdx} is the first track in preferred language and not currently default; setting as default. \n`;
+            } else if (file.ffProbeData.streams[i].disposition.default === 1 && languages.includes(file.ffProbeData.streams[i].tags.language)) {
+                // if any audio with the preferred language is marked default, no action is needed.
+                convert = false;
+                response.infoLog = '';
+                break;
+            }
+          audioIdx += 1;
+        }
+        } catch (err) {
+            // Error
+        }
+    }
+
+    if (convert) {
+        // remove previous default(s)
+        for (let i = 0; i < defaults.length; i++) {
+            if (defaults[i]) {
+                ffmpegCommandInsert += `-disposition:a:${i} 0 `;
+                response.infoLog += `☒Audio stream 0:a:${i} was default but not in preferred language; 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;
+