mirror of
https://github.com/gabehf/Tdarr_Plugins.git
synced 2026-03-17 11:16:26 -07:00
Force rename Radarr/Sonarr logic simplified to just apply the naming policy
This commit is contained in:
jeanchristophe.mqt@gmail.com
parent
d6ac4ba9fa
commit
713923cc1e
2 changed files with 45 additions and 110 deletions
|
|
@ -1,205 +0,0 @@
|
|||
import fileMoveOrCopy from '../../../../FlowHelpers/1.0.0/fileMoveOrCopy';
|
||||
import {
|
||||
getFileAbosluteDir, getFileName,
|
||||
} from '../../../../FlowHelpers/1.0.0/fileUtils';
|
||||
import {
|
||||
IpluginDetails,
|
||||
IpluginInputArgs,
|
||||
IpluginOutputArgs,
|
||||
} from '../../../../FlowHelpers/1.0.0/interfaces/interfaces';
|
||||
|
||||
const details = (): IpluginDetails => ({
|
||||
name: 'Force File Rename Radarr or Sonarr',
|
||||
description: 'Force Radarr or Sonarr to rename a file according to the naming policy',
|
||||
style: {
|
||||
borderColor: 'green',
|
||||
},
|
||||
tags: '',
|
||||
isStartPlugin: false,
|
||||
pType: '',
|
||||
requiresVersion: '2.11.01',
|
||||
sidebarPosition: -1,
|
||||
icon: 'faBell',
|
||||
inputs: [
|
||||
{
|
||||
label: 'Arr',
|
||||
name: 'arr',
|
||||
type: 'string',
|
||||
defaultValue: 'radarr',
|
||||
inputUI: {
|
||||
type: 'dropdown',
|
||||
options: ['radarr', 'sonarr'],
|
||||
},
|
||||
tooltip: 'Specify which arr to use',
|
||||
},
|
||||
{
|
||||
label: 'Arr API Key',
|
||||
name: 'arr_api_key',
|
||||
type: 'string',
|
||||
defaultValue: '',
|
||||
inputUI: {
|
||||
type: 'text',
|
||||
},
|
||||
tooltip: 'Input your arr api key here',
|
||||
},
|
||||
{
|
||||
label: 'Arr Host',
|
||||
name: 'arr_host',
|
||||
type: 'string',
|
||||
defaultValue: 'http://192.168.1.1:7878',
|
||||
inputUI: {
|
||||
type: 'text',
|
||||
},
|
||||
tooltip: 'Input your arr host here.'
|
||||
+ '\\nExample:\\n'
|
||||
+ 'http://192.168.1.1:7878\\n'
|
||||
+ 'http://192.168.1.1:8989\\n'
|
||||
+ 'https://radarr.domain.com\\n'
|
||||
+ 'https://sonarr.domain.com\\n',
|
||||
},
|
||||
],
|
||||
outputs: [
|
||||
{
|
||||
number: 1,
|
||||
tooltip: 'Continue to next plugin',
|
||||
},
|
||||
],
|
||||
});
|
||||
|
||||
const plugin = async (args: IpluginInputArgs): Promise<IpluginOutputArgs> => {
|
||||
const lib = require('../../../../../methods/lib')();
|
||||
// eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign
|
||||
args.inputs = lib.loadDefaultValues(args.inputs, details);
|
||||
|
||||
const { arr, arr_api_key } = args.inputs;
|
||||
const arr_host = String(args.inputs.arr_host).trim();
|
||||
const arrHost = arr_host.endsWith('/') ? arr_host.slice(0, -1) : arr_host;
|
||||
const fileName = getFileName(args.inputFileObj._id);
|
||||
|
||||
interface IRenameDelegates {
|
||||
getId: (parseRequestResult: any) => any,
|
||||
getPreviewRenameResquestUrl: (id: any, parseRequestResult: any) => any,
|
||||
getFileToRename: (previewRenameRequestResult: any) => any,
|
||||
getRenameResquestConfigData: (id: any, fileToRename: any) => any
|
||||
}
|
||||
|
||||
const rename = async (delegates: IRenameDelegates)
|
||||
: Promise<{ existingPath: any, newPath: any }> => {
|
||||
args.jobLog('Going to force rename');
|
||||
args.jobLog(`Renaming ${arr === 'radarr' ? 'Radarr' : 'Sonarr'}...`);
|
||||
|
||||
let existingPath = '', newPath = '';
|
||||
|
||||
const headers = {
|
||||
'Content-Type': 'application/json',
|
||||
'X-Api-Key': arr_api_key,
|
||||
Accept: 'application/json',
|
||||
};
|
||||
|
||||
// Using parse endpoint to get the movie/serie's id.
|
||||
const parseRequestConfig = {
|
||||
method: 'get',
|
||||
url: `${arrHost}/api/v3/parse?title=${encodeURIComponent(fileName)}`,
|
||||
headers,
|
||||
};
|
||||
const parseRequestResult = await args.deps.axios(parseRequestConfig);
|
||||
const id = delegates.getId(parseRequestResult);
|
||||
|
||||
// Using rename endpoint to get ids of all the files that need renaming.
|
||||
const previewRenameRequestConfig = {
|
||||
method: 'get',
|
||||
url: delegates.getPreviewRenameResquestUrl(id, parseRequestResult),
|
||||
headers,
|
||||
};
|
||||
const previewRenameRequestResult = await args.deps.axios(previewRenameRequestConfig);
|
||||
const fileToRename = delegates.getFileToRename(previewRenameRequestResult);
|
||||
|
||||
// Only if there is a rename to execute
|
||||
if (fileToRename !== undefined) {
|
||||
({ existingPath, newPath } = fileToRename);
|
||||
|
||||
// Using command endpoint to schedule the renames.
|
||||
const renameRequestConfig = {
|
||||
method: 'post',
|
||||
url: `${arrHost}/api/v3/command`,
|
||||
headers,
|
||||
data: JSON.stringify(delegates.getRenameResquestConfigData(id, fileToRename))
|
||||
};
|
||||
await args.deps.axios(renameRequestConfig);
|
||||
|
||||
args.jobLog(`✔ Renamed ${arr === 'radarr' ? 'movie' : 'serie'} ${id} in ${arr === 'radarr' ? 'Radarr' : 'Sonarr'} : '${existingPath}' => '${newPath}'.`);
|
||||
} else
|
||||
args.jobLog('✔ No rename necessary.');
|
||||
|
||||
return { existingPath, newPath };
|
||||
};
|
||||
|
||||
let existingPath, newPath = '';
|
||||
if (arr === 'radarr') {
|
||||
({ existingPath, newPath } = await rename({
|
||||
getId: (parseRequestResult) => parseRequestResult.data.movie.movieFile.movieId,
|
||||
getPreviewRenameResquestUrl: (id, parseRequestResult) => `${arrHost}/api/v3/rename?movieId=${id}`,
|
||||
getFileToRename: (previewRenameRequestResult) =>
|
||||
((previewRenameRequestResult.data?.length ?? 0) > 0) ?
|
||||
previewRenameRequestResult.data[0]
|
||||
: undefined,
|
||||
getRenameResquestConfigData: (id, fileToRename) => {
|
||||
return {
|
||||
name: 'RenameFiles',
|
||||
movieId: id,
|
||||
files: [fileToRename.movieFileId]
|
||||
};
|
||||
}
|
||||
}));
|
||||
} else if (arr === 'sonarr') {
|
||||
let episodeNumber = 0;
|
||||
({ existingPath, newPath } = await rename({
|
||||
getId: (parseRequestResult) => parseRequestResult.data.series.id,
|
||||
getPreviewRenameResquestUrl: (id, parseRequestResult) => {
|
||||
episodeNumber = parseRequestResult.data.parsedEpisodeInfo.episodeNumbers[0];
|
||||
return `${arrHost}/api/v3/rename?seriesId=${id}&seasonNumber=${parseRequestResult.data.parsedEpisodeInfo.seasonNumber}`;
|
||||
},
|
||||
getFileToRename: (previewRenameRequestResult) => {
|
||||
return ((previewRenameRequestResult.data?.length ?? 0) > 0) ?
|
||||
previewRenameRequestResult.data.find((episFile: { episodeNumbers: number[]; }) => ((episFile.episodeNumbers?.length ?? 0) > 0) ? episFile.episodeNumbers[0] === episodeNumber : false)
|
||||
: undefined
|
||||
},
|
||||
getRenameResquestConfigData: (id, fileToRename) => {
|
||||
return {
|
||||
name: 'RenameFiles',
|
||||
seriesId: id,
|
||||
files: [fileToRename.episodeFileId]
|
||||
};
|
||||
}
|
||||
}));
|
||||
} else {
|
||||
args.jobLog('No arr specified in plugin inputs.');
|
||||
}
|
||||
|
||||
// If file has been renamed, move the transcoded file accordingly
|
||||
let outputFileObj: {_id : string} = args.inputFileObj;
|
||||
if (existingPath !== newPath) {
|
||||
const destinationPath = `${getFileAbosluteDir(args.inputFileObj._id)}/${getFileName(newPath)}`;
|
||||
|
||||
await fileMoveOrCopy({
|
||||
operation: 'move',
|
||||
sourcePath: args.inputFileObj._id,
|
||||
destinationPath: destinationPath,
|
||||
args,
|
||||
});
|
||||
args.jobLog(`✔ File moved : '${args.inputFileObj._id}' => '${destinationPath}'.`);
|
||||
|
||||
outputFileObj = { _id: destinationPath };
|
||||
}
|
||||
|
||||
return {
|
||||
outputFileObj: outputFileObj,
|
||||
outputNumber: 1,
|
||||
variables: args.variables,
|
||||
};
|
||||
};
|
||||
|
||||
export {
|
||||
details,
|
||||
plugin,
|
||||
};
|
||||
Loading…
Add table
Add a link
Reference in a new issue