Tdarr_Plugins/Community/Tdarr_Plugin_rename_based_on_codec_schadi.js

const loadDefaultValues = require('../methods/loadDefaultValues');
const fs = require('fs');
const path = require('path');

const details = () => {
  return {
    id: "Tdarr_Plugin_rename_based_on_codec_schadi",
    Stage: "Post-processing",
    Name: "Rename based on codec Video and Audio",
    Type: "Video_Audio",
    Operation: "Transcode",
    Description: `
    If the filename contains a codec information like h264, av1 or similar for video and AC3, AAC or trueHD \n\n
    the plugin will read the codec info from the file and rename it accordingly. \n\n
    It also takes care off addiotnal files deffined in the input Option.	\n\n`,
    Version: "1.00",
    Tags: "post-processing",
    Inputs:[
      {
        name: 'rename_audio',
        type: 'boolean',
        defaultValue: false,
        inputUI: {
          type: 'dropdown',
          options: [
            'false',
            'true',
          ],
        },
        tooltip: `Will Rename According to Audio Codec after x264 or x265.
                   \\nExample:\\n
                   true
                   \\nExample:\\n
                   false`,
      },
      {
        name: 'rename_video',
        type: 'boolean',
        defaultValue: false,
        inputUI: {
          type: 'dropdown',
          options: [
            'false',
            'true',
          ],
        },
        tooltip: `Will Rename According to Video Codec after x264 or x265.
                   \\nExample:\\n
                   true
                   \\nExample:\\n
                   false`,
      },
      {
        name: 'additional_extensions',
        type: 'text',
        defaultValue: '.nfo,.srt',
		        inputUI: {
          type: 'text',
        },
        tooltip: `Additional file extensions to rename (comma-separated).
                   \\nExample:\\n
                   .nfo,.srt`,
      },
     ],
  };
};

const plugin = (file, librarySettings, inputs, otherArguments) => {
  inputs = loadDefaultValues(inputs, details);
  const fileNameOld = file._id;
 

	var response = {
      file,
      removeFromDB: false,
      updateDB: true,
	  infoLog: "",
	  processFile: false,
    };


  const codecMap = {
    'aac': 'AAC',
    'ac3': 'AC3',
    'av1': 'AV1',
    'avc': 'h264',
    'dts': 'DTS',
    'eac3': 'EAC3',
    'flac': 'FLAC',
    'hevc': 'h265',
    'mp2': 'MP2',
    'mp3': 'MP3',
    'mpeg2': 'MPEG2',
    'truehd': 'TrueHD',
    'x264': 'h264',
    'x265': 'h265',
    'h264': 'h264',
    'h265': 'h265',
    'dts': 'DTS-X',
    'dts-hd ma': 'DTS-HD MA',
    'dts-es': 'DTS-HD ES',
    'dts-hd hra': 'DTS-HD HRA',
    'dts express ': 'DTS Express',
    'dts 96/24': 'DTS',
  };

  var firstVideoStreamCodec;
  var firstAudioStreamCodec;

  const videoCodecRegex = /(h264|h265|x264|x265|avc|hevc|mpeg2|av1)/gi;
  const audioCodecRegex = /(aac|ac3|eac3|flac|mp2|mp3|truehd|dts[-. ]hd[-. ]ma|dts[-. ]hd[-. ]es|dts[-. ]hd[-. ]hra|dts[-. ]express|dts)/gi;

  const videoStream = file.ffProbeData.streams.find((stream) => stream.codec_type === 'video');

  if (videoStream && inputs.rename_video) {
    const videoCodec = videoStream.codec_name.toLowerCase();
	firstVideoStreamCodec = videoCodec;
	
    if (videoCodec in codecMap) {
      const renamedCodec = codecMap[videoCodec];
      file._id = file._id.replace(videoCodecRegex, renamedCodec);
      file.file = file.file.replace(videoCodecRegex, renamedCodec);
    }
  }

  const audioStream = file.ffProbeData.streams.find((stream) => stream.codec_type === 'audio');

  if (audioStream && inputs.rename_audio) {
    const audioCodec = audioStream.codec_name.toLowerCase();
	firstAudioStreamCodec = audioCodec;
	
    if (audioCodec in codecMap) {
      const renamedCodec = codecMap[audioCodec];
      file._id = file._id.replace(audioCodecRegex, renamedCodec);
      file.file = file.file.replace(audioCodecRegex, renamedCodec);
    }
  }
  
let additionalFilesCount = 0; // Counter for additional files found
  
if ((audioStream && inputs.rename_audio) || (videoStream && inputs.rename_video)) {
  const filename = path.basename(fileNameOld);
  const JustName = path.parse(filename).name;
  const popJustnamen = JustName.split('.');
  popJustnamen.splice(popJustnamen.length - 5);
  const modJustname = popJustnamen.join('.');

  const fileDir = path.dirname(fileNameOld);
  const directoryPath = fileDir;

  const additionalExtensions = inputs.additional_extensions.split(',');

  let fileList = []; // Array to store the file names
  const files = fs.readdirSync(directoryPath);

  files.forEach((supportFile) => {
    fileList.push(supportFile); // Add all files to the fileList array
  });

  const extensionList = additionalExtensions.map(extension => extension.trim()); // Remove leading/trailing spaces from extensions
  const regex = new RegExp(`(${extensionList.join('|')})$`, 'i');

  files.forEach((supportFile) => {
    if (supportFile.startsWith(modJustname) && regex.test(supportFile)) {
      const renamedFileWithVideoCodec = supportFile.replace(videoCodecRegex, codecMap[firstVideoStreamCodec]);
      const renamedFileWithBothCodecs = renamedFileWithVideoCodec.replace(audioCodecRegex, codecMap[firstAudioStreamCodec]);

      fs.renameSync(`${directoryPath}/${supportFile}`, `${directoryPath}/${renamedFileWithBothCodecs}`, {
        overwrite: true,
      });

      response.infoLog += `${directoryPath}/${supportFile} renamed to ${directoryPath}/${renamedFileWithBothCodecs}\n`;
      additionalFilesCount++; // Increment the count for each additional file found
    }
  });

  //const textFilePath = path.join(directoryPath, `${modJustname}.txt`);
  //fs.writeFileSync(textFilePath, fileList.filter(file => file.startsWith(modJustname) && regex.test(file)).join('\n'), 'utf-8');
}



  if (fileNameOld !== file._id) {
    fs.renameSync(fileNameOld, file._id, {
      overwrite: true,
    });
	response.infoLog += `Renamed file to: ${file._id}\n`
	if (additionalFilesCount > 0) {
      response.infoLog += `and: ${additionalFilesCount} additional Files!\n`;
    }
    return response;
  }
	else {
	response.infoLog += `Video File not renamed!\n`	
    if (additionalFilesCount > 0) {
      response.infoLog += `But: ${additionalFilesCount} additional Files!\n`;
    }
	return response;
	}
   
};

module.exports.details = details;
module.exports.plugin = plugin;