mirror of https://github.com/nymea/nymea.git
125 lines
4.6 KiB
C++
125 lines
4.6 KiB
C++
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
|
|
* *
|
|
* Copyright (C) 2015 Simon Stuerz <simon.stuerz@guh.guru> *
|
|
* *
|
|
* This file is part of guh. *
|
|
* *
|
|
* Guh 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, version 2 of the License. *
|
|
* *
|
|
* Guh 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 guh. If not, see <http://www.gnu.org/licenses/>. *
|
|
* *
|
|
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
|
|
|
|
#include "restserver.h"
|
|
#include "loggingcategories.h"
|
|
#include "httprequest.h"
|
|
#include "httpreply.h"
|
|
#include "guhcore.h"
|
|
|
|
#include <QJsonDocument>
|
|
|
|
namespace guhserver {
|
|
|
|
RestServer::RestServer(QObject *parent) :
|
|
QObject(parent)
|
|
{
|
|
m_webserver = new WebServer(this);
|
|
connect(m_webserver, &WebServer::clientConnected, this, &RestServer::clientConnected);
|
|
connect(m_webserver, &WebServer::clientDisconnected, this, &RestServer::clientDisconnected);
|
|
connect(m_webserver, &WebServer::httpRequestReady, this, &RestServer::processHttpRequest);
|
|
|
|
m_webserver->startServer();
|
|
|
|
QMetaObject::invokeMethod(this, "setup", Qt::QueuedConnection);
|
|
}
|
|
|
|
void RestServer::setup()
|
|
{
|
|
// Create resources
|
|
m_deviceResource = new DevicesResource(this);
|
|
m_deviceClassesResource = new DeviceClassesResource(this);
|
|
m_vendorsResource = new VendorsResource(this);
|
|
m_pluginsResource = new PluginsResource(this);
|
|
m_rulesResource = new RulesResource(this);
|
|
m_logsResource = new LogsResource(this);
|
|
|
|
m_resources.insert(m_deviceResource->name(), m_deviceResource);
|
|
m_resources.insert(m_deviceClassesResource->name(), m_deviceClassesResource);
|
|
m_resources.insert(m_vendorsResource->name(), m_vendorsResource);
|
|
m_resources.insert(m_pluginsResource->name(), m_pluginsResource);
|
|
m_resources.insert(m_rulesResource->name(), m_rulesResource);
|
|
m_resources.insert(m_logsResource->name(), m_logsResource);
|
|
}
|
|
|
|
void RestServer::clientConnected(const QUuid &clientId)
|
|
{
|
|
m_clientList.append(clientId);
|
|
}
|
|
|
|
void RestServer::clientDisconnected(const QUuid &clientId)
|
|
{
|
|
m_clientList.removeAll(clientId);
|
|
}
|
|
|
|
void RestServer::processHttpRequest(const QUuid &clientId, const HttpRequest &request)
|
|
{
|
|
QStringList urlTokens = request.url().path().split("/");
|
|
urlTokens.removeAll(QString());
|
|
|
|
// check token count
|
|
if (urlTokens.count() < 3) {
|
|
HttpReply *reply = RestResource::createErrorReply(HttpReply::BadRequest);
|
|
reply->setClientId(clientId);
|
|
m_webserver->sendHttpReply(reply);
|
|
reply->deleteLater();
|
|
return;
|
|
}
|
|
|
|
// check resource
|
|
QString resourceName = urlTokens.at(2);
|
|
if (!m_resources.contains(resourceName)) {
|
|
HttpReply *reply = RestResource::createErrorReply(HttpReply::BadRequest);
|
|
reply->setClientId(clientId);
|
|
m_webserver->sendHttpReply(reply);
|
|
reply->deleteLater();
|
|
}
|
|
|
|
// process request in corresponding resource
|
|
RestResource *resource = m_resources.value(resourceName);
|
|
HttpReply *reply = resource->proccessRequest(request, urlTokens);
|
|
reply->setClientId(clientId);
|
|
if (reply->type() == HttpReply::TypeAsync) {
|
|
connect(reply, &HttpReply::finished, this, &RestServer::asyncReplyFinished);
|
|
m_asyncReplies.insert(clientId, reply);
|
|
reply->startWait();
|
|
return;
|
|
}
|
|
m_webserver->sendHttpReply(reply);
|
|
reply->deleteLater();
|
|
return;
|
|
}
|
|
|
|
void RestServer::asyncReplyFinished()
|
|
{
|
|
HttpReply *reply = qobject_cast<HttpReply*>(sender());
|
|
|
|
qCDebug(dcWebServer) << "Async reply finished";
|
|
|
|
if (reply->timedOut())
|
|
reply->setHttpStatusCode(HttpReply::GatewayTimeout);
|
|
|
|
m_webserver->sendHttpReply(reply);
|
|
reply->deleteLater();
|
|
}
|
|
|
|
|
|
}
|