/****************************************************************************
**
** Copyright (C) 2013 Digia Plc and/or its subsidiary(-ies).
** Contact: http://www.qt-project.org/legal
**
** This file is part of the QtAddOn.JsonStream module of the Qt.
**
** $QT_BEGIN_LICENSE:LGPL$
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and Digia.  For licensing terms and
** conditions see http://qt.digia.com/licensing.  For further information
** use the contact form at http://qt.digia.com/contact-us.
**
** GNU Lesser General Public License Usage
** Alternatively, this file may be used under the terms of the GNU Lesser
** General Public License version 2.1 as published by the Free Software
** Foundation and appearing in the file LICENSE.LGPL included in the
** packaging of this file.  Please review the following information to
** ensure the GNU Lesser General Public License version 2.1 requirements
** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
** In addition, as a special exception, Digia gives you certain additional
** rights.  These rights are described in the Digia Qt LGPL Exception
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3.0 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.  Please review the following information to
** ensure the GNU General Public License version 3.0 requirements will be
** met: http://www.gnu.org/copyleft/gpl.html.
**
**
** $QT_END_LICENSE$
**
****************************************************************************/

#include <stdio.h>
#include <errno.h>
#include <signal.h>
#include <sys/socket.h>

#include "qjsonstream.h"
#include "qjsonserver.h"
#include "qjsonauthority.h"
#include "qjsonserverclient.h"

#include "qjsonschemavalidator.h"

#include <QtCore>
#include <QtNetwork>

QT_BEGIN_NAMESPACE_JSONSTREAM


/*!
  \internal
*/
inline bool canValidate(QJsonServer::ValidatorFlags flags, QJsonSchemaValidator *validator)
{
    return flags != QJsonServer::NoValidation && validator && !validator->isEmpty();
}

class QJsonServerPrivate
{
public:
    QJsonServerPrivate()
        : m_inboundValidator(0)
        , m_outboundValidator(0) {}

    ~QJsonServerPrivate()
    {
        qDeleteAll(m_identifierToClient);
        foreach (QLocalServer *server, m_localServers.keys())
            delete server;

        if (m_inboundValidator)
            delete m_inboundValidator;

        if (m_outboundValidator)
            delete m_outboundValidator;
    }

    QMap<QLocalServer *, QJsonAuthority *>  m_localServers;
    QMultiMap<QString, QJsonServerClient *> m_identifierToClient;
    QMap<QString, QList<QJsonObject> >     m_messageQueues;
    QSet<QString>                          m_multipleConnections;
    QJsonServer::ValidatorFlags             m_validatorFlags;
    QJsonSchemaValidator                       *m_inboundValidator;
    QJsonSchemaValidator                       *m_outboundValidator;
};

/**************************************************************************************************/

/*!
    \class QJsonServer
    \inmodule QtJsonStream
    \brief The QJsonServer class is the server endpoint for JSON communications.

    JSON communications is a interprocess communication (IPC) method between clients and a server.
    It consists of the following components:

    \list
       \li QJsonClient, a client-side class which sends and receives JSON messages
       \li QJsonServer, the server class which sends and receives JSON messages
       \li QJsonAuthority, an abstract authorization class which
          approves connections from clients and assigns client names.
    \endlist

    In most cases, authorization of the client is required.  This
    authorization step determines whether or not the client is
    approved for communication and associates an "identifier" string
    with the client for disambiguating received messages.
    Authorization can occur implicitly by looking at the properties of
    the connected socket (for example, authorize all connections from
    a particular TCP address) or explicitly by receiving a JSON
    message with appropriate authorization data.

    Once a client is authorized, a \l connectionAdded() signal will be
    fired.  When an authorized client disconnects, the \l
    connectionRemoved() signal fires.  No QJsonServer signals are sent
    until a client is authorized.

    After authorization, each message received will result in a \l messageReceived() signal.
    The QJsonServer sends messages to clients with the \l send() slot.

    In some cases, it is desireable to queue up messages for a client
    that has not yet connected.  Calling \c enableQueueing(identifier) will
    enable queueing of messages for that identifier.  Each client must
    be enabled separately; there is no general "queue for everyone" setting.
*/

