ScreenTranslator/Translator.cpp

#include "Translator.h"

#include <QDebug>
#include <QNetworkRequest>
#include <QNetworkReply>
#include <QJsonDocument>
#include <QJsonArray>
#include <QJsonParseError>
#include <QSettings>

#include "Settings.h"
#include "GoogleWebTranslator.h"
#include "StAssert.h"

namespace
{
  const QString translateBaseUrl = "http://translate.google.com/translate_a/"
                                   "t?client=t&text=%1&sl=%2&tl=%3";
}

Translator::Translator(QObject *parent) :
  QObject(parent),
  network_ (this),
  useAlternativeTranslation_ (false)
{
  connect (&network_, SIGNAL (finished (QNetworkReply*)),
           SLOT (replyFinished (QNetworkReply*)));

  GoogleWebTranslator* googleWeb = new GoogleWebTranslator;
  connect (this, SIGNAL (translateAlternative (ProcessingItem)),
           googleWeb, SLOT (translate (ProcessingItem)));
  connect (googleWeb, SIGNAL (translated (ProcessingItem, bool)),
           this, SLOT (translatedAlternative(ProcessingItem, bool)));
  connect (googleWeb, SIGNAL (error (QString)), this, SIGNAL (error (QString)));

  applySettings ();
}

void Translator::applySettings()
{
  QSettings settings;
  settings.beginGroup (settings_names::translationGroup);
  translationLanguage_ = settings.value (settings_names::translationLanguage,
                                         settings_values::translationLanguage).toString ();
  sourceLanguage_ = settings.value (settings_names::sourceLanguage,
                                    settings_values::sourceLanguage).toString ();
}

void Translator::translate(ProcessingItem item)
{
  if (useAlternativeTranslation_) {
    emit translateAlternative(item);
    return;
  }
  ST_ASSERT (!item.recognized.isEmpty ());
  QString sourceLanguage = item.sourceLanguage.isEmpty () ? sourceLanguage_ :
                                                            item.sourceLanguage;
  if (translationLanguage_.isEmpty () || sourceLanguage.isEmpty ())
  {
    emit error (tr ("Неверные парметры для перевода."));
    return;
  }
  QUrl url (translateBaseUrl.arg (item.recognized, sourceLanguage, translationLanguage_));
  QNetworkReply* reply = network_.get (QNetworkRequest (url));
  items_.insert (reply, item);
}

void Translator::translatedAlternative(ProcessingItem item, bool success)
{
  if (success)
  {
    emit translated(item);
  }
  else
  {
    emit error (tr ("Ошибка альтернативного перевода текста: %1").arg (item.recognized));
  }
}

void Translator::replyFinished(QNetworkReply* reply)
{
  ST_ASSERT (items_.contains (reply));
  ProcessingItem item = items_.take (reply);
  ST_ASSERT (reply->isFinished ());
  if (reply->error () != QNetworkReply::NoError)
  {
    useAlternativeTranslation_ = true;
    emit translateAlternative(item);
    reply->deleteLater ();
    return;
  }
  QByteArray data = reply->readAll ();
  reply->deleteLater ();

  while (data.indexOf (",,") != -1)//make json valid
  {
    data.replace (",,", ",");
  }
  QJsonParseError parseError;
  QJsonDocument document = QJsonDocument::fromJson (data, &parseError);
  if (document.isEmpty ())
  {
    useAlternativeTranslation_ = true;
    emit translateAlternative(item);
    return;
  }
  QJsonArray answerArray = document.array ();
  QJsonArray fullTranslation = answerArray.first ().toArray ();
  QString translation = "";
  foreach (QJsonValue part, fullTranslation)
  {
    QJsonArray partTranslation = part.toArray ();
    if (partTranslation.isEmpty ())
    {
      continue;
    }
    translation += partTranslation.at (0).toString ();
  }
  item.translated = translation;
  emit translated (item);
}