Tdarr_Plugins/Community/Tdarr_Plugin_e5c3_CnT_Add_Subtitles.js

const fs = require('fs');
const execSync = require('child_process').execSync;

function details() {
    return {
      id: "Tdarr_Plugin_e5c3_CnT_Add_Subtitles",
      Name: "Add subtitles to MKV files",
      Type: "Video",
      Operation:"Remux",
      Description: `This plugin will check for subtitles, they should be named according to the ISO 639-2 language code.\nA subtitle could look like this: eng.srt\n If there are subtitles found they will be added with FFMPEG, if there are no subs of that language found.\n On first run node module iso-639-2 will be installed in the documents folder.\n Created by @control#0405`,
      Version: "1.3",
      Link: "https://github.com/HaveAGitGat/Tdarr_Plugins/blob/master/Community/Tdarr_Plugin_e5c3_CnT_Add_Subtitles.js",
      Tags:'pre-processing,ffmpeg,subtitle only,configurable',
      Inputs: [
        {
        name: 'install_packages',
        tooltip: `Please change this to "yes", it allows the plugin to install the required nodemodule. (iso-639-2) \\nExample:\\n yes`
        },
    ]
    }
}

function plugin(file, librarySettings, inputs, otherArguments) {
    //default response
    var response = {
        processFile: false, 
        preset: `,`,
        container: '.mkv',
        handBrakeMode: false,
        FFmpegMode: false,
        reQueueAfter: false,
        infoLog: `Searching new subtitles...\n`,
    }

    if (inputs.install_packages == "yes") {
        if (!fs.existsSync(`${otherArguments.homePath}/Tdarr/node_modules/iso-639-2`)) {
            execSync(`cd ${otherArguments.homePath}/Tdarr \n npm install iso-639-2`);
        } 
    } else {
        response.infoLog = `Please take a look at the input options\n A extra nodemodule is required.`
        return response;
    }

    if (fs.existsSync(`/home/Tdarr/Documents/Tdarr/node_modules/iso-639-2`)) {
        var iso6392 = require('/home/Tdarr/Documents/Tdarr/node_modules/iso-639-2');
    } else {
        response.infoLog += `Nodemodule iso-639-2 isn't installed!\nTry Again`
        return response;
    }

    var i = 0; //int for counting lang[position]
    var found_subtitle_stream = 0;
    var sub = 0; //becomes first subtitle stream
    var lang = iso6392; //languages to check against
    var path = file.meta.Directory; //path of media folder
    var exist = 0; //if the language exists should be added this becomes 1
    var new_subs = 0 //count the new subs
    var added_subs = 0; //counts the amount of subs that have been mapped
    var preset_import = '';
    var preset_meta = '';

    //find first subtitle stream
    while (found_subtitle_stream == 0 && sub < file.ffProbeData.streams.length) {
        if (file.ffProbeData.streams[sub].codec_type.toLowerCase() == "subtitle") {
            found_subtitle_stream = 1;
        } else {
        sub++;
        }
    }
    
    response.infoLog += `Path: ${path}\n`;
    for (i = 0; i < lang.length; i++) {
        //check if srt file exists in folder
        if (fs.existsSync(`${path}/${lang[i].iso6392B}.srt`)) {
            response.infoLog += `Found subtitle ${lang[i].name}.srt\n`;

            if (found_subtitle_stream == 1) {
                //check if language already exists
                for (sub_stream = sub; sub_stream < file.ffProbeData.streams.length; sub_stream++) {
                    //response.infoLog += `does ${lang[i].name} exist in stream ${sub_stream}?\n`
                    if (file.ffProbeData.streams[sub_stream].tags.language) {
                        if (file.ffProbeData.streams[sub_stream].tags.language.toLowerCase() == lang[i].iso6392B) {
                            //response.infoLog += `YES\n`
                            exist = 1;
                            response.infoLog += `Language already exists in stream ${sub_stream}\n It will not be added\n`;
                        }
                    }
                }
            } else {
                exist = 0;
            }
            
            //add if it hasn't found the language
            if (exist != 1) {
                preset_import += ` -sub_charenc "UTF-8" -f srt -i "${path}/${lang[i].iso6392B}.srt"`;
                preset_meta += ` -metadata:s:s:${new_subs} language=${lang[i].iso6392B}`;
                new_subs++;
            }
        } 
        //else {
        //    response.infoLog += `did not find sub ${lang[i].iso6392B}.srt\n`
        //}
        exist = 0;
    }

    response.preset += ` ${preset_import}${preset_meta} -map 0:v -map 0:a`
    
    //map new subs
    while (added_subs < new_subs) {
        added_subs++;
        response.preset += ` -map ${added_subs}:s`;
    }

    //if new subs have been found they will be added
    if (new_subs > 0) {
        response.FFmpegMode = true;
        response.processFile = true;
        response.reQueueAfter = true;
        if (found_subtitle_stream == 1) {
            response.preset += ` -map 0:s `;
        }
        response.preset += ` -c copy`;
        response.infoLog += `${new_subs} new subs will be added\n`;
    } else {
        response.infoLog += `No new subtitle languages were found\n`;
    }

    //response.infoLog += `The ffmpeg string is: ${response.preset}\n`

    return response
}

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