nextcloud-desktop/src/mirall/quotainfo.cpp

/*
 * Copyright (C) by Daniel Molkentin <danimo@owncloud.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 * for more details.
 */

#include "mirall/quotainfo.h"
#include "mirall/account.h"
#include "mirall/networkjobs.h"
#include "creds/abstractcredentials.h"

#include <QTimer>
#include <QDebug>

namespace Mirall {

namespace {
static const int defaultIntervalT = 30*1000;
static const int failIntervalT = 5*1000;
static const int initialTimeT = 1*1000;
}

QuotaInfo::QuotaInfo(QObject *parent)
    : QObject(parent)
    , _account(AccountManager::instance()->account())
    , _lastQuotaTotalBytes(0)
    , _lastQuotaUsedBytes(0)
    , _refreshTimer(new QTimer(this))
{
    connect(AccountManager::instance(), SIGNAL(accountChanged(Account*,Account*)),
            SLOT(slotAccountChanged(Account*,Account*)));
    connect(_refreshTimer, SIGNAL(timeout()), SLOT(slotCheckQuota()));
    _refreshTimer->setSingleShot(true);
    _refreshTimer->start(initialTimeT);
}

void QuotaInfo::slotAccountChanged(Account *newAccount, Account *oldAccount)
{
    _account = newAccount;
    disconnect(oldAccount, SIGNAL(stateChanged(int)), this, SLOT(slotAccountStateChanged(int)));
    connect(newAccount, SIGNAL(stateChanged(int)), this, SLOT(slotAccountStateChanged(int)));
}

void QuotaInfo::slotAccountStateChanged(int state)
{
    if (state == Account::Connected) {
        slotCheckQuota();
    } else {
        _refreshTimer->stop();
    }
}

void QuotaInfo::slotRequestFailed()
{
    _refreshTimer->start(failIntervalT);
}

void QuotaInfo::slotCheckQuota()
{
    if (!_account.isNull() && _account->credentials() && _account->credentials()->ready()
            && _account->state() == Account::Connected) {
        CheckQuotaJob *job = new CheckQuotaJob(_account, "/", this);
        connect(job, SIGNAL(quotaRetrieved(qint64,qint64)), SLOT(slotUpdateLastQuota(qint64,qint64)));
        connect(job, SIGNAL(networkError(QNetworkReply*)), SLOT(slotRequestFailed()));
        job->start();
    } else {
        _lastQuotaTotalBytes = 0;
        _lastQuotaUsedBytes = 0;
        _refreshTimer->start(failIntervalT);
    }
}

void QuotaInfo::slotUpdateLastQuota(qint64 total, qint64 used)
{
    _lastQuotaTotalBytes = total;
    _lastQuotaUsedBytes = used;
    emit quotaUpdated(total, used);
    _refreshTimer->start(defaultIntervalT);
}

}