/*!
  \property QJsonServer::validatorFlags
  The current ValidatorFlags set on this server
*/

/*!
  Constructs a new QJsonServer instance with \a parent.
*/
QJsonServer::QJsonServer(QObject *parent)
    : QObject(parent)
    , d_ptr(new QJsonServerPrivate())
{
    initSchemaValidation(); // initialize validation if defined by environment
}

/*!
    Destroy this QJsonServer.  All connections will be immediately dropped, with no more signals.
    We destroy all of the QJsonServerClient objects and remove the QLocalServer objects.
    We do not destroy the QJsonAuthority objects.
*/
QJsonServer::~QJsonServer()
{
}

/*!
  Configure the QJsonServer to listen on a new TCP socket and \a port using QJsonAuthority file \a authority.
  If the \a authority is omitted or NULL, all connections will be automatically authorized and
  a unique identifier generated for each new connection.  Returns true if the socket could be opened.
 */
bool QJsonServer::listen( int port, QJsonAuthority *authority )
{
    Q_UNUSED(port);
    Q_UNUSED(authority);
    return false;
}

/*!
  Configure the QJsonServer to listen on a new Unix local socket \a socketname using QJsonAuthority file \a authority.
  If the \a authority is omitted or NULL, all connections will be automatically authorized and
  a unique identifier generated for each new connection. Returns true if the socket could be opened.

  Does \b{not} take ownership of the \a authority object.
 */
bool QJsonServer::listen(const QString &socketname, QJsonAuthority *authority)
{
    QLocalServer::removeServer(socketname);
    QLocalServer *server = new QLocalServer(this);
    Q_D(QJsonServer);
    d->m_localServers.insert(server, authority);
    QObject::connect(server, SIGNAL(newConnection()), this, SLOT(handleLocalConnection()));
    if (!server->listen(socketname)) {
        qCritical() << Q_FUNC_INFO << "Unable to listen on socket:" << socketname;
        d->m_localServers.remove(server);
        return false;
    }
    return true;
}

/*!
    \internal
    Called with an incoming connection
*/
void QJsonServer::handleLocalConnection()
{
    QLocalServer *server = qobject_cast<QLocalServer *>(sender());
    if (!server)
        return;

    if (QLocalSocket *socket = server->nextPendingConnection()) {
        socket->setReadBufferSize(64*1024);
        Q_D(QJsonServer);
        QJsonAuthority *authority = d->m_localServers.value(server);
        QJsonServerClient *client = new QJsonServerClient(this);
        client->setAuthority(authority);
        client->setSocket(socket);
        connect(client, SIGNAL(authorized(const QString&)),
                this, SLOT(handleClientAuthorized(const QString&)));
        connect(client, SIGNAL(disconnected(const QString&)),
                this, SLOT(clientDisconnected(const QString&)));
        connect(client, SIGNAL(messageReceived(const QString&, const QJsonObject&)),
                this, SLOT(receiveMessage(const QString&, const QJsonObject&)));
        connect(client, SIGNAL(authorizationFailed()),
                this, SLOT(handleAuthorizationFailed()));
        client->start();
    }
}

/*!
  Received when a new client has been authorized by a \c QJsonAuthority.
  The \a identifier must be unique.

  Be sure to call this function if you override it in a subclass.  This function
  will emit \c connectionAdded with the \a identifier of the client.
  t
 */
