/*******************************************************************************
 **
 ** smolbote: yet another qute browser
 ** Copyright (C) 2017  Xian Nox
 **
 ** 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, 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 General Public License for more details.
 **
 ** You should have received a copy of the GNU General Public License
 ** along with this program.  If not, see <http://www.gnu.org/licenses/>.
 **
 ******************************************************************************/

#include "singleapplication.h"
#include <QLocalServer>
#include <QLocalSocket>
#include <QDataStream>

SingleApplication::SingleApplication(int &argc, char **argv) : QApplication(argc, argv)
{
}

SingleApplication::~SingleApplication()
{
    if(m_localServer) {
        if(m_localServer->isListening()) {
            m_localServer->close();
            QLocalServer::removeServer(LOCALSERVER_KEY);
        }
    }
}

void SingleApplication::bindLocalSocket()
{
    m_localServer = new QLocalServer(this);
    connect(m_localServer, &QLocalServer::newConnection, this, &SingleApplication::slot_receiveMessage);

    // check for running local server by connecting to it
    QLocalSocket socket;
    socket.connectToServer(LOCALSERVER_KEY);
    if(socket.waitForConnected(LOCALSERVER_TIMEOUT)) {
        // there is another server
        qWarning("Another server is running");
        socket.close();
        return;
    } else {
        // no other server
        QLocalServer::removeServer(LOCALSERVER_KEY);
        if(!m_localServer->listen(LOCALSERVER_KEY)) {
            qWarning("Cannot bind local server [%s]", qUtf8Printable(LOCALSERVER_KEY));
        }
    }

}

bool SingleApplication::isRunning()
{
    return !m_localServer->isListening();
}

bool SingleApplication::sendMessage(const QHash<QString, QVariant> &params)
{
    QLocalSocket socket;
    socket.connectToServer(LOCALSERVER_KEY);
    if(socket.waitForConnected(LOCALSERVER_TIMEOUT)) {
        QByteArray argumentData;
        QDataStream ds(&argumentData, QIODevice::WriteOnly);
        ds << params;

        socket.write(argumentData);
        socket.waitForBytesWritten(LOCALSERVER_TIMEOUT);
        return true;
    }

    return false;
}

void SingleApplication::slot_receiveMessage()
{
    QLocalSocket *socket = m_localServer->nextPendingConnection();
    if(!socket) {
        // null socket --> return
        return;
    }

    socket->waitForReadyRead();

    QByteArray argumentData = socket->readAll();

    // skip if we got no data
    if(argumentData.isEmpty()) {
        return;
    }

    QHash<QString, QVariant> params;
    QDataStream ds(argumentData);
    ds >> params;

    socket->deleteLater();

    emit messageAvailable(params);
}