/*
Copyright (C) 2012 Sebastian Herbord. All rights reserved.

This file is part of Mod Organizer.

Mod Organizer 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, either version 3 of the License, or
(at your option) any later version.

Mod Organizer 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.

You should have received a copy of the GNU General Public License
along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef NXMACCESSMANAGER_H
#define NXMACCESSMANAGER_H

#include <QNetworkAccessManager>
#include <QNetworkReply>
#include <QProgressDialog>
#include <QTimer>
#include <set>

namespace MOBase {
class IPluginGame;
}

/**
 * @brief access manager extended to handle nxm links
 **/
class NXMAccessManager : public QNetworkAccessManager {
    Q_OBJECT
public:
    explicit NXMAccessManager(QObject* parent, const QString& moVersion);

    ~NXMAccessManager();

    void setNMMVersion(const QString& nmmVersion);

    bool loggedIn() const;

    bool loginAttempted() const;
    bool loginWaiting() const;

    void login(const QString& username, const QString& password);

    void showCookies() const;

    void clearCookies();

    QString userAgent(const QString& subModule = QString()) const;

    void startLoginCheck();

    void refuseLogin();

signals:

    /**
     * @brief emitted when a nxm:// link is opened
     *
     * @param url the nxm-link
     **/
    void requestNXMDownload(const QString& url);

    /**
     * @brief emitted after a successful login or if login was not necessary
     *
     * @param necessary true if a login was necessary and succeeded, false if the user is still logged in
     **/
    void loginSuccessful(bool necessary);

    void loginFailed(const QString& message);

    void credentialsReceived(const QString& userName, bool premium);

private slots:

    void loginChecked();
    void loginFinished();
    void loginError(QNetworkReply::NetworkError errorCode);
    void loginTimeout();

protected:
    virtual QNetworkReply* createRequest(QNetworkAccessManager::Operation operation, const QNetworkRequest& request,
                                         QIODevice* device);

private:
    void pageLogin();
    // void dlLogin();

    bool hasLoginCookies() const;

    void retrieveCredentials();

private:
    static const std::set<int> s_PremiumAccountStates;

private:
    QTimer m_LoginTimeout;
    QNetworkReply* m_LoginReply;
    QProgressDialog* m_ProgressDialog{nullptr};

    QString m_MOVersion;
    QString m_NMMVersion;

    QString m_Username;
    QString m_Password;

    bool m_LoginAttempted;
    enum {
        LOGIN_NOT_CHECKED,
        LOGIN_CHECKING,
        LOGIN_NOT_VALID,
        LOGIN_ATTEMPT_FAILED,
        LOGIN_REFUSED,
        LOGIN_VALID
    } m_LoginState = LOGIN_NOT_CHECKED;
};

#endif // NXMACCESSMANAGER_H