Rick Meijer
5 years ago
parent
69afe2c442
commit
cac1c91bc2
@ -0,0 +1,268 @@
|
|||||||
|
/* eslint-disable no-await-in-loop */
|
||||||
|
module.exports.dependencies = ['axios', '@cospired/i18n-iso-languages', 'path'];
|
||||||
|
const details = () => ({
|
||||||
|
id: 'Tdarr_Plugin_henk_Keep_Native_Lang_Plus_Eng',
|
||||||
|
Stage: 'Pre-processing',
|
||||||
|
Name: 'Remove all langs except native and English',
|
||||||
|
Type: 'Audio',
|
||||||
|
Operation: 'Transcode',
|
||||||
|
Description: `This plugin will remove all language audio tracks except the 'native'
|
||||||
|
(requires TMDB api key) and English.
|
||||||
|
'Native' languages are the ones that are listed on imdb. It does an API call to
|
||||||
|
Radarr, Sonarr to check if the movie/series exists and grabs the IMDB id. As a last resort it
|
||||||
|
falls back to '[imdb-ttDIGITS] in the filename.`,
|
||||||
|
Version: '1.00',
|
||||||
|
Link: 'https://github.com/HaveAGitGat/Tdarr_Plugins/blob/master/Community/'
|
||||||
|
+ 'Tdarr_Plugin_henk_Keep_Native_Lang_Plus_Eng.js',
|
||||||
|
Tags: 'pre-processing,configurable',
|
||||||
|
|
||||||
|
Inputs: [
|
||||||
|
{
|
||||||
|
name: 'user_langs',
|
||||||
|
tooltip: 'Input a comma separated list of ISO-639-2 languages. It will still keep English and undefined tracks.'
|
||||||
|
+ '(https://en.wikipedia.org/wiki/List_of_ISO_639-2_codes 639-2 column)'
|
||||||
|
+ '\\nExample:\\n'
|
||||||
|
+ 'nld,nor',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'priority',
|
||||||
|
tooltip: 'Priority for either Radarr or Sonarr. Leaving it empty defaults to Radarr first.'
|
||||||
|
+ '\\nExample:\\n'
|
||||||
|
+ 'sonarr',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'api_key',
|
||||||
|
tooltip: 'Input your TMDB api key here. (https://www.themoviedb.org/)',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'radarr_api_key',
|
||||||
|
tooltip: 'Input your Radarr api key here.',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'radarr_url',
|
||||||
|
tooltip: 'Input your Radarr url here. (Without http://). Do include the port.'
|
||||||
|
+ '\\nExample:\\n'
|
||||||
|
+ '192.168.1.2:7878',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'sonarr_api_key',
|
||||||
|
tooltip: 'Input your Sonarr api key here.',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'sonarr_url',
|
||||||
|
tooltip: 'Input your Sonarr url here. (Without http://). Do include the port.'
|
||||||
|
+ '\\nExample:\\n'
|
||||||
|
+ '192.168.1.2:8989',
|
||||||
|
},
|
||||||
|
],
|
||||||
|
});
|
||||||
|
const response = {
|
||||||
|
processFile: false,
|
||||||
|
preset: ', -map 0 ',
|
||||||
|
container: '.',
|
||||||
|
handBrakeMode: false,
|
||||||
|
FFmpegMode: true,
|
||||||
|
reQueueAfter: false,
|
||||||
|
infoLog: '',
|
||||||
|
};
|
||||||
|
|
||||||
|
const processStreams = (result, file, user_langs) => {
|
||||||
|
// eslint-disable-next-line global-require,import/no-unresolved
|
||||||
|
const languages = require('@cospired/i18n-iso-languages');
|
||||||
|
const tracks = {
|
||||||
|
keep: [],
|
||||||
|
remove: [],
|
||||||
|
remLangs: '',
|
||||||
|
};
|
||||||
|
let streamIndex = 0;
|
||||||
|
|
||||||
|
const langsTemp = result.original_language;
|
||||||
|
let langs = [];
|
||||||
|
|
||||||
|
if (Array.isArray(langsTemp)) {
|
||||||
|
// For loop because I thought some imdb stuff returns multiple languages
|
||||||
|
// Translates 'en' to 'eng', because imdb uses a different format compared to ffmpeg
|
||||||
|
for (let i = 0; i < langsTemp.length; i += 1) {
|
||||||
|
langs.push(languages.alpha2ToAlpha3B(langsTemp));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
langs.push(languages.alpha2ToAlpha3B(langsTemp));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (user_langs) {
|
||||||
|
langs = langs.concat(user_langs);
|
||||||
|
}
|
||||||
|
if (!langs.includes('eng')) langs.push('eng');
|
||||||
|
if (!langs.includes('und')) langs.push('und');
|
||||||
|
|
||||||
|
response.infoLog += 'Keeping languages: ';
|
||||||
|
// Print languages to UI
|
||||||
|
langs.forEach((l) => {
|
||||||
|
response.infoLog += `${languages.getName(l, 'en')}, `;
|
||||||
|
});
|
||||||
|
|
||||||
|
response.infoLog = `${response.infoLog.slice(0, -2)}\n`;
|
||||||
|
|
||||||
|
for (let i = 0; i < file.ffProbeData.streams.length; i += 1) {
|
||||||
|
const stream = file.ffProbeData.streams[i];
|
||||||
|
|
||||||
|
if (stream.codec_type === 'audio') {
|
||||||
|
if (!stream.tags) {
|
||||||
|
response.infoLog += `☒No tags found on audio track ${streamIndex}. Keeping it. \n`;
|
||||||
|
tracks.keep.push(streamIndex);
|
||||||
|
streamIndex += 1;
|
||||||
|
// eslint-disable-next-line no-continue
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (langs.includes(stream.tags.language)) {
|
||||||
|
tracks.keep.push(streamIndex);
|
||||||
|
} else {
|
||||||
|
tracks.remove.push(streamIndex);
|
||||||
|
response.preset += `-map -0:a:${streamIndex} `;
|
||||||
|
tracks.remLangs += `${languages.getName(stream.tags.language, 'en')}, `;
|
||||||
|
}
|
||||||
|
streamIndex += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
response.preset += ' -c copy -max_muxing_queue_size 9999';
|
||||||
|
return tracks;
|
||||||
|
};
|
||||||
|
|
||||||
|
const tmdbApi = async (filename, api_key, axios) => {
|
||||||
|
let fileName;
|
||||||
|
// If filename begins with tt, it's already an imdb id
|
||||||
|
if (filename) {
|
||||||
|
if (filename.substr(0, 2) === 'tt') {
|
||||||
|
fileName = filename;
|
||||||
|
} else {
|
||||||
|
const idRegex = /\[imdb-(tt\d*)]/;
|
||||||
|
const fileMatch = filename.match(idRegex);
|
||||||
|
// eslint-disable-next-line prefer-destructuring
|
||||||
|
if (fileMatch) fileName = fileMatch[1];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (fileName) {
|
||||||
|
const result = await axios.get(`https://api.themoviedb.org/3/find/${fileName}?api_key=`
|
||||||
|
+ `${api_key}&language=en-US&external_source=imdb_id`)
|
||||||
|
.then((resp) => (resp.data.movie_results.length > 0 ? resp.data.movie_results[0] : resp.data.tv_results[0]));
|
||||||
|
|
||||||
|
if (!result) {
|
||||||
|
response.infoLog += '☒No IMDB result was found. \n';
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
};
|
||||||
|
|
||||||
|
// eslint-disable-next-line consistent-return
|
||||||
|
const parseArrResponse = async (body, filePath, arr) => {
|
||||||
|
// eslint-disable-next-line default-case
|
||||||
|
switch (arr) {
|
||||||
|
case 'radarr':
|
||||||
|
// filePath = file
|
||||||
|
for (let i = 0; i < body.length; i += 1) {
|
||||||
|
if (body[i].movieFile) {
|
||||||
|
if (body[i].movieFile.relativePath) {
|
||||||
|
if (body[i].movieFile.relativePath === filePath) {
|
||||||
|
return body[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 'sonarr':
|
||||||
|
// filePath = directory the file is in
|
||||||
|
// eslint-disable-next-line global-require,import/no-unresolved
|
||||||
|
const path = require('path');
|
||||||
|
for (let i = 0; i < body.length; i += 1) {
|
||||||
|
if (body[i].path) {
|
||||||
|
if (path.basename(body[i].path) === path.basename(path.dirname(filePath))) {
|
||||||
|
return body[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const plugin = async (file, librarySettings, inputs) => {
|
||||||
|
// eslint-disable-next-line global-require,import/no-unresolved
|
||||||
|
const axios = require('axios').default;
|
||||||
|
response.container = `.${file.container}`;
|
||||||
|
let prio = ['radarr', 'sonarr'];
|
||||||
|
let radarrResult = null;
|
||||||
|
let sonarrResult = null;
|
||||||
|
let tmdbResult = null;
|
||||||
|
|
||||||
|
if (inputs.priority) {
|
||||||
|
if (inputs.priority === 'sonarr') {
|
||||||
|
prio = ['sonarr', 'radarr'];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (let i = 0; i < prio.length; i += 1) {
|
||||||
|
let imdbId;
|
||||||
|
// eslint-disable-next-line default-case
|
||||||
|
switch (prio[i]) {
|
||||||
|
case 'radarr':
|
||||||
|
if (tmdbResult) break;
|
||||||
|
if (inputs.radarr_api_key) {
|
||||||
|
radarrResult = await parseArrResponse(
|
||||||
|
await axios.get(`http://${inputs.radarr_url}/api/v3/movie?apiKey=${inputs.radarr_api_key}`)
|
||||||
|
.then((resp) => resp.data),
|
||||||
|
file.meta.FileName, 'radarr',
|
||||||
|
);
|
||||||
|
|
||||||
|
if (radarrResult) {
|
||||||
|
imdbId = radarrResult.imdbId;
|
||||||
|
response.infoLog += `Grabbed ID (${imdbId}) from Radarr \n`;
|
||||||
|
} else {
|
||||||
|
response.infoLog += 'Couldn\'t grab ID from Radarr/Sonarr, grabbing it from file name \n';
|
||||||
|
imdbId = file.meta.FileName;
|
||||||
|
}
|
||||||
|
tmdbResult = await tmdbApi(imdbId, inputs.api_key, axios);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 'sonarr':
|
||||||
|
if (tmdbResult) break;
|
||||||
|
if (inputs.sonarr_api_key) {
|
||||||
|
sonarrResult = await parseArrResponse(
|
||||||
|
await axios.get(`http://${inputs.sonarr_url}/api/series?apikey=${inputs.sonarr_api_key}`)
|
||||||
|
.then((resp) => resp.data),
|
||||||
|
file.meta.Directory, 'sonarr',
|
||||||
|
);
|
||||||
|
|
||||||
|
if (sonarrResult) {
|
||||||
|
imdbId = sonarrResult.imdbId;
|
||||||
|
response.infoLog += `Grabbed ID (${imdbId}) from Sonarr \n`;
|
||||||
|
} else {
|
||||||
|
response.infoLog += 'Couldn\'t grab ID from Radarr/Sonarr, grabbing it from file name \n';
|
||||||
|
imdbId = file.meta.FileName;
|
||||||
|
}
|
||||||
|
tmdbResult = await tmdbApi(imdbId, inputs.api_key, axios);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (tmdbResult) {
|
||||||
|
const tracks = processStreams(tmdbResult, file, inputs.user_langs ? inputs.user_langs.split(',') : '');
|
||||||
|
|
||||||
|
if (tracks.remove.length > 0) {
|
||||||
|
if (tracks.keep.length > 0) {
|
||||||
|
response.infoLog += `☑Removing tracks with languages: ${tracks.remLangs.slice(0, -2)}. \n`;
|
||||||
|
response.processFile = true;
|
||||||
|
response.infoLog += '\n';
|
||||||
|
} else {
|
||||||
|
response.infoLog += '☒Cancelling plugin otherwise all audio tracks would be removed. \n';
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
response.infoLog += '☒No audio tracks to be removed. \n';
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
response.infoLog += '☒Couldn\'t find the IMDB id of this file. Skipping. \n';
|
||||||
|
}
|
||||||
|
return response;
|
||||||
|
};
|
||||||
|
|
||||||
|
module.exports.details = details;
|
||||||
|
module.exports.plugin = plugin;
|
||||||
Loading…
Reference in new issue