glosbe_word_search.js

'use strict'

const request = require('request-promise');
const DictionaryResponseData = require('./dictionary_response_data.js');
const WordMeaning = require('./word_meaning.js');
const DictionaryResult = require('./dictionary_result.js');
const PublicError = require('monochrome-bot').PublicError;

const prettyLanguageForLanguageCode = require('./language_code_maps.js').prettyLanguageForGlosbeLanguageCode;
const TRANSLATE_API = 'https://glosbe.com/gapi_v0_1/translate';
const RESPONSE_FORMAT = 'json';

const threeLetterLanguageCodeForTwoLetterLanguageCode = {
  en: 'eng',
  de: 'deu',
  ja: 'jpn',
  fr: 'fra',
  es: 'spa',
  ru: 'rus',
  zh: 'zho',
  it: 'ita',
  ar: 'ara',
  pl: 'pol',
  jpn: 'jpn',
  eng: 'eng'
};

function addMeaningsForLanguage(outMeanings, inMeanings, toLanguage) {
  for (let meaning of inMeanings) {
    if (meaning.language === toLanguage) {
      outMeanings.push(new WordMeaning(meaning.text, []));
    }
  }
}

function parseGlosbeResponse(inData) {
  let inResults = inData.tuc;
  let fromLanguage = inData.from;
  let toLanguage = inData.dest;
  let phrase = inData.phrase;
  let fromLanguagePretty = prettyLanguageForLanguageCode[fromLanguage];
  let toLanguagePretty = prettyLanguageForLanguageCode[toLanguage];

  let outResults = [];

  if (inResults) {
    for (let inResult of inResults) {
      let inPhrase = inResult.phrase;

      if (inPhrase && inPhrase.language === toLanguage) {
        let resultWord = inPhrase.text;

        let inMeanings = inResult.meanings;
        let outMeanings = [];
        if (inMeanings) {
          addMeaningsForLanguage(outMeanings, inMeanings, fromLanguage);
          addMeaningsForLanguage(outMeanings, inMeanings, toLanguage);

          if (fromLanguage !== 'en' && toLanguage !== 'en') {
            addMeaningsForLanguage(outMeanings, inMeanings, 'en');
          }

          outResults.push(new DictionaryResult([{word: resultWord, readings: []}], [], outMeanings));
        }
      }
    }
  }

  return new DictionaryResponseData(phrase, fromLanguagePretty, toLanguagePretty, true, outResults, '');
}

module.exports.supportsLanguage = function(languageCode) {
  return !!threeLetterLanguageCodeForTwoLetterLanguageCode[languageCode];
}

module.exports.search = function(fromLanguage, toLanguage, suffix) {
  suffix = suffix.trim();
  let requestFromLanguage = threeLetterLanguageCodeForTwoLetterLanguageCode[fromLanguage];
  let requestToLanguage = threeLetterLanguageCodeForTwoLetterLanguageCode[toLanguage];

  return request({
    uri: TRANSLATE_API,
    qs: {
      from: requestFromLanguage,
      dest: requestToLanguage,
      format: RESPONSE_FORMAT,
      phrase: suffix,
    },
    json: true,
    timeout: 10000,
  }).catch(err => {
    throw new PublicError('Sorry, Glosbe dictionary is not responding. Please try again later.', 'Error', err);
  }).then(data => {
    if (data.result !== 'ok') {
      throw new Error('Bad response code from Glosbe: ', data.result);
    }
    return parseGlosbeResponse(data);
  });
};