mirror of
https://github.com/gabehf/Tdarr_Plugins.git
synced 2026-03-17 03:06:27 -07:00
tidy
This commit is contained in:
HaveAGitGat
parent
d28272f6d4
commit
9c8ac98728
60 changed files with 5481 additions and 5980 deletions
|
|
@ -1,143 +1,154 @@
|
|||
const fs = require('fs');
|
||||
const execSync = require('child_process').execSync;
|
||||
const fs = require("fs");
|
||||
const execSync = require("child_process").execSync;
|
||||
|
||||
function details() {
|
||||
return {
|
||||
id: "Tdarr_Plugin_e5c3_CnT_Add_Subtitles",
|
||||
Stage: "Pre-processing",
|
||||
Name: "Add subtitles to MKV files",
|
||||
Type: "Video",
|
||||
Operation:"Remux",
|
||||
Description: `This plugin will check for subtitles, they should be named according to the ISO 639-2 language code.\nA subtitle could look like this: eng.srt\n If there are subtitles found they will be added with FFMPEG, if there are no subs of that language found.\n On first run node module iso-639-2 will be installed in the documents folder.\n Created by @control#0405`,
|
||||
Version: "1.3",
|
||||
Link: "https://github.com/HaveAGitGat/Tdarr_Plugins/blob/master/Community/Tdarr_Plugin_e5c3_CnT_Add_Subtitles.js",
|
||||
Tags:'pre-processing,ffmpeg,subtitle only,configurable',
|
||||
Inputs: [
|
||||
{
|
||||
name: 'install_packages',
|
||||
tooltip: `Please change this to "yes", it allows the plugin to install the required nodemodule. (iso-639-2) \\nExample:\\n yes`
|
||||
},
|
||||
{
|
||||
name: 'container',
|
||||
tooltip: `Enter the output container of the new file.\\n Default: .mkv\\nExample:\\n.mkv`
|
||||
},
|
||||
]
|
||||
}
|
||||
return {
|
||||
id: "Tdarr_Plugin_e5c3_CnT_Add_Subtitles",
|
||||
Stage: "Pre-processing",
|
||||
Name: "Add subtitles to MKV files",
|
||||
Type: "Video",
|
||||
Operation: "Remux",
|
||||
Description: `This plugin will check for subtitles, they should be named according to the ISO 639-2 language code.\nA subtitle could look like this: eng.srt\n If there are subtitles found they will be added with FFMPEG, if there are no subs of that language found.\n On first run node module iso-639-2 will be installed in the documents folder.\n Created by @control#0405`,
|
||||
Version: "1.3",
|
||||
Link:
|
||||
"https://github.com/HaveAGitGat/Tdarr_Plugins/blob/master/Community/Tdarr_Plugin_e5c3_CnT_Add_Subtitles.js",
|
||||
Tags: "pre-processing,ffmpeg,subtitle only,configurable",
|
||||
Inputs: [
|
||||
{
|
||||
name: "install_packages",
|
||||
tooltip: `Please change this to "yes", it allows the plugin to install the required nodemodule. (iso-639-2) \\nExample:\\n yes`,
|
||||
},
|
||||
{
|
||||
name: "container",
|
||||
tooltip: `Enter the output container of the new file.\\n Default: .mkv\\nExample:\\n.mkv`,
|
||||
},
|
||||
],
|
||||
};
|
||||
}
|
||||
|
||||
function plugin(file, librarySettings, inputs, otherArguments) {
|
||||
//default response
|
||||
var response = {
|
||||
processFile: false,
|
||||
preset: `,`,
|
||||
container: '.mkv',
|
||||
handBrakeMode: false,
|
||||
FFmpegMode: false,
|
||||
reQueueAfter: false,
|
||||
infoLog: `Searching new subtitles...\n`,
|
||||
}
|
||||
|
||||
if (inputs.container !== undefined) {
|
||||
response.container = inputs.container;
|
||||
console.log(`Changed container to: ` + inputs.container);
|
||||
}
|
||||
//default response
|
||||
var response = {
|
||||
processFile: false,
|
||||
preset: `,`,
|
||||
container: ".mkv",
|
||||
handBrakeMode: false,
|
||||
FFmpegMode: false,
|
||||
reQueueAfter: false,
|
||||
infoLog: `Searching new subtitles...\n`,
|
||||
};
|
||||
|
||||
if (inputs.install_packages == "yes") {
|
||||
if (!fs.existsSync(`${otherArguments.homePath}/Tdarr/node_modules/iso-639-2`)) {
|
||||
execSync(`cd ${otherArguments.homePath}/Tdarr \n npm install iso-639-2`);
|
||||
}
|
||||
if (inputs.container !== undefined) {
|
||||
response.container = inputs.container;
|
||||
console.log(`Changed container to: ` + inputs.container);
|
||||
}
|
||||
|
||||
if (inputs.install_packages == "yes") {
|
||||
if (
|
||||
!fs.existsSync(`${otherArguments.homePath}/Tdarr/node_modules/iso-639-2`)
|
||||
) {
|
||||
execSync(`cd ${otherArguments.homePath}/Tdarr \n npm install iso-639-2`);
|
||||
}
|
||||
} else {
|
||||
response.infoLog = `Please take a look at the input options\n A extra nodemodule is required.`;
|
||||
return response;
|
||||
}
|
||||
|
||||
if (fs.existsSync(`/home/Tdarr/Documents/Tdarr/node_modules/iso-639-2`)) {
|
||||
var iso6392 = require("/home/Tdarr/Documents/Tdarr/node_modules/iso-639-2");
|
||||
} else {
|
||||
response.infoLog += `Nodemodule iso-639-2 isn't installed!\nTry Again`;
|
||||
return response;
|
||||
}
|
||||
|
||||
var i = 0; //int for counting lang[position]
|
||||
var found_subtitle_stream = 0;
|
||||
var sub = 0; //becomes first subtitle stream
|
||||
var lang = iso6392; //languages to check against
|
||||
var path = file.meta.Directory; //path of media folder
|
||||
var exist = 0; //if the language exists should be added this becomes 1
|
||||
var new_subs = 0; //count the new subs
|
||||
var added_subs = 0; //counts the amount of subs that have been mapped
|
||||
var preset_import = "";
|
||||
var preset_meta = "";
|
||||
|
||||
//find first subtitle stream
|
||||
while (found_subtitle_stream == 0 && sub < file.ffProbeData.streams.length) {
|
||||
if (file.ffProbeData.streams[sub].codec_type.toLowerCase() == "subtitle") {
|
||||
found_subtitle_stream = 1;
|
||||
} else {
|
||||
response.infoLog = `Please take a look at the input options\n A extra nodemodule is required.`
|
||||
return response;
|
||||
sub++;
|
||||
}
|
||||
}
|
||||
|
||||
if (fs.existsSync(`/home/Tdarr/Documents/Tdarr/node_modules/iso-639-2`)) {
|
||||
var iso6392 = require('/home/Tdarr/Documents/Tdarr/node_modules/iso-639-2');
|
||||
} else {
|
||||
response.infoLog += `Nodemodule iso-639-2 isn't installed!\nTry Again`
|
||||
return response;
|
||||
}
|
||||
response.infoLog += `Path: ${path}\n`;
|
||||
for (i = 0; i < lang.length; i++) {
|
||||
//check if srt file exists in folder
|
||||
if (fs.existsSync(`${path}/${lang[i].iso6392B}.srt`)) {
|
||||
response.infoLog += `Found subtitle ${lang[i].name}.srt\n`;
|
||||
|
||||
var i = 0; //int for counting lang[position]
|
||||
var found_subtitle_stream = 0;
|
||||
var sub = 0; //becomes first subtitle stream
|
||||
var lang = iso6392; //languages to check against
|
||||
var path = file.meta.Directory; //path of media folder
|
||||
var exist = 0; //if the language exists should be added this becomes 1
|
||||
var new_subs = 0 //count the new subs
|
||||
var added_subs = 0; //counts the amount of subs that have been mapped
|
||||
var preset_import = '';
|
||||
var preset_meta = '';
|
||||
|
||||
//find first subtitle stream
|
||||
while (found_subtitle_stream == 0 && sub < file.ffProbeData.streams.length) {
|
||||
if (file.ffProbeData.streams[sub].codec_type.toLowerCase() == "subtitle") {
|
||||
found_subtitle_stream = 1;
|
||||
} else {
|
||||
sub++;
|
||||
if (found_subtitle_stream == 1) {
|
||||
//check if language already exists
|
||||
for (
|
||||
sub_stream = sub;
|
||||
sub_stream < file.ffProbeData.streams.length;
|
||||
sub_stream++
|
||||
) {
|
||||
//response.infoLog += `does ${lang[i].name} exist in stream ${sub_stream}?\n`
|
||||
if (file.ffProbeData.streams[sub_stream].tags.language) {
|
||||
if (
|
||||
file.ffProbeData.streams[
|
||||
sub_stream
|
||||
].tags.language.toLowerCase() == lang[i].iso6392B
|
||||
) {
|
||||
//response.infoLog += `YES\n`
|
||||
exist = 1;
|
||||
response.infoLog += `Language already exists in stream ${sub_stream}\n It will not be added\n`;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
response.infoLog += `Path: ${path}\n`;
|
||||
for (i = 0; i < lang.length; i++) {
|
||||
//check if srt file exists in folder
|
||||
if (fs.existsSync(`${path}/${lang[i].iso6392B}.srt`)) {
|
||||
response.infoLog += `Found subtitle ${lang[i].name}.srt\n`;
|
||||
|
||||
if (found_subtitle_stream == 1) {
|
||||
//check if language already exists
|
||||
for (sub_stream = sub; sub_stream < file.ffProbeData.streams.length; sub_stream++) {
|
||||
//response.infoLog += `does ${lang[i].name} exist in stream ${sub_stream}?\n`
|
||||
if (file.ffProbeData.streams[sub_stream].tags.language) {
|
||||
if (file.ffProbeData.streams[sub_stream].tags.language.toLowerCase() == lang[i].iso6392B) {
|
||||
//response.infoLog += `YES\n`
|
||||
exist = 1;
|
||||
response.infoLog += `Language already exists in stream ${sub_stream}\n It will not be added\n`;
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
exist = 0;
|
||||
}
|
||||
|
||||
//add if it hasn't found the language
|
||||
if (exist != 1) {
|
||||
preset_import += ` -sub_charenc "UTF-8" -f srt -i "${path}/${lang[i].iso6392B}.srt"`;
|
||||
preset_meta += ` -metadata:s:s:${new_subs} language=${lang[i].iso6392B}`;
|
||||
new_subs++;
|
||||
}
|
||||
}
|
||||
//else {
|
||||
// response.infoLog += `did not find sub ${lang[i].iso6392B}.srt\n`
|
||||
//}
|
||||
} else {
|
||||
exist = 0;
|
||||
}
|
||||
|
||||
//add if it hasn't found the language
|
||||
if (exist != 1) {
|
||||
preset_import += ` -sub_charenc "UTF-8" -f srt -i "${path}/${lang[i].iso6392B}.srt"`;
|
||||
preset_meta += ` -metadata:s:s:${new_subs} language=${lang[i].iso6392B}`;
|
||||
new_subs++;
|
||||
}
|
||||
}
|
||||
//else {
|
||||
// response.infoLog += `did not find sub ${lang[i].iso6392B}.srt\n`
|
||||
//}
|
||||
exist = 0;
|
||||
}
|
||||
|
||||
response.preset += ` ${preset_import}${preset_meta} -map 0:v -map 0:a`
|
||||
|
||||
//map new subs
|
||||
while (added_subs < new_subs) {
|
||||
added_subs++;
|
||||
response.preset += ` -map ${added_subs}:s`;
|
||||
response.preset += ` ${preset_import}${preset_meta} -map 0:v -map 0:a`;
|
||||
|
||||
//map new subs
|
||||
while (added_subs < new_subs) {
|
||||
added_subs++;
|
||||
response.preset += ` -map ${added_subs}:s`;
|
||||
}
|
||||
|
||||
//if new subs have been found they will be added
|
||||
if (new_subs > 0) {
|
||||
response.FFmpegMode = true;
|
||||
response.processFile = true;
|
||||
response.reQueueAfter = true;
|
||||
if (found_subtitle_stream == 1) {
|
||||
response.preset += ` -map 0:s `;
|
||||
}
|
||||
response.preset += ` -c copy`;
|
||||
response.infoLog += `${new_subs} new subs will be added\n`;
|
||||
} else {
|
||||
response.infoLog += `No new subtitle languages were found\n`;
|
||||
}
|
||||
|
||||
//if new subs have been found they will be added
|
||||
if (new_subs > 0) {
|
||||
response.FFmpegMode = true;
|
||||
response.processFile = true;
|
||||
response.reQueueAfter = true;
|
||||
if (found_subtitle_stream == 1) {
|
||||
response.preset += ` -map 0:s `;
|
||||
}
|
||||
response.preset += ` -c copy`;
|
||||
response.infoLog += `${new_subs} new subs will be added\n`;
|
||||
} else {
|
||||
response.infoLog += `No new subtitle languages were found\n`;
|
||||
}
|
||||
//response.infoLog += `The ffmpeg string is: ${response.preset}\n`
|
||||
|
||||
//response.infoLog += `The ffmpeg string is: ${response.preset}\n`
|
||||
|
||||
return response
|
||||
return response;
|
||||
}
|
||||
|
||||
module.exports.details = details;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue