File indexing completed on 2024-12-08 10:25:53

 /*
    SPDX-FileCopyrightText: 2018-2024 Laurent Montel <montel@kde.org>
 
    SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "rocketchatcache.h"
 #include "avatarmanager.h"
 #include "connection.h"
 #include "rocketchataccount.h"
 #include "ruqola_debug.h"
 #include <QDateTime>
 #include <QDir>
 #include <QSettings>
 #include <QUrlQuery>
 
 RocketChatCache::RocketChatCache(RocketChatAccount *account, QObject *parent)
     : QObject(parent)
     , mAccount(account)
     , mAvatarManager(new AvatarManager(mAccount, this))
 {
     connect(mAvatarManager, &AvatarManager::insertAvatarUrl, this, &RocketChatCache::insertAvatarUrl);
     loadAvatarCache();
 }
 
 RocketChatCache::~RocketChatCache()
 {
     QSettings settings(ManagerDataPaths::self()->accountAvatarConfigPath(mAccount->accountName()), QSettings::IniFormat);
 
     settings.beginGroup(QStringLiteral("Avatar"));
     QHash<QString, QUrl>::const_iterator i = mAvatarUrl.constBegin();
     while (i != mAvatarUrl.constEnd()) {
         if (!i.value().toString().isEmpty()) {
             settings.setValue(i.key(), i.value().toString());
         }
         ++i;
     }
     settings.endGroup();
 }
 
 void RocketChatCache::setRestApiConnection(RocketChatRestApi::Connection *restApi)
 {
     connect(restApi, &RocketChatRestApi::Connection::downloadFileDone, this, &RocketChatCache::slotDataDownloaded);
 }
 
 bool RocketChatCache::fileInCache(const QUrl &url)
 {
     const QFileInfo f(fileCachePath(url));
     return f.exists();
 }
 
 QString RocketChatCache::fileCachePath(const QUrl &url, ManagerDataPaths::PathType type)
 {
     QString cachePath = ManagerDataPaths::self()->path(type, mAccount->accountName());
     QString urlPath = url.path();
     if (type == ManagerDataPaths::PathType::PreviewUrl) {
         urlPath = urlPath.replace(QLatin1Char('/'), QLatin1Char('_'));
     }
     cachePath += QLatin1Char('/') + urlPath;
     if (url.hasQuery()) {
         const QUrlQuery query(url);
         cachePath += query.queryItemValue(QStringLiteral("etag"));
     }
     return cachePath;
 }
 
 void RocketChatCache::slotDataDownloaded(const QUrl &url, const QUrl &localFileUrl)
 {
     mFileInDownload.remove(url.path());
     Q_EMIT fileDownloaded(url.path(), localFileUrl);
 }
 
 void RocketChatCache::loadAvatarCache()
 {
     QSettings settings(ManagerDataPaths::self()->accountAvatarConfigPath(mAccount->accountName()), QSettings::IniFormat);
     settings.beginGroup(QStringLiteral("Avatar"));
     const QStringList keys = settings.childKeys();
     for (const QString &key : keys) {
         mAvatarUrl[key] = QUrl(settings.value(key).toString());
     }
     settings.endGroup();
 }
 
 void RocketChatCache::downloadFile(const QString &url, const QUrl &localFile)
 {
     QFile f(fileCachePath(QUrl(url)));
     if (f.exists()) {
         if (!f.copy(localFile.toLocalFile())) {
             qCWarning(RUQOLA_LOG) << "Impossible to copy" << f.fileName() << "to" << localFile;
         }
         // emit fileDownloaded?
     } else {
         // Not in cache. We need to download it (e.g. file attachment).
         const QUrl downloadUrl = mAccount->urlForLink(url);
         // const QUrl destUrl = storeInCache ? QUrl::fromLocalFile(fileCachePath(downloadUrl)) : localFile;
         mAccount->restApi()->downloadFile(downloadUrl, localFile, QByteArray("text/plain"));
         // this will call slotDataDownloaded
     }
 }
 
 bool RocketChatCache::attachmentIsInLocalCache(const QString &url)
 {
     const QString cachePath = fileCachePath(QUrl(url));
     if (QFileInfo::exists(cachePath)) {
         return true;
     }
     return false;
 }
 
 QUrl RocketChatCache::faviconLogoUrlFromLocalCache(const QString &url)
 {
     return urlFromLocalCache(url, false);
 }
 
 QUrl RocketChatCache::avatarUrlFromLocalCache(const QString &url)
 {
     return urlFromLocalCache(url, false);
 }
 
 QUrl RocketChatCache::urlFromLocalCache(const QString &url, bool needAuthentication, ManagerDataPaths::PathType type)
 {
     const QString cachePath = fileCachePath(QUrl(url), type);
     if (QFileInfo::exists(cachePath)) {
         // QML wants a QUrl here. The widgets code would be simpler with just a QString path.
         return QUrl::fromLocalFile(cachePath);
     } else {
         downloadFileFromServer(url, needAuthentication, type);
     }
     return {};
 }
 
 QUrl RocketChatCache::previewUrlFromLocalCache(const QString &url)
 {
     const QUrl previewUrl = urlFromLocalCache(url, false, ManagerDataPaths::PreviewUrl);
     // qDebug() << "previewUrl  " << previewUrl;
     return previewUrl;
 }
 
 QUrl RocketChatCache::attachmentUrlFromLocalCache(const QString &url)
 {
     return urlFromLocalCache(url, true);
 }
 
 void RocketChatCache::downloadAvatarFromServer(const Utils::AvatarInfo &info)
 {
     mAvatarManager->insertInDownloadQueue(info);
 }
 
 void RocketChatCache::downloadFileFromServer(const QString &filename, bool needAuthentication, ManagerDataPaths::PathType type)
 {
     if (!mFileInDownload.contains(filename)) {
         mFileInDownload.insert(filename);
         const QUrl downloadUrl = mAccount->urlForLink(filename);
         const QUrl destFileUrl = QUrl::fromLocalFile(fileCachePath(downloadUrl, type));
         mAccount->restApi()->downloadFile(mAccount->urlForLink(filename), destFileUrl, "text/plain", needAuthentication);
         // this will call slotDataDownloaded
     }
 }
 
 QString RocketChatCache::avatarUrlFromCacheOnly(const QString &userId)
 {
     const QUrl avatarUrl = mAvatarUrl.value(userId);
     if (!avatarUrl.isEmpty() && fileInCache(avatarUrl)) {
         const QString url = QUrl::fromLocalFile(fileCachePath(avatarUrl)).toString();
         qCDebug(RUQOLA_LOG) << " Use image in cache" << url << " userId " << userId << " mUserAvatarUrl.value(userId) " << mAvatarUrl.value(userId);
         return url;
     }
     return {};
 }
 
 void RocketChatCache::removeAvatar(const QString &avatarIdentifier)
 {
     const QUrl avatarUrl = mAvatarUrl.value(avatarIdentifier);
     if (avatarUrl.isEmpty()) {
         return;
     }
     QFile f(fileCachePath(avatarUrl));
     if (f.exists()) {
         if (!f.remove()) {
             qCWarning(RUQOLA_LOG) << "Impossible to remove f" << f.fileName() << " avartarUrl " << avatarUrl << " userIdentifier  " << avatarIdentifier;
         }
     }
 }
 
 void RocketChatCache::updateAvatar(const Utils::AvatarInfo &info)
 {
     const QString avatarIdentifier = info.generateAvatarIdentifier();
     // qDebug() << " updateAvatar" << info;
     removeAvatar(avatarIdentifier);
     mAvatarUrl.remove(avatarIdentifier);
     insertAvatarUrl(avatarIdentifier, QUrl());
     downloadAvatarFromServer(info);
 }
 
 QString RocketChatCache::avatarUrl(const Utils::AvatarInfo &info)
 {
     if (!info.isValid()) {
         return {};
     }
     // const QString avatarIdentifier = info.identifier;
     const QString avatarIdentifier = info.generateAvatarIdentifier();
     // qDebug() << " RocketChatCache::avatarUrl " << avatarIdentifier;
     // avoid to call this method several time.
     if (!mAvatarUrl.contains(avatarIdentifier)) {
         insertAvatarUrl(avatarIdentifier, QUrl());
         downloadAvatarFromServer(info);
         return {};
     } else {
         const QUrl valueUrl = mAvatarUrl.value(avatarIdentifier);
         // qDebug() << " valueUrl " << valueUrl;
 #if 0
         const QUrlQuery query{valueUrl};
         const QString etagValue = query.queryItemValue(QStringLiteral("etag"));
         if (info.etag.isEmpty() && !etagValue.isEmpty()) {
             insertAvatarUrl(avatarIdentifier, QUrl());
             downloadAvatarFromServer(info);
             return {};
         } else if (!info.etag.isEmpty() && etagValue.isEmpty()) {
             insertAvatarUrl(avatarIdentifier, QUrl());
             qDebug() << " redownload " << info;
             downloadAvatarFromServer(info);
             return {};
         }
 #endif
 
         if (!valueUrl.isEmpty() && fileInCache(valueUrl)) {
             const QString url = QUrl::fromLocalFile(fileCachePath(valueUrl)).toString();
             // qDebug() << " Use image in cache" << url << " userId " << userId << " mUserAvatarUrl.value(userId) "<< mUserAvatarUrl.value(userId);
             // qDebug() << "Use image in cache  " << url;
 
             return url;
         } else {
             downloadAvatarFromServer(info);
         }
         return {};
     }
 }
 
 void RocketChatCache::insertAvatarUrl(const QString &userIdentifier, const QUrl &url)
 {
     mAvatarUrl.insert(userIdentifier, url);
     if (!url.isEmpty() && !fileInCache(url)) {
         mAccount->restApi()->downloadFile(url, QUrl::fromLocalFile(fileCachePath(url)), {});
         // this will call slotDataDownloaded
     }
 }
 
 QString RocketChatCache::recordingVideoPath(const QString &accountName) const
 {
     const QString path = ManagerDataPaths::self()->path(ManagerDataPaths::Video, accountName);
     if (!QDir().mkpath(path)) {
         qCWarning(RUQOLA_LOG) << "Unable to create video folder: " << path;
         return {};
     }
     const QString filePath = path + QLatin1Char('/') + QString::number(QDateTime::currentDateTime().toMSecsSinceEpoch()) + QStringLiteral(".mp4");
     return filePath;
 }
 
 QString RocketChatCache::recordingImagePath(const QString &accountName) const
 {
     const QString path = ManagerDataPaths::self()->path(ManagerDataPaths::Picture, accountName);
     if (!QDir().mkpath(path)) {
         qCWarning(RUQOLA_LOG) << "Unable to create picture folder: " << path;
         return {};
     }
     const QString filePath = path + QLatin1Char('/') + QString::number(QDateTime::currentDateTime().toMSecsSinceEpoch()) + QStringLiteral(".jpg");
     return filePath;
 }
 
 #include "moc_rocketchatcache.cpp"