void QJsonServer::handleClientAuthorized(const QString &identifier)
{
    QJsonServerClient *client = qobject_cast<QJsonServerClient *>(sender());
    Q_D(QJsonServer);
    bool exists = d->m_identifierToClient.contains(identifier);

    if (exists && !d->m_multipleConnections.contains(identifier)) {
        qWarning() << "Error: Multiple disallowed connections for" << identifier;
        client->stop();
    }
    else {
        d->m_identifierToClient.insert(identifier, client);
        QList<QJsonObject> messageQueue = d->m_messageQueues.take(identifier);
        foreach (const QJsonObject& message, messageQueue)
            client->send(message);
        if (!exists)
            emit connectionAdded(identifier);
    }
}

/*!
  Called when an identified client has disconnected.
  This function will emit \c connectionRemoved with the \a identifier of
  the client.
 */

void QJsonServer::clientDisconnected(const QString &identifier) {
    QJsonServerClient *client = qobject_cast<QJsonServerClient *>(sender());
    if (!client)
        return;

    // Only emit the connectionRemoved signal if this was a valid connection
    Q_D(QJsonServer);
    if (!d->m_identifierToClient.remove(identifier, client))
        qWarning() << "Error: Mismatched client for" << identifier;
    else
        emit connectionRemoved(identifier);

    client->deleteLater();
}

/*!
    Called when authorization fails for a client.
*/
void QJsonServer::handleAuthorizationFailed()
{
    emit authorizationFailed();
}

/*!
    Process received \a message originating from application identified by \a identifier.

    The messageReceived signal is emitted when this method is called.
*/
void QJsonServer::receiveMessage(const QString &identifier, const QJsonObject &message)
{
    // do JSON schema validation if required
    Q_D(QJsonServer);
    if (canValidate(validatorFlags(), d->m_inboundValidator)) {
        if (!d->m_inboundValidator->validateSchema(message))
        {
            if (validatorFlags().testFlag(WarnIfInvalid)) {
                emit inboundMessageValidationFailed(message, d->m_inboundValidator->getLastError());
            }
            if (validatorFlags().testFlag(DropIfInvalid)) {
                return;
            }
        }
    }

    emit messageReceived(identifier, message);
}

/*!
    Returns whether there is a connected client with \a identifier.
*/
bool QJsonServer::hasConnection(const QString &identifier) const
{
    Q_D(const QJsonServer);
    return d->m_identifierToClient.contains(identifier);
}

/*!
  Returns a list of identifiers for all current connections.
*/

QStringList QJsonServer::connections() const
{
    Q_D(const QJsonServer);
    return d->m_identifierToClient.uniqueKeys();
}

/*!
    Enables queuing of messages to client identified by \a identifier.

    The queue will be flushed to the client when it connects to the server,
    and all subsequent messages will be sent directly to the client.
*/
void QJsonServer::enableQueuing(const QString &identifier)
{
    if (identifier.isEmpty())
        return;

    Q_D(QJsonServer);
    if (!d->m_messageQueues.contains(identifier))
        d->m_messageQueues.insert(identifier, QList<QJsonObject>());
}

/*!
    Disables queuing of messages for client identified by \a identifier.
*/
void QJsonServer::disableQueuing(const QString &identifier)
{
    if (identifier.isEmpty())
        return;

    Q_D(QJsonServer);
    d->m_messageQueues.remove(identifier);
}

/*!
    Returns whether queuing of messages is enabled for client identified by \a identifier.
*/
bool QJsonServer::isQueuingEnabled(const QString &identifier) const
{
    Q_D(const QJsonServer);
    return d->m_messageQueues.contains(identifier);
}

/*!
    Clears the message queue for client \a identifier.

    Does not disable or enable message queuing for the client.
*/
void QJsonServer::clearQueue(const QString &identifier)
{
    if (identifier.isEmpty())
        return;

    Q_D(QJsonServer);
    if (d->m_messageQueues.contains(identifier))
        d->m_messageQueues.insert(identifier, QList<QJsonObject>());
}

/*!
  Enable multiple clients that match \a identifier.
  Any outgoing messages will be sent to each connection with this
  \a identifier.
*/

