Tdarr_Plugins/FlowPluginsTs/CommunityFlowPlugins/tools/forceFileRenameRadarrOrSonarr/1.0.0/index.ts

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) => {
        args.jobLog(JSON.stringify(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,
};