pilotclient/src/blackcore/db/databaseauthentication.cpp

/* Copyright (C) 2015
 * swift project Community / Contributors
 *
 * This file is part of swift project. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution and at http://www.swift-project.org/license.html. No part of swift project,
 * including this file, may be copied, modified, propagated, or distributed except according to the terms
 * contained in the LICENSE file.
 */

#include "blackcore/application.h"
#include "blackcore/data/globalsetup.h"
#include "blackcore/db/databaseauthentication.h"
#include "blackmisc/json.h"
#include "blackmisc/logcategory.h"
#include "blackmisc/logcategorylist.h"
#include "blackmisc/logmessage.h"
#include "blackmisc/network/authenticateduser.h"
#include "blackmisc/network/networkutils.h"
#include "blackmisc/network/rolelist.h"
#include "blackmisc/network/url.h"
#include "blackmisc/statusmessage.h"

#include <QByteArray>
#include <QJsonObject>
#include <QNetworkReply>
#include <QNetworkRequest>
#include <QScopedPointer>
#include <QScopedPointerDeleteLater>
#include <QString>
#include <QUrl>
#include <QUrlQuery>

using namespace BlackMisc;
using namespace BlackMisc::Network;

namespace BlackCore
{
    namespace Db
    {
        CDatabaseAuthenticationService::CDatabaseAuthenticationService(QObject *parent) :
            QObject(parent)
        {
            // void
        }

        void CDatabaseAuthenticationService::gracefulShutdown()
        {
            if (this->m_shutdown) { return; }
            this->m_shutdown = true;
            this->logoff();
        }

        CAuthenticatedUser CDatabaseAuthenticationService::getDbUser() const
        {
            return this->m_swiftDbUser.get();
        }

        bool CDatabaseAuthenticationService::isUserAuthenticated() const
        {
            const CAuthenticatedUser user(this->getDbUser());
            return user.isAuthenticated();
        }

        CStatusMessageList CDatabaseAuthenticationService::login(const QString &username, const QString &password)
        {
            CStatusMessageList msgs;
            static const CLogCategoryList cats(CLogCategoryList(this).join({ CLogCategory::validation()}));

            if (this->m_shutdown) { msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "Shutdown in progress")); return msgs; }

            const QString un(username.trimmed());
            const QString pw(password.trimmed());
            if (un.isEmpty()) { msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "No user name/id")); }
            if (pw.isEmpty()) { msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "No password")); }
            if (!msgs.isEmpty()) { return msgs; }

            const CUrl url(sApp->getGlobalSetup().getDbLoginServiceUrl());
            QString msg;
            if (!CNetworkUtils::canConnect(url, msg))
            {
                msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, msg));
                return msgs;
            }

            QUrlQuery params;
            params.addQueryItem("username", un);
            params.addQueryItem("password", pw);
            if (sApp->getGlobalSetup().dbDebugFlag()) { CNetworkUtils::addDebugFlag(params); }

            QString query = params.toString();
            const QNetworkRequest request(CNetworkUtils::getSwiftNetworkRequest(url, CNetworkUtils::PostUrlEncoded, sApp->getApplicationNameAndVersion()));
            sApp->postToNetwork(request, CApplication::NoLogRequestId, query.toUtf8(), { this, &CDatabaseAuthenticationService::parseServerResponse});
            static const QString rm("Sent request to authentication server '%1'");
            msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityInfo, rm.arg(url.toQString())));
            return msgs;
        }

        void CDatabaseAuthenticationService::logoff()
        {
            CUrl url(sApp->getGlobalSetup().getDbLoginServiceUrl());
            url.setQuery("logoff=true");
            QNetworkRequest request(CNetworkUtils::getSwiftNetworkRequest(url));
            sApp->getFromNetwork(request, { this, &CDatabaseAuthenticationService::parseServerResponse });
            this->m_swiftDbUser.set(CAuthenticatedUser());
        }

        void CDatabaseAuthenticationService::parseServerResponse(QNetworkReply *nwReplyPtr)
        {
            // always cleanup reply
            QScopedPointer<QNetworkReply, QScopedPointerDeleteLater> nwReply(nwReplyPtr);

            if (this->m_shutdown) { return; }
            QString urlString(nwReply->url().toString());
            if (urlString.contains("logoff", Qt::CaseInsensitive))
            {
                sApp->deleteAllCookies();
                emit logoffFinished();
                return;
            }

            if (nwReply->error() == QNetworkReply::NoError)
            {
                const QString json(nwReply->readAll().trimmed());
                if (json.isEmpty())
                {
                    CLogMessage(this).error("Authentication failed, no response from %1") << urlString;
                    return;
                }
                if (!Json::looksLikeJson(json))
                {
                    CLogMessage(this).error("Illegal JSON object: %1") << CNetworkUtils::removeHtmlPartsFromPhpErrorMessage(json);
                    return;
                }

                static const CLogCategoryList cats(CLogCategoryList(this).join({ CLogCategory::validation()}));
                QJsonObject jsonObj(Json::jsonObjectFromString(json));
                CAuthenticatedUser user = CAuthenticatedUser::fromDatabaseJson(jsonObj.contains("user") ? jsonObj["user"].toObject() : jsonObj);
                CStatusMessageList msgs;
                if (jsonObj.contains("messages"))
                {
                    msgs = CStatusMessageList::fromDatabaseJson(jsonObj["messages"].toArray());
                    msgs.setCategories(cats);
                }

                if (!user.isAuthenticated() || !user.isValid())
                {
                    if (!msgs.hasErrorMessages())
                    {
                        msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "Cannot login, user or password wrong"));
                    }
                }
                else
                {
                    if (!user.isEnabled())
                    {
                        msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "User is disabled"));
                    }
                    if (user.getRoles().isEmpty())
                    {
                        msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "User has no roles"));
                    }
                }
                this->m_swiftDbUser.set(user);
                emit userAuthenticationFinished(user, msgs);
            }
            else
            {
                CLogMessage(this).error("Authentication failed, %1") << nwReply->errorString();
                return;
            }
        }

        void CDatabaseAuthenticationService::userChanged()
        {
            // this->logoff();
        }
    } // ns
} // ns