File indexing completed on 2024-11-24 04:16:57

 /*
   SPDX-FileCopyrightText: 2022-2024 Laurent Montel <montel@kde.org>
 
   SPDX-License-Identifier: GPL-2.0-or-later
 
   based on digikam onlinetranslator code
 */
 
 #include "yandexengineplugin.h"
 
 #include <TextTranslator/TranslatorEngineAccessManager>
 
 #include <KLocalizedString>
 
 #include <QJsonArray>
 #include <QJsonDocument>
 #include <QJsonObject>
 #include <QUrlQuery>
 
 QString YandexEnginePlugin::sYandexKey;
 
 YandexEnginePlugin::YandexEnginePlugin(QObject *parent)
     : TextTranslator::TranslatorEnginePlugin(parent)
 {
 }
 
 YandexEnginePlugin::~YandexEnginePlugin() = default;
 
 void YandexEnginePlugin::translate()
 {
     if (sYandexKey.isEmpty()) {
         const QUrl url(QStringLiteral("https://translate.yandex.com"));
 
         QNetworkReply *reply = TextTranslator::TranslatorEngineAccessManager::self()->networkManager()->get(QNetworkRequest(url));
         connect(reply, &QNetworkReply::finished, this, [this, reply]() {
             parseCredentials(reply);
         });
         connect(reply, &QNetworkReply::errorOccurred, this, [this, reply](QNetworkReply::NetworkError error) {
             slotError(error);
             reply->deleteLater();
         });
     } else {
         translateText();
     }
 }
 
 void YandexEnginePlugin::parseCredentials(QNetworkReply *reply)
 {
     // Check availability of service
     const QByteArray webSiteData = reply->readAll();
     reply->deleteLater();
     if (webSiteData.isEmpty() || webSiteData.contains("<title>Oops!</title>") || webSiteData.contains("<title>302 Found</title>")) {
         Q_EMIT translateFailed(
             i18n("Error: Engine systems have detected suspicious traffic "
                  "from your computer network. Please try your request again later."));
         return;
     }
     // qCDebug(TRANSLATOR_YANDEX_LOG) << "webSiteData  " << webSiteData;
     const QByteArray sidBeginString = "SID: '";
     const int sidBeginStringPos = webSiteData.indexOf(sidBeginString);
 
     if (sidBeginStringPos == -1) {
         Q_EMIT translateFailed(i18n("Error: Unable to find Yandex SID in web version."));
         return;
     }
 
     const int sidBeginPosition = sidBeginStringPos + sidBeginString.size();
     const int sidEndPosition = webSiteData.indexOf('\'', sidBeginPosition);
 
     if (sidEndPosition == -1) {
         Q_EMIT translateFailed(i18n("Error: Unable to extract Yandex SID from web version."));
         return;
     }
 
     // Yandex show reversed parts of session ID, need to decode
 
     const QString sid = QString::fromUtf8(webSiteData.mid(sidBeginPosition, sidEndPosition - sidBeginPosition));
 
     QStringList sidParts = sid.split(QLatin1Char('.'));
 
     for (int i = 0, total = sidParts.size(); i < total; ++i) {
         std::reverse(sidParts[i].begin(), sidParts[i].end());
     }
 
     sYandexKey = sidParts.join(QLatin1Char('.'));
     translateText();
 }
 
 void YandexEnginePlugin::translateText()
 {
     if (verifyFromAndToLanguage()) {
         return;
     }
 
     clear();
     QString lang;
     if (from() == QLatin1String("auto")) {
         lang = languageCode(to());
     } else {
         lang = languageCode(from()) + QLatin1Char('-') + languageCode(to());
     }
     // qDebug() << " lang " << lang;
     // Generate API url
     QUrl url(QStringLiteral("https://translate.yandex.net/api/v1/tr.json/translate"));
     QUrlQuery query;
     query.addQueryItem(QStringLiteral("id"), sYandexKey + QLatin1String("-2-0"));
     query.addQueryItem(QStringLiteral("srv"), QStringLiteral("tr-text"));
     query.addQueryItem(QStringLiteral("text"), inputText());
     query.addQueryItem(QStringLiteral("lang"), lang);
     url.setQuery(query);
 
     // Setup request
     QNetworkRequest request;
     request.setHeader(QNetworkRequest::ContentTypeHeader, QLatin1String("application/x-www-form-urlencoded"));
     request.setUrl(url);
 
     // Make reply
     QNetworkReply *reply = TextTranslator::TranslatorEngineAccessManager::self()->networkManager()->post(request, QByteArray());
     connect(reply, &QNetworkReply::errorOccurred, this, [this, reply](QNetworkReply::NetworkError error) {
         slotError(error);
         reply->deleteLater();
     });
 
     connect(reply, &QNetworkReply::finished, this, [this, reply]() {
         reply->deleteLater();
         parseTranslation(reply);
     });
 }
 
 void YandexEnginePlugin::parseTranslation(QNetworkReply *reply)
 {
     const QJsonDocument jsonResponse = QJsonDocument::fromJson(reply->readAll());
     if (hasDebug()) {
         setJsonDebug(QString::fromUtf8(jsonResponse.toJson(QJsonDocument::Indented)));
     }
     const QJsonObject jsonData = jsonResponse.object();
     reply->deleteLater();
 
     // Parse language
     if (from() == QLatin1String("auto")) {
         QString sourceCode = jsonData.value(QStringLiteral("lang")).toString();
         sourceCode = sourceCode.left(sourceCode.indexOf(QLatin1Char('-')));
         qDebug() << " sourceCode " << sourceCode;
     }
 
     // qDebug() << "jsonData  " << jsonData;
     appendResult(jsonData.value(QStringLiteral("text")).toArray().at(0).toString());
     Q_EMIT translateDone();
 }
 
 QString YandexEnginePlugin::languageCode(const QString &langStr)
 {
     if (langStr == QLatin1String("zh")) {
         return QStringLiteral("zn");
     } else if (langStr == QLatin1String("ja")) {
         return QStringLiteral("jv");
     }
     return langStr;
 }
 
 #include "moc_yandexengineplugin.cpp"