create-translated-files.js

const translateText = require('./libreTranslateWrapper');
const fs = require('fs-extra');
const projectConstants = require('./constants');
const { languagesToTranscribe, translationLanguages } = projectConstants;

const convert = require("cyrillic-to-latin");

let l = console.log;

if(global.debug === 'false'){
  l = function(){}
}

// l('translationLanguages')
// l(translationLanguages);
//
// l('languagesToTranscribe')
// l(languagesToTranscribe);

function getCodeFromLanguageName(languageName){
  return translationLanguages.find(function(filteredLanguage){
    return languageName === filteredLanguage.name;
  }).code
}

// l(getCodeFromLanguageName('English'))


// TODO: add the 'webvtt' thing and don't need these
function cleanUpSubtitles(language, text){
  // l('clean up subtitles');
  // l(language);
  if(language === 'Spanish'){
    // l('Spanish');
    // l(language);
    return text.replace(/-- título/g, '-->');

  } else if(language === 'French'){
    return text.replace(/-- Cancer/g, '-->');

  } else if(language === 'German'){
    return text.replace(/WEBVAT/g, 'WEBVTT');

  }  else if(language === 'Japanese'){
    return text.replace(/ウェブサイト/g, 'WEBVTT');
  } else {
    // Russian works out of the box

    return text
  }

  // can't get Chinese to work (seems a bit complicated)


}

/** for translation **/
async function createTranslatedFiles({
    directoryAndFileName,
    language,
    websocketConnection
}){

  const loopThrough = ['.srt', '.vtt', 'txt'];

  // TODO: translate the rest
  const srtData = await fs.readFile(`${directoryAndFileName}.vtt`, 'utf-8');
  l('srtData');
  l(srtData);

  // copy original as copied
  await fs.copy(`${directoryAndFileName}.vtt`, `${directoryAndFileName}_${language}.vtt`)

  for(const languageToConvertTo of languagesToTranscribe){
    l('languageToConvertTo');
    l(languageToConvertTo);

    l('language');
    l(language);

    try {
      // no need to translate just copy the file
      if(languageToConvertTo !== language){
        websocketConnection.send(JSON.stringify({
          languageUpdate: `Translating into ${languageToConvertTo}..`,
          message: 'languageUpdate'
        }), function () {});

        // hit LibreTranslate backend
        l(`hitting libretranslate: ${language} -> ${languageToConvertTo}`);
        // TODO: to convert to thing
        let translatedText = await translateText({
          sourceLanguage: getCodeFromLanguageName(language), // before these were like 'EN', will full language work?
          targetLanguage: getCodeFromLanguageName(languageToConvertTo),
          text: srtData,
        })
        // l('translatedText');
        // l(translatedText);
        translatedText = cleanUpSubtitles(languageToConvertTo, translatedText);

        translatedText = 'WEBVTT' + translatedText.slice(6);


        // l('translatedText');
        // l(translatedText);


        await fs.writeFile(`${directoryAndFileName}_${languageToConvertTo}.vtt`, translatedText, 'utf-8');
      }
    } catch (err){
      l(err);
      l('error in translation');
      return err
    }
  }
}

// const uploadDirectoryName = 'ef56767d5cba0ae421a9f6f570443205';
// const transcribedFileName = 'ef56767d5cba0ae421a9f6f570443205';
//
// const languageToConvertFrom = 'en';
// const languagesToConvertTo = ['es', 'fr'];

// async function main(){
//   const completed = await createTranslatedSrts({
//     uploadDirectoryName: uploadDirectoryName,
//     transcribedFileName: transcribedFileName,
//     languageToConvertFrom: languageToConvertFrom,
//     languagesToConvertTo: languagesToConvertTo
//   });
//
//   l('completed');
//   l(completed);
// }

// main();

module.exports = createTranslatedFiles;