void QJsonServer::enableMultipleConnections(const QString& identifier)
{
    Q_D(QJsonServer);
    d->m_multipleConnections.insert(identifier);
}

/*!
  Disable multiple clients that match \a identifier.
*/

void QJsonServer::disableMultipleConnections(const QString& identifier)
{
    Q_D(QJsonServer);
    d->m_multipleConnections.remove(identifier);
}



/*!
    Sends an arbitrary \a message to the client application identified by \a identifier and returns true when
    the client was found and the message could be sent. The return value does \b{not} indicate that
    the client actually received the message and reacted to it, only that the QJsonServer had a connection
    open to that client, and was able to dispatch the message.

*/
bool QJsonServer::send(const QString &identifier, const QJsonObject &message)
{
    // do JSON schema validation if required
    Q_D(QJsonServer);
    if (canValidate(validatorFlags(), d->m_outboundValidator)) {
        if (!d->m_outboundValidator->validateSchema(message))
        {
            if (validatorFlags().testFlag(WarnIfInvalid)) {
                emit outboundMessageValidationFailed(message, d->m_outboundValidator->getLastError());
            }
            if (validatorFlags().testFlag(DropIfInvalid)) {
                return false;
            }
        }
    }

    if (isQueuingEnabled(identifier)) {
        QList<QJsonObject> queue = d->m_messageQueues.value(identifier);
        queue << message;
        d->m_messageQueues.insert(identifier, queue);
        return true;
    }

    QList<QJsonServerClient*> clients = d->m_identifierToClient.values(identifier);
    foreach (QJsonServerClient *client, clients) {
        Q_ASSERT(client);
        client->send(message);
    }
    return (clients.size() > 0);
}

/*!
  Broadcast \a message to all connected clients.
*/

void QJsonServer::broadcast(const QJsonObject &message)
{
    // do JSON schema validation if required
    Q_D(QJsonServer);
    if (canValidate(validatorFlags(), d->m_outboundValidator)) {
        if (!d->m_outboundValidator->validateSchema(message))
        {
            if (validatorFlags().testFlag(WarnIfInvalid)) {
                emit outboundMessageValidationFailed(message, d->m_outboundValidator->getLastError());
            }
            if (validatorFlags().testFlag(DropIfInvalid)) {
                return;
            }
        }
    }

    // ### No QJsonServerClient should be repeated in this list
    QList<QJsonServerClient*> clients = d->m_identifierToClient.values();
    foreach (QJsonServerClient *client, clients) {
        Q_ASSERT(client);
        client->send(message);
    }
}

/*!
    Removes all connections created by the created by the application identified by \a identifier.
    If no connections are found, this method does nothing.
    Note that calling this function does not change the QJsonAuthority permissions, so it is
    possible that a new client will connect using the same authorization.

    Calling this function will result in the \l connectionRemoved() signal being raised.
*/
void QJsonServer::removeConnection(const QString &identifier)
{
    Q_D(QJsonServer);
    QList<QJsonServerClient*> clients = d->m_identifierToClient.values(identifier);
    while (!clients.isEmpty()) {
        QJsonServerClient *client = clients.takeFirst();
        Q_ASSERT(client);
        client->stop();
    }
}

/*!
     \enum QJsonServer::ValidatorFlag
     This enum determines what the validators do with packets.  The value is
     specified by combining values from the following list using the bitwise
     OR operator:

     \value DropIfInvalid
         Validate and drop invalid messages.
     \value WarnIfInvalid
         Validate and warn about invalid messages.
     \value ApplyDefaultValues
         If a value is missing then use a default attribute's value fron JSON schema.

     \omitvalue NoValidation
*/

/*!
  Return the current ValidatorFlags
*/
QJsonServer::ValidatorFlags QJsonServer::validatorFlags() const
{
    Q_D(const QJsonServer);
    return d->m_validatorFlags;
}

