import fileMoveOrCopy from '../../../../FlowHelpers/1.0.0/fileMoveOrCopy'; import { getContainer, getFileAbosluteDir, getFileName, } from '../../../../FlowHelpers/1.0.0/fileUtils'; import { IpluginDetails, IpluginInputArgs, IpluginOutputArgs, } from '../../../../FlowHelpers/1.0.0/interfaces/interfaces'; const details = (): IpluginDetails => ({ name: 'Apply Radarr or Sonarr naming policy', description: 'Apply Radarr or Sonarr naming policy to a file. This plugin should be called after the original file has been ' + 'replaced and Radarr or Sonarr has been notified. Radarr or Sonarr should also be notified after this plugin.', 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: 'Radarr or Sonnar notified', }, { number: 2, tooltip: 'Radarr or Sonnar do not know this file', }, ], }); interface IFileNames { originalFileName: string, currentFileName: string } interface IParseRequestResult { data: { movie?: { movieFile: { movieId: number }, }, series?: { id: number }, parsedEpisodeInfo?: { episodeNumbers: number[], seasonNumber: number }, }, } interface IParseRequestResultWrapper { parseRequestResult: IParseRequestResult, id: string } interface IFileToRename { newPath: string episodeNumbers?: number[] } interface IPreviewRenameRequestResult { data: IFileToRename[] } interface IGetNewPathDelegates { getIdFromParseRequestResult: (parseRequestResult: IParseRequestResult) => string, buildPreviewRenameResquestUrl: (parseRequestResult: IParseRequestResultWrapper) => string, getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult: IPreviewRenameRequestResult) => IFileToRename | undefined } interface IGetNewPathType { appName: string, contentName: string, delegates: IGetNewPathDelegates } interface IGetNewPathTypes { radarr: IGetNewPathType, sonarr: IGetNewPathType } interface IGetNewPathOutput { newPath: string, isSuccessful: boolean } const plugin = async (args: IpluginInputArgs): Promise => { 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 filePath = args.originalLibraryFile?._id ?? ''; const fileNames: IFileNames = { originalFileName: getFileName(args.originalLibraryFile?._id ?? ''), currentFileName: getFileName(args.inputFileObj?._id ?? ''), }; const getNewPath = async (getNewPathType: IGetNewPathType) : Promise => { const output: IGetNewPathOutput = { newPath: '', isSuccessful: false, }; args.jobLog('Going to apply new name'); args.jobLog(`Renaming ${getNewPathType.appName}...`); const headers = { 'Content-Type': 'application/json', 'X-Api-Key': arr_api_key, Accept: 'application/json', }; const getParseRequestResultWrapper = async (fileName: string) : Promise => { // 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 = 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 { parseRequestResult, id }; }; let fileName = fileNames.originalFileName; let parseRequestResultWrapper = await getParseRequestResultWrapper(fileName); // In case there has been a name change and the arr app already noticed it. if (parseRequestResultWrapper.id === '-1' && fileNames.currentFileName !== fileNames.originalFileName) { fileName = fileNames.currentFileName; parseRequestResultWrapper = await getParseRequestResultWrapper(fileName); } // Checking that the file has been found. if (parseRequestResultWrapper.id !== '-1') { // Using rename endpoint to get ids of all the files that need renaming. const previewRenameRequestConfig = { method: 'get', url: getNewPathType.delegates.buildPreviewRenameResquestUrl(parseRequestResultWrapper), headers, }; const previewRenameRequestResult = await args.deps.axios(previewRenameRequestConfig); const fileToRename = getNewPathType.delegates .getFileToRenameFromPreviewRenameRequestResult(previewRenameRequestResult); // Only if there is a rename to execute if (fileToRename !== undefined) { output.newPath = `${getFileAbosluteDir(args.inputFileObj._id) }/${getFileName(fileToRename.newPath) }.${getContainer(fileToRename.newPath)}`; output.isSuccessful = await fileMoveOrCopy({ operation: 'move', sourcePath: args.inputFileObj._id, destinationPath: output.newPath, args, }); args.jobLog(`✔ Renamed ${getNewPathType.contentName} ${parseRequestResultWrapper.id} : ` + `'${filePath}' => '${output.newPath}'.`); } else { output.isSuccessful = true; args.jobLog('✔ No rename necessary.'); } } return output; }; let episodeNumber = 0; const getNewPathTypes: IGetNewPathTypes = { radarr: { appName: 'Radarr', contentName: 'movie', delegates: { getIdFromParseRequestResult: (parseRequestResult) => String(parseRequestResult.data?.movie?.movieFile?.movieId ?? -1), buildPreviewRenameResquestUrl: (parseRequestResultWrapper) => `${arrHost}/api/v3/rename?movieId=${parseRequestResultWrapper.id}`, getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult) => (((previewRenameRequestResult.data?.length ?? 0) > 0) ? previewRenameRequestResult.data[0] : undefined), }, }, sonarr: { appName: 'Sonarr', contentName: 'serie', delegates: { getIdFromParseRequestResult: (parseRequestResult) => String(parseRequestResult.data?.series?.id ?? -1), buildPreviewRenameResquestUrl: (parseRequestResultWrapper) => { [episodeNumber] = parseRequestResultWrapper.parseRequestResult.data.parsedEpisodeInfo?.episodeNumbers ?? [1]; return `${arrHost}/api/v3/rename?seriesId=${parseRequestResultWrapper.id}&seasonNumber=` + `${parseRequestResultWrapper.parseRequestResult.data.parsedEpisodeInfo?.seasonNumber ?? 1}`; }, getFileToRenameFromPreviewRenameRequestResult: (previewRenameRequestResult) => (((previewRenameRequestResult.data?.length ?? 0) > 0) ? previewRenameRequestResult.data.find((episodeFile) => episodeFile.episodeNumbers?.at(0) === episodeNumber) : undefined), }, }, }; const newPathOutput = await getNewPath(arr === 'radarr' ? getNewPathTypes.radarr : getNewPathTypes.sonarr); return { outputFileObj: newPathOutput.isSuccessful && newPathOutput.newPath !== '' ? { ...args.inputFileObj, _id: newPathOutput.newPath } : args.inputFileObj, outputNumber: newPathOutput.isSuccessful ? 1 : 2, variables: args.variables, }; }; export { details, plugin, };