commit
02c4cf6f37
@ -0,0 +1,106 @@
|
||||
/* eslint max-len: 0, no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
|
||||
const details = () => ({
|
||||
id: 'Tdarr_Plugin_gabehf_Make_First_Subtitle_In_Lang_Default',
|
||||
Stage: 'Pre-processing',
|
||||
Name: 'Make First Subtitle In The Preferred Language Default',
|
||||
Type: 'Subtitle',
|
||||
Operation: 'Transcode',
|
||||
Description:
|
||||
'If no subtitles in the preferred language are made default already, this plugin sets the first subtitle stream in that language as default. \\n',
|
||||
Version: '1.0',
|
||||
Tags: 'pre-processing,ffmpeg,subtitle only,configurable',
|
||||
Inputs: [{
|
||||
name: 'preferred_language',
|
||||
type: 'string',
|
||||
defaultValue: 'eng,en',
|
||||
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 = ["eng", "en"]; //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 subtitleIdx = 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() === 'subtitle') {
|
||||
defaults[subtitleIdx] = 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:s:${subtitleIdx} default `;
|
||||
response.infoLog += `☒Subtitle stream 0:s:${subtitleIdx} 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 subtitle with the preferred language is marked default, no action is needed.
|
||||
convert = false;
|
||||
response.infoLog = '';
|
||||
break;
|
||||
}
|
||||
subtitleIdx += 1;
|
||||
}
|
||||
} catch (err) {
|
||||
// Error
|
||||
}
|
||||
}
|
||||
|
||||
if (convert) {
|
||||
// remove previous default(s)
|
||||
for (let i = 0; i < defaults.length; i++) {
|
||||
if (defaults[i]) {
|
||||
ffmpegCommandInsert += `-disposition:s:${i} 0 `;
|
||||
response.infoLog += `☒Subtitle stream 0:s:${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 subtitle tracks that require modification.\n";
|
||||
}
|
||||
return response;
|
||||
};
|
||||
module.exports.details = details;
|
||||
module.exports.plugin = plugin;
|
||||
|
||||
@ -0,0 +1,84 @@
|
||||
// tdarrSkipTest
|
||||
/* eslint max-len: 0, no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
|
||||
const details = () => ({
|
||||
id: 'Tdarr_Plugin_gabehf_Set_First_Audio_Stream_Default',
|
||||
Stage: 'Pre-processing',
|
||||
Name: 'Set First Audio Stream As Default',
|
||||
Type: 'Audio',
|
||||
Operation: 'Transcode',
|
||||
Description:
|
||||
'This plugin sets the first audio stream as the default. \n\n',
|
||||
Version: '1.0',
|
||||
Tags: 'audio only,ffmpeg',
|
||||
Inputs: [],
|
||||
});
|
||||
|
||||
// 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);
|
||||
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 defaultHasBeenSet = false;
|
||||
|
||||
// Go through each stream in the file.
|
||||
for (let i = 0; i < file.ffProbeData.streams.length; i++) {
|
||||
// Catch error here incase the metadata is completely missing.
|
||||
try {
|
||||
// Check if stream is audio
|
||||
if (file.ffProbeData.streams[i].codec_type.toLowerCase() === 'audio') {
|
||||
// Check if the audio stream is the first AND it is not default
|
||||
if (audioIdx === 0 && file.ffProbeData.streams[i].disposition.default !== 1) {
|
||||
ffmpegCommandInsert += `-disposition:a:${audioIdx} default `;
|
||||
response.infoLog += `First audio stream 0:a:${audioIdx} is not default; setting as default. \n`;
|
||||
defaultHasBeenSet = true;
|
||||
convert = true;
|
||||
// else if it is not the first stream AND we have changed the default track AND this is the default track
|
||||
} else if (defaultHasBeenSet && audioIdx !== 0 && file.ffProbeData.streams[i].disposition.default === 1) {
|
||||
ffmpegCommandInsert += `-disposition:a:${audioIdx} 0 `;
|
||||
response.infoLog += `Removing audio stream 0:a:${audioIdx} as default. \n`;
|
||||
convert = true; // should already be set by now but whatever
|
||||
}
|
||||
audioIdx += 1;
|
||||
}
|
||||
} catch (err) {
|
||||
// Error
|
||||
}
|
||||
}
|
||||
|
||||
// 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;
|
||||
@ -0,0 +1,234 @@
|
||||
/* eslint max-len: 0, no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
|
||||
const details = () => ({
|
||||
id: 'Tdarr_Plugin_gabehf_Set_Subtitle_Dispositions_By_Title',
|
||||
Stage: 'Pre-processing',
|
||||
Name: 'Set Subtitle Dispositions By Title',
|
||||
Type: 'Subtitle',
|
||||
Operation: 'Transcode',
|
||||
Description:
|
||||
'This plugin sets the disposition of specified subtitle tracks based on track titles. \n\n',
|
||||
Version: '1.0',
|
||||
Tags: 'pre-processing,ffmpeg,subtitle only,configurable',
|
||||
Inputs: [{
|
||||
name: 'titles_to_set_default',
|
||||
type: 'string',
|
||||
defaultValue: '',
|
||||
inputUI: {
|
||||
type: 'text',
|
||||
},
|
||||
tooltip:
|
||||
'Specify titles to set as default, separated by a comma. If a track contains any of those titles, it will be made default. The first matching track will be made default, and any other default tracks will have their dispositions removed.',
|
||||
},
|
||||
{
|
||||
name: 'titles_to_set_forced',
|
||||
type: 'string',
|
||||
defaultValue: '',
|
||||
inputUI: {
|
||||
type: 'text',
|
||||
},
|
||||
tooltip:
|
||||
'Specify titles to set as forced, separated by a comma. If a track contains any of those titles, it will be made forced. The first matching track will be made forced, and any other forced tracks will have their dispositions removed.',
|
||||
},
|
||||
{
|
||||
name: 'titles_to_remove_disposition',
|
||||
type: 'string',
|
||||
defaultValue: '',
|
||||
inputUI: {
|
||||
type: 'text',
|
||||
},
|
||||
tooltip:
|
||||
'Specify titles to remove dispositions from, separated by a comma. If a track contains any of those titles, its disposition will be set to 0 (none).',
|
||||
}],
|
||||
});
|
||||
|
||||
// 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);
|
||||
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 subtitleIdx = 0;
|
||||
let convert = false;
|
||||
let forcedHasBeenSet = false;
|
||||
let defaultHasBeenSet = false;
|
||||
let streamAltered = false;
|
||||
let streamDisposition = {};
|
||||
|
||||
const titles_to_set_forced = inputs.titles_to_set_forced === undefined || inputs.titles_to_set_forced === '' ? [] : inputs.titles_to_set_forced.split(',').map((s) => s.toLowerCase());
|
||||
const titles_to_set_default = inputs.titles_to_set_default === undefined || inputs.titles_to_set_default === '' ? [] : inputs.titles_to_set_default.split(',').map((s) => s.toLowerCase());
|
||||
const titles_to_remove_disposition = inputs.titles_to_remove_disposition === undefined || inputs.titles_to_remove_disposition === '' ? [] : inputs.titles_to_remove_disposition.split(',').map((s) => s.toLowerCase());
|
||||
|
||||
// TODO: Make it more clever and able to ensure all other subtitle tracks have no disposition if both default and forced tracks were changed
|
||||
// 1: English (Default)
|
||||
// 2: Sing/Song (Forced)
|
||||
// 3: Dialogue
|
||||
// 4: Signs/Songs
|
||||
|
||||
// TODO: Make it create a table of current tracks and dispoisitons, then a table on how it should be, then use the difference in the tables to build the ffmpeg command.
|
||||
// TODO: (Maybe for another all-in-one "anime subtitle fixer (sub)") If there is only one English subtitle track, and it is not set default, make it default.
|
||||
|
||||
// Build the track tables
|
||||
// Desired track dispositions;
|
||||
const desiredDispositions = [];
|
||||
// Current track dispositions
|
||||
const currentDispositions = [];
|
||||
for (let i = 0; i < file.ffProbeData.streams.length; i++) {
|
||||
try {
|
||||
if (file.ffProbeData.streams[i].codec_type.toLowerCase() === 'subtitle') {
|
||||
streamAltered = false;
|
||||
streamDisposition = file.ffProbeData.streams[i].disposition;
|
||||
|
||||
if (streamDisposition.default === 1 && streamDisposition.forced === 1) {
|
||||
currentDispositions[subtitleIdx] = 'default+forced';
|
||||
} else if (streamDisposition.default === 1) {
|
||||
currentDispositions[subtitleIdx] = 'default';
|
||||
} else if (streamDisposition.forced === 1) {
|
||||
currentDispositions[subtitleIdx] = 'forced';
|
||||
} else {
|
||||
currentDispositions[subtitleIdx] = '0';
|
||||
}
|
||||
|
||||
if (titles_to_set_default.some((s) => file.ffProbeData.streams[i].tags.title.toLowerCase().includes(s))) {
|
||||
if (defaultHasBeenSet) {
|
||||
desiredDispositions[subtitleIdx] = '0';
|
||||
} else {
|
||||
if (currentDispositions[subtitleIdx] === 'forced' && !forcedHasBeenSet) {
|
||||
desiredDispositions[subtitleIdx] = 'default+forced';
|
||||
} else {
|
||||
desiredDispositions[subtitleIdx] = 'default';
|
||||
}
|
||||
// ensure any previous default will have its disposition removed
|
||||
// note this is only being done when we set a new default so that if the user
|
||||
// does not want to change defaults, this does not get triggered
|
||||
for (let j = 0; j < subtitleIdx; j++) {
|
||||
if (desiredDispositions[j] === 'default') {
|
||||
desiredDispositions[j] = '0';
|
||||
} else if (desiredDispositions[j] === 'default+forced') {
|
||||
desiredDispositions[j] = 'forced';
|
||||
}
|
||||
}
|
||||
}
|
||||
defaultHasBeenSet = true;
|
||||
streamAltered = true;
|
||||
}
|
||||
if (titles_to_set_forced.some((s) => file.ffProbeData.streams[i].tags.title.toLowerCase().includes(s))) {
|
||||
if (forcedHasBeenSet) {
|
||||
desiredDispositions[subtitleIdx] = '0';
|
||||
} else {
|
||||
if (desiredDispositions[subtitleIdx] === 'default' || (currentDispositions[subtitleIdx] === 'default' && !defaultHasBeenSet)) {
|
||||
desiredDispositions[subtitleIdx] = 'default+forced';
|
||||
} else {
|
||||
desiredDispositions[subtitleIdx] = 'forced';
|
||||
}
|
||||
// ensure any previous forced will have its disposition removed
|
||||
// note this is only being done when we set a new forced so that if the user
|
||||
// does not want to change forced, this does not get triggered
|
||||
for (let j = 0; j < subtitleIdx; j++) {
|
||||
if (desiredDispositions[j] === 'forced') {
|
||||
desiredDispositions[j] = '0';
|
||||
} else if (desiredDispositions[j] === 'default+forced') {
|
||||
desiredDispositions[j] = 'default';
|
||||
}
|
||||
}
|
||||
}
|
||||
forcedHasBeenSet = true;
|
||||
streamAltered = true;
|
||||
}
|
||||
if ((defaultHasBeenSet && forcedHasBeenSet && !streamAltered) || titles_to_remove_disposition.some((s) => file.ffProbeData.streams[i].tags.title.toLowerCase().includes(s))) {
|
||||
desiredDispositions[subtitleIdx] = '0';
|
||||
streamAltered = true;
|
||||
}
|
||||
if (!streamAltered && currentDispositions[subtitleIdx] === 'default+forced') {
|
||||
if (defaultHasBeenSet) {
|
||||
desiredDispositions[subtitleIdx] = 'forced';
|
||||
streamAltered = true;
|
||||
} else if (forcedHasBeenSet) {
|
||||
desiredDispositions[subtitleIdx] = 'default';
|
||||
streamAltered = true;
|
||||
}
|
||||
}
|
||||
if (!streamAltered) {
|
||||
desiredDispositions[subtitleIdx] = currentDispositions[subtitleIdx];
|
||||
}
|
||||
subtitleIdx += 1;
|
||||
}
|
||||
} catch (err) {
|
||||
// Error
|
||||
}
|
||||
}
|
||||
|
||||
// Generate ffmpeg commands for the differences in the dispositions
|
||||
for (let i = 0; i < currentDispositions.length; i++) {
|
||||
if (currentDispositions[i] !== desiredDispositions[i]) {
|
||||
convert = true;
|
||||
switch (desiredDispositions[i]) {
|
||||
case 'default':
|
||||
ffmpegCommandInsert += `-disposition:s:${i} default `;
|
||||
if (currentDispositions[i] === 'default+forced') {
|
||||
response.infoLog += `☒Subtitle stream 0:s:${i} has overlapping disposition 'forced'; setting as default. \n`;
|
||||
} else {
|
||||
response.infoLog += `☒Subtitle stream 0:s:${i} detected as being [${titles_to_set_default}]; setting as default. \n`;
|
||||
}
|
||||
break;
|
||||
case 'forced':
|
||||
ffmpegCommandInsert += `-disposition:s:${i} forced `;
|
||||
if (currentDispositions[i] === 'default+forced') {
|
||||
response.infoLog += `☒Subtitle stream 0:s:${i} has overlapping disposition 'default'; setting as forced. \n`;
|
||||
} else {
|
||||
response.infoLog += `☒Subtitle stream 0:s:${i} detected as being [${titles_to_set_forced}]; setting as forced. \n`;
|
||||
}
|
||||
break;
|
||||
case 'default+forced':
|
||||
ffmpegCommandInsert += `-disposition:s:${i} default+forced `;
|
||||
if (currentDispositions[i] === 'default') {
|
||||
response.infoLog += `☒Subtitle stream 0:s:${i} detected as being [${titles_to_set_forced}] and is already default; setting as default+forced. \n`;
|
||||
} else if (currentDispositions[i] === 'forced') {
|
||||
response.infoLog += `☒Subtitle stream 0:s:${i} detected as being [${titles_to_set_default}] and is already forced; setting as default+forced. \n`;
|
||||
} else {
|
||||
response.infoLog += `☒Subtitle stream 0:s:${i} detected as being both [${titles_to_set_default}] and [${titles_to_set_forced}]; setting as default+forced. \n`;
|
||||
}
|
||||
break;
|
||||
case '0':
|
||||
ffmpegCommandInsert += `-disposition:s:${i} 0 `;
|
||||
response.infoLog += `☒Subtitle stream 0:s:${i} detected as being [${titles_to_remove_disposition}] or has disposition overlapping with new default/forced; removing disposition. \n`;
|
||||
break;
|
||||
default:
|
||||
// should not get here, error
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 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 subtitle tracks that require modification.\n";
|
||||
}
|
||||
return response;
|
||||
};
|
||||
module.exports.details = details;
|
||||
module.exports.plugin = plugin;
|
||||
Loading…
Reference in new issue