/*!
  Sets validatorFlags property to \a flags.
*/
void QJsonServer::setValidatorFlags(ValidatorFlags flags)
{
    Q_D(QJsonServer);
    d->m_validatorFlags = flags;
}

/*!
  Returns an inbound JSON schema validator object.
*/
QJsonSchemaValidator *QJsonServer::inboundValidator()
{
    Q_D(QJsonServer);
    if (!d->m_inboundValidator) {
        d->m_inboundValidator = new QJsonSchemaValidator(this);
    }
    return d->m_inboundValidator;
}

/*!
  Returns an outbound JSON schema validator object.
*/
QJsonSchemaValidator *QJsonServer::outboundValidator()
{
    Q_D(QJsonServer);
    if (!d->m_outboundValidator) {
        d->m_outboundValidator = new QJsonSchemaValidator(this);
    }
    return d->m_outboundValidator;
}

/*!
  \internal
  Initialize validation if defined by environment
*/
void QJsonServer::initSchemaValidation()
{
    QString szInboundPath = QString::fromLocal8Bit(qgetenv("JSONSERVER_SCHEMA_INBOUND_PATH"));
    QString szOutboundPath = QString::fromLocal8Bit(qgetenv("JSONSERVER_SCHEMA_OUTBOUND_PATH"));
    QString strSchemaControl = QString::fromLocal8Bit(qgetenv("JSONSERVER_SCHEMA_CONTROL")); // "warn","drop" or "warn":"drop"

    if (!strSchemaControl.isEmpty())
    {
        ValidatorFlags flags(NoValidation);
        QStringList lst = strSchemaControl.toLower().split(QRegExp(QStringLiteral("[,:]")),
                                                           QString::SkipEmptyParts);
        foreach (QString str, lst) {
            if (str == QLatin1String("warn"))
                flags |= WarnIfInvalid;
            else if (str == QLatin1String("drop"))
                flags |= DropIfInvalid;
        }
        setValidatorFlags(flags);
    }

    if (!szInboundPath.isEmpty()) {
        QFileInfo fi(szInboundPath);
        if (fi.exists() && fi.isDir()) {
            inboundValidator()->loadFromFolder(szInboundPath);
        }
    }

    if (!szOutboundPath.isEmpty()) {
        QFileInfo fi(szOutboundPath);
        if (fi.exists() && fi.isDir()) {
            outboundValidator()->loadFromFolder(szOutboundPath);
        }
    }
}

/*!
    \fn void QJsonServer::connectionAdded(const QString &identifier)
    This signal is emitted when a connection identified by the \a identifier has connected and
    been authorized.
*/

/*!
    \fn void QJsonServer::connectionRemoved(const QString &identifier)
    This signal is emitted when a connection identified by \a identifier disconnects.

    Disconnection can happen if the connection is lost, or when the client process is stopped.
*/

/*!
    \fn void QJsonServer::messageReceived(const QString &identifier, const QJsonObject &message)
    This signal is emitted when a message is received from a connected client identified by the application identifier \a identifier.
    \a message contains the JSON message received.
*/

/*!
    \fn void QJsonServer::authorizationFailed()
    This signal is emitted when a client fails to authorize.
*/

/*!
    \fn void QJsonServer::inboundMessageValidationFailed(const QJsonObject &message, const QtAddOn::QtJsonStream::QJsonSchemaError &error)
    This signal is emitted when an inbound \a message message fails a JSON schema validation.  The validation
    error is reported in \a error.
*/

/*!
    \fn void QJsonServer::outboundMessageValidationFailed(const QJsonObject &message, const QtAddOn::QtJsonStream::QJsonSchemaError &error)
    This signal is emitted when an outbound \a message message fails a JSON schema validation.  The validation
    error is reported in \a error.
*/

#include "moc_qjsonserver.cpp"

QT_END_NAMESPACE_JSONSTREAM