File indexing completed on 2024-05-12 05:17:13

 /*
     SPDX-FileCopyrightText: 2009 Andras Mantia <amantia@kde.org>
 
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "getquotarootjob.h"
 
 #include <KLocalizedString>
 
 #include "quotajobbase_p.h"
 #include "response_p.h"
 #include "rfccodecs.h"
 #include "session_p.h"
 
 namespace KIMAP
 {
 class GetQuotaRootJobPrivate : public QuotaJobBasePrivate
 {
 public:
     GetQuotaRootJobPrivate(Session *session, const QString &name)
         : QuotaJobBasePrivate(session, name)
     {
     }
     ~GetQuotaRootJobPrivate()
     {
     }
 
     QString mailBox;
     QList<QByteArray> rootList;
     QMap<QByteArray, QMap<QByteArray, QPair<qint64, qint64>>> quotas;
 };
 }
 
 using namespace KIMAP;
 
 GetQuotaRootJob::GetQuotaRootJob(Session *session)
     : QuotaJobBase(*new GetQuotaRootJobPrivate(session, i18n("GetQuotaRoot")))
 {
 }
 
 GetQuotaRootJob::~GetQuotaRootJob()
 {
 }
 
 void GetQuotaRootJob::doStart()
 {
     Q_D(GetQuotaRootJob);
     d->tags << d->sessionInternal()->sendCommand("GETQUOTAROOT", '\"' + KIMAP::encodeImapFolderName(d->mailBox.toUtf8()) + '\"');
 }
 
 void GetQuotaRootJob::handleResponse(const Response &response)
 {
     Q_D(GetQuotaRootJob);
     if (handleErrorReplies(response) == NotHandled) {
         if (response.content.size() >= 3) {
             if (response.content[1].toString() == "QUOTAROOT") {
                 d->rootList.clear();
                 // some impls don't give the root a name which for us seems as if
                 // there were no message part
                 if (response.content.size() == 3) {
                     d->rootList.append("");
                 } else {
                     int i = 3;
                     while (i < response.content.size()) {
                         d->rootList.append(response.content[i].toString());
                         i++;
                     }
                 }
             } else if (response.content[1].toString() == "QUOTA") {
                 QByteArray rootName;
                 int quotaContentIndex = 3;
                 // some impls don't give the root a name in the response
                 if (response.content.size() == 3) {
                     quotaContentIndex = 2;
                 } else {
                     rootName = response.content[2].toString();
                 }
 
                 const QMap<QByteArray, QPair<qint64, qint64>> &quota = d->readQuota(response.content[quotaContentIndex]);
                 if (d->quotas.contains(rootName)) {
                     d->quotas[rootName].insert(quota);
                 } else {
                     d->quotas[rootName] = quota;
                 }
             }
         }
     }
 }
 
 void GetQuotaRootJob::setMailBox(const QString &mailBox)
 {
     Q_D(GetQuotaRootJob);
     d->mailBox = mailBox;
 }
 
 QString GetQuotaRootJob::mailBox() const
 {
     Q_D(const GetQuotaRootJob);
     return d->mailBox;
 }
 
 QList<QByteArray> GetQuotaRootJob::roots() const
 {
     Q_D(const GetQuotaRootJob);
     return d->rootList;
 }
 
 qint64 GetQuotaRootJob::usage(const QByteArray &root, const QByteArray &resource) const
 {
     Q_D(const GetQuotaRootJob);
     QByteArray r = resource.toUpper();
 
     if (d->quotas.contains(root) && d->quotas[root].contains(r)) {
         return d->quotas[root][r].first;
     }
     return -1;
 }
 
 qint64 GetQuotaRootJob::limit(const QByteArray &root, const QByteArray &resource) const
 {
     Q_D(const GetQuotaRootJob);
 
     QByteArray r = resource.toUpper();
 
     if (d->quotas.contains(root) && d->quotas[root].contains(r)) {
         return d->quotas[root][r].second;
     }
     return -1;
 }
 
 QMap<QByteArray, qint64> GetQuotaRootJob::allUsages(const QByteArray &root) const
 {
     Q_D(const GetQuotaRootJob);
 
     QMap<QByteArray, qint64> result;
 
     if (d->quotas.contains(root)) {
         const QMap<QByteArray, QPair<qint64, qint64>> quota = d->quotas[root];
         QMap<QByteArray, QPair<qint64, qint64>>::const_iterator it = quota.cbegin();
         const QMap<QByteArray, QPair<qint64, qint64>>::const_iterator itEnd = quota.cend();
         for (; it != itEnd; ++it) {
             result[it.key()] = it.value().first;
         }
     }
     return result;
 }
 
 QMap<QByteArray, qint64> GetQuotaRootJob::allLimits(const QByteArray &root) const
 {
     Q_D(const GetQuotaRootJob);
 
     QMap<QByteArray, qint64> result;
 
     if (d->quotas.contains(root)) {
         const QMap<QByteArray, QPair<qint64, qint64>> quota = d->quotas[root];
         QMap<QByteArray, QPair<qint64, qint64>>::const_iterator it = quota.cbegin();
         const QMap<QByteArray, QPair<qint64, qint64>>::const_iterator itEnd = quota.cend();
         for (; it != itEnd; ++it) {
             result[it.key()] = it.value().second;
         }
     }
     return result;
 }
 
 #include "moc_getquotarootjob.cpp"