//////////////////////////////////////////////////////////////////////////////////////
//    akashi - a server for Attorney Online 2                                       //
//    Copyright (C) 2020  scatterflower                                             //
//                                                                                  //
//    This program is free software: you can redistribute it and/or modify          //
//    it under the terms of the GNU Affero General Public License as                //
//    published by the Free Software Foundation, either version 3 of the            //
//    License, or (at your option) any later version.                               //
//                                                                                  //
//    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 Affero General Public License for more details.                           //
//                                                                                  //
//    You should have received a copy of the GNU Affero General Public License      //
//    along with this program.  If not, see <https://www.gnu.org/licenses/>.        //
//////////////////////////////////////////////////////////////////////////////////////
#include "advertiser.h"

#include "config_manager.h"

Advertiser::Advertiser()
{
    m_manager = new QNetworkAccessManager();
    connect(m_manager, &QNetworkAccessManager::finished,
            this, &Advertiser::msRequestFinished);

    m_name = ConfigManager::serverName();
    m_hostname = ConfigManager::advertiserHostname();
    m_description = ConfigManager::serverDescription();
    m_port = ConfigManager::serverPort();

    // Cheap workaround to correctly advertise when Cloudflare tunnel is used.
    if (ConfigManager::advertiserCloudflareMode()) {
        m_ws_port = 80;
    }
    else {
        m_ws_port = ConfigManager::webaoPort();
    }
    m_masterserver = ConfigManager::advertiserIP();
    m_debug = ConfigManager::advertiserDebug();
}

Advertiser::~Advertiser()
{
    m_manager->deleteLater();
}

void Advertiser::msAdvertiseServer()
{
    if (m_masterserver.isValid()) {

        QUrl url(m_masterserver);
        QNetworkRequest request(url);
        request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");

        QJsonObject l_json;

        if (!m_hostname.isEmpty()) {
            l_json["ip"] = m_hostname;
        }

        l_json["port"] = m_port;
        if (m_ws_port != -1) {
            l_json["ws_port"] = m_ws_port;
        }

        l_json["players"] = m_players;
        l_json["name"] = m_name;

        if (!m_description.isEmpty()) {
            l_json["description"] = m_description;
        }

        m_manager->post(request, QJsonDocument(l_json).toJson());

        if (m_debug)
            qDebug().noquote() << "Advertised Server";
        return;
    }
    if (m_debug)
        qWarning().noquote() << "Unable to advertise. Masterserver URL '" + m_masterserver.toString() + "' is not valid.";
    return;
}

void Advertiser::msRequestFinished(QNetworkReply *f_reply)
{
    if (m_debug) {
        if (f_reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt() == 200) {
            qDebug().noquote() << "Succesfully advertised server.";
        }
        else {
            QJsonDocument json = QJsonDocument::fromJson(f_reply->readAll());
            if (json.isNull()) {
                qCritical().noquote() << "Invalid JSON response from" << f_reply->url();
                f_reply->deleteLater();
                return;
            }

            qDebug().noquote() << "Got valid response from" << f_reply->url();
            qDebug() << json;
        }
    }
    f_reply->deleteLater();
}

void Advertiser::updatePlayerCount(int f_current_players)
{
    m_players = f_current_players;
}

void Advertiser::updateAdvertiserSettings()
{
    m_name = ConfigManager::serverName();
    m_hostname = ConfigManager::advertiserHostname();
    m_description = ConfigManager::serverDescription();
    m_masterserver = ConfigManager::advertiserIP();
    m_debug = ConfigManager::advertiserDebug();
}