Plugin for File Renaming

Community/Tdarr_Plugin_rename_based_on_codec_schadi.js

@@ -0,0 +1,204 @@
+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;