gabehf
/
Tdarr_Plugins
mirror of https://github.com/gabehf/Tdarr_Plugins.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added more checks and fallbacks for name checking

Browse Source
make-only-subtitle-default
jeanchristophe.mqt@gmail.com 2 years ago
parent da5806b00f
commit 2a5b73a8eb
2 changed files with 102 additions and 52 deletions
Show Stats Download Patch File Download Diff File

88
FlowPlugins/CommunityFlowPlugins/tools/applyRadarrOrSonarrNamingPolicy/1.0.0/index.js
Unescape View File

@ -116,10 +116,10 @@ var details = function () { return ({
}); }; }); };
exports.details = details; exports.details = details;
var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function () { var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function () {
var lib, _a, arr, arr_api_key, arr_host, arrHost, filePath, fileName, getNewPath, episodeNumber, getNewPathTypes, newPathOutput; var lib, _a, arr, arr_api_key, arr_host, arrHost, filePath, fileNames, getNewPath, episodeNumber, getNewPathTypes, newPathOutput;
var _b, _c; var _b, _c, _d, _e, _f, _g;
return __generator(this, function (_d) { return __generator(this, function (_h) {
switch (_d.label) { switch (_h.label) {
case 0: case 0:
lib = require('../../../../../methods/lib')(); lib = require('../../../../../methods/lib')();
// eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign
@ -128,9 +128,12 @@ var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function
arr_host = String(args.inputs.arr_host).trim(); arr_host = String(args.inputs.arr_host).trim();
arrHost = arr_host.endsWith('/') ? arr_host.slice(0, -1) : arr_host; arrHost = arr_host.endsWith('/') ? arr_host.slice(0, -1) : arr_host;
filePath = (_c = (_b = args.originalLibraryFile) === null || _b === void 0 ? void 0 : _b._id) !== null && _c !== void 0 ? _c : ''; filePath = (_c = (_b = args.originalLibraryFile) === null || _b === void 0 ? void 0 : _b._id) !== null && _c !== void 0 ? _c : '';
fileName = (0, fileUtils_1.getFileName)(filePath); fileNames = {
originalFileName: (0, fileUtils_1.getFileName)((_e = (_d = args.originalLibraryFile) === null || _d === void 0 ? void 0 : _d._id) !== null && _e !== void 0 ? _e : ''),
currentFileName: (0, fileUtils_1.getFileName)((_g = (_f = args.inputFileObj) === null || _f === void 0 ? void 0 : _f._id) !== null && _g !== void 0 ? _g : '')
};
getNewPath = function (getNewPathType) { return __awaiter(void 0, void 0, void 0, function () { getNewPath = function (getNewPathType) { return __awaiter(void 0, void 0, void 0, function () {
var output, headers, parseRequestConfig, parseRequestResult, id, previewRenameRequestConfig, previewRenameRequestResult, fileToRename, _a; var output, headers, getParseRequestResult, fileName, parseRequestResult, previewRenameRequestConfig, previewRenameRequestResult, fileToRename, _a;
return __generator(this, function (_b) { return __generator(this, function (_b) {
switch (_b.label) { switch (_b.label) {
case 0: case 0:
@ -145,26 +148,49 @@ var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function
'X-Api-Key': arr_api_key, 'X-Api-Key': arr_api_key,
Accept: 'application/json', Accept: 'application/json',
}; };
parseRequestConfig = { getParseRequestResult = function (fileName) { return __awaiter(void 0, void 0, void 0, function () {
method: 'get', var parseRequestConfig, parseRequestResult, id;
url: "".concat(arrHost, "/api/v3/parse?title=").concat(encodeURIComponent(fileName)), return __generator(this, function (_a) {
headers: headers, switch (_a.label) {
}; case 0:
return [4 /*yield*/, args.deps.axios(parseRequestConfig)]; parseRequestConfig = {
method: 'get',
url: "".concat(arrHost, "/api/v3/parse?title=").concat(encodeURIComponent(fileName)),
headers: headers,
};
return [4 /*yield*/, args.deps.axios(parseRequestConfig)];
case 1:
parseRequestResult = _a.sent();
id = getNewPathType.delegates.getIdFromParseRequestResult(parseRequestResult);
args.jobLog(id !== '-1' ?
"Found ".concat(getNewPathType.contentName, " ").concat(id, " with a file named '").concat(fileName, "'")
: "Didn't find ".concat(getNewPathType.contentName, " with a file named '").concat(fileName, "' in ").concat(arrHost, "."));
return [2 /*return*/, { requestResult: parseRequestResult, id: id }];
}
});
}); };
fileName = fileNames.originalFileName;
return [4 /*yield*/, getParseRequestResult(fileName)];
case 1: case 1:
parseRequestResult = _b.sent(); parseRequestResult = _b.sent();
id = getNewPathType.delegates.getIdFromParseRequestResult(parseRequestResult); if (!(parseRequestResult.id == '-1' && fileNames.currentFileName !== fileNames.originalFileName)) return [3 /*break*/, 3];
if (!(id !== '-1')) return [3 /*break*/, 6]; fileName = fileNames.currentFileName;
return [4 /*yield*/, getParseRequestResult(fileName)];
case 2:
parseRequestResult = _b.sent();
_b.label = 3;
case 3:
if (!(parseRequestResult.id !== '-1')) return [3 /*break*/, 8];
previewRenameRequestConfig = { previewRenameRequestConfig = {
method: 'get', method: 'get',
url: getNewPathType.delegates.buildPreviewRenameResquestUrl(id, parseRequestResult), url: getNewPathType.delegates.buildPreviewRenameResquestUrl(parseRequestResult),
headers: headers, headers: headers,
}; };
return [4 /*yield*/, args.deps.axios(previewRenameRequestConfig)]; return [4 /*yield*/, args.deps.axios(previewRenameRequestConfig)];
case 2: case 4:
previewRenameRequestResult = _b.sent(); previewRenameRequestResult = _b.sent();
fileToRename = getNewPathType.delegates.getFileToRenameFromPreviewRenameRequestResult(previewRenameRequestResult); fileToRename = getNewPathType.delegates.getFileToRenameFromPreviewRenameRequestResult(previewRenameRequestResult);
if (!(fileToRename !== undefined)) return [3 /*break*/, 4]; if (!(fileToRename !== undefined)) return [3 /*break*/, 6];
output.newPath = "".concat((0, fileUtils_1.getFileAbosluteDir)(args.inputFileObj._id), "/").concat((0, fileUtils_1.getFileName)(fileToRename.newPath), ".").concat((0, fileUtils_1.getContainer)(fileToRename.newPath)); output.newPath = "".concat((0, fileUtils_1.getFileAbosluteDir)(args.inputFileObj._id), "/").concat((0, fileUtils_1.getFileName)(fileToRename.newPath), ".").concat((0, fileUtils_1.getContainer)(fileToRename.newPath));
_a = output; _a = output;
return [4 /*yield*/, (0, fileMoveOrCopy_1.default)({ return [4 /*yield*/, (0, fileMoveOrCopy_1.default)({
@ -173,19 +199,19 @@ var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function
destinationPath: output.newPath, destinationPath: output.newPath,
args: args, args: args,
})]; })];
case 3: case 5:
_a.isSuccessful = _b.sent(); _a.isSuccessful = _b.sent();
args.jobLog("\u2714 Renamed ".concat(getNewPathType.contentName, " ").concat(id, " : '").concat(filePath, "' => '").concat(output.newPath, "'.")); args.jobLog("\u2714 Renamed ".concat(getNewPathType.contentName, " ").concat(parseRequestResult, " : '").concat(filePath, "' => '").concat(output.newPath, "'."));
return [3 /*break*/, 5]; return [3 /*break*/, 7];
case 4: case 6:
output.isSuccessful = true; output.isSuccessful = true;
args.jobLog('✔ No rename necessary.'); args.jobLog('✔ No rename necessary.');
_b.label = 5;
case 5: return [3 /*break*/, 7];
case 6:
args.jobLog("No ".concat(getNewPathType.appName, " with a file named '").concat(fileName, "'."));
_b.label = 7; _b.label = 7;
case 7: return [2 /*return*/, output]; case 7: return [3 /*break*/, 9];
case 8:
args.jobLog("No ".concat(getNewPathType.appName, " with a file named '").concat(fileName, "'."));
_b.label = 9;
case 9: return [2 /*return*/, output];
} }
}); });
}); }; }); };
@ -196,7 +222,7 @@ var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function
contentName: 'movie', contentName: 'movie',
delegates: { delegates: {
getIdFromParseRequestResult: function (parseRequestResult) { var _a, _b, _c, _d; return String((_d = (_c = (_b = (_a = parseRequestResult.data) === null || _a === void 0 ? void 0 : _a.movie) === null || _b === void 0 ? void 0 : _b.movieFile) === null || _c === void 0 ? void 0 : _c.movieId) !== null && _d !== void 0 ? _d : -1); }, getIdFromParseRequestResult: function (parseRequestResult) { var _a, _b, _c, _d; return String((_d = (_c = (_b = (_a = parseRequestResult.data) === null || _a === void 0 ? void 0 : _a.movie) === null || _b === void 0 ? void 0 : _b.movieFile) === null || _c === void 0 ? void 0 : _c.movieId) !== null && _d !== void 0 ? _d : -1); },
buildPreviewRenameResquestUrl: function (id, parseRequestResult) { return "".concat(arrHost, "/api/v3/rename?movieId=").concat(id); }, buildPreviewRenameResquestUrl: function (parseRequestResult) { return "".concat(arrHost, "/api/v3/rename?movieId=").concat(parseRequestResult.id); },
getFileToRenameFromPreviewRenameRequestResult: function (previewRenameRequestResult) { getFileToRenameFromPreviewRenameRequestResult: function (previewRenameRequestResult) {
var _a, _b; var _a, _b;
return (((_b = (_a = previewRenameRequestResult.data) === null || _a === void 0 ? void 0 : _a.length) !== null && _b !== void 0 ? _b : 0) > 0) ? return (((_b = (_a = previewRenameRequestResult.data) === null || _a === void 0 ? void 0 : _a.length) !== null && _b !== void 0 ? _b : 0) > 0) ?
@ -210,9 +236,9 @@ var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function
contentName: 'serie', contentName: 'serie',
delegates: { delegates: {
getIdFromParseRequestResult: function (parseRequestResult) { var _a, _b, _c; return String((_c = (_b = (_a = parseRequestResult.data) === null || _a === void 0 ? void 0 : _a.series) === null || _b === void 0 ? void 0 : _b.id) !== null && _c !== void 0 ? _c : -1); }, getIdFromParseRequestResult: function (parseRequestResult) { var _a, _b, _c; return String((_c = (_b = (_a = parseRequestResult.data) === null || _a === void 0 ? void 0 : _a.series) === null || _b === void 0 ? void 0 : _b.id) !== null && _c !== void 0 ? _c : -1); },
buildPreviewRenameResquestUrl: function (id, parseRequestResult) { buildPreviewRenameResquestUrl: function (parseRequestResult) {
episodeNumber = parseRequestResult.data.parsedEpisodeInfo.episodeNumbers[0]; episodeNumber = parseRequestResult.requestResult.data.parsedEpisodeInfo.episodeNumbers[0];
return "".concat(arrHost, "/api/v3/rename?seriesId=").concat(id, "&seasonNumber=").concat(parseRequestResult.data.parsedEpisodeInfo.seasonNumber); return "".concat(arrHost, "/api/v3/rename?seriesId=").concat(parseRequestResult.id, "&seasonNumber=").concat(parseRequestResult.requestResult.data.parsedEpisodeInfo.seasonNumber);
}, },
getFileToRenameFromPreviewRenameRequestResult: function (previewRenameRequestResult) { getFileToRenameFromPreviewRenameRequestResult: function (previewRenameRequestResult) {
var _a, _b; var _a, _b;
@ -225,7 +251,7 @@ var plugin = function (args) { return __awaiter(void 0, void 0, void 0, function
}; };
return [4 /*yield*/, getNewPath(arr === 'radarr' ? getNewPathTypes.radarr : getNewPathTypes.sonarr)]; return [4 /*yield*/, getNewPath(arr === 'radarr' ? getNewPathTypes.radarr : getNewPathTypes.sonarr)];
case 1: case 1:
newPathOutput = _d.sent(); newPathOutput = _h.sent();
return [2 /*return*/, { return [2 /*return*/, {
outputFileObj: newPathOutput.isSuccessful && newPathOutput.newPath !== '' ? __assign(__assign({}, args.inputFileObj), { _id: newPathOutput.newPath }) : args.inputFileObj, outputFileObj: newPathOutput.isSuccessful && newPathOutput.newPath !== '' ? __assign(__assign({}, args.inputFileObj), { _id: newPathOutput.newPath }) : args.inputFileObj,
outputNumber: newPathOutput.isSuccessful ? 1 : 2, outputNumber: newPathOutput.isSuccessful ? 1 : 2,

66
FlowPluginsTs/CommunityFlowPlugins/tools/applyRadarrOrSonarrNamingPolicy/1.0.0/index.ts
Unescape View File

@ -70,9 +70,17 @@ const details = (): IpluginDetails => ({
] ]
}); });
interface IFileNames {
originalFileName: string,
currentFileName: string
}
interface IParseRequestResult {
requestResult: any,
id: string
}
interface IGetNewPathDelegates { interface IGetNewPathDelegates {
getIdFromParseRequestResult: (parseRequestResult: any) => string, getIdFromParseRequestResult: (parseRequestResult: any) => string,
buildPreviewRenameResquestUrl: (id: string, parseRequestResult: any) => string, buildPreviewRenameResquestUrl: (parseRequestResult: IParseRequestResult) => string,
getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult: any) => any getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult: any) => any
} }
interface IGetNewPathType { interface IGetNewPathType {
@ -98,15 +106,17 @@ const plugin = async (args: IpluginInputArgs): Promise<IpluginOutputArgs> => {
const arr_host = String(args.inputs.arr_host).trim(); const arr_host = String(args.inputs.arr_host).trim();
const arrHost = arr_host.endsWith('/') ? arr_host.slice(0, -1) : arr_host; const arrHost = arr_host.endsWith('/') ? arr_host.slice(0, -1) : arr_host;
const filePath = args.originalLibraryFile?._id ?? ''; const filePath = args.originalLibraryFile?._id ?? '';
const fileName = getFileName(filePath); const fileNames: IFileNames = {
originalFileName: getFileName(args.originalLibraryFile?._id ?? ''),
currentFileName: getFileName(args.inputFileObj?._id ?? '')
};
const getNewPath = async (getNewPathType: IGetNewPathType) const getNewPath = async (getNewPathType: IGetNewPathType)
: Promise<IGetNewPathOutput> => { : Promise<IGetNewPathOutput> => {
const output : IGetNewPathOutput = { const output: IGetNewPathOutput = {
newPath: '', newPath: '',
isSuccessful: false isSuccessful: false
} }
args.jobLog('Going to apply new name'); args.jobLog('Going to apply new name');
args.jobLog(`Renaming ${getNewPathType.appName}...`); args.jobLog(`Renaming ${getNewPathType.appName}...`);
@ -116,22 +126,36 @@ const plugin = async (args: IpluginInputArgs): Promise<IpluginOutputArgs> => {
Accept: 'application/json', Accept: 'application/json',
}; };
// Using parse endpoint to get the movie/serie's id. const getParseRequestResult = async (fileName: string)
const parseRequestConfig = { : Promise<IParseRequestResult> => {
method: 'get', // Using parse endpoint to get the movie/serie's id.
url: `${arrHost}/api/v3/parse?title=${encodeURIComponent(fileName)}`, const parseRequestConfig = {
headers, method: 'get',
}; url: `${arrHost}/api/v3/parse?title=${encodeURIComponent(fileName)}`,
const parseRequestResult = await args.deps.axios(parseRequestConfig); headers,
const id = getNewPathType.delegates.getIdFromParseRequestResult(parseRequestResult); };
const parseRequestResult = await args.deps.axios(parseRequestConfig);
const id = getNewPathType.delegates.getIdFromParseRequestResult(parseRequestResult);
args.jobLog(id !== '-1' ?
`Found ${getNewPathType.contentName} ${id} with a file named '${fileName}'`
: `Didn't find ${getNewPathType.contentName} with a file named '${fileName}' in ${arrHost}.`);
return { requestResult: parseRequestResult, id: id };
}
let fileName = fileNames.originalFileName;
let parseRequestResult = await getParseRequestResult(fileName);
// In case there has been a name change and the arr app already noticed it.
if (parseRequestResult.id == '-1' && fileNames.currentFileName !== fileNames.originalFileName) {
fileName = fileNames.currentFileName;
parseRequestResult = await getParseRequestResult(fileName);
}
// Checking that the file has been found. A file not found might be caused because Radarr/Sonarr hasn't been notified of a file rename (notify plugin missing ?) // Checking that the file has been found.
// or because Radarr/Sonarr has upgraded the movie/serie to another release before the end of the plugin stack execution. if (parseRequestResult.id !== '-1') {
if (id !== '-1') {
// Using rename endpoint to get ids of all the files that need renaming. // Using rename endpoint to get ids of all the files that need renaming.
const previewRenameRequestConfig = { const previewRenameRequestConfig = {
method: 'get', method: 'get',
url: getNewPathType.delegates.buildPreviewRenameResquestUrl(id, parseRequestResult), url: getNewPathType.delegates.buildPreviewRenameResquestUrl(parseRequestResult),
headers, headers,
}; };
const previewRenameRequestResult = await args.deps.axios(previewRenameRequestConfig); const previewRenameRequestResult = await args.deps.axios(previewRenameRequestConfig);
@ -147,7 +171,7 @@ const plugin = async (args: IpluginInputArgs): Promise<IpluginOutputArgs> => {
destinationPath: output.newPath, destinationPath: output.newPath,
args, args,
}); });
args.jobLog(`✔ Renamed ${getNewPathType.contentName} ${id} : '${filePath}' => '${output.newPath}'.`); args.jobLog(`✔ Renamed ${getNewPathType.contentName} ${parseRequestResult} : '${filePath}' => '${output.newPath}'.`);
} else { } else {
output.isSuccessful = true; output.isSuccessful = true;
args.jobLog('✔ No rename necessary.'); args.jobLog('✔ No rename necessary.');
@ -165,7 +189,7 @@ const plugin = async (args: IpluginInputArgs): Promise<IpluginOutputArgs> => {
contentName: 'movie', contentName: 'movie',
delegates: { delegates: {
getIdFromParseRequestResult: (parseRequestResult) => String(parseRequestResult.data?.movie?.movieFile?.movieId ?? -1), getIdFromParseRequestResult: (parseRequestResult) => String(parseRequestResult.data?.movie?.movieFile?.movieId ?? -1),
buildPreviewRenameResquestUrl: (id, parseRequestResult) => `${arrHost}/api/v3/rename?movieId=${id}`, buildPreviewRenameResquestUrl: (parseRequestResult) => `${arrHost}/api/v3/rename?movieId=${parseRequestResult.id}`,
getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult) => getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult) =>
((previewRenameRequestResult.data?.length ?? 0) > 0) ? ((previewRenameRequestResult.data?.length ?? 0) > 0) ?
previewRenameRequestResult.data[0] previewRenameRequestResult.data[0]
@ -177,9 +201,9 @@ const plugin = async (args: IpluginInputArgs): Promise<IpluginOutputArgs> => {
contentName: 'serie', contentName: 'serie',
delegates: { delegates: {
getIdFromParseRequestResult: (parseRequestResult) => String(parseRequestResult.data?.series?.id ?? -1), getIdFromParseRequestResult: (parseRequestResult) => String(parseRequestResult.data?.series?.id ?? -1),
buildPreviewRenameResquestUrl: (id, parseRequestResult) => { buildPreviewRenameResquestUrl: (parseRequestResult) => {
episodeNumber = parseRequestResult.data.parsedEpisodeInfo.episodeNumbers[0]; episodeNumber = parseRequestResult.requestResult.data.parsedEpisodeInfo.episodeNumbers[0];
return `${arrHost}/api/v3/rename?seriesId=${id}&seasonNumber=${parseRequestResult.data.parsedEpisodeInfo.seasonNumber}`; return `${arrHost}/api/v3/rename?seriesId=${parseRequestResult.id}&seasonNumber=${parseRequestResult.requestResult.data.parsedEpisodeInfo.seasonNumber}`;
}, },
getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult) => getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult) =>
((previewRenameRequestResult.data?.length ?? 0) > 0) ? ((previewRenameRequestResult.data?.length ?? 0) > 0) ?

Write Preview
Loading…
Cancel
Save