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

 /*
   SPDX-FileCopyrightText: 2023-2024 Laurent Montel <montel.org>
 
   SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "downloadlanguagejob.h"
 #include "extractlanguagejob.h"
 #include "libbergamot_debug.h"
 #include <KLocalizedString>
 #include <QCryptographicHash>
 #include <QFileInfo>
 #include <QNetworkReply>
 #include <QNetworkRequest>
 #include <QTemporaryFile>
 #include <TextTranslator/TranslatorEngineAccessManager>
 
 DownloadLanguageJob::DownloadLanguageJob(QObject *parent)
     : QObject{parent}
 {
 }
 
 DownloadLanguageJob::~DownloadLanguageJob()
 {
     delete mHash;
 }
 
 void DownloadLanguageJob::start()
 {
     if (!canStart()) {
         qCWarning(TRANSLATOR_LIBBERGAMOT_LOG) << "Impossible to start DownloadLanguageJob";
         deleteLater();
         return;
     }
     mDestination = new QTemporaryFile(this);
     if (!mDestination->open()) {
         Q_EMIT errorText(i18n("Cannot open file for downloading."));
         delete mDestination;
         deleteLater();
         return;
     }
 
     mHash = new QCryptographicHash(QCryptographicHash::Sha256);
 
     QNetworkRequest request(mUrl);
     // qDebug() << " mUrl " << mUrl;
     QNetworkReply *reply = TextTranslator::TranslatorEngineAccessManager::self()->networkManager()->get(request);
     connect(reply, &QNetworkReply::errorOccurred, this, [this, reply](QNetworkReply::NetworkError error) {
         if (error == QNetworkReply::ServiceUnavailableError) {
             Q_EMIT errorText(i18n("Error: Engine systems have detected suspicious traffic from your computer network. Please try your request again later."));
         } else {
             Q_EMIT errorText(i18n("Impossible to access to url: %1", mUrl.toString()));
         }
     });
 
     connect(reply, &QNetworkReply::downloadProgress, this, &DownloadLanguageJob::downloadProgress);
     connect(reply, &QNetworkReply::finished, this, [this, reply]() {
         mDestination->flush();
         mDestination->seek(0);
         reply->deleteLater();
         if (!mCheckSum.isEmpty() && mHash->result().toHex() != mCheckSum.toLatin1()) {
             // qDebug() << " mHash->result() " << mHash->result().toHex() << " mCheckSum " << mCheckSum;
             Q_EMIT errorText(i18n("CheckSum is not correct."));
             deleteLater();
             return;
         } else {
             extractLanguage();
         }
     });
     connect(reply, &QIODevice::readyRead, this, [this, reply] {
         const QByteArray buffer = reply->readAll();
         if (mDestination->write(buffer) == -1) {
             Q_EMIT errorText(i18n("Error during writing on disk: %1", mDestination->errorString()));
             reply->abort();
         }
         mHash->addData(buffer);
     });
 }
 
 bool DownloadLanguageJob::canStart() const
 {
     return !mUrl.isEmpty();
 }
 
 QUrl DownloadLanguageJob::url() const
 {
     return mUrl;
 }
 
 void DownloadLanguageJob::setUrl(const QUrl &newUrl)
 {
     mUrl = newUrl;
 }
 
 void DownloadLanguageJob::extractLanguage()
 {
     auto extraJob = new ExtractLanguageJob(this);
     extraJob->setSource(mDestination->fileName());
     connect(extraJob, &ExtractLanguageJob::errorText, this, &DownloadLanguageJob::errorText);
     connect(extraJob, &ExtractLanguageJob::finished, this, &DownloadLanguageJob::extractDone);
 
     extraJob->start();
 }
 
 QString DownloadLanguageJob::checkSum() const
 {
     return mCheckSum;
 }
 
 void DownloadLanguageJob::setCheckSum(const QString &newCheckSum)
 {
     mCheckSum = newCheckSum;
 }
 
 void DownloadLanguageJob::slotExtractDone()
 {
     Q_EMIT extractDone();
     deleteLater();
 }
 
 #include "moc_downloadlanguagejob.cpp"